@n8n/n8n-nodes-langchain
Version:
49 lines • 1.95 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.processEventStream = processEventStream;
async function processEventStream(ctx, eventStream, itemIndex) {
const agentResult = {
output: '',
};
const toolCalls = [];
ctx.sendChunk('begin', itemIndex);
for await (const event of eventStream) {
switch (event.event) {
case 'on_chat_model_stream':
const chunk = event.data?.chunk;
if (chunk?.content) {
const chunkText = chunk.text;
ctx.sendChunk('item', itemIndex, chunkText);
agentResult.output += chunkText;
}
break;
case 'on_chat_model_end':
if (event.data) {
const output = event.data.output;
if (output?.tool_calls && output.tool_calls.length > 0) {
for (const toolCall of output.tool_calls) {
toolCalls.push({
tool: toolCall.name,
toolInput: toolCall.args,
toolCallId: toolCall.id || 'unknown',
type: toolCall.type || 'tool_call',
log: output.text ||
`Calling ${toolCall.name} with input: ${JSON.stringify(toolCall.args)}`,
messageLog: [output],
additionalKwargs: output.additional_kwargs,
});
}
}
}
break;
default:
break;
}
}
ctx.sendChunk('end', itemIndex);
if (toolCalls.length > 0) {
agentResult.toolCalls = toolCalls;
}
return agentResult;
}
//# sourceMappingURL=processEventStream.js.map