feat: 支持讯飞星火v3.0 [功能请求]: 什么时候支持讯飞星火认知大模型3.0？ #925

GaiZhenbiao · Nov 3, 2023 · 69f0c41 · 69f0c41
1 parent f9abb09
commit 69f0c41
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/modules/models/spark.py b/modules/models/spark.py
@@ -75,6 +75,9 @@ def __init__(self, model_name, appid, api_key, api_secret, user_name="") -> None
         if "2.0" in self.model_name:
             self.spark_url = "wss://spark-api.xf-yun.com/v2.1/chat"
             self.domain = "generalv2"
+        if "3.0" in self.model_name:
+            self.spark_url = "wss://spark-api.xf-yun.com/v3.1/chat"
+            self.domain = "generalv3"
         else:
             self.spark_url = "wss://spark-api.xf-yun.com/v1.1/chat"
             self.domain = "general"

diff --git a/modules/presets.py b/modules/presets.py
@@ -73,6 +73,7 @@
     "yuanai-1.0-rhythm_poems",
     "minimax-abab5-chat",
     "midjourney",
+    "讯飞星火大模型V3.0",
     "讯飞星火大模型V2.0",
     "讯飞星火大模型V1.5",
     "Claude"