github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-16 17:53:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

handle single arg to and/or (#5637)

Browse Source 
@ryderwishart @eyurtsev thoughts on handling this in the parser itself?
related to #5570
This commit is contained in:
Davis Chase 2023-06-03 15:18:46 -07:00 committed by GitHub
parent 934319fc28
commit ae3611730a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
langchain/chains/query_constructor/parser.py
View File

@ -68,11 +68,14 @@ class QueryTransformer(Transformer):
def program(self, *items: Any) -> tuple:
return items
def func_call(self, func_name: Any, *args: Any) -> FilterDirective:
def func_call(self, func_name: Any, args: list) -> FilterDirective:
func = self._match_func_name(str(func_name))
if isinstance(func, Comparator):
return Comparison(comparator=func, attribute=args[0][0], value=args[0][1])
return Operation(operator=func, arguments=args[0])
return Comparison(comparator=func, attribute=args[0], value=args[1])
elif len(args) == 1 and func in (Operator.AND, Operator.OR):
return args[0]
else:
return Operation(operator=func, arguments=args)
def _match_func_name(self, func_name: str) -> Union[Operator, Comparator]:
if func_name in set(Comparator):

8
tests/unit_tests/chains/query_constructor/test_parser.py
View File

@ -114,3 +114,11 @@ def test_parse_bool_value(x: str) -> None:
actual = parsed.value
expected = x.lower() == "true"
assert actual == expected
@pytest.mark.parametrize("op", ("and", "or"))
@pytest.mark.parametrize("arg", ('eq("foo", 2)', 'and(eq("foo", 2), lte("bar", 1.1))'))
def test_parser_unpack_single_arg_operation(op: str, arg: str) -> None:
expected = DEFAULT_PARSER.parse(arg)
actual = DEFAULT_PARSER.parse(f"{op}({arg})")
assert expected == actual