letagents 0.12.12 → 0.12.13

package/dist/mcp/server/tools/messages/read-tool.js

@@ -1,30 +1,15 @@
 import { z } from "zod";
 import { encodeRoomIdPath } from "../../../room-id.js";
-import { appendIncludePromptOnly, currentRoom, ensureAgentIdentity, getFallbackProjectId, getLatestLocalChatMessages, getLocalChatMessages, getTargetRoomId, heartbeatRoomPresence, touchRoomSession, isLocalRoomStorageEnabled, resolveLocalRoomStorageIdentifiers, roomScopedApiCall, toAgentReadableMessages, } from "../../runtime.js";
+import { AGENT_MESSAGE_BODY_MAX_BYTES, appendIncludePromptOnly, boundAgentMessageOutput, buildAgentDeliveryHeaders, currentRoom, ensureAgentIdentity, getFallbackProjectId, getLatestLocalChatMessages, getCurrentAgentSessionSnapshot, getTargetRoomId, heartbeatRoomPresence, touchRoomSession, isLocalRoomStorageEnabled, resolveLocalRoomStorageIdentifiers, roomScopedApiCall, toAgentReadableMessages, } from "../../runtime.js";
+import { requireValidWorkerBearerRuntime } from "../../runtime/worker-bearer.js";
 import { jsonToolResponse } from "./response.js";
 export const DEFAULT_READ_MESSAGES_LIMIT = 100;
 // Both the API and the local store clamp a single page to 500 messages.
-const MAX_MESSAGES_PER_PAGE = 500;
-export function selectRecentMessages(messages, limit) {
-    const total = messages.length;
-    const selected = limit > 0 && total > limit ? messages.slice(total - limit) : messages;
-    return {
-        messages: selected,
-        total_message_count: total,
-        omitted_message_count: total - selected.length,
-    };
-}
-function withRecencyTelemetry(output, selection) {
-    output.total_message_count = selection.total_message_count;
-    if (selection.omitted_message_count > 0) {
-        output.omitted_message_count = selection.omitted_message_count;
-        output.truncated = true;
-    }
-    return output;
-}
+export const MAX_MESSAGES_PER_PAGE = 500;
 // Fetch the most recent messages by paging BACKWARDS from the tail
 // (before=latest), so a limited read never walks the full room history.
 export async function fetchRecentRemoteMessages(input) {
+    const boundedLimit = Math.max(1, Math.min(MAX_MESSAGES_PER_PAGE, Math.floor(input.limit)));
     const collected = [];
     let beforeCursor = "latest";
     let truncated = false;
@@ -33,7 +18,7 @@ export async function fetchRecentRemoteMessages(input) {
     for (;;) {
         const params = new URLSearchParams();
         params.set("before", beforeCursor);
-        params.set("limit", String(Math.min(input.limit - collected.length, MAX_MESSAGES_PER_PAGE)));
+        params.set("limit", String(Math.min(boundedLimit - collected.length, MAX_MESSAGES_PER_PAGE)));
         const qs = params.toString();
         const result = await roomScopedApiCall({
             room_id: input.targetRoomId,
@@ -44,6 +29,7 @@ export async function fetchRecentRemoteMessages(input) {
             // page touch the session would walk last_message_id backwards and make
             // resume_room_session replay already-read history.
             preserve_session_cursor: true,
+            options: { headers: input.deliveryHeaders },
         });
         roomIdFromResponse = roomIdFromResponse || result.room_id || result.project_id;
         const msgs = result.messages ?? [];
@@ -56,7 +42,7 @@ export async function fetchRecentRemoteMessages(input) {
         const hasOlder = Boolean(result.has_older ?? result.has_more);
         if (!hasOlder || msgs.length === 0)
             break;
-        if (collected.length >= input.limit) {
+        if (collected.length >= boundedLimit) {
             truncated = true;
             break;
         }
@@ -73,103 +59,74 @@ export async function fetchRecentRemoteMessages(input) {
     return { messages: collected, truncated, roomIdFromResponse };
 }
 export function registerReadMessagesTool(server) {
-    server.tool("read_messages", "Read recent messages from a Let Agents Chat room (most recent `limit`, default 100; pass limit: 0 for the full history — expensive in busy rooms). Threaded replies include thread_parent_id/thread.root_message_id; use send_thread_message with that id to continue focused side discussion without polluting the main room. For long-running work, prefer wait_for_messages with after_message_id so you only process new lines and do not treat an empty poll as the end of the mission.", {
+    server.tool("read_messages", "Read a bounded recent page from a Let Agents Chat room (most recent `limit`, default 100, maximum 500). Threaded replies include thread_parent_id/thread.root_message_id; use send_thread_message with that id to continue focused side discussion without polluting the main room. For long-running work, prefer wait_for_messages with after_message_id so you only process new lines and do not treat an empty poll as the end of the mission.", {
         room_id: z.string().optional().describe("Canonical room ID. Defaults to the current room."),
         limit: z
             .number()
             .int()
             .min(0)
             .optional()
-            .describe(`Return only the most recent N messages (default ${DEFAULT_READ_MESSAGES_LIMIT}). Pass 0 for the full history. When older messages are omitted, the response carries truncated=true.`),
+            .describe(`Return the most recent N messages (default ${DEFAULT_READ_MESSAGES_LIMIT}, maximum ${MAX_MESSAGES_PER_PAGE}). Zero is retained for compatibility and means the bounded maximum. When older messages are omitted, the response carries truncated=true.`),
     }, async ({ room_id, limit }) => {
-        const effectiveLimit = limit ?? DEFAULT_READ_MESSAGES_LIMIT;
+        const requestedLimit = limit ?? DEFAULT_READ_MESSAGES_LIMIT;
+        const effectiveLimit = requestedLimit > 0
+            ? Math.min(requestedLimit, MAX_MESSAGES_PER_PAGE)
+            : MAX_MESSAGES_PER_PAGE;
         const targetRoomId = getTargetRoomId(room_id);
         const targetProjectId = getFallbackProjectId();
         const localRoomId = targetRoomId ?? currentRoom?.room_id ?? targetProjectId;
+        const sessionRoomId = targetRoomId ?? currentRoom?.room_id ?? null;
+        const workerRuntime = requireValidWorkerBearerRuntime();
+        const agentSessionSnapshot = workerRuntime.mode === "owner"
+            ? getCurrentAgentSessionSnapshot(sessionRoomId)
+            : { complete: true, session: null };
+        const agentSession = agentSessionSnapshot.session;
+        const deliveryHeaders = buildAgentDeliveryHeaders(agentSession);
+        if (!agentSessionSnapshot.complete) {
+            throw new Error("Local agent routing state is unavailable; retry after restoring the state file.");
+        }
         if (localRoomId && await isLocalRoomStorageEnabled(localRoomId)) {
-            const { localRoomId: sqliteRoomId } = await resolveLocalRoomStorageIdentifiers(localRoomId);
+            const { localRoomId: sqliteRoomId, cloudRoomId, } = await resolveLocalRoomStorageIdentifiers(localRoomId);
             const effectiveLocalRoomId = sqliteRoomId || localRoomId;
-            if (effectiveLimit > 0 && effectiveLimit <= MAX_MESSAGES_PER_PAGE) {
-                const result = await getLatestLocalChatMessages(effectiveLocalRoomId, {
-                    limit: effectiveLimit,
-                    include_prompt_only: true,
-                });
-                const output = {
-                    room_id: effectiveLocalRoomId,
-                    messages: toAgentReadableMessages(result.messages ?? []),
-                };
-                if (result.has_more)
-                    output.truncated = true;
-                return jsonToolResponse(output);
-            }
-            const allMessages = [];
-            let afterCursor;
-            for (;;) {
-                const result = await getLocalChatMessages(effectiveLocalRoomId, {
-                    after: afterCursor,
-                    include_prompt_only: true,
-                });
-                const msgs = result.messages ?? [];
-                allMessages.push(...msgs);
-                if (!result.has_more || msgs.length === 0)
-                    break;
-                const lastMsg = msgs[msgs.length - 1];
-                if (!lastMsg?.id)
-                    break;
-                afterCursor = lastMsg.id;
-            }
-            const selection = selectRecentMessages(allMessages, effectiveLimit);
-            return jsonToolResponse(withRecencyTelemetry({
-                room_id: effectiveLocalRoomId,
-                messages: toAgentReadableMessages(selection.messages),
-            }, selection));
-        }
-        if (effectiveLimit > 0) {
-            const recent = await fetchRecentRemoteMessages({
-                targetRoomId,
-                targetProjectId,
+            const result = await getLatestLocalChatMessages(effectiveLocalRoomId, {
                 limit: effectiveLimit,
+                include_prompt_only: true,
+            });
+            const { attachLocalActivationMetadata } = await import("./wait-tool.js");
+            const messages = await attachLocalActivationMetadata(effectiveLocalRoomId, result.messages ?? [], agentSession, {
+                includeTaskOwnerLeases: false,
+                activeSessionRoomId: cloudRoomId || sessionRoomId,
             });
-            await heartbeatRoomPresence(targetRoomId ?? currentRoom?.room_id ?? null, await ensureAgentIdentity());
+            const bounded = boundAgentMessageOutput(toAgentReadableMessages(messages), { direction: "suffix", maxBytes: AGENT_MESSAGE_BODY_MAX_BYTES });
             const output = {
-                messages: toAgentReadableMessages(recent.messages),
+                room_id: effectiveLocalRoomId,
+                messages: bounded.messages,
             };
-            if (recent.truncated)
+            if (result.has_more || bounded.truncated)
                 output.truncated = true;
-            if (recent.roomIdFromResponse) {
-                output[targetRoomId ? "room_id" : "project_id"] = recent.roomIdFromResponse;
+            if (bounded.omittedMessageCount > 0) {
+                output.omitted_message_count = bounded.omittedMessageCount;
             }
             return jsonToolResponse(output);
         }
-        const allMessages = [];
-        let afterCursor;
-        let roomIdFromResponse;
-        for (;;) {
-            const query = new URLSearchParams();
-            if (afterCursor)
-                query.set("after", afterCursor);
-            const qs = query.toString();
-            const result = await roomScopedApiCall({
-                room_id: targetRoomId,
-                project_id: targetProjectId,
-                room_path: (targetRoomId) => appendIncludePromptOnly(`/rooms/${encodeRoomIdPath(targetRoomId)}/messages${qs ? `?${qs}` : ""}`),
-                project_path: (targetProjectId) => appendIncludePromptOnly(`/projects/${encodeURIComponent(targetProjectId)}/messages${qs ? `?${qs}` : ""}`),
-            });
-            roomIdFromResponse = roomIdFromResponse || result.room_id || result.project_id;
-            const msgs = result.messages ?? [];
-            allMessages.push(...msgs);
-            if (!result.has_more || msgs.length === 0)
-                break;
-            const lastMsg = msgs[msgs.length - 1];
-            if (!lastMsg?.id)
-                break;
-            afterCursor = lastMsg.id;
-        }
+        const recent = await fetchRecentRemoteMessages({
+            targetRoomId,
+            targetProjectId,
+            limit: effectiveLimit,
+            deliveryHeaders,
+        });
         await heartbeatRoomPresence(targetRoomId ?? currentRoom?.room_id ?? null, await ensureAgentIdentity());
-        const selection = selectRecentMessages(allMessages, effectiveLimit);
-        const output = withRecencyTelemetry({ messages: toAgentReadableMessages(selection.messages) }, selection);
-        if (roomIdFromResponse) {
-            output[targetRoomId ? "room_id" : "project_id"] = roomIdFromResponse;
+        const bounded = boundAgentMessageOutput(toAgentReadableMessages(recent.messages), { direction: "suffix", maxBytes: AGENT_MESSAGE_BODY_MAX_BYTES });
+        const output = {
+            messages: bounded.messages,
+        };
+        if (recent.truncated || bounded.truncated)
+            output.truncated = true;
+        if (bounded.omittedMessageCount > 0) {
+            output.omitted_message_count = bounded.omittedMessageCount;
+        }
+        if (recent.roomIdFromResponse) {
+            output[targetRoomId ? "room_id" : "project_id"] = recent.roomIdFromResponse;
         }
         return jsonToolResponse(output);
     });

package/dist/mcp/server/tools/messages/reasoning-tool.js

@@ -61,6 +61,8 @@ export function registerPostReasoningTool(server) {
                     sender: identity.actor_label,
                     text: normalizedMilestone,
                     source: "agent",
+                    publisher_agent_key: agentSession?.agent_key ?? null,
+                    publisher_agent_session_id: agentSession?.session_id ?? null,
                 });
                 milestoneMessageId = milestoneMessage.id;
                 touchCurrentRoom(milestoneMessageId);

package/dist/mcp/server/tools/messages/send-tool.js

@@ -87,6 +87,8 @@ async function sendMessageFromTool(input) {
             reply_to: replyTarget,
             ...(resolvedThreadRoot ? { thread_root_id: resolvedThreadRoot } : {}),
             source: "agent",
+            publisher_agent_key: agentSession?.agent_key ?? null,
+            publisher_agent_session_id: agentSession?.session_id ?? null,
         });
         touchCurrentRoom(message.id);
         return jsonToolResponse({

package/dist/mcp/server/tools/messages/status-tool.js

@@ -45,6 +45,8 @@ export function registerPostStatusTool(server) {
                 sender,
                 text: statusText,
                 source: "agent",
+                publisher_agent_key: agentSession?.agent_key ?? null,
+                publisher_agent_session_id: agentSession?.session_id ?? null,
             });
             touchCurrentRoom(message.id);
             return jsonToolResponse({