Update main.py

main.py

@@ -1,9 +1,9 @@
 import os
 import json
-import asyncio
+import time
 import httpx
 from fastapi import FastAPI, Request
-from fastapi.responses import StreamingResponse
+from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 
 app = FastAPI()
@@ -60,42 +60,43 @@ async def chat_completions(request: Request):
         }
     }
     
-    async def generate():
-        # 设置 120 秒超时，防止长文本生成失败
-        async with httpx.AsyncClient(timeout=120.0) as client:
-            # 已修复：改为 app.airops.com
-            url = f"https://app.airops.com/public_api/airops_apps/{AIROPS_APP_ID}/execute"
-            
-            try:
-                response = await client.post(url, headers=headers, json=payload)
-                if response.status_code == 200:
-                    data = response.json()
-                    # 提取 AirOps 返回的文本结果
-                    result_text = data.get("result", "")
-                    
-                    if isinstance(result_text, dict) or isinstance(result_text, list):
-                        result_text = json.dumps(result_text, ensure_ascii=False)
-                    
-                    # 核心：将静态结果转换为完美的 OpenAI 流式打字机输出
-                    chunk_size = 3
-                    for i in range(0, len(result_text), chunk_size):
-                        chunk = result_text[i:i+chunk_size]
-                        openai_chunk = {
-                            "id": "chatcmpl-airops",
-                            "object": "chat.completion.chunk",
-                            "choices": [{"delta": {"content": chunk}}]
+    # 设置 120 秒超时，防止长文本生成失败
+    async with httpx.AsyncClient(timeout=120.0) as client:
+        url = f"https://app.airops.com/public_api/airops_apps/{AIROPS_APP_ID}/execute"
+        
+        try:
+            response = await client.post(url, headers=headers, json=payload)
+            if response.status_code == 200:
+                data = response.json()
+                # 提取 AirOps 返回的文本结果
+                result_text = data.get("result", "")
+                
+                if isinstance(result_text, (dict, list)):
+                    result_text = json.dumps(result_text, ensure_ascii=False)
+                
+                # 核心修改：取消流式，直接返回标准的 OpenAI 格式 JSON 响应
+                return {
+                    "id": f"chatcmpl-airops-{int(time.time())}",
+                    "object": "chat.completion",
+                    "created": int(time.time()),
+                    "model": "airops-model",
+                    "choices": [
+                        {
+                            "index": 0,
+                            "message": {
+                                "role": "assistant",
+                                "content": result_text
+                            },
+                            "finish_reason": "stop"
                         }
-                        yield f"data: {json.dumps(openai_chunk)}\n\n"
-                        await asyncio.sleep(0.01) # 极速打字机效果
-                        
-                    yield "data: [DONE]\n\n"
-                else:
-                    err = {"error": f"AirOps 接口报错: {response.text}"}
-                    yield f"data: {json.dumps(err)}\n\n"
-                    yield "data: [DONE]\n\n"
-            except Exception as e:
-                err = {"error": f"代理服务器内部错误: {str(e)}"}
-                yield f"data: {json.dumps(err)}\n\n"
-                yield "data: [DONE]\n\n"
-
-    return StreamingResponse(generate(), media_type="text/event-stream")
+                    ],
+                    "usage": {
+                        "prompt_tokens": 0,
+                        "completion_tokens": 0,
+                        "total_tokens": 0
+                    }
+                }
+            else:
+                return JSONResponse(status_code=500, content={"error": f"AirOps 接口报错: {response.text}"})
+        except Exception as e:
+            return JSONResponse(status_code=500, content={"error": f"代理服务器内部错误: {str(e)}"})