@treeseed/sdk 0.4.8 → 0.4.10

package/README.md

@@ -168,7 +168,7 @@ const scoped = sdk.scopeForAgent({
 The package also exports:
 
 - workflow helpers such as `TreeseedWorkflowSdk`
-- gateway and queue clients such as `TreeseedGatewayClient` and `CloudflareQueuePullClient`
+- remote and queue clients such as `RemoteTreeseedClient`, `CloudflareQueuePullClient`, and `CloudflareQueuePushClient`
 - model registry, field, and filter utilities
 - plugin/runtime types and helpers
 

package/dist/control-plane-client.d.ts

@@ -0,0 +1,45 @@
+import type { AgentPool, AgentPoolRegistration, AgentPoolScaleDecision, CatalogArtifactVersion, CatalogItem, CatalogItemFilters, CreateProjectDeploymentRequest, PriorityOverride, PrioritySnapshot, ProjectDeployment, ProjectEnvironment, ProjectEnvironmentName, ProjectHosting, ProjectInfrastructureResource, ProjectWorkdaySummary, RecordAgentPoolRegistrationRequest, ScaleDecision, TeamStorageLocator, UpsertAgentPoolRequest, UpsertCatalogArtifactVersionRequest, UpsertCatalogItemRequest, UpsertProjectEnvironmentRequest, UpsertProjectHostingRequest, UpsertProjectInfrastructureResourceRequest, UpsertTeamStorageLocatorRequest, WorkdayPolicy, SdkPriorityOverrideRequest, SdkUpsertWorkPolicyRequest } from './sdk-types.ts';
+export interface ControlPlaneClientOptions {
+    baseUrl: string;
+    accessToken?: string | null;
+    fetchImpl?: typeof fetch;
+}
+export declare class ControlPlaneClient {
+    private readonly baseUrl;
+    private readonly accessToken;
+    private readonly fetchImpl;
+    constructor(options: ControlPlaneClientOptions);
+    private requestUrl;
+    private requestJson;
+    listCatalogItems(filters?: CatalogItemFilters): Promise<CatalogItem[]>;
+    getCatalogItem(itemId: string): Promise<CatalogItem>;
+    listCatalogArtifactVersions(itemId: string): Promise<CatalogArtifactVersion[]>;
+    upsertCatalogItem(teamId: string, input: UpsertCatalogItemRequest): Promise<CatalogItem>;
+    upsertCatalogArtifactVersion(itemId: string, input: UpsertCatalogArtifactVersionRequest): Promise<CatalogArtifactVersion>;
+    getTeamStorageLocator(teamId: string): Promise<TeamStorageLocator | null>;
+    upsertTeamStorageLocator(teamId: string, input: UpsertTeamStorageLocatorRequest): Promise<TeamStorageLocator>;
+    getProjectHosting(projectId: string): Promise<ProjectHosting | null>;
+    upsertProjectHosting(projectId: string, input: UpsertProjectHostingRequest): Promise<ProjectHosting>;
+    listProjectEnvironments(projectId: string): Promise<ProjectEnvironment[]>;
+    upsertProjectEnvironment(projectId: string, environment: ProjectEnvironmentName, input: UpsertProjectEnvironmentRequest): Promise<ProjectEnvironment>;
+    listProjectInfrastructureResources(projectId: string, environment?: ProjectEnvironmentName | null): Promise<ProjectInfrastructureResource[]>;
+    upsertProjectInfrastructureResource(projectId: string, input: UpsertProjectInfrastructureResourceRequest): Promise<ProjectInfrastructureResource>;
+    listProjectDeployments(projectId: string, environment?: ProjectEnvironmentName | null): Promise<ProjectDeployment[]>;
+    createProjectDeployment(projectId: string, input: CreateProjectDeploymentRequest): Promise<ProjectDeployment>;
+    listAgentPools(projectId: string, environment?: ProjectEnvironmentName | null): Promise<AgentPool[]>;
+    upsertAgentPool(projectId: string, input: UpsertAgentPoolRequest): Promise<AgentPool>;
+    listAgentPoolRegistrations(projectId: string, poolId: string): Promise<AgentPoolRegistration[]>;
+    listAgentPoolScaleDecisions(projectId: string, poolId: string): Promise<AgentPoolScaleDecision[]>;
+    getProjectWorkPolicy(projectId: string, environment?: ProjectEnvironmentName): Promise<WorkdayPolicy | null>;
+    upsertProjectWorkPolicy(projectId: string, input: SdkUpsertWorkPolicyRequest): Promise<WorkdayPolicy>;
+    listProjectPriorityOverrides(projectId: string): Promise<PriorityOverride[]>;
+    upsertProjectPriorityOverride(projectId: string, input: SdkPriorityOverrideRequest): Promise<PriorityOverride>;
+    listProjectPrioritySnapshots(projectId: string, workDayId?: string | null): Promise<PrioritySnapshot[]>;
+    recordRunnerEnvironment(projectId: string, environment: ProjectEnvironmentName, input: UpsertProjectEnvironmentRequest): Promise<ProjectEnvironment>;
+    recordRunnerInfrastructureResource(projectId: string, input: UpsertProjectInfrastructureResourceRequest): Promise<ProjectInfrastructureResource>;
+    recordRunnerDeployment(projectId: string, input: CreateProjectDeploymentRequest): Promise<ProjectDeployment>;
+    listRunnerDeployments(projectId: string, environment?: ProjectEnvironmentName | null): Promise<ProjectDeployment[]>;
+    recordRunnerAgentPoolRegistration(projectId: string, poolId: string, input: RecordAgentPoolRegistrationRequest): Promise<AgentPoolRegistration>;
+    recordRunnerScaleDecision(projectId: string, poolName: string, input: Pick<ScaleDecision, 'environment' | 'workDayId' | 'desiredWorkers' | 'observedQueueDepth' | 'observedActiveLeases' | 'reason' | 'metadata'>): Promise<AgentPoolScaleDecision>;
+    recordRunnerWorkdaySummary(projectId: string, input: ProjectWorkdaySummary): Promise<ProjectWorkdaySummary>;
+}

package/dist/control-plane-client.js

@@ -0,0 +1,229 @@
+function normalizeBaseUrl(value) {
+  return value.trim().replace(/\/+$/u, "");
+}
+class ControlPlaneClient {
+  baseUrl;
+  accessToken;
+  fetchImpl;
+  constructor(options) {
+    this.baseUrl = normalizeBaseUrl(options.baseUrl);
+    this.accessToken = typeof options.accessToken === "string" && options.accessToken.trim() ? options.accessToken.trim() : null;
+    this.fetchImpl = options.fetchImpl ?? fetch;
+  }
+  requestUrl(pathname, query) {
+    const url = new URL(pathname, `${this.baseUrl}/`);
+    for (const [key, value] of Object.entries(query ?? {})) {
+      if (typeof value === "string" && value.trim()) {
+        url.searchParams.set(key, value);
+      }
+    }
+    return url;
+  }
+  async requestJson(method, pathname, options = {}) {
+    const headers = new Headers({
+      accept: "application/json"
+    });
+    if (this.accessToken) {
+      headers.set("authorization", `Bearer ${this.accessToken}`);
+    }
+    if (options.body) {
+      headers.set("content-type", "application/json");
+    }
+    const response = await this.fetchImpl(this.requestUrl(pathname, options.query), {
+      method,
+      headers,
+      body: options.body ? JSON.stringify(options.body) : void 0
+    });
+    if (!response.ok) {
+      throw new Error(`Control-plane request failed for ${pathname}: ${response.status} ${response.statusText}`);
+    }
+    const payload = await response.json();
+    if (!payload.ok) {
+      throw new Error(`Control-plane request returned a non-ok envelope for ${pathname}.`);
+    }
+    return payload.payload;
+  }
+  listCatalogItems(filters = {}) {
+    return this.requestJson("GET", "/v1/catalog", {
+      query: {
+        kind: filters.kind ?? null,
+        teamId: filters.teamId ?? null,
+        slug: filters.slug ?? null
+      }
+    });
+  }
+  getCatalogItem(itemId) {
+    return this.requestJson("GET", `/v1/catalog/${encodeURIComponent(itemId)}`);
+  }
+  listCatalogArtifactVersions(itemId) {
+    return this.requestJson("GET", `/v1/catalog/${encodeURIComponent(itemId)}/artifacts`);
+  }
+  upsertCatalogItem(teamId, input) {
+    return this.requestJson("POST", `/v1/teams/${encodeURIComponent(teamId)}/catalog-items`, { body: input });
+  }
+  upsertCatalogArtifactVersion(itemId, input) {
+    return this.requestJson("POST", `/v1/catalog/${encodeURIComponent(itemId)}/artifacts`, { body: input });
+  }
+  getTeamStorageLocator(teamId) {
+    return this.requestJson("GET", `/v1/teams/${encodeURIComponent(teamId)}/storage`);
+  }
+  upsertTeamStorageLocator(teamId, input) {
+    return this.requestJson("PUT", `/v1/teams/${encodeURIComponent(teamId)}/storage`, { body: input });
+  }
+  getProjectHosting(projectId) {
+    return this.requestJson("GET", `/v1/projects/${encodeURIComponent(projectId)}/hosting`);
+  }
+  upsertProjectHosting(projectId, input) {
+    return this.requestJson("PUT", `/v1/projects/${encodeURIComponent(projectId)}/hosting`, { body: input });
+  }
+  listProjectEnvironments(projectId) {
+    return this.requestJson("GET", `/v1/projects/${encodeURIComponent(projectId)}/environments`);
+  }
+  upsertProjectEnvironment(projectId, environment, input) {
+    return this.requestJson(
+      "PUT",
+      `/v1/projects/${encodeURIComponent(projectId)}/environments/${encodeURIComponent(environment)}`,
+      { body: input }
+    );
+  }
+  listProjectInfrastructureResources(projectId, environment) {
+    return this.requestJson(
+      "GET",
+      `/v1/projects/${encodeURIComponent(projectId)}/resources`,
+      { query: { environment: environment ?? null } }
+    );
+  }
+  upsertProjectInfrastructureResource(projectId, input) {
+    return this.requestJson(
+      "POST",
+      `/v1/projects/${encodeURIComponent(projectId)}/resources`,
+      { body: input }
+    );
+  }
+  listProjectDeployments(projectId, environment) {
+    return this.requestJson(
+      "GET",
+      `/v1/projects/${encodeURIComponent(projectId)}/deployments`,
+      { query: { environment: environment ?? null } }
+    );
+  }
+  createProjectDeployment(projectId, input) {
+    return this.requestJson(
+      "POST",
+      `/v1/projects/${encodeURIComponent(projectId)}/deployments`,
+      { body: input }
+    );
+  }
+  listAgentPools(projectId, environment) {
+    return this.requestJson(
+      "GET",
+      `/v1/projects/${encodeURIComponent(projectId)}/agent-pools`,
+      { query: { environment: environment ?? null } }
+    );
+  }
+  upsertAgentPool(projectId, input) {
+    return this.requestJson(
+      "POST",
+      `/v1/projects/${encodeURIComponent(projectId)}/agent-pools`,
+      { body: input }
+    );
+  }
+  listAgentPoolRegistrations(projectId, poolId) {
+    return this.requestJson(
+      "GET",
+      `/v1/projects/${encodeURIComponent(projectId)}/agent-pools/${encodeURIComponent(poolId)}/registrations`
+    );
+  }
+  listAgentPoolScaleDecisions(projectId, poolId) {
+    return this.requestJson(
+      "GET",
+      `/v1/projects/${encodeURIComponent(projectId)}/agent-pools/${encodeURIComponent(poolId)}/scale-decisions`
+    );
+  }
+  getProjectWorkPolicy(projectId, environment = "staging") {
+    return this.requestJson(
+      "GET",
+      `/v1/projects/${encodeURIComponent(projectId)}/work-policy`,
+      { query: { environment } }
+    );
+  }
+  upsertProjectWorkPolicy(projectId, input) {
+    return this.requestJson(
+      "PUT",
+      `/v1/projects/${encodeURIComponent(projectId)}/work-policy`,
+      { body: input }
+    );
+  }
+  listProjectPriorityOverrides(projectId) {
+    return this.requestJson(
+      "GET",
+      `/v1/projects/${encodeURIComponent(projectId)}/priority-overrides`
+    );
+  }
+  upsertProjectPriorityOverride(projectId, input) {
+    return this.requestJson(
+      "POST",
+      `/v1/projects/${encodeURIComponent(projectId)}/priority-overrides`,
+      { body: input }
+    );
+  }
+  listProjectPrioritySnapshots(projectId, workDayId) {
+    return this.requestJson(
+      "GET",
+      `/v1/projects/${encodeURIComponent(projectId)}/priority-snapshots`,
+      { query: { workDayId: workDayId ?? null } }
+    );
+  }
+  recordRunnerEnvironment(projectId, environment, input) {
+    return this.requestJson(
+      "PUT",
+      `/v1/projects/${encodeURIComponent(projectId)}/runner/environments/${encodeURIComponent(environment)}`,
+      { body: input }
+    );
+  }
+  recordRunnerInfrastructureResource(projectId, input) {
+    return this.requestJson(
+      "POST",
+      `/v1/projects/${encodeURIComponent(projectId)}/runner/resources`,
+      { body: input }
+    );
+  }
+  recordRunnerDeployment(projectId, input) {
+    return this.requestJson(
+      "POST",
+      `/v1/projects/${encodeURIComponent(projectId)}/runner/deployments`,
+      { body: input }
+    );
+  }
+  listRunnerDeployments(projectId, environment) {
+    return this.requestJson(
+      "GET",
+      `/v1/projects/${encodeURIComponent(projectId)}/runner/deployments`,
+      { query: { environment: environment ?? null } }
+    );
+  }
+  recordRunnerAgentPoolRegistration(projectId, poolId, input) {
+    return this.requestJson(
+      "POST",
+      `/v1/projects/${encodeURIComponent(projectId)}/runner/agent-pools/${encodeURIComponent(poolId)}/register`,
+      { body: input }
+    );
+  }
+  recordRunnerScaleDecision(projectId, poolName, input) {
+    return this.requestJson(
+      "POST",
+      `/v1/projects/${encodeURIComponent(projectId)}/runner/agent-pools/${encodeURIComponent(poolName)}/scale-decisions`,
+      { body: input }
+    );
+  }
+  recordRunnerWorkdaySummary(projectId, input) {
+    return this.requestJson(
+      "POST",
+      `/v1/projects/${encodeURIComponent(projectId)}/runner/workdays`,
+      { body: input }
+    );
+  }
+}
+export {
+  ControlPlaneClient
+};

package/dist/control-plane.d.ts

@@ -0,0 +1,94 @@
+import type { AgentPoolAutoscalePolicy, ProjectDeploymentKind, ProjectDeploymentStatus, ProjectEnvironmentName, ProjectInfrastructureResourceKind, ProjectInfrastructureResourceProvider, TreeseedHostingKind, TreeseedHostingRegistration } from './sdk-types.ts';
+import type { TreeseedDeployConfig } from './platform/contracts.ts';
+export type ControlPlaneReporterKind = 'noop' | 'market_http' | 'self_http';
+export interface ControlPlaneEnvironmentReport {
+    environment: ProjectEnvironmentName;
+    deploymentProfile: TreeseedHostingKind;
+    baseUrl?: string | null;
+    cloudflareAccountId?: string | null;
+    pagesProjectName?: string | null;
+    workerName?: string | null;
+    r2BucketName?: string | null;
+    d1DatabaseName?: string | null;
+    queueName?: string | null;
+    railwayProjectName?: string | null;
+    metadata?: Record<string, unknown>;
+}
+export interface ControlPlaneResourceReport {
+    environment: ProjectEnvironmentName;
+    provider: ProjectInfrastructureResourceProvider;
+    resourceKind: ProjectInfrastructureResourceKind;
+    logicalName: string;
+    locator?: string | null;
+    metadata?: Record<string, unknown>;
+}
+export interface ControlPlaneDeploymentReport {
+    environment: ProjectEnvironmentName;
+    deploymentKind: ProjectDeploymentKind;
+    status: ProjectDeploymentStatus | 'success';
+    sourceRef?: string | null;
+    releaseTag?: string | null;
+    commitSha?: string | null;
+    triggeredByType?: string | null;
+    triggeredById?: string | null;
+    metadata?: Record<string, unknown>;
+    startedAt?: string | null;
+    finishedAt?: string | null;
+}
+export interface ControlPlaneAgentPoolHeartbeat {
+    teamId: string;
+    environment: ProjectEnvironmentName;
+    poolName: string;
+    managerId?: string | null;
+    serviceName?: string | null;
+    registrationIdentity?: string | null;
+    serviceBaseUrl?: string | null;
+    autoscale?: AgentPoolAutoscalePolicy;
+    desiredWorkers?: number | null;
+    observedQueueDepth?: number | null;
+    observedActiveLeases?: number | null;
+    metadata?: Record<string, unknown>;
+}
+export interface ControlPlaneScaleDecisionReport {
+    environment: ProjectEnvironmentName;
+    poolName: string;
+    workDayId?: string | null;
+    desiredWorkers: number;
+    observedQueueDepth: number;
+    observedActiveLeases: number;
+    reason: string;
+    metadata?: Record<string, unknown>;
+}
+export interface ControlPlaneWorkdaySummaryReport {
+    environment: ProjectEnvironmentName;
+    workDayId: string;
+    kind?: string;
+    state?: string | null;
+    startedAt?: string | null;
+    endedAt?: string | null;
+    summary: Record<string, unknown>;
+    metadata?: Record<string, unknown>;
+}
+export interface ControlPlaneReporter {
+    readonly kind: ControlPlaneReporterKind;
+    readonly enabled: boolean;
+    reportEnvironment(input: ControlPlaneEnvironmentReport): Promise<void>;
+    reportResource(input: ControlPlaneResourceReport): Promise<void>;
+    reportDeployment(input: ControlPlaneDeploymentReport): Promise<void>;
+    registerAgentPoolHeartbeat(input: ControlPlaneAgentPoolHeartbeat): Promise<void>;
+    reportScaleDecision(input: ControlPlaneScaleDecisionReport): Promise<void>;
+    reportWorkdaySummary(input: ControlPlaneWorkdaySummaryReport): Promise<void>;
+}
+export interface ControlPlaneReporterOptions {
+    kind?: ControlPlaneReporterKind | null;
+    projectId?: string | null;
+    baseUrl?: string | null;
+    runnerToken?: string | null;
+    fetchImpl?: typeof fetch;
+}
+export interface ResolveControlPlaneReporterOptions extends ControlPlaneReporterOptions {
+    hostingKind?: TreeseedHostingKind | null;
+    registration?: TreeseedHostingRegistration | null;
+    deployConfig?: Pick<TreeseedDeployConfig, 'hosting'> | null;
+}
+export declare function createControlPlaneReporter(options?: ResolveControlPlaneReporterOptions): ControlPlaneReporter;

package/dist/control-plane.js

@@ -0,0 +1,125 @@
+function normalizeUrl(value) {
+  const normalized = String(value ?? "").trim().replace(/\/+$/u, "");
+  return normalized || null;
+}
+function resolveReporterKind(options) {
+  if (options.kind) {
+    return options.kind;
+  }
+  const hostingKind = options.hostingKind ?? options.deployConfig?.hosting?.kind ?? (process.env.TREESEED_HOSTING_KIND?.trim() || null) ?? "self_hosted_project";
+  const registration = options.registration ?? options.deployConfig?.hosting?.registration ?? (process.env.TREESEED_HOSTING_REGISTRATION?.trim() || null) ?? "none";
+  if (hostingKind === "hosted_project") {
+    return "market_http";
+  }
+  if (hostingKind === "self_hosted_project" && registration === "optional") {
+    return "market_http";
+  }
+  if (hostingKind === "market_control_plane") {
+    return normalizeUrl(options.baseUrl ?? options.deployConfig?.hosting?.marketBaseUrl) ? "self_http" : "noop";
+  }
+  return "noop";
+}
+class NoopControlPlaneReporter {
+  kind = "noop";
+  enabled = false;
+  async reportEnvironment() {
+  }
+  async reportResource() {
+  }
+  async reportDeployment() {
+  }
+  async registerAgentPoolHeartbeat() {
+  }
+  async reportScaleDecision() {
+  }
+  async reportWorkdaySummary() {
+  }
+}
+class HttpControlPlaneReporter {
+  constructor(kind, projectId, baseUrl, runnerToken, fetchImpl = fetch) {
+    this.kind = kind;
+    this.projectId = projectId;
+    this.baseUrl = baseUrl;
+    this.runnerToken = runnerToken;
+    this.fetchImpl = fetchImpl;
+    this.enabled = Boolean(this.projectId && this.baseUrl && this.runnerToken);
+  }
+  kind;
+  projectId;
+  baseUrl;
+  runnerToken;
+  fetchImpl;
+  enabled;
+  async request(method, pathname, body) {
+    if (!this.enabled || !this.baseUrl || !this.runnerToken) {
+      return;
+    }
+    const response = await this.fetchImpl(new URL(pathname, this.baseUrl), {
+      method,
+      headers: {
+        authorization: `Bearer ${this.runnerToken}`,
+        "content-type": "application/json"
+      },
+      body: JSON.stringify(body)
+    });
+    if (!response.ok) {
+      throw new Error(`Control-plane request failed for ${pathname}: ${response.status} ${response.statusText}`);
+    }
+  }
+  async reportEnvironment(input) {
+    if (!this.projectId) return;
+    await this.request("PUT", `/v1/projects/${this.projectId}/runner/environments/${input.environment}`, input);
+  }
+  async reportResource(input) {
+    if (!this.projectId) return;
+    await this.request("POST", `/v1/projects/${this.projectId}/runner/resources`, input);
+  }
+  async reportDeployment(input) {
+    if (!this.projectId) return;
+    const normalizedStatus = input.status === "success" ? "succeeded" : input.status;
+    await this.request("POST", `/v1/projects/${this.projectId}/runner/deployments`, {
+      ...input,
+      status: normalizedStatus
+    });
+  }
+  async registerAgentPoolHeartbeat(input) {
+    if (!this.projectId) return;
+    await this.request(
+      "POST",
+      `/v1/projects/${this.projectId}/runner/agent-pools/${encodeURIComponent(input.poolName)}/register`,
+      input
+    );
+  }
+  async reportScaleDecision(input) {
+    if (!this.projectId) return;
+    await this.request(
+      "POST",
+      `/v1/projects/${this.projectId}/runner/agent-pools/${encodeURIComponent(input.poolName)}/scale-decisions`,
+      input
+    );
+  }
+  async reportWorkdaySummary(input) {
+    if (!this.projectId) return;
+    await this.request("POST", `/v1/projects/${this.projectId}/runner/workdays`, input);
+  }
+}
+function createControlPlaneReporter(options = {}) {
+  const kind = resolveReporterKind(options);
+  if (kind === "noop") {
+    return new NoopControlPlaneReporter();
+  }
+  const projectId = String(
+    options.projectId ?? options.deployConfig?.hosting?.projectId ?? process.env.TREESEED_PROJECT_ID ?? ""
+  ).trim() || null;
+  const baseUrl = normalizeUrl(
+    options.baseUrl ?? options.deployConfig?.hosting?.marketBaseUrl ?? process.env.TREESEED_MARKET_API_BASE_URL ?? null
+  );
+  const runnerToken = String(options.runnerToken ?? process.env.TREESEED_PROJECT_RUNNER_TOKEN ?? "").trim() || null;
+  if (!projectId || !baseUrl || !runnerToken) {
+    return new NoopControlPlaneReporter();
+  }
+  return new HttpControlPlaneReporter(kind, projectId, baseUrl, runnerToken, options.fetchImpl);
+}
+export {
+  createControlPlaneReporter
+};

package/dist/d1-store.d.ts

@@ -1,6 +1,6 @@
 import type { ContentLeaseRecord } from './types/agents.ts';
 import type { D1DatabaseLike } from './types/cloudflare.ts';
-import type { SdkAppendTaskEventRequest, SdkAckMessageRequest, SdkClaimMessageRequest, SdkClaimTaskRequest, SdkCloseWorkDayRequest, SdkCompleteTaskRequest, SdkCreateReportRequest, SdkCreateMessageRequest, SdkCreateTaskRequest, SdkCursorEntity, SdkCursorRequest, SdkFailTaskRequest, SdkFollowRequest, SdkGetRequest, SdkGetCursorRequest, SdkLeaseEntity, SdkLeaseReleaseRequest, SdkManagerContextPayload, SdkMessageEntity, SdkMutationRequest, SdkPickRequest, SdkPickResult, SdkRecordRunRequest, SdkReportEntity, SdkRunEntity, SdkSearchRequest, SdkStartWorkDayRequest, SdkSubscriptionEntity, SdkTaskEntity, SdkTaskSearchRequest, SdkTaskProgressRequest, SdkUpdateRequest, SdkWorkDayEntity } from './sdk-types.ts';
+import type { SdkAppendTaskEventRequest, SdkAckMessageRequest, SdkClaimMessageRequest, SdkClaimTaskRequest, SdkCloseWorkDayRequest, SdkCompleteTaskRequest, SdkCreateReportRequest, SdkCreateMessageRequest, SdkCreatePrioritySnapshotRequest, SdkCreateTaskRequest, SdkCursorEntity, SdkCursorRequest, SdkFailTaskRequest, SdkFollowRequest, SdkGetRequest, SdkGetCursorRequest, SdkLeaseEntity, SdkLeaseReleaseRequest, SdkManagerContextPayload, SdkMessageEntity, SdkMutationRequest, SdkPickRequest, SdkPickResult, SdkPriorityOverrideRequest, SdkRecordRunRequest, SdkRecordScaleDecisionRequest, SdkRecordTaskCreditsRequest, SdkReportEntity, SdkRunEntity, SdkSearchRequest, SdkStartWorkDayRequest, SdkSubscriptionEntity, SdkTaskEntity, SdkTaskSearchRequest, SdkUpsertWorkPolicyRequest, SdkTaskProgressRequest, SdkUpdateRequest, SdkWorkDayEntity, ScaleDecision, TaskCreditLedgerEntry, WorkdayPolicy, PrioritySnapshot } from './sdk-types.ts';
 import { type LeaseClaimInput } from './stores/lease-store.ts';
 export interface TryClaimContentLeaseInput extends LeaseClaimInput {
 }
@@ -34,6 +34,16 @@ export interface AgentDatabase {
     searchTasks(request: SdkTaskSearchRequest): Promise<SdkTaskEntity[]>;
     createReport(request: SdkCreateReportRequest): Promise<SdkReportEntity | null>;
     getManagerContext(taskId: string): Promise<SdkManagerContextPayload>;
+    getWorkPolicy(projectId: string, environment?: string): Promise<WorkdayPolicy | null>;
+    upsertWorkPolicy(request: SdkUpsertWorkPolicyRequest): Promise<WorkdayPolicy | null>;
+    listPriorityOverrides(projectId: string): Promise<Record<string, unknown>[]>;
+    upsertPriorityOverride(request: SdkPriorityOverrideRequest): Promise<Record<string, unknown> | null>;
+    createPrioritySnapshot(request: SdkCreatePrioritySnapshotRequest): Promise<PrioritySnapshot | null>;
+    getLatestPrioritySnapshot(projectId: string, workDayId?: string | null): Promise<PrioritySnapshot | null>;
+    recordTaskCredits(request: SdkRecordTaskCreditsRequest): Promise<TaskCreditLedgerEntry | null>;
+    listTaskCredits(workDayId: string): Promise<TaskCreditLedgerEntry[]>;
+    recordScaleDecision(request: SdkRecordScaleDecisionRequest): Promise<ScaleDecision | null>;
+    getLatestScaleDecision(projectId: string, environment: string, poolName: string): Promise<ScaleDecision | null>;
 }
 export declare class MemoryAgentDatabase implements AgentDatabase {
     private readonly subscriptions;
@@ -47,6 +57,11 @@ export declare class MemoryAgentDatabase implements AgentDatabase {
     private readonly taskOutputs;
     private readonly graphRuns;
     private readonly reports;
+    private readonly workPolicies;
+    private readonly priorityOverrides;
+    private readonly prioritySnapshots;
+    private readonly taskCreditLedger;
+    private readonly scaleDecisions;
     private messageId;
     constructor(seed?: {
         subscriptions?: SdkSubscriptionEntity[];
@@ -114,6 +129,26 @@ export declare class MemoryAgentDatabase implements AgentDatabase {
             graphVersion: string;
         } | null;
     }>;
+    getWorkPolicy(projectId: string, environment?: string): Promise<WorkdayPolicy | null>;
+    upsertWorkPolicy(request: SdkUpsertWorkPolicyRequest): Promise<WorkdayPolicy>;
+    listPriorityOverrides(projectId: string): Promise<Record<string, unknown>[]>;
+    upsertPriorityOverride(request: SdkPriorityOverrideRequest): Promise<{
+        id: string;
+        projectId: string;
+        model: string;
+        subjectId: string;
+        priority: number;
+        estimatedCredits: number | null;
+        metadata: Record<string, unknown>;
+        createdAt: string;
+        updatedAt: string;
+    }>;
+    createPrioritySnapshot(request: SdkCreatePrioritySnapshotRequest): Promise<PrioritySnapshot>;
+    getLatestPrioritySnapshot(projectId: string, workDayId?: string | null): Promise<PrioritySnapshot>;
+    recordTaskCredits(request: SdkRecordTaskCreditsRequest): Promise<TaskCreditLedgerEntry>;
+    listTaskCredits(workDayId: string): Promise<TaskCreditLedgerEntry[]>;
+    recordScaleDecision(request: SdkRecordScaleDecisionRequest): Promise<ScaleDecision>;
+    getLatestScaleDecision(projectId: string, environment: string, poolName: string): Promise<ScaleDecision>;
 }
 export declare class CloudflareD1AgentDatabase implements AgentDatabase {
     readonly db: D1DatabaseLike;
@@ -166,4 +201,24 @@ export declare class CloudflareD1AgentDatabase implements AgentDatabase {
             graphVersion: string;
         } | null;
     }>;
+    getWorkPolicy(projectId: string, environment?: string): Promise<WorkdayPolicy | null>;
+    upsertWorkPolicy(request: SdkUpsertWorkPolicyRequest): Promise<WorkdayPolicy | null>;
+    listPriorityOverrides(projectId: string): Promise<import("./stores/helpers.ts").DatabaseRow[]>;
+    upsertPriorityOverride(request: SdkPriorityOverrideRequest): Promise<{
+        id: string;
+        projectId: string;
+        model: string;
+        subjectId: string;
+        priority: number;
+        estimatedCredits: number | null;
+        metadata: Record<string, unknown>;
+        createdAt: string;
+        updatedAt: string;
+    } | null>;
+    createPrioritySnapshot(request: SdkCreatePrioritySnapshotRequest): Promise<PrioritySnapshot | null>;
+    getLatestPrioritySnapshot(projectId: string, workDayId?: string | null): Promise<PrioritySnapshot | null>;
+    recordTaskCredits(request: SdkRecordTaskCreditsRequest): Promise<TaskCreditLedgerEntry | null>;
+    listTaskCredits(workDayId: string): Promise<TaskCreditLedgerEntry[]>;
+    recordScaleDecision(request: SdkRecordScaleDecisionRequest): Promise<ScaleDecision | null>;
+    getLatestScaleDecision(projectId: string, environment: string, poolName: string): Promise<ScaleDecision | null>;
 }