Fix CORS compatibility with OpenAI, vLLM, TGI, LMDeploy (#1373)

Co-authored-by: Yineng Zhang <me@zhyncs.com>
This commit is contained in:
josephrocca
2024-09-11 00:35:03 +08:00
committed by GitHub
parent e72275cf7f
commit dff2860a69

View File

@@ -37,6 +37,7 @@ import requests
import uvicorn
import uvloop
from fastapi import FastAPI, File, Form, Request, UploadFile
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import JSONResponse, Response, StreamingResponse
from sglang.lang.backend.runtime_endpoint import RuntimeEndpoint
@@ -93,6 +94,14 @@ asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
app = FastAPI()
tokenizer_manager = None
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
@app.get("/health")
async def health() -> Response: