orq-ai-sdk 4.2.8__py3-none-any.whl → 4.2.9__py3-none-any.whl

orq_ai_sdk/models/updateagentop.py

@@ -533,6 +533,8 @@ class ModelConfigurationTimeout(BaseModel):
 class ModelConfigurationParametersTypedDict(TypedDict):
     r"""Model behavior parameters that control how the model generates responses. Common parameters: `temperature` (0-1, randomness), `max_completion_tokens` (max output length), `top_p` (sampling diversity). Advanced: `frequency_penalty`, `presence_penalty`, `response_format` (JSON/structured), `reasoning_effort`, `seed` (reproducibility). Support varies by model - consult AI Gateway documentation."""
 
+    name: NotRequired[str]
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
     audio: NotRequired[Nullable[UpdateAgentModelConfigurationAudioTypedDict]]
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
     frequency_penalty: NotRequired[Nullable[float]]
@@ -602,6 +604,9 @@ class ModelConfigurationParametersTypedDict(TypedDict):
 class ModelConfigurationParameters(BaseModel):
     r"""Model behavior parameters that control how the model generates responses. Common parameters: `temperature` (0-1, randomness), `max_completion_tokens` (max output length), `top_p` (sampling diversity). Advanced: `frequency_penalty`, `presence_penalty`, `response_format` (JSON/structured), `reasoning_effort`, `seed` (reproducibility). Support varies by model - consult AI Gateway documentation."""
 
+    name: Optional[str] = None
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
+
     audio: OptionalNullable[UpdateAgentModelConfigurationAudio] = UNSET
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
 
@@ -697,6 +702,7 @@ class ModelConfigurationParameters(BaseModel):
     def serialize_model(self, handler):
         optional_fields = set(
             [
+                "name",
                 "audio",
                 "frequency_penalty",
                 "max_tokens",
@@ -1373,6 +1379,8 @@ class UpdateAgentFallbackModelConfigurationTimeout(BaseModel):
 class UpdateAgentFallbackModelConfigurationParametersTypedDict(TypedDict):
     r"""Optional model parameters specific to this fallback model. Overrides primary model parameters if this fallback is used."""
 
+    name: NotRequired[str]
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
     audio: NotRequired[Nullable[UpdateAgentFallbackModelConfigurationAudioTypedDict]]
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
     frequency_penalty: NotRequired[Nullable[float]]
@@ -1454,6 +1462,9 @@ class UpdateAgentFallbackModelConfigurationParametersTypedDict(TypedDict):
 class UpdateAgentFallbackModelConfigurationParameters(BaseModel):
     r"""Optional model parameters specific to this fallback model. Overrides primary model parameters if this fallback is used."""
 
+    name: Optional[str] = None
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
+
     audio: OptionalNullable[UpdateAgentFallbackModelConfigurationAudio] = UNSET
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
 
@@ -1557,6 +1568,7 @@ class UpdateAgentFallbackModelConfigurationParameters(BaseModel):
     def serialize_model(self, handler):
         optional_fields = set(
             [
+                "name",
                 "audio",
                 "frequency_penalty",
                 "max_tokens",
@@ -3561,6 +3573,8 @@ class UpdateAgentTimeout(BaseModel):
 class UpdateAgentParametersTypedDict(TypedDict):
     r"""Model behavior parameters (snake_case) stored as part of the agent configuration. These become the default parameters used when the agent is executed. Commonly used: temperature (0-1, controls randomness), max_completion_tokens (response length), top_p (nucleus sampling). Advanced: frequency_penalty, presence_penalty, response_format (JSON/structured output), reasoning_effort (for o1/thinking models), seed (reproducibility), stop sequences. Model-specific support varies. Runtime parameters in agent execution requests can override these defaults."""
 
+    name: NotRequired[str]
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
     audio: NotRequired[Nullable[UpdateAgentAudioTypedDict]]
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
     frequency_penalty: NotRequired[Nullable[float]]
@@ -3630,6 +3644,9 @@ class UpdateAgentParametersTypedDict(TypedDict):
 class UpdateAgentParameters(BaseModel):
     r"""Model behavior parameters (snake_case) stored as part of the agent configuration. These become the default parameters used when the agent is executed. Commonly used: temperature (0-1, controls randomness), max_completion_tokens (response length), top_p (nucleus sampling). Advanced: frequency_penalty, presence_penalty, response_format (JSON/structured output), reasoning_effort (for o1/thinking models), seed (reproducibility), stop sequences. Model-specific support varies. Runtime parameters in agent execution requests can override these defaults."""
 
+    name: Optional[str] = None
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
+
     audio: OptionalNullable[UpdateAgentAudio] = UNSET
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
 
@@ -3725,6 +3742,7 @@ class UpdateAgentParameters(BaseModel):
     def serialize_model(self, handler):
         optional_fields = set(
             [
+                "name",
                 "audio",
                 "frequency_penalty",
                 "max_tokens",
@@ -4367,6 +4385,8 @@ class UpdateAgentFallbackModelConfigurationAgentsTimeout(BaseModel):
 class UpdateAgentFallbackModelConfigurationAgentsParametersTypedDict(TypedDict):
     r"""Optional model parameters specific to this fallback model. Overrides primary model parameters if this fallback is used."""
 
+    name: NotRequired[str]
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
     audio: NotRequired[
         Nullable[UpdateAgentFallbackModelConfigurationAgentsAudioTypedDict]
     ]
@@ -4458,6 +4478,9 @@ class UpdateAgentFallbackModelConfigurationAgentsParametersTypedDict(TypedDict):
 class UpdateAgentFallbackModelConfigurationAgentsParameters(BaseModel):
     r"""Optional model parameters specific to this fallback model. Overrides primary model parameters if this fallback is used."""
 
+    name: Optional[str] = None
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
+
     audio: OptionalNullable[UpdateAgentFallbackModelConfigurationAgentsAudio] = UNSET
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
 
@@ -4567,6 +4590,7 @@ class UpdateAgentFallbackModelConfigurationAgentsParameters(BaseModel):
     def serialize_model(self, handler):
         optional_fields = set(
             [
+                "name",
                 "audio",
                 "frequency_penalty",
                 "max_tokens",

orq_ai_sdk/models/updateconversationop.py

@@ -231,7 +231,7 @@ class UpdateConversationResponseBody(BaseModel):
     r"""Unix timestamp in milliseconds when the conversation was last modified."""
 
     id: Annotated[Optional[str], pydantic.Field(alias="_id")] = (
-        "conv_01kfvjh999wcqyk36wkc4vbjjq"
+        "conv_01kfvn6e02bgrcrkhxtn1kb9g3"
     )
     r"""Unique conversation identifier with `conv_` prefix."""
 

orq_ai_sdk/models/updatedatapointop.py

@@ -1445,7 +1445,7 @@ class UpdateDatapointEvaluations3(BaseModel):
 
     source: Optional[UpdateDatapointEvaluationsDatasetsResponseSource] = "orq"
 
-    reviewed_at: Optional[datetime] = parse_datetime("2026-01-25T21:56:35.391Z")
+    reviewed_at: Optional[datetime] = parse_datetime("2026-01-25T22:43:05.508Z")
     r"""The date and time the item was reviewed"""
 
     @model_serializer(mode="wrap")
@@ -1513,7 +1513,7 @@ class UpdateDatapointEvaluations2(BaseModel):
 
     source: Optional[UpdateDatapointEvaluationsDatasetsSource] = "orq"
 
-    reviewed_at: Optional[datetime] = parse_datetime("2026-01-25T21:56:35.391Z")
+    reviewed_at: Optional[datetime] = parse_datetime("2026-01-25T22:43:05.507Z")
     r"""The date and time the item was reviewed"""
 
     @model_serializer(mode="wrap")
@@ -1581,7 +1581,7 @@ class UpdateDatapointEvaluations1(BaseModel):
 
     source: Optional[UpdateDatapointEvaluationsSource] = "orq"
 
-    reviewed_at: Optional[datetime] = parse_datetime("2026-01-25T21:56:35.390Z")
+    reviewed_at: Optional[datetime] = parse_datetime("2026-01-25T22:43:05.507Z")
     r"""The date and time the item was reviewed"""
 
     @model_serializer(mode="wrap")
@@ -1684,7 +1684,7 @@ class UpdateDatapointResponseBody(BaseModel):
     created: Optional[datetime] = None
     r"""The date and time the resource was created"""
 
-    updated: Optional[datetime] = parse_datetime("2026-01-25T21:56:21.657Z")
+    updated: Optional[datetime] = parse_datetime("2026-01-25T22:42:51.693Z")
     r"""The date and time the resource was last updated"""
 
     @model_serializer(mode="wrap")

orq_ai_sdk/models/updatedatasetop.py

@@ -154,7 +154,7 @@ class UpdateDatasetResponseBody(BaseModel):
     created: Optional[datetime] = None
     r"""The date and time the resource was created"""
 
-    updated: Optional[datetime] = parse_datetime("2026-01-25T21:56:21.657Z")
+    updated: Optional[datetime] = parse_datetime("2026-01-25T22:42:51.693Z")
     r"""The date and time the resource was last updated"""
 
     @model_serializer(mode="wrap")

orq_ai_sdk/models/updatedatasourceop.py

@@ -104,7 +104,7 @@ class UpdateDatasourceResponseBody(BaseModel):
     r"""The number of chunks in the datasource"""
 
     id: Annotated[Optional[str], pydantic.Field(alias="_id")] = (
-        "01KFVJH9S1JQEAHE3SW52D5RXK"
+        "01KFVN6EH2ANE9D6GBKQQW2JCN"
     )
     r"""The unique identifier of the data source"""
 

orq_ai_sdk/models/updateevalop.py

@@ -959,9 +959,9 @@ class ResponseBodyTypescript(BaseModel):
 
     key: str
 
-    created: Optional[str] = "2026-01-25T21:56:23.421Z"
+    created: Optional[str] = "2026-01-25T22:42:53.396Z"
 
-    updated: Optional[str] = "2026-01-25T21:56:23.421Z"
+    updated: Optional[str] = "2026-01-25T22:42:53.396Z"
 
     guardrail_config: Optional[
         UpdateEvalResponseBodyEvalsResponse200ApplicationJSON7GuardrailConfig
@@ -1155,9 +1155,9 @@ class ResponseBodyRagas(BaseModel):
 
     model: str
 
-    created: Optional[str] = "2026-01-25T21:56:23.421Z"
+    created: Optional[str] = "2026-01-25T22:42:53.396Z"
 
-    updated: Optional[str] = "2026-01-25T21:56:23.421Z"
+    updated: Optional[str] = "2026-01-25T22:42:53.396Z"
 
     guardrail_config: Optional[
         UpdateEvalResponseBodyEvalsResponse200ApplicationJSON6GuardrailConfig
@@ -1852,9 +1852,9 @@ class ResponseBodyFunction(BaseModel):
 
     key: str
 
-    created: Optional[str] = "2026-01-25T21:56:23.421Z"
+    created: Optional[str] = "2026-01-25T22:42:53.396Z"
 
-    updated: Optional[str] = "2026-01-25T21:56:23.421Z"
+    updated: Optional[str] = "2026-01-25T22:42:53.396Z"
 
     guardrail_config: Optional[
         UpdateEvalResponseBodyEvalsResponse200ApplicationJSONGuardrailConfig
@@ -2029,9 +2029,9 @@ class UpdateEvalResponseBodyPython(BaseModel):
 
     key: str
 
-    created: Optional[str] = "2026-01-25T21:56:23.421Z"
+    created: Optional[str] = "2026-01-25T22:42:53.396Z"
 
-    updated: Optional[str] = "2026-01-25T21:56:23.421Z"
+    updated: Optional[str] = "2026-01-25T22:42:53.396Z"
 
     guardrail_config: Optional[
         UpdateEvalResponseBodyEvalsResponse200GuardrailConfig
@@ -2213,9 +2213,9 @@ class UpdateEvalResponseBodyHTTP(BaseModel):
 
     key: str
 
-    created: Optional[str] = "2026-01-25T21:56:23.421Z"
+    created: Optional[str] = "2026-01-25T22:42:53.396Z"
 
-    updated: Optional[str] = "2026-01-25T21:56:23.421Z"
+    updated: Optional[str] = "2026-01-25T22:42:53.396Z"
 
     guardrail_config: Optional[UpdateEvalResponseBodyEvalsResponseGuardrailConfig] = (
         None
@@ -2368,9 +2368,9 @@ class UpdateEvalResponseBodyJSON(BaseModel):
 
     key: str
 
-    created: Optional[str] = "2026-01-25T21:56:23.421Z"
+    created: Optional[str] = "2026-01-25T22:42:53.396Z"
 
-    updated: Optional[str] = "2026-01-25T21:56:23.421Z"
+    updated: Optional[str] = "2026-01-25T22:42:53.396Z"
 
     guardrail_config: Optional[UpdateEvalResponseBodyEvalsGuardrailConfig] = None
 
@@ -2522,9 +2522,9 @@ class UpdateEvalResponseBodyLLM(BaseModel):
 
     model: str
 
-    created: Optional[str] = "2026-01-25T21:56:23.421Z"
+    created: Optional[str] = "2026-01-25T22:42:53.396Z"
 
-    updated: Optional[str] = "2026-01-25T21:56:23.421Z"
+    updated: Optional[str] = "2026-01-25T22:42:53.396Z"
 
     guardrail_config: Optional[UpdateEvalResponseBodyGuardrailConfig] = None
 

orq_ai_sdk/models/updateidentityop.py

@@ -194,7 +194,7 @@ class UpdateIdentityResponseBody(BaseModel):
     created: Optional[datetime] = None
     r"""The date and time the resource was created"""
 
-    updated: Optional[datetime] = parse_datetime("2026-01-25T21:56:21.657Z")
+    updated: Optional[datetime] = parse_datetime("2026-01-25T22:42:51.693Z")
     r"""The date and time the resource was last updated"""
 
     @model_serializer(mode="wrap")

orq_ai_sdk/models/updatepromptop.py

@@ -1128,6 +1128,8 @@ class UpdatePromptPromptInputTypedDict(TypedDict):
     r"""Array of messages that make up the conversation. Each message has a role (system, user, assistant, or tool) and content."""
     model: NotRequired[Nullable[str]]
     r"""Model ID used to generate the response, like `openai/gpt-4o` or `anthropic/claude-3-5-sonnet-20241022`. For private models, use format: `{workspaceKey}@{provider}/{model}`. The full list of models can be found at https://docs.orq.ai/docs/ai-gateway-supported-models. Only chat models are supported."""
+    name: NotRequired[str]
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
     audio: NotRequired[Nullable[UpdatePromptAudioTypedDict]]
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
     frequency_penalty: NotRequired[Nullable[float]]
@@ -1203,6 +1205,9 @@ class UpdatePromptPromptInput(BaseModel):
     model: OptionalNullable[str] = UNSET
     r"""Model ID used to generate the response, like `openai/gpt-4o` or `anthropic/claude-3-5-sonnet-20241022`. For private models, use format: `{workspaceKey}@{provider}/{model}`. The full list of models can be found at https://docs.orq.ai/docs/ai-gateway-supported-models. Only chat models are supported."""
 
+    name: Optional[str] = None
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
+
     audio: OptionalNullable[UpdatePromptAudio] = UNSET
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
 
@@ -1300,6 +1305,7 @@ class UpdatePromptPromptInput(BaseModel):
             [
                 "messages",
                 "model",
+                "name",
                 "audio",
                 "frequency_penalty",
                 "max_tokens",
@@ -3503,6 +3509,8 @@ UpdatePromptPromptsResponseMessages = Annotated[
 class UpdatePromptPromptFieldTypedDict(TypedDict):
     r"""Prompt configuration with model and messages. Use this instead of prompt_config."""
 
+    name: NotRequired[str]
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
     audio: NotRequired[Nullable[UpdatePromptPromptsAudioTypedDict]]
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
     frequency_penalty: NotRequired[Nullable[float]]
@@ -3577,6 +3585,9 @@ class UpdatePromptPromptFieldTypedDict(TypedDict):
 class UpdatePromptPromptField(BaseModel):
     r"""Prompt configuration with model and messages. Use this instead of prompt_config."""
 
+    name: Optional[str] = None
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
+
     audio: OptionalNullable[UpdatePromptPromptsAudio] = UNSET
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
 
@@ -3680,6 +3691,7 @@ class UpdatePromptPromptField(BaseModel):
     def serialize_model(self, handler):
         optional_fields = set(
             [
+                "name",
                 "audio",
                 "frequency_penalty",
                 "max_tokens",

orq_ai_sdk/models/updatetoolop.py

@@ -259,7 +259,7 @@ class RequestBodyTools(BaseModel):
 
     schema_: Annotated[UpdateToolRequestBodyToolsSchema, pydantic.Field(alias="schema")]
 
-    id: Optional[str] = "01KFVJH9MMM1CNXGE0VTKKMNRN"
+    id: Optional[str] = "01KFVN6ECAKKZWM7C9VJMS4E18"
 
     description: Optional[str] = None
 
@@ -1214,7 +1214,7 @@ class UpdateToolResponseBodyCodeExecutionTool(BaseModel):
     code_tool: UpdateToolResponseBodyCodeTool
 
     id: Annotated[Optional[str], pydantic.Field(alias="_id")] = (
-        "tool_01KFVJH9MF0CXHV8X272RPNHAF"
+        "tool_01KFVN6EC5MYDCAMFJM4VHKMPY"
     )
 
     display_name: Optional[str] = None
@@ -1345,7 +1345,7 @@ class UpdateToolResponseBodyTools(BaseModel):
         UpdateToolResponseBodyToolsSchema, pydantic.Field(alias="schema")
     ]
 
-    id: Optional[str] = "01KFVJH9ME1ZENDHKDXNW7Z8RS"
+    id: Optional[str] = "01KFVN6EC497801YG8BYE696ZD"
 
     description: Optional[str] = None
 
@@ -1476,7 +1476,7 @@ class UpdateToolResponseBodyMCPTool(BaseModel):
     mcp: UpdateToolResponseBodyMcp
 
     id: Annotated[Optional[str], pydantic.Field(alias="_id")] = (
-        "tool_01KFVJH9MDPNF3TT14QD7FRZHA"
+        "tool_01KFVN6EC35QXNK3VNN8ZREJB2"
     )
 
     display_name: Optional[str] = None
@@ -1777,7 +1777,7 @@ class UpdateToolResponseBodyHTTPTool(BaseModel):
     http: UpdateToolResponseBodyHTTP
 
     id: Annotated[Optional[str], pydantic.Field(alias="_id")] = (
-        "tool_01KFVJH9MB2HNGMN5S4HSZPYJV"
+        "tool_01KFVN6EC13XW1297BB2G6QVF5"
     )
 
     display_name: Optional[str] = None
@@ -1972,7 +1972,7 @@ class UpdateToolResponseBodyJSONSchemaTool(BaseModel):
     json_schema: UpdateToolResponseBodyJSONSchema
 
     id: Annotated[Optional[str], pydantic.Field(alias="_id")] = (
-        "tool_01KFVJH9MAG5B5DMVWMDZWC1AH"
+        "tool_01KFVN6EC06J7W0KTB5SH0JBPX"
     )
 
     display_name: Optional[str] = None
@@ -2171,7 +2171,7 @@ class UpdateToolResponseBodyFunctionTool(BaseModel):
     function: UpdateToolResponseBodyFunction
 
     id: Annotated[Optional[str], pydantic.Field(alias="_id")] = (
-        "tool_01KFVJH9M9G5CPF3JDPCB6TKFG"
+        "tool_01KFVN6EBQNA6CKEPNV8334XCJ"
     )
 
     display_name: Optional[str] = None

orq_ai_sdk/orq_completions.py

@@ -26,6 +26,7 @@ class OrqCompletions(BaseSDK):
         ],
         model: str,
         metadata: Optional[Dict[str, str]] = None,
+        name: Optional[str] = None,
         audio: OptionalNullable[
             Union[
                 models_createchatcompletionop.CreateChatCompletionAudio,
@@ -147,6 +148,7 @@ class OrqCompletions(BaseSDK):
         :param messages: A list of messages comprising the conversation so far.
         :param model: Model ID used to generate the response, like `openai/gpt-4o` or `anthropic/claude-haiku-4-5-20251001`. The AI Gateway offers a wide range of models with different capabilities, performance characteristics, and price points. Refer to the (Supported models)[/docs/proxy/supported-models] to browse available models.
         :param metadata: Set of 16 key-value pairs that can be attached to an object. This can be useful for storing additional information about the object in a structured format. Keys can have a maximum length of 64 characters and values can have a maximum length of 512 characters.
+        :param name: The name to display on the trace. If not specified, the default system name will be used.
         :param audio: Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more.
         :param frequency_penalty: Number between -2.0 and 2.0. Positive values penalize new tokens based on their existing frequency in the text so far, decreasing the model's likelihood to repeat the same line verbatim.
         :param max_tokens: `[Deprecated]`. The maximum number of tokens that can be generated in the chat completion. This value can be used to control costs for text generated via API.
@@ -211,6 +213,7 @@ class OrqCompletions(BaseSDK):
             ),
             model=model,
             metadata=metadata,
+            name=name,
             audio=utils.get_pydantic_model(
                 audio, OptionalNullable[models.CreateChatCompletionAudio]
             ),
@@ -347,6 +350,7 @@ class OrqCompletions(BaseSDK):
         ],
         model: str,
         metadata: Optional[Dict[str, str]] = None,
+        name: Optional[str] = None,
         audio: OptionalNullable[
             Union[
                 models_createchatcompletionop.CreateChatCompletionAudio,
@@ -468,6 +472,7 @@ class OrqCompletions(BaseSDK):
         :param messages: A list of messages comprising the conversation so far.
         :param model: Model ID used to generate the response, like `openai/gpt-4o` or `anthropic/claude-haiku-4-5-20251001`. The AI Gateway offers a wide range of models with different capabilities, performance characteristics, and price points. Refer to the (Supported models)[/docs/proxy/supported-models] to browse available models.
         :param metadata: Set of 16 key-value pairs that can be attached to an object. This can be useful for storing additional information about the object in a structured format. Keys can have a maximum length of 64 characters and values can have a maximum length of 512 characters.
+        :param name: The name to display on the trace. If not specified, the default system name will be used.
         :param audio: Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more.
         :param frequency_penalty: Number between -2.0 and 2.0. Positive values penalize new tokens based on their existing frequency in the text so far, decreasing the model's likelihood to repeat the same line verbatim.
         :param max_tokens: `[Deprecated]`. The maximum number of tokens that can be generated in the chat completion. This value can be used to control costs for text generated via API.
@@ -532,6 +537,7 @@ class OrqCompletions(BaseSDK):
             ),
             model=model,
             metadata=metadata,
+            name=name,
             audio=utils.get_pydantic_model(
                 audio, OptionalNullable[models.CreateChatCompletionAudio]
             ),

orq_ai_sdk/rerank.py

@@ -19,6 +19,37 @@ class Rerank(BaseSDK):
         model: str,
         top_n: Optional[float] = None,
         filename: OptionalNullable[str] = UNSET,
+        name: Optional[str] = None,
+        fallbacks: Optional[
+            Union[
+                List[models_creatererankop.CreateRerankFallbacks],
+                List[models_creatererankop.CreateRerankFallbacksTypedDict],
+            ]
+        ] = None,
+        retry: Optional[
+            Union[
+                models_creatererankop.CreateRerankRetry,
+                models_creatererankop.CreateRerankRetryTypedDict,
+            ]
+        ] = None,
+        cache: Optional[
+            Union[
+                models_creatererankop.CreateRerankCache,
+                models_creatererankop.CreateRerankCacheTypedDict,
+            ]
+        ] = None,
+        load_balancer: Optional[
+            Union[
+                models_creatererankop.CreateRerankLoadBalancer,
+                models_creatererankop.CreateRerankLoadBalancerTypedDict,
+            ]
+        ] = None,
+        timeout: Optional[
+            Union[
+                models_creatererankop.CreateRerankTimeout,
+                models_creatererankop.CreateRerankTimeoutTypedDict,
+            ]
+        ] = None,
         orq: Optional[
             Union[
                 models_creatererankop.CreateRerankOrq,
@@ -39,6 +70,12 @@ class Rerank(BaseSDK):
         :param model: The identifier of the model to use
         :param top_n: The number of most relevant documents or indices to return, defaults to the length of the documents
         :param filename: The filename of the document to rerank
+        :param name: The name to display on the trace. If not specified, the default system name will be used.
+        :param fallbacks: Array of fallback models to use if primary model fails
+        :param retry: Retry configuration for the request
+        :param cache: Cache configuration for the request.
+        :param load_balancer: Load balancer configuration for the request.
+        :param timeout: Timeout configuration to apply to the request. If the request exceeds the timeout, it will be retried or fallback to the next model if configured.
         :param orq:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -64,6 +101,18 @@ class Rerank(BaseSDK):
             model=model,
             top_n=top_n,
             filename=filename,
+            name=name,
+            fallbacks=utils.get_pydantic_model(
+                fallbacks, Optional[List[models.CreateRerankFallbacks]]
+            ),
+            retry=utils.get_pydantic_model(retry, Optional[models.CreateRerankRetry]),
+            cache=utils.get_pydantic_model(cache, Optional[models.CreateRerankCache]),
+            load_balancer=utils.get_pydantic_model(
+                load_balancer, Optional[models.CreateRerankLoadBalancer]
+            ),
+            timeout=utils.get_pydantic_model(
+                timeout, Optional[models.CreateRerankTimeout]
+            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateRerankOrq]),
         )
 
@@ -129,6 +178,37 @@ class Rerank(BaseSDK):
         model: str,
         top_n: Optional[float] = None,
         filename: OptionalNullable[str] = UNSET,
+        name: Optional[str] = None,
+        fallbacks: Optional[
+            Union[
+                List[models_creatererankop.CreateRerankFallbacks],
+                List[models_creatererankop.CreateRerankFallbacksTypedDict],
+            ]
+        ] = None,
+        retry: Optional[
+            Union[
+                models_creatererankop.CreateRerankRetry,
+                models_creatererankop.CreateRerankRetryTypedDict,
+            ]
+        ] = None,
+        cache: Optional[
+            Union[
+                models_creatererankop.CreateRerankCache,
+                models_creatererankop.CreateRerankCacheTypedDict,
+            ]
+        ] = None,
+        load_balancer: Optional[
+            Union[
+                models_creatererankop.CreateRerankLoadBalancer,
+                models_creatererankop.CreateRerankLoadBalancerTypedDict,
+            ]
+        ] = None,
+        timeout: Optional[
+            Union[
+                models_creatererankop.CreateRerankTimeout,
+                models_creatererankop.CreateRerankTimeoutTypedDict,
+            ]
+        ] = None,
         orq: Optional[
             Union[
                 models_creatererankop.CreateRerankOrq,
@@ -149,6 +229,12 @@ class Rerank(BaseSDK):
         :param model: The identifier of the model to use
         :param top_n: The number of most relevant documents or indices to return, defaults to the length of the documents
         :param filename: The filename of the document to rerank
+        :param name: The name to display on the trace. If not specified, the default system name will be used.
+        :param fallbacks: Array of fallback models to use if primary model fails
+        :param retry: Retry configuration for the request
+        :param cache: Cache configuration for the request.
+        :param load_balancer: Load balancer configuration for the request.
+        :param timeout: Timeout configuration to apply to the request. If the request exceeds the timeout, it will be retried or fallback to the next model if configured.
         :param orq:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -174,6 +260,18 @@ class Rerank(BaseSDK):
             model=model,
             top_n=top_n,
             filename=filename,
+            name=name,
+            fallbacks=utils.get_pydantic_model(
+                fallbacks, Optional[List[models.CreateRerankFallbacks]]
+            ),
+            retry=utils.get_pydantic_model(retry, Optional[models.CreateRerankRetry]),
+            cache=utils.get_pydantic_model(cache, Optional[models.CreateRerankCache]),
+            load_balancer=utils.get_pydantic_model(
+                load_balancer, Optional[models.CreateRerankLoadBalancer]
+            ),
+            timeout=utils.get_pydantic_model(
+                timeout, Optional[models.CreateRerankTimeout]
+            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateRerankOrq]),
         )