feat: Support new tools in AI Studio (box/box-openapi#534)

.codegen.json

@@ -1 +1 @@
-{ "engineHash": "fc1155c", "specHash": "b5422f4", "version": "1.15.0" }
+{ "engineHash": "a3ec39e", "specHash": "8402463", "version": "1.15.0" }

box_sdk_gen/schemas/__init__.py

@@ -30,6 +30,10 @@
 
 from box_sdk_gen.schemas.ai_agent_spreadsheet_tool import *
 
+from box_sdk_gen.schemas.ai_studio_agent_spreadsheet_tool import *
+
+from box_sdk_gen.schemas.ai_studio_agent_spreadsheet_tool_response import *
+
 from box_sdk_gen.schemas.ai_agent_basic_text_tool_base import *
 
 from box_sdk_gen.schemas.ai_agent_basic_text_tool_text_gen import *

box_sdk_gen/schemas/ai_studio_agent_ask.py

@@ -2,6 +2,8 @@
 
 from typing import Optional
 
+from typing import List
+
 from box_sdk_gen.internal.base_object import BaseObject
 
 from box_sdk_gen.schemas.ai_studio_agent_long_text_tool import AiStudioAgentLongTextTool
@@ -10,6 +12,10 @@
     AiStudioAgentBasicTextTool,
 )
 
+from box_sdk_gen.schemas.ai_studio_agent_spreadsheet_tool import (
+    AiStudioAgentSpreadsheetTool,
+)
+
 from box_sdk_gen.box.errors import BoxSDKError
 
 
@@ -27,28 +33,38 @@ def __init__(
         *,
         type: AiStudioAgentAskTypeField = AiStudioAgentAskTypeField.AI_AGENT_ASK,
         custom_instructions: Optional[str] = None,
+        suggested_questions: Optional[List[str]] = None,
         long_text: Optional[AiStudioAgentLongTextTool] = None,
         basic_text: Optional[AiStudioAgentBasicTextTool] = None,
+        basic_image: Optional[AiStudioAgentBasicTextTool] = None,
+        spreadsheet: Optional[AiStudioAgentSpreadsheetTool] = None,
         long_text_multi: Optional[AiStudioAgentLongTextTool] = None,
         basic_text_multi: Optional[AiStudioAgentBasicTextTool] = None,
+        basic_image_multi: Optional[AiStudioAgentBasicTextTool] = None,
         **kwargs
     ):
         """
         :param access_state: The state of the AI Agent capability. Possible values are: `enabled` and `disabled`.
         :type access_state: str
-        :param description: The description of the AI Agent.
+        :param description: The description of the AI agent.
         :type description: str
         :param type: The type of AI agent used to handle queries., defaults to AiStudioAgentAskTypeField.AI_AGENT_ASK
         :type type: AiStudioAgentAskTypeField, optional
-        :param custom_instructions: Custom instructions for the agent., defaults to None
+        :param custom_instructions: Custom instructions for the AI agent., defaults to None
         :type custom_instructions: Optional[str], optional
+        :param suggested_questions: Suggested questions for the AI agent. If null, suggested question will be generated. If empty, no suggested questions will be displayed., defaults to None
+        :type suggested_questions: Optional[List[str]], optional
         """
         super().__init__(**kwargs)
         self.access_state = access_state
         self.description = description
         self.type = type
         self.custom_instructions = custom_instructions
+        self.suggested_questions = suggested_questions
         self.long_text = long_text
         self.basic_text = basic_text
+        self.basic_image = basic_image
+        self.spreadsheet = spreadsheet
         self.long_text_multi = long_text_multi
         self.basic_text_multi = basic_text_multi
+        self.basic_image_multi = basic_image_multi

box_sdk_gen/schemas/ai_studio_agent_ask_response.py

@@ -2,6 +2,8 @@
 
 from typing import Optional
 
+from typing import List
+
 from box_sdk_gen.internal.base_object import BaseObject
 
 from box_sdk_gen.schemas.ai_studio_agent_long_text_tool_response import (
@@ -12,6 +14,10 @@
     AiStudioAgentBasicTextToolResponse,
 )
 
+from box_sdk_gen.schemas.ai_studio_agent_spreadsheet_tool_response import (
+    AiStudioAgentSpreadsheetToolResponse,
+)
+
 from box_sdk_gen.box.errors import BoxSDKError
 
 
@@ -29,28 +35,38 @@ def __init__(
         *,
         type: AiStudioAgentAskResponseTypeField = AiStudioAgentAskResponseTypeField.AI_AGENT_ASK,
         custom_instructions: Optional[str] = None,
+        suggested_questions: Optional[List[str]] = None,
         long_text: Optional[AiStudioAgentLongTextToolResponse] = None,
         basic_text: Optional[AiStudioAgentBasicTextToolResponse] = None,
+        basic_image: Optional[AiStudioAgentBasicTextToolResponse] = None,
+        spreadsheet: Optional[AiStudioAgentSpreadsheetToolResponse] = None,
         long_text_multi: Optional[AiStudioAgentLongTextToolResponse] = None,
         basic_text_multi: Optional[AiStudioAgentBasicTextToolResponse] = None,
+        basic_image_multi: Optional[AiStudioAgentBasicTextToolResponse] = None,
         **kwargs
     ):
         """
         :param access_state: The state of the AI Agent capability. Possible values are: `enabled` and `disabled`.
         :type access_state: str
-        :param description: The description of the AI Agent.
+        :param description: The description of the AI agent.
         :type description: str
-        :param type: The type of AI agent used to handle queries., defaults to AiStudioAgentAskResponseTypeField.AI_AGENT_ASK
+        :param type: The type of AI agent used to ask questions., defaults to AiStudioAgentAskResponseTypeField.AI_AGENT_ASK
         :type type: AiStudioAgentAskResponseTypeField, optional
-        :param custom_instructions: Custom instructions for the agent., defaults to None
+        :param custom_instructions: Custom instructions for the AI agent., defaults to None
         :type custom_instructions: Optional[str], optional
+        :param suggested_questions: Suggested questions for the AI agent. If null, suggested question will be generated. If empty, no suggested questions will be displayed., defaults to None
+        :type suggested_questions: Optional[List[str]], optional
         """
         super().__init__(**kwargs)
         self.access_state = access_state
         self.description = description
         self.type = type
         self.custom_instructions = custom_instructions
+        self.suggested_questions = suggested_questions
         self.long_text = long_text
         self.basic_text = basic_text
+        self.basic_image = basic_image
+        self.spreadsheet = spreadsheet
         self.long_text_multi = long_text_multi
         self.basic_text_multi = basic_text_multi
+        self.basic_image_multi = basic_image_multi

box_sdk_gen/schemas/ai_studio_agent_extract.py

@@ -29,16 +29,17 @@ def __init__(
         custom_instructions: Optional[str] = None,
         long_text: Optional[AiStudioAgentLongTextTool] = None,
         basic_text: Optional[AiStudioAgentBasicTextTool] = None,
+        basic_image: Optional[AiStudioAgentBasicTextTool] = None,
         **kwargs
     ):
         """
         :param access_state: The state of the AI Agent capability. Possible values are: `enabled` and `disabled`.
         :type access_state: str
-        :param description: The description of the AI Agent.
+        :param description: The description of the AI agent.
         :type description: str
-        :param type: The type of AI agent to be used for extraction., defaults to AiStudioAgentExtractTypeField.AI_AGENT_EXTRACT
+        :param type: The type of AI agent to be used for metadata extraction., defaults to AiStudioAgentExtractTypeField.AI_AGENT_EXTRACT
         :type type: AiStudioAgentExtractTypeField, optional
-        :param custom_instructions: Custom instructions for the agent., defaults to None
+        :param custom_instructions: Custom instructions for the AI agent., defaults to None
         :type custom_instructions: Optional[str], optional
         """
         super().__init__(**kwargs)
@@ -48,3 +49,4 @@ def __init__(
         self.custom_instructions = custom_instructions
         self.long_text = long_text
         self.basic_text = basic_text
+        self.basic_image = basic_image

box_sdk_gen/schemas/ai_studio_agent_extract_response.py

@@ -31,16 +31,17 @@ def __init__(
         custom_instructions: Optional[str] = None,
         long_text: Optional[AiStudioAgentLongTextToolResponse] = None,
         basic_text: Optional[AiStudioAgentBasicTextToolResponse] = None,
+        basic_image: Optional[AiStudioAgentBasicTextToolResponse] = None,
         **kwargs
     ):
         """
         :param access_state: The state of the AI Agent capability. Possible values are: `enabled` and `disabled`.
         :type access_state: str
-        :param description: The description of the AI Agent.
+        :param description: The description of the AI agent.
         :type description: str
-        :param type: The type of AI agent to be used for extraction., defaults to AiStudioAgentExtractResponseTypeField.AI_AGENT_EXTRACT
+        :param type: The type of AI agent to be used for metadata extraction., defaults to AiStudioAgentExtractResponseTypeField.AI_AGENT_EXTRACT
         :type type: AiStudioAgentExtractResponseTypeField, optional
-        :param custom_instructions: Custom instructions for the agent., defaults to None
+        :param custom_instructions: Custom instructions for the AI agent., defaults to None
         :type custom_instructions: Optional[str], optional
         """
         super().__init__(**kwargs)
@@ -50,3 +51,4 @@ def __init__(
         self.custom_instructions = custom_instructions
         self.long_text = long_text
         self.basic_text = basic_text
+        self.basic_image = basic_image

box_sdk_gen/schemas/ai_studio_agent_spreadsheet_tool.py

@@ -0,0 +1,30 @@
+from typing import Optional
+
+from box_sdk_gen.schemas.ai_llm_endpoint_params import AiLlmEndpointParams
+
+from box_sdk_gen.schemas.ai_agent_spreadsheet_tool import AiAgentSpreadsheetTool
+
+from box_sdk_gen.box.errors import BoxSDKError
+
+
+class AiStudioAgentSpreadsheetTool(AiAgentSpreadsheetTool):
+    def __init__(
+        self,
+        *,
+        model: Optional[str] = None,
+        num_tokens_for_completion: Optional[int] = None,
+        llm_endpoint_params: Optional[AiLlmEndpointParams] = None,
+        **kwargs
+    ):
+        """
+        :param model: The model used for the AI agent for spreadsheets. For specific model values, see the [available models list](g://box-ai/supported-models)., defaults to None
+        :type model: Optional[str], optional
+        :param num_tokens_for_completion: The number of tokens for completion., defaults to None
+        :type num_tokens_for_completion: Optional[int], optional
+        """
+        super().__init__(
+            model=model,
+            num_tokens_for_completion=num_tokens_for_completion,
+            llm_endpoint_params=llm_endpoint_params,
+            **kwargs
+        )

box_sdk_gen/schemas/ai_studio_agent_spreadsheet_tool_response.py

@@ -0,0 +1,40 @@
+from typing import Optional
+
+from typing import List
+
+from box_sdk_gen.schemas.ai_llm_endpoint_params import AiLlmEndpointParams
+
+from box_sdk_gen.schemas.ai_agent_spreadsheet_tool import AiAgentSpreadsheetTool
+
+from box_sdk_gen.schemas.ai_studio_agent_spreadsheet_tool import (
+    AiStudioAgentSpreadsheetTool,
+)
+
+from box_sdk_gen.box.errors import BoxSDKError
+
+
+class AiStudioAgentSpreadsheetToolResponse(AiStudioAgentSpreadsheetTool):
+    def __init__(
+        self,
+        *,
+        warnings: Optional[List[str]] = None,
+        model: Optional[str] = None,
+        num_tokens_for_completion: Optional[int] = None,
+        llm_endpoint_params: Optional[AiLlmEndpointParams] = None,
+        **kwargs
+    ):
+        """
+        :param warnings: Warnings concerning tool., defaults to None
+        :type warnings: Optional[List[str]], optional
+        :param model: The model used for the AI agent for spreadsheets. For specific model values, see the [available models list](g://box-ai/supported-models)., defaults to None
+        :type model: Optional[str], optional
+        :param num_tokens_for_completion: The number of tokens for completion., defaults to None
+        :type num_tokens_for_completion: Optional[int], optional
+        """
+        super().__init__(
+            model=model,
+            num_tokens_for_completion=num_tokens_for_completion,
+            llm_endpoint_params=llm_endpoint_params,
+            **kwargs
+        )
+        self.warnings = warnings

box_sdk_gen/schemas/ai_studio_agent_text_gen.py

@@ -2,6 +2,8 @@
 
 from typing import Optional
 
+from typing import List
+
 from box_sdk_gen.internal.base_object import BaseObject
 
 from box_sdk_gen.schemas.ai_studio_agent_basic_gen_tool import AiStudioAgentBasicGenTool
@@ -23,22 +25,26 @@ def __init__(
         *,
         type: AiStudioAgentTextGenTypeField = AiStudioAgentTextGenTypeField.AI_AGENT_TEXT_GEN,
         custom_instructions: Optional[str] = None,
+        suggested_questions: Optional[List[str]] = None,
         basic_gen: Optional[AiStudioAgentBasicGenTool] = None,
         **kwargs
     ):
         """
         :param access_state: The state of the AI Agent capability. Possible values are: `enabled` and `disabled`.
         :type access_state: str
-        :param description: The description of the AI Agent.
+        :param description: The description of the AI agent.
         :type description: str
         :param type: The type of AI agent used for generating text., defaults to AiStudioAgentTextGenTypeField.AI_AGENT_TEXT_GEN
         :type type: AiStudioAgentTextGenTypeField, optional
-        :param custom_instructions: Custom instructions for the agent., defaults to None
+        :param custom_instructions: Custom instructions for the AI agent., defaults to None
         :type custom_instructions: Optional[str], optional
+        :param suggested_questions: Suggested questions for the AI agent. If null, suggested question will be generated. If empty, no suggested questions will be displayed., defaults to None
+        :type suggested_questions: Optional[List[str]], optional
         """
         super().__init__(**kwargs)
         self.access_state = access_state
         self.description = description
         self.type = type
         self.custom_instructions = custom_instructions
+        self.suggested_questions = suggested_questions
         self.basic_gen = basic_gen

box_sdk_gen/schemas/ai_studio_agent_text_gen_response.py

@@ -2,6 +2,8 @@
 
 from typing import Optional
 
+from typing import List
+
 from box_sdk_gen.internal.base_object import BaseObject
 
 from box_sdk_gen.schemas.ai_studio_agent_basic_gen_tool_response import (
@@ -25,22 +27,26 @@ def __init__(
         *,
         type: AiStudioAgentTextGenResponseTypeField = AiStudioAgentTextGenResponseTypeField.AI_AGENT_TEXT_GEN,
         custom_instructions: Optional[str] = None,
+        suggested_questions: Optional[List[str]] = None,
         basic_gen: Optional[AiStudioAgentBasicGenToolResponse] = None,
         **kwargs
     ):
         """
         :param access_state: The state of the AI Agent capability. Possible values are: `enabled` and `disabled`.
         :type access_state: str
-        :param description: The description of the AI Agent.
+        :param description: The description of the AI agent.
         :type description: str
         :param type: The type of AI agent used for generating text., defaults to AiStudioAgentTextGenResponseTypeField.AI_AGENT_TEXT_GEN
         :type type: AiStudioAgentTextGenResponseTypeField, optional
-        :param custom_instructions: Custom instructions for the agent., defaults to None
+        :param custom_instructions: Custom instructions for the AI agent., defaults to None
         :type custom_instructions: Optional[str], optional
+        :param suggested_questions: Suggested questions for the AI agent. If null, suggested question will be generated. If empty, no suggested questions will be displayed., defaults to None
+        :type suggested_questions: Optional[List[str]], optional
         """
         super().__init__(**kwargs)
         self.access_state = access_state
         self.description = description
         self.type = type
         self.custom_instructions = custom_instructions
+        self.suggested_questions = suggested_questions
         self.basic_gen = basic_gen