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

fix issue #69

This commit is contained in:
François Le Lay 2025-01-05 12:10:40 -05:00
parent 0e712bb93f
commit 6b7f75c6ba
2 changed files with 98 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

92
examples/flux.ipynb Normal file
View File

File diff suppressed because one or more lines are too long

9
src/smolagents/tools.py
View File

@ -208,8 +208,8 @@ class Tool:
# Validate forward function signature, except for PipelineTool # Validate forward function signature, except for PipelineTool
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,