feat: add AWS params (box/box-openapi#478)

.codegen.json

@@ -1 +1 @@
-{ "engineHash": "2efc8ab", "specHash": "e798cb1", "version": "1.5.1" }
+{ "engineHash": "2efc8ab", "specHash": "90cf4e4", "version": "1.5.1" }

box_sdk_gen/schemas/__init__.py

@@ -6,6 +6,8 @@
 
 from box_sdk_gen.schemas.ai_item_base import *
 
+from box_sdk_gen.schemas.ai_llm_endpoint_params_aws import *
+
 from box_sdk_gen.schemas.ai_llm_endpoint_params_google import *
 
 from box_sdk_gen.schemas.ai_llm_endpoint_params_open_ai import *

box_sdk_gen/schemas/ai_agent_basic_gen_tool.py

@@ -6,6 +6,8 @@
 
 from box_sdk_gen.schemas.ai_llm_endpoint_params_google import AiLlmEndpointParamsGoogle
 
+from box_sdk_gen.schemas.ai_llm_endpoint_params_aws import AiLlmEndpointParamsAws
+
 from box_sdk_gen.schemas.ai_agent_basic_text_tool_base import AiAgentBasicTextToolBase
 
 from box_sdk_gen.schemas.ai_agent_basic_text_tool_text_gen import (
@@ -32,7 +34,11 @@ def __init__(
         model: Optional[str] = None,
         num_tokens_for_completion: Optional[int] = None,
         llm_endpoint_params: Optional[
-            Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle]
+            Union[
+                AiLlmEndpointParamsOpenAi,
+                AiLlmEndpointParamsGoogle,
+                AiLlmEndpointParamsAws,
+            ]
         ] = None,
         **kwargs
     ):
@@ -53,7 +59,7 @@ def __init__(
                 :param num_tokens_for_completion: The number of tokens for completion., defaults to None
                 :type num_tokens_for_completion: Optional[int], optional
                 :param llm_endpoint_params: The parameters for the LLM endpoint specific to OpenAI / Google models., defaults to None
-                :type llm_endpoint_params: Optional[Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle]], optional
+                :type llm_endpoint_params: Optional[Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle, AiLlmEndpointParamsAws]], optional
         """
         super().__init__(
             embeddings=embeddings,

box_sdk_gen/schemas/ai_agent_basic_text_tool.py

@@ -6,6 +6,8 @@
 
 from box_sdk_gen.schemas.ai_llm_endpoint_params_google import AiLlmEndpointParamsGoogle
 
+from box_sdk_gen.schemas.ai_llm_endpoint_params_aws import AiLlmEndpointParamsAws
+
 from box_sdk_gen.schemas.ai_agent_basic_text_tool_base import AiAgentBasicTextToolBase
 
 
@@ -18,7 +20,11 @@ def __init__(
         model: Optional[str] = None,
         num_tokens_for_completion: Optional[int] = None,
         llm_endpoint_params: Optional[
-            Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle]
+            Union[
+                AiLlmEndpointParamsOpenAi,
+                AiLlmEndpointParamsGoogle,
+                AiLlmEndpointParamsAws,
+            ]
         ] = None,
         **kwargs
     ):
@@ -34,7 +40,7 @@ def __init__(
                 :param num_tokens_for_completion: The number of tokens for completion., defaults to None
                 :type num_tokens_for_completion: Optional[int], optional
                 :param llm_endpoint_params: The parameters for the LLM endpoint specific to OpenAI / Google models., defaults to None
-                :type llm_endpoint_params: Optional[Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle]], optional
+                :type llm_endpoint_params: Optional[Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle, AiLlmEndpointParamsAws]], optional
         """
         super().__init__(
             model=model,

box_sdk_gen/schemas/ai_agent_basic_text_tool_base.py

@@ -8,6 +8,8 @@
 
 from box_sdk_gen.schemas.ai_llm_endpoint_params_google import AiLlmEndpointParamsGoogle
 
+from box_sdk_gen.schemas.ai_llm_endpoint_params_aws import AiLlmEndpointParamsAws
+
 
 class AiAgentBasicTextToolBase(BaseObject):
     def __init__(
@@ -16,7 +18,11 @@ def __init__(
         model: Optional[str] = None,
         num_tokens_for_completion: Optional[int] = None,
         llm_endpoint_params: Optional[
-            Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle]
+            Union[
+                AiLlmEndpointParamsOpenAi,
+                AiLlmEndpointParamsGoogle,
+                AiLlmEndpointParamsAws,
+            ]
         ] = None,
         **kwargs
     ):
@@ -26,7 +32,7 @@ def __init__(
         :param num_tokens_for_completion: The number of tokens for completion., defaults to None
         :type num_tokens_for_completion: Optional[int], optional
         :param llm_endpoint_params: The parameters for the LLM endpoint specific to OpenAI / Google models., defaults to None
-        :type llm_endpoint_params: Optional[Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle]], optional
+        :type llm_endpoint_params: Optional[Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle, AiLlmEndpointParamsAws]], optional
         """
         super().__init__(**kwargs)
         self.model = model

box_sdk_gen/schemas/ai_agent_basic_text_tool_text_gen.py

@@ -6,6 +6,8 @@
 
 from box_sdk_gen.schemas.ai_llm_endpoint_params_google import AiLlmEndpointParamsGoogle
 
+from box_sdk_gen.schemas.ai_llm_endpoint_params_aws import AiLlmEndpointParamsAws
+
 from box_sdk_gen.schemas.ai_agent_basic_text_tool_base import AiAgentBasicTextToolBase
 
 
@@ -18,7 +20,11 @@ def __init__(
         model: Optional[str] = None,
         num_tokens_for_completion: Optional[int] = None,
         llm_endpoint_params: Optional[
-            Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle]
+            Union[
+                AiLlmEndpointParamsOpenAi,
+                AiLlmEndpointParamsGoogle,
+                AiLlmEndpointParamsAws,
+            ]
         ] = None,
         **kwargs
     ):
@@ -36,7 +42,7 @@ def __init__(
                 :param num_tokens_for_completion: The number of tokens for completion., defaults to None
                 :type num_tokens_for_completion: Optional[int], optional
                 :param llm_endpoint_params: The parameters for the LLM endpoint specific to OpenAI / Google models., defaults to None
-                :type llm_endpoint_params: Optional[Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle]], optional
+                :type llm_endpoint_params: Optional[Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle, AiLlmEndpointParamsAws]], optional
         """
         super().__init__(
             model=model,

box_sdk_gen/schemas/ai_agent_long_text_tool.py

@@ -8,6 +8,8 @@
 
 from box_sdk_gen.schemas.ai_llm_endpoint_params_google import AiLlmEndpointParamsGoogle
 
+from box_sdk_gen.schemas.ai_llm_endpoint_params_aws import AiLlmEndpointParamsAws
+
 from box_sdk_gen.schemas.ai_agent_basic_text_tool_base import AiAgentBasicTextToolBase
 
 from box_sdk_gen.schemas.ai_agent_basic_text_tool import AiAgentBasicTextTool
@@ -59,7 +61,11 @@ def __init__(
         model: Optional[str] = None,
         num_tokens_for_completion: Optional[int] = None,
         llm_endpoint_params: Optional[
-            Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle]
+            Union[
+                AiLlmEndpointParamsOpenAi,
+                AiLlmEndpointParamsGoogle,
+                AiLlmEndpointParamsAws,
+            ]
         ] = None,
         **kwargs
     ):
@@ -75,7 +81,7 @@ def __init__(
                 :param num_tokens_for_completion: The number of tokens for completion., defaults to None
                 :type num_tokens_for_completion: Optional[int], optional
                 :param llm_endpoint_params: The parameters for the LLM endpoint specific to OpenAI / Google models., defaults to None
-                :type llm_endpoint_params: Optional[Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle]], optional
+                :type llm_endpoint_params: Optional[Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle, AiLlmEndpointParamsAws]], optional
         """
         super().__init__(
             system_message=system_message,

box_sdk_gen/schemas/ai_agent_long_text_tool_text_gen.py

@@ -8,6 +8,8 @@
 
 from box_sdk_gen.schemas.ai_llm_endpoint_params_google import AiLlmEndpointParamsGoogle
 
+from box_sdk_gen.schemas.ai_llm_endpoint_params_aws import AiLlmEndpointParamsAws
+
 from box_sdk_gen.schemas.ai_agent_basic_text_tool_base import AiAgentBasicTextToolBase
 
 from box_sdk_gen.schemas.ai_agent_basic_text_tool_text_gen import (
@@ -61,7 +63,11 @@ def __init__(
         model: Optional[str] = None,
         num_tokens_for_completion: Optional[int] = None,
         llm_endpoint_params: Optional[
-            Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle]
+            Union[
+                AiLlmEndpointParamsOpenAi,
+                AiLlmEndpointParamsGoogle,
+                AiLlmEndpointParamsAws,
+            ]
         ] = None,
         **kwargs
     ):
@@ -79,7 +85,7 @@ def __init__(
                 :param num_tokens_for_completion: The number of tokens for completion., defaults to None
                 :type num_tokens_for_completion: Optional[int], optional
                 :param llm_endpoint_params: The parameters for the LLM endpoint specific to OpenAI / Google models., defaults to None
-                :type llm_endpoint_params: Optional[Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle]], optional
+                :type llm_endpoint_params: Optional[Union[AiLlmEndpointParamsOpenAi, AiLlmEndpointParamsGoogle, AiLlmEndpointParamsAws]], optional
         """
         super().__init__(
             system_message=system_message,

box_sdk_gen/schemas/ai_llm_endpoint_params_aws.py

@@ -0,0 +1,39 @@
+from enum import Enum
+
+from typing import Optional
+
+from box_sdk_gen.internal.base_object import BaseObject
+
+
+class AiLlmEndpointParamsAwsTypeField(str, Enum):
+    AWS_PARAMS = 'aws_params'
+
+
+class AiLlmEndpointParamsAws(BaseObject):
+    _discriminator = 'type', {'aws_params'}
+
+    def __init__(
+        self,
+        *,
+        type: AiLlmEndpointParamsAwsTypeField = AiLlmEndpointParamsAwsTypeField.AWS_PARAMS.value,
+        temperature: Optional[float] = None,
+        top_p: Optional[float] = None,
+        **kwargs
+    ):
+        """
+                :param type: The type of the AI LLM endpoint params object for AWS.
+        This parameter is **required**., defaults to AiLlmEndpointParamsAwsTypeField.AWS_PARAMS.value
+                :type type: AiLlmEndpointParamsAwsTypeField, optional
+                :param temperature: What sampling temperature to use, between 0 and 1. Higher values like 0.8 will make the output more random,
+        while lower values like 0.2 will make it more focused and deterministic.
+        We generally recommend altering this or `top_p` but not both., defaults to None
+                :type temperature: Optional[float], optional
+                :param top_p: An alternative to sampling with temperature, called nucleus sampling, where the model considers the results
+        of the tokens with `top_p` probability mass. So 0.1 means only the tokens comprising the top 10% probability
+        mass are considered. We generally recommend altering this or temperature but not both., defaults to None
+                :type top_p: Optional[float], optional
+        """
+        super().__init__(**kwargs)
+        self.type = type
+        self.temperature = temperature
+        self.top_p = top_p