From a3ee037d6d51899afcdbd360a016b6ed199ac83e Mon Sep 17 00:00:00 2001 From: John Wang Date: Mon, 5 Jun 2023 11:23:51 +0800 Subject: [PATCH] feat: optimize output parse failed error (#298) --- api/core/generator/llm_generator.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/api/core/generator/llm_generator.py b/api/core/generator/llm_generator.py index 055e54f4c7..97f4e05ddb 100644 --- a/api/core/generator/llm_generator.py +++ b/api/core/generator/llm_generator.py @@ -1,7 +1,7 @@ import logging from langchain.chat_models.base import BaseChatModel -from langchain.schema import HumanMessage +from langchain.schema import HumanMessage, OutputParserException from core.constant import llm_constant from core.llm.llm_builder import LLMBuilder @@ -153,6 +153,8 @@ class LLMGenerator: try: output = llm(query) rule_config = output_parser.parse(output) + except OutputParserException: + raise ValueError('Please give a valid input for intended audience or hoping to solve problems.') except Exception: logging.exception("Error generating prompt") rule_config = {