Upload 9 files

app.py

@@ -66,31 +66,6 @@ async def list_models():
     }
 
 
-def parse_openrouter_response(response_data: dict) -> str:
-    """从OpenRouter响应中提取内容"""
-    content = ""
-    
-    # 标准OpenAI格式
-    if "choices" in response_data and response_data["choices"]:
-        choices = response_data["choices"]
-        if choices:
-            msg = choices[0].get("message", {})
-            content = msg.get("content", "")
-            if not content:
-                # 可能是delta格式
-                delta = choices[0].get("delta", {})
-                content = delta.get("content", "")
-    
-    # 直接返回的情况
-    if not content and "message" in response_data:
-        content = response_data.get("message", {}).get("content", "")
-    
-    if not content and "content" in response_data:
-        content = response_data.get("content", "")
-    
-    return content
-
-
 @fastapi_app.post("/v1/chat/completions")
 async def chat_completions(request: ChatCompletionRequest):
     prompt = request.messages[-1].content if request.messages else ""
@@ -107,32 +82,8 @@ async def chat_completions(request: ChatCompletionRequest):
     if not result.get("success"):
         raise HTTPException(status_code=400, detail=result.get("error", "Request failed"))
     
-    response_data = result.get("response", {})
-    print(f"[chat_completions] response_data: {response_data}")
-    content = parse_openrouter_response(response_data)
-    print(f"[chat_completions] parsed content: '{content}'")
-    
-    return {
-        "id": response_data.get("id", f"chatcmpl-{random.randint(100000, 999999)}"),
-        "object": "chat.completion",
-        "created": int(datetime.now().timestamp()),
-        "model": result.get("model", request.model or "unknown"),
-        "choices": [
-            {
-                "index": 0,
-                "message": {
-                    "role": "assistant",
-                    "content": content
-                },
-                "finish_reason": "stop"
-            }
-        ],
-        "usage": {
-            "prompt_tokens": len(prompt),
-            "completion_tokens": 10,
-            "total_tokens": len(prompt) + 10
-        }
-    }
+    # 直接返回原始响应
+    return result.get("response", {})
 
 
 async def stream_chat(model_hint: Optional[str], messages: list):