From 50d94a1d433057b17721f5846f46fa74f5b66e71 Mon Sep 17 00:00:00 2001
From: xupei <xupei3360@163.com>
Date: Fri, 7 Feb 2025 15:32:41 +0800
Subject: [PATCH] =?UTF-8?q?=E8=AF=AD=E8=A8=80=E5=88=A4=E6=96=AD=20?=
 =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E7=A4=BA=E4=BE=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../chat_robot/script/service/CallQWen.py     | 24 ++++++++++++++++---
 1 file changed, 21 insertions(+), 3 deletions(-)

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("森林")