Browse Source

chore: :art: format

ThatOneCalculator 1 year ago
parent
commit
699fbefb8a
2 changed files with 18 additions and 11 deletions
  1. 18 10
      backend/apps/openai/main.py
  2. 0 1
      backend/config.py

+ 18 - 10
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":
     if user and user.role == "admin":
         return {"OPENAI_API_BASE_URL": app.state.OPENAI_API_BASE_URL}
         return {"OPENAI_API_BASE_URL": app.state.OPENAI_API_BASE_URL}
     else:
     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")
 @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":
     if user and user.role == "admin":
         app.state.OPENAI_API_BASE_URL = form_data.url
         app.state.OPENAI_API_BASE_URL = form_data.url
         return {"OPENAI_API_BASE_URL": app.state.OPENAI_API_BASE_URL}
         return {"OPENAI_API_BASE_URL": app.state.OPENAI_API_BASE_URL}
     else:
     else:
-        raise HTTPException(status_code=401, detail=ERROR_MESSAGES.ACCESS_PROHIBITED)
+        raise HTTPException(status_code=401,
+                            detail=ERROR_MESSAGES.ACCESS_PROHIBITED)
 
 
 
 
 @app.get("/key")
 @app.get("/key")
@@ -54,16 +57,19 @@ async def get_openai_key(user=Depends(get_current_user)):
     if user and user.role == "admin":
     if user and user.role == "admin":
         return {"OPENAI_API_KEY": app.state.OPENAI_API_KEY}
         return {"OPENAI_API_KEY": app.state.OPENAI_API_KEY}
     else:
     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")
 @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":
     if user and user.role == "admin":
         app.state.OPENAI_API_KEY = form_data.key
         app.state.OPENAI_API_KEY = form_data.key
         return {"OPENAI_API_KEY": app.state.OPENAI_API_KEY}
         return {"OPENAI_API_KEY": app.state.OPENAI_API_KEY}
     else:
     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"])
 @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)
     print(target_url, app.state.OPENAI_API_KEY)
 
 
     if user.role not in ["user", "admin"]:
     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 == "":
     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()
     body = await request.body()
     # headers = dict(request.headers)
     # 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":
             if "openai" in app.state.OPENAI_API_BASE_URL and path == "models":
                 response_data["data"] = list(
                 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
             return response_data
     except Exception as e:
     except Exception as e:

+ 0 - 1
backend/config.py

@@ -26,7 +26,6 @@ if ENV == "prod":
     if OLLAMA_API_BASE_URL == "/ollama/api":
     if OLLAMA_API_BASE_URL == "/ollama/api":
         OLLAMA_API_BASE_URL = "http://host.docker.internal:11434/api"
         OLLAMA_API_BASE_URL = "http://host.docker.internal:11434/api"
 
 
-
 ####################################
 ####################################
 # OPENAI_API
 # OPENAI_API
 ####################################
 ####################################