diff --git a/config.py b/config.py index e668ac90..3e8396ff 100644 --- a/config.py +++ b/config.py @@ -45,6 +45,7 @@ AVAIL_LLM_MODELS = ["qwen-max", "o1-mini", "o1-mini-2024-09-12", "o1", "o1-2024- "gemini-1.5-pro", "chatglm3", "chatglm4", "deepseek-chat", "deepseek-coder", "deepseek-reasoner", "volcengine-deepseek-r1-250120", "volcengine-deepseek-v3-241226", + "deepseek-r1","deepseek-v3", ] EMBEDDING_MODEL = "text-embedding-3-small" diff --git a/request_llms/bridge_all.py b/request_llms/bridge_all.py index e911a765..cabfbe71 100644 --- a/request_llms/bridge_all.py +++ b/request_llms/bridge_all.py @@ -813,8 +813,9 @@ if "qwen-local" in AVAIL_LLM_MODELS: }) except: logger.error(trimmed_format_exc()) -# -=-=-=-=-=-=- 通义-在线模型 -=-=-=-=-=-=- -qwen_models = ["qwen-max-latest", "qwen-max-2025-01-25","qwen-max","qwen-turbo","qwen-plus"] + +# -=-=-=-=-=-=- 阿里云百炼(通义)-在线模型 -=-=-=-=-=-=- +qwen_models = ["qwen-max-latest", "qwen-max-2025-01-25","qwen-max","qwen-turbo","qwen-plus","deepseek-r1","deepseek-v3"] if any(item in qwen_models for item in AVAIL_LLM_MODELS): try: from .bridge_qwen import predict_no_ui_long_connection as qwen_noui @@ -864,10 +865,30 @@ if any(item in qwen_models for item in AVAIL_LLM_MODELS): "max_token": 30720, "tokenizer": tokenizer_gpt35, "token_cnt": get_token_num_gpt35, + }, + "deepseek-r1": { + "fn_with_ui": qwen_ui, + "fn_without_ui": qwen_noui, + "enable_reasoning": True, + "can_multi_thread": True, + "endpoint": None, + "max_token": 57344, + "tokenizer": tokenizer_gpt35, + "token_cnt": get_token_num_gpt35, + }, + "deepseek-v3": { + "fn_with_ui": qwen_ui, + "fn_without_ui": qwen_noui, + "can_multi_thread": True, + "endpoint": None, + "max_token": 57344, + "tokenizer": tokenizer_gpt35, + "token_cnt": get_token_num_gpt35, } }) except: logger.error(trimmed_format_exc()) + # -=-=-=-=-=-=- 零一万物模型 -=-=-=-=-=-=- yi_models = ["yi-34b-chat-0205","yi-34b-chat-200k","yi-large","yi-medium","yi-spark","yi-large-turbo","yi-large-preview"] if any(item in yi_models for item in AVAIL_LLM_MODELS):