airia 0.1.36__tar.gz → 0.1.37__tar.gz

{airia-0.1.36 → airia-0.1.37}/PKG-INFO

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

{airia-0.1.36 → airia-0.1.37}/airia/client/models/async_models.py

@@ -95,6 +95,74 @@ class AsyncModels(BaseModels):
 
         return [ModelItem(**item) for item in resp["items"]]
 
+    async def get_model(
+        self,
+        model_id: str,
+        correlation_id: Optional[str] = None,
+    ) -> ModelItem:
+        """
+        Retrieve a model by its ID.
+
+        This method fetches detailed information about a specific model, including
+        its configuration, pricing, capabilities, and associated project details.
+
+        Args:
+            model_id (str): The unique identifier of the model to retrieve.
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+
+        Returns:
+            ModelItem: A ModelItem object containing comprehensive model information
+                including configuration, pricing, capabilities, and project associations.
+
+        Raises:
+            AiriaAPIError: If the API request fails, including cases where:
+                - Model not found (404)
+                - Authentication fails (401)
+                - Access is forbidden (403)
+                - Server errors (5xx)
+
+        Example:
+            ```python
+            from airia import AiriaAsyncClient
+
+            client = AiriaAsyncClient(api_key="your_api_key")
+
+            # Get a specific model
+            model = await client.models.get_model(
+                model_id="447589d8-f82f-4a0c-ad15-215a2eaee7b8"
+            )
+
+            print(f"Model: {model.display_name}")
+            print(f"Provider: {model.provider}")
+            print(f"Model Name: {model.model_name}")
+            print(f"Has tool support: {model.has_tool_support}")
+            print(f"Has stream support: {model.has_stream_support}")
+
+            # Access pricing information
+            if model.user_provided_details:
+                print(f"Input token price: {model.user_provided_details.input_token_price}")
+                print(f"Output token price: {model.user_provided_details.output_token_price}")
+
+            # Get model with correlation ID for tracking
+            model = await client.models.get_model(
+                model_id="447589d8-f82f-4a0c-ad15-215a2eaee7b8",
+                correlation_id="my-correlation-id"
+            )
+            ```
+
+        Note:
+            The model must be accessible to the authenticated user. Users will only
+            be able to retrieve models they have been granted access to.
+        """
+        request_data = self._pre_get_model(
+            model_id=model_id,
+            correlation_id=correlation_id,
+            api_version=ApiVersion.V1.value,
+        )
+        resp = await self._request_handler.make_request("GET", request_data)
+        return ModelItem(**resp)
+
     async def delete_model(
         self,
         model_id: str,

{airia-0.1.36 → airia-0.1.37}/airia/client/models/base_models.py

@@ -67,6 +67,40 @@ class BaseModels:
 
         return request_data
 
+    def _pre_get_model(
+        self,
+        model_id: str,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V1.value,
+    ):
+        """
+        Prepare request data for retrieving a single model.
+
+        Args:
+            model_id: The ID of the model to retrieve
+            correlation_id: Optional correlation ID for tracing
+            api_version: API version to use for the request
+
+        Returns:
+            RequestData: Prepared request data for the get endpoint
+
+        Raises:
+            ValueError: If an invalid API version is provided
+        """
+        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._request_handler.base_url,
+            f"{api_version}/Models/{model_id}",
+        )
+        request_data = self._request_handler.prepare_request(
+            url, correlation_id=correlation_id
+        )
+        return request_data
+
     def _pre_delete_model(
         self,
         model_id: str,

{airia-0.1.36 → airia-0.1.37}/airia/client/models/sync_models.py

@@ -95,6 +95,74 @@ class Models(BaseModels):
 
         return [ModelItem(**item) for item in resp["items"]]
 
+    def get_model(
+        self,
+        model_id: str,
+        correlation_id: Optional[str] = None,
+    ) -> ModelItem:
+        """
+        Retrieve a model by its ID.
+
+        This method fetches detailed information about a specific model, including
+        its configuration, pricing, capabilities, and associated project details.
+
+        Args:
+            model_id (str): The unique identifier of the model to retrieve.
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+
+        Returns:
+            ModelItem: A ModelItem object containing comprehensive model information
+                including configuration, pricing, capabilities, and project associations.
+
+        Raises:
+            AiriaAPIError: If the API request fails, including cases where:
+                - Model not found (404)
+                - Authentication fails (401)
+                - Access is forbidden (403)
+                - Server errors (5xx)
+
+        Example:
+            ```python
+            from airia import AiriaClient
+
+            client = AiriaClient(api_key="your_api_key")
+
+            # Get a specific model
+            model = client.models.get_model(
+                model_id="447589d8-f82f-4a0c-ad15-215a2eaee7b8"
+            )
+
+            print(f"Model: {model.display_name}")
+            print(f"Provider: {model.provider}")
+            print(f"Model Name: {model.model_name}")
+            print(f"Has tool support: {model.has_tool_support}")
+            print(f"Has stream support: {model.has_stream_support}")
+
+            # Access pricing information
+            if model.user_provided_details:
+                print(f"Input token price: {model.user_provided_details.input_token_price}")
+                print(f"Output token price: {model.user_provided_details.output_token_price}")
+
+            # Get model with correlation ID for tracking
+            model = client.models.get_model(
+                model_id="447589d8-f82f-4a0c-ad15-215a2eaee7b8",
+                correlation_id="my-correlation-id"
+            )
+            ```
+
+        Note:
+            The model must be accessible to the authenticated user. Users will only
+            be able to retrieve models they have been granted access to.
+        """
+        request_data = self._pre_get_model(
+            model_id=model_id,
+            correlation_id=correlation_id,
+            api_version=ApiVersion.V1.value,
+        )
+        resp = self._request_handler.make_request("GET", request_data)
+        return ModelItem(**resp)
+
     def delete_model(
         self,
         model_id: str,

{airia-0.1.36 → airia-0.1.37}/airia.egg-info/PKG-INFO

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

{airia-0.1.36 → airia-0.1.37}/pyproject.toml

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