From 699fbefb8a3f4f3f02634bd714e8d0ac8d72fcfe Mon Sep 17 00:00:00 2001 From: ThatOneCalculator Date: Thu, 4 Jan 2024 20:05:19 -0800 Subject: [PATCH] chore: :art: format --- backend/apps/openai/main.py | 28 ++++++++++++++++++---------- backend/config.py | 1 - 2 files changed, 18 insertions(+), 11 deletions(-) diff --git a/backend/apps/openai/main.py b/backend/apps/openai/main.py index 03d4621b..ef9330c5 100644 --- a/backend/apps/openai/main.py +++ b/backend/apps/openai/main.py @@ -37,16 +37,19 @@ async def get_openai_url(user=Depends(get_current_user)): if user and user.role == "admin": return {"OPENAI_API_BASE_URL": app.state.OPENAI_API_BASE_URL} else: - raise HTTPException(status_code=401, detail=ERROR_MESSAGES.ACCESS_PROHIBITED) + raise HTTPException(status_code=401, + detail=ERROR_MESSAGES.ACCESS_PROHIBITED) @app.post("/url/update") -async def update_openai_url(form_data: UrlUpdateForm, user=Depends(get_current_user)): +async def update_openai_url(form_data: UrlUpdateForm, + user=Depends(get_current_user)): if user and user.role == "admin": app.state.OPENAI_API_BASE_URL = form_data.url return {"OPENAI_API_BASE_URL": app.state.OPENAI_API_BASE_URL} else: - raise HTTPException(status_code=401, detail=ERROR_MESSAGES.ACCESS_PROHIBITED) + raise HTTPException(status_code=401, + detail=ERROR_MESSAGES.ACCESS_PROHIBITED) @app.get("/key") @@ -54,16 +57,19 @@ async def get_openai_key(user=Depends(get_current_user)): if user and user.role == "admin": return {"OPENAI_API_KEY": app.state.OPENAI_API_KEY} else: - raise HTTPException(status_code=401, detail=ERROR_MESSAGES.ACCESS_PROHIBITED) + raise HTTPException(status_code=401, + detail=ERROR_MESSAGES.ACCESS_PROHIBITED) @app.post("/key/update") -async def update_openai_key(form_data: KeyUpdateForm, user=Depends(get_current_user)): +async def update_openai_key(form_data: KeyUpdateForm, + user=Depends(get_current_user)): if user and user.role == "admin": app.state.OPENAI_API_KEY = form_data.key return {"OPENAI_API_KEY": app.state.OPENAI_API_KEY} else: - raise HTTPException(status_code=401, detail=ERROR_MESSAGES.ACCESS_PROHIBITED) + raise HTTPException(status_code=401, + detail=ERROR_MESSAGES.ACCESS_PROHIBITED) @app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE"]) @@ -72,9 +78,11 @@ async def proxy(path: str, request: Request, user=Depends(get_current_user)): print(target_url, app.state.OPENAI_API_KEY) if user.role not in ["user", "admin"]: - raise HTTPException(status_code=401, detail=ERROR_MESSAGES.ACCESS_PROHIBITED) + raise HTTPException(status_code=401, + detail=ERROR_MESSAGES.ACCESS_PROHIBITED) if app.state.OPENAI_API_KEY == "": - raise HTTPException(status_code=401, detail=ERROR_MESSAGES.API_KEY_NOT_FOUND) + raise HTTPException(status_code=401, + detail=ERROR_MESSAGES.API_KEY_NOT_FOUND) body = await request.body() # headers = dict(request.headers) @@ -117,8 +125,8 @@ async def proxy(path: str, request: Request, user=Depends(get_current_user)): if "openai" in app.state.OPENAI_API_BASE_URL and path == "models": response_data["data"] = list( - filter(lambda model: "gpt" in model["id"], response_data["data"]) - ) + filter(lambda model: "gpt" in model["id"], + response_data["data"])) return response_data except Exception as e: diff --git a/backend/config.py b/backend/config.py index e6b815d5..4c518d13 100644 --- a/backend/config.py +++ b/backend/config.py @@ -26,7 +26,6 @@ if ENV == "prod": if OLLAMA_API_BASE_URL == "/ollama/api": OLLAMA_API_BASE_URL = "http://host.docker.internal:11434/api" - #################################### # OPENAI_API ####################################