fix: tools valves

This commit is contained in:
Timothy J. Baek 2024-06-23 19:10:52 -07:00
parent db9e5e008f
commit 26e735618e

View File

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