Selaa lähdekoodia

Clean API configuration

StrayWarrior 1 kuukausi sitten
vanhempi
commit
7dfec71c8d
3 muutettua tiedostoa jossa 12 lisäystä ja 6 poistoa
  1. 3 3
      agent_service.py
  2. 4 0
      chat_service.py
  3. 5 3
      user_profile_extractor.py

+ 3 - 3
agent_service.py

@@ -43,11 +43,11 @@ class AgentService:
         self.agent_registry: Dict[str, DialogueManager] = {}
 
         self.llm_client = OpenAI(
-            api_key='5e275c38-44fd-415f-abcf-4b59f6377f72',
-            base_url="https://ark.cn-beijing.volces.com/api/v3"
+            api_key=chat_service.VOLCENGINE_API_TOKEN,
+            base_url=chat_service.VOLCENGINE_BASE_URL
         )
         # DeepSeek on Volces
-        self.model_name = "ep-20250213194558-rrmr2"
+        self.model_name = chat_service.VOLCENGINE_MODEL_DEEPSEEK_V3
         self.coze_client = CozeChat(
             token=chat_service.COZE_API_TOKEN,
             base_url=chat_service.COZE_CN_BASE_URL

+ 4 - 0
chat_service.py

@@ -11,6 +11,10 @@ from cozepy import Coze, TokenAuth, Message, ChatStatus, MessageContentType, Cha
 
 COZE_API_TOKEN = os.getenv("COZE_API_TOKEN")
 COZE_CN_BASE_URL = 'https://api.coze.cn'
+VOLCENGINE_API_TOKEN = '5e275c38-44fd-415f-abcf-4b59f6377f72'
+VOLCENGINE_BASE_URL = "https://ark.cn-beijing.volces.com/api/v3"
+VOLCENGINE_MODEL_DEEPSEEK_V3 = "ep-20250213194558-rrmr2"
+VOLCENGINE_MODEL_DOUBAO_PRO_1_5 = 'ep-20250307150409-4blz9'
 
 class ChatServiceType(Enum):
     OPENAI_COMPATIBLE = auto

+ 5 - 3
user_profile_extractor.py

@@ -4,6 +4,8 @@
 
 import json
 from typing import Dict, Any, Optional
+
+import chat_service
 from prompt_templates import USER_PROFILE_EXTRACT_PROMPT
 from openai import OpenAI
 import logging
@@ -14,10 +16,10 @@ import global_flags
 class UserProfileExtractor:
     def __init__(self):
         self.llm_client = OpenAI(
-            api_key='5e275c38-44fd-415f-abcf-4b59f6377f72',
-            base_url="https://ark.cn-beijing.volces.com/api/v3"
+            api_key=chat_service.VOLCENGINE_API_TOKEN,
+            base_url=chat_service.VOLCENGINE_BASE_URL
         )
-        self.model_name = 'ep-20250307150409-4blz9'
+        self.model_name = chat_service.VOLCENGINE_MODEL_DOUBAO_PRO_1_5
 
     def get_extraction_function(self) -> Dict:
         """