Browse Source

Update chat_service: rename

StrayWarrior 2 days ago
parent
commit
ad1705fa33
1 changed files with 5 additions and 5 deletions
  1. 5 5
      pqai_agent/chat_service.py

+ 5 - 5
pqai_agent/chat_service.py

@@ -22,9 +22,9 @@ 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 = "deepseek-v3-250324"
-VOLCENGINE_MODEL_DOUBAO_PRO_1_5 = 'ep-20250307150409-4blz9'
-VOLCENGINE_MODEL_DOUBAO_PRO_32K = 'ep-20250414202859-6nkz5'
-VOLCENGINE_MODEL_DOUBAO_1_5_VISION_PRO = 'ep-20250421193334-nz5wd'
+VOLCENGINE_MODEL_DOUBAO_PRO_1_5_32K = 'doubao-1-5-pro-32k-250115'
+VOLCENGINE_MODEL_DOUBAO_PRO_32K = 'doubao-pro-32k-241215'
+VOLCENGINE_MODEL_DOUBAO_1_5_VISION_PRO = 'doubao-1-5-vision-pro-32k-250115'
 DEEPSEEK_API_TOKEN = 'sk-67daad8f424f4854bda7f1fed7ef220b'
 DEEPSEEK_BASE_URL = 'https://api.deepseek.com/'
 DEEPSEEK_CHAT_MODEL = 'deepseek-chat'
@@ -79,7 +79,7 @@ class ModelPrice:
 class OpenAICompatible:
     volcengine_models = [
         VOLCENGINE_MODEL_DOUBAO_PRO_32K,
-        VOLCENGINE_MODEL_DOUBAO_PRO_1_5,
+        VOLCENGINE_MODEL_DOUBAO_PRO_1_5_32K,
         VOLCENGINE_MODEL_DOUBAO_1_5_VISION_PRO,
         VOLCENGINE_MODEL_DEEPSEEK_V3
     ]
@@ -97,7 +97,7 @@ class OpenAICompatible:
     model_prices = {
         VOLCENGINE_MODEL_DEEPSEEK_V3: ModelPrice(input_price=2, output_price=8),
         VOLCENGINE_MODEL_DOUBAO_PRO_32K: ModelPrice(input_price=0.8, output_price=2),
-        VOLCENGINE_MODEL_DOUBAO_PRO_1_5: ModelPrice(input_price=0.8, output_price=2),
+        VOLCENGINE_MODEL_DOUBAO_PRO_1_5_32K: ModelPrice(input_price=0.8, output_price=2),
         VOLCENGINE_MODEL_DOUBAO_1_5_VISION_PRO: ModelPrice(input_price=3, output_price=9),
         DEEPSEEK_CHAT_MODEL: ModelPrice(input_price=2, output_price=8),
         OPENAI_MODEL_GPT_4o: ModelPrice(input_price=2.5, output_price=10, currency='USD'),