airia 0.1.36__py3-none-any.whl → 0.1.37__py3-none-any.whl

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/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/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.dist-info → airia-0.1.37.dist-info}/METADATA

@@ -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.dist-info → airia-0.1.37.dist-info}/RECORD

@@ -31,9 +31,9 @@ airia/client/library/async_library.py,sha256=fRG3eUxZkXDMnDTAt18KxD1LQMC5VB6cssW
 airia/client/library/base_library.py,sha256=VkcWr25fp2es2sRB97tORQmtAi59NZ6YRzwLymlQHIk,4664
 airia/client/library/sync_library.py,sha256=u_oBscUTYRNcsDbxCgpx_UvY5UAGgum2DW7eWt8y_i8,4191
 airia/client/models/__init__.py,sha256=YbxGLwMb7RQKfPo_2L_O_W9j8Szwnb0vtWiUNxkScfs,107
-airia/client/models/async_models.py,sha256=gwW3tWwzSZ9_3JmGlwryY48w9KrFXoLET6Jpsj0hX0I,5853
-airia/client/models/base_models.py,sha256=FchsA7P-Fc7fxzlN88jJ3BEVXPWDcIHSzci6wtSArqg,3439
-airia/client/models/sync_models.py,sha256=FQLM4xCoxcBk1NuLf6z7jVVejlDKx5KpMRBz2tQm71o,5748
+airia/client/models/async_models.py,sha256=GihX15MjqH-XUUl141aae0fbHG-FVH9SdNenIAY9mFU,8505
+airia/client/models/base_models.py,sha256=NNe24dcCirraVzpWJR5nNdPA3_nGwj98Ga1bk1f-TMI,4522
+airia/client/models/sync_models.py,sha256=36-ejKBLJ6FCB4RWXCSdBy84VpOmVYaK9PdoUJMVgCg,8366
 airia/client/pipeline_execution/__init__.py,sha256=7qEZsPRTLySC71zlwYioBuJs6B4BwRCgFL3TQyFWXmc,175
 airia/client/pipeline_execution/async_pipeline_execution.py,sha256=EH9EexSYjhSxWAB5kOLQSh5Sfl5Qn5lQGyJrU72wFLw,23824
 airia/client/pipeline_execution/base_pipeline_execution.py,sha256=8x99rm8cPq-i9-UA97-XLNo1qAwdEPkmZ815fRSceFE,11168
@@ -95,8 +95,8 @@ airia/types/api/tools/_tools.py,sha256=PSJYFok7yQdE4it55iQmbryFzKN54nT6N161X1Rkp
 airia/types/sse/__init__.py,sha256=KWnNTfsQnthfrU128pUX6ounvSS7DvjC-Y21FE-OdMk,1863
 airia/types/sse/sse_messages.py,sha256=asq9KG5plT2XSgQMz-Nqo0WcKlXvE8UT3E-WLhCegPk,30244
 airia/utils/sse_parser.py,sha256=XCTkuaroYWaVQOgBq8VpbseQYSAVruF69AvKUwZQKTA,4251
-airia-0.1.36.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
-airia-0.1.36.dist-info/METADATA,sha256=y-Qw5v6SJTdKg1-5E1hmSay-dhoIqE0Dpxctm48Ju7E,4949
-airia-0.1.36.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-airia-0.1.36.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
-airia-0.1.36.dist-info/RECORD,,
+airia-0.1.37.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
+airia-0.1.37.dist-info/METADATA,sha256=JIQioSlT7ZRhwTVDMaVEtdncWYlZhODaWScapoYJdfY,4949
+airia-0.1.37.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+airia-0.1.37.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
+airia-0.1.37.dist-info/RECORD,,