-
Notifications
You must be signed in to change notification settings - Fork 2.1k
/
output_parser.ts
58 lines (54 loc) · 1.88 KB
/
output_parser.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import { AgentAction, AgentFinish } from "@langchain/core/agents";
import { OutputParserException } from "@langchain/core/output_parsers";
import { AgentActionOutputParser } from "../types.js";
/**
* @example
* ```typescript
* const prompt = ChatPromptTemplate.fromMessages([
* HumanMessagePromptTemplate.fromTemplate(AGENT_INSTRUCTIONS),
* new MessagesPlaceholder("agent_scratchpad"),
* ]);
* const runnableAgent = RunnableSequence.from([
* ...rest of runnable
* prompt,
* new ChatAnthropic({ modelName: "claude-2", temperature: 0 }).bind({
* stop: ["</tool_input>", "</final_answer>"],
* }),
* new XMLAgentOutputParser(),
* ]);
* const result = await executor.invoke({
* input: "What is the weather in Honolulu?",
* tools: [],
* });
* ```
*/
export class XMLAgentOutputParser extends AgentActionOutputParser {
lc_namespace = ["langchain", "agents", "xml"];
static lc_name() {
return "XMLAgentOutputParser";
}
/**
* Parses the output text from the agent and returns an AgentAction or
* AgentFinish object.
* @param text The output text from the agent.
* @returns An AgentAction or AgentFinish object.
*/
async parse(text: string): Promise<AgentAction | AgentFinish> {
if (text.includes("</tool>")) {
const [tool, toolInput] = text.split("</tool>");
const _tool = tool.split("<tool>")[1];
const _toolInput = toolInput.split("<tool_input>")[1];
return { tool: _tool, toolInput: _toolInput, log: text };
} else if (text.includes("<final_answer>")) {
const [, answer] = text.split("<final_answer>");
return { returnValues: { output: answer }, log: text };
} else {
throw new OutputParserException(`Could not parse LLM output: ${text}`);
}
}
getFormatInstructions(): string {
throw new Error(
"getFormatInstructions not implemented inside OpenAIFunctionsAgentOutputParser."
);
}
}