airia 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl

airia/client/base_client.py

@@ -212,7 +212,7 @@ class AiriaBaseClient:
             raise ValueError(
                 f"Invalid API version: {api_version}. Valid versions are: {', '.join(ApiVersion.as_list())}"
             )
-        url = urljoin(self.base_url, f"{api_version}/PipelinesConfig/{pipeline_id}")
+        url = urljoin(self.base_url, f"{api_version}/PipelinesConfig/export/{pipeline_id}")
         request_data = self._prepare_request(url, correlation_id=correlation_id)
 
         return request_data

airia/types/api/get_pipeline_config.py

@@ -1,65 +1,150 @@
-from datetime import datetime
 from typing import Any, Dict, List, Optional
 
 from pydantic import BaseModel, Field
 
 
-class Version(BaseModel):
-    pipeline_id: str = Field(alias="pipelineId")
-    major_version: int = Field(alias="majorVersion")
-    minor_version: int = Field(alias="minorVersion")
-    version_number: str = Field(alias="versionNumber")
-    is_draft_version: bool = Field(alias="isDraftVersion")
-    is_latest: bool = Field(alias="isLatest")
-    steps: Optional[List[Dict[str, Any]]] = None
-    alignment: str
+class Metadata(BaseModel):
     id: str
-    tenant_id: str = Field(alias="tenantId")
-    project_id: str = Field(alias="projectId")
-    created_at: datetime = Field(alias="createdAt")
-    updated_at: datetime = Field(alias="updatedAt")
-    user_id: str = Field(alias="userId")
+    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 ExecutionStats(BaseModel):
-    success_count: int = Field(alias="successCount")
-    failure_count: int = Field(alias="failureCount")
+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 GetPipelineConfigResponse(BaseModel):
-    deployment_id: Optional[str] = Field(alias="deploymentId")
-    deployment_name: Optional[str] = Field(alias="deploymentName")
-    deployment_description: Optional[str] = Field(alias="deploymentDescription")
-    user_keys: Dict[str, Any] = Field(alias="userKeys")
-    group_keys: Dict[str, Any] = Field(alias="groupKeys")
-    agent_icon: Optional[str] = Field(alias="agentIcon")
-    external: bool
-    active_version_id: str = Field(alias="activeVersionId")
+class PromptMessage(BaseModel):
+    text: str
+    order: int
+
+
+class Prompt(BaseModel):
     name: str
-    execution_name: str = Field(alias="executionName")
-    description: str
-    video_link: Optional[str] = Field(alias="videoLink")
-    agent_icon_id: Optional[str] = Field(alias="agentIconId")
-    versions: List[Version]
-    execution_stats: ExecutionStats = Field(alias="executionStats")
-    industry: Optional[str]
-    sub_industries: List[str] = Field(alias="subIndustries")
-    agent_details: Dict[str, Any] = Field(alias="agentDetails")
-    agent_details_tags: List[str] = Field(alias="agentDetailsTags")
-    active_version: Version = Field(alias="activeVersion")
-    backup_pipeline_id: Optional[str] = Field(alias="backupPipelineId")
-    deployment: Optional[Any]
-    library_agent_id: Optional[str] = Field(alias="libraryAgentId")
-    library_imported_hash: Optional[str] = Field(alias="libraryImportedHash")
-    library_imported_version: Optional[str] = Field(alias="libraryImportedVersion")
-    is_deleted: Optional[bool] = Field(alias="isDeleted")
-    agent_trigger: Optional[Any] = Field(alias="agentTrigger")
-    api_key_id: Optional[str] = Field(alias="apiKeyId")
-    is_seeded: bool = Field(alias="isSeeded")
-    behaviours: List[Any]
+    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
-    tenant_id: str = Field(alias="tenantId")
-    project_id: str = Field(alias="projectId")
-    created_at: datetime = Field(alias="createdAt")
-    updated_at: datetime = Field(alias="updatedAt")
-    user_id: str = Field(alias="userId")
+
+
+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: 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 GetPipelineConfigResponse(BaseModel):
+    metadata: Metadata
+    agent: Agent
+    data_sources: Optional[List[Any]] = 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.6.dist-info → airia-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airia
-Version: 0.1.6
+Version: 0.1.8
 Summary: Python SDK for Airia API
 Author-email: Airia LLC <support@airia.com>
 License: MIT
@@ -348,11 +348,7 @@ client = AiriaClient(api_key="your_api_key")
 config = client.get_pipeline_config(pipeline_id="your_pipeline_id")
 
 # Access configuration details
-print(f"Pipeline Name: {config.deployment_name}")
-print(f"Description: {config.deployment_description}")
-print(f"Active Version: {config.active_version.version_number}")
-print(f"Success Count: {config.execution_stats.success_count}")
-print(f"Failure Count: {config.execution_stats.failure_count}")
+print(f"Pipeline Name: {config.agent.name}")
 ```
 
 ## Gateway Usage

{airia-0.1.6.dist-info → airia-0.1.8.dist-info}/RECORD

@@ -3,17 +3,17 @@ airia/exceptions.py,sha256=4Z55n-cRJrtTa5-pZBIK2oZD4-Z99aUtKx_kfTFYY5o,1146
 airia/logs.py,sha256=17YZ4IuzOF0m5bgofj9-QYlJ2BYR2kRZbBVQfFSLFEk,5441
 airia/client/__init__.py,sha256=6gSQ9bl7j79q1HPE0o5py3IRdkwWWuU_7J4h05Dd2o8,127
 airia/client/async_client.py,sha256=QtnH1MNfF_ONx2YHsTz7Y4pFBTrggHWz0LLJiITREwc,23694
-airia/client/base_client.py,sha256=IQYksAw6xCFmAarPUGxyp2bHm95iTzdZUv8FNUq4fHk,7962
+airia/client/base_client.py,sha256=YWhQle7cmtiuc5BwiavW-NYjTC6kXyclpse4irwkMuI,7969
 airia/client/sync_client.py,sha256=iUqRn7Bv5hi-Qcg3Vgfc-Z1gx5T42f9Vq4pUTPsyG5M,22750
 airia/types/__init__.py,sha256=s6_uMrrKzhk2wg1lpaRRIl4AQ8hUxKbCNLfz-VK7tAs,2166
 airia/types/api_version.py,sha256=Uzom6O2ZG92HN_Z2h-lTydmO2XYX9RVs4Yi4DJmXytE,255
 airia/types/request_data.py,sha256=m8lBwFliK2_kZU2TgeLdhiO_7v826lVRhA5_8NYB_NM,206
 airia/types/sse_messages.py,sha256=wSdowY07AjEO8R73SJrFPJtkfIBS4satUpNytjKQq2U,8305
-airia/types/api/get_pipeline_config.py,sha256=hHyIfzpH-miayAAUtgf5ywXfqOg53dUMiEfDHW1I-tg,2837
+airia/types/api/get_pipeline_config.py,sha256=HAKgcpiJKeG_TuWtxo28CeWiodhlsVBwgbpzDgd1Eoo,5401
 airia/types/api/pipeline_execution.py,sha256=4zGM5W4aXiAibAdgWQCIZBC_blW6QYAlAHoVUMbnCF0,752
 airia/utils/sse_parser.py,sha256=h3TcBvXqUIngTqgY6yYxZCGLnC1eI6meQzYr13aFAb8,2791
-airia-0.1.6.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
-airia-0.1.6.dist-info/METADATA,sha256=1gvnnQKQhkJR31GkL4RG9Vw91UTi9Tclcjv5d9nxpMk,13471
-airia-0.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-airia-0.1.6.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
-airia-0.1.6.dist-info/RECORD,,
+airia-0.1.8.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
+airia-0.1.8.dist-info/METADATA,sha256=Toz3tWc8psnI2QbUrhjfzomljqp5BSrlaJH11mohHos,13218
+airia-0.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+airia-0.1.8.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
+airia-0.1.8.dist-info/RECORD,,