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

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,,

airia/types/api/conversations.py

@@ -1,14 +0,0 @@
-from typing import Optional
-
-from pydantic import BaseModel, Field
-
-
-class CreateConversationResponse(BaseModel):
-    user_id: str = Field(alias="userId")
-    conversation_id: str = Field(alias="conversationId")
-    websocket_url: str = Field(alias="websocketUrl")
-    deployment_id: str = Field(alias="deploymentId")
-    icon_id: Optional[str] = Field(None, alias="iconId")
-    icon_url: Optional[str] = Field(None, alias="iconUrl")
-    description: Optional[str] = None
-    space_name: Optional[str] = Field(None, alias="spaceName")

airia/types/api/get_pipeline_config.py

@@ -1,183 +0,0 @@
-from typing import Any, Dict, List, Optional
-
-from pydantic import BaseModel, Field
-
-
-class Metadata(BaseModel):
-    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):
-    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):
-    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):
-    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/types/api/get_projects.py

@@ -1,35 +0,0 @@
-from datetime import datetime
-from typing import Any, List, Optional
-
-from pydantic import BaseModel, Field
-
-
-class Pipeline(BaseModel):
-    id: str
-    name: str
-
-
-class ProjectItem(BaseModel):
-    tenant_id: str = Field(alias="tenantId")
-    created_at: datetime = Field(alias="createdAt")
-    require_classification: bool = Field(alias="requireClassification")
-    budget_amount: Optional[Any] = Field(None, alias="budgetAmount")
-    budget_period: Optional[Any] = Field(None, alias="budgetPeriod")
-    budget_alert: Optional[Any] = Field(None, alias="budgetAlert")
-    budget_stop: bool = Field(alias="budgetStop")
-    used_budget_amount: Optional[Any] = Field(None, alias="usedBudgetAmount")
-    resume_ends_at: Optional[datetime] = Field(None, alias="resumeEndsAt")
-    updated_at: datetime = Field(alias="updatedAt")
-    pipelines: List[Pipeline]
-    models: Optional[Any] = None
-    data_sources: List[Any] = Field(alias="dataSources")
-    prompts: Optional[Any] = None
-    api_keys: Optional[Any] = Field(alias="apiKeys")
-    memories: Optional[Any] = None
-    project_icon: Optional[str] = Field(None, alias="projectIcon")
-    project_icon_id: Optional[str] = Field(None, alias="projectIconId")
-    description: Optional[str] = None
-    project_type: str = Field(alias="projectType")
-    classifications: Optional[Any] = None
-    id: str
-    name: str

airia/types/api/pipeline_execution.py

@@ -1,29 +0,0 @@
-from typing import Any, AsyncIterator, Dict, Iterator
-
-from pydantic import BaseModel, ConfigDict, Field
-
-from ..sse import SSEMessage
-
-
-class PipelineExecutionResponse(BaseModel):
-    result: str
-    report: None
-    is_backup_pipeline: bool = Field(alias="isBackupPipeline")
-
-
-class PipelineExecutionDebugResponse(BaseModel):
-    result: str
-    report: Dict[str, Any]
-    is_backup_pipeline: bool = Field(alias="isBackupPipeline")
-
-
-class PipelineExecutionStreamedResponse(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True)
-
-    stream: Iterator[SSEMessage]
-
-
-class PipelineExecutionAsyncStreamedResponse(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True)
-
-    stream: AsyncIterator[SSEMessage]