diff --git a/app/service/chat_robot/script/service/CallQWen.py b/app/service/chat_robot/script/service/CallQWen.py index cb7669b..780d206 100644 --- a/app/service/chat_robot/script/service/CallQWen.py +++ b/app/service/chat_robot/script/service/CallQWen.py @@ -304,20 +304,38 @@ def tutorial_tool(): def get_language(message: str) -> str: messages = [ { - "content": message, # 用户message + "content": GET_LANGUAGE_PREFIX, # ai message + "role": "system" + }, + { + "content": "Tree", # 用户message "role": "user" }, { - "content": GET_LANGUAGE_PREFIX, # ai message + "content": "English", # 用户message "role": "assistant" + }, + { + "content": "玩具", # 用户message + "role": "user" + }, + { + "content": "Chinese", # 用户message + "role": "assistant" + }, + { + "content": message, # 用户message + "role": "user" } ] first_response = get_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}") return assistant_output if __name__ == '__main__': - get_language("") + for _ in range(4): + get_language("森林")