@adminforth/agent 2.2.5 → 2.3.0

package/dist/transport/http/context.d.ts

@@ -1,7 +1,8 @@
 import type { ChatSurfaceAdapter, ChatSurfaceEventSink, ChatSurfaceIncomingMessage, IAdminForth } from "adminforth";
-import type { HandleSpeechTurnInput, HandleTurnInput, RunAndPersistAgentResponseInput, RunAndPersistAgentResponseResult } from "../../domain/turnTypes.js";
+import type { HandleEditTurnInput, HandleSpeechTurnInput, HandleTurnInput, RunAndPersistAgentResponseInput, RunAndPersistAgentResponseResult } from "../../domain/turnTypes.js";
 import type { PluginOptions } from "../../types.js";
 export type SessionTurn = {
+    id: string;
     prompt: string;
     response: string;
 };
@@ -9,6 +10,7 @@ export type AgentEndpointsContext = {
     adminforth: IAdminForth;
     options: PluginOptions;
     handleTurn(input: HandleTurnInput): Promise<RunAndPersistAgentResponseResult>;
+    handleEditTurn(input: HandleEditTurnInput): Promise<RunAndPersistAgentResponseResult>;
     handleSpeechTurn(input: HandleSpeechTurnInput): Promise<RunAndPersistAgentResponseResult | null>;
     steer(input: {
         sessionId: string;
@@ -24,6 +26,6 @@ export type AgentEndpointsContext = {
     appendSteerToCurrentTurn(sessionId: string, steerText: string): Promise<void>;
     handleChatSurfaceMessage(adapter: ChatSurfaceAdapter, incoming: ChatSurfaceIncomingMessage, sink: ChatSurfaceEventSink): Promise<void>;
 };
-export type CoreEndpointsContext = Pick<AgentEndpointsContext, "options" | "handleTurn" | "handleSpeechTurn" | "steer">;
+export type CoreEndpointsContext = Pick<AgentEndpointsContext, "options" | "handleTurn" | "handleEditTurn" | "handleSpeechTurn" | "steer">;
 export type SessionEndpointsContext = Pick<AgentEndpointsContext, "adminforth" | "options" | "getSessionTurns" | "createNewTurn" | "createSystemTurn" | "appendSteerToCurrentTurn">;
 export type ChatSurfaceEndpointsContext = Pick<AgentEndpointsContext, "adminforth" | "options" | "handleChatSurfaceMessage">;

package/dist/transport/http/coreEndpoints.js

@@ -30,6 +30,14 @@ const agentSteerBodySchema = z.object({
     sessionId: z.string(),
     message: z.string().min(1),
 }).strict();
+const agentEditBodySchema = z.object({
+    sessionId: z.string(),
+    turnId: z.string(),
+    message: z.string().min(1),
+    mode: z.string().nullish(),
+    timeZone: z.string().optional(),
+    currentPage: currentPageContextSchema.optional(),
+}).strict();
 // Sent as multipart/form-data (via multer), so every field arrives as a plain string on the wire.
 // `currentPage` is JSON.stringify'd by the client into a form field and must be parsed back into
 // an object. Zod -> JSON-Schema conversion cannot represent `.transform()` at all (it throws), so
@@ -103,6 +111,33 @@ export function setupCoreEndpoints(ctx, server) {
             return null;
         })
     });
+    server.endpoint({
+        method: 'POST',
+        path: `/agent/edit`,
+        request_schema: agentEditBodySchema,
+        handler: (_a) => __awaiter(this, [_a], void 0, function* ({ body, adminUser, _raw_express_res, abortSignal }) {
+            var _b;
+            const data = body;
+            const emit = createSseEventEmitter(_raw_express_res, {
+                vercelAiUiMessageStream: true,
+                closeActiveBlockOnToolStart: true,
+            });
+            yield ctx.handleEditTurn({
+                prompt: data.message,
+                turnId: data.turnId,
+                sessionId: data.sessionId,
+                modeName: data.mode,
+                userTimeZone: (_b = data.timeZone) !== null && _b !== void 0 ? _b : 'UTC',
+                currentPage: data.currentPage,
+                abortSignal,
+                adminUser: adminUser,
+                emit,
+                failureLogMessage: "Agent message edit streaming failed",
+                abortLogMessage: "Agent message edit streaming aborted by the client",
+            });
+            return null;
+        })
+    });
     server.endpoint({
         method: 'POST',
         path: `/agent/approval`,

package/dist/transport/http/sessionEndpoints.js

@@ -81,6 +81,7 @@ export function setupSessionEndpoints(ctx, server) {
                             messages.push({
                                 text: turn.response,
                                 role: 'system',
+                                turnId: turn.id,
                             });
                             return messages;
                         }
@@ -88,12 +89,14 @@ export function setupSessionEndpoints(ctx, server) {
                             messages.push({
                                 text: turn.prompt,
                                 role: 'user',
+                                turnId: turn.id,
                             });
                         }
                         if (turn.response && turn.response !== "not_finished") {
                             messages.push({
                                 text: turn.response,
                                 role: 'assistant',
+                                turnId: turn.id,
                             });
                         }
                         return messages;

package/dist/transport/sse/sseWriter.js

@@ -124,6 +124,15 @@ function createAgentEventStream(res, options = {}) {
                 },
             });
         },
+        turnPersisted(sessionId, turnId) {
+            stream.send({
+                type: isAiUiMessageStream ? "data-turn-persisted" : "turn-persisted",
+                data: {
+                    sessionId,
+                    turnId,
+                },
+            });
+        },
         response(text, sessionId, turnId) {
             stream.send({
                 type: isAiUiMessageStream ? "data-response" : "response",
@@ -240,6 +249,9 @@ export function createSseEventEmitter(res, options = {}) {
             case "transcript":
                 stream.transcript(event.text, event.language);
                 break;
+            case "turn-persisted":
+                stream.turnPersisted(event.sessionId, event.turnId);
+                break;
             case "response":
                 stream.response(event.text, event.sessionId, event.turnId);
                 break;

package/dist/types.d.ts

@@ -21,6 +21,7 @@ interface ITurnResource {
     promptField: string;
     responseField: string;
     debugField?: string;
+    checkpointIdField?: string;
 }
 interface ICheckpointResource {
     resourceId: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@adminforth/agent",
-  "version": "2.2.5",
+  "version": "2.3.0",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "type": "module",