From c9f63a1f12fb9078863a2384c0c43b3abb2afe2e Mon Sep 17 00:00:00 2001 From: xupei Date: Thu, 20 Feb 2025 17:14:57 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E8=AF=AD=E8=A8=80=E5=88=A4?= =?UTF-8?q?=E6=96=AD=E8=B0=83=E7=94=A8=E7=9A=84=E6=96=B9=E6=B3=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/chat_robot/script/service/CallQWen.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/app/service/chat_robot/script/service/CallQWen.py b/app/service/chat_robot/script/service/CallQWen.py index 780d206..9e9f1a5 100644 --- a/app/service/chat_robot/script/service/CallQWen.py +++ b/app/service/chat_robot/script/service/CallQWen.py @@ -200,6 +200,18 @@ def get_response(messages): return response +def get_assistant_response(messages): + response = Generation.call( + model='qwen-max', + api_key=QWEN_API_KEY, + messages=messages, + # seed=random.randint(1, 10000), # 设置随机数种子seed，如果没有设置，则随机数种子默认为1234 + result_format='message', # 将输出设置为message形式 + enable_search='false' + ) + return response + + def call_with_messages(message, gender): global tool_info user_input = message @@ -329,7 +341,7 @@ def get_language(message: str) -> str: } ] - first_response = get_response(messages) + first_response = get_assistant_response(messages) assistant_output = first_response.output.choices[0].message.content logging.info(f"大模型输出信息：{first_response}\n判断用户输入的语言为：{assistant_output}") # print(f"大模型输出信息：{first_response}\n判断用户输入的语言为：{assistant_output}")