From ac4e9c2277c6a71c679b8454028405dd968b669c Mon Sep 17 00:00:00 2001 From: kingbri Date: Thu, 16 Nov 2023 22:19:32 -0500 Subject: [PATCH] API: Add CORS support Tell CORS to go fly a kite. Signed-off-by: kingbri --- main.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index b5238dd..9b37840 100644 --- a/main.py +++ b/main.py @@ -3,6 +3,7 @@ import yaml import pathlib from auth import check_admin_key, check_api_key, load_auth_keys from fastapi import FastAPI, Request, HTTPException, Depends +from fastapi.middleware.cors import CORSMiddleware from model import ModelContainer from progress.bar import IncrementalBar from sse_starlette import EventSourceResponse @@ -36,9 +37,18 @@ def _check_model_container(): if model_container is None or model_container.model is None: raise HTTPException(400, "No models are loaded.") +# ALlow CORS requests +app.add_middleware( + CORSMiddleware, + allow_origins=["*"], + allow_credentials=True, + allow_methods=["*"], + allow_headers=["*"], +) + # Model list endpoint -@app.get("/v1/models", dependencies=[Depends(check_api_key)]) -@app.get("/v1/model/list", dependencies=[Depends(check_api_key)]) +@app.get("/v1/models") +@app.get("/v1/model/list") async def list_models(): model_config = config.get("model", {}) if "model_dir" in model_config: