airia 0.1.7__tar.gz → 0.1.9__tar.gz

{airia-0.1.7 → airia-0.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airia
-Version: 0.1.7
+Version: 0.1.9
 Summary: Python SDK for Airia API
 Author-email: Airia LLC <support@airia.com>
 License: MIT

{airia-0.1.7 → airia-0.1.9}/airia/client/async_client.py

@@ -14,6 +14,7 @@ from ..types import (
     PipelineExecutionResponse,
     PipelineExecutionV1StreamedResponse,
     PipelineExecutionV2AsyncStreamedResponse,
+    ProjectItem,
     RequestData,
 )
 from ..utils.sse_parser import async_parse_sse_stream_chunked
@@ -199,6 +200,7 @@ class AiriaAsyncClient(AiriaBaseClient):
                 method=method,
                 url=request_data.url,
                 json=request_data.payload,
+                params=request_data.params,
                 headers=request_data.headers,
                 timeout=self.timeout,
             ) as response:
@@ -250,6 +252,7 @@ class AiriaAsyncClient(AiriaBaseClient):
                 method=method,
                 url=request_data.url,
                 json=request_data.payload,
+                params=request_data.params,
                 headers=request_data.headers,
                 timeout=self.timeout,
                 chunked=True,
@@ -419,11 +422,11 @@ class AiriaAsyncClient(AiriaBaseClient):
             aiohttp.ClientError: For other request-related errors.
 
         Example:
-            >>> async with AiriaAsyncClient(api_key="your_api_key") as client:
-            ...     response = await client.execute_pipeline(
-            ...         pipeline_id="pipeline_123",
-            ...         user_input="Tell me about quantum computing"
-            ...     )
+            >>> client = AiriaAsyncClient(api_key="your_api_key")
+            ... response = await client.execute_pipeline(
+            ...     pipeline_id="pipeline_123",
+            ...     user_input="Tell me about quantum computing"
+            ... )
             >>> print(response.result)
         """
         request_data = self._pre_execute_pipeline(
@@ -462,9 +465,9 @@ class AiriaAsyncClient(AiriaBaseClient):
                 self.base_url, f"{api_version}/StreamSocketConfig/GenerateUrl"
             )
             request_data = self._prepare_request(
-                url,
-                {"socketIdentifier": resp},
-                request_data.headers["X-Correlation-ID"],
+                url=url,
+                payload={"socketIdentifier": resp},
+                correlation_id=request_data.headers["X-Correlation-ID"],
             )
             resp = await self._make_request("POST", request_data)
 
@@ -472,43 +475,126 @@ class AiriaAsyncClient(AiriaBaseClient):
 
         return PipelineExecutionV2AsyncStreamedResponse(stream=resp)
 
+    async def get_projects(
+        self,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V1.value,
+    ) -> List[ProjectItem]:
+        """
+        Retrieve a list of all projects accessible to the authenticated user.
+
+        This method fetches comprehensive information about all projects that the
+        current user has access to, including project metadata, creation details,
+        and status information.
+
+        Args:
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+            api_version (str, optional): The API version to use for the request.
+                Defaults to "v1". Valid versions are defined in ApiVersion enum.
+
+        Returns:
+            List[ProjectItem]: A list of ProjectItem objects containing project
+                information. Returns an empty list if no projects are accessible
+                or found.
+
+        Raises:
+            ValueError: If the provided api_version is not valid.
+            AiriaAPIError: If the API request fails, including cases where:
+                - Authentication fails (401)
+                - Access is forbidden (403)
+                - Server errors (5xx)
+
+        Example:
+            ```python
+            from airia import AiriaAsyncClient
+
+            client = AiriaAsyncClient(api_key="your_api_key")
+
+            # Get all accessible projects
+            projects = await client.get_projects()
+
+            for project in projects:
+                print(f"Project: {project.name}")
+                print(f"ID: {project.id}")
+                print(f"Description: {project.description}")
+                print(f"Created: {project.created_at}")
+                print("---")
+            ```
+
+        Note:
+            The returned projects are filtered based on the authenticated user's
+            permissions. Users will only see projects they have been granted
+            access to.
+        """
+        request_data = self._pre_get_projects(
+            correlation_id=correlation_id, api_version=api_version
+        )
+        resp = await self._make_request("GET", request_data)
+
+        if "items" not in resp or len(resp["items"]) == 0:
+            return []
+
+        return [ProjectItem(**item) for item in resp["items"]]
+
     async def get_active_pipelines_ids(
         self,
+        project_id: Optional[str] = None,
         correlation_id: Optional[str] = None,
         api_version: str = ApiVersion.V1.value,
     ) -> List[str]:
         """
         Retrieve a list of active pipeline IDs.
-        
-        This method fetches all currently active pipeline IDs from the Airia API.
-        These IDs can be used with other methods like execute_pipeline() or 
-        get_pipeline_config().
-        
+
+        This method fetches the IDs of all active pipelines, optionally filtered by project.
+        Active pipelines are those that are currently deployed and available for execution.
+
         Args:
-            api_version (str, optional): API version to use for the request. 
-                Must be one of the supported versions. Defaults to "v1".
-            correlation_id (str, optional): Unique identifier for request tracing
-                and logging. If not provided, a new UUID will be automatically 
-                generated.
-        
+            project_id (str, optional): The unique identifier of the project to filter
+                pipelines by. If not provided, returns active pipelines from all projects
+                accessible to the authenticated user.
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+            api_version (str, optional): The API version to use for the request.
+                Defaults to "v1". Valid versions are defined in ApiVersion enum.
+
         Returns:
-            List[str]: A list of active pipeline ID strings. Returns an empty list
-                if no active pipelines are found.
-        
+            List[str]: A list of pipeline IDs that are currently active. Returns an
+                empty list if no active pipelines are found.
+
         Raises:
-            ValueError: If the provided API version is not supported.
-            AiriaAPIError: If the API request fails, including network errors,
-                authentication failures, or server errors.
-        
+            ValueError: If the provided api_version is not valid.
+            AiriaAPIError: If the API request fails, including cases where:
+                - The project_id doesn't exist (404)
+                - Authentication fails (401)
+                - Access is forbidden (403)
+                - Server errors (5xx)
+
         Example:
-            >>> client = AiriaClient(api_key="your_api_key")
-            >>> pipeline_ids = client.get_active_pipelines_ids()
-            >>> print(f"Found {len(pipeline_ids)} active pipelines")
-            >>> for pipeline_id in pipeline_ids:
-            ...     print(f"Pipeline ID: {pipeline_id}")
+            ```python
+            from airia import AiriaAsyncClient
+
+            client = AiriaAsyncClient(api_key="your_api_key")
+
+            # Get all active pipeline IDs
+            pipeline_ids = await client.get_active_pipelines_ids()
+            print(f"Found {len(pipeline_ids)} active pipelines")
+
+            # Get active pipeline IDs for a specific project
+            project_pipelines = await client.get_active_pipelines_ids(
+                project_id="your_project_id"
+            )
+            print(f"Project has {len(project_pipelines)} active pipelines")
+            ```
+
+        Note:
+            Only pipelines with active versions are returned. Inactive or archived
+            pipelines are not included in the results.
         """
         request_data = self._pre_get_active_pipelines_ids(
-            correlation_id=correlation_id, api_version=api_version
+            project_id=project_id,
+            correlation_id=correlation_id,
+            api_version=api_version,
         )
         resp = await self._make_request("GET", request_data)
 
@@ -553,12 +639,12 @@ class AiriaAsyncClient(AiriaBaseClient):
 
         Example:
             ```python
-            from airia import AiriaClient
+            from airia import AiriaAsyncClient
 
-            client = AiriaClient(api_key="your_api_key")
+            client = AiriaAsyncClient(api_key="your_api_key")
 
             # Get pipeline configuration
-            config = client.get_pipeline_config(
+            config = await client.get_pipeline_config(
                 pipeline_id="your_pipeline_id"
             )
 

{airia-0.1.7 → airia-0.1.9}/airia/client/base_client.py

@@ -70,29 +70,9 @@ class AiriaBaseClient:
         self,
         url: str,
         payload: Optional[Dict[str, Any]] = None,
+        params: Optional[Dict[str, Any]] = None,
         correlation_id: Optional[str] = None,
     ):
-        """
-        Prepare the request parameters for an API call.
-
-        Args:
-            url (str): The endpoint URL for the API request.
-            payload (Optional[Dict[str, Any]]): The request payload/body to be sent.
-            correlation_id (Optional[str]): A unique identifier for tracing the request. If None, one will be generated.
-
-        Returns:
-            dict: A dictionary containing the prepared request parameters with the following keys:
-                - url: The request URL
-                - payload: The request payload
-                - headers: Request headers including API key and correlation ID
-                - correlation_id: The correlation ID used for the request
-
-        Note:
-            This method handles:
-            - Setting/generating correlation IDs
-            - Adding authentication headers
-            - Logging requests (if enabled) with sensitive information redacted
-        """
         # Set correlation ID if provided or generate a new one
         correlation_id = set_correlation_id(correlation_id)
 
@@ -105,29 +85,27 @@ class AiriaBaseClient:
 
         # Log the request if enabled
         if self.log_requests:
-            # Create a sanitized copy of headers for logging
+            # Create a sanitized copy of headers and params for logging
             log_headers = headers.copy()
+            log_params = params.copy() if params is not None else {}
 
             # Filter out sensitive headers
             if "X-API-KEY" in log_headers:
                 log_headers["X-API-KEY"] = "[REDACTED]"
 
             # Process payload for logging
-            log_payload = None
-            if payload is not None:
-                log_payload = payload.copy()
-
-                if "images" in log_payload and log_payload["images"] is not None:
-                    log_payload["images"] = f"{len(log_payload['images'])} images"
-                if "files" in log_payload and log_payload["files"] is not None:
-                    log_payload["files"] = f"{len(log_payload['files'])} files"
-
-                log_payload = json.dumps(log_payload)
+            log_payload = payload.copy() if payload is not None else {}
+            if "images" in log_payload and log_payload["images"] is not None:
+                log_payload["images"] = f"{len(log_payload['images'])} images"
+            if "files" in log_payload and log_payload["files"] is not None:
+                log_payload["files"] = f"{len(log_payload['files'])} files"
+            log_payload = json.dumps(log_payload)
 
             self.logger.info(
                 f"API Request: POST {url}\n"
                 f"Headers: {json.dumps(log_headers)}\n"
                 f"Payload: {log_payload}"
+                f"Params: {json.dumps(log_params)}\n"
             )
 
         return RequestData(
@@ -135,6 +113,7 @@ class AiriaBaseClient:
                 "url": url,
                 "payload": payload,
                 "headers": headers,
+                "params": params,
                 "correlation_id": correlation_id,
             }
         )
@@ -184,12 +163,29 @@ class AiriaBaseClient:
             "promptVariables": prompt_variables,
         }
 
-        request_data = self._prepare_request(url, payload, correlation_id)
+        request_data = self._prepare_request(
+            url=url, payload=payload, correlation_id=correlation_id
+        )
+
+        return request_data
+
+    def _pre_get_projects(
+        self,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V1.value,
+    ):
+        if api_version not in ApiVersion.as_list():
+            raise ValueError(
+                f"Invalid API version: {api_version}. Valid versions are: {', '.join(ApiVersion.as_list())}"
+            )
+        url = urljoin(self.base_url, f"{api_version}/Project/paginated")
+        request_data = self._prepare_request(url, correlation_id=correlation_id)
 
         return request_data
 
     def _pre_get_active_pipelines_ids(
         self,
+        project_id: Optional[str] = None,
         correlation_id: Optional[str] = None,
         api_version: str = ApiVersion.V1.value,
     ):
@@ -198,7 +194,10 @@ class AiriaBaseClient:
                 f"Invalid API version: {api_version}. Valid versions are: {', '.join(ApiVersion.as_list())}"
             )
         url = urljoin(self.base_url, f"{api_version}/PipelinesConfig")
-        request_data = self._prepare_request(url, correlation_id=correlation_id)
+        params = {"projectId": project_id} if project_id is not None else None
+        request_data = self._prepare_request(
+            url, params=params, correlation_id=correlation_id
+        )
 
         return request_data
 
@@ -212,7 +211,23 @@ 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/export/{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
+
+    def _pre_get_projects(
+        self,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V1.value,
+    ):
+        if api_version not in ApiVersion.as_list():
+            raise ValueError(
+                f"Invalid API version: {api_version}. Valid versions are: {', '.join(ApiVersion.as_list())}"
+            )
+        url = urljoin(self.base_url, f"{api_version}/Project/paginated")
         request_data = self._prepare_request(url, correlation_id=correlation_id)
 
         return request_data

{airia-0.1.7 → airia-0.1.9}/airia/client/sync_client.py

@@ -12,6 +12,7 @@ from ..types import (
     PipelineExecutionResponse,
     PipelineExecutionV1StreamedResponse,
     PipelineExecutionV2StreamedResponse,
+    ProjectItem,
     RequestData,
 )
 from ..utils.sse_parser import parse_sse_stream_chunked
@@ -181,6 +182,7 @@ class AiriaClient(AiriaBaseClient):
                 method=method,
                 url=request_data.url,
                 json=request_data.payload,
+                params=request_data.params,
                 headers=request_data.headers,
                 timeout=self.timeout,
             )
@@ -231,6 +233,7 @@ class AiriaClient(AiriaBaseClient):
             response = self.session.request(
                 method=method,
                 url=request_data.url,
+                params=request_data.params,
                 json=request_data.payload,
                 headers=request_data.headers,
                 timeout=self.timeout,
@@ -445,8 +448,8 @@ class AiriaClient(AiriaBaseClient):
             )
             request_data = self._prepare_request(
                 url,
-                {"socketIdentifier": resp},
-                request_data.headers["X-Correlation-ID"],
+                payload={"socketIdentifier": resp},
+                correlation_id=request_data.headers["X-Correlation-ID"],
             )
             resp = self._make_request("POST", request_data)
 
@@ -454,43 +457,126 @@ class AiriaClient(AiriaBaseClient):
 
         return PipelineExecutionV2StreamedResponse(stream=resp)
 
+    def get_projects(
+        self,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V1.value,
+    ) -> List[ProjectItem]:
+        """
+        Retrieve a list of all projects accessible to the authenticated user.
+
+        This method fetches comprehensive information about all projects that the
+        current user has access to, including project metadata, creation details,
+        and status information.
+
+        Args:
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+            api_version (str, optional): The API version to use for the request.
+                Defaults to "v1". Valid versions are defined in ApiVersion enum.
+
+        Returns:
+            List[ProjectItem]: A list of ProjectItem objects containing project
+                information. Returns an empty list if no projects are accessible
+                or found.
+
+        Raises:
+            ValueError: If the provided api_version is not valid.
+            AiriaAPIError: If the API request fails, including cases where:
+                - Authentication fails (401)
+                - Access is forbidden (403)
+                - Server errors (5xx)
+
+        Example:
+            ```python
+            from airia import AiriaClient
+
+            client = AiriaClient(api_key="your_api_key")
+
+            # Get all accessible projects
+            projects = client.get_projects()
+
+            for project in projects:
+                print(f"Project: {project.name}")
+                print(f"ID: {project.id}")
+                print(f"Description: {project.description}")
+                print(f"Created: {project.created_at}")
+                print("---")
+            ```
+
+        Note:
+            The returned projects are filtered based on the authenticated user's
+            permissions. Users will only see projects they have been granted
+            access to.
+        """
+        request_data = self._pre_get_projects(
+            correlation_id=correlation_id, api_version=api_version
+        )
+        resp = self._make_request("GET", request_data)
+
+        if "items" not in resp or len(resp["items"]) == 0:
+            return []
+
+        return [ProjectItem(**item) for item in resp["items"]]
+
     def get_active_pipelines_ids(
         self,
+        project_id: Optional[str] = None,
         correlation_id: Optional[str] = None,
         api_version: str = ApiVersion.V1.value,
     ) -> List[str]:
         """
         Retrieve a list of active pipeline IDs.
 
-        This method fetches all currently active pipeline IDs from the Airia API.
-        These IDs can be used with other methods like execute_pipeline() or
-        get_pipeline_config().
+        This method fetches the IDs of all active pipelines, optionally filtered by project.
+        Active pipelines are those that are currently deployed and available for execution.
 
         Args:
-            api_version (str, optional): API version to use for the request.
-                Must be one of the supported versions. Defaults to "v1".
-            correlation_id (str, optional): Unique identifier for request tracing
-                and logging. If not provided, a new UUID will be automatically
-                generated.
+            project_id (str, optional): The unique identifier of the project to filter
+                pipelines by. If not provided, returns active pipelines from all projects
+                accessible to the authenticated user.
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+            api_version (str, optional): The API version to use for the request.
+                Defaults to "v1". Valid versions are defined in ApiVersion enum.
 
         Returns:
-            List[str]: A list of active pipeline ID strings. Returns an empty list
-                if no active pipelines are found.
+            List[str]: A list of pipeline IDs that are currently active. Returns an
+                empty list if no active pipelines are found.
 
         Raises:
-            ValueError: If the provided API version is not supported.
-            AiriaAPIError: If the API request fails, including network errors,
-                authentication failures, or server errors.
+            ValueError: If the provided api_version is not valid.
+            AiriaAPIError: If the API request fails, including cases where:
+                - The project_id doesn't exist (404)
+                - Authentication fails (401)
+                - Access is forbidden (403)
+                - Server errors (5xx)
 
         Example:
-            >>> client = AiriaClient(api_key="your_api_key")
-            >>> pipeline_ids = client.get_active_pipelines_ids()
-            >>> print(f"Found {len(pipeline_ids)} active pipelines")
-            >>> for pipeline_id in pipeline_ids:
-            ...     print(f"Pipeline ID: {pipeline_id}")
+            ```python
+            from airia import AiriaClient
+
+            client = AiriaClient(api_key="your_api_key")
+
+            # Get all active pipeline IDs
+            pipeline_ids = client.get_active_pipelines_ids()
+            print(f"Found {len(pipeline_ids)} active pipelines")
+
+            # Get active pipeline IDs for a specific project
+            project_pipelines = client.get_active_pipelines_ids(
+                project_id="your_project_id"
+            )
+            print(f"Project has {len(project_pipelines)} active pipelines")
+            ```
+
+        Note:
+            Only pipelines with active versions are returned. Inactive or archived
+            pipelines are not included in the results.
         """
         request_data = self._pre_get_active_pipelines_ids(
-            correlation_id=correlation_id, api_version=api_version
+            project_id=project_id,
+            correlation_id=correlation_id,
+            api_version=api_version,
         )
         resp = self._make_request("GET", request_data)
 

{airia-0.1.7 → airia-0.1.9}/airia/types/__init__.py

@@ -1,3 +1,4 @@
+from .api.get_projects import ProjectItem
 from .api.get_pipeline_config import GetPipelineConfigResponse
 from .api.pipeline_execution import (
     PipelineExecutionDebugResponse,
@@ -43,6 +44,7 @@ __all__ = [
     "PipelineExecutionV2AsyncStreamedResponse",
     "PipelineExecutionV2StreamedResponse",
     "GetPipelineConfigResponse",
+    "ProjectItem",
     "RequestData",
     "AgentPingMessage",
     "AgentStartMessage",

{airia-0.1.7 → airia-0.1.9}/airia/types/api/get_pipeline_config.py

@@ -75,7 +75,9 @@ class Tool(BaseModel):
     tool_description: str = Field(alias="toolDescription")
     purpose: str
     api_endpoint: str = Field(alias="apiEndpoint")
-    credentials_definition: CredentialsDefinition = Field(alias="credentialsDefinition")
+    credentials_definition: Optional[CredentialsDefinition] = Field(
+        alias="credentialsDefinition"
+    )
     headers_definition: List[HeaderDefinition] = Field(alias="headersDefinition")
     body: str
     parameters_definition: List[ParameterDefinition] = Field(
@@ -99,7 +101,9 @@ class Model(BaseModel):
     url: str
     input_type: str = Field(alias="inputType")
     provider: str
-    credentials_definition: CredentialsDefinition = Field(alias="credentialsDefinition")
+    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)
@@ -110,14 +114,14 @@ class Model(BaseModel):
     uploaded_container_id: Optional[str] = Field(
         alias="uploadedContainerId", default=None
     )
-    library_model_id: str = Field(alias="libraryModelId")
+    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: str
+    author: Optional[str]
     price_type: str = Field(alias="priceType")
 
 
@@ -133,10 +137,41 @@ class Router(BaseModel):
     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[Any]] = Field(alias="dataSources", default_factory=list)
+    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)

airia-0.1.9/airia/types/api/get_projects.py

@@ -0,0 +1,35 @@
+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-0.1.7 → airia-0.1.9}/airia/types/request_data.py

@@ -6,5 +6,6 @@ from pydantic import BaseModel
 class RequestData(BaseModel):
     url: str
     payload: Optional[Dict[str, Any]]
+    params: Optional[Dict[str, Any]]
     headers: Dict[str, Any]
     correlation_id: str

{airia-0.1.7 → airia-0.1.9}/airia.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airia
-Version: 0.1.7
+Version: 0.1.9
 Summary: Python SDK for Airia API
 Author-email: Airia LLC <support@airia.com>
 License: MIT

{airia-0.1.7 → airia-0.1.9}/airia.egg-info/SOURCES.txt

@@ -18,10 +18,12 @@ airia/types/api_version.py
 airia/types/request_data.py
 airia/types/sse_messages.py
 airia/types/api/get_pipeline_config.py
+airia/types/api/get_projects.py
 airia/types/api/pipeline_execution.py
 airia/utils/sse_parser.py
 tests/test_anthropic_gateway.py
 tests/test_execute_pipeline.py
 tests/test_get_active_pipelines_ids.py
 tests/test_get_pipeline_config.py
+tests/test_get_projects.py
 tests/test_openai_gateway.py

{airia-0.1.7 → airia-0.1.9}/pyproject.toml

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

{airia-0.1.7 → airia-0.1.9}/tests/test_anthropic_gateway.py

@@ -4,7 +4,7 @@ from dotenv import load_dotenv
 
 from airia import AiriaAsyncClient, AiriaClient
 
-load_dotenv()
+load_dotenv(override=True)
 
 
 # Fixtures for the sync client

{airia-0.1.7 → airia-0.1.9}/tests/test_execute_pipeline.py

@@ -6,7 +6,7 @@ from dotenv import load_dotenv
 
 from airia import AiriaAsyncClient, AiriaClient
 
-load_dotenv()
+load_dotenv(override=True)
 PYTHON_PIPELINE = "0134da17-c5a5-4730-a576-92f8eaf0926f"
 GPT_4O_MINI_PIPELINE = "367969f1-7a11-40f0-bab6-c8f901fdb537"
 USER_INPUT = "test"

{airia-0.1.7 → airia-0.1.9}/tests/test_get_active_pipelines_ids.py

@@ -56,6 +56,20 @@ class TestSyncGetActivePipelineIds:
             assert isinstance(result, list)
             assert len(result) == 0
 
+    def test_get_active_pipelines_ids_with_custom_project_id(
+        self, sync_client: AiriaClient
+    ):
+        """Test using custom Project ID."""
+        with patch.object(sync_client, "_make_request") as mock_request:
+            mock_request.return_value = SAMPLE_PIPELINE_IDS
+
+            result = sync_client.get_active_pipelines_ids(
+                project_id="custom-project-id"
+            )
+
+            assert result == ["pipeline-id-1", "pipeline-id-2", "pipeline-id-3"]
+            mock_request.assert_called_once()
+
     def test_get_active_pipelines_ids_with_custom_api_version(
         self, sync_client: AiriaClient
     ):
@@ -104,18 +118,6 @@ class TestSyncGetActivePipelineIds:
             assert "Pipelines not found" in str(exc_info.value)
             assert exc_info.value.status_code == 404
 
-    def test_get_active_pipelines_ids_network_error(self, sync_client: AiriaClient):
-        """Test error handling for network errors."""
-        with patch.object(sync_client, "_make_request") as mock_request:
-            mock_request.side_effect = AiriaAPIError(
-                status_code=500, message="Internal server error"
-            )
-
-            with pytest.raises(AiriaAPIError) as exc_info:
-                sync_client.get_active_pipelines_ids()
-
-            assert exc_info.value.status_code == 500
-
 
 class TestAsyncGetActivePipelineIds:
     """Test cases for asynchronous get_active_pipelines_ids method."""
@@ -166,6 +168,21 @@ class TestAsyncGetActivePipelineIds:
             assert result == ["pipeline-id-1", "pipeline-id-2", "pipeline-id-3"]
             mock_request.assert_called_once()
 
+    @pytest.mark.asyncio
+    async def test_get_active_pipelines_ids_with_custom_project_id(
+        self, async_client: AiriaClient
+    ):
+        """Test using custom Project ID."""
+        with patch.object(async_client, "_make_request") as mock_request:
+            mock_request.return_value = SAMPLE_PIPELINE_IDS
+
+            result = await async_client.get_active_pipelines_ids(
+                project_id="custom-project-id"
+            )
+
+            assert result == ["pipeline-id-1", "pipeline-id-2", "pipeline-id-3"]
+            mock_request.assert_called_once()
+
     @pytest.mark.asyncio
     async def test_get_active_pipelines_ids_invalid_api_version(
         self, async_client: AiriaAsyncClient

{airia-0.1.7 → airia-0.1.9}/tests/test_get_pipeline_config.py

@@ -8,7 +8,7 @@ from airia import AiriaAPIError, AiriaAsyncClient, AiriaClient
 from airia.types import GetPipelineConfigResponse
 
 # Load environment variables for testing
-load_dotenv()
+load_dotenv(override=True)
 PYTHON_PIPELINE = "0134da17-c5a5-4730-a576-92f8eaf0926f"
 
 

airia-0.1.9/tests/test_get_projects.py

@@ -0,0 +1,203 @@
+from unittest.mock import patch
+
+import pytest
+import pytest_asyncio
+from dotenv import load_dotenv
+
+from airia import AiriaAsyncClient, AiriaClient
+from airia.exceptions import AiriaAPIError
+
+# Load environment variables from .env file
+load_dotenv(override=True)
+
+# Test constants
+SAMPLE_PROJECTS = {
+    "items": [
+        {
+            "tenantId": "2ce49ae0-c3ff-421a-91b7-830d0c73b348",
+            "createdAt": "2025-01-22T20:00:19.8504590Z",
+            "requireClassification": False,
+            "budgetAmount": None,
+            "budgetPeriod": None,
+            "budgetAlert": None,
+            "budgetStop": False,
+            "usedBudgetAmount": None,
+            "resumeEndsAt": None,
+            "updatedAt": "2025-01-22T20:00:19.8509410Z",
+            "pipelines": [
+                {
+                    "id": "940275e2-7eae-4bc9-9830-4ef595727b87",
+                    "name": "Cloudkit Records",
+                },
+                {"id": "b8c24cc9-39de-4e76-b7dd-debac381d1a8", "name": "RAG_test"},
+                {
+                    "id": "17554b31-2370-4d2d-91de-4fe4805b9ef9",
+                    "name": "Omni Agent - Test Agent to be Described",
+                },
+            ],
+            "models": None,
+            "dataSources": [],
+            "prompts": None,
+            "apiKeys": None,
+            "memories": None,
+            "projectIcon": "https://airiaimagesprod.blob.core.windows.net/airia-default/8a8f75f5-bd6c-470b-bf23-e82c58d81912?sv=2025-05-05&spr=https&st=2025-06-17T20%3A23%3A39Z&se=2025-06-18T20%3A23%3A39Z&sr=b&sp=r&sig=TE7B5sgtCp%2BMpeQizbuhZShOXRcGPol%2BGitQdKGDBKg%3D",
+            "projectIconId": "8a8f75f5-bd6c-470b-bf23-e82c58d81912",
+            "description": "Agents Developed by AI Core Team",
+            "projectType": "Standard",
+            "classifications": None,
+            "id": "01948f99-f78a-7415-a187-b250c6e04458",
+            "name": "AI Core",
+        }
+    ],
+    "totalCount": 1,
+}
+
+
+@pytest.fixture
+def sync_client():
+    return AiriaClient(log_requests=True)
+
+
+@pytest_asyncio.fixture
+async def async_client():
+    return AiriaAsyncClient(log_requests=True)
+
+
+class TestSyncGetProjectsIds:
+    """Test cases for synchronous get_projects method."""
+
+    def test_get_projects_success(self, sync_client: AiriaClient):
+        """Test successful retrieval of projectss."""
+        with patch.object(sync_client, "_make_request") as mock_request:
+            mock_request.return_value = SAMPLE_PROJECTS
+
+            result = sync_client.get_projects()
+
+            assert isinstance(result, list)
+            assert len(result) == 1
+            mock_request.assert_called_once()
+
+    def test_get_projects_empty_dict(self, sync_client: AiriaClient):
+        """Test handling of empty project response."""
+        with patch.object(sync_client, "_make_request") as mock_request:
+            mock_request.return_value = {}
+
+            result = sync_client.get_projects()
+
+            assert result == []
+            assert isinstance(result, list)
+            assert len(result) == 0
+
+    def test_get_projects_with_custom_api_version(self, sync_client: AiriaClient):
+        """Test using custom API version."""
+        with patch.object(sync_client, "_make_request") as mock_request:
+            mock_request.return_value = SAMPLE_PROJECTS
+
+            result = sync_client.get_projects(api_version="v1")
+
+            assert isinstance(result, list)
+            assert len(result) == 1
+            mock_request.assert_called_once()
+
+    def test_get_projects_with_correlation_id(self, sync_client: AiriaClient):
+        """Test using custom correlation ID."""
+        custom_correlation_id = "test-correlation-123"
+
+        with patch.object(sync_client, "_make_request") as mock_request:
+            mock_request.return_value = SAMPLE_PROJECTS
+
+            result = sync_client.get_projects(correlation_id=custom_correlation_id)
+
+            assert isinstance(result, list)
+            assert len(result) == 1
+            mock_request.assert_called_once()
+
+    def test_get_projects_invalid_api_version(
+        self, sync_client: AiriaClient
+    ):
+        """Test error handling for invalid API version."""
+        with pytest.raises(ValueError, match="Invalid API version"):
+            sync_client.get_projects(api_version="invalid")
+
+    def test_get_projects_api_error(self, sync_client: AiriaClient):
+        """Test error handling for API errors."""
+        with patch.object(sync_client, "_make_request") as mock_request:
+            mock_request.side_effect = AiriaAPIError(
+                status_code=404, message="Projects not found"
+            )
+
+            with pytest.raises(AiriaAPIError) as exc_info:
+                sync_client.get_projects()
+
+            assert "Projects not found" in str(exc_info.value)
+            assert exc_info.value.status_code == 404
+
+
+class TestAsyncGetActivePipelineIds:
+    """Test cases for asynchronous get_active_pipelines_ids method."""
+
+    @pytest.mark.asyncio
+    async def test_get_projects_success(self, async_client: AiriaAsyncClient):
+        """Test successful retrieval of projects."""
+        with patch.object(async_client, "_make_request") as mock_request:
+            mock_request.return_value = SAMPLE_PROJECTS
+
+            result = await async_client.get_projects()
+
+            assert isinstance(result, list)
+            assert len(result) == 1
+            mock_request.assert_called_once()
+
+    @pytest.mark.asyncio
+    async def test_get_projects_empty_list(self, async_client: AiriaAsyncClient):
+        """Test handling of empty project list response."""
+        with patch.object(async_client, "_make_request") as mock_request:
+            mock_request.return_value = []
+
+            result = await async_client.get_projects()
+
+            assert result == []
+            assert isinstance(result, list)
+            assert len(result) == 0
+
+    @pytest.mark.asyncio
+    async def test_get_projects_with_custom_params(
+        self, async_client: AiriaAsyncClient
+    ):
+        """Test using custom API version and correlation ID."""
+        custom_correlation_id = "async-test-correlation-456"
+
+        with patch.object(async_client, "_make_request") as mock_request:
+            mock_request.return_value = SAMPLE_PROJECTS
+
+            result = await async_client.get_projects(
+                api_version="v1", correlation_id=custom_correlation_id
+            )
+
+            assert isinstance(result, list)
+            assert len(result) == 1
+            mock_request.assert_called_once()
+
+    @pytest.mark.asyncio
+    async def test_get_projects_invalid_api_version(
+        self, async_client: AiriaAsyncClient
+    ):
+        """Test error handling for invalid API version."""
+        with pytest.raises(ValueError, match="Invalid API version"):
+            await async_client.get_projects(api_version="invalid")
+
+    @pytest.mark.asyncio
+    async def test_get_projects_api_error(
+        self, async_client: AiriaAsyncClient
+    ):
+        """Test error handling for API errors."""
+        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_projects()
+
+            assert "Access forbidden" in str(exc_info.value)
+            assert exc_info.value.status_code == 403

{airia-0.1.7 → airia-0.1.9}/tests/test_openai_gateway.py

@@ -4,7 +4,7 @@ from dotenv import load_dotenv
 
 from airia import AiriaAsyncClient, AiriaClient
 
-load_dotenv()
+load_dotenv(override=True)
 
 
 # Fixtures for the sync client