| from langchain.output_parsers.regex import RegexParser | |
| def load_output_parser(config: dict) -> dict: | |
| """Load an output parser. | |
| Args: | |
| config: config dict | |
| Returns: | |
| config dict with output parser loaded | |
| """ | |
| 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 | |