mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 16:48:30 +00:00
47 lines
2.4 KiB
Python
47 lines
2.4 KiB
Python
import json
|
|
import pydantic_core
|
|
import pytest
|
|
from crewai_tools import BaseTool, tool
|
|
|
|
def test_creating_a_tool_using_annotation():
|
|
@tool("Name of my tool")
|
|
def my_tool(question: str) -> str:
|
|
"""Clear description for what this tool is useful for, you agent will need this information to use it."""
|
|
return question
|
|
|
|
# Assert all the right attributes were defined
|
|
assert my_tool.name == "Name of my tool"
|
|
assert my_tool.description == "Clear description for what this tool is useful for, you agent will need this information to use it."
|
|
assert my_tool.args_schema.schema()["properties"] == {'question': {'title': 'Question', 'type': 'string'}}
|
|
assert my_tool.func("What is the meaning of life?") == "What is the meaning of life?"
|
|
|
|
# Assert the langchain tool conversion worked as expected
|
|
converted_tool = my_tool.to_langchain()
|
|
assert converted_tool.name == "Name of my tool"
|
|
assert converted_tool.description == "Clear description for what this tool is useful for, you agent will need this information to use it."
|
|
assert converted_tool.args_schema.schema()["properties"] == {'question': {'title': 'Question', 'type': 'string'}}
|
|
assert converted_tool.func("What is the meaning of life?") == "What is the meaning of life?"
|
|
|
|
def test_creating_a_tool_using_baseclass():
|
|
class MyCustomTool(BaseTool):
|
|
name: str = "Name of my tool"
|
|
description: str = "Clear description for what this tool is useful for, you agent will need this information to use it."
|
|
|
|
def _run(self, question: str) -> str:
|
|
return question
|
|
|
|
my_tool = MyCustomTool()
|
|
# Assert all the right attributes were defined
|
|
assert my_tool.name == "Name of my tool"
|
|
assert my_tool.description == "Clear description for what this tool is useful for, you agent will need this information to use it."
|
|
assert my_tool.args_schema.schema()["properties"] == {'question': {'title': 'Question', 'type': 'string'}}
|
|
assert my_tool.run("What is the meaning of life?") == "What is the meaning of life?"
|
|
|
|
# Assert the langchain tool conversion worked as expected
|
|
converted_tool = my_tool.to_langchain()
|
|
assert converted_tool.name == "Name of my tool"
|
|
assert converted_tool.description == "Clear description for what this tool is useful for, you agent will need this information to use it."
|
|
assert converted_tool.args_schema.schema()["properties"] == {'question': {'title': 'Question', 'type': 'string'}}
|
|
assert converted_tool.run("What is the meaning of life?") == "What is the meaning of life?"
|
|
|