Skip to content

Commit

Permalink
Removed instructor from test
Browse files Browse the repository at this point in the history
  • Loading branch information
VRSEN committed Aug 15, 2024
1 parent 6d93928 commit 035a43c
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions tests/test_tool_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,7 @@
import unittest
from typing import List, Optional

from instructor import OpenAISchema
from pydantic import Field
from pydantic import Field, BaseModel

sys.path.insert(0, '../agency-swarm')
from agency_swarm.tools import ToolFactory
Expand All @@ -34,15 +33,15 @@ def test_move_file_tool(self):
tool.run()

def test_complex_schema(self):
class FriendDetail(OpenAISchema):
class FriendDetail(BaseModel):
"test 123"
id: int = Field(..., description="Unique identifier for each friend.")
name: str = Field(..., description="Name of the friend.")
age: Optional[int] = Field(25, description="Age of the friend.")
email: Optional[str] = Field(None, description="Email address of the friend.")
is_active: Optional[bool] = Field(None, description="Indicates if the friend is currently active.")

class UserDetail(OpenAISchema):
class UserDetail(BaseModel):
"""Hey this is a test?"""
id: int = Field(..., description="Unique identifier for each user.")
age: int
Expand All @@ -55,7 +54,7 @@ class RelationshipType(Enum):
FRIEND = "friend"
COLLEAGUE = "colleague"

class UserRelationships(OpenAISchema):
class UserRelationships(BaseModel):
"""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 035a43c

Please sign in to comment.