语言判断 添加示例

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