update assistant type

phidatahq · Jan 9, 2024 · 2d9da05 · 2d9da05
1 parent 2163687
commit 2d9da05
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 8 deletions.
diff --git a/phi/agent/python.py b/phi/agent/python.py
@@ -10,9 +10,6 @@
 
 
 class PythonAgent(Agent):
-    name: str = "python_agent"
-    description: str = "achieve tasks using Python code"
-
     files: Optional[List[File]] = None
     file_information: Optional[str] = None
 

diff --git a/phi/assistant/assistant.py b/phi/assistant/assistant.py
@@ -11,7 +11,7 @@
 
 try:
     from openai import OpenAI
-    from openai.types.beta.assistant import Assistant as OpenAIAssistant
+    from openai.types.beta.assistant import Assistant as OpenAIAssistantType
     from openai.types.beta.assistant_deleted import AssistantDeleted as OpenAIAssistantDeleted
 except ImportError:
     logger.error("`openai` not installed")
@@ -76,7 +76,7 @@ class Assistant(BaseModel):
     # Enable monitoring on phidata.com
     monitoring: bool = False
 
-    openai_assistant: Optional[OpenAIAssistant] = None
+    openai_assistant: Optional[OpenAIAssistantType] = None
 
     model_config = ConfigDict(arbitrary_types_allowed=True)
 
@@ -112,7 +112,7 @@ def __enter__(self):
     def __exit__(self, exc_type, exc_value, traceback):
         self.delete()
 
-    def load_from_openai(self, openai_assistant: OpenAIAssistant):
+    def load_from_openai(self, openai_assistant: OpenAIAssistantType):
         self.id = openai_assistant.id
         self.object = openai_assistant.object
         self.created_at = openai_assistant.created_at
@@ -169,7 +169,7 @@ def create(self) -> "Assistant":
     def get_id(self) -> Optional[str]:
         return self.id or self.openai_assistant.id if self.openai_assistant else None
 
-    def get_from_openai(self) -> OpenAIAssistant:
+    def get_from_openai(self) -> OpenAIAssistantType:
         _assistant_id = self.get_id()
         if _assistant_id is None:
             raise AssistantIdNotSet("Assistant.id not set")

diff --git a/phi/conversation/conversation.py b/phi/conversation/conversation.py
@@ -449,7 +449,6 @@ def _run(self, message: Optional[Union[List[Dict], str]] = None, stream: bool =
             "tasks": all_tasks_meta_data,
             "messages": [m.model_dump(exclude_none=True) for m in all_tasks_messages if m is not None],
             "references": [r.model_dump(exclude_none=True) for r in all_tasks_references if r is not None],
-            "markdown": self.markdown,
         }
         event_data = {
             "user_message": message,