Skip to content

Commit

Permalink
Fix test
Browse files Browse the repository at this point in the history
  • Loading branch information
VRSEN committed Aug 15, 2024
1 parent 4e3ddde commit 74d7dac
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test_tool_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -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")
Expand Down

0 comments on commit 74d7dac

Please sign in to comment.