# Part of Odoo. See LICENSE file for full copyright and licensing details.
from typing import NamedTuple

from odoo.api import Environment
from odoo.exceptions import UserError


class Provider(NamedTuple):
    name: str
    display_name: str
    embedding_model: str
    embedding_config: dict
    llms: list[tuple[str, str]]
    deprecated_models: list[str]


PROVIDERS = [
    Provider(
        "openai",
        "OpenAI",
        "text-embedding-3-small",
        {
            # https://platform.openai.com/docs/api-reference/embeddings/create
            "max_batch_size": 2048,
            "max_tokens_per_request": 200000,
        },
        [
            ("gpt-3.5-turbo", "GPT-3.5 Turbo"),
            ("gpt-4", "GPT-4"),
            ("gpt-4o", "GPT-4o"),
            ("gpt-4.1", "GPT-4.1"),
            ("gpt-4.1-mini", "GPT-4.1 Mini"),
            ("gpt-5", "GPT-5"),
            ("gpt-5-mini", "GPT-5 Mini"),
        ],
        [],
    ),
    Provider(
        "google",
        "Google",
        "gemini-embedding-2",
        {
            # https://googleapis.dev/python/generativelanguage/latest/_modules/google/ai/generativelanguage_v1alpha/types/text_service.html#BatchEmbedTextRequest
            "max_batch_size": 100,
            "max_tokens_per_request": 10000,
        },
        [
            ("gemini-2.5-pro", "Gemini 2.5 Pro"),
            ("gemini-2.5-flash", "Gemini 2.5 Flash"),
            ("gemini-1.5-pro", "Gemini 1.5 Pro"),
            ("gemini-1.5-flash", "Gemini 1.5 Flash"),
        ],
        ["gemini-1.5-pro", "gemini-1.5-flash", "gemini-embedding-001"],
    ),
]

DEPRECATED_MODELS = [model for provider in PROVIDERS for model in provider.deprecated_models]

EMBEDDING_MODELS_SELECTION = [
    (provider.embedding_model, provider.display_name) for provider in PROVIDERS
]


def get_provider_for_embedding_model(env, embedding_model):
    for p in PROVIDERS:
        if p.embedding_model == embedding_model or embedding_model in p.deprecated_models:
            return p.name
    raise UserError(env._("No provider found for the embedding model"))


def get_provider(env, llm_model):
    for p in PROVIDERS:
        if llm_model in [m[0] for m in p.llms]:
            return p.name
    raise UserError(env._("No provider found for the selected model"))


def get_embedding_config(env, provider):
    for p in PROVIDERS:
        if p.name == provider:
            return p.embedding_config
    raise UserError(env._("No embedding configuration found for the provider"))


def check_model_depreciation(env: Environment, model: str) -> None:
    if model in DEPRECATED_MODELS:
        for provider in PROVIDERS:
            for name, label in provider.llms:
                if name == model:
                    raise UserError(env._("%s is no longer available. Please select a newer model.", label))


def get_embedding_model(provider_name: str):
    for provider in PROVIDERS:
        if provider.name == provider_name:
            return provider.embedding_model