Timothy J. Baek пре 10 месеци
родитељ
комит
26e735618e
1 измењених фајлова са 3 додато и 3 уклоњено
  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(