Fix Bug in from_langchain in tools.py (#268)
* Fix from_langchain in tools.py
This commit is contained in:
parent
3178b18aab
commit
35f719164b
|
@ -188,7 +188,7 @@ class Tool:
|
|||
|
||||
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 (
|
||||
hasattr(self, "skip_forward_signature_validation")
|
||||
and getattr(self, "skip_forward_signature_validation") is True
|
||||
|
@ -676,6 +676,8 @@ class Tool:
|
|||
"""
|
||||
|
||||
class LangChainToolWrapper(Tool):
|
||||
skip_forward_signature_validation = True
|
||||
|
||||
def __init__(self, _langchain_tool):
|
||||
self.name = _langchain_tool.name.lower()
|
||||
self.description = _langchain_tool.description
|
||||
|
@ -686,6 +688,7 @@ class Tool:
|
|||
input_content["description"] = ""
|
||||
self.output_type = "string"
|
||||
self.langchain_tool = _langchain_tool
|
||||
self.is_initialized = True
|
||||
|
||||
def forward(self, *args, **kwargs):
|
||||
tool_input = kwargs.copy()
|
||||
|
|
Loading…
Reference in New Issue