ark_client.py 458 B

1234567891011121314
  1. from volcenginesdkarkruntime import Ark
  2. from openai import OpenAI
  3. from ..config.config import Config
  4. config = Config()
  5. # 初始化客户端
  6. # client = Ark(api_key=config.API_KEY, base_url=config.BASE_URL,default_headers={"ark-beta-knowledge-search": "true"}) # 启用私域知识库搜索
  7. client = OpenAI(
  8. base_url=config.BASE_URL,
  9. api_key=config.API_KEY,
  10. default_headers={"ark-beta-knowledge-search": "true"} # 启用私域知识库搜索
  11. )