Tuchuanhuhuhu commited on
Commit
69f0c41
·
1 Parent(s): f9abb09

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

Browse files
Files changed (2) hide show
  1. modules/models/spark.py +3 -0
  2. modules/presets.py +1 -0
modules/models/spark.py CHANGED
@@ -75,6 +75,9 @@ class Spark_Client(BaseLLMModel):
75
  if "2.0" in self.model_name:
76
  self.spark_url = "wss://spark-api.xf-yun.com/v2.1/chat"
77
  self.domain = "generalv2"
 
 
 
78
  else:
79
  self.spark_url = "wss://spark-api.xf-yun.com/v1.1/chat"
80
  self.domain = "general"
 
75
  if "2.0" in self.model_name:
76
  self.spark_url = "wss://spark-api.xf-yun.com/v2.1/chat"
77
  self.domain = "generalv2"
78
+ if "3.0" in self.model_name:
79
+ self.spark_url = "wss://spark-api.xf-yun.com/v3.1/chat"
80
+ self.domain = "generalv3"
81
  else:
82
  self.spark_url = "wss://spark-api.xf-yun.com/v1.1/chat"
83
  self.domain = "general"
modules/presets.py CHANGED
@@ -73,6 +73,7 @@ ONLINE_MODELS = [
73
  "yuanai-1.0-rhythm_poems",
74
  "minimax-abab5-chat",
75
  "midjourney",
 
76
  "讯飞星火大模型V2.0",
77
  "讯飞星火大模型V1.5",
78
  "Claude"
 
73
  "yuanai-1.0-rhythm_poems",
74
  "minimax-abab5-chat",
75
  "midjourney",
76
+ "讯飞星火大模型V3.0",
77
  "讯飞星火大模型V2.0",
78
  "讯飞星火大模型V1.5",
79
  "Claude"