airia 0.1.35__py3-none-any.whl → 0.1.37__py3-none-any.whl

airia/client/models/async_models.py

@@ -95,6 +95,74 @@ class AsyncModels(BaseModels):
 
         return [ModelItem(**item) for item in resp["items"]]
 
+    async def get_model(
+        self,
+        model_id: str,
+        correlation_id: Optional[str] = None,
+    ) -> ModelItem:
+        """
+        Retrieve a model by its ID.
+
+        This method fetches detailed information about a specific model, including
+        its configuration, pricing, capabilities, and associated project details.
+
+        Args:
+            model_id (str): The unique identifier of the model to retrieve.
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+
+        Returns:
+            ModelItem: A ModelItem object containing comprehensive model information
+                including configuration, pricing, capabilities, and project associations.
+
+        Raises:
+            AiriaAPIError: If the API request fails, including cases where:
+                - Model not found (404)
+                - Authentication fails (401)
+                - Access is forbidden (403)
+                - Server errors (5xx)
+
+        Example:
+            ```python
+            from airia import AiriaAsyncClient
+
+            client = AiriaAsyncClient(api_key="your_api_key")
+
+            # Get a specific model
+            model = await client.models.get_model(
+                model_id="447589d8-f82f-4a0c-ad15-215a2eaee7b8"
+            )
+
+            print(f"Model: {model.display_name}")
+            print(f"Provider: {model.provider}")
+            print(f"Model Name: {model.model_name}")
+            print(f"Has tool support: {model.has_tool_support}")
+            print(f"Has stream support: {model.has_stream_support}")
+
+            # Access pricing information
+            if model.user_provided_details:
+                print(f"Input token price: {model.user_provided_details.input_token_price}")
+                print(f"Output token price: {model.user_provided_details.output_token_price}")
+
+            # Get model with correlation ID for tracking
+            model = await client.models.get_model(
+                model_id="447589d8-f82f-4a0c-ad15-215a2eaee7b8",
+                correlation_id="my-correlation-id"
+            )
+            ```
+
+        Note:
+            The model must be accessible to the authenticated user. Users will only
+            be able to retrieve models they have been granted access to.
+        """
+        request_data = self._pre_get_model(
+            model_id=model_id,
+            correlation_id=correlation_id,
+            api_version=ApiVersion.V1.value,
+        )
+        resp = await self._request_handler.make_request("GET", request_data)
+        return ModelItem(**resp)
+
     async def delete_model(
         self,
         model_id: str,

airia/client/models/base_models.py

@@ -67,6 +67,40 @@ class BaseModels:
 
         return request_data
 
+    def _pre_get_model(
+        self,
+        model_id: str,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V1.value,
+    ):
+        """
+        Prepare request data for retrieving a single model.
+
+        Args:
+            model_id: The ID of the model to retrieve
+            correlation_id: Optional correlation ID for tracing
+            api_version: API version to use for the request
+
+        Returns:
+            RequestData: Prepared request data for the get endpoint
+
+        Raises:
+            ValueError: If an invalid API version is provided
+        """
+        if api_version not in ApiVersion.as_list():
+            raise ValueError(
+                f"Invalid API version: {api_version}. Valid versions are: {', '.join(ApiVersion.as_list())}"
+            )
+
+        url = urljoin(
+            self._request_handler.base_url,
+            f"{api_version}/Models/{model_id}",
+        )
+        request_data = self._request_handler.prepare_request(
+            url, correlation_id=correlation_id
+        )
+        return request_data
+
     def _pre_delete_model(
         self,
         model_id: str,

airia/client/models/sync_models.py

@@ -95,6 +95,74 @@ class Models(BaseModels):
 
         return [ModelItem(**item) for item in resp["items"]]
 
+    def get_model(
+        self,
+        model_id: str,
+        correlation_id: Optional[str] = None,
+    ) -> ModelItem:
+        """
+        Retrieve a model by its ID.
+
+        This method fetches detailed information about a specific model, including
+        its configuration, pricing, capabilities, and associated project details.
+
+        Args:
+            model_id (str): The unique identifier of the model to retrieve.
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+
+        Returns:
+            ModelItem: A ModelItem object containing comprehensive model information
+                including configuration, pricing, capabilities, and project associations.
+
+        Raises:
+            AiriaAPIError: If the API request fails, including cases where:
+                - Model not found (404)
+                - Authentication fails (401)
+                - Access is forbidden (403)
+                - Server errors (5xx)
+
+        Example:
+            ```python
+            from airia import AiriaClient
+
+            client = AiriaClient(api_key="your_api_key")
+
+            # Get a specific model
+            model = client.models.get_model(
+                model_id="447589d8-f82f-4a0c-ad15-215a2eaee7b8"
+            )
+
+            print(f"Model: {model.display_name}")
+            print(f"Provider: {model.provider}")
+            print(f"Model Name: {model.model_name}")
+            print(f"Has tool support: {model.has_tool_support}")
+            print(f"Has stream support: {model.has_stream_support}")
+
+            # Access pricing information
+            if model.user_provided_details:
+                print(f"Input token price: {model.user_provided_details.input_token_price}")
+                print(f"Output token price: {model.user_provided_details.output_token_price}")
+
+            # Get model with correlation ID for tracking
+            model = client.models.get_model(
+                model_id="447589d8-f82f-4a0c-ad15-215a2eaee7b8",
+                correlation_id="my-correlation-id"
+            )
+            ```
+
+        Note:
+            The model must be accessible to the authenticated user. Users will only
+            be able to retrieve models they have been granted access to.
+        """
+        request_data = self._pre_get_model(
+            model_id=model_id,
+            correlation_id=correlation_id,
+            api_version=ApiVersion.V1.value,
+        )
+        resp = self._request_handler.make_request("GET", request_data)
+        return ModelItem(**resp)
+
     def delete_model(
         self,
         model_id: str,

airia/types/api/pipelines_config/__init__.py

@@ -24,13 +24,17 @@ from .export_pipeline_definition import (
     AgentDetailItemDefinition,
     ExportPipelineDefinitionResponse,
     ExportChunkingConfig,
+    ExportConditionalBranchConfigDefinition,
     ExportCredentialDataList,
     ExportCredentials,
+    ExportCustomCredentials,
     ExportDataSource,
     ExportDataSourceFile,
     ExportDependency,
     ExportDeployment,
     ExportHandle,
+    ExportJsonFormatterStepConfiguration,
+    ExportLoopStepConfiguration,
     ExportMemory,
     ExportMetadata,
     ExportModel,
@@ -38,56 +42,68 @@ from .export_pipeline_definition import (
     ExportPipelineStep,
     ExportPosition,
     ExportPrompt,
-    ExportPromptMessageList,
     ExportPythonCodeBlock,
+    ExportRetryConfiguration,
     ExportRouter,
     ExportRouterConfig,
+    ExportSDKStepCredentialReference,
+    ExportSDKStepDefinition,
     ExportTool,
     ExportToolHeaders,
     ExportToolParameters,
     ExportUserPrompt,
+    ExportVectorStore,
+    ExportWebhookApprovalRequest,
 )
 
 __all__ = [
     "AboutDeploymentMetadata",
     "AgentDetailsEntry",
+    "AgentDetailItemDefinition",
     "AgentTrigger",
     "Deployment",
     "DeploymentAssignment",
     "DeploymentUserPrompt",
-    "ExportPipelineDefinitionResponse",
-    "GetPipelinesConfigResponse",
-    "Pipeline",
-    "PipelineConfigItem",
-    "PipelineConfigResponse",
-    "PipelineExecutionStats",
-    "PipelineStep",
-    "PipelineStepDependency",
-    "PipelineStepHandle",
-    "PipelineStepPosition",
-    "PipelineVersion",
-    "AgentDetailItemDefinition",
     "ExportChunkingConfig",
+    "ExportConditionalBranchConfigDefinition",
     "ExportCredentialDataList",
     "ExportCredentials",
+    "ExportCustomCredentials",
     "ExportDataSource",
     "ExportDataSourceFile",
     "ExportDependency",
     "ExportDeployment",
     "ExportHandle",
+    "ExportJsonFormatterStepConfiguration",
+    "ExportLoopStepConfiguration",
     "ExportMemory",
     "ExportMetadata",
     "ExportModel",
     "ExportPipeline",
+    "ExportPipelineDefinitionResponse",
     "ExportPipelineStep",
     "ExportPosition",
     "ExportPrompt",
-    "ExportPromptMessageList",
     "ExportPythonCodeBlock",
+    "ExportRetryConfiguration",
     "ExportRouter",
     "ExportRouterConfig",
+    "ExportSDKStepCredentialReference",
+    "ExportSDKStepDefinition",
     "ExportTool",
     "ExportToolHeaders",
     "ExportToolParameters",
     "ExportUserPrompt",
+    "ExportVectorStore",
+    "ExportWebhookApprovalRequest",
+    "GetPipelinesConfigResponse",
+    "Pipeline",
+    "PipelineConfigItem",
+    "PipelineConfigResponse",
+    "PipelineExecutionStats",
+    "PipelineStep",
+    "PipelineStepDependency",
+    "PipelineStepHandle",
+    "PipelineStepPosition",
+    "PipelineVersion",
 ]

airia/types/api/pipelines_config/export_pipeline_definition.py

@@ -25,6 +25,26 @@ class ExportCredentialDataList(BaseModel):
     value: str = Field(..., description="Gets or sets the value.")
 
 
+class ExportCustomCredentials(BaseModel):
+    """Represents custom credentials definition.
+
+    Attributes:
+        id: The unique identifier
+        name: The name of the credential
+        credential_type: The type of credential
+        credential_data: Optional credential data
+    """
+
+    id: str = Field(..., description="Gets or sets the ID.")
+    name: str = Field(..., description="Gets or sets the name.")
+    credential_type: str = Field(
+        ..., description="Gets or sets the credential type.", alias="credentialType"
+    )
+    credential_data: Optional[str] = Field(
+        None, description="Gets or sets the credential data.", alias="credentialData"
+    )
+
+
 class ExportCredentials(BaseModel):
     """Represents credential information for external service authentication.
 
@@ -36,9 +56,9 @@ class ExportCredentials(BaseModel):
         credential_type: The type of credential (API key, OAuth, etc.)
         source_type: The source where the credential originates
         credential_data_list: List of key-value pairs containing credential data
-        display_identifier_name: The display name for the credential identifier
-        administrative_scope: The administrative scope of the credential
-        origin: The origin of the credential
+        display_identifier_name: Optional display name for the credential identifier
+        administrative_scope: Optional administrative scope of the credential
+        origin: Optional origin of the credential
         custom_credentials: Optional custom credentials configuration
         id: The unique identifier for the credential set
     """
@@ -55,18 +75,20 @@ class ExportCredentials(BaseModel):
         description="Gets or sets the credential data list.",
         alias="credentialDataList",
     )
-    display_identifier_name: str = Field(
-        ...,
+    display_identifier_name: Optional[str] = Field(
+        None,
         description="Gets or sets the display identifier name.",
         alias="displayIdentifierName",
     )
-    administrative_scope: str = Field(
-        ...,
+    administrative_scope: Optional[str] = Field(
+        None,
         description="Gets or sets the administrative scope.",
         alias="administrativeScope",
     )
-    origin: str = Field(..., description="Gets or sets the origin.")
-    custom_credentials: Optional[dict] = Field(
+    origin: Optional[str] = Field(
+        None, description="Gets or sets the origin."
+    )
+    custom_credentials: Optional[ExportCustomCredentials] = Field(
         None,
         description="Gets or sets the custom credentials.",
         alias="customCredentials",
@@ -155,6 +177,95 @@ class ExportDependency(BaseModel):
     )
 
 
+class ExportLoopStepConfiguration(BaseModel):
+    """Represents loop container step configuration.
+
+    Attributes:
+        stop_on_error: Whether to stop on error
+        max_iterations: The maximum number of iterations
+    """
+
+    stop_on_error: bool = Field(
+        ...,
+        description="Gets a value indicating whether the pipeline execution should stop if an error occurs during loop execution.",
+        alias="stopOnError",
+    )
+    max_iterations: int = Field(
+        ...,
+        description="Gets the maximum number of iterations for the loop.",
+        alias="maxIterations",
+    )
+
+
+class ExportJsonFormatterStepConfiguration(BaseModel):
+    """Represents JSON formatter step configuration.
+
+    Attributes:
+        template: The JSON formatter template
+    """
+
+    template: str = Field(
+        ..., description="Gets the JSON formatter template."
+    )
+
+
+class ExportConditionalBranchConfigDefinition(BaseModel):
+    """Represents conditional branch configuration definition.
+
+    Attributes:
+        id: The unique identifier
+        label: Optional route label
+    """
+
+    id: str = Field(..., description="Gets or sets the Id.")
+    label: Optional[str] = Field(
+        None,
+        description="Gets or sets the route label (e.g., \"Route 1\", \"High Priority\").",
+    )
+
+
+class ExportRetryConfiguration(BaseModel):
+    """Represents retry configuration for step execution.
+
+    Attributes:
+        enabled: Whether retries are enabled
+        max_retries: The maximum number of retry attempts
+        delay: The base delay between retries in milliseconds
+        max_delay: The maximum delay between retries in milliseconds
+        backoff_multiplier: The backoff multiplier for exponential backoff
+        use_jitter: Whether to use jitter to randomize retry delays
+    """
+
+    enabled: bool = Field(
+        ...,
+        description="Gets a value indicating whether retries are enabled or disabled.",
+    )
+    max_retries: int = Field(
+        ...,
+        description="Gets the maximum number of retry attempts.",
+        alias="maxRetries",
+    )
+    delay: int = Field(
+        ...,
+        description="Gets the base delay between retries in milliseconds.",
+    )
+    max_delay: int = Field(
+        ...,
+        description="Gets the maximum delay between retries in milliseconds.",
+        alias="maxDelay",
+    )
+    backoff_multiplier: float = Field(
+        ...,
+        description="Gets the backoff multiplier for exponential backoff.",
+        alias="backoffMultiplier",
+    )
+    use_jitter: bool = Field(
+        ...,
+        description="Gets a value indicating whether to use jitter to randomize retry delays.",
+        alias="useJitter",
+    )
+
+
 class ExportPipelineStep(BaseModel):
     """Represents a complete pipeline step definition for export.
 
@@ -202,11 +313,21 @@ class ExportPipelineStep(BaseModel):
     temperature: Optional[float] = Field(
         None, description="Gets or sets the temperature."
     )
+    reasoning_effort: Optional[str] = Field(
+        None,
+        description="Gets or sets the reasoning effort level for OpenAI reasoning models.",
+        alias="reasoningEffort",
+    )
     include_date_time_context: Optional[bool] = Field(
         None,
         description="Gets or sets a value indicating whether to include date time context.",
         alias="includeDateTimeContext",
     )
+    include_user_details_context: Optional[bool] = Field(
+        None,
+        description="Gets or sets a value indicating whether to include user details context.",
+        alias="includeUserDetailsContext",
+    )
     prompt_id: Optional[str] = Field(
         None, description="Gets or sets the prompt ID.", alias="promptId"
     )
@@ -227,7 +348,7 @@ class ExportPipelineStep(BaseModel):
     top_k: Optional[int] = Field(
         None, description="Gets or sets the top K.", alias="topK"
     )
-    relevance_threshold: Optional[int] = Field(
+    relevance_threshold: Optional[float] = Field(
         None,
         description="Gets or sets the relevance threshold.",
         alias="relevanceThreshold",
@@ -245,12 +366,48 @@ class ExportPipelineStep(BaseModel):
     database_type: Optional[str] = Field(
         None, description="Gets or sets the database type.", alias="databaseType"
     )
+    enable_sql_queries: Optional[bool] = Field(
+        None,
+        description="Gets or sets a value indicating whether SQL queries are enabled.",
+        alias="enableSqlQueries",
+    )
+    fuzzy_search: Optional[bool] = Field(
+        None,
+        description="Gets or sets a value indicating whether fuzzy search is enabled.",
+        alias="fuzzySearch",
+    )
+    sql_result_limit: Optional[int] = Field(
+        None,
+        description="Gets or sets the maximum number of rows to return from SQL queries.",
+        alias="sqlResultLimit",
+    )
+    hybrid_search_fusion_type: Optional[str] = Field(
+        None,
+        description="Gets or sets the hybrid search fusion type.",
+        alias="hybridSearchFusionType",
+    )
+    user_input_on_vector_search: Optional[bool] = Field(
+        None,
+        description="Gets or sets a value indicating whether user input is considered during vector search.",
+        alias="userInputOnVectorSearch",
+    )
     memory_id: Optional[str] = Field(
         None, description="Gets or sets the memory.", alias="memoryId"
     )
     python_code_block_id: Optional[str] = Field(
         None, description="Gets or sets the code.", alias="pythonCodeBlockId"
     )
+    approval_request_id: Optional[str] = Field(
+        None, description="Gets or sets the approval request ID.", alias="approvalRequestId"
+    )
+    webhook_approval_request_id: Optional[str] = Field(
+        None,
+        description="Gets or sets the webhook approval request ID.",
+        alias="webhookApprovalRequestId",
+    )
+    agent_card_id: Optional[str] = Field(
+        None, description="Gets or sets the A2A Card ID.", alias="agentCardId"
+    )
     background_color: Optional[str] = Field(
         None, description="Gets or sets the backgroundColor.", alias="backgroundColor"
     )
@@ -258,22 +415,142 @@ class ExportPipelineStep(BaseModel):
     step_title: str = Field(
         ..., description="Gets or sets the Step Title.", alias="stepTitle"
     )
+    parent_id: Optional[str] = Field(
+        None,
+        description="Gets or sets the identifier of the parent step that this step belongs to.",
+        alias="parentId",
+    )
     width: Optional[int] = Field(
-        None, description="Gets or sets the width of the step."
+        None, description="Gets or sets the width of the step in the pipeline UI."
     )
     height: Optional[int] = Field(
-        None, description="Gets or sets the height of the step."
+        None, description="Gets or sets the height of the step in the pipeline UI."
     )
-    ai_operation_step_include_chat_history: Optional[bool] = Field(
+    loop_step_configuration: Optional[ExportLoopStepConfiguration] = Field(
         None,
-        description="Gets or sets whether to include chat history in AI operations.",
-        alias="aiOperationStepIncludeChatHistory",
+        description="Gets or sets the loop step configuration.",
+        alias="loopStepConfiguration",
+    )
+    json_formatter_step_configuration: Optional[ExportJsonFormatterStepConfiguration] = Field(
+        None,
+        description="Gets or sets the JSON formatter step configuration.",
+        alias="jsonFormatterStepConfiguration",
     )
     input_variables_json: Optional[str] = Field(
         None,
-        description="Gets or sets the input variables as JSON string.",
+        description="Gets or sets a dictionary containing input variable configuration as a json.",
         alias="inputVariablesJson",
     )
+    input_template: Optional[str] = Field(
+        None,
+        description="Gets or sets the input template that will be used with the input variables.",
+        alias="inputTemplate",
+    )
+    include_chat_history: Optional[bool] = Field(
+        None,
+        description="Gets a value indicating whether to include chat history.",
+        alias="includeChatHistory",
+    )
+    is_multi_route: Optional[bool] = Field(
+        None,
+        description="Gets a value indicating whether the router is multi-route or single route.",
+        alias="isMultiRoute",
+    )
+    router_config_json: Optional[str] = Field(
+        None,
+        description="Gets the Router Configuration in JSON format.",
+        alias="routerConfigJson",
+    )
+    conditional_branch_code: Optional[str] = Field(
+        None,
+        description="Gets the Python code to be executed in this step.",
+        alias="conditionalBranchCode",
+    )
+    conditional_branch_config_json: Optional[str] = Field(
+        None,
+        description="Gets or sets the conditional branch configuration in JSON format.",
+        alias="conditionalBranchConfigJson",
+    )
+    router_config: Optional[Dict[str, ExportConditionalBranchConfigDefinition]] = Field(
+        None,
+        description="Gets the Router Configuration.",
+        alias="routerConfig",
+    )
+    selected_prompt_version: Optional[str] = Field(
+        None,
+        description="Gets the Selected Prompt Version.",
+        alias="selectedPromptVersion",
+    )
+    step_params: Optional[Dict[str, str]] = Field(
+        None,
+        description="Gets a dictionary containing step parameters, where the key is the parameter name and the value is the parameter value.",
+        alias="stepParams",
+    )
+    ai_operation_step_include_chat_history: Optional[bool] = Field(
+        None,
+        description="Gets a value indicating whether to include the chat history in AI Operation step.",
+        alias="aiOperationStepIncludeChatHistory",
+    )
+    ai_operation_step_include_attachments: Optional[bool] = Field(
+        None,
+        description="Gets a value indicating whether to include the attachments in AI Operation step.",
+        alias="aiOperationStepIncludeAttachments",
+    )
+    ai_operation_step_use_data_source: Optional[bool] = Field(
+        None,
+        description="Gets a value indicating whether to use data sources during agent execution in AI Operation step.",
+        alias="aiOperationStepUseDataSource",
+    )
+    selected_data_sources: Optional[List[str]] = Field(
+        None,
+        description="Gets the selected data sources (data stores) to be used during agent execution in AI Operation step.",
+        alias="selectedDataSources",
+    )
+    memory_store_step_append_text: Optional[bool] = Field(
+        None,
+        description="Gets a value indicating whether to append the text from previous step/steps into the Memory step.",
+        alias="memoryStoreStepAppendText",
+    )
+    ai_operation_step_chat_history_messages_limit: Optional[int] = Field(
+        None,
+        description="Gets or sets the maximum number of messages to include from the chat history.",
+        alias="aiOperationStepChatHistoryMessagesLimit",
+    )
+    ai_operation_step_output_schema_json: Optional[str] = Field(
+        None,
+        description="Gets or sets the output schema in JSON format.",
+        alias="aiOperationStepOutputSchemaJson",
+    )
+    execute_pipeline_id: Optional[str] = Field(
+        None,
+        description="Gets or sets the ID of the pipeline to be executed for ExecutePipelineStep.",
+        alias="executePipelineId",
+    )
+    selected_pipeline_version: Optional[str] = Field(
+        None,
+        description="Gets or sets the selected pipeline version to be executed for ExecutePipelineStep.",
+        alias="selectedPipelineVersion",
+    )
+    embedded_agent_variables_json: Optional[str] = Field(
+        None,
+        description="Gets or sets the embedded agent variables in JSON format for ExecutePipelineStep.",
+        alias="embeddedAgentVariablesJson",
+    )
+    preferred_pdf_parser_id: Optional[str] = Field(
+        None,
+        description="Gets or sets the preferred PDF parser ID for document ingestion.",
+        alias="preferredPdfParserId",
+    )
+    custom_pdf_parser_prompt: Optional[str] = Field(
+        None,
+        description="Gets or sets the custom prompt for the PDF parser.",
+        alias="customPdfParserPrompt",
+    )
+    retry_configuration: Optional[ExportRetryConfiguration] = Field(
+        None,
+        description="Gets or sets the retry configuration for step execution.",
+        alias="retryConfiguration",
+    )
 
 
 class AgentDetailItemDefinition(BaseModel):
@@ -590,33 +867,17 @@ class ExportDataSource(BaseModel):
     )
 
 
-class ExportPromptMessageList(BaseModel):
-    """Represents a single message within a prompt sequence.
-
-    Defines individual messages that make up a prompt, including
-    content and ordering information.
-
-    Attributes:
-        text: The content of the message
-        order: The order of this message in the prompt sequence
-    """
-
-    text: str = Field(..., description="Gets or sets the text.")
-    order: int = Field(..., description="Gets or sets the order.")
-
-
 class ExportPrompt(BaseModel):
     """Represents a complete prompt definition for export.
 
     Contains all information for a prompt including name, version information,
-    and the sequence of messages that make up the prompt.
+    and the prompt message.
 
     Attributes:
         name: The name of the prompt
         version_change_description: Description of changes in this version
         prompt_message: Optional consolidated prompt message text
         is_agent_specific: Whether the prompt is specific to an agent
-        prompt_message_list: List of messages that make up the prompt
         id: The unique identifier of the prompt
     """
 
@@ -628,7 +889,7 @@ class ExportPrompt(BaseModel):
     )
     prompt_message: Optional[str] = Field(
         None,
-        description="Gets or sets the consolidated prompt message.",
+        description="Gets or sets the prompt message.",
         alias="promptMessage",
     )
     is_agent_specific: Optional[bool] = Field(
@@ -636,11 +897,6 @@ class ExportPrompt(BaseModel):
         description="Gets or sets whether the prompt is agent specific.",
         alias="isAgentSpecific",
     )
-    prompt_message_list: Optional[List[ExportPromptMessageList]] = Field(
-        None,
-        description="Gets or sets the prompt message list.",
-        alias="promptMessageList",
-    )
     id: str = Field(..., description="Gets or sets the ID.")
 
 
@@ -768,13 +1024,16 @@ class ExportTool(BaseModel):
     method_type: str = Field(
         ..., description="Gets or sets the method type.", alias="methodType"
     )
-    route_through_acc: bool = Field(
-        ...,
+    route_through_acc: Optional[bool] = Field(
+        None,
         description="Gets or sets a value indicating whether the tool should route through the ACC.",
         alias="routeThroughACC",
     )
     acc_group: Optional[str] = Field(
-        None, description="Gets or sets the access control group.", alias="accGroup"
+        None, description="Gets or sets the ACC group.", alias="accGroup"
+    )
+    icon_url: Optional[str] = Field(
+        None, description="Gets or sets the Icon URL.", alias="iconUrl"
     )
     use_user_credentials: bool = Field(
         ...,
@@ -786,16 +1045,18 @@ class ExportTool(BaseModel):
         description="Gets or sets a value indicating what the credential type is when the tool use user based credentials.",
         alias="useUserCredentialsType",
     )
-    provider: str = Field(..., description="Gets or sets the provider.")
-    body_type: str = Field(
-        ..., description="Gets or sets the body type.", alias="bodyType"
+    provider: Optional[str] = Field(
+        None, description="Gets or sets the provider."
+    )
+    body_type: Optional[str] = Field(
+        None, description="Gets or sets the body type.", alias="bodyType"
     )
-    number_of_pages: Optional[int] = Field(
+    number_of_pages: Optional[str] = Field(
         None, description="Gets or sets the number of pages.", alias="numberOfPages"
     )
-    should_retrieve_full_page_content: Optional[bool] = Field(
+    should_retrieve_full_page_content: Optional[str] = Field(
         None,
-        description="Gets or sets whether to retrieve full page content.",
+        description="Gets or sets should retrieve full page content.",
         alias="shouldRetrieveFullPageContent",
     )
     schema_definition: Optional[str] = Field(
@@ -819,6 +1080,11 @@ class ExportTool(BaseModel):
         description="Gets or sets the credentials source type.",
         alias="credentialsSourceType",
     )
+    use_jwt_forwarding: Optional[bool] = Field(
+        None,
+        description="Gets or sets a value indicating whether the tool should use JWT forwarding for authentication.",
+        alias="useJwtForwarding",
+    )
     id: str = Field(..., description="Gets or sets the ID.")
 
 
@@ -955,6 +1221,7 @@ class ExportMemory(BaseModel):
         id: The unique identifier of the memory
         name: The name of the memory
         is_user_specific: Whether the memory is specific to individual users
+        is_conversation_scoped: Optional flag indicating if memory is conversation specific
     """
 
     id: str = Field(..., description="Gets or sets the memory id.")
@@ -964,6 +1231,11 @@ class ExportMemory(BaseModel):
         description="Gets or sets a value indicating whether the memory is user specific.",
         alias="isUserSpecific",
     )
+    is_conversation_scoped: Optional[bool] = Field(
+        None,
+        description="Gets or sets a value indicating whether the memory is conversation specific.",
+        alias="isConversationScoped",
+    )
 
 
 class ExportPythonCodeBlock(BaseModel):
@@ -1103,8 +1375,8 @@ class ExportQuickAction(BaseModel):
     quick_action_type: str = Field(
         ..., description="Gets or sets the quick action type.", alias="quickActionType"
     )
-    assigned_agent: ExportAssignedAgent = Field(
-        ..., description="Gets or sets the assigned agent.", alias="assignedAgent"
+    assigned_agent: Optional[ExportAssignedAgent] = Field(
+        None, description="Gets or sets the assigned agent.", alias="assignedAgent"
     )
 
 
@@ -1268,10 +1540,14 @@ class ExportMetadata(BaseModel):
     video_link: Optional[str] = Field(
         None, description="Gets or sets the video link.", alias="videoLink"
     )
-    readiness: str = Field(..., description="Gets or sets the readiness status.")
-    department: str = Field(..., description="Gets or sets the department.")
-    agent_last_updated: str = Field(
-        ...,
+    readiness: Optional[str] = Field(
+        None, description="Gets or sets the readiness level."
+    )
+    department: Optional[str] = Field(
+        None, description="Gets or sets the department."
+    )
+    agent_last_updated: Optional[str] = Field(
+        None,
         description="Gets or sets the timestamp when the agent was last updated.",
         alias="agentLastUpdated",
     )
@@ -1316,6 +1592,47 @@ class ExportApprovalRequest(BaseModel):
     )
 
 
+class ExportWebhookApprovalRequest(BaseModel):
+    """Represents a webhook approval request for export.
+
+    Attributes:
+        id: The unique identifier
+        webhook_url: The webhook URL where the approval request will be sent
+        timeout_seconds: The timeout in seconds for the webhook request
+        max_retry_attempts: The maximum number of retry attempts
+        message: Optional message to include in the webhook payload
+        approved_handle_id: The approved handle ID
+        denied_handle_id: The denied handle ID
+    """
+
+    id: str = Field(..., description="Gets or sets the ID.")
+    webhook_url: str = Field(
+        ...,
+        description="Gets or sets the webhook URL where the approval request will be sent.",
+        alias="webhookUrl",
+    )
+    timeout_seconds: int = Field(
+        ...,
+        description="Gets or sets the timeout in seconds for the webhook request (default: 300 seconds / 5 minutes).",
+        alias="timeoutSeconds",
+    )
+    max_retry_attempts: int = Field(
+        ...,
+        description="Gets or sets the maximum number of retry attempts for the webhook request (default: 3).",
+        alias="maxRetryAttempts",
+    )
+    message: Optional[str] = Field(
+        None,
+        description="Gets or sets the optional Message to include in the webhook payload.",
+    )
+    approved_handle_id: str = Field(
+        ..., description="Gets the approved handle identifier.", alias="approvedHandleId"
+    )
+    denied_handle_id: str = Field(
+        ..., description="Gets the denied handle identifier.", alias="deniedHandleId"
+    )
+
+
 class ExportAgentCardProvider(BaseModel):
     """Represents a provider for an agent card.
 
@@ -1357,7 +1674,7 @@ class ExportAgentCardAuthentication(BaseModel):
     """
 
     schemes: List[str] = Field(..., description="Gets or sets the schemes.")
-    credentials: Optional[dict] = Field(
+    credentials: Optional[str] = Field(
         None, description="Gets or sets the credentials."
     )
 
@@ -1371,20 +1688,20 @@ class ExportAgentCardSkill(BaseModel):
         description: Optional description
         tags: Optional tags
         examples: Optional examples
-        input_modes: List of input modes
-        output_modes: List of output modes
+        input_modes: Optional list of input modes
+        output_modes: Optional list of output modes
     """
 
     id: str = Field(..., description="Gets or sets the ID.")
     name: str = Field(..., description="Gets or sets the name.")
     description: Optional[str] = Field(None, description="Gets or sets the description.")
-    tags: Optional[str] = Field(None, description="Gets or sets the tags.")
-    examples: Optional[str] = Field(None, description="Gets or sets the examples.")
-    input_modes: List[str] = Field(
-        ..., description="Gets or sets the input modes.", alias="inputModes"
+    tags: Optional[List[str]] = Field(None, description="Gets or sets the tags.")
+    examples: Optional[List[str]] = Field(None, description="Gets or sets the examples.")
+    input_modes: Optional[List[str]] = Field(
+        None, description="Gets or sets the input modes.", alias="inputModes"
     )
-    output_modes: List[str] = Field(
-        ..., description="Gets or sets the output modes.", alias="outputModes"
+    output_modes: Optional[List[str]] = Field(
+        None, description="Gets or sets the output modes.", alias="outputModes"
     )
 
 
@@ -1410,20 +1727,22 @@ class ExportAgentCard(BaseModel):
         ..., description="Gets or sets the agent card ID.", alias="agentCardId"
     )
     name: str = Field(..., description="Gets or sets the name.")
-    description: str = Field(..., description="Gets or sets the description.")
+    description: Optional[str] = Field(
+        None, description="Gets or sets the description."
+    )
     url: str = Field(..., description="Gets or sets the URL.")
-    provider: ExportAgentCardProvider = Field(
-        ..., description="Gets or sets the provider."
+    provider: Optional[ExportAgentCardProvider] = Field(
+        None, description="Gets or sets the provider."
     )
     version: str = Field(..., description="Gets or sets the version.")
     documentation_url: Optional[str] = Field(
         None, description="Gets or sets the documentation URL.", alias="documentationUrl"
     )
-    capabilities: ExportAgentCardCapabilities = Field(
-        ..., description="Gets or sets the capabilities."
+    capabilities: Optional[ExportAgentCardCapabilities] = Field(
+        None, description="Gets or sets the capabilities."
     )
-    authentication: ExportAgentCardAuthentication = Field(
-        ..., description="Gets or sets the authentication."
+    authentication: Optional[ExportAgentCardAuthentication] = Field(
+        None, description="Gets or sets the authentication."
     )
     default_input_modes: List[str] = Field(
         ..., description="Gets or sets the default input modes.", alias="defaultInputModes"
@@ -1454,16 +1773,190 @@ class ExportEmailInbox(BaseModel):
     )
 
 
+class ExportMCPAnnotations(BaseModel):
+    """Represents MCP annotations for a tool interface.
+
+    Attributes:
+        destructive_hint: Optional hint indicating if tool has destructive side effects
+        idempotent_hint: Optional hint indicating if tool is idempotent
+        open_world_hint: Optional hint indicating if tool uses open-world knowledge
+        read_only_hint: Optional hint indicating if tool is read-only
+        title: Optional tool title
+    """
+
+    destructive_hint: Optional[bool] = Field(
+        None,
+        description="Gets or sets a value indicating whether the tool has destructive side effects.",
+        alias="destructiveHint",
+    )
+    idempotent_hint: Optional[bool] = Field(
+        None,
+        description="Gets or sets a value indicating whether the tool is idempotent.",
+        alias="idempotentHint",
+    )
+    open_world_hint: Optional[bool] = Field(
+        None,
+        description="Gets or sets a value indicating whether the tool uses open-world knowledge.",
+        alias="openWorldHint",
+    )
+    read_only_hint: Optional[bool] = Field(
+        None,
+        description="Gets or sets a value indicating whether the tool is read-only.",
+        alias="readOnlyHint",
+    )
+    title: Optional[str] = Field(
+        None, description="Gets or sets the tool title."
+    )
+
+
+class ExportToolInterface(BaseModel):
+    """Represents a tool interface definition for agent export.
+
+    Attributes:
+        name: The tool name (used as function name)
+        title: Optional title for the tool
+        description: The tool description
+        input_schema: JSON schema for input parameters
+        output_schema: Optional JSON schema for output structure
+        enabled: Whether the tool interface is enabled
+        annotations: Optional MCP annotations for the tool
+    """
+
+    name: str = Field(
+        ..., description="Gets or sets the tool name (used as function name)."
+    )
+    title: Optional[str] = Field(
+        None, description="Gets or sets the optional title for the tool (display name)."
+    )
+    description: str = Field(
+        ..., description="Gets or sets the tool description (shown to models)."
+    )
+    input_schema: str = Field(
+        ...,
+        description="Gets or sets the JSON schema for input parameters (required).",
+        alias="inputSchema",
+    )
+    output_schema: Optional[str] = Field(
+        None,
+        description="Gets or sets the JSON schema for output structure (optional).",
+        alias="outputSchema",
+    )
+    enabled: bool = Field(
+        ...,
+        description="Gets or sets a value indicating whether the tool interface is enabled.",
+    )
+    annotations: Optional[ExportMCPAnnotations] = Field(
+        None, description="Gets or sets the MCP annotations for the tool."
+    )
+
+
 class ExportInterfaces(BaseModel):
     """Represents interfaces configuration.
 
     Attributes:
         email_inbox: Optional email inbox configuration
+        tool_interface: Optional tool interface configuration
     """
 
     email_inbox: Optional[ExportEmailInbox] = Field(
         None, description="Gets or sets the email inbox.", alias="emailInbox"
     )
+    tool_interface: Optional[ExportToolInterface] = Field(
+        None, description="Gets or sets the tool interface.", alias="toolInterface"
+    )
+
+
+class ExportSDKStepCredentialReference(BaseModel):
+    """Represents a credential reference within an SDK step.
+
+    Attributes:
+        property_name: The property name in the parameters class
+        original_credential_id: The original credential ID from the exported environment
+        credentials_definition: The credential definition
+        required: Whether this credential is required
+        supported_types: The supported credential types for this reference
+    """
+
+    property_name: str = Field(
+        ...,
+        description="Gets or sets the property name in the parameters class (e.g., \"CredentialId\").",
+        alias="propertyName",
+    )
+    original_credential_id: str = Field(
+        ...,
+        description="Gets or sets the original credential ID from the exported environment.",
+        alias="originalCredentialId",
+    )
+    credentials_definition: ExportCredentials = Field(
+        ...,
+        description="Gets or sets the credential definition (name, type, encrypted/placeholder values).",
+        alias="credentialsDefinition",
+    )
+    required: bool = Field(
+        ...,
+        description="Gets or sets a value indicating whether this credential is required by the SDK step.",
+    )
+    supported_types: List[str] = Field(
+        ...,
+        description="Gets or sets the supported credential types for this reference.",
+        alias="supportedTypes",
+    )
+
+
+class ExportSDKStepDefinition(BaseModel):
+    """Represents an SDK step definition for export/import.
+
+    Attributes:
+        id: The unique identifier of the SDK step
+        name: The name of the SDK step
+        sdk_step_type: The type of SDK step
+        category: The SDK step category
+        group: The SDK step group
+        version: The version of the SDK step definition
+        step_parameters: The serialized step parameters (JSON)
+        credential_references: The credential references used by this SDK step
+        supported_credential_types: The supported credential types for this step
+    """
+
+    id: str = Field(
+        ...,
+        description="Gets or sets the unique identifier of the SDK step.",
+    )
+    name: str = Field(
+        ..., description="Gets or sets the name of the SDK step."
+    )
+    sdk_step_type: str = Field(
+        ...,
+        description="Gets or sets the type of SDK step (e.g., \"HttpRequest\", \"OneDrive_ListItems\").",
+        alias="sdkStepType",
+    )
+    category: str = Field(
+        ...,
+        description="Gets or sets the SDK step category (e.g., \"Actions\", \"Logic\").",
+    )
+    group: str = Field(
+        ...,
+        description="Gets or sets the SDK step group (e.g., \"Http\", \"OneDrive\").",
+    )
+    version: str = Field(
+        ...,
+        description="Gets or sets the version of the SDK step definition.",
+    )
+    step_parameters: str = Field(
+        ...,
+        description="Gets or sets the serialized step parameters (JSON).",
+        alias="stepParameters",
+    )
+    credential_references: List[ExportSDKStepCredentialReference] = Field(
+        ...,
+        description="Gets or sets the credential references used by this SDK step.",
+        alias="credentialReferences",
+    )
+    supported_credential_types: List[str] = Field(
+        ...,
+        description="Gets or sets the supported credential types for this step.",
+        alias="supportedCredentialTypes",
+    )
 
 
 class ExportPipelineDefinitionResponse(BaseModel):
@@ -1484,9 +1977,12 @@ class ExportPipelineDefinitionResponse(BaseModel):
         python_code_blocks: Optional list of Python code blocks used by the pipeline
         routers: Optional list of router configurations used by the pipeline
         approval_requests: Optional list of approval requests
+        webhook_approval_requests: Optional list of webhook approval requests
         agent_cards: Optional list of agent cards
+        sub_agents: Optional list of sub-agents (nested agents)
         deployment: Optional deployment configuration for the pipeline
         interfaces: Optional list of interface configurations
+        sdk_steps: Optional list of SDK steps used by the agent
     """
 
     available: bool = Field(
@@ -1520,12 +2016,27 @@ class ExportPipelineDefinitionResponse(BaseModel):
         description="Gets or sets the approval requests.",
         alias="approvalRequests",
     )
+    webhook_approval_requests: Optional[List[ExportWebhookApprovalRequest]] = Field(
+        None,
+        description="Gets or sets the webhook approval requests.",
+        alias="webhookApprovalRequests",
+    )
     agent_cards: Optional[List[ExportAgentCard]] = Field(
         None, description="Gets or sets the agent cards.", alias="agentCards"
     )
+    sub_agents: Optional[List[ExportPipeline]] = Field(
+        None,
+        description="Gets or sets the sub-agents (nested agents) used by this agent.",
+        alias="subAgents",
+    )
     deployment: Optional[ExportDeployment] = Field(
         None, description="Gets or sets the deployment."
     )
     interfaces: Optional[ExportInterfaces] = Field(
         None, description="Gets or sets the interfaces."
     )
+    sdk_steps: Optional[List[ExportSDKStepDefinition]] = Field(
+        None,
+        description="Gets or sets the SDK steps used by this agent.",
+        alias="sdkSteps",
+    )

{airia-0.1.35.dist-info → airia-0.1.37.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airia
-Version: 0.1.35
+Version: 0.1.37
 Summary: Python SDK for Airia API
 Author-email: Airia LLC <support@airia.com>
 License: MIT

{airia-0.1.35.dist-info → airia-0.1.37.dist-info}/RECORD

@@ -31,9 +31,9 @@ airia/client/library/async_library.py,sha256=fRG3eUxZkXDMnDTAt18KxD1LQMC5VB6cssW
 airia/client/library/base_library.py,sha256=VkcWr25fp2es2sRB97tORQmtAi59NZ6YRzwLymlQHIk,4664
 airia/client/library/sync_library.py,sha256=u_oBscUTYRNcsDbxCgpx_UvY5UAGgum2DW7eWt8y_i8,4191
 airia/client/models/__init__.py,sha256=YbxGLwMb7RQKfPo_2L_O_W9j8Szwnb0vtWiUNxkScfs,107
-airia/client/models/async_models.py,sha256=gwW3tWwzSZ9_3JmGlwryY48w9KrFXoLET6Jpsj0hX0I,5853
-airia/client/models/base_models.py,sha256=FchsA7P-Fc7fxzlN88jJ3BEVXPWDcIHSzci6wtSArqg,3439
-airia/client/models/sync_models.py,sha256=FQLM4xCoxcBk1NuLf6z7jVVejlDKx5KpMRBz2tQm71o,5748
+airia/client/models/async_models.py,sha256=GihX15MjqH-XUUl141aae0fbHG-FVH9SdNenIAY9mFU,8505
+airia/client/models/base_models.py,sha256=NNe24dcCirraVzpWJR5nNdPA3_nGwj98Ga1bk1f-TMI,4522
+airia/client/models/sync_models.py,sha256=36-ejKBLJ6FCB4RWXCSdBy84VpOmVYaK9PdoUJMVgCg,8366
 airia/client/pipeline_execution/__init__.py,sha256=7qEZsPRTLySC71zlwYioBuJs6B4BwRCgFL3TQyFWXmc,175
 airia/client/pipeline_execution/async_pipeline_execution.py,sha256=EH9EexSYjhSxWAB5kOLQSh5Sfl5Qn5lQGyJrU72wFLw,23824
 airia/client/pipeline_execution/base_pipeline_execution.py,sha256=8x99rm8cPq-i9-UA97-XLNo1qAwdEPkmZ815fRSceFE,11168
@@ -81,8 +81,8 @@ airia/types/api/pipeline_execution/_pipeline_execution.py,sha256=Y167Tdt0XHY2ZGV
 airia/types/api/pipeline_execution/get_pipeline_execution.py,sha256=Mkp3bV9sajVq4GARVyJviFWPR2BEtLCbZZLHTgQqcSI,2925
 airia/types/api/pipeline_import/__init__.py,sha256=5m8e4faOYGCEFLQWJgj-v5H4NXPNtnlAKAxA4tGJUbw,267
 airia/types/api/pipeline_import/create_agent_from_pipeline_definition.py,sha256=vn2aZGPnahmstHYKKMMN1hyTMRWFa2uAjpAYGwaI2xg,4394
-airia/types/api/pipelines_config/__init__.py,sha256=tNYV8AEGsKMfH4nMb-KGDfH1kvHO2b6VSoTB7TYQ7N8,2188
-airia/types/api/pipelines_config/export_pipeline_definition.py,sha256=0ieujueaeHFRXD2h8TzjuvwmaW-f5OFxfQPtjeXcKFg,59941
+airia/types/api/pipelines_config/__init__.py,sha256=6nxX-y1cAp66aupdzG_EvHeizBNoJ56i5nhRQIp2kbg,2752
+airia/types/api/pipelines_config/export_pipeline_definition.py,sha256=2rrniWhNihzyDK-w8ppYMxksl-hdqisWQFdfogiiZYw,79193
 airia/types/api/pipelines_config/get_pipeline_config.py,sha256=gvyp_xGpxr3Elcsu04JSQRPDvjmxRCPDAAR0rbE-oGs,23538
 airia/types/api/pipelines_config/get_pipelines_config.py,sha256=RbiX5zISxzGRxzPGHe7QpO-Ro-0woQsPGLxtiP4Y4K4,15955
 airia/types/api/project/__init__.py,sha256=ervHvCeqt08JkMRsSrG1ZnQshE70of-8kf4VeW2HG9c,113
@@ -95,8 +95,8 @@ airia/types/api/tools/_tools.py,sha256=PSJYFok7yQdE4it55iQmbryFzKN54nT6N161X1Rkp
 airia/types/sse/__init__.py,sha256=KWnNTfsQnthfrU128pUX6ounvSS7DvjC-Y21FE-OdMk,1863
 airia/types/sse/sse_messages.py,sha256=asq9KG5plT2XSgQMz-Nqo0WcKlXvE8UT3E-WLhCegPk,30244
 airia/utils/sse_parser.py,sha256=XCTkuaroYWaVQOgBq8VpbseQYSAVruF69AvKUwZQKTA,4251
-airia-0.1.35.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
-airia-0.1.35.dist-info/METADATA,sha256=FTD1cTvHWUy7elHFeAgBL8c-St2jkaqSsoKD36ws72M,4949
-airia-0.1.35.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-airia-0.1.35.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
-airia-0.1.35.dist-info/RECORD,,
+airia-0.1.37.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
+airia-0.1.37.dist-info/METADATA,sha256=JIQioSlT7ZRhwTVDMaVEtdncWYlZhODaWScapoYJdfY,4949
+airia-0.1.37.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+airia-0.1.37.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
+airia-0.1.37.dist-info/RECORD,,