blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

conv_chat: raise parsing error on output parser

Browse Source
This commit is contained in:
blob42 2023-05-18 16:43:12 +02:00
parent 8dcad0f272
commit 4e42c737f8
2 changed files with 23 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
langchain/agents/conversational_chat/output_parser.py
View File

@ -5,7 +5,7 @@ from typing import Union
from langchain.agents import AgentOutputParser from langchain.agents import AgentOutputParser
from langchain.agents.conversational_chat.prompt import FORMAT_INSTRUCTIONS from langchain.agents.conversational_chat.prompt import FORMAT_INSTRUCTIONS
from langchain.schema import AgentAction, AgentFinish from langchain.schema import AgentAction, AgentFinish, OutputParserException
class ConvoOutputParser(AgentOutputParser): class ConvoOutputParser(AgentOutputParser):
@ -13,6 +13,7 @@ class ConvoOutputParser(AgentOutputParser):
return FORMAT_INSTRUCTIONS return FORMAT_INSTRUCTIONS
def parse(self, text: str) -> Union[AgentAction, AgentFinish]: def parse(self, text: str) -> Union[AgentAction, AgentFinish]:
try:
cleaned_output = text.strip() cleaned_output = text.strip()
if "```json" in cleaned_output: if "```json" in cleaned_output:
_, cleaned_output = cleaned_output.split("```json") _, cleaned_output = cleaned_output.split("```json")
@ -31,6 +32,8 @@ class ConvoOutputParser(AgentOutputParser):
return AgentFinish({"output": action_input}, text) return AgentFinish({"output": action_input}, text)
else: else:
return AgentAction(action, action_input, text) return AgentAction(action, action_input, text)
except Exception as e:
raise OutputParserException(f"Could not parse LLM output: {text}") from e
@property @property
def _type(self) -> str: def _type(self) -> str:

2
langchain/schema.py
View File

@ -360,7 +360,7 @@ class BaseOutputParser(BaseModel, ABC, Generic[T]):
return output_parser_dict return output_parser_dict
class OutputParserException(Exception): class OutputParserException(ValueError):
"""Exception that output parsers should raise to signify a parsing error. """Exception that output parsers should raise to signify a parsing error.
This exists to differentiate parsing errors from other code or execution errors This exists to differentiate parsing errors from other code or execution errors