@getpaseo/client 0.1.109 → 0.2.0-beta.1

package/dist/daemon-client.d.ts

@@ -1,7 +1,7 @@
 import type { z } from "zod";
 import { type ClientCapability } from "@getpaseo/protocol/client-capabilities";
 import { AgentRefreshedStatusPayloadSchema, CheckoutRenameBranchResponseSchema, RenameTerminalResponseSchema, RestartRequestedStatusPayloadSchema, ShutdownRequestedStatusPayloadSchema, DaemonUpdateResponseSchema, type ServerInfoStatusPayload } from "@getpaseo/protocol/messages";
-import type { AgentStreamEventPayload, AgentSnapshotPayload, ProjectPlacementPayload, AgentPermissionResolvedMessage, CreateAgentRequestMessage, CreatePaseoWorktreeRequest, FileDownloadTokenResponse, FileUploadResponse, FileExplorerResponse, FetchAgentTimelineResponseMessage, AgentForkContextResponseMessage, GitSetupOptions, CheckoutStatusResponse, CheckoutCommitResponse, CheckoutMergeResponse, CheckoutMergeFromBaseResponse, CheckoutPullResponse, CheckoutPushResponse, CheckoutRefreshResponse, CheckoutPrCreateResponse, CheckoutPrMergeResponse, CheckoutPrMergeMethod, CheckoutGithubSetAutoMergeResponse, CheckoutGithubGetCheckDetailsResponse, CheckoutPrStatusResponse, PullRequestTimelineResponse, CheckoutSwitchBranchResponse, StashSaveResponse, StashPopResponse, StashListResponse, ValidateBranchResponse, BranchSuggestionsResponse, GitHubSearchResponse, GitHubSearchRequest, DirectorySuggestionsResponse, PaseoWorktreeListResponse, PaseoWorktreeArchiveResponse, ProjectIconResponse, ProjectAddResponse, ProjectCreateDirectoryResponse, OpenProjectResponseMessage, WorkspaceGithubSearchRepositoriesResponse, ProjectGithubCloneProtocol, ProjectGithubCloneResponse, ArchiveWorkspaceResponseMessage, WorkspaceSetupStatusResponseMessage, ListCommandsResponse, ListProviderFeaturesResponseMessage, ListProviderModelsResponseMessage, ListProviderModesResponseMessage, ListAvailableProvidersResponse, GetProvidersSnapshotResponseMessage, RefreshProvidersSnapshotResponseMessage, ProviderDiagnosticResponseMessage, ProviderUsageListResponseMessage, DaemonGetStatusResponse, DaemonGetPairingOfferResponse, DiagnosticsResponse, AgentRewindResponseMessage, ListTerminalsResponse, CreateTerminalResponse, SubscribeTerminalResponse, SubscribeTerminalRequest, CloseItemsResponse, KillTerminalResponse, CaptureTerminalResponse, TerminalInput, SessionInboundMessage, SessionOutboundMessage, SendAgentMessageRequest, PaseoConfigRaw, PaseoConfigRevision, WorkspaceCreateRequest } from "@getpaseo/protocol/messages";
+import type { AgentStreamEventPayload, AgentSnapshotPayload, ProjectPlacementPayload, AgentPermissionResolvedMessage, CreateAgentRequestMessage, CreatePaseoWorktreeRequest, FileDownloadTokenResponse, FileUploadResponse, FileExplorerResponse, FetchAgentTimelineResponseMessage, AgentForkContextResponseMessage, GitSetupOptions, CheckoutStatusResponse, CheckoutCommit, ParsedDiffFile, CheckoutCommitResponse, CheckoutMergeResponse, CheckoutMergeFromBaseResponse, CheckoutPullResponse, CheckoutPushResponse, CheckoutRefreshResponse, CheckoutPrCreateResponse, CheckoutPrMergeResponse, CheckoutPrMergeMethod, CheckoutForgeSetAutoMergeResponse, CheckoutGithubSetAutoMergeResponse, CheckoutForgeGetCheckDetailsResponse, CheckoutGithubGetCheckDetailsResponse, CheckoutPrStatusResponse, PullRequestTimelineResponse, CheckoutSwitchBranchResponse, StashSaveResponse, StashPopResponse, StashListResponse, ValidateBranchResponse, BranchSuggestionsResponse, ForgeSearchResponse, ForgeSearchRequest, GitHubSearchResponse, GitHubSearchRequest, DirectorySuggestionsResponse, PaseoWorktreeListResponse, PaseoWorktreeArchiveResponse, ProjectIconResponse, ProjectAddResponse, ProjectCreateDirectoryResponse, OpenProjectResponseMessage, WorkspaceGithubSearchRepositoriesResponse, ProjectGithubCloneProtocol, ProjectGithubCloneResponse, ArchiveWorkspaceResponseMessage, WorkspaceSetupStatusResponseMessage, ListCommandsResponse, ListProviderFeaturesResponseMessage, ListProviderModelsResponseMessage, ListProviderModesResponseMessage, ListAvailableProvidersResponse, GetProvidersSnapshotResponseMessage, RefreshProvidersSnapshotResponseMessage, ProviderDiagnosticResponseMessage, ProviderUsageListResponseMessage, DaemonGetStatusResponse, DaemonGetPairingOfferResponse, DiagnosticsResponse, AgentRewindResponseMessage, ListTerminalsResponse, CreateTerminalResponse, SubscribeTerminalResponse, SubscribeTerminalRequest, CloseItemsResponse, KillTerminalResponse, CaptureTerminalResponse, TerminalInput, SessionInboundMessage, SessionOutboundMessage, SendAgentMessageRequest, PaseoConfigRaw, PaseoConfigRevision, WorkspaceCreateRequest, WorkspaceRecoveryState } from "@getpaseo/protocol/messages";
 import type { AgentPermissionRequest, AgentPermissionResponse, AgentPersistenceHandle, AgentProviderNotice, AgentProvider, AgentSessionConfig } from "@getpaseo/protocol/agent-types";
 import type { MutableDaemonConfig, MutableDaemonConfigPatch } from "@getpaseo/protocol/messages";
 import { type DaemonTransportFactory, type WebSocketFactory } from "./daemon-client-transport.js";
@@ -15,6 +15,7 @@ export interface Logger {
 }
 interface ImportAgentInputBase {
     cwd?: string;
+    workspaceId?: string;
     labels?: Record<string, string>;
 }
 export type ImportAgentInput = (ImportAgentInputBase & {
@@ -146,7 +147,7 @@ export interface CreateAgentRequestOptions extends AgentConfigOverrides {
     requestId?: string;
     labels?: Record<string, string>;
 }
-export interface CreatePaseoWorktreeInput extends Pick<CreatePaseoWorktreeRequest, "cwd" | "projectId" | "worktreeSlug" | "firstAgentContext" | "refName" | "action" | "githubPrNumber"> {
+export interface CreatePaseoWorktreeInput extends Pick<CreatePaseoWorktreeRequest, "cwd" | "projectId" | "worktreeSlug" | "firstAgentContext" | "refName" | "action" | "checkoutSource" | "githubPrNumber"> {
 }
 type CheckoutStatusPayload = CheckoutStatusResponse["payload"];
 type SubscribeCheckoutDiffPayload = Extract<SessionOutboundMessage, {
@@ -161,7 +162,9 @@ type CheckoutPushPayload = CheckoutPushResponse["payload"];
 type CheckoutRefreshPayload = CheckoutRefreshResponse["payload"];
 type CheckoutPrCreatePayload = CheckoutPrCreateResponse["payload"];
 type CheckoutPrMergePayload = CheckoutPrMergeResponse["payload"];
+type CheckoutForgeSetAutoMergePayload = CheckoutForgeSetAutoMergeResponse["payload"];
 type CheckoutGithubSetAutoMergePayload = CheckoutGithubSetAutoMergeResponse["payload"];
+type CheckoutForgeGetCheckDetailsPayload = CheckoutForgeGetCheckDetailsResponse["payload"];
 type CheckoutGithubGetCheckDetailsPayload = CheckoutGithubGetCheckDetailsResponse["payload"];
 type CheckoutPrStatusPayload = CheckoutPrStatusResponse["payload"];
 type PullRequestTimelinePayload = PullRequestTimelineResponse["payload"];
@@ -172,6 +175,7 @@ type StashPopPayload = StashPopResponse["payload"];
 type StashListPayload = StashListResponse["payload"];
 type ValidateBranchPayload = ValidateBranchResponse["payload"];
 type BranchSuggestionsPayload = BranchSuggestionsResponse["payload"];
+type ForgeSearchPayload = ForgeSearchResponse["payload"];
 type GitHubSearchPayload = GitHubSearchResponse["payload"];
 type DirectorySuggestionsPayload = DirectorySuggestionsResponse["payload"];
 type PaseoWorktreeListPayload = PaseoWorktreeListResponse["payload"];
@@ -733,6 +737,8 @@ export declare class DaemonClient {
     setWorkspacePinned(workspaceId: string, pinned: boolean, requestId?: string): Promise<{
         pinnedAt: string | null;
     }>;
+    inspectWorkspaceRecovery(workspaceId: string, requestId?: string): Promise<WorkspaceRecoveryState>;
+    restoreWorkspace(workspaceId: string, requestId?: string): Promise<void>;
     resumeAgent(handle: AgentPersistenceHandle, overrides?: Partial<AgentSessionConfig>): Promise<AgentSnapshotPayload>;
     importAgent(input: ImportAgentInput): Promise<AgentSnapshotPayload>;
     refreshAgent(agentId: string, requestId?: string): Promise<AgentRefreshedStatusPayload>;
@@ -799,6 +805,13 @@ export declare class DaemonClient {
     checkoutPull(cwd: string, requestId?: string): Promise<CheckoutPullPayload>;
     checkoutPush(cwd: string, requestId?: string): Promise<CheckoutPushPayload>;
     checkoutRefresh(cwd: string, requestId?: string): Promise<CheckoutRefreshPayload>;
+    listCheckoutCommits(cwd: string, requestId?: string): Promise<{
+        baseRef: string | null;
+        commits: CheckoutCommit[];
+    }>;
+    getCommitFileDiff(cwd: string, sha: string, path: string, requestId?: string): Promise<{
+        file: ParsedDiffFile | null;
+    }>;
     checkoutPrCreate(cwd: string, input: {
         title?: string;
         body?: string;
@@ -807,17 +820,31 @@ export declare class DaemonClient {
     checkoutPrMerge(cwd: string, input: {
         method: CheckoutPrMergeMethod;
     }, requestId?: string): Promise<CheckoutPrMergePayload>;
+    checkoutForgeSetAutoMerge(cwd: string, input: {
+        enabled: true;
+        method: CheckoutPrMergeMethod;
+    } | {
+        enabled: false;
+    }, requestId?: string): Promise<CheckoutForgeSetAutoMergePayload>;
     checkoutGithubSetAutoMerge(cwd: string, input: {
         enabled: true;
         method: CheckoutPrMergeMethod;
     } | {
         enabled: false;
     }, requestId?: string): Promise<CheckoutGithubSetAutoMergePayload>;
+    checkoutForgeGetCheckDetails(input: {
+        cwd: string;
+        repoOwner?: string;
+        repoName?: string;
+        checkRunId?: number;
+        workflowRunId?: number;
+        changeRequestNumber?: number;
+    }, requestId?: string): Promise<CheckoutForgeGetCheckDetailsPayload>;
     checkoutGithubGetCheckDetails(input: {
         cwd: string;
-        repoOwner: string;
-        repoName: string;
-        checkRunId: number;
+        repoOwner?: string;
+        repoName?: string;
+        checkRunId?: number;
         workflowRunId?: number;
     }, requestId?: string): Promise<CheckoutGithubGetCheckDetailsPayload>;
     checkoutPrStatus(cwd: string, requestId?: string): Promise<CheckoutPrStatusPayload>;
@@ -862,6 +889,12 @@ export declare class DaemonClient {
         query?: string;
         limit?: number;
     }, requestId?: string): Promise<BranchSuggestionsPayload>;
+    searchForge(options: {
+        cwd: string;
+        query: string;
+        limit?: number;
+        kinds?: ForgeSearchRequest["kinds"];
+    }, requestId?: string): Promise<ForgeSearchPayload>;
     searchGitHub(options: {
         cwd: string;
         query: string;
@@ -1015,6 +1048,7 @@ export declare class DaemonClient {
     private recordLivenessFailure;
     private handleSessionMessage;
     private resolveWaiters;
+    private rejectWaitersForRequestId;
     private clearWaiters;
     private toEvent;
     private waitForWithCancel;

package/dist/daemon-client.js

@@ -24,6 +24,34 @@ function normalizePassword(value) {
     }
     return value.length > 0 ? value : null;
 }
+function extractCorrelatedResponseIdentity(input) {
+    if (!input || typeof input !== "object") {
+        return null;
+    }
+    const envelope = input;
+    if (envelope.type !== "session" || !envelope.message || typeof envelope.message !== "object") {
+        return null;
+    }
+    const message = envelope.message;
+    if (typeof message.type !== "string" ||
+        !(message.type === "rpc_error" ||
+            message.type.endsWith("_response") ||
+            message.type.endsWith(".response") ||
+            message.type.endsWith("/response"))) {
+        return null;
+    }
+    if (!message.payload || typeof message.payload !== "object") {
+        return null;
+    }
+    const payload = message.payload;
+    if (typeof payload.requestId !== "string") {
+        return null;
+    }
+    return {
+        requestId: payload.requestId,
+        responseType: message.type,
+    };
+}
 // COMPAT(daemon-client-object-options): added in v0.1.102; remove after
 // 2026-12-29 once SDK callers have migrated to object parameters.
 function normalizeFetchAgentOptions(input, legacyOptions) {
@@ -58,6 +86,16 @@ class DaemonRpcError extends Error {
         this.code = params.code;
     }
 }
+class DaemonProtocolError extends Error {
+    constructor(identity) {
+        const responseLabel = identity.responseType ?? "unknown response";
+        super(`Response validation failed for ${responseLabel}`);
+        this.code = "invalid_response";
+        this.name = "DaemonProtocolError";
+        this.requestId = identity.requestId;
+        this.responseType = identity.responseType;
+    }
+}
 class PingTimeoutError extends Error {
     constructor(timeoutMs) {
         super(`Ping timed out (${timeoutMs}ms)`);
@@ -645,7 +683,7 @@ export class DaemonClient {
                 return null;
             }
             return { kind: "ok", value };
-        }, timeout, params.options);
+        }, timeout, { ...params.options, requestId: params.requestId });
         try {
             await this.sendSessionMessageOrThrow(params.message);
         }
@@ -1339,6 +1377,29 @@ export class DaemonClient {
         }
         return { pinnedAt: payload.pinnedAt };
     }
+    async inspectWorkspaceRecovery(workspaceId, requestId) {
+        const payload = await this.sendNamespacedCorrelatedSessionRequest({
+            requestId,
+            message: {
+                type: "workspace.recovery.inspect.request",
+                workspaceId,
+            },
+        });
+        return payload.state;
+    }
+    async restoreWorkspace(workspaceId, requestId) {
+        const payload = await this.sendNamespacedCorrelatedSessionRequest({
+            requestId,
+            message: {
+                type: "workspace.recovery.restore.request",
+                workspaceId,
+            },
+            timeout: 150000,
+        });
+        if (!payload.accepted) {
+            throw new Error(payload.error ?? "Workspace recovery was rejected by the host");
+        }
+    }
     async resumeAgent(handle, overrides) {
         const requestId = this.createRequestId();
         const message = SessionInboundMessageSchema.parse({
@@ -1373,6 +1434,7 @@ export class DaemonClient {
                 ? { providerId: input.providerId, providerHandleId: input.providerHandleId }
                 : { provider: input.provider, sessionId: input.sessionId }),
             ...(input.cwd ? { cwd: input.cwd } : {}),
+            ...(input.workspaceId ? { workspaceId: input.workspaceId } : {}),
             ...(input.labels && Object.keys(input.labels).length > 0 ? { labels: input.labels } : {}),
         });
         const status = await this.sendRequest({
@@ -2197,6 +2259,36 @@ export class DaemonClient {
             responseType: "checkout.refresh.response",
         });
     }
+    async listCheckoutCommits(cwd, requestId) {
+        const payload = await this.sendNamespacedCorrelatedSessionRequest({
+            requestId,
+            message: {
+                type: "checkout.commits.list.request",
+                cwd,
+            },
+            timeout: 60000,
+        });
+        if (payload.error) {
+            throw new Error(payload.error.message);
+        }
+        return { baseRef: payload.baseRef, commits: payload.commits };
+    }
+    async getCommitFileDiff(cwd, sha, path, requestId) {
+        const payload = await this.sendNamespacedCorrelatedSessionRequest({
+            requestId,
+            message: {
+                type: "checkout.commits.file_diff.request",
+                cwd,
+                sha,
+                path,
+            },
+            timeout: 60000,
+        });
+        if (payload.error) {
+            throw new Error(payload.error.message);
+        }
+        return { file: payload.file };
+    }
     async checkoutPrCreate(cwd, input, requestId) {
         return this.sendCorrelatedSessionRequest({
             requestId,
@@ -2221,6 +2313,18 @@ export class DaemonClient {
             responseType: "checkout_pr_merge_response",
         });
     }
+    async checkoutForgeSetAutoMerge(cwd, input, requestId) {
+        return this.sendNamespacedCorrelatedSessionRequest({
+            requestId,
+            message: {
+                type: "checkout.forge.set_auto_merge.request",
+                cwd,
+                enabled: input.enabled,
+                ...(input.enabled ? { mergeMethod: input.method } : {}),
+            },
+            timeout: 60000,
+        });
+    }
     async checkoutGithubSetAutoMerge(cwd, input, requestId) {
         return this.sendNamespacedCorrelatedSessionRequest({
             requestId,
@@ -2232,6 +2336,21 @@ export class DaemonClient {
             },
         });
     }
+    async checkoutForgeGetCheckDetails(input, requestId) {
+        return this.sendNamespacedCorrelatedSessionRequest({
+            requestId,
+            message: {
+                type: "checkout.forge.get_check_details.request",
+                cwd: input.cwd,
+                repoOwner: input.repoOwner,
+                repoName: input.repoName,
+                checkRunId: input.checkRunId,
+                workflowRunId: input.workflowRunId,
+                changeRequestNumber: input.changeRequestNumber,
+            },
+            timeout: 60000,
+        });
+    }
     async checkoutGithubGetCheckDetails(input, requestId) {
         return this.sendNamespacedCorrelatedSessionRequest({
             requestId,
@@ -2361,6 +2480,7 @@ export class DaemonClient {
                     : {}),
                 ...(input.refName !== undefined ? { refName: input.refName } : {}),
                 ...(input.action !== undefined ? { action: input.action } : {}),
+                ...(input.checkoutSource !== undefined ? { checkoutSource: input.checkoutSource } : {}),
                 ...(input.githubPrNumber !== undefined ? { githubPrNumber: input.githubPrNumber } : {}),
             },
             responseType: "create_paseo_worktree_response",
@@ -2403,6 +2523,20 @@ export class DaemonClient {
             responseType: "branch_suggestions_response",
         });
     }
+    async searchForge(options, requestId) {
+        return this.sendCorrelatedSessionRequest({
+            requestId,
+            message: {
+                type: "forge.search.request",
+                cwd: options.cwd,
+                query: options.query,
+                limit: options.limit,
+                kinds: options.kinds,
+            },
+            responseType: "forge.search.response",
+            timeout: 15000,
+        });
+    }
     async searchGitHub(options, requestId) {
         return this.sendCorrelatedSessionRequest({
             requestId,
@@ -3427,13 +3561,18 @@ export class DaemonClient {
         }
         const parsed = validateWSOutboundMessage(parsedJson);
         if (!parsed.success) {
-            const msgType = parsedJson != null &&
+            const responseIdentity = extractCorrelatedResponseIdentity(parsedJson);
+            const envelopeType = parsedJson != null &&
                 typeof parsedJson === "object" &&
                 "type" in parsedJson &&
                 typeof parsedJson.type === "string"
                 ? parsedJson.type
                 : "unknown";
+            const msgType = responseIdentity?.responseType ?? envelopeType;
             this.logger.warn({ msgType, error: parsed.error.message }, "Message validation failed");
+            if (responseIdentity) {
+                this.rejectWaitersForRequestId(responseIdentity.requestId, new DaemonProtocolError(responseIdentity));
+            }
             return;
         }
         this.consecutiveLivenessFailures = 0;
@@ -3707,6 +3846,18 @@ export class DaemonClient {
             }
         }
     }
+    rejectWaitersForRequestId(requestId, error) {
+        for (const waiter of Array.from(this.waiters)) {
+            if (waiter.requestId !== requestId) {
+                continue;
+            }
+            this.waiters.delete(waiter);
+            if (waiter.timeoutHandle) {
+                clearTimeout(waiter.timeoutHandle);
+            }
+            waiter.reject(error);
+        }
+    }
     clearWaiters(error) {
         for (const waiter of Array.from(this.waiters)) {
             if (waiter.timeoutHandle) {
@@ -3771,7 +3922,7 @@ export class DaemonClient {
                 return null;
         }
     }
-    waitForWithCancel(predicate, timeout = 30000, _options) {
+    waitForWithCancel(predicate, timeout = 30000, options) {
         // Capture stack trace at call site, not inside setTimeout
         const timeoutError = new Error(`Timeout waiting for message (${timeout}ms)`);
         let waiter = null;
@@ -3804,6 +3955,7 @@ export class DaemonClient {
                 resolve: wrappedResolve,
                 reject: wrappedReject,
                 timeoutHandle,
+                requestId: options?.requestId,
             };
             this.waiters.add(waiter);
         });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@getpaseo/client",
-  "version": "0.1.109",
+  "version": "0.2.0-beta.1",
   "description": "Paseo client SDK package",
   "files": [
     "dist",
@@ -35,8 +35,8 @@
     "test": "vitest run"
   },
   "dependencies": {
-    "@getpaseo/protocol": "0.1.109",
-    "@getpaseo/relay": "0.1.109",
+    "@getpaseo/protocol": "0.2.0-beta.1",
+    "@getpaseo/relay": "0.2.0-beta.1",
     "zod": "^4.4.3"
   },
   "devDependencies": {