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

feat: optimize output parse (#906)

Browse Source
This commit is contained in:
takatost 2023-08-18 17:00:40 +08:00 committed by GitHub
parent 1d9cc5ca05
commit d9afebe216
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/agent/agent/output_parser/structured_chat.py
View File

@ -10,7 +10,7 @@ from langchain.schema import AgentAction, AgentFinish, OutputParserException
class StructuredChatOutputParser(LCStructuredChatOutputParser): class StructuredChatOutputParser(LCStructuredChatOutputParser):
def parse(self, text: str) -> Union[AgentAction, AgentFinish]: def parse(self, text: str) -> Union[AgentAction, AgentFinish]:
try: try:
action_match = re.search(r"```(.*?)\n?(.*?)```", text, re.DOTALL) action_match = re.search(r"```(\w*)\n?({.*?)```", text, re.DOTALL)
if action_match is not None: if action_match is not None:
response = json.loads(action_match.group(2).strip(), strict=False) response = json.loads(action_match.group(2).strip(), strict=False)
if isinstance(response, list): if isinstance(response, list):
@ -26,4 +26,4 @@ class StructuredChatOutputParser(LCStructuredChatOutputParser):
else: else:
return AgentFinish({"output": text}, text) return AgentFinish({"output": text}, text)
except Exception as e: except Exception as e:
raise OutputParserException(f"Could not parse LLM output: {text}") from e raise OutputParserException(f"Could not parse LLM output: {text}")