fix: change tool to a dict

Author: bennyai21

ai21/clients/common/beta/assistant/assistants.py

@@ -24,7 +24,7 @@ def create(
         description: str | NotGiven = NOT_GIVEN,
         optimization: Optimization | NotGiven = NOT_GIVEN,
         models: List[str] | NotGiven = NOT_GIVEN,
-        tools: List[Tool] | NotGiven = NOT_GIVEN,
+        tools: List[dict[str, Tool]] | NotGiven = NOT_GIVEN,
         tool_resources: ToolResources | NotGiven = NOT_GIVEN,
         **kwargs,
     ) -> AssistantResponse:
@@ -71,7 +71,7 @@ def modify(
         optimization: Optimization | NotGiven = NOT_GIVEN,
         is_archived: bool | NotGiven = NOT_GIVEN,
         models: List[str] | NotGiven = NOT_GIVEN,
-        tools: List[Tool] | NotGiven = NOT_GIVEN,
+        tools: List[dict[str, Tool]] | NotGiven = NOT_GIVEN,
         tool_resources: ToolResources | NotGiven = NOT_GIVEN,
     ) -> AssistantResponse:
         pass

ai21/clients/studio/resources/beta/assistant/assistant.py

@@ -30,7 +30,7 @@ def create(
         description: str | NotGiven = NOT_GIVEN,
         optimization: str | NotGiven = NOT_GIVEN,
         models: List[str] | NotGiven = NOT_GIVEN,
-        tools: List[Tool] | NotGiven = NOT_GIVEN,
+        tools: List[dict[str, Tool]] | NotGiven = NOT_GIVEN,
         tool_resources: ToolResources | NotGiven = NOT_GIVEN,
         **kwargs,
     ) -> AssistantResponse:
@@ -61,7 +61,7 @@ def modify(
         optimization: str | NotGiven = NOT_GIVEN,
         is_archived: bool | NotGiven = NOT_GIVEN,
         models: List[str] | NotGiven = NOT_GIVEN,
-        tools: List[Tool] | NotGiven = NOT_GIVEN,
+        tools: List[dict[str, Tool]] | NotGiven = NOT_GIVEN,
         tool_resources: ToolResources | NotGiven = NOT_GIVEN,
     ) -> AssistantResponse:
         body = self._create_body(
@@ -94,7 +94,7 @@ async def create(
         description: str | NotGiven = NOT_GIVEN,
         optimization: str | NotGiven = NOT_GIVEN,
         models: List[str] | NotGiven = NOT_GIVEN,
-        tools: List[Tool] | NotGiven = NOT_GIVEN,
+        tools: List[dict[str, Tool]] | NotGiven = NOT_GIVEN,
         tool_resources: ToolResources | NotGiven = NOT_GIVEN,
         **kwargs,
     ) -> AssistantResponse:
@@ -125,7 +125,7 @@ async def modify(
         optimization: str | NotGiven = NOT_GIVEN,
         is_archived: bool | NotGiven = NOT_GIVEN,
         models: List[str] | NotGiven = NOT_GIVEN,
-        tools: List[Tool] | NotGiven = NOT_GIVEN,
+        tools: List[dict[str, Tool]] | NotGiven = NOT_GIVEN,
         tool_resources: ToolResources | NotGiven = NOT_GIVEN,
     ) -> AssistantResponse:
         body = self._create_body(

ai21/models/responses/assistant_response.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Optional, List, Literal
+from typing import Optional, List, Literal, Any
 
 from ai21.models.ai21_base_model import AI21BaseModel
 from ai21.models.assistant.assistant import ToolResources
@@ -18,7 +18,7 @@ class AssistantResponse(AI21BaseModel):
     avatar: Optional[str] = None
     is_archived: bool = False
     models: Optional[List[str]] = None
-    tools: Optional[List[str]] = None
+    tools: Optional[List[dict[str, Any]]] = None
     tool_resources: Optional[ToolResources] = None