fix: tools valves

This commit is contained in:
Timothy J. Baek 2024-06-23 19:10:52 -07:00
parent db9e5e008f
commit 26e735618e
1 changed files with 3 additions and 3 deletions

View File

@ -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(