浏览代码

fix: only edit body with whitelisted paths

Timothy J. Baek 1 年之前
父节点
当前提交
dfcc314283
共有 1 个文件被更改,包括 5 次插入1 次删除
  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