mirascope.core.cohere.tool

from mirascope.core import prompt_template
from mirascope.core.cohere import cohere_call


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


@cohere_call("command-r-plus", tools=[format_book])
def recommend_book(genre: str) -> str:
    return f"Recommend a {genre} book"


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

from mirascope.core.cohere import CohereTool


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


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