@getpaseo/server 0.1.81 → 0.1.83

package/dist/server/server/session.js

@@ -23,7 +23,7 @@ import { DictationStreamManager, } from "./dictation/dictation-stream-manager.js
 import { createVoiceTurnController, } from "./voice/voice-turn-controller.js";
 import { buildConfigOverrides, extractTimestamps, isStoredAgentProviderAvailable, toAgentPersistenceHandle, } from "./persistence-hooks.js";
 import { ensureAgentLoaded } from "./agent/agent-loading.js";
-import { formatSystemNotificationPrompt, sendPromptToAgent, unarchiveAgentState, } from "./agent/agent-prompt.js";
+import { formatSystemNotificationPrompt, sendPromptToAgent, startCreatedAgentInitialPrompt, waitForAgentRunStartWithTimeout, unarchiveAgentState, } from "./agent/agent-prompt.js";
 import { experimental_createMCPClient } from "ai";
 import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
 import { buildWorkspaceScriptPayloads, readPaseoConfigForProjection, } from "./script-status-projection.js";
@@ -616,10 +616,11 @@ export class Session {
             const handleProviderSnapshotChange = (entries, cwd) => {
                 // COMPAT(providersSnapshot): keep provider visibility gating for older clients.
                 const visibleEntries = entries.filter((entry) => this.isProviderVisibleToClient(entry.provider));
+                const snapshotCwd = cwd === resolveSnapshotCwd() ? undefined : cwd;
                 this.emit({
                     type: "providers_snapshot_update",
                     payload: {
-                        cwd,
+                        ...(snapshotCwd ? { cwd: snapshotCwd } : {}),
                         entries: visibleEntries,
                         generatedAt: new Date().toISOString(),
                     },
@@ -757,7 +758,7 @@ export class Session {
     }
     async buildAgentPayload(agent) {
         const storedRecord = await this.agentStorage.get(agent.id);
-        const title = storedRecord?.title ?? storedRecord?.config?.title ?? null;
+        const title = storedRecord?.title ?? null;
         const payload = toAgentPayload(agent, { title });
         const storedUpdatedAt = storedRecord ? resolveStoredAgentPayloadUpdatedAt(storedRecord) : null;
         if (storedUpdatedAt) {
@@ -1038,6 +1039,7 @@ export class Session {
     }
     async dispatchInboundMessage(msg) {
         const promise = this.dispatchVoiceAndControlMessage(msg) ??
+            this.dispatchAgentRewindMessage(msg) ??
             this.dispatchAgentLifecycleMessage(msg) ??
             this.dispatchAgentConfigMessage(msg) ??
             this.dispatchCheckoutMessage(msg) ??
@@ -1098,6 +1100,14 @@ export class Session {
                 return undefined;
         }
     }
+    dispatchAgentRewindMessage(msg) {
+        switch (msg.type) {
+            case "agent.rewind.request":
+                return this.handleAgentRewindRequest(msg);
+            default:
+                return undefined;
+        }
+    }
     async handleDictationStreamStart(msg) {
         const unavailable = this.resolveVoiceFeatureUnavailableContext("dictation");
         if (unavailable) {
@@ -2095,7 +2105,6 @@ export class Session {
                 agentManager: this.agentManager,
                 agentStorage: this.agentStorage,
                 agentId,
-                userMessageText: text,
                 prompt,
                 messageId,
                 runOptions,
@@ -2125,10 +2134,7 @@ export class Session {
                 configTitle: config.title,
                 initialPrompt: trimmedPrompt,
             });
-            const resolvedConfig = {
-                ...config,
-                ...(provisionalTitle ? { title: provisionalTitle } : {}),
-            };
+            const resolvedConfig = config;
             const firstAgentContext = {
                 ...(trimmedPrompt ? { prompt: trimmedPrompt } : {}),
                 ...(attachments && attachments.length > 0 ? { attachments } : {}),
@@ -2156,6 +2162,7 @@ export class Session {
                 workspaceId: resolvedWorkspace.workspaceId,
                 initialPrompt: trimmedPrompt,
                 env,
+                initialTitle: provisionalTitle,
             });
             createdAgentId = snapshot.id;
             await this.forwardAgentUpdate(snapshot);
@@ -2164,7 +2171,7 @@ export class Session {
                 agentId: snapshot.id,
                 createdWorktree,
             });
-            await this.sendInitialCreateAgentPrompt({
+            const liveSnapshot = await this.sendInitialCreateAgentPrompt({
                 snapshot,
                 trimmedPrompt,
                 images,
@@ -2174,12 +2181,12 @@ export class Session {
                 explicitTitle,
             });
             if (requestId) {
-                const agentPayload = await this.buildAgentPayload(snapshot);
+                const agentPayload = await this.buildAgentPayload(liveSnapshot);
                 this.emit({
                     type: "status",
                     payload: {
                         status: "agent_created",
-                        agentId: snapshot.id,
+                        agentId: liveSnapshot.id,
                         requestId,
                         agent: agentPayload,
                     },
@@ -2225,7 +2232,7 @@ export class Session {
         const hasImages = (images?.length ?? 0) > 0;
         const hasAttachments = (attachments?.length ?? 0) > 0;
         if (!hasPrompt && !hasImages && !hasAttachments) {
-            return;
+            return snapshot;
         }
         scheduleAgentMetadataGeneration({
             agentManager: this.agentManager,
@@ -2237,10 +2244,17 @@ export class Session {
             paseoHome: this.paseoHome,
             logger: this.sessionLogger,
         });
-        const started = await this.handleSendAgentMessage(snapshot.id, trimmedPrompt || "", resolveClientMessageId(clientMessageId), images, attachments, outputSchema ? { outputSchema } : undefined);
-        if (!started.ok) {
-            throw new Error(started.error);
-        }
+        const prompt = this.buildAgentPrompt(trimmedPrompt || "", images, attachments);
+        return await startCreatedAgentInitialPrompt({
+            agentManager: this.agentManager,
+            agentId: snapshot.id,
+            snapshot,
+            prompt,
+            runOptions: outputSchema ? { outputSchema } : undefined,
+            logger: this.sessionLogger.child({
+                clientMessageId: resolveClientMessageId(clientMessageId),
+            }),
+        });
     }
     async handleResumeAgentRequest(msg) {
         const { handle, overrides, requestId } = msg;
@@ -2464,6 +2478,31 @@ export class Session {
             this.handleAgentRunError(agentId, error, "Failed to cancel running agent on request");
         }
     }
+    async handleAgentRewindRequest(msg) {
+        try {
+            await this.agentManager.rewind(msg.agentId, msg.messageId, msg.mode);
+            this.emit({
+                type: "agent.rewind.response",
+                payload: {
+                    requestId: msg.requestId,
+                    agentId: msg.agentId,
+                    ok: true,
+                    error: null,
+                },
+            });
+        }
+        catch (error) {
+            this.emit({
+                type: "agent.rewind.response",
+                payload: {
+                    requestId: msg.requestId,
+                    agentId: msg.agentId,
+                    ok: false,
+                    error: error instanceof Error ? error.message : "Failed to rewind agent",
+                },
+            });
+        }
+    }
     async buildAgentSessionConfig(config, gitOptions, legacyWorktreeName, firstAgentContext) {
         return buildWorktreeAgentSessionConfig({
             paseoHome: this.paseoHome,
@@ -5917,7 +5956,6 @@ export class Session {
                     agentManager: this.agentManager,
                     agentStorage: this.agentStorage,
                     agentId,
-                    userMessageText: msg.text,
                     prompt,
                     messageId: msg.messageId,
                     logger: this.sessionLogger,
@@ -5949,11 +5987,8 @@ export class Session {
                 });
                 return;
             }
-            const startAbort = new AbortController();
-            const startTimeoutMs = 15000;
-            const startTimeout = setTimeout(() => startAbort.abort("timeout"), startTimeoutMs);
             try {
-                await this.agentManager.waitForAgentRunStart(agentId, { signal: startAbort.signal });
+                await waitForAgentRunStartWithTimeout(this.agentManager, agentId);
             }
             catch (error) {
                 this.emit({
@@ -5967,9 +6002,6 @@ export class Session {
                 });
                 return;
             }
-            finally {
-                clearTimeout(startTimeout);
-            }
             this.emit({
                 type: "send_agent_message_response",
                 payload: {
@@ -6767,7 +6799,6 @@ export class Session {
                         agentId,
                         prompt: formatSystemNotificationPrompt(text),
                         unarchive: false,
-                        recordUserMessage: false,
                         logger: this.sessionLogger,
                     });
                 },