feat: Support AI Studio API (box/box-codegen#626) (#483)

.codegen.json

@@ -1 +1 @@
-{ "engineHash": "59337f7", "specHash": "59747aa", "version": "1.11.1" }
+{ "engineHash": "8a9cc1d", "specHash": "f20ba3f", "version": "1.11.1" }

box_sdk_gen/client.py

@@ -164,6 +164,8 @@
 
 from box_sdk_gen.managers.ai import AiManager
 
+from box_sdk_gen.managers.ai_studio import AiStudioManager
+
 from box_sdk_gen.managers.docgen_template import DocgenTemplateManager
 
 from box_sdk_gen.managers.docgen import DocgenManager
@@ -401,6 +403,9 @@ def __init__(self, auth: Authentication, *, network_session: NetworkSession = No
             auth=self.auth, network_session=self.network_session
         )
         self.ai = AiManager(auth=self.auth, network_session=self.network_session)
+        self.ai_studio = AiStudioManager(
+            auth=self.auth, network_session=self.network_session
+        )
         self.docgen_template = DocgenTemplateManager(
             auth=self.auth, network_session=self.network_session
         )

box_sdk_gen/managers/__init__.py

@@ -138,6 +138,8 @@
 
 from box_sdk_gen.managers.ai import *
 
+from box_sdk_gen.managers.ai_studio import *
+
 from box_sdk_gen.managers.docgen_template import *
 
 from box_sdk_gen.managers.docgen import *

box_sdk_gen/managers/ai.py

@@ -8,18 +8,20 @@
 
 from typing import Dict
 
+from typing import Union
+
 from box_sdk_gen.serialization.json import serialize
 
 from box_sdk_gen.internal.utils import to_string
 
 from box_sdk_gen.serialization.json import deserialize
 
-from typing import Union
-
 from box_sdk_gen.schemas.ai_item_ask import AiItemAsk
 
 from box_sdk_gen.schemas.ai_dialogue_history import AiDialogueHistory
 
+from box_sdk_gen.schemas.ai_agent_reference import AiAgentReference
+
 from box_sdk_gen.networking.fetch_options import ResponseFormat
 
 from box_sdk_gen.schemas.ai_item_base import AiItemBase
@@ -220,7 +222,7 @@ def create_ai_ask(
         *,
         dialogue_history: Optional[List[AiDialogueHistory]] = None,
         include_citations: Optional[bool] = None,
-        ai_agent: Optional[AiAgentAsk] = None,
+        ai_agent: Optional[Union[AiAgentAsk, AiAgentReference]] = None,
         extra_headers: Optional[Dict[str, Optional[str]]] = None
     ) -> Optional[AiResponseFull]:
         """
@@ -275,7 +277,7 @@ def create_ai_text_gen(
         items: List[CreateAiTextGenItems],
         *,
         dialogue_history: Optional[List[AiDialogueHistory]] = None,
-        ai_agent: Optional[AiAgentTextGen] = None,
+        ai_agent: Optional[Union[AiAgentReference, AiAgentTextGen]] = None,
         extra_headers: Optional[Dict[str, Optional[str]]] = None
     ) -> AiResponse:
         """
@@ -371,7 +373,7 @@ def create_ai_extract(
         prompt: str,
         items: List[AiItemBase],
         *,
-        ai_agent: Optional[AiAgentExtract] = None,
+        ai_agent: Optional[Union[AiAgentReference, AiAgentExtract]] = None,
         extra_headers: Optional[Dict[str, Optional[str]]] = None
     ) -> AiResponse:
         """
@@ -415,7 +417,7 @@ def create_ai_extract_structured(
         *,
         metadata_template: Optional[CreateAiExtractStructuredMetadataTemplate] = None,
         fields: Optional[List[CreateAiExtractStructuredFields]] = None,
-        ai_agent: Optional[AiAgentExtractStructured] = None,
+        ai_agent: Optional[Union[AiAgentReference, AiAgentExtractStructured]] = None,
         extra_headers: Optional[Dict[str, Optional[str]]] = None
     ) -> AiExtractStructuredResponse:
         """