Update output_parser.py

This commit is contained in:
Marco Vinciguerra 2024-09-28 11:05:09 +02:00
parent 560f079d4c
commit 27ae896cb7

View File

@ -23,8 +23,7 @@ def get_structured_output_parser(schema: Union[Dict[str, Any],
return _dict_output_parser
def get_pydantic_output_parser(schema: Union[Dict[str, Any],
Type[BaseModelV1 | BaseModelV2], Type])
-> JsonOutputParser:
Type[BaseModelV1 | BaseModelV2], Type]) -> JsonOutputParser:
"""
Get the correct output parser for the LLM model.