Jelajahi Sumber

feat: optimize output parse failed error (#298)

John Wang 1 tahun lalu
induk
melakukan
a3ee037d6d
1 mengubah file dengan 3 tambahan dan 1 penghapusan
  1. 3 1
      api/core/generator/llm_generator.py

+ 3 - 1
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 = {