wip

Author: harel-tussi

ai21/clients/studio/resources/agents/agents.py

@@ -116,15 +116,15 @@ def modify(
         self,
         agent_id: str,
         *,
-        name: Union[str, NotGiven] = NOT_GIVEN,
-        description: Union[str, NotGiven] = NOT_GIVEN,
-        models: Union[List[str], NotGiven] = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        description: str | NotGiven = NOT_GIVEN,
+        models: List[str] | NotGiven = NOT_GIVEN,
         tools: Union[List[Dict[str, Any]], NotGiven] = NOT_GIVEN,
-        tool_resources: Union[Dict[str, Any], NotGiven] = NOT_GIVEN,
-        requirements: Union[List[AgentRequirement], NotGiven] = NOT_GIVEN,
-        budget: Union[BudgetLevel, NotGiven] = NOT_GIVEN,
-        visibility: Union[Visibility, NotGiven] = NOT_GIVEN,
-        response_language: Union[ResponseLanguage, NotGiven] = NOT_GIVEN,
+        tool_resources: Dict[str, Any] | NotGiven = NOT_GIVEN,
+        requirements: List[AgentRequirement] | NotGiven = NOT_GIVEN,
+        budget: BudgetLevel | NotGiven = NOT_GIVEN,
+        visibility: Visibility | NotGiven = NOT_GIVEN,
+        response_language: ResponseLanguage | NotGiven = NOT_GIVEN,
         **kwargs,
     ) -> Agent:
         """Modify an existing agent"""