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

Merge pull request #75 from lelayf/fix_space_tool_fwd_sig_validation 

Fix forward signature validation in `SpaceToolWrapper`
This commit is contained in:
Aymeric Roucher 2025-01-06 14:43:42 +01:00 committed by GitHub
parent d742a844de f0fc41d328
commit c3a11e09da
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/smolagents/tools.py
View File

@ -206,10 +206,10 @@ class Tool:
assert getattr(self, "output_type", None) in AUTHORIZED_TYPES assert getattr(self, "output_type", None) in AUTHORIZED_TYPES
# Validate forward function signature, except for PipelineTool # Validate forward function signature, except for Tools that use a "generic" signature (PipelineTool, SpaceToolWrapper)
if not ( if not (
hasattr(self, "is_pipeline_tool") hasattr(self, "skip_forward_signature_validation")
and getattr(self, "is_pipeline_tool") is True and getattr(self, "skip_forward_signature_validation") is True
): ):
signature = inspect.signature(self.forward) signature = inspect.signature(self.forward)
@ -575,6 +575,9 @@ class Tool:
from gradio_client import Client, handle_file from gradio_client import Client, handle_file
class SpaceToolWrapper(Tool): class SpaceToolWrapper(Tool):
skip_forward_signature_validation = True
def __init__( def __init__(
self, self,
space_id: str, space_id: str,
@ -1098,7 +1101,7 @@ class PipelineTool(Tool):
name = "pipeline" name = "pipeline"
inputs = {"prompt": str} inputs = {"prompt": str}
output_type = str output_type = str
is_pipeline_tool = True skip_forward_signature_validation = True
def __init__( def __init__(
self, self,