Nit: list to sequence (#3678)

fix_agent_callbacks
Davis Chase 1 year ago committed by GitHub
parent b807a114e4
commit 212aadd4af
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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,

Loading…
Cancel
Save