Update app.py

app.py

@@ -23,6 +23,7 @@ def format_openai_response(content, finish_reason=None):
     }
 
 @app.route('/hf/v1/chat/completions', methods=['POST'])
+
 def chat_completions():
     data = request.json
     messages = data.get('messages', [])
@@ -38,20 +39,30 @@ def chat_completions():
     conversation += "\n请关注并回复user最近的消息并避免总结对话历史的回答"
 
     model = data.get('model', 'gpt-4o')
+    
+    # 根据模型名称设置endpoint
+    if model.startswith('gpt'):
+        endpoint = "openAI"
+    elif model.startswith('claude'):
+        endpoint = "claude"
+    else:
+        return {"error": "Unsupported model"}, 400
 
     original_api_url = 'https://chatpro.ai-pro.org/api/ask/openAI'
+    
     headers = {
         'content-type': 'application/json',
         'X-Forwarded-For': generate_random_ip(),
         'origin': 'https://chatpro.ai-pro.org',
         'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/127.0.0.0 Safari/537.36'
     }
+    
     payload = {
         "text": conversation,
-        "endpoint": "openAI",
+        "endpoint": endpoint,
         "model": model
     }
-
+    
     def generate():
         last_content = ""
         response = requests.post(original_api_url, headers=headers, json=payload, stream=True)