forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
import json
|
|
from typing import Union
|
|
|
|
from langchain.agents.agent import AgentOutputParser
|
|
from langchain.agents.chat.prompt import FORMAT_INSTRUCTIONS
|
|
from langchain.schema import AgentAction, AgentFinish, OutputParserException
|
|
|
|
FINAL_ANSWER_ACTION = "Final Answer:"
|
|
|
|
|
|
class ChatOutputParser(AgentOutputParser):
|
|
def get_format_instructions(self) -> str:
|
|
return FORMAT_INSTRUCTIONS
|
|
|
|
def parse(self, text: str) -> Union[AgentAction, AgentFinish]:
|
|
includes_answer = FINAL_ANSWER_ACTION in text
|
|
try:
|
|
action = text.split("```")[1]
|
|
response = json.loads(action.strip())
|
|
includes_action = "action" in response and "action_input" in response
|
|
if includes_answer and includes_action:
|
|
raise OutputParserException(
|
|
"Parsing LLM output produced a final answer "
|
|
f"and a parse-able action: {text}"
|
|
)
|
|
return AgentAction(response["action"], response["action_input"], text)
|
|
|
|
except Exception:
|
|
if not includes_answer:
|
|
raise OutputParserException(f"Could not parse LLM output: {text}")
|
|
return AgentFinish(
|
|
{"output": text.split(FINAL_ANSWER_ACTION)[-1].strip()}, text
|
|
)
|
|
|
|
@property
|
|
def _type(self) -> str:
|
|
return "chat"
|