Bläddra i källkod

fix: tools valves

Timothy J. Baek 10 månader sedan
förälder
incheckning
26e735618e
1 ändrade filer med 3 tillägg och 3 borttagningar
  1. 3 3
      backend/apps/webui/routers/tools.py

+ 3 - 3
backend/apps/webui/routers/tools.py

@@ -164,9 +164,9 @@ async def get_toolkit_valves_spec_by_id(
             toolkit_module = load_toolkit_module_by_id(id)
             request.app.state.TOOLS[id] = toolkit_module
 
-        if hasattr(toolkit_module, "UserValves"):
-            UserValves = toolkit_module.UserValves
-            return UserValves.schema()
+        if hasattr(toolkit_module, "Valves"):
+            Valves = toolkit_module.Valves
+            return Valves.schema()
         return None
     else:
         raise HTTPException(