fix: 🐛 use List for typing instead of list

AI21Labs · Nov 28, 2024 · fc45f19 · fc45f19
1 parent 16ef903
commit fc45f19
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ai21/models/responses/assistant_response.py b/ai21/models/responses/assistant_response.py
@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Optional
+from typing import Optional, List
 
 from ai21.models.ai21_base_model import AI21BaseModel
 
@@ -16,6 +16,6 @@ class AssistantResponse(AI21BaseModel):
     user_id: str
     avatar: Optional[str] = None
     is_archived: bool = False
-    models: Optional[list[str]] = None
-    tools: Optional[list[str]] = None
+    models: Optional[List[str]] = None
+    tools: Optional[List[str]] = None
     tool_resources: Optional[dict] = None