KeyArgo
/
smolagents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Enable smolagent tools in Hugging Chat! 🚀 (#132) 

* Add correct signature, api_name, and description
This commit is contained in:
Aymeric Roucher 2025-01-09 15:15:06 +01:00 committed by GitHub
parent 067ae9bc90
commit cf04285cc1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 200 additions and 199 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

389
examples/benchmark.ipynb
View File

File diff suppressed because one or more lines are too long

10
src/smolagents/tools.py
View File

@ -779,8 +779,10 @@ def launch_gradio_demo(tool: Tool):
"number": gr.Textbox, "number": gr.Textbox,
} }
def fn(*args, **kwargs): def tool_forward(*args, **kwargs):
return tool(*args, **kwargs, sanitize_inputs_outputs=True) return tool(*args, sanitize_inputs_outputs=True, **kwargs)
tool_forward.__signature__ = inspect.signature(tool.forward)
gradio_inputs = [] gradio_inputs = []
for input_name, input_details in tool.inputs.items(): for input_name, input_details in tool.inputs.items():
@ -794,11 +796,13 @@ def launch_gradio_demo(tool: Tool):
gradio_output = output_gradio_componentclass(label="Output") gradio_output = output_gradio_componentclass(label="Output")
gr.Interface( gr.Interface(
fn=fn, fn=tool_forward,
inputs=gradio_inputs, inputs=gradio_inputs,
outputs=gradio_output, outputs=gradio_output,
title=tool.name, title=tool.name,
article=tool.description, article=tool.description,
description=tool.description,
api_name=tool.name,
).launch() ).launch()