@inkeep/agents-api 0.0.1 → 0.43.0

package/dist/domains/run/utils/stream-helpers.js

@@ -5,6 +5,7 @@ import { parsePartialJson } from "ai";
 var SSEStreamHelper = class {
 	isTextStreaming = false;
 	queuedEvents = [];
+	toolCallIndexById = /* @__PURE__ */ new Map();
 	constructor(stream, requestId, timestamp) {
 		this.stream = stream;
 		this.requestId = requestId;
@@ -40,6 +41,25 @@ var SSEStreamHelper = class {
 			}]
 		}) });
 	}
+	getToolIndex(toolCallId) {
+		const existing = this.toolCallIndexById.get(toolCallId);
+		if (existing !== void 0) return existing;
+		const next = this.toolCallIndexById.size;
+		this.toolCallIndexById.set(toolCallId, next);
+		return next;
+	}
+	async writeToolCallsDelta(toolCalls) {
+		await this.stream.writeSSE({ data: JSON.stringify({
+			id: this.requestId,
+			object: "chat.completion.chunk",
+			created: this.timestamp,
+			choices: [{
+				index: 0,
+				delta: { tool_calls: toolCalls },
+				finish_reason: null
+			}]
+		}) });
+	}
 	/**
 	* Stream text word by word with optional delay
 	*/
@@ -97,6 +117,65 @@ var SSEStreamHelper = class {
 			}]
 		}) });
 	}
+	async writeToolInputStart(params) {
+		const index = this.getToolIndex(params.toolCallId);
+		await this.writeToolCallsDelta([{
+			index,
+			id: params.toolCallId,
+			type: "function",
+			function: {
+				name: params.toolName,
+				arguments: ""
+			}
+		}]);
+	}
+	async writeToolInputDelta(params) {
+		const index = this.getToolIndex(params.toolCallId);
+		await this.writeToolCallsDelta([{
+			index,
+			id: null,
+			type: null,
+			function: {
+				name: null,
+				arguments: params.inputTextDelta
+			}
+		}]);
+	}
+	async writeToolInputAvailable(params) {
+		const fullArgs = JSON.stringify(params.input ?? {});
+		if (fullArgs) await this.writeToolInputDelta({
+			toolCallId: params.toolCallId,
+			inputTextDelta: fullArgs
+		});
+	}
+	async writeToolOutputAvailable(params) {
+		await this.writeContent(JSON.stringify({
+			type: "tool-output-available",
+			toolCallId: params.toolCallId,
+			output: params.output
+		}));
+	}
+	async writeToolOutputError(params) {
+		await this.writeContent(JSON.stringify({
+			type: "tool-output-error",
+			toolCallId: params.toolCallId,
+			errorText: params.errorText,
+			output: params.output ?? null
+		}));
+	}
+	async writeToolApprovalRequest(params) {
+		await this.writeContent(JSON.stringify({
+			type: "tool-approval-request",
+			approvalId: params.approvalId,
+			toolCallId: params.toolCallId
+		}));
+	}
+	async writeToolOutputDenied(params) {
+		await this.writeContent(JSON.stringify({
+			type: "tool-output-denied",
+			toolCallId: params.toolCallId
+		}));
+	}
 	async writeSummary(summary) {
 		if (this.isTextStreaming) {
 			this.queuedEvents.push({
@@ -166,6 +245,7 @@ var VercelDataStreamHelper = class VercelDataStreamHelper {
 			this.forceCleanup("Connection lifetime exceeded");
 		}, STREAM_MAX_LIFETIME_MS);
 	}
+	setSessionId(_sessionId) {}
 	async writeRole(_ = "assistant") {}
 	async writeContent(content) {
 		if (this.isCompleted) {
@@ -272,6 +352,64 @@ var VercelDataStreamHelper = class VercelDataStreamHelper {
 			type: "error"
 		});
 	}
+	async writeToolInputStart(params) {
+		if (this.isCompleted) return;
+		this.writer.write({
+			type: "tool-input-start",
+			toolCallId: params.toolCallId,
+			toolName: params.toolName
+		});
+	}
+	async writeToolInputDelta(params) {
+		if (this.isCompleted) return;
+		this.writer.write({
+			type: "tool-input-delta",
+			toolCallId: params.toolCallId,
+			inputTextDelta: params.inputTextDelta
+		});
+	}
+	async writeToolInputAvailable(params) {
+		if (this.isCompleted) return;
+		this.writer.write({
+			type: "tool-input-available",
+			toolCallId: params.toolCallId,
+			toolName: params.toolName,
+			input: params.input,
+			...params.providerMetadata ? { providerMetadata: params.providerMetadata } : {}
+		});
+	}
+	async writeToolOutputAvailable(params) {
+		if (this.isCompleted) return;
+		this.writer.write({
+			type: "tool-output-available",
+			toolCallId: params.toolCallId,
+			output: params.output
+		});
+	}
+	async writeToolOutputError(params) {
+		if (this.isCompleted) return;
+		this.writer.write({
+			type: "tool-output-error",
+			toolCallId: params.toolCallId,
+			errorText: params.errorText,
+			output: params.output ?? null
+		});
+	}
+	async writeToolApprovalRequest(params) {
+		if (this.isCompleted) return;
+		this.writer.write({
+			type: "tool-approval-request",
+			approvalId: params.approvalId,
+			toolCallId: params.toolCallId
+		});
+	}
+	async writeToolOutputDenied(params) {
+		if (this.isCompleted) return;
+		this.writer.write({
+			type: "tool-output-denied",
+			toolCallId: params.toolCallId
+		});
+	}
 	async streamData(data) {
 		await this.writeContent(JSON.stringify(data));
 	}
@@ -458,6 +596,7 @@ var BufferingStreamHelper = class {
 	capturedSummaries = [];
 	hasError = false;
 	errorMessage = "";
+	setSessionId(_sessionId) {}
 	async writeRole(_role) {}
 	async writeContent(content) {
 		this.capturedText += content;
@@ -487,6 +626,49 @@ var BufferingStreamHelper = class {
 		this.hasError = true;
 		this.errorMessage = typeof error === "string" ? error : error.message;
 	}
+	async writeToolInputStart(params) {
+		this.capturedData.push({
+			type: "tool-input-start",
+			...params
+		});
+	}
+	async writeToolInputDelta(params) {
+		this.capturedData.push({
+			type: "tool-input-delta",
+			...params
+		});
+	}
+	async writeToolInputAvailable(params) {
+		this.capturedData.push({
+			type: "tool-input-available",
+			...params
+		});
+	}
+	async writeToolOutputAvailable(params) {
+		this.capturedData.push({
+			type: "tool-output-available",
+			...params
+		});
+	}
+	async writeToolOutputError(params) {
+		this.capturedData.push({
+			type: "tool-output-error",
+			...params,
+			output: params.output ?? null
+		});
+	}
+	async writeToolApprovalRequest(params) {
+		this.capturedData.push({
+			type: "tool-approval-request",
+			...params
+		});
+	}
+	async writeToolOutputDenied(params) {
+		this.capturedData.push({
+			type: "tool-output-denied",
+			...params
+		});
+	}
 	async complete() {}
 	/**
 	* Get the captured response for non-streaming output