fix: assistant object

Author: shamspias

assistant_manager.py

@@ -0,0 +1,59 @@
+from typing import Optional
+
+
+class AssistantManager:
+    def __init__(self, client):
+        self.client = client
+
+    async def list_assistants(self):
+        response = await self.client.beta.assistants.list()
+        return {assistant.name: assistant.id for assistant in response.data}
+
+    async def retrieve_assistant(self, assistant_id: str):
+        return await self.client.beta.assistants.retrieve(assistant_id)
+
+    async def create_assistant(self, name: str, instructions: str, tools: list, model: str):
+        return await self.client.beta.assistants.create(
+            name=name,
+            instructions=instructions,
+            tools=tools,
+            model=model
+        )
+
+    async def update_assistant(self, assistant_id: str, name: Optional[str] = None, description: Optional[str] = None,
+                               instructions: Optional[str] = None, tools: Optional[list] = None):
+        update_fields = {}
+        if name is not None:
+            update_fields['name'] = name
+        if description is not None:
+            update_fields['description'] = description
+        if instructions is not None:
+            update_fields['instructions'] = instructions
+        if tools is not None:
+            update_fields['tools'] = tools
+        return await self.client.beta.assistants.update(assistant_id, **update_fields)
+
+    async def delete_assistant(self, assistant_id: str):
+        return await self.client.beta.assistants.delete(assistant_id)
+
+    async def create_assistant_file(self, assistant_id: str, file_id: str):
+        return await self.client.beta.assistants.files.create(assistant_id=assistant_id, file_id=file_id)
+
+    async def delete_assistant_file(self, assistant_id: str, file_id: str):
+        return await self.client.beta.assistants.files.delete(assistant_id, file_id)
+
+    async def list_assistant_files(self, assistant_id: str):
+        return await self.client.beta.assistants.files.list(assistant_id)
+
+    async def get_assistant_id_by_name(self, name: str):
+        """
+        Get the ID of an assistant by its name.
+
+        Args:
+            name (str): The name of the assistant.
+
+        Returns:
+            str: The ID of the assistant if found, otherwise None.
+        """
+        assistants = await self.list_assistants()
+        return assistants.get(name)