PyPI - airia - Versions diffs - 0.1.6__tar.gz → 0.1.8__tar.gz - Mend

airia 0.1.6tar.gz → 0.1.8tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (31) hide show

{airia-0.1.6 → airia-0.1.8}/PKG-INFO RENAMED Viewed

@@ -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 → airia-0.1.8}/README.md RENAMED Viewed

@@ -318,11 +318,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 → airia-0.1.8}/airia/client/base_client.py RENAMED Viewed

@@ -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-0.1.8/airia/types/api/get_pipeline_config.py ADDED Viewed

@@ -0,0 +1,150 @@
+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: 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 → airia-0.1.8}/airia.egg-info/PKG-INFO RENAMED Viewed

@@ -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 → airia-0.1.8}/pyproject.toml RENAMED Viewed

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "airia"
-version = "0.1.6"
+version = "0.1.8"
 description = "Python SDK for Airia API"
 license = { text = "MIT" }
 authors = [{ name = "Airia LLC", email = "support@airia.com" }]

airia-0.1.8/tests/test_get_pipeline_config.py ADDED Viewed

@@ -0,0 +1,449 @@
+from unittest.mock import patch
+import pytest
+import pytest_asyncio
+from dotenv import load_dotenv
+from airia import AiriaAPIError, AiriaAsyncClient, AiriaClient
+from airia.types import GetPipelineConfigResponse
+# Load environment variables for testing
+load_dotenv()
+PYTHON_PIPELINE = "0134da17-c5a5-4730-a576-92f8eaf0926f"
+@pytest.fixture
+def sync_client():
+    return AiriaClient(log_requests=True)
+@pytest_asyncio.fixture
+async def async_client():
+    return AiriaAsyncClient(log_requests=True)
+# Mock response data for testing
+MOCK_PIPELINE_CONFIG_RESPONSE = {
+    "metadata": {
+        "id": "17554b31-2370-4d2d-91de-4fe4805b9ef9",
+        "exportVersion": "20250612041452_AddAiInternalMcpServerActionsToUsage",
+        "tagline": "",
+        "agentDescription": "",
+        "industry": "",
+        "tasks": "",
+        "credentialExportOption": "Placeholder",
+        "dataSourceExportOption": "Placeholder",
+        "versionInformation": "7.00",
+        "state": "Preview",
+    },
+    "agent": {
+        "name": "Omni Agent - Test Agent to be Described",
+        "executionName": "untitled_agent_%281%29",
+        "agentDescription": "",
+        "videoLink": None,
+        "industry": None,
+        "subIndustries": [],
+        "agentDetails": {},
+        "id": "17554b31-2370-4d2d-91de-4fe4805b9ef9",
+        "agentIcon": None,
+        "steps": [
+            {
+                "id": "1d180b16-d097-45e8-bf3a-031a4c4b959d",
+                "stepType": "pythonStep",
+                "position": {"x": "12", "y": "378"},
+                "handles": [
+                    {
+                        "uuid": "c75a1538-8ff5-4abb-acfd-d0554738e34f",
+                        "type": "target",
+                        "label": "",
+                        "tooltip": "",
+                        "x": 87,
+                        "y": -3,
+                    },
+                    {
+                        "uuid": "9734a974-9d0c-486b-b00d-aa2135c62bbc",
+                        "type": "source",
+                        "label": "",
+                        "tooltip": "",
+                        "x": 87,
+                        "y": 53.299988,
+                    },
+                ],
+                "dependenciesObject": [
+                    {
+                        "parentId": "f896b741-87d5-49de-92f9-881677da1dca",
+                        "parentHandleId": "7099c54b-ebd1-46a4-b1b8-602ccbbab1d5",
+                        "handleId": "c75a1538-8ff5-4abb-acfd-d0554738e34f",
+                    }
+                ],
+                "pythonCodeBlockId": "1d180b16-d097-45e8-bf3a-031a4c4b959d",
+                "stepTitle": "Python Code",
+            },
+            {
+                "id": "5a03ca6c-8b1c-4cee-bad2-32507e767e41",
+                "stepType": "outputStep",
+                "position": {"x": "507", "y": "790"},
+                "handles": [
+                    {
+                        "uuid": "04c6e2b3-b430-45bb-9bff-bb9e5b0ecdbb",
+                        "type": "target",
+                        "label": "",
+                        "tooltip": "",
+                        "x": 57,
+                        "y": -3,
+                    }
+                ],
+                "dependenciesObject": [
+                    {
+                        "parentId": "ac6467bf-999d-4aab-a747-73fa0c96b01e",
+                        "parentHandleId": "36ddab85-8bdd-433b-9516-35391bc363da",
+                        "handleId": "04c6e2b3-b430-45bb-9bff-bb9e5b0ecdbb",
+                    }
+                ],
+                "stepTitle": "Output",
+            },
+            {
+                "id": "9f6f9f66-712f-497d-bffd-3fe2823bd7ab",
+                "stepType": "inputStep",
+                "position": {"x": "483", "y": "12"},
+                "handles": [
+                    {
+                        "uuid": "64b87a80-7e5c-47c5-8819-a2b0f780e600",
+                        "type": "source",
+                        "label": "",
+                        "tooltip": "",
+                        "x": 69,
+                        "y": 39.300003,
+                    }
+                ],
+                "dependenciesObject": [],
+                "stepTitle": "Input",
+            },
+            {
+                "id": "ac6467bf-999d-4aab-a747-73fa0c96b01e",
+                "stepType": "AIOperation",
+                "position": {"x": "447", "y": "584"},
+                "handles": [
+                    {
+                        "uuid": "36ddab85-8bdd-433b-9516-35391bc363da",
+                        "type": "source",
+                        "label": "",
+                        "tooltip": "",
+                        "x": 87,
+                        "y": 62.69998,
+                    },
+                    {
+                        "uuid": "9921db77-05b9-4129-8a18-0c55a2ac038d",
+                        "type": "target",
+                        "label": "",
+                        "tooltip": "",
+                        "x": 87,
+                        "y": -3,
+                    },
+                ],
+                "dependenciesObject": [
+                    {
+                        "parentId": "1d180b16-d097-45e8-bf3a-031a4c4b959d",
+                        "parentHandleId": "9734a974-9d0c-486b-b00d-aa2135c62bbc",
+                        "handleId": "9921db77-05b9-4129-8a18-0c55a2ac038d",
+                    },
+                    {
+                        "parentId": "ddd98da0-e1c7-42c1-abdf-8c4bf86d6fc2",
+                        "parentHandleId": "a8988cf5-03ba-427c-9793-60173b7ab23d",
+                        "handleId": "9921db77-05b9-4129-8a18-0c55a2ac038d",
+                    },
+                    {
+                        "parentId": "f896b741-87d5-49de-92f9-881677da1dca",
+                        "parentHandleId": "7099c54b-ebd1-46a4-b1b8-602ccbbab1d5",
+                        "handleId": "9921db77-05b9-4129-8a18-0c55a2ac038d",
+                    },
+                ],
+                "temperature": 0.7,
+                "includeDateTimeContext": False,
+                "promptId": "cfce7cf4-0ba3-47fd-9376-1138fd44ad45",
+                "modelId": "4c511bda-95f6-4717-b210-0b9b4e331720",
+                "toolIds": ["01976577-2452-7d96-815b-d590e7f16364"],
+                "toolParams": {},
+                "stepTitle": "AI Model 1",
+            },
+            {
+                "id": "ddd98da0-e1c7-42c1-abdf-8c4bf86d6fc2",
+                "stepType": "dataSearch",
+                "position": {"x": "447", "y": "360"},
+                "handles": [
+                    {
+                        "uuid": "a8988cf5-03ba-427c-9793-60173b7ab23d",
+                        "type": "source",
+                        "label": "",
+                        "tooltip": "",
+                        "x": 87,
+                        "y": 71.29999,
+                    },
+                    {
+                        "uuid": "427c9f8b-3b7d-4a52-a76f-9ce23548caac",
+                        "type": "target",
+                        "label": "",
+                        "tooltip": "",
+                        "x": 87,
+                        "y": -3,
+                    },
+                ],
+                "dependenciesObject": [
+                    {
+                        "parentId": "f896b741-87d5-49de-92f9-881677da1dca",
+                        "parentHandleId": "7099c54b-ebd1-46a4-b1b8-602ccbbab1d5",
+                        "handleId": "427c9f8b-3b7d-4a52-a76f-9ce23548caac",
+                    }
+                ],
+                "dataSourceId": "01976170-1aa6-7900-8b6d-238bb32caa35",
+                "topK": 5,
+                "relevanceThreshold": 20,
+                "neighboringChunksCount": 1,
+                "hybridSearchAlpha": 0.5,
+                "databaseType": "",
+                "stepTitle": "Data Source",
+            },
+            {
+                "id": "f896b741-87d5-49de-92f9-881677da1dca",
+                "stepType": "routerStep",
+                "position": {"x": "447", "y": "172"},
+                "handles": [
+                    {
+                        "uuid": "8f4953a3-fe39-4b4c-bfbd-31ea473862f5",
+                        "type": "target",
+                        "label": "",
+                        "tooltip": "",
+                        "x": 87,
+                        "y": -3,
+                    },
+                    {
+                        "uuid": "7099c54b-ebd1-46a4-b1b8-602ccbbab1d5",
+                        "type": "source",
+                        "label": "Route 1",
+                        "tooltip": "",
+                        "x": 87,
+                        "y": 53.299988,
+                    },
+                ],
+                "dependenciesObject": [
+                    {
+                        "parentId": "9f6f9f66-712f-497d-bffd-3fe2823bd7ab",
+                        "parentHandleId": "64b87a80-7e5c-47c5-8819-a2b0f780e600",
+                        "handleId": "8f4953a3-fe39-4b4c-bfbd-31ea473862f5",
+                    }
+                ],
+                "stepTitle": "Router",
+            },
+        ],
+    },
+    "dataSources": [],
+    "prompts": [
+        {
+            "name": "Helpful Agent",
+            "versionChangeDescription": "Initial version",
+            "promptMessageList": [{"text": "You are a helpful agent.", "order": 0}],
+            "id": "cfce7cf4-0ba3-47fd-9376-1138fd44ad45",
+        }
+    ],
+    "tools": [
+        {
+            "toolType": "YoutubeSearch",
+            "name": "Search Youtube",
+            "standardizedName": "search_youtube",
+            "toolDescription": "Search YouTube for videos on a specific subject",
+            "purpose": "To search YouTube for videos on a specific subject",
+            "apiEndpoint": "https://www.googleapis.com/youtube/v3/search?part=snippet&q=<Subject/>&maxResults=<maxResults/>",
+            "credentialsDefinition": {
+                "name": "YoutubeSearch",
+                "credentialType": "Youtube",
+                "sourceType": "library",
+                "credentialDataList": [
+                    {"key": "youtubeApiKey", "value": "**placeholder**"}
+                ],
+                "id": "4be7b206-c5be-405e-af59-ba292cd68bd9",
+            },
+            "headersDefinition": [{"key": "Accept", "value": "application/json"}],
+            "body": "",
+            "parametersDefinition": [
+                {
+                    "name": "Subject",
+                    "parameterType": "string",
+                    "parameterDescription": "The subject to search for",
+                    "default": "",
+                    "validOptions": [],
+                    "id": "00000000-0000-0000-0000-000000000000",
+                },
+                {
+                    "name": "maxResults",
+                    "parameterType": "integer",
+                    "parameterDescription": "The maximum number of results to return. If no values is provided from the user the default value is 5. If a value requested is less than 5 set it to 5.",
+                    "default": "",
+                    "validOptions": [],
+                    "id": "00000000-0000-0000-0000-000000000000",
+                },
+            ],
+            "methodType": "GET",
+            "routeThroughACC": False,
+            "useUserCredentials": False,
+            "useUserCredentialsType": "YoutubeKey",
+            "id": "01976577-2452-7d96-815b-d590e7f16364",
+        }
+    ],
+    "models": [
+        {
+            "id": "4c511bda-95f6-4717-b210-0b9b4e331720",
+            "displayName": "GPT 4.1",
+            "modelName": "gpt-4.1-2025-04-14",
+            "promptId": None,
+            "systemPromptDefinition": None,
+            "url": "https://api.openai.com/v1/chat/completions",
+            "inputType": "image",
+            "provider": "OpenAI",
+            "credentialsDefinition": {
+                "name": "OpenAI",
+                "credentialType": "OpenAI",
+                "sourceType": "library",
+                "credentialDataList": [
+                    {"key": "openaiApiKey", "value": "**placeholder**"}
+                ],
+                "id": "580e51d7-b4b0-410c-86ff-11e34cee1242",
+            },
+            "deploymentType": "marketplace",
+            "sourceType": "library",
+            "connectionString": None,
+            "containerName": None,
+            "deployedKey": None,
+            "deployedUrl": None,
+            "state": None,
+            "uploadedContainerId": None,
+            "libraryModelId": "7e9a1b22-3c5f-4a57-a9d4-8b6d1e5f8a27",
+            "inputTokenPrice": "0.00000206",
+            "outputTokenPrice": "0.00000824",
+            "tokenUnits": 1000,
+            "hasToolSupport": True,
+            "allowAiriaCredentials": True,
+            "allowBYOKCredentials": True,
+            "author": "OpenAI",
+            "priceType": "AITextOutputModelPrice",
+        }
+    ],
+    "memories": None,
+    "pythonCodeBlocks": [
+        {
+            "id": "1d180b16-d097-45e8-bf3a-031a4c4b959d",
+            "code": "output = \"Always start your response with 'Let's do this!'\"",
+        }
+    ],
+    "routers": [
+        {
+            "id": "f896b741-87d5-49de-92f9-881677da1dca",
+            "modelId": "2e005300-b320-40ff-af41-b641486b559c",
+            "model": None,
+            "routerConfig": {
+                "7099c54b-ebd1-46a4-b1b8-602ccbbab1d5": {
+                    "id": "00000000-0000-0000-0000-000000000000",
+                    "prompt": "",
+                    "isDefault": True,
+                }
+            },
+        }
+    ],
+    "deployment": None,
+}
+class TestSyncGetPipelineConfig:
+    """Test cases for synchronous get_pipeline_config method."""
+    def test_get_pipeline_config_success(self, sync_client: AiriaClient):
+        """Test successful pipeline configuration retrieval."""
+        response = sync_client.get_pipeline_config(pipeline_id=PYTHON_PIPELINE)
+        # Verify the response is properly typed
+        assert isinstance(response, GetPipelineConfigResponse)
+    def test_get_pipeline_config_with_custom_api_version(
+        self, sync_client: AiriaClient
+    ):
+        """Test pipeline configuration retrieval with custom API version."""
+        with patch.object(sync_client, "_make_request") as mock_request:
+            mock_request.return_value = MOCK_PIPELINE_CONFIG_RESPONSE
+            response = sync_client.get_pipeline_config(
+                pipeline_id="test-pipeline-123", api_version="v1"
+            )
+            assert isinstance(response, GetPipelineConfigResponse)
+    def test_get_pipeline_config_with_correlation_id(self, sync_client: AiriaClient):
+        """Test pipeline configuration retrieval with custom correlation ID."""
+        with patch.object(sync_client, "_make_request") as mock_request:
+            mock_request.return_value = MOCK_PIPELINE_CONFIG_RESPONSE
+            custom_correlation_id = "test-correlation-123"
+            response = sync_client.get_pipeline_config(
+                pipeline_id="test-pipeline-123", correlation_id=custom_correlation_id
+            )
+            assert isinstance(response, GetPipelineConfigResponse)
+    def test_get_pipeline_config_invalid_api_version(self, sync_client: AiriaClient):
+        """Test that invalid API version raises ValueError."""
+        with pytest.raises(ValueError, match="Invalid API version"):
+            sync_client.get_pipeline_config(
+                pipeline_id="test-pipeline-123", api_version="invalid_version"
+            )
+    def test_get_pipeline_config_api_error(self, sync_client: AiriaClient):
+        """Test handling of API errors."""
+        with patch.object(sync_client, "_make_request") as mock_request:
+            mock_request.side_effect = AiriaAPIError(
+                status_code=404, message="Pipeline not found"
+            )
+            with pytest.raises(AiriaAPIError) as exc_info:
+                sync_client.get_pipeline_config(pipeline_id="nonexistent-pipeline")
+            assert exc_info.value.status_code == 404
+            assert "Pipeline not found" in str(exc_info.value)
+class TestAsyncGetPipelineConfig:
+    """Test cases for asynchronous get_pipeline_config method."""
+    @pytest.mark.asyncio
+    async def test_get_pipeline_config_success(self, async_client: AiriaAsyncClient):
+        """Test successful asynchronous pipeline configuration retrieval."""
+        response = await async_client.get_pipeline_config(pipeline_id=PYTHON_PIPELINE)
+        # Verify the response is properly typed
+        assert isinstance(response, GetPipelineConfigResponse)
+    @pytest.mark.asyncio
+    async def test_get_pipeline_config_with_custom_params(
+        self, async_client: AiriaAsyncClient
+    ):
+        """Test asynchronous pipeline configuration retrieval with custom parameters."""
+        with patch.object(async_client, "_make_request") as mock_request:
+            mock_request.return_value = MOCK_PIPELINE_CONFIG_RESPONSE
+            response = await async_client.get_pipeline_config(
+                pipeline_id="test-pipeline-123",
+                api_version="v1",
+                correlation_id="async-test-correlation",
+            )
+            assert isinstance(response, GetPipelineConfigResponse)
+    @pytest.mark.asyncio
+    async def test_get_pipeline_config_api_error(self, async_client: AiriaAsyncClient):
+        """Test handling of API errors in async context."""
+        with patch.object(async_client, "_make_request") as mock_request:
+            mock_request.side_effect = AiriaAPIError(
+                status_code=403, message="Access forbidden"
+            )
+            with pytest.raises(AiriaAPIError) as exc_info:
+                await async_client.get_pipeline_config(pipeline_id="forbidden-pipeline")
+            assert exc_info.value.status_code == 403
+            assert "Access forbidden" in str(exc_info.value)

airia-0.1.6/airia/types/api/get_pipeline_config.py DELETED Viewed

@@ -1,65 +0,0 @@
-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
-    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 ExecutionStats(BaseModel):
-    success_count: int = Field(alias="successCount")
-    failure_count: int = Field(alias="failureCount")
-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")
-    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]
-    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")

airia-0.1.6/tests/test_get_pipeline_config.py DELETED Viewed

@@ -1,394 +0,0 @@
-from unittest.mock import patch
-import pytest
-import pytest_asyncio
-from dotenv import load_dotenv
-from airia import AiriaAPIError, AiriaAsyncClient, AiriaClient
-from airia.types import GetPipelineConfigResponse
-# Load environment variables for testing
-load_dotenv()
-PYTHON_PIPELINE = "0134da17-c5a5-4730-a576-92f8eaf0926f"
-@pytest.fixture
-def sync_client():
-    return AiriaClient(log_requests=True)
-@pytest_asyncio.fixture
-async def async_client():
-    return AiriaAsyncClient(log_requests=True)
-# Mock response data for testing
-MOCK_PIPELINE_CONFIG_RESPONSE = {
-    "deploymentId": None,
-    "deploymentName": None,
-    "deploymentDescription": None,
-    "userKeys": {},
-    "groupKeys": {},
-    "agentIcon": None,
-    "external": False,
-    "activeVersionId": "6e98f6ae-d666-4db0-9495-b5b40e45b482",
-    "name": "Python Block Test",
-    "executionName": "python_block_test",
-    "description": "",
-    "videoLink": None,
-    "agentIconId": None,
-    "versions": [
-        {
-            "pipelineId": "0134da17-c5a5-4730-a576-92f8eaf0926f",
-            "majorVersion": 1,
-            "minorVersion": 0,
-            "versionNumber": "1.00",
-            "isDraftVersion": False,
-            "isLatest": False,
-            "steps": None,
-            "alignment": "Horizontal",
-            "id": "1756d834-4c2a-4271-beff-b6a1b8b4737f",
-            "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-            "projectId": "01948f99-f78a-7415-a187-b250c6e04458",
-            "createdAt": "2025-03-26T20:33:31.9410340Z",
-            "updatedAt": "2025-03-26T22:37:25.8345390Z",
-            "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-        },
-        {
-            "pipelineId": "0134da17-c5a5-4730-a576-92f8eaf0926f",
-            "majorVersion": 2,
-            "minorVersion": 0,
-            "versionNumber": "2.00",
-            "isDraftVersion": False,
-            "isLatest": False,
-            "steps": None,
-            "alignment": "Horizontal",
-            "id": "0ac78be5-182c-4350-b5a9-128fb8904ded",
-            "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-            "projectId": "01948f99-f78a-7415-a187-b250c6e04458",
-            "createdAt": "2025-03-26T22:37:25.8318150Z",
-            "updatedAt": "2025-03-26T22:40:49.4444470Z",
-            "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-        },
-        {
-            "pipelineId": "0134da17-c5a5-4730-a576-92f8eaf0926f",
-            "majorVersion": 3,
-            "minorVersion": 0,
-            "versionNumber": "3.00",
-            "isDraftVersion": False,
-            "isLatest": False,
-            "steps": None,
-            "alignment": "Horizontal",
-            "id": "8b4cff91-22c9-4f49-8d56-c7679e97d896",
-            "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-            "projectId": "01948f99-f78a-7415-a187-b250c6e04458",
-            "createdAt": "2025-03-26T22:40:49.4406740Z",
-            "updatedAt": "2025-03-26T22:50:14.2888930Z",
-            "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-        },
-        {
-            "pipelineId": "0134da17-c5a5-4730-a576-92f8eaf0926f",
-            "majorVersion": 4,
-            "minorVersion": 0,
-            "versionNumber": "4.00",
-            "isDraftVersion": False,
-            "isLatest": False,
-            "steps": None,
-            "alignment": "Horizontal",
-            "id": "a51ab51f-f680-46fe-bb92-9ccdf35d05ab",
-            "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-            "projectId": "01948f99-f78a-7415-a187-b250c6e04458",
-            "createdAt": "2025-03-26T22:50:53.2168600Z",
-            "updatedAt": "2025-03-26T23:52:29.1746240Z",
-            "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-        },
-        {
-            "pipelineId": "0134da17-c5a5-4730-a576-92f8eaf0926f",
-            "majorVersion": 5,
-            "minorVersion": 0,
-            "versionNumber": "5.00",
-            "isDraftVersion": False,
-            "isLatest": True,
-            "steps": None,
-            "alignment": "Horizontal",
-            "id": "6e98f6ae-d666-4db0-9495-b5b40e45b482",
-            "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-            "projectId": "01948f99-f78a-7415-a187-b250c6e04458",
-            "createdAt": "2025-03-26T23:52:29.1722390Z",
-            "updatedAt": "2025-03-26T23:52:29.1746230Z",
-            "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-        },
-    ],
-    "executionStats": {"successCount": 1039, "failureCount": 1},
-    "industry": None,
-    "subIndustries": [],
-    "agentDetails": {},
-    "agentDetailsTags": [],
-    "activeVersion": {
-        "pipelineId": "0134da17-c5a5-4730-a576-92f8eaf0926f",
-        "majorVersion": 5,
-        "minorVersion": 0,
-        "versionNumber": "5.00",
-        "isDraftVersion": False,
-        "isLatest": True,
-        "steps": [
-            {
-                "code": '"""\nThis Block processes input data and client context to produce an output.\n\nAvailable Variables:\n- input (str): Input from previous steps.\n- client_data (dict): Contains user context with the following keys:\n    - user_id (str): The user\'s unique identifier.\n    - conversation_id (str): The conversation\'s unique identifier.\n    - user_input (str): The user\'s input text.\n    - user_roles (List[str]): The user\'s roles.\n    - user_groups (List[str]): The user\'s groups.\n    - images (List[str]): An array of image URLs.\n    - additional_info (List[Any]): an array of objects containing additional information passed to the agent.\n    - step_inputs (List[Any]): The inputs for the current step.\n    - files (List[str]): The files associated with the request.    \n- execution_parameters (dict): Contains the following keys:\n    - execution_id (str): The unique identifier for the execution.\n    - step_results_by_type (dict): A dictionary containing the results of the previous steps, ordered by completion, with the following keys:\n       - model (List[Any]): The results of the model steps.\n       - data_source (List[Any]): The results of the data search steps.\n       - memory_load (List[Any]): The results of the memory load steps.\n       - memory_store (List[Any]): The results of the memory store steps.    \n       - python (List[Any]): The results of the python steps.\n       - router (List[Any]): The results of the router steps.\n       - tool_action (List[Any]): The results of the tool action steps.\n       - agent (List[Any]): The results of the agent steps.\n                     \nReturns:\n- output (str): The processed output, initially set to the input value.\n"""\n\n# Your code here\noutput = input  # Assign your output to this variable\n\nif len(client_data["images"]) > 0:\n    output = f\'{len(client_data["images"])} images\'\n\nif len(client_data["files"]) > 0:\n    output = f\'{len(client_data["files"])} files\'\n\nif len(client_data["additional_info"]) > 0:\n    output = str(client_data["additional_info"])\n',
-                "stepType": "pythonStep",
-                "position": None,
-                "positionId": "a8c9cfa0-6313-4f1b-a28b-cdd2350b05d3",
-                "handles": [
-                    {
-                        "pipelineStepId": "48c1e273-e1ba-47df-a713-35eee1c2d7d7",
-                        "uuid": "198db5b2-75d6-4fe1-92d9-b347207449a4",
-                        "type": "source",
-                        "label": "",
-                        "tooltip": "",
-                        "x": 255.9375,
-                        "y": 47.738647,
-                        "id": "95f413c2-0e3f-47c5-af31-38549dd72bef",
-                        "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-                        "projectId": "00000000-0000-0000-0000-000000000000",
-                        "createdAt": "2025-03-26T23:52:29.1722350Z",
-                        "updatedAt": "2025-03-26T23:52:29.1746230Z",
-                        "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-                    },
-                    {
-                        "pipelineStepId": "48c1e273-e1ba-47df-a713-35eee1c2d7d7",
-                        "uuid": "ca16a5b9-f32e-454d-873d-1e9efa0ce167",
-                        "type": "target",
-                        "label": "",
-                        "tooltip": "",
-                        "x": -5.4454956,
-                        "y": 47.738647,
-                        "id": "fc0b1989-1e49-4e3c-8f22-094426513669",
-                        "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-                        "projectId": "00000000-0000-0000-0000-000000000000",
-                        "createdAt": "2025-03-26T23:52:29.1722360Z",
-                        "updatedAt": "2025-03-26T23:52:29.1746230Z",
-                        "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-                    },
-                ],
-                "dependenciesObject": [
-                    {
-                        "pipelineStepId": "48c1e273-e1ba-47df-a713-35eee1c2d7d7",
-                        "parentId": "ae8b9ff8-7f95-4b62-9fde-2c7127448a10",
-                        "parentHandleId": "b81a7e4e-3e55-4797-889d-a4f6015cab9b",
-                        "handleId": "ca16a5b9-f32e-454d-873d-1e9efa0ce167",
-                        "id": "e59f33f2-33c8-4b04-a2cd-3df416979771",
-                        "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-                        "projectId": "00000000-0000-0000-0000-000000000000",
-                        "createdAt": "2025-03-26T23:52:29.1722330Z",
-                        "updatedAt": "2025-03-26T23:52:29.1746230Z",
-                        "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-                    }
-                ],
-                "pipelineVersionId": "6e98f6ae-d666-4db0-9495-b5b40e45b482",
-                "stepTitle": "Python Code",
-                "id": "48c1e273-e1ba-47df-a713-35eee1c2d7d7",
-                "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-                "projectId": "01948f99-f78a-7415-a187-b250c6e04458",
-                "createdAt": "2025-03-26T20:33:31.9370040Z",
-                "updatedAt": "2025-03-26T23:52:29.1746230Z",
-                "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-            },
-            {
-                "stepType": "outputStep",
-                "position": None,
-                "positionId": "fef7c865-1b6e-4896-b1c4-7d89fbfad260",
-                "handles": [
-                    {
-                        "pipelineStepId": "a0f14314-602d-46d8-b6be-d350865231c9",
-                        "uuid": "b5de5347-6897-4414-ae3c-0d4f513f6461",
-                        "type": "target",
-                        "label": "",
-                        "tooltip": "",
-                        "x": -5.4453125,
-                        "y": 32.491364,
-                        "id": "95be20fa-a8d0-40a8-9047-20cb7546c8e7",
-                        "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-                        "projectId": "00000000-0000-0000-0000-000000000000",
-                        "createdAt": "2025-03-26T23:52:29.1722320Z",
-                        "updatedAt": "2025-03-26T23:52:29.1746230Z",
-                        "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-                    }
-                ],
-                "dependenciesObject": [
-                    {
-                        "pipelineStepId": "a0f14314-602d-46d8-b6be-d350865231c9",
-                        "parentId": "48c1e273-e1ba-47df-a713-35eee1c2d7d7",
-                        "parentHandleId": "198db5b2-75d6-4fe1-92d9-b347207449a4",
-                        "handleId": "b5de5347-6897-4414-ae3c-0d4f513f6461",
-                        "id": "c11e97cf-994d-479a-97da-195a60c3bef0",
-                        "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-                        "projectId": "00000000-0000-0000-0000-000000000000",
-                        "createdAt": "2025-03-26T23:52:29.1722290Z",
-                        "updatedAt": "2025-03-26T23:52:29.1746230Z",
-                        "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-                    }
-                ],
-                "pipelineVersionId": "6e98f6ae-d666-4db0-9495-b5b40e45b482",
-                "stepTitle": "Output",
-                "id": "a0f14314-602d-46d8-b6be-d350865231c9",
-                "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-                "projectId": "01948f99-f78a-7415-a187-b250c6e04458",
-                "createdAt": "2025-03-26T20:33:31.9369710Z",
-                "updatedAt": "2025-03-26T23:52:29.1746230Z",
-                "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-            },
-            {
-                "inputVariables": [],
-                "inputTemplate": None,
-                "stepType": "inputStep",
-                "position": None,
-                "positionId": "b7e17fb7-1ee4-42e3-803b-6d3053673f23",
-                "handles": [
-                    {
-                        "pipelineStepId": "ae8b9ff8-7f95-4b62-9fde-2c7127448a10",
-                        "uuid": "b81a7e4e-3e55-4797-889d-a4f6015cab9b",
-                        "type": "source",
-                        "label": "",
-                        "tooltip": "",
-                        "x": 255.93744,
-                        "y": 32.491364,
-                        "id": "b6be5af9-7534-450f-850b-14cb7ad654bf",
-                        "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-                        "projectId": "00000000-0000-0000-0000-000000000000",
-                        "createdAt": "2025-03-26T23:52:29.1722380Z",
-                        "updatedAt": "2025-03-26T23:52:29.1746240Z",
-                        "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-                    }
-                ],
-                "dependenciesObject": [],
-                "pipelineVersionId": "6e98f6ae-d666-4db0-9495-b5b40e45b482",
-                "stepTitle": "Input",
-                "id": "ae8b9ff8-7f95-4b62-9fde-2c7127448a10",
-                "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-                "projectId": "01948f99-f78a-7415-a187-b250c6e04458",
-                "createdAt": "2025-03-26T20:33:31.9369340Z",
-                "updatedAt": "2025-03-26T23:52:29.1746230Z",
-                "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-            },
-        ],
-        "alignment": "Horizontal",
-        "id": "6e98f6ae-d666-4db0-9495-b5b40e45b482",
-        "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-        "projectId": "01948f99-f78a-7415-a187-b250c6e04458",
-        "createdAt": "2025-03-26T23:52:29.1722390Z",
-        "updatedAt": "2025-03-26T23:52:29.1746230Z",
-        "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-    },
-    "backupPipelineId": None,
-    "deployment": None,
-    "libraryAgentId": None,
-    "libraryImportedHash": None,
-    "libraryImportedVersion": None,
-    "isDeleted": None,
-    "agentTrigger": None,
-    "apiKeyId": None,
-    "isSeeded": False,
-    "behaviours": [],
-    "id": "0134da17-c5a5-4730-a576-92f8eaf0926f",
-    "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
-    "projectId": "01948f99-f78a-7415-a187-b250c6e04458",
-    "createdAt": "2025-03-26T20:33:31.9410260Z",
-    "updatedAt": "2025-03-26T23:52:31.1933780Z",
-    "userId": "9f7e215f-5563-414e-9635-570efa078c35",
-}
-class TestSyncGetPipelineConfig:
-    """Test cases for synchronous get_pipeline_config method."""
-    def test_get_pipeline_config_success(self, sync_client: AiriaClient):
-        """Test successful pipeline configuration retrieval."""
-        response = sync_client.get_pipeline_config(pipeline_id=PYTHON_PIPELINE)
-        # Verify the response is properly typed
-        assert isinstance(response, GetPipelineConfigResponse)
-    def test_get_pipeline_config_with_custom_api_version(
-        self, sync_client: AiriaClient
-    ):
-        """Test pipeline configuration retrieval with custom API version."""
-        with patch.object(sync_client, "_make_request") as mock_request:
-            mock_request.return_value = MOCK_PIPELINE_CONFIG_RESPONSE
-            response = sync_client.get_pipeline_config(
-                pipeline_id="test-pipeline-123", api_version="v1"
-            )
-            assert isinstance(response, GetPipelineConfigResponse)
-    def test_get_pipeline_config_with_correlation_id(self, sync_client: AiriaClient):
-        """Test pipeline configuration retrieval with custom correlation ID."""
-        with patch.object(sync_client, "_make_request") as mock_request:
-            mock_request.return_value = MOCK_PIPELINE_CONFIG_RESPONSE
-            custom_correlation_id = "test-correlation-123"
-            response = sync_client.get_pipeline_config(
-                pipeline_id="test-pipeline-123", correlation_id=custom_correlation_id
-            )
-            assert isinstance(response, GetPipelineConfigResponse)
-    def test_get_pipeline_config_invalid_api_version(self, sync_client: AiriaClient):
-        """Test that invalid API version raises ValueError."""
-        with pytest.raises(ValueError, match="Invalid API version"):
-            sync_client.get_pipeline_config(
-                pipeline_id="test-pipeline-123", api_version="invalid_version"
-            )
-    def test_get_pipeline_config_api_error(self, sync_client: AiriaClient):
-        """Test handling of API errors."""
-        with patch.object(sync_client, "_make_request") as mock_request:
-            mock_request.side_effect = AiriaAPIError(
-                status_code=404, message="Pipeline not found"
-            )
-            with pytest.raises(AiriaAPIError) as exc_info:
-                sync_client.get_pipeline_config(pipeline_id="nonexistent-pipeline")
-            assert exc_info.value.status_code == 404
-            assert "Pipeline not found" in str(exc_info.value)
-class TestAsyncGetPipelineConfig:
-    """Test cases for asynchronous get_pipeline_config method."""
-    @pytest.mark.asyncio
-    async def test_get_pipeline_config_success(self, async_client: AiriaAsyncClient):
-        """Test successful asynchronous pipeline configuration retrieval."""
-        response = await async_client.get_pipeline_config(pipeline_id=PYTHON_PIPELINE)
-        # Verify the response is properly typed
-        assert isinstance(response, GetPipelineConfigResponse)
-    @pytest.mark.asyncio
-    async def test_get_pipeline_config_with_custom_params(
-        self, async_client: AiriaAsyncClient
-    ):
-        """Test asynchronous pipeline configuration retrieval with custom parameters."""
-        with patch.object(async_client, "_make_request") as mock_request:
-            mock_request.return_value = MOCK_PIPELINE_CONFIG_RESPONSE
-            response = await async_client.get_pipeline_config(
-                pipeline_id="test-pipeline-123",
-                api_version="v1",
-                correlation_id="async-test-correlation",
-            )
-            assert isinstance(response, GetPipelineConfigResponse)
-    @pytest.mark.asyncio
-    async def test_get_pipeline_config_api_error(self, async_client: AiriaAsyncClient):
-        """Test handling of API errors in async context."""
-        with patch.object(async_client, "_make_request") as mock_request:
-            mock_request.side_effect = AiriaAPIError(
-                status_code=403, message="Access forbidden"
-            )
-            with pytest.raises(AiriaAPIError) as exc_info:
-                await async_client.get_pipeline_config(pipeline_id="forbidden-pipeline")
-            assert exc_info.value.status_code == 403
-            assert "Access forbidden" in str(exc_info.value)