بناء طبقة موحدة لعدة مزوّدين: Claude و OpenAI والنماذج المحلية

import os
from typing import Callable

Message = dict[str, str]  # {"role": "user" | "assistant", "content": "..."}
Provider = Callable[[str, list[Message]], str]  # (system_prompt, messages) -> reply

def call_anthropic(system: str, messages: list[Message]) -> str:
    import anthropic
    client = anthropic.Anthropic(api_key=os.environ["AI_API_KEY"])
    response = client.messages.create(
        model=os.environ["AI_MODEL_NAME"],
        max_tokens=600,
        system=system,
        messages=messages,
    )
    return response.content[0].text

def call_openai_compatible(system: str, messages: list[Message]) -> str:
    from openai import OpenAI
    client = OpenAI(
        api_key=os.environ.get("AI_API_KEY") or "not-needed",
        base_url=os.environ.get("AI_API_BASE") or None,  # Ollama / vLLM / OpenAI
    )
    completion = client.chat.completions.create(
        model=os.environ["AI_MODEL_NAME"],
        messages=[{"role": "system", "content": system}, *messages],
    )
    return completion.choices[0].message.content

PROVIDERS: dict[str, Provider] = {
    "anthropic": call_anthropic,
    "openai": call_openai_compatible,
    "ollama": call_openai_compatible,   # Ollama يعرض واجهة متوافقة مع OpenAI
    "custom": call_openai_compatible,
}

import logging
logger = logging.getLogger(__name__)

def generate_reply(system, messages, user_text):
    name = os.getenv("AI_PROVIDER", "knowledge_base")
    provider = PROVIDERS.get(name)
    if provider:
        try:
            return provider(system, messages), name
        except Exception:
            logger.exception("provider %s failed, using fallback", name)
    return knowledge_base_reply(user_text), "knowledge_base"