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

Fix Bug in from_langchain in tools.py (#268) 

* Fix from_langchain in tools.py
This commit is contained in:
RolandJAAI 2025-01-20 10:50:03 +01:00 committed by GitHub
parent 3178b18aab
commit 35f719164b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/smolagents/tools.py
View File

@ -188,7 +188,7 @@ 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 Tools that use a "generic" signature (PipelineTool, SpaceToolWrapper) # Validate forward function signature, except for Tools that use a "generic" signature (PipelineTool, SpaceToolWrapper, LangChainToolWrapper)
if not ( if not (
hasattr(self, "skip_forward_signature_validation") hasattr(self, "skip_forward_signature_validation")
and getattr(self, "skip_forward_signature_validation") is True and getattr(self, "skip_forward_signature_validation") is True
@ -676,6 +676,8 @@ class Tool:
""" """
class LangChainToolWrapper(Tool): class LangChainToolWrapper(Tool):
skip_forward_signature_validation = True
def __init__(self, _langchain_tool): def __init__(self, _langchain_tool):
self.name = _langchain_tool.name.lower() self.name = _langchain_tool.name.lower()
self.description = _langchain_tool.description self.description = _langchain_tool.description
@ -686,6 +688,7 @@ class Tool:
input_content["description"] = "" input_content["description"] = ""
self.output_type = "string" self.output_type = "string"
self.langchain_tool = _langchain_tool self.langchain_tool = _langchain_tool
self.is_initialized = True
def forward(self, *args, **kwargs): def forward(self, *args, **kwargs):
tool_input = kwargs.copy() tool_input = kwargs.copy()