From 26e735618e46b5fce735e6aa269fe3b81a23ec1b Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Sun, 23 Jun 2024 19:10:52 -0700 Subject: [PATCH] fix: tools valves --- backend/apps/webui/routers/tools.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/apps/webui/routers/tools.py b/backend/apps/webui/routers/tools.py index 900e17999..3ab75187a 100644 --- a/backend/apps/webui/routers/tools.py +++ b/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(