Преглед на файлове

Merge pull request #451 from goecho/main

Fix bug: Header attributes (Host, Authorization, Origin, Referer) not sanitized.
Timothy Jaeryang Baek преди 1 година
родител
ревизия
5c5bde3b85
променени са 1 файла, в които са добавени 4 реда и са изтрити 4 реда
  1. 4 4
      backend/apps/ollama/main.py

+ 4 - 4
backend/apps/ollama/main.py

@@ -65,10 +65,10 @@ async def proxy(path: str, request: Request, user=Depends(get_current_user)):
     else:
         raise HTTPException(status_code=401, detail=ERROR_MESSAGES.ACCESS_PROHIBITED)
 
-    headers.pop("Host", None)
-    headers.pop("Authorization", None)
-    headers.pop("Origin", None)
-    headers.pop("Referer", None)
+    headers.pop("host", None)
+    headers.pop("authorization", None)
+    headers.pop("origin", None)
+    headers.pop("referer", None)
 
     r = None