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