Update app.py

app.py

@@ -219,6 +219,76 @@ def process_function_call(function_name, function_args):
 def split_message(message, max_length=500):
     return [message[i:i+max_length] for i in range(0, len(message), max_length)]
 
+# @app.route('/api/wx', methods=['GET', 'POST'])
+# def wechatai():
+#     if request.method == 'GET':
+#         return verify_wechat(request)
+#     else:
+#         # 处理POST请求
+#         xml_str = request.data
+#         if not xml_str:
+#             return ""
+        
+#         user_message_content, from_user_name, to_user_name = getUserMessageContentFromXML(xml_str)
+        
+#         if from_user_name not in user_sessions:
+#             user_sessions[from_user_name] = {'model': 'gpt-4o-mini', 'messages': [], 'pending_response': []}
+
+#         session = user_sessions[from_user_name]
+        
+#         if user_message_content.lower() == '/models':
+#             response_content = f"可用的模型列表：\n{list_available_models()}\n\n使用 /model 模型名称 来切换模型"
+#             return generate_response_xml(from_user_name, to_user_name, response_content)
+#         elif user_message_content.lower().startswith('/model'):
+#             model = user_message_content.split(' ')[1]
+#             if model in AVAILABLE_MODELS:
+#                 session['model'] = model
+#                 response_content = f'模型已切换为 {AVAILABLE_MODELS[model]}'
+#             else:
+#                 response_content = f'无效的模型名称。可用的模型有：\n{list_available_models()}'
+#             return generate_response_xml(from_user_name, to_user_name, response_content)
+#         elif user_message_content.lower() == '继续':
+#             if session['pending_response']:
+#                 response_content = session['pending_response'].pop(0)
+#                 if session['pending_response']:
+#                     response_content += '\n\n回复"继续"获取下一部分。'
+#                 else:
+#                     response_content += '\n\n回复结束。'
+#             else:
+#                 response_content = "没有待发送的消息。"
+#             return generate_response_xml(from_user_name, to_user_name, response_content)
+        
+#         session['messages'].append({"role": "user", "content": user_message_content})
+        
+#         # 调用OpenAI API
+#         ai_response = get_openai_response(session['messages'], model=session['model'], functions=FUNCTIONS, function_call="auto")
+        
+#         if ai_response.function_call:
+#             function_name = ai_response.function_call.name
+#             function_args = json.loads(ai_response.function_call.arguments)
+#             function_result = process_function_call(function_name, function_args)
+            
+#             session['messages'].append(ai_response.model_dump())
+#             session['messages'].append({
+#                 "role": "function",
+#                 "name": function_name,
+#                 "content": json.dumps(function_result, ensure_ascii=False)
+#             })
+            
+#             final_response = get_openai_response(session['messages'], model=session['model'])
+#             response_content = final_response.content
+#         else:
+#             response_content = ai_response.content
+        
+#         session['messages'].append({"role": "assistant", "content": response_content})
+        
+#         # 处理长消息
+#         response_parts = split_message(response_content)
+#         if len(response_parts) > 1:
+#             session['pending_response'] = response_parts[1:]
+#             response_content = response_parts[0] + '\n\n回复"继续"获取下一部分。'
+        
+#         return generate_response_xml(from_user_name, to_user_name, response_content)
 @app.route('/api/wx', methods=['GET', 'POST'])
 def wechatai():
     if request.method == 'GET':
@@ -232,7 +302,7 @@ def wechatai():
         user_message_content, from_user_name, to_user_name = getUserMessageContentFromXML(xml_str)
         
         if from_user_name not in user_sessions:
-            user_sessions[from_user_name] = {'model': 'gpt-4o-mini', 'messages': [], 'pending_response': []}
+            user_sessions[from_user_name] = {'model': 'gpt-3.5-turbo', 'messages': [], 'pending_response': []}
 
         session = user_sessions[from_user_name]
         
@@ -275,6 +345,7 @@ def wechatai():
                 "content": json.dumps(function_result, ensure_ascii=False)
             })
             
+            # 再次调用OpenAI API，将函数执行结果作为上下文
             final_response = get_openai_response(session['messages'], model=session['model'])
             response_content = final_response.content
         else:
@@ -289,7 +360,7 @@ def wechatai():
             response_content = response_parts[0] + '\n\n回复"继续"获取下一部分。'
         
         return generate_response_xml(from_user_name, to_user_name, response_content)
-
+        
 def list_available_models():
     return "\n".join([f"{key}: {value}" for key, value in AVAILABLE_MODELS.items()])