Prechádzať zdrojové kódy

fix: tools valves access

Timothy Jaeryang Baek 3 mesiacov pred
rodič
commit
8fc5532e2f
1 zmenil súbory, kde vykonal 11 pridanie a 0 odobranie
  1. 11 0
      backend/open_webui/routers/tools.py

+ 11 - 0
backend/open_webui/routers/tools.py

@@ -309,6 +309,17 @@ async def update_tools_valves_by_id(
             status_code=status.HTTP_401_UNAUTHORIZED,
             detail=ERROR_MESSAGES.NOT_FOUND,
         )
+
+    if (
+        tools.user_id != user.id
+        and not has_access(user.id, "write", tools.access_control)
+        and user.role != "admin"
+    ):
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
+        )
+
     if id in request.app.state.TOOLS:
         tools_module = request.app.state.TOOLS[id]
     else: