AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 02:35:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: optimize output parse failed error (#298)

Browse Source
This commit is contained in:
John Wang 2023-06-05 11:23:51 +08:00 committed by GitHub
parent 2de18a6490
commit a3ee037d6d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/generator/llm_generator.py
View File

@ -1,7 +1,7 @@
import logging import logging
from langchain.chat_models.base import BaseChatModel 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.constant import llm_constant
from core.llm.llm_builder import LLMBuilder from core.llm.llm_builder import LLMBuilder
@ -153,6 +153,8 @@ class LLMGenerator:
try: try:
output = llm(query) output = llm(query)
rule_config = output_parser.parse(output) 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: except Exception:
logging.exception("Error generating prompt") logging.exception("Error generating prompt")
rule_config = { rule_config = {