1

app.py

@@ -1,8 +1,9 @@
 # uvicorn app:app --host 0.0.0.0 --port 7860 --reload
 
-from fastapi import FastAPI, Request, HTTPException
-import re
+from fastapi import FastAPI, Request, HTTPException, Response
+import re, json
 import httpx
+import uuid
 
 app = FastAPI()
 
@@ -38,17 +39,26 @@ async def proxy_url(dest_url: str, request: Request):
             if response.status_code == 200:
                 # 检查响应内容类型是否为 JSON
                 if 'application/json' in response.headers.get('content-type', ''):
-                    return response.json()
+                    json_response = response.json()
+                    json_response['id'] = f"chatcmpl-{uuid.uuid4()}"
+                    # resp = Response(content=str(json_response), media_type="application/json")
+                    # resp = Response(content=str(json_response), media_type="application/json")
+                    resp = Response(content=json.dumps(json_response), media_type="application/json")
+                    resp.headers["Access-Control-Allow-Origin"] = "*"
+                    return resp
                 else:
-                    return {"error": "Response is not in JSON format"}
+                    return {"error": "Response is not in JSON format", "id": f"chatcmpl-{uuid.uuid4()}"}
             else:
                 # 将错误响应转换为 JSON 格式并返回给客户端
                 try:
                     error_data = response.json()
+                    error_data['id'] = f"chatcmpl-{uuid.uuid4()}"
                 except ValueError:
-                    error_data = {"status_code": response.status_code, "detail": response.text}
-                return error_data
+                    error_data = {"status_code": response.status_code, "detail": response.text, "id": f"chatcmpl-{uuid.uuid4()}"}
+                resp = Response(content=str(error_data), media_type="application/json")
+                resp.headers["Access-Control-Allow-Origin"] = "*"
+                return resp
         
         except httpx.RequestError as e:
             # 处理请求错误
-            raise HTTPException(status_code=500, detail=str(e))
+            raise HTTPException(status_code=500, detail=str(e))