@economic/agents 2.14.1 → 2.14.2

package/dist/index.d.mts

@@ -519,6 +519,8 @@ declare abstract class ChatAgent<RequestContext extends Record<string, unknown>
    * returned immediately and each event reaches the client as it is produced.
    */
   private streamMessage;
+  private streamToolApproval;
+  private streamTurn;
   /**
    * Submit feedback for a message by its id.
    * @param messageId - The id of the message to give feedback on.

package/dist/index.mjs

@@ -4,6 +4,7 @@ import { Output, convertToModelMessages, generateText, jsonSchema, pruneMessages
 import { createRemoteJWKSet, decodeJwt, errors, jwtVerify } from "jose";
 import { BasicTracerProvider, SimpleSpanProcessor } from "@opentelemetry/sdk-trace-base";
 import { genericObservability } from "agents/observability";
+import { applyToolUpdate, toolApprovalUpdate } from "agents/chat";
 import { createCompactFunction, estimateMessageTokens, estimateStringTokens, sanitizeToolPairs, truncateOlderMessages } from "agents/experimental/memory/utils";
 import { nanoid } from "nanoid";
 //#region src/server/features/auth/index.ts
@@ -2846,6 +2847,22 @@ var ChatAgent = class extends Agent {
 		if (action === "getMessages" && request.method === "GET") return Response.json(await this.getMessages());
 		if (action === "getInvocableSkills" && request.method === "GET") return Response.json(await this.getInvocableSkills(auth.userContext));
 		if (action === "sendMessage" && request.method === "POST") return this.streamMessage(request);
+		if (action === "submitToolApproval" && request.method === "POST") return this.streamToolApproval(request);
+		if (action === "getMessageFeedback" && request.method === "GET") return Response.json(await this.getMessageFeedback());
+		if (action === "submitMessageFeedback" && request.method === "POST") {
+			let payload;
+			try {
+				payload = await request.json();
+			} catch {
+				return Response.json({ error: "Request body must be valid JSON" }, { status: 400 });
+			}
+			if (!payload || typeof payload !== "object" || Array.isArray(payload)) return Response.json({ error: "Request body must be an object" }, { status: 400 });
+			const { messageId, rating, comment } = payload;
+			if (!messageId) return Response.json({ error: "Message ID is required" }, { status: 400 });
+			if (rating !== -1 && rating !== 0 && rating !== 1) return Response.json({ error: "Rating must be -1, 0, or 1" }, { status: 400 });
+			if (comment !== void 0 && typeof comment !== "string") return Response.json({ error: "Comment must be a string" }, { status: 400 });
+			return Response.json(await this.submitMessageFeedback(messageId, rating, comment));
+		}
 		if (action === "cancelChat" && request.method === "POST") {
 			const { requestId } = await request.json();
 			this.cancelChat(requestId);
@@ -2882,6 +2899,39 @@ var ChatAgent = class extends Agent {
 			this._turn.context = this.createToolContextForCurrentTurn(context);
 			return [message];
 		} : message;
+		return this.streamTurn(input);
+	}
+	async streamToolApproval(request) {
+		let payload;
+		try {
+			payload = await request.json();
+		} catch {
+			return Response.json({ error: "Request body must be valid JSON" }, { status: 400 });
+		}
+		if (!payload || typeof payload !== "object" || Array.isArray(payload)) return Response.json({ error: "Request body must be an object" }, { status: 400 });
+		const { toolCallId, approved, context } = payload;
+		if (!toolCallId) return Response.json({ error: "Tool call ID is required" }, { status: 400 });
+		if (typeof approved !== "boolean") return Response.json({ error: "Approved must be a boolean" }, { status: 400 });
+		if (context !== void 0 && (context === null || typeof context !== "object" || Array.isArray(context))) return Response.json({ error: "Context must be an object" }, { status: 400 });
+		return this.streamTurn((messages) => {
+			if (context) this._turn.context = this.createToolContextForCurrentTurn(context);
+			const update = toolApprovalUpdate(toolCallId, approved);
+			let found = false;
+			const updatedMessages = messages.map((message) => {
+				if (found || message.role !== "assistant") return message;
+				const result = applyToolUpdate(message.parts, update);
+				if (!result) return message;
+				found = true;
+				return {
+					...message,
+					parts: result.parts
+				};
+			});
+			if (!found) throw new Error(`No pending tool approval found for ${toolCallId}`);
+			return updatedMessages;
+		});
+	}
+	streamTurn(input) {
 		const encoder = new TextEncoder();
 		const readable = new ReadableStream({ start: async (controller) => {
 			const send = (event, data) => controller.enqueue(encoder.encode(`event: ${event}\ndata: ${data}\n\n`));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@economic/agents",
-  "version": "2.14.1",
+  "version": "2.14.2",
   "description": "A starter for creating a TypeScript package.",
   "license": "MIT",
   "files": [