Timothy J. Baek 11 月之前
父節點
當前提交
8181ae3918
共有 1 個文件被更改,包括 9 次插入9 次删除
  1. 9 9
      backend/main.py

+ 9 - 9
backend/main.py

@@ -118,15 +118,15 @@ origins = ["*"]
 
 
 
 
 # Custom middleware to add security headers
 # Custom middleware to add security headers
-class SecurityHeadersMiddleware(BaseHTTPMiddleware):
-    async def dispatch(self, request: Request, call_next):
-        response: Response = await call_next(request)
-        response.headers["Cross-Origin-Opener-Policy"] = "same-origin"
-        response.headers["Cross-Origin-Embedder-Policy"] = "require-corp"
-        return response
+# class SecurityHeadersMiddleware(BaseHTTPMiddleware):
+#     async def dispatch(self, request: Request, call_next):
+#         response: Response = await call_next(request)
+#         response.headers["Cross-Origin-Opener-Policy"] = "same-origin"
+#         response.headers["Cross-Origin-Embedder-Policy"] = "require-corp"
+#         return response
 
 
 
 
-app.add_middleware(SecurityHeadersMiddleware)
+# app.add_middleware(SecurityHeadersMiddleware)
 
 
 
 
 class RAGMiddleware(BaseHTTPMiddleware):
 class RAGMiddleware(BaseHTTPMiddleware):
@@ -289,8 +289,8 @@ class ModelFilterConfigForm(BaseModel):
 async def update_model_filter_config(
 async def update_model_filter_config(
     form_data: ModelFilterConfigForm, user=Depends(get_admin_user)
     form_data: ModelFilterConfigForm, user=Depends(get_admin_user)
 ):
 ):
-    app.state.config.ENABLE_MODEL_FILTER, form_data.enabled
-    app.state.config.MODEL_FILTER_LIST, form_data.models
+    app.state.config.ENABLE_MODEL_FILTER = form_data.enabled
+    app.state.config.MODEL_FILTER_LIST = form_data.models
 
 
     ollama_app.state.ENABLE_MODEL_FILTER = app.state.config.ENABLE_MODEL_FILTER
     ollama_app.state.ENABLE_MODEL_FILTER = app.state.config.ENABLE_MODEL_FILTER
     ollama_app.state.MODEL_FILTER_LIST = app.state.config.MODEL_FILTER_LIST
     ollama_app.state.MODEL_FILTER_LIST = app.state.config.MODEL_FILTER_LIST