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)