diff --git a/langchain/chains/query_constructor/base.py b/langchain/chains/query_constructor/base.py index 33d64ab7..3fb80c43 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,