Commit 26e73561 authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

fix: tools valves

parent db9e5e00
......@@ -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(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment