Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-16 06:13:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix runnable vistitor for funcs without pos args (#15182)

Browse Source
This commit is contained in:
Nuno Campos 2023-12-26 12:42:24 -08:00 committed by GitHub
parent b4a0d206d9
commit 7e26559256
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/core/langchain_core/runnables/utils.py
View File

@ -107,14 +107,20 @@ class IsFunctionArgDict(ast.NodeVisitor):
self.keys: Set[str] = set() self.keys: Set[str] = set()
def visit_Lambda(self, node: ast.Lambda) -> Any: def visit_Lambda(self, node: ast.Lambda) -> Any:
if not node.args.args:
return
input_arg_name = node.args.args[0].arg input_arg_name = node.args.args[0].arg
IsLocalDict(input_arg_name, self.keys).visit(node.body) IsLocalDict(input_arg_name, self.keys).visit(node.body)
def visit_FunctionDef(self, node: ast.FunctionDef) -> Any: def visit_FunctionDef(self, node: ast.FunctionDef) -> Any:
if not node.args.args:
return
input_arg_name = node.args.args[0].arg input_arg_name = node.args.args[0].arg
IsLocalDict(input_arg_name, self.keys).visit(node) IsLocalDict(input_arg_name, self.keys).visit(node)
def visit_AsyncFunctionDef(self, node: ast.AsyncFunctionDef) -> Any: def visit_AsyncFunctionDef(self, node: ast.AsyncFunctionDef) -> Any:
if not node.args.args:
return
input_arg_name = node.args.args[0].arg input_arg_name = node.args.args[0].arg
IsLocalDict(input_arg_name, self.keys).visit(node) IsLocalDict(input_arg_name, self.keys).visit(node)