@cat-factory/node-server 0.29.0 → 0.31.0

package/dist/repositories/drizzle.js

@@ -1,7 +1,7 @@
 import { LLM_WARNING_FINISH_REASONS } from '@cat-factory/kernel';
 import { blockInsertValues, blockPatchToColumns, rowToBlock, rowToExecution, executionToDetail, rowToPipeline, rowToSandboxExperiment, rowToSandboxFixture, rowToSandboxGrade, rowToSandboxPromptVersion, rowToSandboxRun, rowToWorkspace, } from '@cat-factory/server';
-import { and, desc, eq, gte, inArray, isNull, lt, ne, or, sql } from 'drizzle-orm';
-import { accountInvitations, passwordResetTokens, accountSettings, localSettings, accounts, agentContextSnapshots, agentRuns, blocks, consensusSessions, incidentEnrichmentConnections, observabilityConnections, emailConnections, llmCallMetrics, provisioningLog, memberships, mergeThresholdPresets, releaseHealthConfigs, pipelineScheduleRuns, pipelineSchedules, pipelines, requirementReviews, kaizenGradings, kaizenVerifiedCombos, clarityReviews, brainstormSessions, sandboxPromptVersions, sandboxFixtures, sandboxExperiments, sandboxRuns, sandboxGrades, services, tokenUsage, trackerSettings, modelPresets, userIdentities, users, workspaceFragmentDefaults, workspaceServices, workspaceSettings, workspaces, } from '../db/schema.js';
+import { and, asc, count, desc, eq, gte, inArray, isNull, lt, ne, or, sql } from 'drizzle-orm';
+import { accountInvitations, passwordResetTokens, accountSettings, localSettings, accounts, agentContextSnapshots, agentRuns, blocks, consensusSessions, incidentEnrichmentConnections, observabilityConnections, emailConnections, llmCallMetrics, provisioningLog, memberships, mergeThresholdPresets, releaseHealthConfigs, pipelineScheduleRuns, pipelineSchedules, pipelines, requirementReviews, kaizenGradings, kaizenVerifiedCombos, clarityReviews, binaryArtifacts, brainstormSessions, sandboxPromptVersions, sandboxFixtures, sandboxExperiments, sandboxRuns, sandboxGrades, services, tokenUsage, trackerSettings, modelPresets, userIdentities, users, workspaceFragmentDefaults, workspaceServices, workspaceSettings, workspaces, } from '../db/schema.js';
 // Drizzle/Postgres implementations of the core kernel repository ports. The
 // row<->domain mapping is the SAME shared mapping the Cloudflare D1 repos use
 // (@cat-factory/server), so behaviour matches across stores; this layer only owns
@@ -898,6 +898,7 @@ class DrizzleLlmCallMetricRepository {
             agentKind: llmCallMetrics.agent_kind,
             calls: sql `count(*)::int`,
             promptTokens: sql `coalesce(sum(${llmCallMetrics.prompt_tokens}), 0)::int`,
+            cachedPromptTokens: sql `coalesce(sum(${llmCallMetrics.cached_prompt_tokens}), 0)::int`,
             completionTokens: sql `coalesce(sum(${llmCallMetrics.completion_tokens}), 0)::int`,
             peakCompletionTokens: sql `coalesce(max(${llmCallMetrics.completion_tokens}), 0)::int`,
             maxOutputTokens: sql `max(${llmCallMetrics.request_max_tokens})`,
@@ -918,6 +919,7 @@ class DrizzleLlmCallMetricRepository {
             agentKind: r.agentKind,
             calls: Number(r.calls),
             promptTokens: Number(r.promptTokens),
+            cachedPromptTokens: Number(r.cachedPromptTokens),
             completionTokens: Number(r.completionTokens),
             peakCompletionTokens: Number(r.peakCompletionTokens),
             maxOutputTokens: r.maxOutputTokens == null ? null : Number(r.maxOutputTokens),
@@ -1003,6 +1005,95 @@ class DrizzleAgentContextSnapshotRepository {
         return deleted.length;
     }
 }
+function rowToBinaryArtifact(row) {
+    return {
+        id: row.id,
+        workspaceId: row.workspace_id,
+        executionId: row.execution_id,
+        blockId: row.block_id,
+        kind: row.kind,
+        view: row.view,
+        contentType: row.content_type,
+        byteSize: row.byte_size,
+        hash: row.hash,
+        storage: row.storage,
+        storageKey: row.storage_key,
+        createdAt: row.created_at,
+    };
+}
+/** Drizzle/Postgres metadata store for binary artifacts (mirror of D1 migration 0017). */
+class DrizzleBinaryArtifactMetadataStore {
+    db;
+    constructor(db) {
+        this.db = db;
+    }
+    async insert(record) {
+        await this.db.insert(binaryArtifacts).values({
+            workspace_id: record.workspaceId,
+            id: record.id,
+            execution_id: record.executionId,
+            block_id: record.blockId,
+            kind: record.kind,
+            view: record.view,
+            content_type: record.contentType,
+            byte_size: record.byteSize,
+            hash: record.hash,
+            storage: record.storage,
+            storage_key: record.storageKey,
+            created_at: record.createdAt,
+        });
+    }
+    async get(workspaceId, id) {
+        const rows = await this.db
+            .select()
+            .from(binaryArtifacts)
+            .where(and(eq(binaryArtifacts.workspace_id, workspaceId), eq(binaryArtifacts.id, id)))
+            .limit(1);
+        return rows[0] ? rowToBinaryArtifact(rows[0]) : null;
+    }
+    async listByExecution(workspaceId, executionId) {
+        const rows = await this.db
+            .select()
+            .from(binaryArtifacts)
+            .where(and(eq(binaryArtifacts.workspace_id, workspaceId), eq(binaryArtifacts.execution_id, executionId)))
+            .orderBy(asc(binaryArtifacts.created_at), asc(binaryArtifacts.id));
+        return rows.map(rowToBinaryArtifact);
+    }
+    async countByExecution(workspaceId, executionId) {
+        const rows = await this.db
+            .select({ n: count() })
+            .from(binaryArtifacts)
+            .where(and(eq(binaryArtifacts.workspace_id, workspaceId), eq(binaryArtifacts.execution_id, executionId)));
+        return rows[0]?.n ?? 0;
+    }
+    async listByBlock(workspaceId, blockId) {
+        const rows = await this.db
+            .select()
+            .from(binaryArtifacts)
+            .where(and(eq(binaryArtifacts.workspace_id, workspaceId), eq(binaryArtifacts.block_id, blockId)))
+            .orderBy(asc(binaryArtifacts.created_at), asc(binaryArtifacts.id));
+        return rows.map(rowToBinaryArtifact);
+    }
+    async delete(workspaceId, id) {
+        await this.db
+            .delete(binaryArtifacts)
+            .where(and(eq(binaryArtifacts.workspace_id, workspaceId), eq(binaryArtifacts.id, id)));
+    }
+    async listOlderThan(workspaceId, olderThan) {
+        const rows = await this.db
+            .select()
+            .from(binaryArtifacts)
+            .where(and(eq(binaryArtifacts.workspace_id, workspaceId), lt(binaryArtifacts.created_at, olderThan)));
+        return rows.map(rowToBinaryArtifact);
+    }
+    async deleteOlderThan(workspaceId, olderThan) {
+        const deleted = await this.db
+            .delete(binaryArtifacts)
+            .where(and(eq(binaryArtifacts.workspace_id, workspaceId), lt(binaryArtifacts.created_at, olderThan)))
+            .returning({ id: binaryArtifacts.id });
+        return deleted.length;
+    }
+}
 function rowToProvisioningLog(row) {
     return {
         id: row.id,
@@ -2709,6 +2800,7 @@ export class DrizzleWorkspaceSettingsRepository {
             taskLimitShared: row.task_limit_shared,
             taskLimitPerType: perType,
             storeAgentContext: row.store_agent_context === 1,
+            artifactRetentionDays: row.artifact_retention_days,
             kaizenEnabled: row.kaizen_enabled === 1,
             delegateAgentsToRunnerPool: row.delegate_agents_to_runner_pool === 1,
             delegateTestEnvToProvider: row.delegate_test_env_to_provider === 1,
@@ -2726,6 +2818,7 @@ export class DrizzleWorkspaceSettingsRepository {
                 ? JSON.stringify(settings.taskLimitPerType)
                 : null,
             store_agent_context: settings.storeAgentContext ? 1 : 0,
+            artifact_retention_days: settings.artifactRetentionDays,
             kaizen_enabled: settings.kaizenEnabled ? 1 : 0,
             delegate_agents_to_runner_pool: settings.delegateAgentsToRunnerPool ? 1 : 0,
             delegate_test_env_to_provider: settings.delegateTestEnvToProvider ? 1 : 0,
@@ -2743,6 +2836,7 @@ export class DrizzleWorkspaceSettingsRepository {
                 task_limit_shared: values.task_limit_shared,
                 task_limit_per_type: values.task_limit_per_type,
                 store_agent_context: values.store_agent_context,
+                artifact_retention_days: values.artifact_retention_days,
                 kaizen_enabled: values.kaizen_enabled,
                 delegate_agents_to_runner_pool: values.delegate_agents_to_runner_pool,
                 delegate_test_env_to_provider: values.delegate_test_env_to_provider,
@@ -3054,6 +3148,7 @@ export function createDrizzleRepositories(db, clock) {
         tokenUsageRepository: new DrizzleTokenUsageRepository(db),
         llmCallMetricRepository: new DrizzleLlmCallMetricRepository(db),
         agentContextSnapshotRepository: new DrizzleAgentContextSnapshotRepository(db),
+        binaryArtifactMetadataStore: new DrizzleBinaryArtifactMetadataStore(db),
         agentRunRepository: new DrizzleAgentRunRepository(db),
         modelPresetRepository: new DrizzleModelPresetRepository(db),
         serviceFragmentDefaultsRepository: new DrizzleServiceFragmentDefaultsRepository(db),