@treeseed/sdk 0.11.0 → 0.12.4

package/dist/api/types.d.ts

@@ -1,6 +1,6 @@
 import type { Hono } from 'hono';
-import type { AgentSdk } from '../sdk.ts';
-import type { ApiPrincipal, ApiScope, DeviceCodeApproveRequest as SdkDeviceCodeApproveRequest, DeviceCodePollRequest, DeviceCodePollResponse, DeviceCodeStartRequest, DeviceCodeStartResponse, RemoteWorkflowOperationRequest as WorkflowHttpOperationRequest, RemoteWorkflowOperationResponse as ApiWorkflowOperationResponse, RemoteSdkOperationRequest as SdkHttpOperationRequest, TokenRefreshRequest, TokenRefreshResponse } from '../remote.ts';
+import type { AgentSdk } from '../sdk.js';
+import type { ApiPrincipal, ApiScope, DeviceCodeApproveRequest as SdkDeviceCodeApproveRequest, DeviceCodePollRequest, DeviceCodePollResponse, DeviceCodeStartRequest, DeviceCodeStartResponse, RemoteWorkflowOperationRequest as WorkflowHttpOperationRequest, RemoteWorkflowOperationResponse as ApiWorkflowOperationResponse, RemoteSdkOperationRequest as SdkHttpOperationRequest, TokenRefreshRequest, TokenRefreshResponse } from '../remote.js';
 export type { ApiPrincipal, ApiScope, DeviceCodePollRequest, DeviceCodePollResponse, DeviceCodeStartRequest, DeviceCodeStartResponse, WorkflowHttpOperationRequest, ApiWorkflowOperationResponse, SdkHttpOperationRequest, TokenRefreshRequest, TokenRefreshResponse, };
 export type DeviceCodeApproveRequest = SdkDeviceCodeApproveRequest;
 export interface ApiAuthProvider {
@@ -114,6 +114,7 @@ export interface ApiConfig {
     projectApiPermissions: string[];
     cloudflareAccountId?: string;
     cloudflareApiToken?: string;
+    apiDatabaseUrl?: string;
     d1DatabaseId?: string;
     d1DatabaseName?: string;
     d1LocalPersistTo?: string;

package/dist/capacity-provider.d.ts

@@ -1,19 +1,28 @@
-import type { NativeUsageObservation } from './sdk-types.ts';
-import type { ProjectRepositoryTopology } from './sdk-types.ts';
+import type { CapacityGrant, NativeUsageObservation } from './sdk-types.js';
+import type { ProjectRepositoryTopology } from './sdk-types.js';
+import type { SeedProjectArchitecture } from './seeds/types.js';
+import type { ProviderAssignmentLifecycleRequest, ProviderAssignmentLifecycleResult, ProviderCheckInRequest, ProviderNextAssignmentRequest } from './agent-capacity.js';
 export declare const CAPACITY_PROVIDER_ENDPOINTS: {
     readonly register: "/v1/provider/register";
     readonly heartbeat: "/v1/provider/heartbeat";
+    readonly checkIn: "/v1/provider/check-in";
+    readonly sessions: "/v1/provider/sessions";
     readonly portfolio: "/v1/provider/portfolio";
     readonly workdays: "/v1/provider/workdays";
-    readonly claimTask: "/v1/provider/tasks/claim";
-    readonly taskEvents: (taskId: string) => string;
-    readonly completeTask: (taskId: string) => string;
-    readonly failTask: (taskId: string) => string;
+    readonly nextAssignment: "/v1/provider/assignments/next";
+    readonly assignment: (assignmentId: string) => string;
+    readonly assignmentRenew: (assignmentId: string) => string;
+    readonly assignmentReturn: (assignmentId: string) => string;
+    readonly assignmentComplete: (assignmentId: string) => string;
+    readonly assignmentFail: (assignmentId: string) => string;
+    readonly assignmentModeRuns: (assignmentId: string) => string;
+    readonly assignmentWorkflowOperationDispatch: (assignmentId: string, operationId: string) => string;
+    readonly assignmentExplanation: (assignmentId: string) => string;
     readonly usage: "/v1/provider/usage";
     readonly reports: "/v1/provider/reports";
 };
-export declare const CAPACITY_PROVIDER_SCOPES: readonly ["provider:register", "provider:heartbeat", "provider:portfolio:read", "provider:tasks:claim", "provider:tasks:update", "provider:usage:report", "provider:reports:write", "provider:capabilities:write"];
-export declare const CAPACITY_PROVIDER_ENV_KEYS: readonly ["TREESEED_MARKET_URL", "TREESEED_MANAGER_ID", "TREESEED_CAPACITY_PROVIDER_API_KEY", "TREESEED_PROVIDER_HOST_DATA_DIR", "TREESEED_PROVIDER_DATA_DIR", "TREESEED_PROVIDER_API_PORT", "TREESEED_PROVIDER_HOST_API_PORT", "TREESEED_PROVIDER_ENVIRONMENT", "TREESEED_PROVIDER_CAPABILITIES_FILE", "TREESEED_PROVIDER_BUDGET_FILE", "TREESEED_PROVIDER_MAX_CONCURRENT_WORKDAYS", "TREESEED_PROVIDER_MAX_CONCURRENT_RUNNERS", "TREESEED_PROVIDER_DAILY_CREDIT_BUDGET", "TREESEED_PROVIDER_MONTHLY_CREDIT_BUDGET", "TREESEED_PROVIDER_STARTUP_MODE", "TREESEED_CODEX_AUTH_FILE", "TREESEED_CODEX_AUTH_JSON_B64", "TREESEED_CODEX_AUTH_OVERWRITE"];
+export declare const CAPACITY_PROVIDER_SCOPES: readonly ["provider:register", "provider:heartbeat", "provider:portfolio:read", "provider:assignments:read", "provider:assignments:write", "provider:usage:report", "provider:reports:write", "provider:capabilities:write"];
+export declare const CAPACITY_PROVIDER_ENV_KEYS: readonly ["TREESEED_MANAGEMENT_API_URL", "TREESEED_MARKET_URL", "TREESEED_MARKET_ID", "TREESEED_MANAGER_ID", "TREESEED_CAPACITY_PROVIDER_ID", "TREESEED_CAPACITY_PROVIDER_TEAM_ID", "TREESEED_CAPACITY_PROVIDER_API_KEY", "TREESEED_PROVIDER_HOST_DATA_DIR", "TREESEED_HOST_CODEX_AUTH_FILE", "TREESEED_PROVIDER_DATA_DIR", "TREESEED_PROVIDER_ENVIRONMENT", "TREESEED_PROVIDER_CAPABILITIES_FILE", "TREESEED_PROVIDER_BUDGET_FILE", "TREESEED_PROVIDER_MAX_CONCURRENT_WORKDAYS", "TREESEED_PROVIDER_MAX_CONCURRENT_RUNNERS", "TREESEED_PROVIDER_DAILY_CREDIT_BUDGET", "TREESEED_PROVIDER_MONTHLY_CREDIT_BUDGET", "TREESEED_PROVIDER_STARTUP_MODE", "TREESEED_CODEX_AUTH_FILE", "TREESEED_CODEX_AUTH_JSON_B64", "TREESEED_CODEX_AUTH_OVERWRITE", "TREESEED_TREEDX_BASE_URL", "TREESEED_TREEDX_URL", "TREESEED_TREEDX_TOKEN", "TREESEED_TREEDX_REPO_ID", "TREESEED_TREEDX_REF", "TREESEED_TREEDX_WORKSPACE_ID", "TREESEED_TREEDX_JWT_ISSUER", "TREESEED_TREEDX_JWT_AUDIENCE", "TREESEED_TREEDX_JWT_HS256_SECRET", "TREESEED_TREEDX_PROXY_ACTOR_ID", "TREESEED_TREEDX_PROXY_TENANT_ID"];
 export type CapacityProviderScope = (typeof CAPACITY_PROVIDER_SCOPES)[number];
 export type CapacityProviderEnvironmentName = 'local' | 'staging' | 'prod' | 'production' | string;
 export type CapacityProviderLaunchMode = 'self_hosted' | 'managed_market_host' | 'connected_host';
@@ -22,6 +31,11 @@ export type CapacityProviderConnectionState = 'waiting_for_provider' | 'connecte
 export type CapacityProviderDeploymentStatus = 'not_deployed' | 'deploying' | 'deployed' | 'failed' | string;
 export type CapacityProviderDeploymentServiceRole = 'api' | 'manager' | 'runner';
 export declare const CAPACITY_PROVIDER_DEPLOYMENT_SERVICE_ROLES: readonly ["api", "manager", "runner"];
+export declare const DEFAULT_CAPACITY_PROVIDER_ROLE_IMAGES: {
+    readonly api: "treeseed/agent-provider-api";
+    readonly manager: "treeseed/agent-manager";
+    readonly runner: "treeseed/agent-runner";
+};
 export interface CapacityProviderRuntimeInfo {
     package: '@treeseed/agent' | string;
     version: string;
@@ -98,7 +112,7 @@ export interface CapacityProviderHealthState {
     [key: string]: unknown;
 }
 export interface CapacityProviderRegistrationRequest {
-    marketId: string;
+    marketId?: string;
     runtime: CapacityProviderRuntimeInfo;
     capabilities: CapacityProviderCapability[];
     budgets: CapacityProviderBudgetCapacity;
@@ -115,9 +129,11 @@ export interface CapacityProviderRegistrationResponse {
     };
     portfolioManifestUrl: string;
     heartbeatIntervalSeconds: number;
+    sessionToken?: string | null;
+    sessionExpiresAt?: string | null;
 }
 export interface CapacityProviderHeartbeatRequest {
-    marketId: string;
+    marketId?: string;
     providerId?: string | null;
     runtime?: CapacityProviderRuntimeInfo;
     capabilities?: CapacityProviderCapability[];
@@ -137,6 +153,7 @@ export interface CapacityProviderPortfolioManifest {
         slug: string;
         name: string;
     };
+    grants?: CapacityGrant[];
     projects: CapacityProviderPortfolioProject[];
 }
 export interface CapacityProviderPortfolioProject {
@@ -156,6 +173,7 @@ export interface CapacityProviderPortfolioProject {
         webUrl?: string | null;
     };
     repositoryTopology?: ProjectRepositoryTopology;
+    architecture?: SeedProjectArchitecture;
     agentSpecs: {
         root: string;
         testsRoot: string;
@@ -183,49 +201,6 @@ export interface ProviderWorkdayResponse {
     ok: true;
     workDay: Record<string, unknown>;
 }
-export interface ProviderTaskClaimRequest {
-    providerId?: string | null;
-    runnerId?: string | null;
-    projectId?: string | null;
-    environment?: CapacityProviderEnvironmentName | null;
-    limit?: number;
-    capabilities?: string[];
-}
-export interface ProviderTaskClaimResponse {
-    ok: true;
-    tasks: Record<string, unknown>[];
-    leaseSeconds?: number;
-}
-export interface ProviderTaskEventRequest {
-    kind: string;
-    data?: Record<string, unknown>;
-    runnerId?: string | null;
-}
-export interface ProviderTaskEventResponse {
-    ok: true;
-    event?: Record<string, unknown>;
-}
-export interface ProviderTaskCompleteRequest {
-    output?: Record<string, unknown> | null;
-    outputRef?: string | null;
-    summary?: Record<string, unknown> | null;
-    usage?: ProviderUsageReport | null;
-}
-export interface ProviderTaskCompleteResponse {
-    ok: true;
-    task: Record<string, unknown>;
-}
-export interface ProviderTaskFailRequest {
-    errorCode?: string | null;
-    errorMessage: string;
-    retryable?: boolean;
-    nextVisibleAt?: string | null;
-    usage?: ProviderUsageReport | null;
-}
-export interface ProviderTaskFailResponse {
-    ok: true;
-    task: Record<string, unknown>;
-}
 export interface ProviderUsageReport {
     taskId?: string | null;
     workDayId?: string | null;
@@ -318,6 +293,7 @@ export interface CapacityProviderDeploymentPrimitiveInput {
     env: Record<string, string>;
     redactedEnv?: Record<string, string>;
     imageRef?: string | null;
+    roleImageRefs?: Partial<Record<CapacityProviderDeploymentServiceRole, string>>;
     serviceNamePrefix?: string | null;
     adapter?: CapacityProviderDeploymentAdapter;
     now?: Date;
@@ -326,7 +302,7 @@ export interface CapacityProviderDeploymentPrimitiveResult {
     ok: boolean;
     launchMode: CapacityProviderLaunchMode;
     hostKind: string;
-    imageRef: string;
+    roleImageRefs: Record<CapacityProviderDeploymentServiceRole, string>;
     status: CapacityProviderDeploymentStatus;
     serviceRefs: Record<CapacityProviderDeploymentServiceRole, CapacityProviderDeploymentServiceResult>;
     envRefs: Record<string, unknown>;
@@ -338,8 +314,9 @@ export interface CapacityProviderEnvironmentInput {
     marketUrl: string;
     marketId: string;
     apiKey: string;
+    providerId?: string;
+    teamId?: string;
     providerDataDir?: string;
-    providerApiPort?: number | string;
     providerEnvironment?: CapacityProviderEnvironmentName;
     providerHostDataDir?: string;
     capabilitiesFile?: string;
@@ -352,6 +329,57 @@ export interface CapacityProviderEnvironmentInput {
     codexAuthJsonB64?: string;
     codexAuthOverwrite?: boolean | number | string;
 }
+export interface CapacityProviderRoleImageConfig {
+    image?: string;
+    tag?: string;
+}
+export interface CapacityProviderImageExtensionConfig {
+    enabled?: boolean;
+    baseImage?: string;
+    dockerfile?: string;
+    context?: string;
+    image?: string;
+    tag?: string;
+    buildArgs?: Record<string, string | number | boolean | null | undefined>;
+}
+export interface CapacityProviderLaunchManifest {
+    schemaVersion: 1;
+    provider?: {
+        id?: string;
+        environment?: string;
+        dataDir?: string;
+        market?: {
+            profile?: string;
+            url?: string;
+            id?: string;
+        };
+    };
+    runtime?: {
+        images?: {
+            tag?: string;
+            architecture?: 'auto' | 'amd64' | 'arm64' | string;
+            pullPolicy?: string;
+            roles?: Partial<Record<CapacityProviderDeploymentServiceRole, CapacityProviderRoleImageConfig>>;
+        };
+        manager?: {
+            pollSeconds?: number | string;
+        };
+        runners?: {
+            count?: number | string;
+            maxConcurrent?: number | string;
+            dataDir?: string;
+        };
+    };
+    extensions?: Partial<Record<CapacityProviderDeploymentServiceRole, CapacityProviderImageExtensionConfig>>;
+    executionProviders?: unknown[];
+    capabilities?: unknown[];
+}
+export interface CapacityProviderLaunchPlan {
+    manifest: CapacityProviderLaunchManifest;
+    roleImages: Record<CapacityProviderDeploymentServiceRole, string>;
+    composeEnv: Record<string, string>;
+    redactedEnv: Record<string, string>;
+}
 export interface CapacityProviderSelfHostInstructions {
     composeFile: string;
     commands: string[];
@@ -389,6 +417,7 @@ export interface CapacityProviderConnectionConfigResult {
 export interface MarketProviderClientOptions extends CapacityProviderEnvironmentInput {
     fetchImpl?: typeof fetch;
     userAgent?: string;
+    requestTimeoutMs?: number;
 }
 export declare class CapacityProviderApiError extends Error {
     readonly status: number;
@@ -409,15 +438,18 @@ export declare function persistCapacityProviderConnectionToTreeseedConfig(input:
 export declare function redactCapacityProviderSecret(value: string): string;
 export declare function isCapacityProviderSecretEnvKey(key: string): boolean;
 export declare function redactCapacityProviderEnv(env: Record<string, string>): Record<string, string>;
+export declare function parseCapacityProviderLaunchManifest(source: string | Record<string, unknown>): CapacityProviderLaunchManifest;
+export declare function resolveCapacityProviderLaunchPlan(manifest: CapacityProviderLaunchManifest): CapacityProviderLaunchPlan;
 export declare function renderCapacityProviderSelfHostInstructions(input: CapacityProviderEnvironmentInput): CapacityProviderSelfHostInstructions;
 export declare function deployCapacityProviderToRailway(input: CapacityProviderDeploymentPrimitiveInput): Promise<CapacityProviderDeploymentPrimitiveResult>;
 export declare function deployCapacityProviderToManagedMarketHost(input: CapacityProviderDeploymentPrimitiveInput): Promise<CapacityProviderDeploymentPrimitiveResult>;
 export declare class MarketProviderClient {
     private readonly marketUrl;
     private readonly marketId;
-    private readonly apiKey;
+    private apiKey;
     private readonly fetchImpl;
     private readonly userAgent?;
+    private readonly requestTimeoutMs;
     constructor(options: MarketProviderClientOptions);
     private requestJson;
     register(request: Omit<CapacityProviderRegistrationRequest, 'marketId'> & {
@@ -426,12 +458,37 @@ export declare class MarketProviderClient {
     heartbeat(request?: Omit<CapacityProviderHeartbeatRequest, 'marketId'> & {
         marketId?: string;
     }): Promise<CapacityProviderHeartbeatResponse>;
+    createAvailabilitySession(request?: Record<string, unknown>): Promise<{
+        ok: true;
+        payload: Record<string, unknown>;
+    }>;
+    checkIn(request?: ProviderCheckInRequest): Promise<{
+        ok: true;
+        payload: Record<string, unknown>;
+    }>;
     portfolio(): Promise<CapacityProviderPortfolioManifest>;
     createWorkday(request: ProviderWorkdayRequest): Promise<ProviderWorkdayResponse>;
-    claimTask(request?: ProviderTaskClaimRequest): Promise<ProviderTaskClaimResponse>;
-    appendTaskEvent(taskId: string, request: ProviderTaskEventRequest): Promise<ProviderTaskEventResponse>;
-    completeTask(taskId: string, request?: ProviderTaskCompleteRequest): Promise<ProviderTaskCompleteResponse>;
-    failTask(taskId: string, request: ProviderTaskFailRequest): Promise<ProviderTaskFailResponse>;
+    assignment(assignmentId: string): Promise<{
+        ok: true;
+        payload: Record<string, unknown>;
+    }>;
+    assignmentExplanation(assignmentId: string): Promise<{
+        ok: true;
+        payload: Record<string, unknown>;
+    }>;
+    nextAssignment(request?: ProviderNextAssignmentRequest): Promise<ProviderAssignmentLifecycleResult>;
+    renewAssignment(assignmentId: string, request?: ProviderAssignmentLifecycleRequest): Promise<ProviderAssignmentLifecycleResult>;
+    returnAssignment(assignmentId: string, request?: ProviderAssignmentLifecycleRequest): Promise<ProviderAssignmentLifecycleResult>;
+    completeAssignment(assignmentId: string, request?: ProviderAssignmentLifecycleRequest): Promise<ProviderAssignmentLifecycleResult>;
+    failAssignment(assignmentId: string, request?: ProviderAssignmentLifecycleRequest): Promise<ProviderAssignmentLifecycleResult>;
+    createAssignmentModeRun(assignmentId: string, request: Record<string, unknown>): Promise<{
+        ok: true;
+        payload: Record<string, unknown>;
+    }>;
+    dispatchAssignmentWorkflowOperation(assignmentId: string, operationId: string, request: Record<string, unknown>): Promise<{
+        ok: true;
+        payload: Record<string, unknown>;
+    }>;
     reportUsage(request: ProviderUsageReport): Promise<ProviderUsageReportResponse>;
     writeReport(request: ProviderReportRequest): Promise<ProviderReportResponse>;
 }