forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
16 lines
595 B
Python
16 lines
595 B
Python
from langchain.output_parsers.regex import RegexParser
|
|
|
|
|
|
def load_output_parser(config: dict) -> dict:
|
|
"""Load output parser."""
|
|
if "output_parsers" in config:
|
|
if config["output_parsers"] is not None:
|
|
_config = config["output_parsers"]
|
|
output_parser_type = _config["_type"]
|
|
if output_parser_type == "regex_parser":
|
|
output_parser = RegexParser(**_config)
|
|
else:
|
|
raise ValueError(f"Unsupported output parser {output_parser_type}")
|
|
config["output_parsers"] = output_parser
|
|
return config
|