airia 0.1.14__py3-none-any.whl → 0.1.16__py3-none-any.whl

airia/types/api/project/__init__.py

@@ -1,3 +1,3 @@
-from .get_projects import ProjectItem
+from .get_projects import DataSource, Pipeline, ProjectItem
 
-__all__ = ["ProjectItem"]
+__all__ = ["DataSource", "Pipeline", "ProjectItem"]

airia/types/api/store/__init__.py

@@ -1,4 +1,19 @@
-from .get_file import File, GetFileResponse
+from .get_file import (
+    DownloadInfo,
+    File,
+    GetFileResponse,
+    IngestionProcessingStatus,
+    PreviewInfo,
+    Timestamp,
+)
 from .get_files import GetFilesResponse
 
-__all__ = ["GetFileResponse", "GetFilesResponse", "File"]
+__all__ = [
+    "PreviewInfo",
+    "DownloadInfo",
+    "IngestionProcessingStatus",
+    "Timestamp",
+    "GetFileResponse",
+    "GetFilesResponse",
+    "File",
+]

airia/types/sse/sse_messages.py

@@ -19,34 +19,62 @@ class MessageType(str, Enum):
 
     These message types correspond to different events that occur during
     pipeline execution, agent processing, and streaming responses.
-    """
 
-    AGENT_PING = "AgentPingMessage"
-    AGENT_START = "AgentStartMessage"
-    AGENT_INPUT = "AgentInputMessage"
-    AGENT_END = "AgentEndMessage"
-    AGENT_STEP_START = "AgentStepStartMessage"
-    AGENT_STEP_HALT = "AgentStepHaltMessage"
-    AGENT_STEP_END = "AgentStepEndMessage"
-    AGENT_OUTPUT = "AgentOutputMessage"
-    AGENT_AGENT_CARD = "AgentAgentCardMessage"
-    AGENT_DATASEARCH = "AgentDatasearchMessage"
-    AGENT_INVOCATION = "AgentInvocationMessage"
-    AGENT_MODEL = "AgentModelMessage"
-    AGENT_PYTHON_CODE = "AgentPythonCodeMessage"
-    AGENT_TOOL_ACTION = "AgentToolActionMessage"
-    AGENT_MODEL_STREAM_START = "AgentModelStreamStartMessage"
-    AGENT_MODEL_STREAM_END = "AgentModelStreamEndMessage"
-    AGENT_MODEL_STREAM_ERROR = "AgentModelStreamErrorMessage"
-    AGENT_MODEL_STREAM_USAGE = "AgentModelStreamUsageMessage"
-    AGENT_MODEL_STREAM_FRAGMENT = "AgentModelStreamFragmentMessage"
-    MODEL_STREAM_FRAGMENT = "ModelStreamFragment"
-    AGENT_AGENT_CARD_STREAM_START = "AgentAgentCardStreamStartMessage"
-    AGENT_AGENT_CARD_STREAM_ERROR = "AgentAgentCardStreamErrorMessage"
-    AGENT_AGENT_CARD_STREAM_FRAGMENT = "AgentAgentCardStreamFragmentMessage"
-    AGENT_AGENT_CARD_STREAM_END = "AgentAgentCardStreamEndMessage"
-    AGENT_TOOL_REQUEST = "AgentToolRequestMessage"
-    AGENT_TOOL_RESPONSE = "AgentToolResponseMessage"
+    Attributes:
+        AGENT_PING: Ping message sent periodically to maintain connection health.
+        AGENT_START: Message indicating that an agent has started processing.
+        AGENT_INPUT: Message indicating that an agent has received input to process.
+        AGENT_END: Message indicating that an agent has finished processing.
+        AGENT_STEP_START: Message indicating that a processing step has started.
+        AGENT_STEP_HALT: Message indicating that a step has been halted pending approval.
+        AGENT_STEP_END: Message indicating that a processing step has completed.
+        AGENT_OUTPUT: Message containing the output result from a completed step.
+        AGENT_AGENT_CARD: Message indicating that an agent card step is being processed.
+        AGENT_DATASEARCH: Message indicating that data source search is being performed.
+        AGENT_INVOCATION: Message indicating that another agent is being invoked.
+        AGENT_MODEL: Message indicating that a language model is being called.
+        AGENT_PYTHON_CODE: Message indicating that Python code execution is taking place.
+        AGENT_TOOL_ACTION: Message indicating that a tool or external service is being called.
+        AGENT_MODEL_STREAM_START: Message indicating that model text streaming has begun.
+        AGENT_MODEL_STREAM_END: Message indicating that model text streaming has completed.
+        AGENT_MODEL_STREAM_ERROR: Message indicating that an error occurred during model streaming.
+        AGENT_MODEL_STREAM_USAGE: Message containing token usage and cost information for model calls.
+        AGENT_MODEL_STREAM_FRAGMENT: Fragment of streaming text content from a language model.
+        MODEL_STREAM_FRAGMENT: Alternative fragment message type for model streaming.
+        AGENT_AGENT_CARD_STREAM_START: Message indicating that agent card streaming has begun.
+        AGENT_AGENT_CARD_STREAM_ERROR: Message indicating that an error occurred during agent card streaming.
+        AGENT_AGENT_CARD_STREAM_FRAGMENT: Fragment of streaming agent card content.
+        AGENT_AGENT_CARD_STREAM_END: Message indicating that agent card streaming has completed.
+        AGENT_TOOL_REQUEST: Message indicating that a tool request has been initiated.
+        AGENT_TOOL_RESPONSE: Message indicating that a tool request has completed.
+    """
+
+    AGENT_PING: str = "AgentPingMessage"
+    AGENT_START: str = "AgentStartMessage"
+    AGENT_INPUT: str = "AgentInputMessage"
+    AGENT_END: str = "AgentEndMessage"
+    AGENT_STEP_START: str = "AgentStepStartMessage"
+    AGENT_STEP_HALT: str = "AgentStepHaltMessage"
+    AGENT_STEP_END: str = "AgentStepEndMessage"
+    AGENT_OUTPUT: str = "AgentOutputMessage"
+    AGENT_AGENT_CARD: str = "AgentAgentCardMessage"
+    AGENT_DATASEARCH: str = "AgentDatasearchMessage"
+    AGENT_INVOCATION: str = "AgentInvocationMessage"
+    AGENT_MODEL: str = "AgentModelMessage"
+    AGENT_PYTHON_CODE: str = "AgentPythonCodeMessage"
+    AGENT_TOOL_ACTION: str = "AgentToolActionMessage"
+    AGENT_MODEL_STREAM_START: str = "AgentModelStreamStartMessage"
+    AGENT_MODEL_STREAM_END: str = "AgentModelStreamEndMessage"
+    AGENT_MODEL_STREAM_ERROR: str = "AgentModelStreamErrorMessage"
+    AGENT_MODEL_STREAM_USAGE: str = "AgentModelStreamUsageMessage"
+    AGENT_MODEL_STREAM_FRAGMENT: str = "AgentModelStreamFragmentMessage"
+    MODEL_STREAM_FRAGMENT: str = "ModelStreamFragment"
+    AGENT_AGENT_CARD_STREAM_START: str = "AgentAgentCardStreamStartMessage"
+    AGENT_AGENT_CARD_STREAM_ERROR: str = "AgentAgentCardStreamErrorMessage"
+    AGENT_AGENT_CARD_STREAM_FRAGMENT: str = "AgentAgentCardStreamFragmentMessage"
+    AGENT_AGENT_CARD_STREAM_END: str = "AgentAgentCardStreamEndMessage"
+    AGENT_TOOL_REQUEST: str = "AgentToolRequestMessage"
+    AGENT_TOOL_RESPONSE: str = "AgentToolResponseMessage"
 
 
 class BaseSSEMessage(BaseModel):
@@ -55,6 +83,9 @@ class BaseSSEMessage(BaseModel):
 
     All SSE messages include a message_type field that identifies the specific
     type of event being reported.
+
+    Attributes:
+        message_type: The type of SSE message, identifying the specific event being reported.
     """
 
     model_config = ConfigDict(use_enum_values=True)
@@ -67,6 +98,10 @@ class AgentPingMessage(BaseSSEMessage):
 
     These messages help verify that the connection is still active during
     long-running pipeline executions.
+
+    Attributes:
+        message_type: Always set to AGENT_PING for ping messages.
+        timestamp: The time when the ping message was sent.
     """
 
     message_type: MessageType = MessageType.AGENT_PING
@@ -82,6 +117,10 @@ class BaseAgentMessage(BaseSSEMessage):
 
     All agent messages include identifiers for the specific agent
     and execution session.
+
+    Attributes:
+        agent_id: Unique identifier for the agent generating this message.
+        execution_id: Unique identifier for the current execution session.
     """
 
     agent_id: str
@@ -91,6 +130,11 @@ class BaseAgentMessage(BaseSSEMessage):
 class AgentStartMessage(BaseAgentMessage):
     """
     Message indicating that an agent has started processing.
+
+    Attributes:
+        message_type: Always set to AGENT_START for agent start messages.
+        agent_id: Unique identifier for the agent that started.
+        execution_id: Unique identifier for the execution session.
     """
 
     message_type: MessageType = MessageType.AGENT_START
@@ -99,6 +143,11 @@ class AgentStartMessage(BaseAgentMessage):
 class AgentInputMessage(BaseAgentMessage):
     """
     Message indicating that an agent has received input to process.
+
+    Attributes:
+        message_type: Always set to AGENT_INPUT for agent input messages.
+        agent_id: Unique identifier for the agent receiving input.
+        execution_id: Unique identifier for the execution session.
     """
 
     message_type: MessageType = MessageType.AGENT_INPUT
@@ -107,6 +156,11 @@ class AgentInputMessage(BaseAgentMessage):
 class AgentEndMessage(BaseAgentMessage):
     """
     Message indicating that an agent has finished processing.
+
+    Attributes:
+        message_type: Always set to AGENT_END for agent end messages.
+        agent_id: Unique identifier for the agent that finished.
+        execution_id: Unique identifier for the execution session.
     """
 
     message_type: MessageType = MessageType.AGENT_END
@@ -121,6 +175,13 @@ class BaseStepMessage(BaseAgentMessage):
 
     Steps represent discrete operations or tasks that an agent performs
     as part of its overall processing workflow.
+
+    Attributes:
+        agent_id: Unique identifier for the agent performing the step.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
     """
 
     step_id: str
@@ -131,6 +192,15 @@ class BaseStepMessage(BaseAgentMessage):
 class AgentStepStartMessage(BaseStepMessage):
     """
     Message indicating that a processing step has started.
+
+    Attributes:
+        message_type: Always set to AGENT_STEP_START for step start messages.
+        agent_id: Unique identifier for the agent performing the step.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step that started.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
+        start_time: The timestamp when the step began execution.
     """
 
     message_type: MessageType = MessageType.AGENT_STEP_START
@@ -143,6 +213,15 @@ class AgentStepHaltMessage(BaseStepMessage):
 
     This occurs when human approval is required before proceeding
     with potentially sensitive or high-impact operations.
+
+    Attributes:
+        message_type: Always set to AGENT_STEP_HALT for step halt messages.
+        agent_id: Unique identifier for the agent performing the step.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step that was halted.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
+        approval_id: Unique identifier for the approval request.
     """
 
     message_type: MessageType = MessageType.AGENT_STEP_HALT
@@ -154,6 +233,17 @@ class AgentStepEndMessage(BaseStepMessage):
     Message indicating that a processing step has completed.
 
     Includes timing information and the final status of the step.
+
+    Attributes:
+        message_type: Always set to AGENT_STEP_END for step end messages.
+        agent_id: Unique identifier for the agent performing the step.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step that ended.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
+        end_time: The timestamp when the step completed execution.
+        duration: The total time the step took to execute.
+        status: The final status of the step (e.g., 'success', 'failed').
     """
 
     message_type: MessageType = MessageType.AGENT_STEP_END
@@ -165,6 +255,15 @@ class AgentStepEndMessage(BaseStepMessage):
 class AgentOutputMessage(BaseStepMessage):
     """
     Message containing the output result from a completed step.
+
+    Attributes:
+        message_type: Always set to AGENT_OUTPUT for output messages.
+        agent_id: Unique identifier for the agent that produced the output.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step that generated output.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
+        step_result: The output result or content from the completed step.
     """
 
     message_type: MessageType = MessageType.AGENT_OUTPUT
@@ -180,6 +279,13 @@ class BaseStatusMessage(BaseStepMessage):
 
     Status messages provide real-time updates about what operations
     are being performed during step execution.
+
+    Attributes:
+        agent_id: Unique identifier for the agent performing the step.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
     """
 
     pass
@@ -191,6 +297,15 @@ class AgentAgentCardMessage(BaseStatusMessage):
 
     Agent cards represent interactive UI components or displays
     that provide rich information to users during pipeline execution.
+
+    Attributes:
+        message_type: Always set to AGENT_AGENT_CARD for agent card messages.
+        agent_id: Unique identifier for the agent processing the card.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
+        step_name: The name of the agent card step being processed.
     """
 
     message_type: MessageType = MessageType.AGENT_AGENT_CARD
@@ -203,6 +318,17 @@ class AgentDatasearchMessage(BaseStatusMessage):
 
     This message is sent when an agent is querying or searching
     through configured data sources to retrieve relevant information.
+
+    Attributes:
+        message_type: Always set to AGENT_DATASEARCH for data search messages.
+        agent_id: Unique identifier for the agent performing the search.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
+        datastore_id: Unique identifier for the data source being searched.
+        datastore_type: The type of data source (e.g., 'database', 'file', 'api').
+        datastore_name: Human-readable name of the data source.
     """
 
     message_type: MessageType = MessageType.AGENT_DATASEARCH
@@ -217,6 +343,15 @@ class AgentInvocationMessage(BaseStatusMessage):
 
     This occurs when the current agent calls or delegates work
     to another specialized agent in the pipeline.
+
+    Attributes:
+        message_type: Always set to AGENT_INVOCATION for agent invocation messages.
+        agent_id: Unique identifier for the agent making the invocation.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
+        agent_name: The name of the agent being invoked.
     """
 
     message_type: MessageType = MessageType.AGENT_INVOCATION
@@ -229,6 +364,15 @@ class AgentModelMessage(BaseStatusMessage):
 
     This message is sent when an agent begins interacting with
     a language model for text generation or processing.
+
+    Attributes:
+        message_type: Always set to AGENT_MODEL for model messages.
+        agent_id: Unique identifier for the agent calling the model.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
+        model_name: The name of the language model being called.
     """
 
     message_type: MessageType = MessageType.AGENT_MODEL
@@ -241,6 +385,15 @@ class AgentPythonCodeMessage(BaseStatusMessage):
 
     This message is sent when an agent executes custom Python code
     blocks as part of its processing workflow.
+
+    Attributes:
+        message_type: Always set to AGENT_PYTHON_CODE for Python code messages.
+        agent_id: Unique identifier for the agent executing the code.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
+        step_name: The name of the Python code step being executed.
     """
 
     message_type: MessageType = MessageType.AGENT_PYTHON_CODE
@@ -253,6 +406,16 @@ class AgentToolActionMessage(BaseStatusMessage):
 
     This message is sent when an agent invokes an external tool,
     API, or service to perform a specific action or retrieve data.
+
+    Attributes:
+        message_type: Always set to AGENT_TOOL_ACTION for tool action messages.
+        agent_id: Unique identifier for the agent calling the tool.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
+        step_name: The name of the tool action step being performed.
+        tool_name: The name of the external tool being called.
     """
 
     message_type: MessageType = MessageType.AGENT_TOOL_ACTION
@@ -269,6 +432,12 @@ class BaseModelStreamMessage(BaseAgentMessage):
 
     Model streaming allows real-time display of text generation
     as it occurs, providing better user experience for long responses.
+
+    Attributes:
+        agent_id: Unique identifier for the agent performing the streaming.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        stream_id: Unique identifier for the streaming session.
     """
 
     step_id: str
@@ -278,6 +447,14 @@ class BaseModelStreamMessage(BaseAgentMessage):
 class AgentModelStreamStartMessage(BaseModelStreamMessage):
     """
     Message indicating that model text streaming has begun.
+
+    Attributes:
+        message_type: Always set to AGENT_MODEL_STREAM_START for stream start messages.
+        agent_id: Unique identifier for the agent performing the streaming.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        stream_id: Unique identifier for the streaming session.
+        model_name: The name of the language model being streamed.
     """
 
     message_type: MessageType = MessageType.AGENT_MODEL_STREAM_START
@@ -287,6 +464,14 @@ class AgentModelStreamStartMessage(BaseModelStreamMessage):
 class AgentModelStreamErrorMessage(BaseModelStreamMessage):
     """
     Message indicating that an error occurred during model streaming.
+
+    Attributes:
+        message_type: Always set to AGENT_MODEL_STREAM_ERROR for stream error messages.
+        agent_id: Unique identifier for the agent performing the streaming.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        stream_id: Unique identifier for the streaming session.
+        error_message: Description of the error that occurred during streaming.
     """
 
     message_type: MessageType = MessageType.AGENT_MODEL_STREAM_ERROR
@@ -299,6 +484,15 @@ class AgentModelStreamFragmentMessage(BaseModelStreamMessage):
 
     These messages contain individual chunks of text as they are generated
     by the model, allowing for real-time display of results.
+
+    Attributes:
+        message_type: Always set to AGENT_MODEL_STREAM_FRAGMENT for fragment messages.
+        agent_id: Unique identifier for the agent performing the streaming.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        stream_id: Unique identifier for the streaming session.
+        index: The sequential index of this fragment within the stream.
+        content: The text content of this fragment, or None if no content.
     """
 
     message_type: MessageType = MessageType.AGENT_MODEL_STREAM_FRAGMENT
@@ -309,6 +503,15 @@ class AgentModelStreamFragmentMessage(BaseModelStreamMessage):
 class AgentModelStreamEndMessage(BaseModelStreamMessage):
     """
     Message indicating that model text streaming has completed.
+
+    Attributes:
+        message_type: Always set to AGENT_MODEL_STREAM_END for stream end messages.
+        agent_id: Unique identifier for the agent performing the streaming.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        stream_id: Unique identifier for the streaming session.
+        content_id: Unique identifier for the generated content.
+        duration: Optional duration of the streaming session in seconds.
     """
 
     message_type: MessageType = MessageType.AGENT_MODEL_STREAM_END
@@ -319,6 +522,15 @@ class AgentModelStreamEndMessage(BaseModelStreamMessage):
 class AgentModelStreamUsageMessage(BaseModelStreamMessage):
     """
     Message containing token usage and cost information for model calls.
+
+    Attributes:
+        message_type: Always set to AGENT_MODEL_STREAM_USAGE for usage messages.
+        agent_id: Unique identifier for the agent performing the streaming.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        stream_id: Unique identifier for the streaming session.
+        token: Optional number of tokens used during the model call.
+        tokens_cost: Optional cost in dollars for the tokens used.
     """
 
     message_type: MessageType = MessageType.AGENT_MODEL_STREAM_USAGE
@@ -335,6 +547,12 @@ class BaseAgentAgentCardStreamMessage(BaseAgentMessage):
 
     Agent card streaming allows real-time updates to interactive
     UI components during their generation or processing.
+
+    Attributes:
+        agent_id: Unique identifier for the agent performing the streaming.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        stream_id: Unique identifier for the streaming session.
     """
 
     step_id: str
@@ -344,6 +562,14 @@ class BaseAgentAgentCardStreamMessage(BaseAgentMessage):
 class AgentAgentCardStreamStartMessage(BaseAgentAgentCardStreamMessage):
     """
     Message indicating that agent card streaming has begun.
+
+    Attributes:
+        message_type: Always set to AGENT_AGENT_CARD_STREAM_START for card stream start messages.
+        agent_id: Unique identifier for the agent performing the streaming.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        stream_id: Unique identifier for the streaming session.
+        content: Optional initial content for the agent card.
     """
 
     message_type: MessageType = MessageType.AGENT_AGENT_CARD_STREAM_START
@@ -353,6 +579,14 @@ class AgentAgentCardStreamStartMessage(BaseAgentAgentCardStreamMessage):
 class AgentAgentCardStreamErrorMessage(BaseAgentAgentCardStreamMessage):
     """
     Message indicating that an error occurred during agent card streaming.
+
+    Attributes:
+        message_type: Always set to AGENT_AGENT_CARD_STREAM_ERROR for card stream error messages.
+        agent_id: Unique identifier for the agent performing the streaming.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        stream_id: Unique identifier for the streaming session.
+        error_message: Description of the error that occurred during card streaming.
     """
 
     message_type: MessageType = MessageType.AGENT_AGENT_CARD_STREAM_ERROR
@@ -365,6 +599,15 @@ class AgentAgentCardStreamFragmentMessage(BaseAgentAgentCardStreamMessage):
 
     These messages contain individual chunks of agent card data
     as they are generated, allowing for real-time UI updates.
+
+    Attributes:
+        message_type: Always set to AGENT_AGENT_CARD_STREAM_FRAGMENT for card fragment messages.
+        agent_id: Unique identifier for the agent performing the streaming.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        stream_id: Unique identifier for the streaming session.
+        index: The sequential index of this fragment within the stream.
+        content: The card content of this fragment, or None if no content.
     """
 
     message_type: MessageType = MessageType.AGENT_AGENT_CARD_STREAM_FRAGMENT
@@ -375,6 +618,14 @@ class AgentAgentCardStreamFragmentMessage(BaseAgentAgentCardStreamMessage):
 class AgentAgentCardStreamEndMessage(BaseAgentAgentCardStreamMessage):
     """
     Message indicating that agent card streaming has completed.
+
+    Attributes:
+        message_type: Always set to AGENT_AGENT_CARD_STREAM_END for card stream end messages.
+        agent_id: Unique identifier for the agent performing the streaming.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        stream_id: Unique identifier for the streaming session.
+        content: Optional final content for the agent card.
     """
 
     message_type: MessageType = MessageType.AGENT_AGENT_CARD_STREAM_END
@@ -390,6 +641,15 @@ class BaseAgentToolMessage(BaseStepMessage):
 
     Tool messages track the lifecycle of external tool or service
     calls made by agents during pipeline execution.
+
+    Attributes:
+        agent_id: Unique identifier for the agent performing the step.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
+        id: Unique identifier for the tool execution.
+        name: The name of the tool being executed.
     """
 
     id: str
@@ -402,6 +662,16 @@ class AgentToolRequestMessage(BaseAgentToolMessage):
 
     This message is sent when an agent begins calling an external
     tool or service to perform a specific operation.
+
+    Attributes:
+        message_type: Always set to AGENT_TOOL_REQUEST for tool request messages.
+        agent_id: Unique identifier for the agent making the tool request.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
+        id: Unique identifier for the tool execution.
+        name: The name of the tool being requested.
     """
 
     message_type: MessageType = MessageType.AGENT_TOOL_REQUEST
@@ -413,6 +683,18 @@ class AgentToolResponseMessage(BaseAgentToolMessage):
 
     This message contains the results and timing information
     from a completed tool or service call.
+
+    Attributes:
+        message_type: Always set to AGENT_TOOL_RESPONSE for tool response messages.
+        agent_id: Unique identifier for the agent that made the tool request.
+        execution_id: Unique identifier for the execution session.
+        step_id: Unique identifier for the processing step.
+        step_type: The type/category of the processing step.
+        step_title: Optional human-readable title for the step.
+        id: Unique identifier for the tool execution.
+        name: The name of the tool that was executed.
+        duration: The time it took for the tool to complete execution.
+        success: Whether the tool execution was successful.
     """
 
     message_type: MessageType = MessageType.AGENT_TOOL_RESPONSE

{airia-0.1.14.dist-info → airia-0.1.16.dist-info}/METADATA

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