farcasclaudiu/Flowise
1
0
Fork 0
mirror of https://github.com/farcasclaudiu/Flowise.git synced 2026-06-28 21:00:58 +03:00
Code Issues Packages Projects Releases Wiki Activity

Bugfix/stream custom tool return direct (#3003)

Browse Source 
stream custom tool return direct
This commit is contained in:
Henry Heng
2024-08-12 18:35:15 +01:00
committed by GitHub
parent f57dc2477f
commit b9f0ec3a3f
4 changed files with 44 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/agents/XMLAgent/XMLAgent.ts
+11
View File
@@ -142,6 +142,17 @@ class XMLAgent_Agents implements INode {
options.socketIO.to(options.socketIOClientId).emit('usedTools', res.usedTools)
usedTools = res.usedTools
}
// If the tool is set to returnDirect, stream the output to the client
if (res.usedTools && res.usedTools.length) {
let inputTools = nodeData.inputs?.tools
inputTools = flatten(inputTools)
for (const tool of res.usedTools) {
const inputTool = inputTools.find((inputTool: Tool) => inputTool.name === tool.tool)
if (inputTool && inputTool.returnDirect) {
options.socketIO.to(options.socketIOClientId).emit('token', tool.toolOutput)
}
}
}
} else {
res = await executor.invoke({ input }, { callbacks: [loggerHandler, ...callbacks] })
if (res.sourceDocuments) {