From 74d7dac75c8d23b6d31d170bdb96d591ac8b78e2 Mon Sep 17 00:00:00 2001 From: Arsenii Shatokhin Date: Thu, 15 Aug 2024 10:35:45 +0400 Subject: [PATCH] Fix test --- tests/test_tool_factory.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_tool_factory.py b/tests/test_tool_factory.py index 7cf11f74..5b25ba84 100644 --- a/tests/test_tool_factory.py +++ b/tests/test_tool_factory.py @@ -9,7 +9,7 @@ from pydantic import Field, BaseModel sys.path.insert(0, '../agency-swarm') -from agency_swarm.tools import ToolFactory +from agency_swarm.tools import ToolFactory, BaseTool from agency_swarm.util.schema import dereference_schema, reference_schema from langchain.tools import MoveFileTool, YouTubeSearchTool @@ -54,7 +54,7 @@ class RelationshipType(Enum): FRIEND = "friend" COLLEAGUE = "colleague" - class UserRelationships(BaseModel): + class UserRelationships(BaseTool): """Hey this is a test?""" users: List[UserDetail] = Field(..., description="Collection of users, correctly capturing the relationships among them.", title="Users")