chat-ai-api/app/core/ark_client.py

from openai import OpenAI
from ..db.ai_config import get_config_by_app_name


def get_client(app_name: str = "com.yunxiangshengtai") -> OpenAI:
    """根据appName动态初始化client"""
    config = get_config_by_app_name(app_name)
    if not config:
        raise ValueError(f"未找到appName '{app_name}' 的配置")

    return OpenAI(
        base_url=config["baseUrl"],
        api_key=config["apiKey"],
        default_headers={"ark-beta-knowledge-search": "true"}
    )