From f357bb485ddb6243398ab51c14bb7f8795d27e9d Mon Sep 17 00:00:00 2001 From: box-sdk-build Date: Mon, 22 Jul 2024 04:41:53 -0700 Subject: [PATCH] fix: Merge schemas for AI Agent response and request parameter (box/box-openapi#441) --- .codegen.json | 2 +- box_sdk_gen/managers/ai.py | 14 ++----- box_sdk_gen/schemas/__init__.py | 4 -- box_sdk_gen/schemas/ai_agent_ask.py | 7 ++-- box_sdk_gen/schemas/ai_agent_ask_response.py | 38 ------------------- box_sdk_gen/schemas/ai_agent_text_gen.py | 7 ++-- .../schemas/ai_agent_text_gen_response.py | 30 --------------- box_sdk_gen/schemas/app_item_event_source.py | 2 +- docs/ai.md | 2 +- 9 files changed, 13 insertions(+), 93 deletions(-) delete mode 100644 box_sdk_gen/schemas/ai_agent_ask_response.py delete mode 100644 box_sdk_gen/schemas/ai_agent_text_gen_response.py diff --git a/.codegen.json b/.codegen.json index eea0921..88a015b 100644 --- a/.codegen.json +++ b/.codegen.json @@ -1 +1 @@ -{ "engineHash": "c2b0bec", "specHash": "6573c3a", "version": "1.1.0" } +{ "engineHash": "c2b0bec", "specHash": "d36b9f0", "version": "1.1.0" } diff --git a/box_sdk_gen/managers/ai.py b/box_sdk_gen/managers/ai.py index 55c1fcd..2b7cbe1 100644 --- a/box_sdk_gen/managers/ai.py +++ b/box_sdk_gen/managers/ai.py @@ -18,10 +18,6 @@ from box_sdk_gen.internal.utils import DateTime -from box_sdk_gen.schemas.ai_agent_ask import AiAgentAsk - -from box_sdk_gen.schemas.ai_agent_text_gen import AiAgentTextGen - from box_sdk_gen.schemas.ai_response import AiResponse from box_sdk_gen.schemas.client_error import ClientError @@ -30,9 +26,9 @@ from box_sdk_gen.schemas.ai_text_gen import AiTextGen -from box_sdk_gen.schemas.ai_agent_ask_response import AiAgentAskResponse +from box_sdk_gen.schemas.ai_agent_ask import AiAgentAsk -from box_sdk_gen.schemas.ai_agent_text_gen_response import AiAgentTextGenResponse +from box_sdk_gen.schemas.ai_agent_text_gen import AiAgentTextGen from box_sdk_gen.networking.auth import Authentication @@ -259,7 +255,7 @@ def get_ai_agent_default( language: Optional[str] = None, model: Optional[str] = None, extra_headers: Optional[Dict[str, Optional[str]]] = None - ) -> Union[AiAgentAskResponse, AiAgentTextGenResponse]: + ) -> Union[AiAgentAsk, AiAgentTextGen]: """ Get the AI agent default config :param mode: The mode to filter the agent config to return. @@ -293,6 +289,4 @@ def get_ai_agent_default( network_session=self.network_session, ), ) - return deserialize( - response.data, Union[AiAgentAskResponse, AiAgentTextGenResponse] - ) + return deserialize(response.data, Union[AiAgentAsk, AiAgentTextGen]) diff --git a/box_sdk_gen/schemas/__init__.py b/box_sdk_gen/schemas/__init__.py index 39f2a82..b682b5f 100644 --- a/box_sdk_gen/schemas/__init__.py +++ b/box_sdk_gen/schemas/__init__.py @@ -338,16 +338,12 @@ from box_sdk_gen.schemas.ai_agent_basic_gen_tool import * -from box_sdk_gen.schemas.ai_agent_text_gen_response import * - from box_sdk_gen.schemas.ai_agent_text_gen import * from box_sdk_gen.schemas.ai_text_gen import * from box_sdk_gen.schemas.ai_agent_basic_text_tool_ask import * -from box_sdk_gen.schemas.ai_agent_ask_response import * - from box_sdk_gen.schemas.ai_agent_ask import * from box_sdk_gen.schemas.ai_ask import * diff --git a/box_sdk_gen/schemas/ai_agent_ask.py b/box_sdk_gen/schemas/ai_agent_ask.py index 807da59..11ebfc5 100644 --- a/box_sdk_gen/schemas/ai_agent_ask.py +++ b/box_sdk_gen/schemas/ai_agent_ask.py @@ -19,7 +19,7 @@ class AiAgentAsk(BaseObject): def __init__( self, *, - type: AiAgentAskTypeField = AiAgentAskTypeField.AI_AGENT_ASK.value, + type: Optional[AiAgentAskTypeField] = None, long_text: Optional[AiAgentLongTextTool] = None, basic_text: Optional[AiAgentBasicTextToolAsk] = None, long_text_multi: Optional[AiAgentLongTextTool] = None, @@ -27,9 +27,8 @@ def __init__( **kwargs ): """ - :param type: The type of AI agent used to handle queries. - This parameter is **required**., defaults to AiAgentAskTypeField.AI_AGENT_ASK.value - :type type: AiAgentAskTypeField, optional + :param type: The type of AI agent used to handle queries., defaults to None + :type type: Optional[AiAgentAskTypeField], optional """ super().__init__(**kwargs) self.type = type diff --git a/box_sdk_gen/schemas/ai_agent_ask_response.py b/box_sdk_gen/schemas/ai_agent_ask_response.py deleted file mode 100644 index 69b04ca..0000000 --- a/box_sdk_gen/schemas/ai_agent_ask_response.py +++ /dev/null @@ -1,38 +0,0 @@ -from enum import Enum - -from typing import Optional - -from box_sdk_gen.internal.base_object import BaseObject - -from box_sdk_gen.schemas.ai_agent_long_text_tool import AiAgentLongTextTool - -from box_sdk_gen.schemas.ai_agent_basic_text_tool_ask import AiAgentBasicTextToolAsk - - -class AiAgentAskResponseTypeField(str, Enum): - AI_AGENT_ASK = 'ai_agent_ask' - - -class AiAgentAskResponse(BaseObject): - _discriminator = 'type', {'ai_agent_ask'} - - def __init__( - self, - *, - type: Optional[AiAgentAskResponseTypeField] = None, - long_text: Optional[AiAgentLongTextTool] = None, - basic_text: Optional[AiAgentBasicTextToolAsk] = None, - long_text_multi: Optional[AiAgentLongTextTool] = None, - basic_text_multi: Optional[AiAgentBasicTextToolAsk] = None, - **kwargs - ): - """ - :param type: The type of AI agent used to handle queries., defaults to None - :type type: Optional[AiAgentAskResponseTypeField], optional - """ - super().__init__(**kwargs) - self.type = type - self.long_text = long_text - self.basic_text = basic_text - self.long_text_multi = long_text_multi - self.basic_text_multi = basic_text_multi diff --git a/box_sdk_gen/schemas/ai_agent_text_gen.py b/box_sdk_gen/schemas/ai_agent_text_gen.py index f3a47f9..930d2e7 100644 --- a/box_sdk_gen/schemas/ai_agent_text_gen.py +++ b/box_sdk_gen/schemas/ai_agent_text_gen.py @@ -17,14 +17,13 @@ class AiAgentTextGen(BaseObject): def __init__( self, *, - type: AiAgentTextGenTypeField = AiAgentTextGenTypeField.AI_AGENT_TEXT_GEN.value, + type: Optional[AiAgentTextGenTypeField] = None, basic_gen: Optional[AiAgentBasicGenTool] = None, **kwargs ): """ - :param type: The type of AI agent used for generating text. - This parameter is **required**., defaults to AiAgentTextGenTypeField.AI_AGENT_TEXT_GEN.value - :type type: AiAgentTextGenTypeField, optional + :param type: The type of AI agent used for generating text., defaults to None + :type type: Optional[AiAgentTextGenTypeField], optional """ super().__init__(**kwargs) self.type = type diff --git a/box_sdk_gen/schemas/ai_agent_text_gen_response.py b/box_sdk_gen/schemas/ai_agent_text_gen_response.py deleted file mode 100644 index 820060e..0000000 --- a/box_sdk_gen/schemas/ai_agent_text_gen_response.py +++ /dev/null @@ -1,30 +0,0 @@ -from enum import Enum - -from typing import Optional - -from box_sdk_gen.internal.base_object import BaseObject - -from box_sdk_gen.schemas.ai_agent_basic_gen_tool import AiAgentBasicGenTool - - -class AiAgentTextGenResponseTypeField(str, Enum): - AI_AGENT_TEXT_GEN = 'ai_agent_text_gen' - - -class AiAgentTextGenResponse(BaseObject): - _discriminator = 'type', {'ai_agent_text_gen'} - - def __init__( - self, - *, - type: Optional[AiAgentTextGenResponseTypeField] = None, - basic_gen: Optional[AiAgentBasicGenTool] = None, - **kwargs - ): - """ - :param type: The type of AI agent used for generating text., defaults to None - :type type: Optional[AiAgentTextGenResponseTypeField], optional - """ - super().__init__(**kwargs) - self.type = type - self.basic_gen = basic_gen diff --git a/box_sdk_gen/schemas/app_item_event_source.py b/box_sdk_gen/schemas/app_item_event_source.py index cc98da0..d8abacf 100644 --- a/box_sdk_gen/schemas/app_item_event_source.py +++ b/box_sdk_gen/schemas/app_item_event_source.py @@ -31,7 +31,7 @@ def __init__( :type id: str :param app_item_type: The type of the `AppItem` :type app_item_type: str - :param type: The type of the source that this event represents. Can be only `app_item`. + :param type: The type of the source that this event represents. Can only be `app_item`. , defaults to AppItemEventSourceTypeField.APP_ITEM.value :type type: AppItemEventSourceTypeField, optional """ diff --git a/docs/ai.md b/docs/ai.md index e476b68..2bf7785 100644 --- a/docs/ai.md +++ b/docs/ai.md @@ -132,7 +132,7 @@ _Currently we don't have an example for calling `get_ai_agent_default` in integr ### Returns -This function returns a value of type `Union[AiAgentAskResponse, AiAgentTextGenResponse]`. +This function returns a value of type `Union[AiAgentAsk, AiAgentTextGen]`. A successful response including the default agent configuration. This response can be one of the following two objects: