Sultan
/
Flowise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix OpenAIFunctionAgent that function not return string result 

refer to https://github.com/langchain-ai/langchain/blob/master/libs/langchain/langchain/agents/format_scratchpad/openai_functions.py#L29

and fix the role of systemMessage from `ai` to `system`.
This commit is contained in:
YISH 2024-01-10 17:41:53 +08:00 committed by GitHub
parent 6e2bf91c21
commit be0fff4d9d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
packages/components/nodes/agents/OpenAIFunctionAgent/OpenAIFunctionAgent.ts
View File

@ -109,9 +109,18 @@ class OpenAIFunctionAgent_Agents implements INode {
const formatAgentSteps = (steps: AgentStep[]): BaseMessage[] => const formatAgentSteps = (steps: AgentStep[]): BaseMessage[] =>
steps.flatMap(({ action, observation }) => { steps.flatMap(({ action, observation }) => {
const create_function_message = (observation: string, action: AgentAction) => {
let content: string
if (typeof observation !== 'string') {
content = JSON.stringify(observation)
} else {
content = observation
}
return new FunctionMessage(content, action.tool)
}
if ('messageLog' in action && action.messageLog !== undefined) { if ('messageLog' in action && action.messageLog !== undefined) {
const log = action.messageLog as BaseMessage[] const log = action.messageLog as BaseMessage[]
return log.concat(new FunctionMessage(observation, action.tool)) return log.concat(create_function_message(observation, action))
} else { } else {
return [new AIMessage(action.log)] return [new AIMessage(action.log)]
} }
@ -127,7 +136,7 @@ const prepareAgent = (nodeData: INodeData, sessionId?: string) => {
const inputKey = memory.inputKey ? memory.inputKey : 'input' const inputKey = memory.inputKey ? memory.inputKey : 'input'
const prompt = ChatPromptTemplate.fromMessages([ const prompt = ChatPromptTemplate.fromMessages([
['ai', systemMessage ? systemMessage : `You are a helpful AI assistant.`], ['system', systemMessage ? systemMessage : `You are a helpful AI assistant.`],
new MessagesPlaceholder(memoryKey), new MessagesPlaceholder(memoryKey),
['human', `{${inputKey}}`], ['human', `{${inputKey}}`],
new MessagesPlaceholder('agent_scratchpad') new MessagesPlaceholder('agent_scratchpad')