ogx_open_client 1.0.3.dev0__py3-none-any.whl

ogx_client/api/models_api.py

@@ -0,0 +1,451 @@
+# coding: utf-8
+
+# Copyright (c) The OGX Contributors.
+# All rights reserved.
+#
+# This source code is licensed under the terms described in the LICENSE file in
+# the root directory of this source tree.
+
+"""
+    OGX Specification - Stable & Experimental APIs
+
+    This is the specification of the OGX that provides                     a set of endpoints and their corresponding interfaces that are     tailored to                     best leverage Llama Models.      **🔗 COMBINED**: This specification includes both stable production-ready APIs     and experimental pre-release APIs. Use stable APIs for production deployments     and experimental APIs for testing new features.
+
+    The version of the OpenAPI document: v1
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+import importlib
+import json
+import logging
+import re
+import sys
+import warnings
+
+from pydantic import validate_call, Field, SkipValidation, StrictFloat, StrictStr, StrictInt
+from typing import Any
+from typing_extensions import Annotated
+
+from pydantic import Field, StrictStr
+from typing import Optional
+from typing_extensions import Annotated
+from ogx_client.models.get_model_v1_models_model_id_get200_response import GetModelV1ModelsModelIdGet200Response
+
+from ogx_client.api_client import ApiClient, RequestSerialized
+from ogx_client.api_response import ApiResponse
+from ogx_client.rest import RESTResponseType
+from ogx_client.stream import Stream
+from ogx_client.lib._utils import pascal_to_snake_case
+from ogx_client.api.openai_api import OpenaiApi
+
+
+class ModelsApi:
+    """NOTE: This class is auto generated by OpenAPI Generator
+    Ref: https://openapi-generator.tech
+
+    Do not edit the class manually.
+    """
+
+    def __init__(self, api_client=None) -> None:
+        if api_client is None:
+            api_client = ApiClient.get_default()
+        self.api_client = api_client
+        self.openai: Optional[OpenaiApi] = None
+        self.logger = logging.getLogger(ModelsApi.__name__)
+
+        # Child API attributes (set by OgxClient based on x-nesting-path)
+
+
+
+    def list(self, *args, **kwargs):
+        """Proxy to self.openai.list()."""
+        return self.openai.list(*args, **kwargs)
+    def _create_event_stream(self, response_data: RESTResponseType, stream_type_name: str) -> Stream[Any]:
+        """Create a typed event stream from an SSE response.
+
+        Args:
+            response_data: The raw HTTP response containing the SSE stream.
+            stream_type_name: The schema name for the streaming response type
+                (e.g., 'OpenAIChatCompletionChunk', 'OpenAIResponseObjectStream').
+        """
+        # Import the stream type class once
+        stream_class = None
+        discriminator_map = None
+        discriminator_property = None
+        try:
+            module_name = pascal_to_snake_case(stream_type_name)
+            model_module = importlib.import_module('ogx_client.models.' + module_name)
+            stream_class = getattr(model_module, stream_type_name)
+            # Check if the stream type has a discriminator (e.g., oneOf with propertyName)
+            discriminator_map = getattr(stream_class, 'discriminator_value_class_map', None)
+            discriminator_property = getattr(stream_class, 'discriminator_property_name', None)
+        except (ImportError, ModuleNotFoundError, AttributeError) as e:
+            self.logger.debug(f"Could not import stream type {stream_type_name}: {e}")
+
+        def stream_decoder(data_str: str) -> Any:
+            if not data_str:
+                return None
+            try:
+                data = json.loads(data_str)
+            except json.JSONDecodeError:
+                return data_str
+
+            # If we have a discriminator, resolve the specific variant class
+            if discriminator_map and discriminator_property:
+                disc_value = data.get(discriminator_property, '')
+                variant_class_name = discriminator_map.get(disc_value)
+                if variant_class_name:
+                    try:
+                        variant_module_name = pascal_to_snake_case(variant_class_name)
+                        variant_module = importlib.import_module('ogx_client.models.' + variant_module_name)
+                        variant_class = getattr(variant_module, variant_class_name)
+                        return variant_class.from_dict(data)
+                    except Exception as e:
+                        self.logger.debug(f"Failed to deserialize as {variant_class_name}: {e}")
+
+            # Non-discriminated: deserialize as stream type directly
+            if stream_class is not None:
+                try:
+                    result = stream_class.from_dict(data)
+                    # Unwrap oneOf/anyOf wrappers
+                    if hasattr(result, 'actual_instance') and result.actual_instance is not None:
+                        return result.actual_instance
+                    return result
+                except Exception as e:
+                    self.logger.debug(f"Failed to deserialize as {stream_type_name}: {e}")
+
+            # Final fallback: return raw dict
+            return data
+
+        return Stream(
+            response=response_data.response,
+            client=self.api_client,
+            decoder=stream_decoder,
+        )
+
+
+    @validate_call
+    def retrieve(
+        self,
+        model_id: Annotated[StrictStr, Field(description="The ID of the model to get.")],
+        anthropic_version: Optional[StrictStr] = None,
+        x_goog_api_key: Optional[StrictStr] = None,
+        x_goog_user_project: Optional[StrictStr] = None,
+        x_goog_api_client: Optional[StrictStr] = None,
+        _request_timeout: Annotated[StrictFloat, Field(gt=0)] | tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ] | None = None,
+        _request_auth: dict[StrictStr, Any] | None = None,
+        _content_type: StrictStr | None = None,
+        _headers: dict[StrictStr, Any] | None = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0
+    ) -> GetModelV1ModelsModelIdGet200Response:
+        """Get a model by its identifier.
+
+        Get a model by its identifier. Returns OpenAI, Anthropic, or Google response format based on SDK detection headers.
+
+        :param model_id: The ID of the model to get. (required)
+        :type model_id: str
+        :param anthropic_version:
+        :type anthropic_version: str
+        :param x_goog_api_key:
+        :type x_goog_api_key: str
+        :param x_goog_user_project:
+        :type x_goog_user_project: str
+        :param x_goog_api_client:
+        :type x_goog_api_client: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+        
+
+        _param = self._retrieve_serialize(
+            model_id=model_id,
+            anthropic_version=anthropic_version,
+            x_goog_api_key=x_goog_api_key,
+            x_goog_user_project=x_goog_user_project,
+            x_goog_api_client=x_goog_api_client,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: dict[str, str | None] = {
+            '200': "GetModelV1ModelsModelIdGet200Response",
+            '400': "Error",
+            '429': "Error",
+            '500': "Error",
+        }
+
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+
+        response_data.read()
+        _deserialized = self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+        # Unwrap List*Response wrappers to return the data field directly
+        return _deserialized
+
+    @validate_call
+    def retrieve_with_http_info(
+        self,
+        model_id: Annotated[StrictStr, Field(description="The ID of the model to get.")],
+        anthropic_version: Optional[StrictStr] = None,
+        x_goog_api_key: Optional[StrictStr] = None,
+        x_goog_user_project: Optional[StrictStr] = None,
+        x_goog_api_client: Optional[StrictStr] = None,
+        _request_timeout: Annotated[StrictFloat, Field(gt=0)] | tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ] | None = None,
+        _request_auth: dict[StrictStr, Any] | None = None,
+        _content_type: StrictStr | None = None,
+        _headers: dict[StrictStr, Any] | None = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0
+    ) -> ApiResponse[GetModelV1ModelsModelIdGet200Response]:
+        """Get a model by its identifier.
+
+        Get a model by its identifier. Returns OpenAI, Anthropic, or Google response format based on SDK detection headers.
+
+        :param model_id: The ID of the model to get. (required)
+        :type model_id: str
+        :param anthropic_version:
+        :type anthropic_version: str
+        :param x_goog_api_key:
+        :type x_goog_api_key: str
+        :param x_goog_user_project:
+        :type x_goog_user_project: str
+        :param x_goog_api_client:
+        :type x_goog_api_client: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+        
+
+        _param = self._retrieve_serialize(
+            model_id=model_id,
+            anthropic_version=anthropic_version,
+            x_goog_api_key=x_goog_api_key,
+            x_goog_user_project=x_goog_user_project,
+            x_goog_api_client=x_goog_api_client,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: dict[str, str | None] = {
+            '200': "GetModelV1ModelsModelIdGet200Response",
+            '400': "Error",
+            '429': "Error",
+            '500': "Error",
+        }
+
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+    @validate_call
+    def retrieve_without_preload_content(
+        self,
+        model_id: Annotated[StrictStr, Field(description="The ID of the model to get.")],
+        anthropic_version: Optional[StrictStr] = None,
+        x_goog_api_key: Optional[StrictStr] = None,
+        x_goog_user_project: Optional[StrictStr] = None,
+        x_goog_api_client: Optional[StrictStr] = None,
+        _request_timeout: Annotated[StrictFloat, Field(gt=0)] | tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ] | None = None,
+        _request_auth: dict[StrictStr, Any] | None = None,
+        _content_type: StrictStr | None = None,
+        _headers: dict[StrictStr, Any] | None = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0
+    ) -> RESTResponseType:
+        """Get a model by its identifier.
+
+        Get a model by its identifier. Returns OpenAI, Anthropic, or Google response format based on SDK detection headers.
+
+        :param model_id: The ID of the model to get. (required)
+        :type model_id: str
+        :param anthropic_version:
+        :type anthropic_version: str
+        :param x_goog_api_key:
+        :type x_goog_api_key: str
+        :param x_goog_user_project:
+        :type x_goog_user_project: str
+        :param x_goog_api_client:
+        :type x_goog_api_client: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+        
+
+        _param = self._retrieve_serialize(
+            model_id=model_id,
+            anthropic_version=anthropic_version,
+            x_goog_api_key=x_goog_api_key,
+            x_goog_user_project=x_goog_user_project,
+            x_goog_api_client=x_goog_api_client,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: dict[str, str | None] = {
+            '200': "GetModelV1ModelsModelIdGet200Response",
+            '400': "Error",
+            '429': "Error",
+            '500': "Error",
+        }
+
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+
+    def _retrieve_serialize(
+        self,
+        model_id,
+        anthropic_version,
+        x_goog_api_key,
+        x_goog_user_project,
+        x_goog_api_client,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: dict[str, str] = {
+        }
+
+        _path_params: dict[str, str] = {}
+        _query_params: list[tuple[str, str]] = []
+        _header_params: dict[str, str | None] = _headers or {}
+        _form_params: list[tuple[str, str]] = []
+        _files: dict[
+            str, str | bytes | list[str] | list[bytes] | list[tuple[str, bytes]]
+        ] = {}
+        _body_params: bytes | None = None
+
+        # process the path parameters
+        if model_id is not None:
+            _path_params['model_id'] = model_id
+        # process the query parameters
+        # process the header parameters
+        if anthropic_version is not None:
+            _header_params['anthropic-version'] = anthropic_version
+        if x_goog_api_key is not None:
+            _header_params['x-goog-api-key'] = x_goog_api_key
+        if x_goog_user_project is not None:
+            _header_params['x-goog-user-project'] = x_goog_user_project
+        if x_goog_api_client is not None:
+            _header_params['x-goog-api-client'] = x_goog_api_client
+        # process the form parameters
+        # process the body parameter
+
+
+        # set the HTTP header `Accept`
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'application/json'
+                ]
+            )
+
+
+        # authentication setting
+        _auth_settings: list[str] = [
+        ]
+
+        return self.api_client.param_serialize(
+            method='GET',
+            resource_path='/v1/models/{model_id}',
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth
+        )
+
+