airia 0.1.27__py3-none-any.whl → 0.1.29__py3-none-any.whl

airia/client/models/async_models.py

@@ -94,3 +94,63 @@ class AsyncModels(BaseModels):
             return []
 
         return [ModelItem(**item) for item in resp["items"]]
+
+    async def delete_model(
+        self,
+        model_id: str,
+        correlation_id: Optional[str] = None,
+    ) -> None:
+        """
+        Delete a model by its ID.
+
+        This method permanently deletes a model from the Airia platform. Once deleted,
+        the model cannot be recovered. Only models that the authenticated user has
+        permission to delete can be removed.
+
+        Args:
+            model_id (str): The unique identifier of the model to delete.
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+
+        Returns:
+            None
+
+        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")
+
+            # Delete a specific model
+            await client.models.delete_model(
+                model_id="12345678-1234-1234-1234-123456789abc"
+            )
+            print("Model deleted successfully")
+
+            # Delete with correlation ID for tracking
+            await client.models.delete_model(
+                model_id="12345678-1234-1234-1234-123456789abc",
+                correlation_id="my-correlation-id"
+            )
+            ```
+
+        Note:
+            This operation is irreversible. Ensure you have the correct model ID
+            before calling this method. You must have delete permissions for the
+            specified model.
+        """
+        request_data = self._pre_delete_model(
+            model_id=model_id,
+            correlation_id=correlation_id,
+            api_version=ApiVersion.V1.value,
+        )
+        await self._request_handler.make_request(
+            "DELETE", request_data, return_json=False
+        )

airia/client/models/base_models.py

@@ -66,3 +66,37 @@ class BaseModels:
         )
 
         return request_data
+
+    def _pre_delete_model(
+        self,
+        model_id: str,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V1.value,
+    ):
+        """
+        Prepare request data for deleting a model.
+
+        Args:
+            model_id: The ID of the model to delete
+            correlation_id: Optional correlation ID for tracing
+            api_version: API version to use for the request
+
+        Returns:
+            RequestData: Prepared request data for the delete 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

airia/client/models/sync_models.py

@@ -94,3 +94,61 @@ class Models(BaseModels):
             return []
 
         return [ModelItem(**item) for item in resp["items"]]
+
+    def delete_model(
+        self,
+        model_id: str,
+        correlation_id: Optional[str] = None,
+    ) -> None:
+        """
+        Delete a model by its ID.
+
+        This method permanently deletes a model from the Airia platform. Once deleted,
+        the model cannot be recovered. Only models that the authenticated user has
+        permission to delete can be removed.
+
+        Args:
+            model_id (str): The unique identifier of the model to delete.
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+
+        Returns:
+            None
+
+        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")
+
+            # Delete a specific model
+            client.models.delete_model(
+                model_id="12345678-1234-1234-1234-123456789abc"
+            )
+            print("Model deleted successfully")
+
+            # Delete with correlation ID for tracking
+            client.models.delete_model(
+                model_id="12345678-1234-1234-1234-123456789abc",
+                correlation_id="my-correlation-id"
+            )
+            ```
+
+        Note:
+            This operation is irreversible. Ensure you have the correct model ID
+            before calling this method. You must have delete permissions for the
+            specified model.
+        """
+        request_data = self._pre_delete_model(
+            model_id=model_id,
+            correlation_id=correlation_id,
+            api_version=ApiVersion.V1.value,
+        )
+        self._request_handler.make_request("DELETE", request_data, return_json=False)

airia/types/api/pipelines_config/export_pipeline_definition.py

@@ -36,6 +36,10 @@ 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
+        custom_credentials: Optional custom credentials configuration
         id: The unique identifier for the credential set
     """
 
@@ -51,6 +55,22 @@ class ExportCredentials(BaseModel):
         description="Gets or sets the credential data list.",
         alias="credentialDataList",
     )
+    display_identifier_name: str = Field(
+        ...,
+        description="Gets or sets the display identifier name.",
+        alias="displayIdentifierName",
+    )
+    administrative_scope: str = Field(
+        ...,
+        description="Gets or sets the administrative scope.",
+        alias="administrativeScope",
+    )
+    origin: str = Field(..., description="Gets or sets the origin.")
+    custom_credentials: Optional[dict] = Field(
+        None,
+        description="Gets or sets the custom credentials.",
+        alias="customCredentials",
+    )
     id: str = Field(..., description="Gets or sets the ID.")
 
 
@@ -152,7 +172,7 @@ class ExportPipelineStep(BaseModel):
         prompt_id: Optional ID of the associated prompt
         model_id: Optional ID of the associated model
         tool_ids: Optional list of tool IDs used by this step
-        tool_params: Optional default parameters for tools
+        tool_params_json: Optional JSON string with default parameters for tools
         data_source_id: Optional ID of the associated data source
         top_k: Optional number of top results to retrieve
         relevance_threshold: Optional relevance threshold for retrieval
@@ -164,6 +184,10 @@ class ExportPipelineStep(BaseModel):
         background_color: Optional background color for the step
         content: Optional content or configuration data
         step_title: The human-readable title of the step
+        width: The width of the step in the UI
+        height: The height of the step in the UI
+        ai_operation_step_include_chat_history: Optional flag for including chat history in AI operations
+        input_variables_json: Optional JSON string with input variables configuration
     """
 
     id: str = Field(..., description="Gets or sets the ID.")
@@ -192,8 +216,10 @@ class ExportPipelineStep(BaseModel):
     tool_ids: Optional[List[str]] = Field(
         None, description="Gets or sets the tool IDs.", alias="toolIds"
     )
-    tool_params: Optional[Dict[str, str]] = Field(
-        None, description="Gets or sets the tool default params.", alias="toolParams"
+    tool_params_json: Optional[str] = Field(
+        None,
+        description="Gets or sets the tool default params as JSON string.",
+        alias="toolParamsJson",
     )
     data_source_id: Optional[str] = Field(
         None, description="Gets or sets the data source.", alias="dataSourceId"
@@ -232,6 +258,22 @@ class ExportPipelineStep(BaseModel):
     step_title: str = Field(
         ..., description="Gets or sets the Step Title.", alias="stepTitle"
     )
+    width: Optional[int] = Field(
+        None, description="Gets or sets the width of the step."
+    )
+    height: Optional[int] = Field(
+        None, description="Gets or sets the height of the step."
+    )
+    ai_operation_step_include_chat_history: Optional[bool] = Field(
+        None,
+        description="Gets or sets whether to include chat history in AI operations.",
+        alias="aiOperationStepIncludeChatHistory",
+    )
+    input_variables_json: Optional[str] = Field(
+        None,
+        description="Gets or sets the input variables as JSON string.",
+        alias="inputVariablesJson",
+    )
 
 
 class AgentDetailItemDefinition(BaseModel):
@@ -273,25 +315,44 @@ class ExportPipeline(BaseModel):
     identification, description, industry classification, and step definitions.
 
     Attributes:
+        id: The unique identifier of the pipeline
         name: The name of the pipeline
         execution_name: The name used for programmatic execution
+        agent_icon: Optional icon reference
         agent_description: Detailed description of the pipeline's purpose
+        markdown_description: Markdown-formatted description
+        tagline: Optional brief tagline for the pipeline
         video_link: Optional URL to instructional video
         industry: The primary industry served by this pipeline
         sub_industries: Additional industry classifications
+        tags: Optional list of tags for categorization
         agent_details: Dictionary of configurable agent parameters
-        id: The unique identifier of the pipeline
-        agent_icon: Optional base64-encoded icon image
         steps: List of pipeline steps that make up the workflow
+        behaviours: List of pipeline behaviours
+        agent_unicode_icon: Optional unicode icon for the agent
+        agent_icon_base64: Optional base64-encoded icon image
+        alignment: The alignment/layout of the pipeline
     """
 
+    id: str = Field(..., description="Gets or sets the ID.")
     name: str = Field(..., description="Gets or sets the name.")
     execution_name: Optional[str] = Field(
         None, description="Gets or sets the execution name.", alias="executionName"
     )
+    agent_icon: Optional[str] = Field(
+        None, description="Gets or sets the Agent Icon.", alias="agentIcon"
+    )
     agent_description: str = Field(
         ..., description="Gets or sets the description.", alias="agentDescription"
     )
+    markdown_description: Optional[str] = Field(
+        None,
+        description="Gets or sets the markdown description.",
+        alias="markdownDescription",
+    )
+    tagline: Optional[str] = Field(
+        None, description="Gets or sets the tagline."
+    )
     video_link: Optional[str] = Field(
         None, description="Gets or sets the video link.", alias="videoLink"
     )
@@ -299,14 +360,27 @@ class ExportPipeline(BaseModel):
     sub_industries: Optional[List[str]] = Field(
         None, description="Gets or sets the sub industries.", alias="subIndustries"
     )
-    agent_details: Optional[Dict[str, List[AgentDetailItemDefinition]]] = Field(
-        None, description="Gets or sets the industry.", alias="agentDetails"
+    tags: Optional[List[str]] = Field(
+        None, description="Gets or sets the tags for categorization."
     )
-    id: str = Field(..., description="Gets or sets the ID.")
-    agent_icon: Optional[str] = Field(
-        None, description="Gets or sets the Agent Icon.", alias="agentIcon"
+    agent_details: Optional[Dict[str, List[AgentDetailItemDefinition]]] = Field(
+        None, description="Gets or sets the agent details.", alias="agentDetails"
     )
     steps: List[ExportPipelineStep] = Field(..., description="Gets or sets the steps.")
+    behaviours: Optional[List[str]] = Field(
+        ..., description="Gets or sets the pipeline behaviours."
+    )
+    agent_unicode_icon: Optional[str] = Field(
+        None,
+        description="Gets or sets the agent unicode icon.",
+        alias="agentUnicodeIcon",
+    )
+    agent_icon_base64: Optional[str] = Field(
+        None,
+        description="Gets or sets the base64-encoded agent icon.",
+        alias="agentIconBase64",
+    )
+    alignment: str = Field(..., description="Gets or sets the pipeline alignment.")
 
 
 class ExportDataSourceFile(BaseModel):
@@ -338,6 +412,53 @@ class ExportDataSourceFile(BaseModel):
     )
 
 
+class ExportVectorStore(BaseModel):
+    """Represents vector store configuration for data sources.
+
+    Defines the vector store settings including provider, hosting type,
+    and embedding configuration.
+
+    Attributes:
+        hosting_type: The hosting type for the vector store
+        modality: The modality of the vector store
+        provider: The vector store provider
+        store_type_id: The ID of the store type
+        credential_id: Optional credential ID for the vector store
+        configuration_json: Optional JSON configuration
+        sparse_vector_enabled: Whether sparse vectors are enabled
+        embedding_provider: The embedding provider
+        embedding_configuration_json: Optional embedding configuration JSON
+    """
+
+    hosting_type: str = Field(
+        ..., description="Gets or sets the hosting type.", alias="hostingType"
+    )
+    modality: str = Field(..., description="Gets or sets the modality.")
+    provider: str = Field(..., description="Gets or sets the provider.")
+    store_type_id: str = Field(
+        ..., description="Gets or sets the store type ID.", alias="storeTypeId"
+    )
+    credential_id: Optional[str] = Field(
+        None, description="Gets or sets the credential ID.", alias="credentialId"
+    )
+    configuration_json: Optional[str] = Field(
+        None, description="Gets or sets the configuration JSON.", alias="configurationJson"
+    )
+    sparse_vector_enabled: bool = Field(
+        ...,
+        description="Gets or sets whether sparse vectors are enabled.",
+        alias="sparseVectorEnabled",
+    )
+    embedding_provider: str = Field(
+        ..., description="Gets or sets the embedding provider.", alias="embeddingProvider"
+    )
+    embedding_configuration_json: Optional[str] = Field(
+        None,
+        description="Gets or sets the embedding configuration JSON.",
+        alias="embeddingConfigurationJson",
+    )
+
+
 class ExportChunkingConfig(BaseModel):
     """Represents chunking configuration for data processing.
 
@@ -385,6 +506,13 @@ class ExportDataSource(BaseModel):
         configuration_json: Optional JSON configuration string
         credentials: Optional credential information for access
         is_image_processing_enabled: Whether image processing is enabled
+        description: Optional description of the data source
+        vector_store: Optional vector store configuration
+        scan_document_for_images: Whether to scan documents for images
+        image_processing_prompt: Optional prompt for image processing
+        store_type: Optional store type
+        table_document_processing_mode: Optional table document processing mode
+        parser_configuration_json: Optional parser configuration JSON
     """
 
     id: str = Field(..., description="Gets the id.")
@@ -400,11 +528,11 @@ class ExportDataSource(BaseModel):
     data_source_type: str = Field(
         ..., description="Gets the data source type.", alias="dataSourceType"
     )
-    database_type: str = Field(
-        ..., description="Gets the database type.", alias="databaseType"
+    database_type: Optional[str] = Field(
+        None, description="Gets the database type.", alias="databaseType"
     )
-    embedding_provider: str = Field(
-        ..., description="Gets the Embedding provider type.", alias="embeddingProvider"
+    embedding_provider: Optional[str] = Field(
+        None, description="Gets the Embedding provider type.", alias="embeddingProvider"
     )
     is_user_specific: bool = Field(
         ...,
@@ -431,6 +559,35 @@ class ExportDataSource(BaseModel):
         description="Gets or sets a value indicating whether the image processing is enabled.",
         alias="isImageProcessingEnabled",
     )
+    description: Optional[str] = Field(
+        None, description="Gets or sets the description."
+    )
+    vector_store: Optional[ExportVectorStore] = Field(
+        None, description="Gets or sets the vector store.", alias="vectorStore"
+    )
+    scan_document_for_images: bool = Field(
+        ...,
+        description="Gets or sets whether to scan documents for images.",
+        alias="scanDocumentForImages",
+    )
+    image_processing_prompt: Optional[str] = Field(
+        None,
+        description="Gets or sets the image processing prompt.",
+        alias="imageProcessingPrompt",
+    )
+    store_type: Optional[str] = Field(
+        None, description="Gets or sets the store type.", alias="storeType"
+    )
+    table_document_processing_mode: Optional[str] = Field(
+        None,
+        description="Gets or sets the table document processing mode.",
+        alias="tableDocumentProcessingMode",
+    )
+    parser_configuration_json: Optional[str] = Field(
+        None,
+        description="Gets or sets the parser configuration JSON.",
+        alias="parserConfigurationJson",
+    )
 
 
 class ExportPromptMessageList(BaseModel):
@@ -457,6 +614,8 @@ class ExportPrompt(BaseModel):
     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
     """
@@ -467,6 +626,16 @@ class ExportPrompt(BaseModel):
         description="Gets or sets the version change description.",
         alias="versionChangeDescription",
     )
+    prompt_message: Optional[str] = Field(
+        None,
+        description="Gets or sets the consolidated prompt message.",
+        alias="promptMessage",
+    )
+    is_agent_specific: Optional[bool] = Field(
+        None,
+        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.",
@@ -502,6 +671,9 @@ class ExportToolParameters(BaseModel):
         parameter_description: Description of the parameter's purpose
         default: The default value for the parameter
         valid_options: Optional list of valid values for the parameter
+        array_item_type: Optional type of items if this is an array parameter
+        parameters: Optional nested parameters for complex types
+        requirement: Whether the parameter is required or optional
         id: The unique identifier of the parameter
     """
 
@@ -518,6 +690,17 @@ class ExportToolParameters(BaseModel):
         description="Gets or sets the list of valid options.",
         alias="validOptions",
     )
+    array_item_type: Optional[str] = Field(
+        None,
+        description="Gets or sets the array item type.",
+        alias="arrayItemType",
+    )
+    parameters: Optional[List["ExportToolParameters"]] = Field(
+        None, description="Gets or sets nested parameters for complex types."
+    )
+    requirement: str = Field(
+        ..., description="Gets or sets whether the parameter is required."
+    )
     id: Optional[str] = Field(None, description="Gets or sets the ID.")
 
 
@@ -540,8 +723,17 @@ class ExportTool(BaseModel):
         parameters_definition: Optional parameter definitions
         method_type: The HTTP method type (GET, POST, etc.)
         route_through_acc: Whether to route through the Access Control Center
+        acc_group: The access control group
         use_user_credentials: Whether to use user-specific credentials
         use_user_credentials_type: The type of user credentials to use
+        provider: The provider of the tool
+        body_type: The type of the request body
+        number_of_pages: Optional number of pages to retrieve
+        should_retrieve_full_page_content: Optional flag for full page content retrieval
+        schema_definition: Optional schema definition for the tool
+        request_timeout: The timeout for requests in seconds
+        should_reroute: Whether requests should be rerouted
+        credentials_source_type: The source type of the credentials
         id: The unique identifier of the tool
     """
 
@@ -581,6 +773,9 @@ class ExportTool(BaseModel):
         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"
+    )
     use_user_credentials: bool = Field(
         ...,
         description="Gets or sets a value indicating whether the tool should use user based credentials.",
@@ -591,6 +786,39 @@ 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"
+    )
+    number_of_pages: Optional[int] = Field(
+        None, description="Gets or sets the number of pages.", alias="numberOfPages"
+    )
+    should_retrieve_full_page_content: Optional[bool] = Field(
+        None,
+        description="Gets or sets whether to retrieve full page content.",
+        alias="shouldRetrieveFullPageContent",
+    )
+    schema_definition: Optional[str] = Field(
+        None,
+        description="Gets or sets the schema definition.",
+        alias="schemaDefinition",
+    )
+    request_timeout: Optional[int] = Field(
+        None, description="Gets or sets the request timeout.", alias="requestTimeout"
+    )
+    annotations: Optional[dict] = Field(
+        None, description="Gets or sets the annotations."
+    )
+    should_reroute: Optional[bool] = Field(
+        None,
+        description="Gets or sets whether requests should be rerouted.",
+        alias="shouldReroute",
+    )
+    credentials_source_type: Optional[str] = Field(
+        None,
+        description="Gets or sets the credentials source type.",
+        alias="credentialsSourceType",
+    )
     id: str = Field(..., description="Gets or sets the ID.")
 
 
@@ -626,7 +854,8 @@ class ExportModel(BaseModel):
         allow_airia_credentials: Whether to allow Airia-provided credentials
         allow_byok_credentials: Whether to allow bring-your-own-key credentials
         author: Optional author information
-        price_type: The pricing model type
+        price_type: Optional pricing model type
+        category: Optional category classification
     """
 
     id: str = Field(..., description="Gets or sets the ID.")
@@ -711,6 +940,9 @@ class ExportModel(BaseModel):
     price_type: Optional[str] = Field(
         None, description="Gets or sets the price type.", alias="priceType"
     )
+    category: Optional[str] = Field(
+        None, description="Gets or sets the category classification."
+    )
 
 
 class ExportMemory(BaseModel):
@@ -779,17 +1011,33 @@ class ExportRouter(BaseModel):
     Attributes:
         id: The unique identifier of the router
         model_id: Optional ID of the model used for routing decisions
-        model: Optional AI model definition for routing
+        router_config_json: Optional JSON string of router configuration
         router_config: Dictionary of routing configurations
+        is_multi_route: Optional flag indicating if this is a multi-route router
+        include_chat_history: Whether to include chat history in routing decisions
     """
 
     id: str = Field(..., description="Gets or sets the Router identifier.")
     model_id: Optional[str] = Field(
         None, description="Gets or sets the Model identifier.", alias="modelId"
     )
-    model: Optional[ExportModel] = Field(None, description="Gets or sets the AI Model.")
-    router_config: Dict[str, ExportRouterConfig] = Field(
-        ..., description="Gets or sets the Router Configuration.", alias="routerConfig"
+    router_config_json: Optional[str] = Field(
+        None,
+        description="Gets or sets the router configuration JSON.",
+        alias="routerConfigJson",
+    )
+    router_config: Optional[Dict[str, ExportRouterConfig]] = Field(
+        None, description="Gets or sets the Router Configuration.", alias="routerConfig"
+    )
+    is_multi_route: Optional[bool] = Field(
+        None,
+        description="Gets or sets whether this is a multi-route router.",
+        alias="isMultiRoute",
+    )
+    include_chat_history: bool = Field(
+        ...,
+        description="Gets or sets whether to include chat history.",
+        alias="includeChatHistory",
     )
 
 
@@ -814,6 +1062,68 @@ class ExportUserPrompt(BaseModel):
     )
 
 
+class ExportAssignedAgent(BaseModel):
+    """Represents an assigned agent for quick actions.
+
+    Attributes:
+        deployment_id: The deployment ID
+        agent_name: The name of the agent
+        agent_url: The URL of the agent
+        is_configurable: Whether the agent is configurable
+    """
+
+    deployment_id: str = Field(
+        ..., description="Gets or sets the deployment ID.", alias="deploymentId"
+    )
+    agent_name: str = Field(
+        ..., description="Gets or sets the agent name.", alias="agentName"
+    )
+    agent_url: str = Field(
+        ..., description="Gets or sets the agent URL.", alias="agentUrl"
+    )
+    is_configurable: bool = Field(
+        ..., description="Gets or sets whether configurable.", alias="isConfigurable"
+    )
+
+
+class ExportQuickAction(BaseModel):
+    """Represents a quick action for browser extensions.
+
+    Attributes:
+        id: The unique identifier
+        display_name: The display name
+        quick_action_type: The type of quick action
+        assigned_agent: The assigned agent
+    """
+
+    id: str = Field(..., description="Gets or sets the ID.")
+    display_name: str = Field(
+        ..., description="Gets or sets the display name.", alias="displayName"
+    )
+    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"
+    )
+
+
+class ExportBrowserExtensionConfig(BaseModel):
+    """Represents browser extension configuration.
+
+    Attributes:
+        browser_type: The type of browser
+        quick_actions: List of quick actions
+    """
+
+    browser_type: str = Field(
+        ..., description="Gets or sets the browser type.", alias="browserType"
+    )
+    quick_actions: List[ExportQuickAction] = Field(
+        ..., description="Gets or sets the quick actions.", alias="quickActions"
+    )
+
+
 class ExportDeployment(BaseModel):
     """Represents a deployment configuration for export.
 
@@ -831,6 +1141,9 @@ class ExportDeployment(BaseModel):
         deployment_type: The type of deployment
         conversation_type: The type of conversation interface
         about_json: Optional JSON metadata about the deployment
+        supported_input_modes: Optional list of supported input modes
+        browser_extension_config: Optional browser extension configuration
+        display_consumption_info: Whether to display consumption information
     """
 
     name: str = Field(..., description="Gets the Deployment Name.")
@@ -867,6 +1180,21 @@ class ExportDeployment(BaseModel):
         description="Gets information about the deployment. This is a json value.",
         alias="aboutJson",
     )
+    supported_input_modes: Optional[List[str]] = Field(
+        None,
+        description="Gets the supported input modes.",
+        alias="supportedInputModes",
+    )
+    browser_extension_config: Optional[ExportBrowserExtensionConfig] = Field(
+        None,
+        description="Gets the browser extension config.",
+        alias="browserExtensionConfig",
+    )
+    display_consumption_info: bool = Field(
+        ...,
+        description="Gets whether to display consumption info.",
+        alias="displayConsumptionInfo",
+    )
 
 
 class ExportMetadata(BaseModel):
@@ -886,6 +1214,14 @@ class ExportMetadata(BaseModel):
         data_source_export_option: How data sources are handled in export
         version_information: Information about the pipeline version
         state: The current state of the agent
+        contributor_id: Optional ID of the contributor
+        contributor_given_name: Optional given name of the contributor
+        contributor_surname: Optional surname of the contributor
+        video_link: Optional URL to instructional video
+        readiness: The readiness status of the pipeline
+        department: The department this pipeline is associated with
+        agent_last_updated: Timestamp of when the agent was last updated
+        country: Optional country information
     """
 
     id: str = Field(..., description="Gets or sets the id.")
@@ -916,6 +1252,218 @@ class ExportMetadata(BaseModel):
         alias="versionInformation",
     )
     state: str = Field(..., description="Gets or sets the state of the agent.")
+    contributor_id: Optional[str] = Field(
+        None, description="Gets or sets the contributor ID.", alias="contributorId"
+    )
+    contributor_given_name: Optional[str] = Field(
+        None,
+        description="Gets or sets the contributor given name.",
+        alias="contributorGivenName",
+    )
+    contributor_surname: Optional[str] = Field(
+        None,
+        description="Gets or sets the contributor surname.",
+        alias="contributorSurname",
+    )
+    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(
+        ...,
+        description="Gets or sets the timestamp when the agent was last updated.",
+        alias="agentLastUpdated",
+    )
+    country: Optional[str] = Field(
+        None, description="Gets or sets the country information."
+    )
+
+
+class ExportApprovalRequest(BaseModel):
+    """Represents an approval request for export.
+
+    Attributes:
+        id: The unique identifier
+        message: Optional message
+        email_notification: Whether to send email notification
+        approval_description: Optional approval description
+        denial_description: Optional denial description
+        approved_handle_id: The approved handle ID
+        denied_handle_id: The denied handle ID
+    """
+
+    id: str = Field(..., description="Gets or sets the ID.")
+    message: Optional[str] = Field(None, description="Gets or sets the message.")
+    email_notification: bool = Field(
+        ..., description="Gets or sets email notification.", alias="emailNotification"
+    )
+    approval_description: Optional[str] = Field(
+        None,
+        description="Gets or sets the approval description.",
+        alias="approvalDescription",
+    )
+    denial_description: Optional[str] = Field(
+        None,
+        description="Gets or sets the denial description.",
+        alias="denialDescription",
+    )
+    approved_handle_id: str = Field(
+        ..., description="Gets or sets the approved handle ID.", alias="approvedHandleId"
+    )
+    denied_handle_id: str = Field(
+        ..., description="Gets or sets the denied handle ID.", alias="deniedHandleId"
+    )
+
+
+class ExportAgentCardProvider(BaseModel):
+    """Represents a provider for an agent card.
+
+    Attributes:
+        organization: The organization name
+        url: Optional URL
+    """
+
+    organization: str = Field(..., description="Gets or sets the organization.")
+    url: Optional[str] = Field(None, description="Gets or sets the URL.")
+
+
+class ExportAgentCardCapabilities(BaseModel):
+    """Represents capabilities of an agent card.
+
+    Attributes:
+        streaming: Optional streaming capability
+        push_notifications: Optional push notifications capability
+        state_transition_history: Optional state transition history capability
+    """
+
+    streaming: Optional[bool] = Field(None, description="Gets or sets streaming.")
+    push_notifications: Optional[bool] = Field(
+        None, description="Gets or sets push notifications.", alias="pushNotifications"
+    )
+    state_transition_history: Optional[bool] = Field(
+        None,
+        description="Gets or sets state transition history.",
+        alias="stateTransitionHistory",
+    )
+
+
+class ExportAgentCardAuthentication(BaseModel):
+    """Represents authentication for an agent card.
+
+    Attributes:
+        schemes: List of authentication schemes
+        credentials: Optional credentials
+    """
+
+    schemes: List[str] = Field(..., description="Gets or sets the schemes.")
+    credentials: Optional[dict] = Field(
+        None, description="Gets or sets the credentials."
+    )
+
+
+class ExportAgentCardSkill(BaseModel):
+    """Represents a skill for an agent card.
+
+    Attributes:
+        id: The unique identifier
+        name: The name
+        description: Optional description
+        tags: Optional tags
+        examples: Optional examples
+        input_modes: List of input modes
+        output_modes: 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"
+    )
+    output_modes: List[str] = Field(
+        ..., description="Gets or sets the output modes.", alias="outputModes"
+    )
+
+
+class ExportAgentCard(BaseModel):
+    """Represents an agent card for export.
+
+    Attributes:
+        agent_card_id: The agent card ID
+        name: The name
+        description: The description
+        url: The URL
+        provider: The provider
+        version: The version
+        documentation_url: Optional documentation URL
+        capabilities: The capabilities
+        authentication: The authentication
+        default_input_modes: List of default input modes
+        default_output_modes: List of default output modes
+        skills: List of skills
+    """
+
+    agent_card_id: str = Field(
+        ..., 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.")
+    url: str = Field(..., description="Gets or sets the URL.")
+    provider: ExportAgentCardProvider = Field(
+        ..., 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."
+    )
+    authentication: ExportAgentCardAuthentication = Field(
+        ..., description="Gets or sets the authentication."
+    )
+    default_input_modes: List[str] = Field(
+        ..., description="Gets or sets the default input modes.", alias="defaultInputModes"
+    )
+    default_output_modes: List[str] = Field(
+        ...,
+        description="Gets or sets the default output modes.",
+        alias="defaultOutputModes",
+    )
+    skills: List[ExportAgentCardSkill] = Field(
+        ..., description="Gets or sets the skills."
+    )
+
+
+class ExportEmailInbox(BaseModel):
+    """Represents email inbox interface configuration.
+
+    Attributes:
+        allowed_type: Optional allowed type
+        email_action: Optional email action
+    """
+
+    allowed_type: Optional[str] = Field(
+        None, description="Gets or sets the allowed type.", alias="allowedType"
+    )
+    email_action: Optional[str] = Field(
+        None, description="Gets or sets the email action.", alias="emailAction"
+    )
+
+
+class ExportInterfaces(BaseModel):
+    """Represents interfaces configuration.
+
+    Attributes:
+        email_inbox: Optional email inbox configuration
+    """
+
+    email_inbox: Optional[ExportEmailInbox] = Field(
+        None, description="Gets or sets the email inbox.", alias="emailInbox"
+    )
 
 
 class ExportPipelineDefinitionResponse(BaseModel):
@@ -925,6 +1473,7 @@ class ExportPipelineDefinitionResponse(BaseModel):
     including the pipeline definition, associated resources, and metadata.
 
     Attributes:
+        available: Whether the pipeline is available for export
         metadata: Export metadata and configuration information
         agent: The main pipeline definition and configuration
         data_sources: Optional list of data sources used by the pipeline
@@ -934,9 +1483,15 @@ class ExportPipelineDefinitionResponse(BaseModel):
         memories: Optional list of memory definitions used by the pipeline
         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
+        agent_cards: Optional list of agent cards
         deployment: Optional deployment configuration for the pipeline
+        interfaces: Optional list of interface configurations
     """
 
+    available: bool = Field(
+        ..., description="Gets or sets whether the pipeline is available."
+    )
     metadata: ExportMetadata = Field(..., description="Gets or sets the Metadata.")
     agent: ExportPipeline = Field(..., description="Gets or sets the pipeline.")
     data_sources: Optional[List[ExportDataSource]] = Field(
@@ -960,6 +1515,17 @@ class ExportPipelineDefinitionResponse(BaseModel):
     routers: Optional[List[ExportRouter]] = Field(
         None, description="Gets or sets the Routers."
     )
+    approval_requests: Optional[List[ExportApprovalRequest]] = Field(
+        None,
+        description="Gets or sets the approval requests.",
+        alias="approvalRequests",
+    )
+    agent_cards: Optional[List[ExportAgentCard]] = Field(
+        None, description="Gets or sets the agent cards.", alias="agentCards"
+    )
     deployment: Optional[ExportDeployment] = Field(
         None, description="Gets or sets the deployment."
     )
+    interfaces: Optional[ExportInterfaces] = Field(
+        None, description="Gets or sets the interfaces."
+    )

airia/types/api/tools/_tools.py

@@ -63,6 +63,18 @@ class ToolCredentialSettings(BaseModel):
     )
 
 
+class CredentialEntry(BaseModel):
+    """Represents a deserialized object for each entry of a credential.
+
+    Attributes:
+        key: The property name of the API Key for credentials data
+        value: The value of the credential
+    """
+
+    key: str
+    value: str
+
+
 class ToolCredential(BaseModel):
     """Tool credential information.
 
@@ -77,6 +89,11 @@ class ToolCredential(BaseModel):
         expires_on: When the credential expires
         administrative_scope: Scope of the credential (e.g., "Tenant")
         user_id: ID of the user who owns the credential
+        credential_data: List of credential entries
+        custom_credentials: Custom credentials data
+        custom_credentials_id: ID of custom credentials
+        tenant_id: ID of the tenant this credential belongs to
+        origin: Origin of the credential (Platform or Chat)
     """
 
     id: str
@@ -89,6 +106,11 @@ class ToolCredential(BaseModel):
     expires_on: Optional[datetime] = Field(None, alias="expiresOn")
     administrative_scope: str = Field(alias="administrativeScope")
     user_id: Optional[str] = Field(None, alias="userId")
+    credential_data: List[CredentialEntry] = Field(default_factory=list, alias="credentialData")
+    custom_credentials: Optional[dict] = Field(None, alias="customCredentials")
+    custom_credentials_id: Optional[str] = Field(None, alias="customCredentialsId")
+    tenant_id: Optional[str] = Field(None, alias="tenantId")
+    origin: str
 
 
 class ToolCredentials(BaseModel):

{airia-0.1.27.dist-info → airia-0.1.29.dist-info}/METADATA

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

{airia-0.1.27.dist-info → airia-0.1.29.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=kcFywYQr4Ni1_eSH3dnC3ZVfDoHIAJZDb94Zsegt3pQ,3797
-airia/client/models/base_models.py,sha256=rBXPK3f5YPn1ntkmXNdHyMlzYpOH5d3CA0hHvQM1GTI,2361
-airia/client/models/sync_models.py,sha256=O_nD2bD9N3EIWz6nCt5KUQaRmIGaBhrLs006B-1d65U,3748
+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/pipeline_execution/__init__.py,sha256=7qEZsPRTLySC71zlwYioBuJs6B4BwRCgFL3TQyFWXmc,175
 airia/client/pipeline_execution/async_pipeline_execution.py,sha256=alglTaHtWEipJx56WOUdV8MNhpOmo4GuhCXI9wkv7Mw,18022
 airia/client/pipeline_execution/base_pipeline_execution.py,sha256=fVGG__zHPpKLZdcwOL3GYmEyjCC7V4GZVuZ4CagGRP4,9608
@@ -80,7 +80,7 @@ airia/types/api/pipeline_execution/_pipeline_execution.py,sha256=4E8rmr3Ok9EHkTz
 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=UdXoG2c_MpzRF7apLz-4K-wsSyu_Jw0xAliCkCWlFD8,38156
+airia/types/api/pipelines_config/export_pipeline_definition.py,sha256=OzB9gYgm8Dy7crYrcrLMGchATTSmVwFiL2GJ4JK3gRs,59930
 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
@@ -89,12 +89,12 @@ airia/types/api/store/__init__.py,sha256=BgViwV_SHE9cxtilPnA2xWRk6MkAbxYxansmGeZ
 airia/types/api/store/get_file.py,sha256=Li3CpWUktQruNeoKSTlHJPXzNMaysG_Zy-fXGji8zs8,6174
 airia/types/api/store/get_files.py,sha256=v22zmOuTSFqzrS73L5JL_FgBeF5a5wutv1nK4IHAoW0,700
 airia/types/api/tools/__init__.py,sha256=r4jcHknQcLIb3jVkGR5lcmuapHqln7-rot3EBCNmZtI,146
-airia/types/api/tools/_tools.py,sha256=M8nVRqhTpdbIzPs2z4Pi11BFtOnbgtNk3_RnDyViM_M,8348
+airia/types/api/tools/_tools.py,sha256=PSJYFok7yQdE4it55iQmbryFzKN54nT6N161X1Rkp5U,9241
 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.27.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
-airia-0.1.27.dist-info/METADATA,sha256=F_fbsYlh5Av4u2zFd10Yx3CFTYYeplBWx2tIZK9cdU8,4506
-airia-0.1.27.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-airia-0.1.27.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
-airia-0.1.27.dist-info/RECORD,,
+airia-0.1.29.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
+airia-0.1.29.dist-info/METADATA,sha256=SjH5WPody0kAfZ8TreZ_Vr3A1PLkxXNb-CzSLJ2BQtQ,4506
+airia-0.1.29.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+airia-0.1.29.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
+airia-0.1.29.dist-info/RECORD,,