Browse Source

Fix/2770 suggestions for next steps (#2788)

tags/0.5.10
crazywoola 1 year ago
parent
commit
6bd2730317
No account linked to committer's email address

+ 4
- 5
api/core/prompt/output_parser/suggested_questions_after_answer.py View File



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




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".*(\[\".+\"\]).*", json_string, re.DOTALL)
action_match = re.search(r"\[.*?\]", text.strip(), 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

Loading…
Cancel
Save