From df3d0f61bdddea1c2ece955956487ce69c6aa057 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Mon, 24 Feb 2025 15:43:32 +0800 Subject: [PATCH] Fix base url missing for deepseek from Tongyi. (#5294) ### What problem does this PR solve? ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- rag/llm/chat_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rag/llm/chat_model.py b/rag/llm/chat_model.py index af98e2644..a8ba8413c 100644 --- a/rag/llm/chat_model.py +++ b/rag/llm/chat_model.py @@ -259,7 +259,7 @@ class QWenChat(Base): dashscope.api_key = key self.model_name = model_name if model_name.lower().find("deepseek") >= 0: - super().__init__(key, model_name) + super().__init__(key, model_name, "https://dashscope.aliyuncs.com/compatible-mode/v1") def chat(self, system, history, gen_conf): if self.model_name.lower().find("deepseek") >= 0: @@ -334,7 +334,7 @@ class QWenChat(Base): def chat_streamly(self, system, history, gen_conf): if self.model_name.lower().find("deepseek") >= 0: return super.chat_streamly(system, history, gen_conf) - + return self._chat_streamly(system, history, gen_conf)