Browse Source

fix: only edit body with whitelisted paths

Timothy J. Baek 1 year ago
parent
commit
dfcc314283
1 changed files with 5 additions and 1 deletions
  1. 5 1
      backend/main.py

+ 5 - 1
backend/main.py

@@ -62,8 +62,12 @@ async def on_startup():
 class RAGMiddleware(BaseHTTPMiddleware):
     async def dispatch(self, request: Request, call_next):
 
+        "chat/completions" in request.url.path
+
         print(request.url.path)
-        if request.method == "POST":
+        if request.method == "POST" and (
+            "/api/chat" in request.url.path or "/chat/completions" in request.url.path
+        ):
             # Read the original request body
             body = await request.body()
             # Decode body to string