Source code for genai.client

from __future__ import annotations

from typing import Optional, overload

from genai._types import ModelLike
from genai._utils.api_client import ApiClient
from genai._utils.api_client import BaseConfig as ApiClientConfig
from genai._utils.general import to_model_instance
from genai._utils.service import (
    BaseService,
    BaseServiceConfig,
    BaseServiceServices,
)
from genai.credentials import Credentials
from genai.deployment import DeploymentService as _DeploymentService
from genai.file import FileService as _FileService
from genai.folder import FolderService as _FolderService
from genai.model import ModelService as _ModelService
from genai.prompt import PromptService as _PromptService
from genai.request import RequestService as _RequestService
from genai.system_prompt import SystemPromptService as _SystemPromptService
from genai.tag import TagService as _TagService
from genai.task import TaskService as _TaskService
from genai.text import TextService as _TextService
from genai.tune import TuneService as _TuneService
from genai.user import UserService as _UserService

__all__ = ["Client", "BaseConfig", "BaseServices"]




class BaseServices(BaseServiceServices):
    """Appropriate services used by the Client"""

    TextService: type[_TextService] = _TextService
    RequestService: type[_RequestService] = _RequestService
    TuneService: type[_TuneService] = _TuneService
    ModelService: type[_ModelService] = _ModelService
    FileService: type[_FileService] = _FileService
    PromptService: type[_PromptService] = _PromptService
    SystemPromptService: type[_SystemPromptService] = _SystemPromptService
    UserService: type[_UserService] = _UserService
    TagService: type[_TagService] = _TagService
    FolderService: type[_FolderService] = _FolderService
    TaskService: type[_TaskService] = _TaskService
    DeploymentService: type[_DeploymentService] = _DeploymentService





class BaseConfig(BaseServiceConfig):
    """Client's configuration model"""

    api_client_config: Optional[ModelLike[ApiClientConfig]] = None





class Client(BaseService[BaseConfig, BaseServices]):
    """
    The `Client` class provides an interface for interacting with various services through an API client.
    It can be initialized with either an `api_client` or `credentials` along with optional configurations and services.

    Example::

        from genai import Credentials, Client

        credentials = Credentials.from_env()
        client = Client(credentials=credentials)

    Attributes:
        text: An instance of the `TextService` class for text-related operations.
        request: An instance of the `RequestService` class for making request-related operations.
        tune: An instance of the `TuneService` class for tuning models.
        model: An instance of the `ModelService` class for managing models.
        file: An instance of the `FileService` class for managing files.
        prompt: An instance of the `PromptService` class for working with prompts.
        system_prompt: An instance of the `SystemPromptService` class for working with system prompts.
        user: An instance of the `UserService` class for managing user-related operations.
        tag: An instance of the `TagService` class for working with tags.
        folder: An instance of the `FolderService` class for working with folder.
        task: An instance of the `TaskService` class for working with tasks.
    """

    Config = BaseConfig
    Services = BaseServices

    @overload
    def __init__(
        self,
        *,
        api_client: ApiClient,
        config: Optional[ModelLike[BaseConfig]] = None,
        services: Optional[BaseServices] = None,
    ) -> None: ...

    @overload
    def __init__(
        self,
        *,
        credentials: Credentials,
        config: Optional[ModelLike[BaseConfig]] = None,
        services: Optional[BaseServices] = None,
    ) -> None: ...



    def __init__(
        self,
        *,
        credentials: Optional[Credentials] = None,
        api_client: Optional[ApiClient] = None,
        config: Optional[ModelLike[BaseConfig]] = None,
        services: Optional[BaseServices] = None,
    ) -> None:
        """
        Args:
            credentials: The credentials used to authenticate the API client.
            api_client: The API client used to make requests to the API.
            config: The configuration for the API client.
            services: The services object containing instances of various service classes.

        Raises:
            ValueError: Either 'api_client' or 'credentials' needs to be passed.

        Note:
            The `api_client` parameter must be provided either directly or through `credentials`.
            If no `services` parameter is provided, the client will use the default one.
        """
        config = to_model_instance(config, self.Config)

        if (not api_client and not credentials) or (api_client and credentials):
            raise ValueError("Either 'api_client' or 'credentials' needs to passed.")

        if not api_client:
            assert credentials
            api_client = ApiClient(credentials=credentials, config=config.api_client_config)

        assert api_client and config
        super().__init__(api_client=api_client, config=config)

        if not services:
            services = self.Services()

        self.text = services.TextService(api_client=api_client)
        self.request = services.RequestService(api_client=api_client)
        self.tune = services.TuneService(api_client=api_client)
        self.model = services.ModelService(api_client=api_client)
        self.file = services.FileService(api_client=api_client)
        self.prompt = services.PromptService(api_client=api_client)
        self.system_prompt = services.SystemPromptService(api_client=api_client)
        self.user = services.UserService(api_client=api_client)
        self.tag = services.TagService(api_client=api_client)
        self.folder = services.FolderService(api_client=api_client)
        self.task = services.TaskService(api_client=api_client)
        self.deployment = services.DeploymentService(api_client=api_client)