KMS/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix/2770 suggestions for next steps (#2788)

Browse Source
This commit is contained in:
crazywoola 2024-03-12 16:27:55 +08:00 committed by GitHub
parent f734cca337
commit 6bd2730317
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
api/core/prompt/output_parser/suggested_questions_after_answer.py
View File

@ -4,7 +4,6 @@ from typing import Any
from langchain.schema import BaseOutputParser from langchain.schema import BaseOutputParser
from core.model_runtime.errors.invoke import InvokeError
from core.prompt.prompts import SUGGESTED_QUESTIONS_AFTER_ANSWER_INSTRUCTION_PROMPT from core.prompt.prompts import SUGGESTED_QUESTIONS_AFTER_ANSWER_INSTRUCTION_PROMPT
@ -14,11 +13,11 @@ class SuggestedQuestionsAfterAnswerOutputParser(BaseOutputParser):
return SUGGESTED_QUESTIONS_AFTER_ANSWER_INSTRUCTION_PROMPT return SUGGESTED_QUESTIONS_AFTER_ANSWER_INSTRUCTION_PROMPT
def parse(self, text: str) -> Any: def parse(self, text: str) -> Any:
json_string = text.strip() action_match = re.search(r"\[.*?\]", text.strip(), re.DOTALL)
action_match = re.search(r".*(\[\".+\"\]).*", json_string, re.DOTALL)
if action_match is not None: if action_match is not None:
json_obj = json.loads(action_match.group(1).strip(), strict=False) json_obj = json.loads(action_match.group(0).strip())
else: else:
raise InvokeError("Could not parse LLM output: {text}") json_obj= []
print(f"Could not parse LLM output: {text}")
return json_obj return json_obj