hotfix: response type

Portkey-AI · Oct 9, 2024 · c31b029 · c31b029
1 parent 4c57bc5
commit c31b029
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 17 deletions.
diff --git a/portkey_ai/api_resources/types/thread_message_type.py b/portkey_ai/api_resources/types/thread_message_type.py
@@ -75,7 +75,7 @@ class ThreadMessage(BaseModel, extra="allow"):
     assistant_id: Optional[str] = None
     content: Optional[List[Content]]
     created_at: Optional[int]
-    file_ids: Optional[List[str]]
+    file_ids: Optional[List[str]] = None
     metadata: Optional[object] = None
     object: Optional[str]
     role: Optional[str]
@@ -107,21 +107,6 @@ def get_headers(self) -> Optional[Dict[str, str]]:
         return parse_headers(self._headers)
 
 
-# class MessageFile(BaseModel, extra="allow"):
-#     id: Optional[str]
-#     object: Optional[str]
-#     created_at: Optional[int]
-#     message_id: Optional[str]
-#     _headers: Optional[httpx.Headers] = PrivateAttr()
-
-#     def __str__(self):
-#         del self._headers
-#         return json.dumps(self.dict(), indent=4)
-
-#     def get_headers(self) -> Optional[Dict[str, str]]:
-#         return parse_headers(self._headers)
-
-
 class ThreadMessageDeleted(BaseModel):
     id: str
     deleted: bool

diff --git a/portkey_ai/api_resources/types/thread_run_type.py b/portkey_ai/api_resources/types/thread_run_type.py
@@ -171,7 +171,7 @@ class Run(BaseModel):
     created_at: Optional[int]
     expires_at: Optional[int]
     failed_at: Optional[int] = None
-    file_ids: Optional[List[str]]
+    file_ids: Optional[List[str]] = None
     instructions: Optional[str]
     last_error: Optional[LastError] = None
     metadata: Optional[object] = None