fix enabling sparkv4 (#1936)

2024-08-12 21:45:08 +08:00
parent f9384e4e5f
commit a119ab36fe
1 changed files with 1 additions and 1 deletions
--- a/request_llms/bridge_all.py
+++ b/request_llms/bridge_all.py
@@ -881,7 +881,7 @@ if "sparkv2" in AVAIL_LLM_MODELS:   # 讯飞星火认知大模型
        })
    except:
        print(trimmed_format_exc())
-if "sparkv3" in AVAIL_LLM_MODELS or "sparkv3.5" in AVAIL_LLM_MODELS:   # 讯飞星火认知大模型
+if any(x in AVAIL_LLM_MODELS for x in ("sparkv3", "sparkv3.5", "sparkv4")):   # 讯飞星火认知大模型
    try:
        from .bridge_spark import predict_no_ui_long_connection as spark_noui
        from .bridge_spark import predict as spark_ui