@assemblyline-agents/runtime 2.0.0 → 3.1.0

package/dist/workspace-service.d.ts

@@ -0,0 +1,173 @@
+import { type SandboxSession } from "@assemblyline-agents/core";
+import type { BlobAdapter } from "./runtime-types.js";
+import type { MemoryEmbeddingProvider, ResolvedStateAdapter, RunRecord, Workspace, WorkspaceCheckpoint, WorkspaceCommitResult, WorkspaceManifest, WorkspaceManifestEntry, WorkspaceSearchIndex, WorkspaceSearchResult, WorkspaceVersion } from "./state.js";
+import { type WorkspaceLimits } from "./workspace-manifest.js";
+import { type WorkspaceSearchLimits } from "./workspace-search.js";
+import { type WorkspaceDiagnostics, type WorkspaceGarbageCollectionReport, type WorkspaceReachabilityReport, type WorkspaceRetentionReport, type WorkspaceUsageReport, type WorkspaceVerificationReport } from "./workspace-operations.js";
+import { type WorkspaceOperationLimiter } from "./workspace-operation-limiter.js";
+export interface WorkspaceServiceHost {
+    state: ResolvedStateAdapter;
+    blob: BlobAdapter;
+    agentScope: string;
+    now(): Date;
+    limits?: WorkspaceLimits;
+    searchLimits?: WorkspaceSearchLimits;
+    embeddingProvider?: MemoryEmbeddingProvider;
+    operationLimiter?: WorkspaceOperationLimiter;
+    maxCheckpoints?: number;
+}
+export interface WorkspaceHydrationResult {
+    workspace: Workspace;
+    version?: WorkspaceVersion;
+    manifest: WorkspaceManifest;
+    skippedDirty: boolean;
+}
+export interface WorkspaceSyncResult extends WorkspaceCommitResult {
+    manifest: WorkspaceManifest;
+    changedPaths: string[];
+    deletedPaths: string[];
+}
+export interface WorkspaceStatus {
+    workspace: Workspace;
+    head?: WorkspaceVersion;
+    checkpoints: WorkspaceCheckpoint[];
+    versionCount: number;
+}
+export interface WorkspaceCommittedFile {
+    workspaceId: string;
+    versionId: string;
+    entry: WorkspaceManifestEntry;
+    contents: Uint8Array;
+}
+export interface WorkspaceGrepResult {
+    path: string;
+    versionId: string;
+    line: number;
+    column: number;
+    text: string;
+}
+export interface WorkspaceSearchResponse {
+    workspaceId: string;
+    versionId: string;
+    indexedVersionId?: string;
+    indexStale: boolean;
+    results: WorkspaceSearchResult[];
+}
+export declare class WorkspaceService {
+    private readonly host;
+    private readonly limits;
+    private readonly searchLimits;
+    private readonly operationLimiter;
+    private readonly maxCheckpoints;
+    constructor(host: WorkspaceServiceHost);
+    resolveForRun(run: RunRecord): Promise<Workspace>;
+    loadManifest(workspace: Workspace, versionId?: string | null | undefined): Promise<{
+        version?: WorkspaceVersion;
+        manifest: WorkspaceManifest;
+    }>;
+    hydrate(run: RunRecord, sandbox: SandboxSession, sessionKey: string): Promise<WorkspaceHydrationResult>;
+    sync(input: {
+        run: RunRecord;
+        sandbox: SandboxSession;
+        syncJobId: string;
+        sessionKey: string;
+        workspaceId: string;
+        baseVersionId: string | null;
+    }): Promise<WorkspaceSyncResult>;
+    status(workspaceId: string): Promise<WorkspaceStatus>;
+    statusForRun(run: RunRecord): Promise<WorkspaceStatus>;
+    retention(workspaceId: string, input?: {
+        tailCount?: number;
+        dryRun?: boolean;
+    }): Promise<WorkspaceRetentionReport>;
+    reachability(): Promise<WorkspaceReachabilityReport>;
+    garbageCollect(input?: {
+        dryRun?: boolean;
+        minAgeMs?: number;
+    }): Promise<WorkspaceGarbageCollectionReport>;
+    usage(): Promise<WorkspaceUsageReport>;
+    diagnostics(): Promise<WorkspaceDiagnostics>;
+    verify(workspaceId?: string): Promise<WorkspaceVerificationReport>;
+    repairBlob(input: {
+        workspaceId: string;
+        versionId: string;
+        path?: string;
+        replacement: Uint8Array;
+    }): Promise<{
+        workspaceId: string;
+        versionId: string;
+        path?: string;
+        blobKey: string;
+        sha256: string;
+        sizeBytes: number;
+    }>;
+    repairHead(input: {
+        workspaceId: string;
+        targetVersionId: string;
+        expectedHeadVersionId?: string | null;
+    }): Promise<Workspace>;
+    readFile(workspaceId: string, path: string, versionId?: string): Promise<WorkspaceCommittedFile>;
+    listFiles(workspaceId: string, input?: {
+        versionId?: string;
+        prefix?: string;
+    }): Promise<{
+        workspaceId: string;
+        versionId: string;
+        entries: WorkspaceManifestEntry[];
+    }>;
+    grep(workspaceId: string, pattern: string, input?: {
+        versionId?: string;
+        literal?: boolean;
+        caseSensitive?: boolean;
+        limit?: number;
+    }): Promise<{
+        workspaceId: string;
+        versionId: string;
+        results: WorkspaceGrepResult[];
+    }>;
+    indexVersion(workspaceId: string, versionId?: string): Promise<WorkspaceSearchIndex & {
+        excluded: Array<{
+            path: string;
+            reason: "binary" | "oversized" | "ignored" | "secret";
+        }>;
+    }>;
+    search(workspaceId: string, query: string, input?: {
+        versionId?: string;
+        limit?: number;
+    }): Promise<WorkspaceSearchResponse>;
+    createCheckpoint(input: {
+        workspaceId: string;
+        name: string;
+        replace?: boolean;
+        runId?: string;
+    }): Promise<WorkspaceCheckpoint>;
+    listCheckpoints(workspaceId: string): Promise<WorkspaceCheckpoint[]>;
+    getCheckpoint(workspaceId: string, name: string): Promise<WorkspaceCheckpoint | undefined>;
+    deleteCheckpoint(input: {
+        workspaceId: string;
+        name: string;
+        runId?: string;
+    }): Promise<boolean>;
+    restoreCheckpoint(input: {
+        workspaceId: string;
+        name: string;
+        idempotencyKey?: string;
+        runId?: string;
+    }): Promise<WorkspaceCommitResult>;
+    fork(input: {
+        sourceWorkspaceId: string;
+        sourceVersionId?: string;
+        checkpointName?: string;
+        workspaceId?: string;
+        name?: string;
+        tenantId?: string;
+    }): Promise<Workspace>;
+    private tryIndexVersion;
+    private requireWorkspace;
+    private isInheritedForkManifest;
+    private operations;
+    private assertVersionSize;
+}
+export declare function workspaceContentBlobKey(sha256: string): string;
+export declare function workspaceManifestBlobKey(sha256: string): string;
+//# sourceMappingURL=workspace-service.d.ts.map

package/dist/workspace-service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace-service.d.ts","sourceRoot":"","sources":["../src/workspace-service.ts"],"names":[],"mappings":"AACA,OAAO,EAAsD,KAAK,cAAc,EAAE,MAAM,2BAA2B,CAAC;AACpH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,KAAK,EACV,uBAAuB,EACvB,oBAAoB,EACpB,SAAS,EACT,SAAS,EACT,mBAAmB,EACnB,qBAAqB,EACrB,iBAAiB,EACjB,sBAAsB,EACtB,oBAAoB,EACpB,qBAAqB,EACrB,gBAAgB,EACjB,MAAM,YAAY,CAAC;AAEpB,OAAO,EAQL,KAAK,eAAe,EACrB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAML,KAAK,qBAAqB,EAC3B,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAEL,KAAK,oBAAoB,EACzB,KAAK,gCAAgC,EACrC,KAAK,2BAA2B,EAChC,KAAK,wBAAwB,EAC7B,KAAK,oBAAoB,EACzB,KAAK,2BAA2B,EACjC,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAEL,KAAK,yBAAyB,EAC/B,MAAM,kCAAkC,CAAC;AAG1C,MAAM,WAAW,oBAAoB;IACnC,KAAK,EAAE,oBAAoB,CAAC;IAC5B,IAAI,EAAE,WAAW,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,GAAG,IAAI,IAAI,CAAC;IACZ,MAAM,CAAC,EAAE,eAAe,CAAC;IACzB,YAAY,CAAC,EAAE,qBAAqB,CAAC;IACrC,iBAAiB,CAAC,EAAE,uBAAuB,CAAC;IAC5C,gBAAgB,CAAC,EAAE,yBAAyB,CAAC;IAC7C,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,wBAAwB;IACvC,SAAS,EAAE,SAAS,CAAC;IACrB,OAAO,CAAC,EAAE,gBAAgB,CAAC;IAC3B,QAAQ,EAAE,iBAAiB,CAAC;IAC5B,YAAY,EAAE,OAAO,CAAC;CACvB;AAED,MAAM,WAAW,mBAAoB,SAAQ,qBAAqB;IAChE,QAAQ,EAAE,iBAAiB,CAAC;IAC5B,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,YAAY,EAAE,MAAM,EAAE,CAAC;CACxB;AAED,MAAM,WAAW,eAAe;IAC9B,SAAS,EAAE,SAAS,CAAC;IACrB,IAAI,CAAC,EAAE,gBAAgB,CAAC;IACxB,WAAW,EAAE,mBAAmB,EAAE,CAAC;IACnC,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,sBAAsB;IACrC,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,sBAAsB,CAAC;IAC9B,QAAQ,EAAE,UAAU,CAAC;CACtB;AAED,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,uBAAuB;IACtC,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,UAAU,EAAE,OAAO,CAAC;IACpB,OAAO,EAAE,qBAAqB,EAAE,CAAC;CAClC;AAED,qBAAa,gBAAgB;IAMf,OAAO,CAAC,QAAQ,CAAC,IAAI;IALjC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAkB;IACzC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAwB;IACrD,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAA4B;IAC7D,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAS;gBAEX,IAAI,EAAE,oBAAoB;IAOjD,aAAa,CAAC,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;IAIjD,YAAY,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,GAAE,MAAM,GAAG,IAAI,GAAG,SAAmC,GAAG,OAAO,CAAC;QAChH,OAAO,CAAC,EAAE,gBAAgB,CAAC;QAC3B,QAAQ,EAAE,iBAAiB,CAAC;KAC7B,CAAC;IAsBI,OAAO,CAAC,GAAG,EAAE,SAAS,EAAE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,wBAAwB,CAAC;IAmDvG,IAAI,CAAC,KAAK,EAAE;QAChB,GAAG,EAAE,SAAS,CAAC;QACf,OAAO,EAAE,cAAc,CAAC;QACxB,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,EAAE,MAAM,CAAC;QACpB,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;KAC9B,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAiF1B,MAAM,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;IAcrD,YAAY,CAAC,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,eAAe,CAAC;IAI5D,SAAS,CAAC,WAAW,EAAE,MAAM,EAAE,KAAK,GAAE;QAAE,SAAS,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,OAAO,CAAA;KAAO,GAAG,OAAO,CAAC,wBAAwB,CAAC;IAIvH,YAAY,IAAI,OAAO,CAAC,2BAA2B,CAAC;IAIpD,cAAc,CAAC,KAAK,GAAE;QAAE,MAAM,CAAC,EAAE,OAAO,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAO,GAAG,OAAO,CAAC,gCAAgC,CAAC;IAI9G,KAAK,IAAI,OAAO,CAAC,oBAAoB,CAAC;IAItC,WAAW,IAAI,OAAO,CAAC,oBAAoB,CAAC;IAI5C,MAAM,CAAC,WAAW,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,2BAA2B,CAAC;IAIlE,UAAU,CAAC,KAAK,EAAE;QAChB,WAAW,EAAE,MAAM,CAAC;QACpB,SAAS,EAAE,MAAM,CAAC;QAClB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,WAAW,EAAE,UAAU,CAAC;KACzB;;;;;;;;IAID,UAAU,CAAC,KAAK,EAAE;QAChB,WAAW,EAAE,MAAM,CAAC;QACpB,eAAe,EAAE,MAAM,CAAC;QACxB,qBAAqB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;KACvC,GAAG,OAAO,CAAC,SAAS,CAAC;IAIhB,QAAQ,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAehG,SAAS,CAAC,WAAW,EAAE,MAAM,EAAE,KAAK,GAAE;QAAE,SAAS,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAO,GAAG,OAAO,CAAC;QACjG,WAAW,EAAE,MAAM,CAAC;QACpB,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,sBAAsB,EAAE,CAAC;KACnC,CAAC;IAWI,IAAI,CAAC,WAAW,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,GAAE;QACtD,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,OAAO,CAAC,EAAE,OAAO,CAAC;QAClB,aAAa,CAAC,EAAE,OAAO,CAAC;QACxB,KAAK,CAAC,EAAE,MAAM,CAAC;KACX,GAAG,OAAO,CAAC;QAAE,WAAW,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,mBAAmB,EAAE,CAAA;KAAE,CAAC;IA+BtF,YAAY,CAAC,WAAW,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,oBAAoB,GAAG;QAC1F,QAAQ,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,QAAQ,GAAG,WAAW,GAAG,SAAS,GAAG,QAAQ,CAAA;SAAE,CAAC,CAAC;KAC1F,CAAC;IAsBI,MAAM,CAAC,WAAW,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,GAAE;QAAE,SAAS,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAO,GAAG,OAAO,CAAC,uBAAuB,CAAC;IAuDhI,gBAAgB,CAAC,KAAK,EAAE;QAC5B,WAAW,EAAE,MAAM,CAAC;QACpB,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,CAAC,EAAE,OAAO,CAAC;QAClB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,OAAO,CAAC,mBAAmB,CAAC;IA+B1B,eAAe,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAKpE,aAAa,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,GAAG,SAAS,CAAC;IAK1F,gBAAgB,CAAC,KAAK,EAAE;QAAE,WAAW,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,OAAO,CAAC;IAahG,iBAAiB,CAAC,KAAK,EAAE;QAC7B,WAAW,EAAE,MAAM,CAAC;QACpB,IAAI,EAAE,MAAM,CAAC;QACb,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAuC5B,IAAI,CAAC,KAAK,EAAE;QAChB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,GAAG,OAAO,CAAC,SAAS,CAAC;YAsCR,eAAe;YAuBf,gBAAgB;YAShB,uBAAuB;IASrC,OAAO,CAAC,UAAU;IAWlB,OAAO,CAAC,iBAAiB;CAM1B;AAED,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAE9D;AAED,wBAAgB,wBAAwB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAE/D"}