From 212aadd4afc8a03ee55172e069947d88bdfb1b40 Mon Sep 17 00:00:00 2001 From: Davis Chase <130488702+dev2049@users.noreply.github.com> Date: Thu, 27 Apr 2023 14:41:59 -0700 Subject: [PATCH] Nit: list to sequence (#3678) --- langchain/chains/query_constructor/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/langchain/chains/query_constructor/base.py b/langchain/chains/query_constructor/base.py index 33d64ab781a..3fb80c4332f 100644 --- a/langchain/chains/query_constructor/base.py +++ b/langchain/chains/query_constructor/base.py @@ -55,7 +55,7 @@ class StructuredQueryOutputParser(BaseOutputParser[StructuredQuery]): return cls(ast_parse=ast_parser.parse) -def _format_attribute_info(info: List[AttributeInfo]) -> str: +def _format_attribute_info(info: Sequence[AttributeInfo]) -> str: info_dicts = {} for i in info: i_dict = dict(i) @@ -65,7 +65,7 @@ def _format_attribute_info(info: List[AttributeInfo]) -> str: def _get_prompt( document_contents: str, - attribute_info: List[AttributeInfo], + attribute_info: Sequence[AttributeInfo], examples: Optional[List] = None, allowed_comparators: Optional[Sequence[Comparator]] = None, allowed_operators: Optional[Sequence[Operator]] = None,