diff --git a/modules/models/spark.py b/modules/models/spark.py index 1c5790f5..2bce33a8 100644 --- a/modules/models/spark.py +++ b/modules/models/spark.py @@ -75,9 +75,15 @@ 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: + elif "3.0" in self.model_name: self.spark_url = "wss://spark-api.xf-yun.com/v3.1/chat" self.domain = "generalv3" + elif "3.5" in self.model_name: + self.spark_url = "wss://spark-api.xf-yun.com/v3.5/chat" + self.domain = "generalv3.5" + elif "4.0" in self.model_name: + self.spark_url = "wss://spark-api.xf-yun.com/v4.0/chat" + self.domain = "4.0Ultra" 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 index 6808881c..ad59ff1d 100644 --- a/modules/presets.py +++ b/modules/presets.py @@ -84,6 +84,8 @@ "yuanai-1.0-rhythm_poems", "minimax-abab5-chat", "midjourney", + "讯飞星火大模型V4.0", + "讯飞星火大模型V3.5", "讯飞星火大模型V3.0", "讯飞星火大模型V2.0", "讯飞星火大模型V1.5", @@ -380,6 +382,8 @@ "yuanai-1.0-rhythm_poems": {"model_name": "yuanai-1.0-rhythm_poems"}, "minimax-abab5-chat": {"model_name": "minimax-abab5-chat"}, "midjourney": {"model_name": "midjourney"}, + "讯飞星火大模型V4.0": {"model_name": "讯飞星火大模型V4.0"}, + "讯飞星火大模型V3.5": {"model_name": "讯飞星火大模型V3.5"}, "讯飞星火大模型V3.0": {"model_name": "讯飞星火大模型V3.0"}, "讯飞星火大模型V2.0": {"model_name": "讯飞星火大模型V2.0"}, "讯飞星火大模型V1.5": {"model_name": "讯飞星火大模型V1.5"},