| @@ -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 = { | |||