fix: only edit body with whitelisted paths

dfcc3142 · Timothy J. Baek · 6c58bb59 · dfcc3142
Commit dfcc3142 authored Mar 08, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

backend/main.py backend/main.py +5 -1

No files found.
--- a/backend/main.py
+++ b/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