weave-typescript 0.49.0 → 0.51.0

package/dist/weavesql/weavedb/document_table_sql.js

@@ -0,0 +1,1352 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.retryDocumentBlobDeletionJobQuery = exports.completeDocumentBlobDeletionJobQuery = exports.claimDocumentBlobDeletionJobQuery = exports.listReadyDocumentBlobDeletionJobsQuery = exports.enqueueDocumentBlobDeletionJobQuery = exports.listSessionTabularRetrievalContextQuery = exports.insertSessionTabularRetrievalContextQuery = exports.findDocumentTableArtifactScopeMismatchesQuery = exports.loadAuthorizedDocumentTableArtifactQuery = exports.deleteDocumentTableMetadataByDocumentIDQuery = exports.listDocumentBlobRefsForDeletionQuery = exports.listCandidateDocumentTablesByDocumentIDsQuery = exports.listCandidateDocumentTablesByIDsQuery = exports.searchCandidateDocumentTablesByMetadataQuery = exports.listThreadTableLinksByThreadIDsQuery = exports.attachThreadTableLinkQuery = exports.getDocumentTableByPermissionQuery = exports.listDocumentTablesByPermissionQuery = exports.markDocumentTablesInactiveQuery = exports.setDocumentTableActiveQuery = exports.createDocumentTableQuery = void 0;
+exports.createDocumentTable = createDocumentTable;
+exports.setDocumentTableActive = setDocumentTableActive;
+exports.listDocumentTablesByPermission = listDocumentTablesByPermission;
+exports.getDocumentTableByPermission = getDocumentTableByPermission;
+exports.attachThreadTableLink = attachThreadTableLink;
+exports.listThreadTableLinksByThreadIDs = listThreadTableLinksByThreadIDs;
+exports.searchCandidateDocumentTablesByMetadata = searchCandidateDocumentTablesByMetadata;
+exports.listCandidateDocumentTablesByIDs = listCandidateDocumentTablesByIDs;
+exports.listCandidateDocumentTablesByDocumentIDs = listCandidateDocumentTablesByDocumentIDs;
+exports.listDocumentBlobRefsForDeletion = listDocumentBlobRefsForDeletion;
+exports.loadAuthorizedDocumentTableArtifact = loadAuthorizedDocumentTableArtifact;
+exports.findDocumentTableArtifactScopeMismatches = findDocumentTableArtifactScopeMismatches;
+exports.insertSessionTabularRetrievalContext = insertSessionTabularRetrievalContext;
+exports.listSessionTabularRetrievalContext = listSessionTabularRetrievalContext;
+exports.enqueueDocumentBlobDeletionJob = enqueueDocumentBlobDeletionJob;
+exports.listReadyDocumentBlobDeletionJobs = listReadyDocumentBlobDeletionJobs;
+exports.claimDocumentBlobDeletionJob = claimDocumentBlobDeletionJob;
+exports.createDocumentTableQuery = `-- name: CreateDocumentTable :one
+INSERT INTO weave.document_tables (
+    id,
+    organization_id,
+    document_id,
+    name,
+    description,
+    summary,
+    source_format,
+    parser_name,
+    table_version,
+    active,
+    artifact_id,
+    artifact_version,
+    artifact_storage_ref,
+    manifest_storage_ref,
+    manifest_version,
+    manifest_sha256,
+    manifest_size_bytes,
+    artifact_scope_kind,
+    artifact_scope_key,
+    authorization_snapshot,
+    safe_table_alias,
+    row_count,
+    column_count,
+    search_text,
+    source_location,
+    metadata
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6,
+    $7,
+    $8,
+    $9,
+    $10,
+    $11,
+    $12,
+    $13,
+    $14,
+    $15,
+    $16,
+    $17,
+    $18,
+    $19,
+    $20,
+    $21,
+    $22,
+    $23,
+    $24,
+    $25,
+    $26
+)
+RETURNING
+    id,
+    organization_id,
+    document_id,
+    name,
+    description,
+    summary,
+    source_format,
+    parser_name,
+    table_version,
+    active,
+    artifact_id,
+    artifact_version,
+    artifact_storage_ref,
+    manifest_storage_ref,
+    manifest_version,
+    manifest_sha256,
+    manifest_size_bytes,
+    artifact_scope_kind,
+    artifact_scope_key,
+    authorization_snapshot,
+    safe_table_alias,
+    row_count,
+    column_count,
+    search_text,
+    source_location,
+    metadata,
+    created_at,
+    updated_at`;
+async function createDocumentTable(client, args) {
+    const result = await client.query({
+        text: exports.createDocumentTableQuery,
+        values: [args.id, args.organizationId, args.documentId, args.name, args.description, args.summary, args.sourceFormat, args.parserName, args.tableVersion, args.active, args.artifactId, args.artifactVersion, args.artifactStorageRef, args.manifestStorageRef, args.manifestVersion, args.manifestSha256, args.manifestSizeBytes, args.artifactScopeKind, args.artifactScopeKey, args.authorizationSnapshot, args.safeTableAlias, args.rowCount, args.columnCount, args.searchText, args.sourceLocation, args.metadata],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        documentId: row[2],
+        name: row[3],
+        description: row[4],
+        summary: row[5],
+        sourceFormat: row[6],
+        parserName: row[7],
+        tableVersion: row[8],
+        active: row[9],
+        artifactId: row[10],
+        artifactVersion: row[11],
+        artifactStorageRef: row[12],
+        manifestStorageRef: row[13],
+        manifestVersion: row[14],
+        manifestSha256: row[15],
+        manifestSizeBytes: row[16],
+        artifactScopeKind: row[17],
+        artifactScopeKey: row[18],
+        authorizationSnapshot: row[19],
+        safeTableAlias: row[20],
+        rowCount: row[21],
+        columnCount: row[22],
+        searchText: row[23],
+        sourceLocation: row[24],
+        metadata: row[25],
+        createdAt: row[26],
+        updatedAt: row[27]
+    };
+}
+exports.setDocumentTableActiveQuery = `-- name: SetDocumentTableActive :one
+UPDATE weave.document_tables
+SET
+    active = $1,
+    artifact_id = $2,
+    artifact_version = $3,
+    artifact_storage_ref = $4,
+    manifest_storage_ref = $5,
+    manifest_version = $6,
+    manifest_sha256 = $7,
+    manifest_size_bytes = $8,
+    artifact_scope_kind = $9,
+    artifact_scope_key = $10,
+    authorization_snapshot = $11,
+    row_count = $12,
+    column_count = $13,
+    search_text = $14,
+    metadata = $15,
+    updated_at = now()
+WHERE organization_id = $16
+  AND document_id = $17
+  AND id = $18
+RETURNING
+    id,
+    organization_id,
+    document_id,
+    name,
+    description,
+    summary,
+    source_format,
+    parser_name,
+    table_version,
+    active,
+    artifact_id,
+    artifact_version,
+    artifact_storage_ref,
+    manifest_storage_ref,
+    manifest_version,
+    manifest_sha256,
+    manifest_size_bytes,
+    artifact_scope_kind,
+    artifact_scope_key,
+    authorization_snapshot,
+    safe_table_alias,
+    row_count,
+    column_count,
+    search_text,
+    source_location,
+    metadata,
+    created_at,
+    updated_at`;
+async function setDocumentTableActive(client, args) {
+    const result = await client.query({
+        text: exports.setDocumentTableActiveQuery,
+        values: [args.active, args.artifactId, args.artifactVersion, args.artifactStorageRef, args.manifestStorageRef, args.manifestVersion, args.manifestSha256, args.manifestSizeBytes, args.artifactScopeKind, args.artifactScopeKey, args.authorizationSnapshot, args.rowCount, args.columnCount, args.searchText, args.metadata, args.organizationId, args.documentId, args.tableId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        documentId: row[2],
+        name: row[3],
+        description: row[4],
+        summary: row[5],
+        sourceFormat: row[6],
+        parserName: row[7],
+        tableVersion: row[8],
+        active: row[9],
+        artifactId: row[10],
+        artifactVersion: row[11],
+        artifactStorageRef: row[12],
+        manifestStorageRef: row[13],
+        manifestVersion: row[14],
+        manifestSha256: row[15],
+        manifestSizeBytes: row[16],
+        artifactScopeKind: row[17],
+        artifactScopeKey: row[18],
+        authorizationSnapshot: row[19],
+        safeTableAlias: row[20],
+        rowCount: row[21],
+        columnCount: row[22],
+        searchText: row[23],
+        sourceLocation: row[24],
+        metadata: row[25],
+        createdAt: row[26],
+        updatedAt: row[27]
+    };
+}
+exports.markDocumentTablesInactiveQuery = `-- name: MarkDocumentTablesInactive :execrows
+UPDATE weave.document_tables
+SET
+    active = FALSE,
+    updated_at = now()
+WHERE organization_id = $1
+  AND document_id = $2
+  AND active = TRUE`;
+exports.listDocumentTablesByPermissionQuery = `-- name: ListDocumentTablesByPermission :many
+SELECT
+    dt.id,
+    dt.organization_id,
+    dt.document_id,
+    d.filename AS document_name,
+    dt.name,
+    dt.description,
+    dt.summary,
+    dt.source_format,
+    dt.parser_name,
+    dt.table_version,
+    dt.active,
+    dt.artifact_id,
+    dt.artifact_version,
+    dt.artifact_storage_ref,
+    dt.manifest_storage_ref,
+    dt.manifest_version,
+    dt.manifest_sha256,
+    dt.manifest_size_bytes,
+    dt.artifact_scope_kind,
+    dt.artifact_scope_key,
+    dt.authorization_snapshot,
+    dt.safe_table_alias,
+    dt.row_count,
+    dt.column_count,
+    dt.search_text,
+    dt.source_location,
+    dt.metadata,
+    dt.created_at,
+    dt.updated_at
+FROM weave.document_tables dt
+JOIN weave.documents d
+  ON d.organization_id = dt.organization_id
+ AND d.id = dt.document_id
+WHERE dt.organization_id = $1
+  AND dt.document_id = $2
+  AND ($3::boolean OR dt.active = TRUE)
+  AND (
+      d.scope = 'DOCUMENT_SCOPE_ORG'
+      OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'
+          AND d.uploaded_by_user_id = $4)
+      OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'
+          AND (
+          EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_user_memberships rgum
+                ON rgum.organization_id = drg.organization_id
+               AND rgum.restricted_group_id = drg.restricted_group_id
+               AND rgum.user_id = $4
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )
+          OR EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_team_memberships rgtm
+                ON rgtm.organization_id = drg.organization_id
+               AND rgtm.restricted_group_id = drg.restricted_group_id
+              JOIN weave.org_team_memberships otm
+                ON otm.organization_id = rgtm.organization_id
+               AND otm.org_team_id = rgtm.org_team_id
+               AND otm.user_id = $4
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )))
+  )
+ORDER BY dt.table_version DESC, dt.name ASC, dt.id ASC
+LIMIT $6 OFFSET $5`;
+async function listDocumentTablesByPermission(client, args) {
+    const result = await client.query({
+        text: exports.listDocumentTablesByPermissionQuery,
+        values: [args.organizationId, args.documentId, args.includeInactive, args.actingUserId, args.pageOffset, args.pageSize],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            organizationId: row[1],
+            documentId: row[2],
+            documentName: row[3],
+            name: row[4],
+            description: row[5],
+            summary: row[6],
+            sourceFormat: row[7],
+            parserName: row[8],
+            tableVersion: row[9],
+            active: row[10],
+            artifactId: row[11],
+            artifactVersion: row[12],
+            artifactStorageRef: row[13],
+            manifestStorageRef: row[14],
+            manifestVersion: row[15],
+            manifestSha256: row[16],
+            manifestSizeBytes: row[17],
+            artifactScopeKind: row[18],
+            artifactScopeKey: row[19],
+            authorizationSnapshot: row[20],
+            safeTableAlias: row[21],
+            rowCount: row[22],
+            columnCount: row[23],
+            searchText: row[24],
+            sourceLocation: row[25],
+            metadata: row[26],
+            createdAt: row[27],
+            updatedAt: row[28]
+        };
+    });
+}
+exports.getDocumentTableByPermissionQuery = `-- name: GetDocumentTableByPermission :one
+SELECT
+    dt.id,
+    dt.organization_id,
+    dt.document_id,
+    d.filename AS document_name,
+    dt.name,
+    dt.description,
+    dt.summary,
+    dt.source_format,
+    dt.parser_name,
+    dt.table_version,
+    dt.active,
+    dt.artifact_id,
+    dt.artifact_version,
+    dt.artifact_storage_ref,
+    dt.manifest_storage_ref,
+    dt.manifest_version,
+    dt.manifest_sha256,
+    dt.manifest_size_bytes,
+    dt.artifact_scope_kind,
+    dt.artifact_scope_key,
+    dt.authorization_snapshot,
+    dt.safe_table_alias,
+    dt.row_count,
+    dt.column_count,
+    dt.search_text,
+    dt.source_location,
+    dt.metadata,
+    dt.created_at,
+    dt.updated_at
+FROM weave.document_tables dt
+JOIN weave.documents d
+  ON d.organization_id = dt.organization_id
+ AND d.id = dt.document_id
+WHERE dt.organization_id = $1
+  AND dt.document_id = $2
+  AND dt.id = $3
+  AND dt.active = TRUE
+  AND (
+      d.scope = 'DOCUMENT_SCOPE_ORG'
+      OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'
+          AND d.uploaded_by_user_id = $4)
+      OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'
+          AND (
+          EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_user_memberships rgum
+                ON rgum.organization_id = drg.organization_id
+               AND rgum.restricted_group_id = drg.restricted_group_id
+               AND rgum.user_id = $4
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )
+          OR EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_team_memberships rgtm
+                ON rgtm.organization_id = drg.organization_id
+               AND rgtm.restricted_group_id = drg.restricted_group_id
+              JOIN weave.org_team_memberships otm
+                ON otm.organization_id = rgtm.organization_id
+               AND otm.org_team_id = rgtm.org_team_id
+               AND otm.user_id = $4
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )))
+  )`;
+async function getDocumentTableByPermission(client, args) {
+    const result = await client.query({
+        text: exports.getDocumentTableByPermissionQuery,
+        values: [args.organizationId, args.documentId, args.tableId, args.actingUserId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        documentId: row[2],
+        documentName: row[3],
+        name: row[4],
+        description: row[5],
+        summary: row[6],
+        sourceFormat: row[7],
+        parserName: row[8],
+        tableVersion: row[9],
+        active: row[10],
+        artifactId: row[11],
+        artifactVersion: row[12],
+        artifactStorageRef: row[13],
+        manifestStorageRef: row[14],
+        manifestVersion: row[15],
+        manifestSha256: row[16],
+        manifestSizeBytes: row[17],
+        artifactScopeKind: row[18],
+        artifactScopeKey: row[19],
+        authorizationSnapshot: row[20],
+        safeTableAlias: row[21],
+        rowCount: row[22],
+        columnCount: row[23],
+        searchText: row[24],
+        sourceLocation: row[25],
+        metadata: row[26],
+        createdAt: row[27],
+        updatedAt: row[28]
+    };
+}
+exports.attachThreadTableLinkQuery = `-- name: AttachThreadTableLink :one
+INSERT INTO weave.thread_table_links (
+    organization_id,
+    document_id,
+    thread_id,
+    table_id,
+    artifact_id,
+    artifact_version,
+    table_name,
+    table_summary,
+    safe_table_alias,
+    source_location,
+    authorization_snapshot
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6,
+    $7,
+    $8,
+    $9,
+    $10,
+    $11
+)
+ON CONFLICT (thread_id, table_id) DO UPDATE
+SET
+    artifact_id = EXCLUDED.artifact_id,
+    artifact_version = EXCLUDED.artifact_version,
+    table_name = EXCLUDED.table_name,
+    table_summary = EXCLUDED.table_summary,
+    safe_table_alias = EXCLUDED.safe_table_alias,
+    source_location = EXCLUDED.source_location,
+    authorization_snapshot = EXCLUDED.authorization_snapshot
+RETURNING
+    organization_id,
+    document_id,
+    thread_id,
+    table_id,
+    artifact_id,
+    artifact_version,
+    table_name,
+    table_summary,
+    safe_table_alias,
+    source_location,
+    authorization_snapshot,
+    created_at`;
+async function attachThreadTableLink(client, args) {
+    const result = await client.query({
+        text: exports.attachThreadTableLinkQuery,
+        values: [args.organizationId, args.documentId, args.threadId, args.tableId, args.artifactId, args.artifactVersion, args.tableName, args.tableSummary, args.safeTableAlias, args.sourceLocation, args.authorizationSnapshot],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        organizationId: row[0],
+        documentId: row[1],
+        threadId: row[2],
+        tableId: row[3],
+        artifactId: row[4],
+        artifactVersion: row[5],
+        tableName: row[6],
+        tableSummary: row[7],
+        safeTableAlias: row[8],
+        sourceLocation: row[9],
+        authorizationSnapshot: row[10],
+        createdAt: row[11]
+    };
+}
+exports.listThreadTableLinksByThreadIDsQuery = `-- name: ListThreadTableLinksByThreadIDs :many
+SELECT
+    ttl.organization_id,
+    ttl.document_id,
+    ttl.thread_id,
+    ttl.table_id,
+    ttl.artifact_id,
+    ttl.artifact_version,
+    ttl.table_name,
+    ttl.table_summary,
+    ttl.safe_table_alias,
+    ttl.source_location,
+    ttl.authorization_snapshot,
+    ttl.created_at,
+    dt.source_format,
+    dt.parser_name,
+    dt.row_count,
+    dt.column_count,
+    dt.manifest_storage_ref,
+    dt.manifest_version,
+    dt.manifest_sha256,
+    dt.manifest_size_bytes
+FROM weave.thread_table_links ttl
+JOIN weave.document_tables dt
+  ON dt.organization_id = ttl.organization_id
+ AND dt.id = ttl.table_id
+ AND dt.active = TRUE
+WHERE ttl.organization_id = $1
+  AND ttl.thread_id = ANY($2::uuid[])
+ORDER BY ttl.thread_id ASC, ttl.created_at ASC`;
+async function listThreadTableLinksByThreadIDs(client, args) {
+    const result = await client.query({
+        text: exports.listThreadTableLinksByThreadIDsQuery,
+        values: [args.organizationId, args.threadIds],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            organizationId: row[0],
+            documentId: row[1],
+            threadId: row[2],
+            tableId: row[3],
+            artifactId: row[4],
+            artifactVersion: row[5],
+            tableName: row[6],
+            tableSummary: row[7],
+            safeTableAlias: row[8],
+            sourceLocation: row[9],
+            authorizationSnapshot: row[10],
+            createdAt: row[11],
+            sourceFormat: row[12],
+            parserName: row[13],
+            rowCount: row[14],
+            columnCount: row[15],
+            manifestStorageRef: row[16],
+            manifestVersion: row[17],
+            manifestSha256: row[18],
+            manifestSizeBytes: row[19]
+        };
+    });
+}
+exports.searchCandidateDocumentTablesByMetadataQuery = `-- name: SearchCandidateDocumentTablesByMetadata :many
+SELECT
+    dt.id,
+    dt.organization_id,
+    dt.document_id,
+    d.filename AS document_name,
+    dt.name,
+    dt.description,
+    dt.summary,
+    dt.source_format,
+    dt.parser_name,
+    dt.table_version,
+    dt.artifact_id,
+    dt.artifact_version,
+    dt.artifact_storage_ref,
+    dt.manifest_storage_ref,
+    dt.manifest_version,
+    dt.manifest_sha256,
+    dt.manifest_size_bytes,
+    dt.artifact_scope_kind,
+    dt.artifact_scope_key,
+    dt.authorization_snapshot,
+    dt.safe_table_alias,
+    dt.row_count,
+    dt.column_count,
+    dt.source_location,
+    ts_rank_cd(
+        to_tsvector('simple', dt.search_text),
+        plainto_tsquery('simple', $1)
+    )::real AS score
+FROM weave.document_tables dt
+JOIN weave.documents d
+  ON d.organization_id = dt.organization_id
+ AND d.id = dt.document_id
+WHERE dt.organization_id = $2
+  AND dt.active = TRUE
+  AND (
+      d.scope = 'DOCUMENT_SCOPE_ORG'
+      OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'
+          AND d.uploaded_by_user_id = $3)
+      OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'
+          AND (
+          EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_user_memberships rgum
+                ON rgum.organization_id = drg.organization_id
+               AND rgum.restricted_group_id = drg.restricted_group_id
+               AND rgum.user_id = $3
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )
+          OR EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_team_memberships rgtm
+                ON rgtm.organization_id = drg.organization_id
+               AND rgtm.restricted_group_id = drg.restricted_group_id
+              JOIN weave.org_team_memberships otm
+                ON otm.organization_id = rgtm.organization_id
+               AND otm.org_team_id = rgtm.org_team_id
+               AND otm.user_id = $3
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )))
+  )
+  AND to_tsvector('simple', dt.search_text) @@ plainto_tsquery('simple', $1)
+ORDER BY score DESC, dt.updated_at DESC
+LIMIT $4`;
+async function searchCandidateDocumentTablesByMetadata(client, args) {
+    const result = await client.query({
+        text: exports.searchCandidateDocumentTablesByMetadataQuery,
+        values: [args.query, args.organizationId, args.actingUserId, args.resultLimit],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            organizationId: row[1],
+            documentId: row[2],
+            documentName: row[3],
+            name: row[4],
+            description: row[5],
+            summary: row[6],
+            sourceFormat: row[7],
+            parserName: row[8],
+            tableVersion: row[9],
+            artifactId: row[10],
+            artifactVersion: row[11],
+            artifactStorageRef: row[12],
+            manifestStorageRef: row[13],
+            manifestVersion: row[14],
+            manifestSha256: row[15],
+            manifestSizeBytes: row[16],
+            artifactScopeKind: row[17],
+            artifactScopeKey: row[18],
+            authorizationSnapshot: row[19],
+            safeTableAlias: row[20],
+            rowCount: row[21],
+            columnCount: row[22],
+            sourceLocation: row[23],
+            score: row[24]
+        };
+    });
+}
+exports.listCandidateDocumentTablesByIDsQuery = `-- name: ListCandidateDocumentTablesByIDs :many
+SELECT
+    dt.id,
+    dt.organization_id,
+    dt.document_id,
+    d.filename AS document_name,
+    dt.name,
+    dt.summary,
+    dt.source_format,
+    dt.parser_name,
+    dt.artifact_id,
+    dt.artifact_version,
+    dt.artifact_storage_ref,
+    dt.manifest_storage_ref,
+    dt.manifest_version,
+    dt.manifest_sha256,
+    dt.manifest_size_bytes,
+    dt.artifact_scope_kind,
+    dt.artifact_scope_key,
+    dt.authorization_snapshot,
+    dt.safe_table_alias,
+    dt.row_count,
+    dt.column_count,
+    dt.source_location
+FROM weave.document_tables dt
+JOIN weave.documents d
+  ON d.organization_id = dt.organization_id
+ AND d.id = dt.document_id
+WHERE dt.organization_id = $1
+  AND dt.active = TRUE
+  AND dt.id = ANY($2::uuid[])
+  AND (
+      d.scope = 'DOCUMENT_SCOPE_ORG'
+      OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'
+          AND d.uploaded_by_user_id = $3)
+      OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'
+          AND (
+          EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_user_memberships rgum
+                ON rgum.organization_id = drg.organization_id
+               AND rgum.restricted_group_id = drg.restricted_group_id
+               AND rgum.user_id = $3
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )
+          OR EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_team_memberships rgtm
+                ON rgtm.organization_id = drg.organization_id
+               AND rgtm.restricted_group_id = drg.restricted_group_id
+              JOIN weave.org_team_memberships otm
+                ON otm.organization_id = rgtm.organization_id
+               AND otm.org_team_id = rgtm.org_team_id
+               AND otm.user_id = $3
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )))
+  )
+ORDER BY dt.updated_at DESC, dt.id ASC
+LIMIT $4`;
+async function listCandidateDocumentTablesByIDs(client, args) {
+    const result = await client.query({
+        text: exports.listCandidateDocumentTablesByIDsQuery,
+        values: [args.organizationId, args.tableIds, args.actingUserId, args.resultLimit],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            organizationId: row[1],
+            documentId: row[2],
+            documentName: row[3],
+            name: row[4],
+            summary: row[5],
+            sourceFormat: row[6],
+            parserName: row[7],
+            artifactId: row[8],
+            artifactVersion: row[9],
+            artifactStorageRef: row[10],
+            manifestStorageRef: row[11],
+            manifestVersion: row[12],
+            manifestSha256: row[13],
+            manifestSizeBytes: row[14],
+            artifactScopeKind: row[15],
+            artifactScopeKey: row[16],
+            authorizationSnapshot: row[17],
+            safeTableAlias: row[18],
+            rowCount: row[19],
+            columnCount: row[20],
+            sourceLocation: row[21]
+        };
+    });
+}
+exports.listCandidateDocumentTablesByDocumentIDsQuery = `-- name: ListCandidateDocumentTablesByDocumentIDs :many
+SELECT
+    dt.id,
+    dt.organization_id,
+    dt.document_id,
+    d.filename AS document_name,
+    dt.name,
+    dt.summary,
+    dt.source_format,
+    dt.parser_name,
+    dt.artifact_id,
+    dt.artifact_version,
+    dt.artifact_storage_ref,
+    dt.manifest_storage_ref,
+    dt.manifest_version,
+    dt.manifest_sha256,
+    dt.manifest_size_bytes,
+    dt.artifact_scope_kind,
+    dt.artifact_scope_key,
+    dt.authorization_snapshot,
+    dt.safe_table_alias,
+    dt.row_count,
+    dt.column_count,
+    dt.source_location
+FROM weave.document_tables dt
+JOIN weave.documents d
+  ON d.organization_id = dt.organization_id
+ AND d.id = dt.document_id
+WHERE dt.organization_id = $1
+  AND dt.active = TRUE
+  AND dt.document_id = ANY($2::uuid[])
+  AND (
+      d.scope = 'DOCUMENT_SCOPE_ORG'
+      OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'
+          AND d.uploaded_by_user_id = $3)
+      OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'
+          AND (
+          EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_user_memberships rgum
+                ON rgum.organization_id = drg.organization_id
+               AND rgum.restricted_group_id = drg.restricted_group_id
+               AND rgum.user_id = $3
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )
+          OR EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_team_memberships rgtm
+                ON rgtm.organization_id = drg.organization_id
+               AND rgtm.restricted_group_id = drg.restricted_group_id
+              JOIN weave.org_team_memberships otm
+                ON otm.organization_id = rgtm.organization_id
+               AND otm.org_team_id = rgtm.org_team_id
+               AND otm.user_id = $3
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )))
+  )
+ORDER BY dt.updated_at DESC, dt.id ASC
+LIMIT $4`;
+async function listCandidateDocumentTablesByDocumentIDs(client, args) {
+    const result = await client.query({
+        text: exports.listCandidateDocumentTablesByDocumentIDsQuery,
+        values: [args.organizationId, args.documentIds, args.actingUserId, args.resultLimit],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            organizationId: row[1],
+            documentId: row[2],
+            documentName: row[3],
+            name: row[4],
+            summary: row[5],
+            sourceFormat: row[6],
+            parserName: row[7],
+            artifactId: row[8],
+            artifactVersion: row[9],
+            artifactStorageRef: row[10],
+            manifestStorageRef: row[11],
+            manifestVersion: row[12],
+            manifestSha256: row[13],
+            manifestSizeBytes: row[14],
+            artifactScopeKind: row[15],
+            artifactScopeKey: row[16],
+            authorizationSnapshot: row[17],
+            safeTableAlias: row[18],
+            rowCount: row[19],
+            columnCount: row[20],
+            sourceLocation: row[21]
+        };
+    });
+}
+exports.listDocumentBlobRefsForDeletionQuery = `-- name: ListDocumentBlobRefsForDeletion :many
+SELECT
+    d.id AS document_id,
+    'DOCUMENT_BLOB_KIND_SOURCE'::text AS blob_kind,
+    NULL::uuid AS artifact_id,
+    d.storage_ref
+FROM weave.documents d
+WHERE d.organization_id = $1
+  AND d.id = $2
+  AND d.storage_ref <> ''
+UNION ALL
+SELECT
+    dt.document_id,
+    'DOCUMENT_BLOB_KIND_TABLE_ARTIFACT'::text AS blob_kind,
+    dt.artifact_id,
+    dt.artifact_storage_ref AS storage_ref
+FROM weave.document_tables dt
+WHERE dt.organization_id = $1
+  AND dt.document_id = $2
+  AND dt.artifact_storage_ref <> ''
+UNION ALL
+SELECT
+    dt.document_id,
+    'DOCUMENT_BLOB_KIND_TABLE_MANIFEST'::text AS blob_kind,
+    dt.artifact_id,
+    dt.manifest_storage_ref AS storage_ref
+FROM weave.document_tables dt
+WHERE dt.organization_id = $1
+  AND dt.document_id = $2
+  AND dt.manifest_storage_ref <> ''
+ORDER BY blob_kind ASC, artifact_id ASC`;
+async function listDocumentBlobRefsForDeletion(client, args) {
+    const result = await client.query({
+        text: exports.listDocumentBlobRefsForDeletionQuery,
+        values: [args.organizationId, args.documentId],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            documentId: row[0],
+            blobKind: row[1],
+            artifactId: row[2],
+            storageRef: row[3]
+        };
+    });
+}
+exports.deleteDocumentTableMetadataByDocumentIDQuery = `-- name: DeleteDocumentTableMetadataByDocumentID :execrows
+DELETE FROM weave.document_tables
+WHERE organization_id = $1
+  AND document_id = $2`;
+exports.loadAuthorizedDocumentTableArtifactQuery = `-- name: LoadAuthorizedDocumentTableArtifact :one
+SELECT
+    dt.id,
+    dt.organization_id,
+    dt.document_id,
+    d.filename AS document_name,
+    dt.name,
+    dt.artifact_id,
+    dt.artifact_version,
+    dt.artifact_storage_ref,
+    dt.artifact_scope_kind,
+    dt.artifact_scope_key,
+    dt.authorization_snapshot,
+    dt.safe_table_alias,
+    dt.source_location
+FROM weave.document_tables dt
+JOIN weave.documents d
+  ON d.organization_id = dt.organization_id
+ AND d.id = dt.document_id
+WHERE dt.organization_id = $1
+  AND dt.active = TRUE
+  AND dt.artifact_id = $2
+  AND ($3::uuid IS NULL OR dt.document_id = $3::uuid)
+  AND ($4::uuid IS NULL OR dt.id = $4::uuid)
+  AND (
+      d.scope = 'DOCUMENT_SCOPE_ORG'
+      OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'
+          AND d.uploaded_by_user_id = $5)
+      OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'
+          AND (
+          EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_user_memberships rgum
+                ON rgum.organization_id = drg.organization_id
+               AND rgum.restricted_group_id = drg.restricted_group_id
+               AND rgum.user_id = $5
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )
+          OR EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_team_memberships rgtm
+                ON rgtm.organization_id = drg.organization_id
+               AND rgtm.restricted_group_id = drg.restricted_group_id
+              JOIN weave.org_team_memberships otm
+                ON otm.organization_id = rgtm.organization_id
+               AND otm.org_team_id = rgtm.org_team_id
+               AND otm.user_id = $5
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )))
+  )`;
+async function loadAuthorizedDocumentTableArtifact(client, args) {
+    const result = await client.query({
+        text: exports.loadAuthorizedDocumentTableArtifactQuery,
+        values: [args.organizationId, args.artifactId, args.documentId, args.tableId, args.actingUserId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        documentId: row[2],
+        documentName: row[3],
+        name: row[4],
+        artifactId: row[5],
+        artifactVersion: row[6],
+        artifactStorageRef: row[7],
+        artifactScopeKind: row[8],
+        artifactScopeKey: row[9],
+        authorizationSnapshot: row[10],
+        safeTableAlias: row[11],
+        sourceLocation: row[12]
+    };
+}
+exports.findDocumentTableArtifactScopeMismatchesQuery = `-- name: FindDocumentTableArtifactScopeMismatches :many
+SELECT
+    dt.id,
+    dt.organization_id,
+    dt.document_id,
+    dt.artifact_id,
+    dt.artifact_scope_kind,
+    dt.artifact_scope_key,
+    d.scope AS current_scope,
+    CASE
+        WHEN d.scope = 'DOCUMENT_SCOPE_ORG'
+            THEN ''::text
+        WHEN d.scope = 'DOCUMENT_SCOPE_PRIVATE'
+            THEN d.uploaded_by_user_id::text
+        WHEN d.scope = 'DOCUMENT_SCOPE_RESTRICTED'
+            THEN COALESCE(rg_scope.restricted_group_id::text, '')
+        ELSE ''::text
+    END AS current_scope_key
+FROM weave.document_tables dt
+JOIN weave.documents d
+  ON d.organization_id = dt.organization_id
+ AND d.id = dt.document_id
+LEFT JOIN LATERAL (
+    SELECT drg.restricted_group_id
+    FROM weave.document_restricted_groups drg
+    WHERE drg.organization_id = d.organization_id
+      AND drg.document_id = d.id
+    ORDER BY drg.restricted_group_id ASC
+    LIMIT 1
+) rg_scope ON TRUE
+WHERE dt.organization_id = $1
+  AND dt.active = TRUE
+  AND (
+      dt.artifact_scope_kind <> d.scope
+      OR dt.artifact_scope_key <> CASE
+          WHEN d.scope = 'DOCUMENT_SCOPE_ORG'
+              THEN ''::text
+          WHEN d.scope = 'DOCUMENT_SCOPE_PRIVATE'
+              THEN d.uploaded_by_user_id::text
+          WHEN d.scope = 'DOCUMENT_SCOPE_RESTRICTED'
+              THEN COALESCE(rg_scope.restricted_group_id::text, '')
+          ELSE ''::text
+      END
+  )`;
+async function findDocumentTableArtifactScopeMismatches(client, args) {
+    const result = await client.query({
+        text: exports.findDocumentTableArtifactScopeMismatchesQuery,
+        values: [args.organizationId],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            organizationId: row[1],
+            documentId: row[2],
+            artifactId: row[3],
+            artifactScopeKind: row[4],
+            artifactScopeKey: row[5],
+            currentScope: row[6],
+            currentScopeKey: row[7]
+        };
+    });
+}
+exports.insertSessionTabularRetrievalContextQuery = `-- name: InsertSessionTabularRetrievalContext :one
+INSERT INTO weave.session_tabular_retrieval_context (
+    session_id,
+    turn_number,
+    organization_id,
+    table_id,
+    artifact_id,
+    query,
+    sql,
+    result_row_count,
+    trace
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6,
+    $7,
+    $8,
+    $9
+)
+RETURNING
+    session_id,
+    turn_number,
+    organization_id,
+    table_id,
+    artifact_id,
+    query,
+    sql,
+    result_row_count,
+    trace,
+    retrieved_at`;
+async function insertSessionTabularRetrievalContext(client, args) {
+    const result = await client.query({
+        text: exports.insertSessionTabularRetrievalContextQuery,
+        values: [args.sessionId, args.turnNumber, args.organizationId, args.tableId, args.artifactId, args.query, args.sql, args.resultRowCount, args.trace],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        sessionId: row[0],
+        turnNumber: row[1],
+        organizationId: row[2],
+        tableId: row[3],
+        artifactId: row[4],
+        query: row[5],
+        sql: row[6],
+        resultRowCount: row[7],
+        trace: row[8],
+        retrievedAt: row[9]
+    };
+}
+exports.listSessionTabularRetrievalContextQuery = `-- name: ListSessionTabularRetrievalContext :many
+SELECT
+    session_id,
+    turn_number,
+    organization_id,
+    table_id,
+    artifact_id,
+    query,
+    sql,
+    result_row_count,
+    trace,
+    retrieved_at
+FROM weave.session_tabular_retrieval_context
+WHERE session_id = $1
+  AND organization_id = $2
+ORDER BY turn_number DESC, retrieved_at DESC
+LIMIT $3`;
+async function listSessionTabularRetrievalContext(client, args) {
+    const result = await client.query({
+        text: exports.listSessionTabularRetrievalContextQuery,
+        values: [args.sessionId, args.organizationId, args.resultLimit],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            sessionId: row[0],
+            turnNumber: row[1],
+            organizationId: row[2],
+            tableId: row[3],
+            artifactId: row[4],
+            query: row[5],
+            sql: row[6],
+            resultRowCount: row[7],
+            trace: row[8],
+            retrievedAt: row[9]
+        };
+    });
+}
+exports.enqueueDocumentBlobDeletionJobQuery = `-- name: EnqueueDocumentBlobDeletionJob :one
+INSERT INTO weave.document_blob_deletion_jobs (
+    id,
+    organization_id,
+    document_id,
+    artifact_id,
+    blob_kind,
+    storage_ref,
+    status,
+    next_retry_at
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4::uuid,
+    $5,
+    $6,
+    $7,
+    $8
+)
+RETURNING
+    id,
+    organization_id,
+    document_id,
+    artifact_id,
+    blob_kind,
+    storage_ref,
+    status,
+    attempt_count,
+    next_retry_at,
+    last_error,
+    created_at,
+    updated_at`;
+async function enqueueDocumentBlobDeletionJob(client, args) {
+    const result = await client.query({
+        text: exports.enqueueDocumentBlobDeletionJobQuery,
+        values: [args.id, args.organizationId, args.documentId, args.artifactId, args.blobKind, args.storageRef, args.status, args.nextRetryAt],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        documentId: row[2],
+        artifactId: row[3],
+        blobKind: row[4],
+        storageRef: row[5],
+        status: row[6],
+        attemptCount: row[7],
+        nextRetryAt: row[8],
+        lastError: row[9],
+        createdAt: row[10],
+        updatedAt: row[11]
+    };
+}
+exports.listReadyDocumentBlobDeletionJobsQuery = `-- name: ListReadyDocumentBlobDeletionJobs :many
+SELECT
+    id,
+    organization_id,
+    document_id,
+    artifact_id,
+    blob_kind,
+    storage_ref,
+    status,
+    attempt_count,
+    next_retry_at,
+    last_error,
+    created_at,
+    updated_at
+FROM weave.document_blob_deletion_jobs
+WHERE status IN (
+    'DOCUMENT_BLOB_DELETION_JOB_STATUS_PENDING',
+    'DOCUMENT_BLOB_DELETION_JOB_STATUS_RETRYABLE'
+)
+  AND next_retry_at <= now()
+ORDER BY next_retry_at ASC, created_at ASC
+LIMIT $1`;
+async function listReadyDocumentBlobDeletionJobs(client, args) {
+    const result = await client.query({
+        text: exports.listReadyDocumentBlobDeletionJobsQuery,
+        values: [args.resultLimit],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            organizationId: row[1],
+            documentId: row[2],
+            artifactId: row[3],
+            blobKind: row[4],
+            storageRef: row[5],
+            status: row[6],
+            attemptCount: row[7],
+            nextRetryAt: row[8],
+            lastError: row[9],
+            createdAt: row[10],
+            updatedAt: row[11]
+        };
+    });
+}
+exports.claimDocumentBlobDeletionJobQuery = `-- name: ClaimDocumentBlobDeletionJob :one
+UPDATE weave.document_blob_deletion_jobs
+SET
+    status = 'DOCUMENT_BLOB_DELETION_JOB_STATUS_CLAIMED',
+    attempt_count = attempt_count + 1,
+    updated_at = now()
+WHERE id = $1
+  AND status IN (
+      'DOCUMENT_BLOB_DELETION_JOB_STATUS_PENDING',
+      'DOCUMENT_BLOB_DELETION_JOB_STATUS_RETRYABLE'
+  )
+RETURNING
+    id,
+    organization_id,
+    document_id,
+    artifact_id,
+    blob_kind,
+    storage_ref,
+    status,
+    attempt_count,
+    next_retry_at,
+    last_error,
+    created_at,
+    updated_at`;
+async function claimDocumentBlobDeletionJob(client, args) {
+    const result = await client.query({
+        text: exports.claimDocumentBlobDeletionJobQuery,
+        values: [args.id],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        documentId: row[2],
+        artifactId: row[3],
+        blobKind: row[4],
+        storageRef: row[5],
+        status: row[6],
+        attemptCount: row[7],
+        nextRetryAt: row[8],
+        lastError: row[9],
+        createdAt: row[10],
+        updatedAt: row[11]
+    };
+}
+exports.completeDocumentBlobDeletionJobQuery = `-- name: CompleteDocumentBlobDeletionJob :execrows
+UPDATE weave.document_blob_deletion_jobs
+SET
+    status = 'DOCUMENT_BLOB_DELETION_JOB_STATUS_COMPLETE',
+    last_error = '',
+    updated_at = now()
+WHERE id = $1`;
+exports.retryDocumentBlobDeletionJobQuery = `-- name: RetryDocumentBlobDeletionJob :execrows
+UPDATE weave.document_blob_deletion_jobs
+SET
+    status = $1,
+    last_error = $2,
+    next_retry_at = $3,
+    updated_at = now()
+WHERE id = $4`;