Skip to content

mirascope.core.groq.tool

The GroqTool class for easy tool usage with Groq LLM calls.

Usage Documentation

Tools

GroqTool

Bases: BaseTool[ChatCompletionToolParam]

A class for defining tools for Groq LLM calls.

Example:

from mirascope.core import prompt_template
from mirascope.core.groq import groq_call


def format_book(title: str, author: str) -> str:
    return f"{title} by {author}"


@groq_call("llama-3.1-8b-instant", tools=[format_book])
@prompt_template("Recommend a {genre} book")
def recommend_book(genre: str):
    ...


response = recommend_book("fantasy")
if tool := response.tool:  # returns an `GroqTool` instance
    print(tool.call())

tool_schema classmethod

tool_schema() -> ChatCompletionToolParam

Constructs a JSON Schema tool schema from the BaseModel schema defined.

Example:

from mirascope.core.groq import GroqTool


def format_book(title: str, author: str) -> str:
    return f"{title} by {author}"


tool_type = GroqTool.type_from_fn(format_book)
print(tool_type.tool_schema())  # prints the Groq-specific tool schema

Source code in mirascope/core/groq/tool.py
@classmethod
def tool_schema(cls) -> ChatCompletionToolParam:
    """Constructs a JSON Schema tool schema from the `BaseModel` schema defined.

    Example:
    ```python
    from mirascope.core.groq import GroqTool


    def format_book(title: str, author: str) -> str:
        return f"{title} by {author}"


    tool_type = GroqTool.type_from_fn(format_book)
    print(tool_type.tool_schema())  # prints the Groq-specific tool schema
    ```
    """
    fn = FunctionDefinition(name=cls._name(), description=cls._description())
    model_schema = cls.model_json_schema()
    if model_schema["properties"]:
        fn["parameters"] = model_schema
    return ChatCompletionToolParam(function=fn, type="function")

from_tool_call classmethod

from_tool_call(
    tool_call: ChatCompletionMessageToolCall,
) -> GroqTool

Constructs an GroqTool instance from a tool_call.

Parameters:

Name Type Description Default
tool_call ChatCompletionMessageToolCall

The Groq tool call from which to construct this tool instance.

required
Source code in mirascope/core/groq/tool.py
@classmethod
def from_tool_call(cls, tool_call: ChatCompletionMessageToolCall) -> GroqTool:
    """Constructs an `GroqTool` instance from a `tool_call`.

    Args:
        tool_call: The Groq tool call from which to construct this tool instance.
    """
    model_json = jiter.from_json(tool_call.function.arguments.encode())
    model_json["tool_call"] = tool_call.model_dump()
    return cls.model_validate(model_json)