airia 0.1.23__py3-none-any.whl → 0.1.24__py3-none-any.whl

airia/types/api/pipeline_execution/_pipeline_execution.py

@@ -5,44 +5,89 @@ This module defines the response models returned by pipeline execution endpoints
 including both synchronous and streaming response types.
 """
 
-from typing import Any, AsyncIterator, Dict, Iterator, Optional
+from datetime import datetime
+from typing import Any, AsyncIterator, Dict, Iterator, List, Literal, Optional, Union
+from uuid import UUID
 
 from pydantic import BaseModel, ConfigDict, Field
 
 from ...sse import SSEMessage
 
 
-class PipelineExecutionResponse(BaseModel):
-    """Response model for standard pipeline execution requests.
+class TimeTrackingData(BaseModel):
+    """
+    Time tracking data for pipeline execution.
+
+    Attributes:
+        duration: Human-readable duration string (e.g., "2.5s", "1m 30s"). None if execution not completed.
+        started_at: Timestamp when the execution started.
+        finished_at: Timestamp when the execution finished. None if still running or failed.
+    """
+
+    duration: Optional[str] = None
+    started_at: datetime = Field(alias="startedAt")
+    finished_at: Optional[datetime] = Field(None, alias="finishedAt")
 
-    This model represents the response when executing a pipeline in normal mode
-    (not debug mode and not streaming).
+
+class AgentBaseStepResult(BaseModel):
+    """
+    Base class for agent step results.
 
     Attributes:
-        result: The execution result as a string
-        report: Always None for standard executions
-        is_backup_pipeline: Whether a backup pipeline was used for execution
+        step_type: Type of the step (e.g., "LLMStep", "ToolStep", "ConditionalStep").
+        step_id: Unique identifier for this step execution.
+        type: Discriminator field for polymorphic deserialization ($type field).
     """
 
-    result: str
-    report: None
-    is_backup_pipeline: bool = Field(alias="isBackupPipeline")
+    step_type: str = Field(alias="stepType")
+    step_id: UUID = Field(alias="stepId")
+    type: str = Field(alias="$type")
 
 
-class PipelineExecutionDebugResponse(BaseModel):
-    """Response model for pipeline execution requests in debug mode.
+class PipelineStepResult(BaseModel):
+    """
+    Result of a pipeline step execution.
 
-    This model includes additional debugging information in the report field
-    that provides insights into the pipeline's execution process.
+    Attributes:
+        step_id: Unique identifier for the step.
+        step_type: Type of step that was executed (e.g., "LLMStep", "ToolStep").
+        step_title: Human-readable title/name of the step.
+        result: Output result from the step execution. None if step failed or produced no output.
+        input: List of input data that was passed to this step.
+        success: Whether the step executed successfully. None if status is indeterminate.
+        completion_status: Status of step completion (e.g., "Success", "Failed", "Skipped").
+        time_tracking_data: Timing information for this step's execution.
+        exception_message: Error message if the step failed. None if successful.
+        debug_information: Additional debug data and metadata from step execution.
+    """
+
+    step_id: UUID = Field(alias="stepId")
+    step_type: Optional[str] = Field(None, alias="stepType")
+    step_title: str = Field(alias="stepTitle")
+    result: Optional[AgentBaseStepResult] = None
+    input: List[AgentBaseStepResult] = []
+    success: Optional[bool] = None
+    completion_status: str = Field(alias="completionStatus")
+    time_tracking_data: Optional[TimeTrackingData] = Field(
+        None, alias="timeTrackingData"
+    )
+    exception_message: Optional[str] = Field(None, alias="exceptionMessage")
+    debug_information: Dict[str, Any] = Field(
+        default_factory=dict, alias="debugInformation"
+    )
+
+
+class PipelineExecutionResponse(BaseModel):
+    """Response model for pipeline execution requests in mode.
 
     Attributes:
         result: The execution result as a string
-        report: Dictionary containing debugging information and execution details
+        report: Optional Dictionary containing debugging information and execution details
         is_backup_pipeline: Whether a backup pipeline was used for execution
     """
 
     result: Optional[str]
-    report: Dict[str, Any]
+    report: Optional[Dict[str, PipelineStepResult]] = None
     is_backup_pipeline: bool = Field(alias="isBackupPipeline")
 
 
@@ -74,3 +119,57 @@ class PipelineExecutionAsyncStreamedResponse(BaseModel):
     model_config = ConfigDict(arbitrary_types_allowed=True)
 
     stream: AsyncIterator[SSEMessage]
+
+
+class TemporaryAssistantResponse(BaseModel):
+    """Response model for TemporaryAssistant pipeline execution.
+
+    This model represents the response from the TemporaryAssistant endpoint,
+    which can return either string or array results based on the discriminator.
+
+    Attributes:
+        type: Discriminator field indicating the response type
+        report: Pipeline execution debug report (optional)
+        is_backup_pipeline: Whether a backup pipeline was used
+        execution_id: Unique execution identifier
+        user_input_id: Identifier for the user input
+        files: Files processed during execution
+        images: Images processed during execution
+    """
+
+    type: Literal["string", "objectArray"] = Field(alias="$type")
+    result: Union[List[Any], str] = Field(...)
+    report: Optional[Dict[str, PipelineStepResult]] = None
+    is_backup_pipeline: bool = Field(alias="isBackupPipeline")
+    execution_id: UUID = Field(alias="executionId")
+    user_input_id: Optional[UUID] = Field(None, alias="userInputId")
+    files: Optional[List[str]] = None
+    images: Optional[List[str]] = None
+
+
+class TemporaryAssistantStreamedResponse(BaseModel):
+    """
+    Response model for streaming TemporaryAssistant requests (synchronous client).
+
+    Attributes:
+        stream: Iterator that yields SSEMessage objects as they arrive from the streaming response.
+                Each message contains incremental data from the assistant's response.
+    """
+
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+
+    stream: Iterator[SSEMessage]
+
+
+class TemporaryAssistantAsyncStreamedResponse(BaseModel):
+    """
+    Response model for streaming TemporaryAssistant requests (asynchronous client).
+
+    Attributes:
+        stream: Async iterator that yields SSEMessage objects as they arrive from the streaming response.
+                Each message contains incremental data from the assistant's response.
+    """
+
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+
+    stream: AsyncIterator[SSEMessage]

{airia-0.1.23.dist-info → airia-0.1.24.dist-info}/METADATA

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

{airia-0.1.23.dist-info → airia-0.1.24.dist-info}/RECORD

@@ -3,9 +3,9 @@ airia/constants.py,sha256=oKABowOTsJPIQ1AgtVpNN73oKuhh1DekBC5axYSm8lY,647
 airia/exceptions.py,sha256=M98aESxodebVLafqDQSqCKLUUT_b-lNSVOpYGJ6Cmn0,1114
 airia/logs.py,sha256=oHU8ByekHu-udBB-5abY39wUct_26t8io-A1Kcl7D3U,4538
 airia/client/__init__.py,sha256=6gSQ9bl7j79q1HPE0o5py3IRdkwWWuU_7J4h05Dd2o8,127
-airia/client/async_client.py,sha256=rj2RfVZqvwrsDDsmeVzMvX_1sceXqM3Xs3nFDnHb8BI,7143
+airia/client/async_client.py,sha256=l1mx6m49OPlKlf_ZRn7ko1aY2i7LGsXpVoFPlbWXpFs,7236
 airia/client/base_client.py,sha256=ftyLi2E3Hb4CQFYfiknitJA0sWDTylCiUbde0qkWYhg,3182
-airia/client/sync_client.py,sha256=NpZIbgheAUxWWTtmKA0PjmmCjz0jdNs6cRFpV6iwnfU,6992
+airia/client/sync_client.py,sha256=MbnQPfnJJTh96GDvxm3HfUWDxFa7e5Rm4_hoccId9kA,7075
 airia/client/_request_handler/__init__.py,sha256=FOdJMfzjN0Tw0TeD8mDJwHgte70Fw_j08EljtfulCvw,157
 airia/client/_request_handler/async_request_handler.py,sha256=8225uM82kgOFu1E2DULVuk0qBN7vmPmcBtYXstUZaR8,10931
 airia/client/_request_handler/base_request_handler.py,sha256=dObRK6e_V_cGslHzC3WrNQ5lZqmMamjxNhVpb7P1L_w,3516
@@ -26,10 +26,14 @@ airia/client/deployments/__init__.py,sha256=Xu_MqbeUrtqYE4tH6hbmvUk8eQScWJ6MEyYF
 airia/client/deployments/async_deployments.py,sha256=IqRQZtkulBxxuviHpbcoi4UFQ-XRH4oRqZ90Q4H-FXs,4321
 airia/client/deployments/base_deployments.py,sha256=luiw4Fuj2YxD0j0-rXrsWkae_CNipxwRV7wIrHBFxpI,3230
 airia/client/deployments/sync_deployments.py,sha256=gEq2M0uUi5GLw8lxf8QasAHx48FZHzzRwrt7TAcyA9s,4230
+airia/client/library/__init__.py,sha256=jlcw1FEbKn7bN2ox-5aeJp3VQk4qb5iUbyjxbE2lsjo,165
+airia/client/library/async_library.py,sha256=JwGTmkQNZivGbxscUPB2X3pRSguJawjSOaGG0ajtT54,4279
+airia/client/library/base_library.py,sha256=VkcWr25fp2es2sRB97tORQmtAi59NZ6YRzwLymlQHIk,4664
+airia/client/library/sync_library.py,sha256=MTyH3-JDGhteA1C3Xe2Xd0sRsb1E-0HZnUzvYOQNuw8,4190
 airia/client/pipeline_execution/__init__.py,sha256=7qEZsPRTLySC71zlwYioBuJs6B4BwRCgFL3TQyFWXmc,175
-airia/client/pipeline_execution/async_pipeline_execution.py,sha256=RDXqaMxfA1QlIsF4v8g6oPCi2SichNzWy7DdlFmc3zE,9734
-airia/client/pipeline_execution/base_pipeline_execution.py,sha256=wmzVTjbvFcgq2TxSb-7v47HhvkLhhkid_H40Mn2F85A,4548
-airia/client/pipeline_execution/sync_pipeline_execution.py,sha256=7RptUGCO0XBighyC19oH1ulrENMwRxHyXzr9Y7pUilI,9614
+airia/client/pipeline_execution/async_pipeline_execution.py,sha256=bbQguB4-KNVgDDUMWgkaeLBH0fG-esLch39nmGWwyOU,18022
+airia/client/pipeline_execution/base_pipeline_execution.py,sha256=dkLb9zUrs0FSN5uZcjce9joOi80UavQxMmqTNTuRMiw,9608
+airia/client/pipeline_execution/sync_pipeline_execution.py,sha256=S1hLy5jhhMi4-iYXUcD5tsYFamtL_SOGvorqLB9RvXI,18106
 airia/client/pipelines_config/__init__.py,sha256=Mjn3ie-bQo6zjayxrJDvoJG2vKis72yGt_CQkvtREw4,163
 airia/client/pipelines_config/async_pipelines_config.py,sha256=H3W5yCUWsd93cq1i-PdQXyq6MG_dW0Jq7pjU2ZfqNf0,7611
 airia/client/pipelines_config/base_pipelines_config.py,sha256=UotK6-SB19fUSlUEHM_aNQTWD4GLlRUynF-Gs0Nvu0c,3917
@@ -55,8 +59,10 @@ airia/types/api/data_vector_search/get_file_chunks.py,sha256=ywP1zH_OXnyP5ZyJ_ep
 airia/types/api/deployments/__init__.py,sha256=ONQgkKr_8i6FhcBsHyzT4mYOP_H0OPUAsxvxroZWBMU,538
 airia/types/api/deployments/get_deployment.py,sha256=KBr9edTu-e-tC3u9bpAdz0CMWzk0DFQxusrVU6OZ7mc,4398
 airia/types/api/deployments/get_deployments.py,sha256=5Dm7pTkEFmIZ4p4Scle9x9p3Nqy5tnXxeft3H4O_Fa8,8718
-airia/types/api/pipeline_execution/__init__.py,sha256=01Cf0Cx-RZygtgQvYajSrikxmtUYbsDcBzFk7hkNnGc,360
-airia/types/api/pipeline_execution/_pipeline_execution.py,sha256=hMMR-nqxEoxvLU6-u4Tc_0gIQxoIZrdpP4oS-fXnO_M,2449
+airia/types/api/library/__init__.py,sha256=b2TLWT90EhQ41f8LcGMtvzEftd3ol_eKc41ian6zXX8,201
+airia/types/api/library/_library_models.py,sha256=sROM8OEFRWxvl3A6x-0PAezPwrh5UKV2e_7iUdDCIHs,10399
+airia/types/api/pipeline_execution/__init__.py,sha256=jlBLNN3yzd9TBQEfi7JJWlp3ur2NfXmON3YDnMkYdEY,674
+airia/types/api/pipeline_execution/_pipeline_execution.py,sha256=4E8rmr3Ok9EHkTzHjeZHQFlQVIXQIXIpbEeDtKAh3sQ,6425
 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/get_pipeline_config.py,sha256=UEGnvzBPHAJMJEcMOKLZWXenaL96iyYoWt0zZjyzEUg,23514
@@ -69,8 +75,8 @@ airia/types/api/store/get_files.py,sha256=v22zmOuTSFqzrS73L5JL_FgBeF5a5wutv1nK4I
 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.23.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
-airia-0.1.23.dist-info/METADATA,sha256=1t1g5uKAet39HfnPFTbcsKRTgojdfPSJyAmD6f3odcY,4506
-airia-0.1.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-airia-0.1.23.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
-airia-0.1.23.dist-info/RECORD,,
+airia-0.1.24.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
+airia-0.1.24.dist-info/METADATA,sha256=uoX2TWhod1iwgEwwv2v1mK1MBi90vCS2Rw_EPEBLFM0,4506
+airia-0.1.24.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+airia-0.1.24.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
+airia-0.1.24.dist-info/RECORD,,