fix: 🏷️ tool_resources use type

AI21Labs · Nov 28, 2024 · 5de2d8c · 5de2d8c
1 parent 90f44f5
commit 5de2d8c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ai21/models/responses/assistant_response.py b/ai21/models/responses/assistant_response.py
@@ -28,7 +28,7 @@ class AssistantResponse(AI21BaseModel):
     is_archived: bool = False
     models: Optional[List[str]] = None
     tools: Optional[List[str]] = None
-    tool_resources: Optional[dict] = None
+    tool_resources: Optional[ToolResources] = None
 
 
 class ListAssistantResponse(AI21BaseModel):