diff --git a/app/ai/models.py b/app/ai/models.py index 4d60727..c85e22d 100644 --- a/app/ai/models.py +++ b/app/ai/models.py @@ -16,6 +16,8 @@ from app.config import settings # 供应商连接配置 _VOLCENGINE = (lambda: settings.volcengine_api_key, lambda: settings.volcengine_base_url) _CARDIAC = (lambda: settings.cardiacBrder_api_key, lambda: settings.cardiacBrder_base_url) +_JIEKOU = (lambda: settings.jiekou_api_key, lambda: settings.jiekou_base_url) + class LLM(Enum): @@ -30,10 +32,10 @@ class LLM(Enum): DOUBAO_SEED_PRO = ("doubao-seed-2-0-pro-260215", *_VOLCENGINE) # 心缘 - GPT_4O = ("gpt-4o", *_CARDIAC) - GPT_4O_MINI = ("gpt-4o-mini", *_CARDIAC) - CLAUDE_SONNET_4 = ("claude-sonnet-4-20250514", *_CARDIAC) - GEMINI_FLASH = ("gemini-2.5-flash", *_CARDIAC) + GPT_4O = ("gpt-4o", *_JIEKOU) + GPT_4O_MINI = ("gpt-4o-mini", *_JIEKOU) + CLAUDE_SONNET_4 = ("claude-sonnet-4-20250514", *_JIEKOU) + GEMINI_FLASH = ("gemini-2.5-flash", *_JIEKOU) def __init__(self, model_name: str, api_key_fn, base_url_fn): self.model_name = model_name diff --git a/app/config/settings.py b/app/config/settings.py index 40ede2f..269084d 100644 --- a/app/config/settings.py +++ b/app/config/settings.py @@ -42,6 +42,10 @@ class Settings(BaseSettings): cardiacBrder_api_key: str = "sk-8NxoLe7ZTJveGSmtPENBm4NwN9ai4YLGw8y6fqueZrPTo4Uu" cardiacBrder_base_url: str = "https://api-i.xykjy.com/v1" + # jiekou + jiekou_api_key: str = "sk_4BXi4_yvwx-NW4_zuVt2LufrrlGyhs2q9qMN5Vq_j7Q" + jiekou_base_url: str = "https://api.jiekou.ai/openai" + # JWT jwt_secret: str = "Aa123123" token_expire_seconds: int = 5184000