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

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.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airia
-Version: 0.1.13
+Version: 0.1.15
 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.15.dist-info/RECORD

@@ -0,0 +1,62 @@
+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=aeFRugnHfQHGVx3R3XVFGC1PH0q_OqCbZjGKNopaE4g,6901
+airia/client/base_client.py,sha256=ftyLi2E3Hb4CQFYfiknitJA0sWDTylCiUbde0qkWYhg,3182
+airia/client/sync_client.py,sha256=p2qUfDnzBMgiAjoECI43O15sLOoL9nrtcCkA9LT0bPU,6770
+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/deployments/__init__.py,sha256=Xu_MqbeUrtqYE4tH6hbmvUk8eQScWJ6MEyYFlVEwlRw,310
+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/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=CzwEMQKnJlMPPixvRenSP4AEUtBEhc2H_ApxnGQ2aIA,2442
+airia/client/pipelines_config/base_pipelines_config.py,sha256=w7o9eZL0xLCpAPOrQlPayUqW9YD0MZE3PifSz4-QKJ8,1492
+airia/client/pipelines_config/sync_pipelines_config.py,sha256=ENlgwwaw5a1rjJM6GXMFkdaM7UINKJVyPzOy528SQfs,2399
+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=W6GlNVZCAY5eViJOoPl1bY9_etRBWeUnYZJJt7WHtfI,269
+airia/types/api/conversations/_conversations.py,sha256=O7VHs2_Ykg0s6Phe0sAVvbO7YVu6d6R2M2-JkWAipxI,4845
+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=cJ2icsQvG5K15crNNbymje9B2tJduc_D64OnNXF043U,2429
+airia/types/api/pipelines_config/__init__.py,sha256=peAX82IUN_ZM1kp3Opv0Zk-1NgIHmeL_IOUihBHfnUQ,780
+airia/types/api/pipelines_config/get_pipeline_config.py,sha256=UEGnvzBPHAJMJEcMOKLZWXenaL96iyYoWt0zZjyzEUg,23514
+airia/types/api/project/__init__.py,sha256=ervHvCeqt08JkMRsSrG1ZnQshE70of-8kf4VeW2HG9c,113
+airia/types/api/project/get_projects.py,sha256=Ot8mq6VnXrGXZs7FQ0UuRYSyuCeER7YeCCZlGb4ZyQo,3646
+airia/types/api/store/__init__.py,sha256=BgViwV_SHE9cxtilPnA2xWRk6MkAbxYxansmGeZR3nw,341
+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=asq9KG5plT2XSgQMz-Nqo0WcKlXvE8UT3E-WLhCegPk,30244
+airia/utils/sse_parser.py,sha256=XCTkuaroYWaVQOgBq8VpbseQYSAVruF69AvKUwZQKTA,4251
+airia-0.1.15.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
+airia-0.1.15.dist-info/METADATA,sha256=rp4FO6skpbu9Lz13R0gyI61GnVrHAcnwHVwMZS9s22o,4506
+airia-0.1.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+airia-0.1.15.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
+airia-0.1.15.dist-info/RECORD,,

airia/types/api/get_pipeline_config.py

@@ -1,214 +0,0 @@
-"""
-Pydantic models for pipeline configuration API responses.
-
-This module defines comprehensive data structures for pipeline configuration exports,
-including all components like agents, models, tools, data sources, and deployment settings.
-"""
-from typing import Any, Dict, List, Optional
-
-from pydantic import BaseModel, Field
-
-
-class Metadata(BaseModel):
-    """
-    Pipeline metadata and export configuration.
-    
-    Contains version information, export settings, and descriptive metadata
-    about the pipeline configuration.
-    """
-    id: str
-    export_version: str = Field(alias="exportVersion")
-    tagline: Optional[str] = None
-    agent_description: Optional[str] = Field(alias="agentDescription", default=None)
-    industry: Optional[str] = None
-    tasks: Optional[str] = None
-    credential_export_option: str = Field(alias="credentialExportOption")
-    data_source_export_option: str = Field(alias="dataSourceExportOption")
-    version_information: str = Field(alias="versionInformation")
-    state: str
-
-
-class Agent(BaseModel):
-    """
-    AI agent configuration and workflow definition.
-    
-    Represents the core agent that executes the pipeline, including its
-    identity, industry specialization, and step-by-step workflow configuration.
-    """
-    name: str
-    execution_name: str = Field(alias="executionName")
-    agent_description: Optional[str] = Field(alias="agentDescription", default=None)
-    video_link: Optional[str] = Field(alias="videoLink", default=None)
-    industry: Optional[str] = None
-    sub_industries: List[str] = Field(alias="subIndustries", default_factory=list)
-    agent_details: Dict[str, Any] = Field(alias="agentDetails", default_factory=dict)
-    id: str
-    agent_icon: Optional[str] = Field(alias="agentIcon", default=None)
-    steps: List[Dict[str, Any]]
-
-
-class PromptMessage(BaseModel):
-    text: str
-    order: int
-
-
-class Prompt(BaseModel):
-    name: str
-    version_change_description: str = Field(alias="versionChangeDescription")
-    prompt_message_list: List[PromptMessage] = Field(alias="promptMessageList")
-    id: str
-
-
-class CredentialData(BaseModel):
-    key: str
-    value: str
-
-
-class CredentialsDefinition(BaseModel):
-    name: str
-    credential_type: str = Field(alias="credentialType")
-    source_type: str = Field(alias="sourceType")
-    credential_data_list: List[CredentialData] = Field(alias="credentialDataList")
-    id: str
-
-
-class HeaderDefinition(BaseModel):
-    key: str
-    value: str
-
-
-class ParameterDefinition(BaseModel):
-    name: str
-    parameter_type: str = Field(alias="parameterType")
-    parameter_description: str = Field(alias="parameterDescription")
-    default: str
-    valid_options: List[str] = Field(alias="validOptions", default_factory=list)
-    id: str
-
-
-class Tool(BaseModel):
-    tool_type: str = Field(alias="toolType")
-    name: str
-    standardized_name: str = Field(alias="standardizedName")
-    tool_description: str = Field(alias="toolDescription")
-    purpose: str
-    api_endpoint: str = Field(alias="apiEndpoint")
-    credentials_definition: Optional[CredentialsDefinition] = Field(
-        alias="credentialsDefinition"
-    )
-    headers_definition: List[HeaderDefinition] = Field(alias="headersDefinition")
-    body: str
-    parameters_definition: List[ParameterDefinition] = Field(
-        alias="parametersDefinition"
-    )
-    method_type: str = Field(alias="methodType")
-    route_through_acc: bool = Field(alias="routeThroughACC")
-    use_user_credentials: bool = Field(alias="useUserCredentials")
-    use_user_credentials_type: str = Field(alias="useUserCredentialsType")
-    id: str
-
-
-class Model(BaseModel):
-    """
-    Language model configuration and deployment settings.
-    
-    Defines an AI model used in the pipeline, including its deployment details,
-    pricing configuration, authentication settings, and capabilities.
-    """
-    id: str
-    display_name: str = Field(alias="displayName")
-    model_name: str = Field(alias="modelName")
-    prompt_id: Optional[str] = Field(alias="promptId", default=None)
-    system_prompt_definition: Optional[Any] = Field(
-        alias="systemPromptDefinition", default=None
-    )
-    url: str
-    input_type: str = Field(alias="inputType")
-    provider: str
-    credentials_definition: Optional[CredentialsDefinition] = Field(
-        alias="credentialsDefinition"
-    )
-    deployment_type: str = Field(alias="deploymentType")
-    source_type: str = Field(alias="sourceType")
-    connection_string: Optional[str] = Field(alias="connectionString", default=None)
-    container_name: Optional[str] = Field(alias="containerName", default=None)
-    deployed_key: Optional[str] = Field(alias="deployedKey", default=None)
-    deployed_url: Optional[str] = Field(alias="deployedUrl", default=None)
-    state: Optional[str] = None
-    uploaded_container_id: Optional[str] = Field(
-        alias="uploadedContainerId", default=None
-    )
-    library_model_id: Optional[str] = Field(alias="libraryModelId")
-    input_token_price: str = Field(alias="inputTokenPrice")
-    output_token_price: str = Field(alias="outputTokenPrice")
-    token_units: int = Field(alias="tokenUnits")
-    has_tool_support: bool = Field(alias="hasToolSupport")
-    allow_airia_credentials: bool = Field(alias="allowAiriaCredentials")
-    allow_byok_credentials: bool = Field(alias="allowBYOKCredentials")
-    author: Optional[str]
-    price_type: str = Field(alias="priceType")
-
-
-class PythonCodeBlock(BaseModel):
-    id: str
-    code: str
-
-
-class Router(BaseModel):
-    id: str
-    model_id: str = Field(alias="modelId")
-    model: Optional[Any] = None
-    router_config: Dict[str, Dict[str, Any]] = Field(alias="routerConfig")
-
-
-class ChunkingConfig(BaseModel):
-    id: str
-    chunk_size: int = Field(alias="chunkSize")
-    chunk_overlap: int = Field(alias="chunkOverlap")
-    strategy_type: str = Field(alias="strategyType")
-
-
-class DataSourceFile(BaseModel):
-    data_source_id: str = Field(alias="dataSourceId")
-    file_path: Optional[str] = Field(None, alias="filePath")
-    input_token: Optional[str] = Field(None, alias="inputToken")
-    file_count: Optional[int] = Field(None, alias="fileCount")
-
-
-class DataSource(BaseModel):
-    id: str = Field(alias="id")
-    name: Optional[str] = None
-    execution_name: Optional[str] = Field(None, alias="executionName")
-    chunking_config: ChunkingConfig = Field(alias="chunkingConfig")
-    data_source_type: str = Field(alias="dataSourceType")
-    database_type: str = Field(alias="databaseType")
-    embedding_provider: str = Field(alias="embeddingProvider")
-    is_user_specific: bool = Field(alias="isUserSpecific")
-    files: Optional[List[DataSourceFile]] = None
-    configuration_json: Optional[str] = Field(None, alias="configurationJson")
-    credentials: Optional[CredentialsDefinition]
-    is_image_processing_enabled: bool = Field(alias="isImageProcessingEnabled")
-
-
-class GetPipelineConfigResponse(BaseModel):
-    """
-    Complete pipeline configuration export response.
-    
-    This is the root response model containing all components of a pipeline
-    configuration, including the agent definition, associated resources,
-    and deployment settings.
-    """
-    metadata: Metadata
-    agent: Agent
-    data_sources: Optional[List[DataSource]] = Field(
-        alias="dataSources", default_factory=list
-    )
-    prompts: Optional[List[Prompt]] = Field(default_factory=list)
-    tools: Optional[List[Tool]] = Field(default_factory=list)
-    models: Optional[List[Model]] = Field(default_factory=list)
-    memories: Optional[Any] = None
-    python_code_blocks: Optional[List[PythonCodeBlock]] = Field(
-        alias="pythonCodeBlocks", default_factory=list
-    )
-    routers: Optional[List[Router]] = Field(default_factory=list)
-    deployment: Optional[Any] = None

airia-0.1.13.dist-info/RECORD

@@ -1,24 +0,0 @@
-airia/__init__.py,sha256=T39gO8E5T5zxlw-JP78ruxOu7-LeKOJCJzz6t40kdQo,231
-airia/constants.py,sha256=QDCxP-j3QQp5JGftYIYoqMu1fwXd56KZYAPD6_5YmPU,646
-airia/exceptions.py,sha256=4Z55n-cRJrtTa5-pZBIK2oZD4-Z99aUtKx_kfTFYY5o,1146
-airia/logs.py,sha256=_JO55-TcCtS1aVmmWUWEFKd2bMW1oYsz1j_8ajgPIdA,4570
-airia/client/__init__.py,sha256=6gSQ9bl7j79q1HPE0o5py3IRdkwWWuU_7J4h05Dd2o8,127
-airia/client/async_client.py,sha256=smYbJj90gBer24eLi4vwl_5z5jgPU6KA0wAWSv8GDxw,33290
-airia/client/base_client.py,sha256=LdAcTU1URuq6b1XXStd-7Sa-LiYm8H2TgzDyJSYegls,17456
-airia/client/sync_client.py,sha256=gH0VISV1ZGUjP53ignIrn8yN_iUOPgFV0EFZ8TsafVk,32118
-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=EUYMWQzKlX8Isf_VYP1bcb3uau6fjzRm9guXSaXPiwE,1041
-airia/types/api/__init__.py,sha256=gP4vOYxOG8Gfq8E2EEe1K_KW73nbJ--HUt3n7BUzu4A,913
-airia/types/api/conversations.py,sha256=Kkb3ybdBCQBCBVHe-gRUjGV2wqGA4Y5d5EGTitMuJ1Q,2981
-airia/types/api/get_pipeline_config.py,sha256=GTmY2IoCmXNCCmwOI9gQ5pkEJTyhQdaJxzhWaQhEmsM,7781
-airia/types/api/get_projects.py,sha256=krGV2gAxH81E_5OtUzUNdWZN7S-2jkAV5TiqJmW8A48,3479
-airia/types/api/pipeline_execution.py,sha256=-lcae-UqNrXqqy8DuyXEbMFBm3dVXQj0cl2jpeFeb78,1831
-airia/types/sse/__init__.py,sha256=DOFbw7DASl96g_UQwSS8iHY1mX4T6CLX7It5QWxtwJ0,1862
-airia/types/sse/sse_messages.py,sha256=q4DOqcYbYnsS3mBYO_-V7rZScGIChrCuh15jrcO0aII,14320
-airia/utils/sse_parser.py,sha256=YCnlXHqkAFUeKoy60RH-IUgQ9ZkrzK_UQ4VsnQldCUU,4274
-airia-0.1.13.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
-airia-0.1.13.dist-info/METADATA,sha256=coFoSzERDhm84fg0oaXgYalPIBAWWWOiJ3uexckVLRI,4364
-airia-0.1.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-airia-0.1.13.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
-airia-0.1.13.dist-info/RECORD,,