fix: 🏷️ remove Model Literal typing

Author: benshuk

ai21/clients/common/assistant/assistant.py

@@ -7,7 +7,6 @@
     AssistantResponse,
     Optimization,
     ToolResources,
-    Model,
     Tool,
     ListAssistantResponse,
 )
@@ -26,7 +25,7 @@ def create(
         description: str | NotGiven = NOT_GIVEN,
         optimization: Optimization | NotGiven = NOT_GIVEN,
         avatar: str | NotGiven = NOT_GIVEN,
-        models: List[Model] | NotGiven = NOT_GIVEN,
+        models: List[str] | NotGiven = NOT_GIVEN,
         tools: List[Tool] | NotGiven = NOT_GIVEN,
         tool_resources: ToolResources | NotGiven = NOT_GIVEN,
         **kwargs,
@@ -76,7 +75,7 @@ def modify(
         optimization: Optimization | NotGiven = NOT_GIVEN,
         avatar: str | NotGiven = NOT_GIVEN,
         is_archived: bool | NotGiven = NOT_GIVEN,
-        models: List[Model] | NotGiven = NOT_GIVEN,
+        models: List[str] | NotGiven = NOT_GIVEN,
         tools: List[Tool] | NotGiven = NOT_GIVEN,
         tool_resources: ToolResources | NotGiven = NOT_GIVEN,
     ) -> AssistantResponse:

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

@@ -9,7 +9,6 @@
 )
 from ai21.models.responses.assistant_response import (
     AssistantResponse,
-    Model,
     Tool,
     ToolResources,
     ListAssistantResponse,
@@ -25,7 +24,7 @@ def create(
         description: str | NotGiven = NOT_GIVEN,
         optimization: str | NotGiven = NOT_GIVEN,
         avatar: str | NotGiven = NOT_GIVEN,
-        models: List[Model] | NotGiven = NOT_GIVEN,
+        models: List[str] | NotGiven = NOT_GIVEN,
         tools: List[Tool] | NotGiven = NOT_GIVEN,
         tool_resources: ToolResources | NotGiven = NOT_GIVEN,
         **kwargs,
@@ -58,7 +57,7 @@ def modify(
         optimization: str | NotGiven = NOT_GIVEN,
         avatar: str | NotGiven = NOT_GIVEN,
         is_archived: bool | NotGiven = NOT_GIVEN,
-        models: List[Model] | NotGiven = NOT_GIVEN,
+        models: List[str] | NotGiven = NOT_GIVEN,
         tools: List[Tool] | NotGiven = NOT_GIVEN,
         tool_resources: ToolResources | NotGiven = NOT_GIVEN,
     ) -> AssistantResponse:
@@ -84,7 +83,7 @@ async def create(
         description: str | NotGiven = NOT_GIVEN,
         optimization: str | NotGiven = NOT_GIVEN,
         avatar: str | NotGiven = NOT_GIVEN,
-        models: List[Model] | NotGiven = NOT_GIVEN,
+        models: List[str] | NotGiven = NOT_GIVEN,
         tools: List[Tool] | NotGiven = NOT_GIVEN,
         tool_resources: ToolResources | NotGiven = NOT_GIVEN,
         **kwargs,
@@ -117,7 +116,7 @@ async def modify(
         optimization: str | NotGiven = NOT_GIVEN,
         avatar: str | NotGiven = NOT_GIVEN,
         is_archived: bool | NotGiven = NOT_GIVEN,
-        models: List[Model] | NotGiven = NOT_GIVEN,
+        models: List[str] | NotGiven = NOT_GIVEN,
         tools: List[Tool] | NotGiven = NOT_GIVEN,
         tool_resources: ToolResources | NotGiven = NOT_GIVEN,
     ) -> AssistantResponse:

ai21/models/responses/assistant_response.py

@@ -5,7 +5,6 @@
 
 
 Optimization = Literal["cost", "latency"]
-Model = Literal["jamba-1-5", "jamba-1-5-large"]
 Tool = Literal["rag", "internet_research", "plan_approval"]
 
 
@@ -21,7 +20,7 @@ class AssistantResponse(AI21BaseModel):
     updated_at: datetime
     object: str
     name: str
-    description: Optional[str]
+    description: Optional[str] = None
     optimization: str
     organization_id: str
     user_id: str