瀏覽代碼

feat: optimize output parse failed error (#298)

tags/0.3.2
John Wang 2 年之前
父節點
當前提交
a3ee037d6d
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 3 行新增1 行删除
  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 = {

Loading…
取消
儲存