From 04cddfba0d1be3a6a919a05e8ce0852aa31e94b4 Mon Sep 17 00:00:00 2001 From: Bagatur <22008038+baskaryan@users.noreply.github.com> Date: Mon, 10 Jul 2023 03:21:23 -0400 Subject: [PATCH] Add lark import error (#7465) --- langchain/chains/query_constructor/parser.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/langchain/chains/query_constructor/parser.py b/langchain/chains/query_constructor/parser.py index c0c10acfb23..5c06188cbbf 100644 --- a/langchain/chains/query_constructor/parser.py +++ b/langchain/chains/query_constructor/parser.py @@ -145,6 +145,11 @@ def get_parser( Returns: Lark parser for the query language. """ + # QueryTransformer is None when Lark cannot be imported. + if QueryTransformer is None: + raise ImportError( + "Cannot import lark, please install it with 'pip install lark'." + ) transformer = QueryTransformer( allowed_comparators=allowed_comparators, allowed_operators=allowed_operators )