from volcenginesdkarkruntime import Ark
from openai import OpenAI
from ..config.config import Config

config = Config()
# 初始化客户端
# client = Ark(api_key=config.API_KEY, base_url=config.BASE_URL,default_headers={"ark-beta-knowledge-search": "true"})  # 启用私域知识库搜索

client = OpenAI(
    base_url=config.BASE_URL,
    api_key=config.API_KEY,
    default_headers={"ark-beta-knowledge-search": "true"}  # 启用私域知识库搜索
)