Update app.py

app.py

@@ -73,6 +73,7 @@ def root():
                 "body": {
                     "messages": "Array of message objects",
                     "stream": "Boolean (true for streaming response)",
+                    "model": "Model to be used (optional, defaults to claude-3-opus-20240229)"
                 }
             }
         }
@@ -86,6 +87,7 @@ def messages():
 
     try:
         json_body = request.json
+        model = json_body.get('model', 'claude-3-opus-20240229')  # 动态获取模型，默认 claude-3-opus-20240229
         stream = json_body.get('stream', True)  # 默认为True
         previous_messages = "\n\n".join([msg['content'] for msg in json_body['messages']])
         msg_id = str(uuid.uuid4())
@@ -94,7 +96,7 @@ def messages():
 
         if not stream:
             # 处理 stream 为 false 的情况
-            return handle_non_stream(previous_messages, msg_id)
+            return handle_non_stream(previous_messages, msg_id, model)
 
         # 记录日志：此时请求上下文仍然有效
         log_request(request.remote_addr, request.path, 200)
@@ -107,7 +109,7 @@ def messages():
                     "type": "message",
                     "role": "assistant",
                     "content": [],
-                    "model": "claude-3-opus-20240229",
+                    "model": model,  # 动态模型
                     "stop_reason": None,
                     "stop_sequence": None,
                     "usage": {"input_tokens": 8, "output_tokens": 1},
@@ -124,8 +126,6 @@ def messages():
                     "attachments": [],
                     "language": "en-GB",
                     "timezone": "Europe/London",
-                    "search_focus": "writing",
-                    "frontend_uuid": str(uuid.uuid4()),
                     "mode": "concise",
                     "is_related_query": False,
                     "is_default_related_query": False,
@@ -205,7 +205,7 @@ def messages():
         log_request(request.remote_addr, request.path, 400)
         return jsonify({"error": str(e)}), 400
 
-def handle_non_stream(previous_messages, msg_id):
+def handle_non_stream(previous_messages, msg_id, model):
     """
     处理 stream 为 false 的情况，返回完整的响应。
     """
@@ -221,8 +221,6 @@ def handle_non_stream(previous_messages, msg_id):
                 "attachments": [],
                 "language": "en-GB",
                 "timezone": "Europe/London",
-                "search_focus": "writing",
-                "frontend_uuid": str(uuid.uuid4()),
                 "mode": "concise",
                 "is_related_query": False,
                 "is_default_related_query": False,
@@ -268,7 +266,7 @@ def handle_non_stream(previous_messages, msg_id):
         full_response = {
             "id": msg_id,
             "content": [{"text": ''.join(response_text)}],  # 合并所有文本块
-            "model": "claude-3-opus-20240229",
+            "model": model,  # 动态模型
             "stop_reason": "end_turn",
             "stop_sequence": None,
             "usage": {