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

airia/types/sse/sse_messages.py

@@ -5,6 +5,7 @@ This module defines all possible SSE message types that can be received during
 pipeline execution, including agent lifecycle events, processing steps, model
 streaming, and tool execution messages.
 """
+
 from datetime import datetime, time
 from enum import Enum
 from typing import Optional, Union
@@ -15,10 +16,11 @@ from pydantic import BaseModel, ConfigDict
 class MessageType(str, Enum):
     """
     Enumeration of all possible SSE message types from the Airia API.
-    
+
     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"
@@ -50,10 +52,11 @@ class MessageType(str, Enum):
 class BaseSSEMessage(BaseModel):
     """
     Base class for all Server-Sent Event (SSE) messages from the Airia API.
-    
+
     All SSE messages include a message_type field that identifies the specific
     type of event being reported.
     """
+
     model_config = ConfigDict(use_enum_values=True)
     message_type: MessageType
 
@@ -61,10 +64,11 @@ class BaseSSEMessage(BaseModel):
 class AgentPingMessage(BaseSSEMessage):
     """
     Ping message sent periodically to maintain connection health.
-    
+
     These messages help verify that the connection is still active during
     long-running pipeline executions.
     """
+
     message_type: MessageType = MessageType.AGENT_PING
     timestamp: datetime
 
@@ -75,10 +79,11 @@ class AgentPingMessage(BaseSSEMessage):
 class BaseAgentMessage(BaseSSEMessage):
     """
     Base class for messages related to agent execution.
-    
+
     All agent messages include identifiers for the specific agent
     and execution session.
     """
+
     agent_id: str
     execution_id: str
 
@@ -87,6 +92,7 @@ class AgentStartMessage(BaseAgentMessage):
     """
     Message indicating that an agent has started processing.
     """
+
     message_type: MessageType = MessageType.AGENT_START
 
 
@@ -94,6 +100,7 @@ class AgentInputMessage(BaseAgentMessage):
     """
     Message indicating that an agent has received input to process.
     """
+
     message_type: MessageType = MessageType.AGENT_INPUT
 
 
@@ -101,6 +108,7 @@ class AgentEndMessage(BaseAgentMessage):
     """
     Message indicating that an agent has finished processing.
     """
+
     message_type: MessageType = MessageType.AGENT_END
 
 
@@ -110,10 +118,11 @@ class AgentEndMessage(BaseAgentMessage):
 class BaseStepMessage(BaseAgentMessage):
     """
     Base class for messages related to individual processing steps within an agent.
-    
+
     Steps represent discrete operations or tasks that an agent performs
     as part of its overall processing workflow.
     """
+
     step_id: str
     step_type: str
     step_title: Optional[str] = None
@@ -123,6 +132,7 @@ class AgentStepStartMessage(BaseStepMessage):
     """
     Message indicating that a processing step has started.
     """
+
     message_type: MessageType = MessageType.AGENT_STEP_START
     start_time: datetime
 
@@ -130,10 +140,11 @@ class AgentStepStartMessage(BaseStepMessage):
 class AgentStepHaltMessage(BaseStepMessage):
     """
     Message indicating that a step has been halted pending approval.
-    
+
     This occurs when human approval is required before proceeding
     with potentially sensitive or high-impact operations.
     """
+
     message_type: MessageType = MessageType.AGENT_STEP_HALT
     approval_id: str
 
@@ -141,9 +152,10 @@ class AgentStepHaltMessage(BaseStepMessage):
 class AgentStepEndMessage(BaseStepMessage):
     """
     Message indicating that a processing step has completed.
-    
+
     Includes timing information and the final status of the step.
     """
+
     message_type: MessageType = MessageType.AGENT_STEP_END
     end_time: datetime
     duration: time
@@ -154,6 +166,7 @@ class AgentOutputMessage(BaseStepMessage):
     """
     Message containing the output result from a completed step.
     """
+
     message_type: MessageType = MessageType.AGENT_OUTPUT
     step_result: str
 
@@ -164,20 +177,22 @@ class AgentOutputMessage(BaseStepMessage):
 class BaseStatusMessage(BaseStepMessage):
     """
     Base class for status update messages within processing steps.
-    
+
     Status messages provide real-time updates about what operations
     are being performed during step execution.
     """
+
     pass
 
 
 class AgentAgentCardMessage(BaseStatusMessage):
     """
     Message indicating that an agent card step is being processed.
-    
+
     Agent cards represent interactive UI components or displays
     that provide rich information to users during pipeline execution.
     """
+
     message_type: MessageType = MessageType.AGENT_AGENT_CARD
     step_name: str
 
@@ -185,10 +200,11 @@ class AgentAgentCardMessage(BaseStatusMessage):
 class AgentDatasearchMessage(BaseStatusMessage):
     """
     Message indicating that data source search is being performed.
-    
+
     This message is sent when an agent is querying or searching
     through configured data sources to retrieve relevant information.
     """
+
     message_type: MessageType = MessageType.AGENT_DATASEARCH
     datastore_id: str
     datastore_type: str
@@ -198,10 +214,11 @@ class AgentDatasearchMessage(BaseStatusMessage):
 class AgentInvocationMessage(BaseStatusMessage):
     """
     Message indicating that another agent is being invoked.
-    
+
     This occurs when the current agent calls or delegates work
     to another specialized agent in the pipeline.
     """
+
     message_type: MessageType = MessageType.AGENT_INVOCATION
     agent_name: str
 
@@ -209,10 +226,11 @@ class AgentInvocationMessage(BaseStatusMessage):
 class AgentModelMessage(BaseStatusMessage):
     """
     Message indicating that a language model is being called.
-    
+
     This message is sent when an agent begins interacting with
     a language model for text generation or processing.
     """
+
     message_type: MessageType = MessageType.AGENT_MODEL
     model_name: str
 
@@ -220,10 +238,11 @@ class AgentModelMessage(BaseStatusMessage):
 class AgentPythonCodeMessage(BaseStatusMessage):
     """
     Message indicating that Python code execution is taking place.
-    
+
     This message is sent when an agent executes custom Python code
     blocks as part of its processing workflow.
     """
+
     message_type: MessageType = MessageType.AGENT_PYTHON_CODE
     step_name: str
 
@@ -231,10 +250,11 @@ class AgentPythonCodeMessage(BaseStatusMessage):
 class AgentToolActionMessage(BaseStatusMessage):
     """
     Message indicating that a tool or external service is being called.
-    
+
     This message is sent when an agent invokes an external tool,
     API, or service to perform a specific action or retrieve data.
     """
+
     message_type: MessageType = MessageType.AGENT_TOOL_ACTION
     step_name: str
     tool_name: str
@@ -246,10 +266,11 @@ class AgentToolActionMessage(BaseStatusMessage):
 class BaseModelStreamMessage(BaseAgentMessage):
     """
     Base class for language model streaming messages.
-    
+
     Model streaming allows real-time display of text generation
     as it occurs, providing better user experience for long responses.
     """
+
     step_id: str
     stream_id: str
 
@@ -258,6 +279,7 @@ class AgentModelStreamStartMessage(BaseModelStreamMessage):
     """
     Message indicating that model text streaming has begun.
     """
+
     message_type: MessageType = MessageType.AGENT_MODEL_STREAM_START
     model_name: str
 
@@ -266,6 +288,7 @@ class AgentModelStreamErrorMessage(BaseModelStreamMessage):
     """
     Message indicating that an error occurred during model streaming.
     """
+
     message_type: MessageType = MessageType.AGENT_MODEL_STREAM_ERROR
     error_message: str
 
@@ -273,10 +296,11 @@ class AgentModelStreamErrorMessage(BaseModelStreamMessage):
 class AgentModelStreamFragmentMessage(BaseModelStreamMessage):
     """
     Fragment of streaming text content from a language model.
-    
+
     These messages contain individual chunks of text as they are generated
     by the model, allowing for real-time display of results.
     """
+
     message_type: MessageType = MessageType.AGENT_MODEL_STREAM_FRAGMENT
     index: int
     content: Optional[str] = None
@@ -286,6 +310,7 @@ class AgentModelStreamEndMessage(BaseModelStreamMessage):
     """
     Message indicating that model text streaming has completed.
     """
+
     message_type: MessageType = MessageType.AGENT_MODEL_STREAM_END
     content_id: str
     duration: Optional[float] = None
@@ -295,6 +320,7 @@ class AgentModelStreamUsageMessage(BaseModelStreamMessage):
     """
     Message containing token usage and cost information for model calls.
     """
+
     message_type: MessageType = MessageType.AGENT_MODEL_STREAM_USAGE
     token: Optional[int] = None
     tokens_cost: Optional[float] = None
@@ -306,10 +332,11 @@ class AgentModelStreamUsageMessage(BaseModelStreamMessage):
 class BaseAgentAgentCardStreamMessage(BaseAgentMessage):
     """
     Base class for agent card streaming messages.
-    
+
     Agent card streaming allows real-time updates to interactive
     UI components during their generation or processing.
     """
+
     step_id: str
     stream_id: str
 
@@ -318,6 +345,7 @@ class AgentAgentCardStreamStartMessage(BaseAgentAgentCardStreamMessage):
     """
     Message indicating that agent card streaming has begun.
     """
+
     message_type: MessageType = MessageType.AGENT_AGENT_CARD_STREAM_START
     content: Optional[str] = None
 
@@ -326,6 +354,7 @@ class AgentAgentCardStreamErrorMessage(BaseAgentAgentCardStreamMessage):
     """
     Message indicating that an error occurred during agent card streaming.
     """
+
     message_type: MessageType = MessageType.AGENT_AGENT_CARD_STREAM_ERROR
     error_message: str
 
@@ -333,10 +362,11 @@ class AgentAgentCardStreamErrorMessage(BaseAgentAgentCardStreamMessage):
 class AgentAgentCardStreamFragmentMessage(BaseAgentAgentCardStreamMessage):
     """
     Fragment of streaming agent card content.
-    
+
     These messages contain individual chunks of agent card data
     as they are generated, allowing for real-time UI updates.
     """
+
     message_type: MessageType = MessageType.AGENT_AGENT_CARD_STREAM_FRAGMENT
     index: int
     content: Optional[str]
@@ -346,6 +376,7 @@ class AgentAgentCardStreamEndMessage(BaseAgentAgentCardStreamMessage):
     """
     Message indicating that agent card streaming has completed.
     """
+
     message_type: MessageType = MessageType.AGENT_AGENT_CARD_STREAM_END
     content: Optional[str] = None
 
@@ -356,10 +387,11 @@ class AgentAgentCardStreamEndMessage(BaseAgentAgentCardStreamMessage):
 class BaseAgentToolMessage(BaseStepMessage):
     """
     Base class for tool execution messages.
-    
+
     Tool messages track the lifecycle of external tool or service
     calls made by agents during pipeline execution.
     """
+
     id: str
     name: str
 
@@ -367,20 +399,22 @@ class BaseAgentToolMessage(BaseStepMessage):
 class AgentToolRequestMessage(BaseAgentToolMessage):
     """
     Message indicating that a tool request has been initiated.
-    
+
     This message is sent when an agent begins calling an external
     tool or service to perform a specific operation.
     """
+
     message_type: MessageType = MessageType.AGENT_TOOL_REQUEST
 
 
 class AgentToolResponseMessage(BaseAgentToolMessage):
     """
     Message indicating that a tool request has completed.
-    
+
     This message contains the results and timing information
     from a completed tool or service call.
     """
+
     message_type: MessageType = MessageType.AGENT_TOOL_RESPONSE
     duration: time
     success: bool

airia/utils/sse_parser.py

@@ -5,6 +5,7 @@ This module provides functions for parsing SSE streams from both synchronous and
 asynchronous HTTP responses. It handles the SSE protocol format and converts
 raw stream data into typed message objects.
 """
+
 import json
 import re
 from typing import AsyncIterable, AsyncIterator, Iterable, Iterator
@@ -28,7 +29,7 @@ def _to_snake_case(name: str) -> str:
 def parse_sse_stream_chunked(stream_chunks: Iterable[bytes]) -> Iterator[SSEMessage]:
     """
     Parse Server-Sent Events (SSE) stream from an iterable of byte chunks.
-    
+
     This function processes streaming data from HTTP responses that follow the SSE protocol,
     parsing event blocks and converting them into typed SSE message objects. It handles
     incomplete chunks by buffering data until complete events are received.
@@ -38,7 +39,7 @@ def parse_sse_stream_chunked(stream_chunks: Iterable[bytes]) -> Iterator[SSEMess
 
     Yields:
         SSEMessage: Typed SSE message objects corresponding to the parsed events
-        
+
     Note:
         Events are expected to follow the SSE format with 'event:' and 'data:' lines,
         terminated by double newlines (\n\n). The data portion should contain valid JSON.
@@ -78,7 +79,7 @@ async def async_parse_sse_stream_chunked(
 ) -> AsyncIterator[SSEMessage]:
     """
     Asynchronously parse Server-Sent Events (SSE) stream from an async iterable of byte chunks.
-    
+
     This is the async version of parse_sse_stream_chunked, designed for use with
     asynchronous HTTP clients. It processes streaming data that follows the SSE protocol,
     parsing event blocks and converting them into typed SSE message objects.
@@ -88,7 +89,7 @@ async def async_parse_sse_stream_chunked(
 
     Yields:
         SSEMessage: Typed SSE message objects corresponding to the parsed events
-        
+
     Note:
         Events are expected to follow the SSE format with 'event:' and 'data:' lines,
         terminated by double newlines (\n\n). The data portion should contain valid JSON.

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airia
-Version: 0.1.13
+Version: 0.1.14
 Summary: Python SDK for Airia API
 Author-email: Airia LLC <support@airia.com>
 License: MIT
@@ -47,7 +47,9 @@ The Airia Python library provides a clean and intuitive interface to interact wi
 
 ## Documentation and Quick Start Guides
 
-Run locally with `mkdocs`:
+Full documentation and quick start guides are available [here](https://airiallc.github.io/airia-python).
+
+You can also run the documentation page locally with `mkdocs`:
 
 1. [Install development dependencies](#install-with-development-dependencies)
 2. Run `mkdocs serve` to start the local server

airia-0.1.14.dist-info/RECORD

@@ -0,0 +1,55 @@
+airia/__init__.py,sha256=T39gO8E5T5zxlw-JP78ruxOu7-LeKOJCJzz6t40kdQo,231
+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=-PHKGBmbtw1SL589052-mZ11YfmVxcd5_iCpw9LzoHs,6792
+airia/client/base_client.py,sha256=ftyLi2E3Hb4CQFYfiknitJA0sWDTylCiUbde0qkWYhg,3182
+airia/client/sync_client.py,sha256=5xSoYjJDDbtnEd5mFpQCvvHJIKPXiAbjZDilbHuiPMA,6671
+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=nYZIpWw-we0WMfBIOv1ZD5cNDT6hdNrbrFSTuBb6rbU,3838
+airia/client/_request_handler/sync_request_handler.py,sha256=IFMauU8Fk3AvrMQrSOMTuWwUw44zmWimeW8LkIYj7Ws,10132
+airia/client/conversations/__init__.py,sha256=5gbcHEYLHOn5cK95tlM52GPOmar8gpyK-BivyASmFEo,149
+airia/client/conversations/async_conversations.py,sha256=XbSErt_6RKcQwDdGHKr_roSf2EkxhR6ogBiN3oMFNhk,7448
+airia/client/conversations/base_conversations.py,sha256=YpIvA66xFZbOC4L8P17pISCqyWlJvju34FHX7Pcj8Aw,4848
+airia/client/conversations/sync_conversations.py,sha256=IelRpGuoV0sBwOfZd44igV3i5P35Hv-zzZwCC7y6mlA,7200
+airia/client/pipeline_execution/__init__.py,sha256=7qEZsPRTLySC71zlwYioBuJs6B4BwRCgFL3TQyFWXmc,175
+airia/client/pipeline_execution/async_pipeline_execution.py,sha256=tesbtLYuTffPPHb3a2lw8aNzSzuDQ80yRWboSFEhyRs,7477
+airia/client/pipeline_execution/base_pipeline_execution.py,sha256=Jkpp_D2OrNg3koudrIua-d0VKwI4hlATDqi8WwRCifY,3965
+airia/client/pipeline_execution/sync_pipeline_execution.py,sha256=wIifP_KjRvBgXeqs1g237IlAXiuU7ThxS5YDj6qB7gM,7392
+airia/client/pipelines_config/__init__.py,sha256=Mjn3ie-bQo6zjayxrJDvoJG2vKis72yGt_CQkvtREw4,163
+airia/client/pipelines_config/async_pipelines_config.py,sha256=iC6B5nVlQOA02LsUY_oBWTGpJ6bf8micwrGwWtXZhBI,4971
+airia/client/pipelines_config/base_pipelines_config.py,sha256=gfbjXzNUDxsRS2a55YtGMd-k7ii0MHnonflKVH72qe0,2701
+airia/client/pipelines_config/sync_pipelines_config.py,sha256=vJuNpbTtt13G2KuJZnvKaD52MQ-Ru2b65QBylIad50s,4894
+airia/client/project/__init__.py,sha256=GGEGfxtNzSO_BMAJ8Wfo8ZTq8BIdR6MHf6gSwhPv9mE,113
+airia/client/project/async_project.py,sha256=OrIiqopKSXAhkYJi6yjoR-YYVUxzll1ldDJkJW1rIBM,4534
+airia/client/project/base_project.py,sha256=4vP_dKGAt17mdzp3eW3ER8E0C0XMQ9P7JAEH2EH0imE,2432
+airia/client/project/sync_project.py,sha256=sbDt02WKH-65p_hCgz4tpUJVhp0T1oneeXPccRQ06sw,4449
+airia/client/store/__init__.py,sha256=-rbtYQ8PTaFA6Fs2dm7Db2C3RDNvagaK3iE4PpLRcjg,101
+airia/client/store/async_store.py,sha256=wlYbK2AlTuJCrk3otdpO68W208166xyKpZymtuA5PyA,13557
+airia/client/store/base_store.py,sha256=RUIGkKb2KiKZfzmDJIiPrLSes_VAnJA4xouIqxkEazI,8106
+airia/client/store/sync_store.py,sha256=DNBhTckdoVnTqTVwFp_J42l-XkSHX3sCKVR7J4b79RU,12578
+airia/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+airia/types/_api_version.py,sha256=Uzom6O2ZG92HN_Z2h-lTydmO2XYX9RVs4Yi4DJmXytE,255
+airia/types/_request_data.py,sha256=q8t7KfO2WvgbPVYPvPWiwYb8LyP0kovlOgHFhZIU6ns,1278
+airia/types/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+airia/types/api/conversations/__init__.py,sha256=POaFtK2Wi5qpObAEz5--vJf3AT6vfPIb0P0Qe2_YvLQ,149
+airia/types/api/conversations/_conversations.py,sha256=O7VHs2_Ykg0s6Phe0sAVvbO7YVu6d6R2M2-JkWAipxI,4845
+airia/types/api/pipeline_execution/__init__.py,sha256=01Cf0Cx-RZygtgQvYajSrikxmtUYbsDcBzFk7hkNnGc,360
+airia/types/api/pipeline_execution/_pipeline_execution.py,sha256=cJ2icsQvG5K15crNNbymje9B2tJduc_D64OnNXF043U,2429
+airia/types/api/pipelines_config/__init__.py,sha256=A_udsGV3tvMoGu4HcK91r0HxdT47AEn8uobN_nj1TJ0,100
+airia/types/api/pipelines_config/get_pipeline_config.py,sha256=M-2w5ER8MsdTJobVeArBwc9fhqXMDOuhsTpIrT-Rh1Q,15365
+airia/types/api/project/__init__.py,sha256=W2X2D9NrIDXH3bXWr2FNt9SVe999gsWKlhtJ0zY1VsU,65
+airia/types/api/project/get_projects.py,sha256=Ot8mq6VnXrGXZs7FQ0UuRYSyuCeER7YeCCZlGb4ZyQo,3646
+airia/types/api/store/__init__.py,sha256=sj4_I4lfaHnw5U_-dRmyK5Nr-wuAEjT5S3upJxbuRtc,143
+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/sse/__init__.py,sha256=KWnNTfsQnthfrU128pUX6ounvSS7DvjC-Y21FE-OdMk,1863
+airia/types/sse/sse_messages.py,sha256=-jnOD-ZkmIC6tm1occDLkZ88l_ZHxMFQPAgJf4Z4HNc,14270
+airia/utils/sse_parser.py,sha256=XCTkuaroYWaVQOgBq8VpbseQYSAVruF69AvKUwZQKTA,4251
+airia-0.1.14.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
+airia-0.1.14.dist-info/METADATA,sha256=kP3EFgdsqUsR5o1l0jIG156RyJjGmGP11Gw02ISTpkU,4506
+airia-0.1.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+airia-0.1.14.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
+airia-0.1.14.dist-info/RECORD,,