From 475053801e771fc314c9c14428ec01ebbd6119d3 Mon Sep 17 00:00:00 2001 From: zchengrong <124802516+zchengrong@users.noreply.github.com> Date: Tue, 22 Apr 2025 14:19:15 +0800 Subject: [PATCH] =?UTF-8?q?feat=EF=BC=88=E6=96=B0=E5=8A=9F=E8=83=BD?= =?UTF-8?q?=EF=BC=89:=20fix=EF=BC=88=E4=BF=AE=E5=A4=8Dbug=EF=BC=89:=20=20?= =?UTF-8?q?=20chat=20robot=20=20=E2=80=9Cparameters=E2=80=9D=E4=BF=AE?= =?UTF-8?q?=E5=A4=8D=20docs=EF=BC=88=E6=96=87=E6=A1=A3=E5=8F=98=E6=9B=B4?= =?UTF-8?q?=EF=BC=89:=20refactor=EF=BC=88=E9=87=8D=E6=9E=84=EF=BC=89:=20te?= =?UTF-8?q?st(=E5=A2=9E=E5=8A=A0=E6=B5=8B=E8=AF=95):?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../chat_robot/script/service/CallQWen.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/app/service/chat_robot/script/service/CallQWen.py b/app/service/chat_robot/script/service/CallQWen.py index 780d206..d2f28a0 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 @@ -271,7 +283,7 @@ def call_with_messages(message, gender): elif assistant_output.tool_calls[0]['function']['name'] == 'get_image_from_vector_db': content = json.loads(assistant_output.tool_calls[0]['function']['arguments']) tool_info = {"name": "get_image_from_vector_db", "role": "tool", - 'content': get_image_from_vector_db(gender, content['parameters']['content'])} + 'content': get_image_from_vector_db(gender, content['parameters']['content'] if "parameters" in content.keys() else content['content'])} flag = False result_content = tool_info['content'] response_type = "image" @@ -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}")