From 955801db2e773cc8855a2be91b8d671a833093d7 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Tue, 25 Feb 2025 17:42:29 +0800 Subject: [PATCH] Resolve super class invokation error. (#5337) ### 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 5e23c9326..4580e3019 100644 --- a/rag/llm/chat_model.py +++ b/rag/llm/chat_model.py @@ -265,7 +265,7 @@ class QWenChat(Base): def chat(self, system, history, gen_conf): if self.model_name.lower().find("deepseek") >= 0: - return super.chat(system, history, gen_conf) + return super().chat(system, history, gen_conf) stream_flag = str(os.environ.get('QWEN_CHAT_BY_STREAM', 'true')).lower() == 'true' if not stream_flag: @@ -335,7 +335,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 super().chat_streamly(system, history, gen_conf) return self._chat_streamly(system, history, gen_conf)