weave-typescript 0.44.0 → 0.46.0

package/dist/weavesql/weavedb/ingestion_sql.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.listEntitiesByThreadQuery = exports.upsertEntityRelationshipQuery = exports.upsertThreadEntityQuery = exports.upsertEntityQuery = exports.listThreadActivityByThreadQuery = exports.createThreadActivityQuery = exports.listThreadReconciliationByPendingThreadQuery = exports.upsertThreadReconciliationQuery = exports.deleteThreadReconciliationByPendingThreadQuery = exports.getIngestionRunThreadCountsQuery = exports.listThreadsByIDsQuery = exports.listThreadsBySupersedesQuery = exports.markThreadsSupersededQuery = exports.bulkRejectThreadsQuery = exports.rejectThreadQuery = exports.bulkApproveThreadsQuery = exports.approveThreadQuery = exports.deleteDocumentThreadByOrganizationAndIDQuery = exports.editDocumentThreadQuery = exports.editPendingThreadQuery = exports.updateThreadClassificationQuery = exports.promoteThreadNextEmbeddingsByOrganizationQuery = exports.setThreadNextEmbeddingQuery = exports.updateThreadEmbeddingQuery = exports.listPendingThreadsByOrganizationQuery = exports.listPendingThreadsByRunQuery = exports.listThreadsByDocumentQuery = exports.listThreadsByIngestionRunQuery = exports.getThreadByOrganizationAndIDQuery = exports.createThreadQuery = exports.deleteDocumentByOrganizationAndIDQuery = exports.listDocumentTagsByDocumentIDQuery = exports.createDocumentTagQuery = exports.deleteDocumentTagByDocumentIDAndTagQuery = exports.deleteDocumentTagsByDocumentIDQuery = exports.listIngestionHistoryQuery = exports.listIngestionRunsByDocumentQuery = exports.listIngestionRunsByOrganizationQuery = exports.updateIngestionRunStateQuery = exports.getIngestionRunByOrganizationAndIDQuery = exports.createIngestionRunQuery = exports.getDatabaseConfigQuery = exports.getAccessibleDocumentSummaryQuery = exports.listAccessibleDocumentSummariesQuery = exports.listDocumentsByOrganizationQuery = exports.setDocumentStatusQuery = exports.setDocumentParseResultQuery = exports.getDocumentByOrganizationAndFingerprintQuery = exports.getDocumentByOrganizationAndIDQuery = exports.createDocumentQuery = void 0;
-exports.deactivateFacetConfigQuery = exports.upsertFacetConfigQuery = exports.listActiveFacetConfigsQuery = exports.listFacetConfigsQuery = exports.syncOrganizationIngestionReasoningDefaultsQuery = exports.upsertOrganizationIngestionSettingsQuery = exports.getOrganizationIngestionSettingsQuery = exports.ensureOrganizationIngestionSettingsQuery = void 0;
+exports.deactivateFacetConfigQuery = exports.upsertFacetConfigQuery = exports.listActiveFacetConfigsQuery = exports.listFacetConfigsQuery = exports.deleteThreadFacetsByThreadQuery = exports.upsertThreadFacetQuery = exports.listThreadFacetsByThreadIDsQuery = exports.syncOrganizationIngestionReasoningDefaultsQuery = exports.upsertOrganizationIngestionStageSettingsQuery = exports.upsertOrganizationIngestionSettingsQuery = exports.getOrganizationIngestionSettingsQuery = exports.ensureOrganizationIngestionSettingsQuery = void 0;
 exports.createDocument = createDocument;
 exports.getDocumentByOrganizationAndID = getDocumentByOrganizationAndID;
 exports.getDocumentByOrganizationAndFingerprint = getDocumentByOrganizationAndFingerprint;
@@ -50,7 +50,10 @@ exports.listEntitiesByThread = listEntitiesByThread;
 exports.ensureOrganizationIngestionSettings = ensureOrganizationIngestionSettings;
 exports.getOrganizationIngestionSettings = getOrganizationIngestionSettings;
 exports.upsertOrganizationIngestionSettings = upsertOrganizationIngestionSettings;
+exports.upsertOrganizationIngestionStageSettings = upsertOrganizationIngestionStageSettings;
 exports.syncOrganizationIngestionReasoningDefaults = syncOrganizationIngestionReasoningDefaults;
+exports.listThreadFacetsByThreadIDs = listThreadFacetsByThreadIDs;
+exports.upsertThreadFacet = upsertThreadFacet;
 exports.listFacetConfigs = listFacetConfigs;
 exports.listActiveFacetConfigs = listActiveFacetConfigs;
 exports.upsertFacetConfig = upsertFacetConfig;
@@ -684,7 +687,9 @@ INSERT INTO weave.ingestion_runs (
     reasoning_provider_configuration_id,
     embedding_model_id,
     embedding_provider_configuration_id,
-    embedding_dimensions
+    embedding_dimensions,
+    stage_model_settings,
+    refine_loop_count
 ) VALUES (
     $1,
     $2,
@@ -694,7 +699,9 @@ INSERT INTO weave.ingestion_runs (
     $6,
     $7,
     $8,
-    $9
+    $9,
+    $10,
+    $11
 )
 RETURNING
     id,
@@ -706,6 +713,8 @@ RETURNING
     embedding_model_id,
     embedding_provider_configuration_id,
     embedding_dimensions,
+    stage_model_settings,
+    refine_loop_count,
     thread_count,
     pending_count,
     approved_count,
@@ -718,7 +727,7 @@ RETURNING
 async function createIngestionRun(client, args) {
     const result = await client.query({
         text: exports.createIngestionRunQuery,
-        values: [args.id, args.organizationId, args.documentId, args.status, args.reasoningModelId, args.reasoningProviderConfigurationId, args.embeddingModelId, args.embeddingProviderConfigurationId, args.embeddingDimensions],
+        values: [args.id, args.organizationId, args.documentId, args.status, args.reasoningModelId, args.reasoningProviderConfigurationId, args.embeddingModelId, args.embeddingProviderConfigurationId, args.embeddingDimensions, args.stageModelSettings, args.refineLoopCount],
         rowMode: "array"
     });
     if (result.rows.length !== 1) {
@@ -735,15 +744,17 @@ async function createIngestionRun(client, args) {
         embeddingModelId: row[6],
         embeddingProviderConfigurationId: row[7],
         embeddingDimensions: row[8],
-        threadCount: row[9],
-        pendingCount: row[10],
-        approvedCount: row[11],
-        rejectedCount: row[12],
-        error: row[13],
-        tokenUsage: row[14],
-        startedAt: row[15],
-        completedAt: row[16],
-        committedAt: row[17]
+        stageModelSettings: row[9],
+        refineLoopCount: row[10],
+        threadCount: row[11],
+        pendingCount: row[12],
+        approvedCount: row[13],
+        rejectedCount: row[14],
+        error: row[15],
+        tokenUsage: row[16],
+        startedAt: row[17],
+        completedAt: row[18],
+        committedAt: row[19]
     };
 }
 exports.getIngestionRunByOrganizationAndIDQuery = `-- name: GetIngestionRunByOrganizationAndID :one
@@ -757,6 +768,8 @@ SELECT
     embedding_model_id,
     embedding_provider_configuration_id,
     embedding_dimensions,
+    stage_model_settings,
+    refine_loop_count,
     thread_count,
     pending_count,
     approved_count,
@@ -789,15 +802,17 @@ async function getIngestionRunByOrganizationAndID(client, args) {
         embeddingModelId: row[6],
         embeddingProviderConfigurationId: row[7],
         embeddingDimensions: row[8],
-        threadCount: row[9],
-        pendingCount: row[10],
-        approvedCount: row[11],
-        rejectedCount: row[12],
-        error: row[13],
-        tokenUsage: row[14],
-        startedAt: row[15],
-        completedAt: row[16],
-        committedAt: row[17]
+        stageModelSettings: row[9],
+        refineLoopCount: row[10],
+        threadCount: row[11],
+        pendingCount: row[12],
+        approvedCount: row[13],
+        rejectedCount: row[14],
+        error: row[15],
+        tokenUsage: row[16],
+        startedAt: row[17],
+        completedAt: row[18],
+        committedAt: row[19]
     };
 }
 exports.updateIngestionRunStateQuery = `-- name: UpdateIngestionRunState :one
@@ -824,6 +839,8 @@ RETURNING
     embedding_model_id,
     embedding_provider_configuration_id,
     embedding_dimensions,
+    stage_model_settings,
+    refine_loop_count,
     thread_count,
     pending_count,
     approved_count,
@@ -853,15 +870,17 @@ async function updateIngestionRunState(client, args) {
         embeddingModelId: row[6],
         embeddingProviderConfigurationId: row[7],
         embeddingDimensions: row[8],
-        threadCount: row[9],
-        pendingCount: row[10],
-        approvedCount: row[11],
-        rejectedCount: row[12],
-        error: row[13],
-        tokenUsage: row[14],
-        startedAt: row[15],
-        completedAt: row[16],
-        committedAt: row[17]
+        stageModelSettings: row[9],
+        refineLoopCount: row[10],
+        threadCount: row[11],
+        pendingCount: row[12],
+        approvedCount: row[13],
+        rejectedCount: row[14],
+        error: row[15],
+        tokenUsage: row[16],
+        startedAt: row[17],
+        completedAt: row[18],
+        committedAt: row[19]
     };
 }
 exports.listIngestionRunsByOrganizationQuery = `-- name: ListIngestionRunsByOrganization :many
@@ -875,6 +894,8 @@ SELECT
     embedding_model_id,
     embedding_provider_configuration_id,
     embedding_dimensions,
+    stage_model_settings,
+    refine_loop_count,
     thread_count,
     pending_count,
     approved_count,
@@ -905,15 +926,17 @@ async function listIngestionRunsByOrganization(client, args) {
             embeddingModelId: row[6],
             embeddingProviderConfigurationId: row[7],
             embeddingDimensions: row[8],
-            threadCount: row[9],
-            pendingCount: row[10],
-            approvedCount: row[11],
-            rejectedCount: row[12],
-            error: row[13],
-            tokenUsage: row[14],
-            startedAt: row[15],
-            completedAt: row[16],
-            committedAt: row[17]
+            stageModelSettings: row[9],
+            refineLoopCount: row[10],
+            threadCount: row[11],
+            pendingCount: row[12],
+            approvedCount: row[13],
+            rejectedCount: row[14],
+            error: row[15],
+            tokenUsage: row[16],
+            startedAt: row[17],
+            completedAt: row[18],
+            committedAt: row[19]
         };
     });
 }
@@ -928,6 +951,8 @@ SELECT
     embedding_model_id,
     embedding_provider_configuration_id,
     embedding_dimensions,
+    stage_model_settings,
+    refine_loop_count,
     thread_count,
     pending_count,
     approved_count,
@@ -958,15 +983,17 @@ async function listIngestionRunsByDocument(client, args) {
             embeddingModelId: row[6],
             embeddingProviderConfigurationId: row[7],
             embeddingDimensions: row[8],
-            threadCount: row[9],
-            pendingCount: row[10],
-            approvedCount: row[11],
-            rejectedCount: row[12],
-            error: row[13],
-            tokenUsage: row[14],
-            startedAt: row[15],
-            completedAt: row[16],
-            committedAt: row[17]
+            stageModelSettings: row[9],
+            refineLoopCount: row[10],
+            threadCount: row[11],
+            pendingCount: row[12],
+            approvedCount: row[13],
+            rejectedCount: row[14],
+            error: row[15],
+            tokenUsage: row[16],
+            startedAt: row[17],
+            completedAt: row[18],
+            committedAt: row[19]
         };
     });
 }
@@ -2612,6 +2639,19 @@ WITH upsert AS (
         reasoning_provider_configuration_id,
         embedding_model_id,
         embedding_provider_configuration_id,
+        classify_model_id,
+        classify_provider_configuration_id,
+        refine_model_id,
+        refine_provider_configuration_id,
+        extract_model_id,
+        extract_provider_configuration_id,
+        embed_model_id,
+        embed_provider_configuration_id,
+        reconcile_model_id,
+        reconcile_provider_configuration_id,
+        diff_model_id,
+        diff_provider_configuration_id,
+        refine_loop_count,
         updated_at
 )
 SELECT
@@ -2620,7 +2660,21 @@ SELECT
     upsert.reasoning_provider_configuration_id,
     upsert.embedding_model_id,
     upsert.embedding_provider_configuration_id,
+    upsert.classify_model_id,
+    upsert.classify_provider_configuration_id,
+    upsert.refine_model_id,
+    upsert.refine_provider_configuration_id,
+    upsert.extract_model_id,
+    upsert.extract_provider_configuration_id,
+    upsert.embed_model_id,
+    upsert.embed_provider_configuration_id,
+    upsert.reconcile_model_id,
+    upsert.reconcile_provider_configuration_id,
+    upsert.diff_model_id,
+    upsert.diff_provider_configuration_id,
     cfg.embedding_dimension AS embedding_dimensions,
+    cfg.embedding_dimension AS embed_dimensions,
+    upsert.refine_loop_count,
     upsert.updated_at
 FROM upsert
 JOIN weave.database_config cfg
@@ -2641,8 +2695,22 @@ async function ensureOrganizationIngestionSettings(client, args) {
         reasoningProviderConfigurationId: row[2],
         embeddingModelId: row[3],
         embeddingProviderConfigurationId: row[4],
-        embeddingDimensions: row[5],
-        updatedAt: row[6]
+        classifyModelId: row[5],
+        classifyProviderConfigurationId: row[6],
+        refineModelId: row[7],
+        refineProviderConfigurationId: row[8],
+        extractModelId: row[9],
+        extractProviderConfigurationId: row[10],
+        embedModelId: row[11],
+        embedProviderConfigurationId: row[12],
+        reconcileModelId: row[13],
+        reconcileProviderConfigurationId: row[14],
+        diffModelId: row[15],
+        diffProviderConfigurationId: row[16],
+        embeddingDimensions: row[17],
+        embedDimensions: row[18],
+        refineLoopCount: row[19],
+        updatedAt: row[20]
     };
 }
 exports.getOrganizationIngestionSettingsQuery = `-- name: GetOrganizationIngestionSettings :one
@@ -2652,7 +2720,21 @@ SELECT
     settings.reasoning_provider_configuration_id,
     settings.embedding_model_id,
     settings.embedding_provider_configuration_id,
+    settings.classify_model_id,
+    settings.classify_provider_configuration_id,
+    settings.refine_model_id,
+    settings.refine_provider_configuration_id,
+    settings.extract_model_id,
+    settings.extract_provider_configuration_id,
+    settings.embed_model_id,
+    settings.embed_provider_configuration_id,
+    settings.reconcile_model_id,
+    settings.reconcile_provider_configuration_id,
+    settings.diff_model_id,
+    settings.diff_provider_configuration_id,
     cfg.embedding_dimension AS embedding_dimensions,
+    cfg.embedding_dimension AS embed_dimensions,
+    settings.refine_loop_count,
     settings.updated_at
 FROM weave.org_ingestion_settings settings
 JOIN weave.database_config cfg
@@ -2674,8 +2756,22 @@ async function getOrganizationIngestionSettings(client, args) {
         reasoningProviderConfigurationId: row[2],
         embeddingModelId: row[3],
         embeddingProviderConfigurationId: row[4],
-        embeddingDimensions: row[5],
-        updatedAt: row[6]
+        classifyModelId: row[5],
+        classifyProviderConfigurationId: row[6],
+        refineModelId: row[7],
+        refineProviderConfigurationId: row[8],
+        extractModelId: row[9],
+        extractProviderConfigurationId: row[10],
+        embedModelId: row[11],
+        embedProviderConfigurationId: row[12],
+        reconcileModelId: row[13],
+        reconcileProviderConfigurationId: row[14],
+        diffModelId: row[15],
+        diffProviderConfigurationId: row[16],
+        embeddingDimensions: row[17],
+        embedDimensions: row[18],
+        refineLoopCount: row[19],
+        updatedAt: row[20]
     };
 }
 exports.upsertOrganizationIngestionSettingsQuery = `-- name: UpsertOrganizationIngestionSettings :one
@@ -2690,14 +2786,38 @@ WITH cfg AS (
         reasoning_model_id,
         reasoning_provider_configuration_id,
         embedding_model_id,
-        embedding_provider_configuration_id
+        embedding_provider_configuration_id,
+        classify_model_id,
+        classify_provider_configuration_id,
+        refine_model_id,
+        refine_provider_configuration_id,
+        extract_model_id,
+        extract_provider_configuration_id,
+        embed_model_id,
+        embed_provider_configuration_id,
+        reconcile_model_id,
+        reconcile_provider_configuration_id,
+        diff_model_id,
+        diff_provider_configuration_id
     )
     SELECT
         $2,
         $3,
         $4,
         $5,
-        $6
+        $6,
+        $3,
+        $4,
+        $3,
+        $4,
+        $3,
+        $4,
+        $5,
+        $6,
+        $3,
+        $4,
+        $3,
+        $4
     FROM cfg
     ON CONFLICT (organization_id) DO UPDATE
     SET
@@ -2705,6 +2825,18 @@ WITH cfg AS (
         reasoning_provider_configuration_id = EXCLUDED.reasoning_provider_configuration_id,
         embedding_model_id = EXCLUDED.embedding_model_id,
         embedding_provider_configuration_id = EXCLUDED.embedding_provider_configuration_id,
+        classify_model_id = EXCLUDED.classify_model_id,
+        classify_provider_configuration_id = EXCLUDED.classify_provider_configuration_id,
+        refine_model_id = EXCLUDED.refine_model_id,
+        refine_provider_configuration_id = EXCLUDED.refine_provider_configuration_id,
+        extract_model_id = EXCLUDED.extract_model_id,
+        extract_provider_configuration_id = EXCLUDED.extract_provider_configuration_id,
+        embed_model_id = EXCLUDED.embed_model_id,
+        embed_provider_configuration_id = EXCLUDED.embed_provider_configuration_id,
+        reconcile_model_id = EXCLUDED.reconcile_model_id,
+        reconcile_provider_configuration_id = EXCLUDED.reconcile_provider_configuration_id,
+        diff_model_id = EXCLUDED.diff_model_id,
+        diff_provider_configuration_id = EXCLUDED.diff_provider_configuration_id,
         updated_at = now()
     RETURNING
         organization_id,
@@ -2712,6 +2844,19 @@ WITH cfg AS (
         reasoning_provider_configuration_id,
         embedding_model_id,
         embedding_provider_configuration_id,
+        classify_model_id,
+        classify_provider_configuration_id,
+        refine_model_id,
+        refine_provider_configuration_id,
+        extract_model_id,
+        extract_provider_configuration_id,
+        embed_model_id,
+        embed_provider_configuration_id,
+        reconcile_model_id,
+        reconcile_provider_configuration_id,
+        diff_model_id,
+        diff_provider_configuration_id,
+        refine_loop_count,
         updated_at
 )
 SELECT
@@ -2720,7 +2865,21 @@ SELECT
     upsert.reasoning_provider_configuration_id,
     upsert.embedding_model_id,
     upsert.embedding_provider_configuration_id,
+    upsert.classify_model_id,
+    upsert.classify_provider_configuration_id,
+    upsert.refine_model_id,
+    upsert.refine_provider_configuration_id,
+    upsert.extract_model_id,
+    upsert.extract_provider_configuration_id,
+    upsert.embed_model_id,
+    upsert.embed_provider_configuration_id,
+    upsert.reconcile_model_id,
+    upsert.reconcile_provider_configuration_id,
+    upsert.diff_model_id,
+    upsert.diff_provider_configuration_id,
     cfg.embedding_dimension AS embedding_dimensions,
+    cfg.embedding_dimension AS embed_dimensions,
+    upsert.refine_loop_count,
     upsert.updated_at
 FROM upsert
 JOIN cfg
@@ -2741,8 +2900,145 @@ async function upsertOrganizationIngestionSettings(client, args) {
         reasoningProviderConfigurationId: row[2],
         embeddingModelId: row[3],
         embeddingProviderConfigurationId: row[4],
-        embeddingDimensions: row[5],
-        updatedAt: row[6]
+        classifyModelId: row[5],
+        classifyProviderConfigurationId: row[6],
+        refineModelId: row[7],
+        refineProviderConfigurationId: row[8],
+        extractModelId: row[9],
+        extractProviderConfigurationId: row[10],
+        embedModelId: row[11],
+        embedProviderConfigurationId: row[12],
+        reconcileModelId: row[13],
+        reconcileProviderConfigurationId: row[14],
+        diffModelId: row[15],
+        diffProviderConfigurationId: row[16],
+        embeddingDimensions: row[17],
+        embedDimensions: row[18],
+        refineLoopCount: row[19],
+        updatedAt: row[20]
+    };
+}
+exports.upsertOrganizationIngestionStageSettingsQuery = `-- name: UpsertOrganizationIngestionStageSettings :one
+WITH cfg AS (
+    SELECT embedding_dimension
+    FROM weave.database_config
+    WHERE id = 1
+      AND ($1 = 0 OR $1 = embedding_dimension)
+), upsert AS (
+    INSERT INTO weave.org_ingestion_settings (
+        organization_id,
+        classify_model_id,
+        classify_provider_configuration_id,
+        refine_model_id,
+        refine_provider_configuration_id,
+        extract_model_id,
+        extract_provider_configuration_id,
+        embed_model_id,
+        embed_provider_configuration_id,
+        reconcile_model_id,
+        reconcile_provider_configuration_id,
+        diff_model_id,
+        diff_provider_configuration_id,
+        refine_loop_count
+    )
+    SELECT
+        $2,
+        $3,
+        $4,
+        $5,
+        $6,
+        $7,
+        $8,
+        $9,
+        $10,
+        $11,
+        $12,
+        $13,
+        $14,
+        $15
+    FROM cfg
+    ON CONFLICT (organization_id) DO UPDATE
+    SET
+        classify_model_id = EXCLUDED.classify_model_id,
+        classify_provider_configuration_id = EXCLUDED.classify_provider_configuration_id,
+        refine_model_id = EXCLUDED.refine_model_id,
+        refine_provider_configuration_id = EXCLUDED.refine_provider_configuration_id,
+        extract_model_id = EXCLUDED.extract_model_id,
+        extract_provider_configuration_id = EXCLUDED.extract_provider_configuration_id,
+        embed_model_id = EXCLUDED.embed_model_id,
+        embed_provider_configuration_id = EXCLUDED.embed_provider_configuration_id,
+        reconcile_model_id = EXCLUDED.reconcile_model_id,
+        reconcile_provider_configuration_id = EXCLUDED.reconcile_provider_configuration_id,
+        diff_model_id = EXCLUDED.diff_model_id,
+        diff_provider_configuration_id = EXCLUDED.diff_provider_configuration_id,
+        refine_loop_count = EXCLUDED.refine_loop_count,
+        updated_at = now()
+    RETURNING
+        organization_id,
+        classify_model_id,
+        classify_provider_configuration_id,
+        refine_model_id,
+        refine_provider_configuration_id,
+        extract_model_id,
+        extract_provider_configuration_id,
+        embed_model_id,
+        embed_provider_configuration_id,
+        reconcile_model_id,
+        reconcile_provider_configuration_id,
+        diff_model_id,
+        diff_provider_configuration_id,
+        refine_loop_count,
+        updated_at
+)
+SELECT
+    upsert.organization_id,
+    upsert.classify_model_id,
+    upsert.classify_provider_configuration_id,
+    upsert.refine_model_id,
+    upsert.refine_provider_configuration_id,
+    upsert.extract_model_id,
+    upsert.extract_provider_configuration_id,
+    upsert.embed_model_id,
+    upsert.embed_provider_configuration_id,
+    upsert.reconcile_model_id,
+    upsert.reconcile_provider_configuration_id,
+    upsert.diff_model_id,
+    upsert.diff_provider_configuration_id,
+    cfg.embedding_dimension AS embedding_dimensions,
+    cfg.embedding_dimension AS embed_dimensions,
+    upsert.refine_loop_count,
+    upsert.updated_at
+FROM upsert
+JOIN cfg
+  ON TRUE`;
+async function upsertOrganizationIngestionStageSettings(client, args) {
+    const result = await client.query({
+        text: exports.upsertOrganizationIngestionStageSettingsQuery,
+        values: [args.embedDimensions, args.organizationId, args.classifyModelId, args.classifyProviderConfigurationId, args.refineModelId, args.refineProviderConfigurationId, args.extractModelId, args.extractProviderConfigurationId, args.embedModelId, args.embedProviderConfigurationId, args.reconcileModelId, args.reconcileProviderConfigurationId, args.diffModelId, args.diffProviderConfigurationId, args.refineLoopCount],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        organizationId: row[0],
+        classifyModelId: row[1],
+        classifyProviderConfigurationId: row[2],
+        refineModelId: row[3],
+        refineProviderConfigurationId: row[4],
+        extractModelId: row[5],
+        extractProviderConfigurationId: row[6],
+        embedModelId: row[7],
+        embedProviderConfigurationId: row[8],
+        reconcileModelId: row[9],
+        reconcileProviderConfigurationId: row[10],
+        diffModelId: row[11],
+        diffProviderConfigurationId: row[12],
+        embeddingDimensions: row[13],
+        embedDimensions: row[14],
+        refineLoopCount: row[15],
+        updatedAt: row[16]
     };
 }
 exports.syncOrganizationIngestionReasoningDefaultsQuery = `-- name: SyncOrganizationIngestionReasoningDefaults :one
@@ -2754,11 +3050,31 @@ WITH cfg AS (
     INSERT INTO weave.org_ingestion_settings (
         organization_id,
         reasoning_model_id,
-        reasoning_provider_configuration_id
+        reasoning_provider_configuration_id,
+        classify_model_id,
+        classify_provider_configuration_id,
+        refine_model_id,
+        refine_provider_configuration_id,
+        extract_model_id,
+        extract_provider_configuration_id,
+        reconcile_model_id,
+        reconcile_provider_configuration_id,
+        diff_model_id,
+        diff_provider_configuration_id
     )
     SELECT
         cs.organization_id,
         cs.default_chat_model_id,
+        cs.default_provider_configuration_id,
+        cs.default_chat_model_id,
+        cs.default_provider_configuration_id,
+        cs.default_chat_model_id,
+        cs.default_provider_configuration_id,
+        cs.default_chat_model_id,
+        cs.default_provider_configuration_id,
+        cs.default_chat_model_id,
+        cs.default_provider_configuration_id,
+        cs.default_chat_model_id,
         cs.default_provider_configuration_id
     FROM weave.organization_chat_settings cs
     WHERE cs.organization_id = $1
@@ -2769,6 +3085,31 @@ WITH cfg AS (
             weave.org_ingestion_settings.reasoning_provider_configuration_id,
             EXCLUDED.reasoning_provider_configuration_id
         ),
+        classify_model_id = COALESCE(weave.org_ingestion_settings.classify_model_id, EXCLUDED.classify_model_id),
+        classify_provider_configuration_id = COALESCE(
+            weave.org_ingestion_settings.classify_provider_configuration_id,
+            EXCLUDED.classify_provider_configuration_id
+        ),
+        refine_model_id = COALESCE(weave.org_ingestion_settings.refine_model_id, EXCLUDED.refine_model_id),
+        refine_provider_configuration_id = COALESCE(
+            weave.org_ingestion_settings.refine_provider_configuration_id,
+            EXCLUDED.refine_provider_configuration_id
+        ),
+        extract_model_id = COALESCE(weave.org_ingestion_settings.extract_model_id, EXCLUDED.extract_model_id),
+        extract_provider_configuration_id = COALESCE(
+            weave.org_ingestion_settings.extract_provider_configuration_id,
+            EXCLUDED.extract_provider_configuration_id
+        ),
+        reconcile_model_id = COALESCE(weave.org_ingestion_settings.reconcile_model_id, EXCLUDED.reconcile_model_id),
+        reconcile_provider_configuration_id = COALESCE(
+            weave.org_ingestion_settings.reconcile_provider_configuration_id,
+            EXCLUDED.reconcile_provider_configuration_id
+        ),
+        diff_model_id = COALESCE(weave.org_ingestion_settings.diff_model_id, EXCLUDED.diff_model_id),
+        diff_provider_configuration_id = COALESCE(
+            weave.org_ingestion_settings.diff_provider_configuration_id,
+            EXCLUDED.diff_provider_configuration_id
+        ),
         updated_at = now()
     RETURNING
         organization_id,
@@ -2776,6 +3117,19 @@ WITH cfg AS (
         reasoning_provider_configuration_id,
         embedding_model_id,
         embedding_provider_configuration_id,
+        classify_model_id,
+        classify_provider_configuration_id,
+        refine_model_id,
+        refine_provider_configuration_id,
+        extract_model_id,
+        extract_provider_configuration_id,
+        embed_model_id,
+        embed_provider_configuration_id,
+        reconcile_model_id,
+        reconcile_provider_configuration_id,
+        diff_model_id,
+        diff_provider_configuration_id,
+        refine_loop_count,
         updated_at
 )
 SELECT
@@ -2784,7 +3138,21 @@ SELECT
     upsert.reasoning_provider_configuration_id,
     upsert.embedding_model_id,
     upsert.embedding_provider_configuration_id,
+    upsert.classify_model_id,
+    upsert.classify_provider_configuration_id,
+    upsert.refine_model_id,
+    upsert.refine_provider_configuration_id,
+    upsert.extract_model_id,
+    upsert.extract_provider_configuration_id,
+    upsert.embed_model_id,
+    upsert.embed_provider_configuration_id,
+    upsert.reconcile_model_id,
+    upsert.reconcile_provider_configuration_id,
+    upsert.diff_model_id,
+    upsert.diff_provider_configuration_id,
     cfg.embedding_dimension AS embedding_dimensions,
+    cfg.embedding_dimension AS embed_dimensions,
+    upsert.refine_loop_count,
     upsert.updated_at
 FROM upsert
 JOIN cfg
@@ -2805,10 +3173,147 @@ async function syncOrganizationIngestionReasoningDefaults(client, args) {
         reasoningProviderConfigurationId: row[2],
         embeddingModelId: row[3],
         embeddingProviderConfigurationId: row[4],
-        embeddingDimensions: row[5],
-        updatedAt: row[6]
+        classifyModelId: row[5],
+        classifyProviderConfigurationId: row[6],
+        refineModelId: row[7],
+        refineProviderConfigurationId: row[8],
+        extractModelId: row[9],
+        extractProviderConfigurationId: row[10],
+        embedModelId: row[11],
+        embedProviderConfigurationId: row[12],
+        reconcileModelId: row[13],
+        reconcileProviderConfigurationId: row[14],
+        diffModelId: row[15],
+        diffProviderConfigurationId: row[16],
+        embeddingDimensions: row[17],
+        embedDimensions: row[18],
+        refineLoopCount: row[19],
+        updatedAt: row[20]
+    };
+}
+exports.listThreadFacetsByThreadIDsQuery = `-- name: ListThreadFacetsByThreadIDs :many
+SELECT
+    thread_id,
+    organization_id,
+    knowledge_type,
+    score,
+    facet_fields,
+    rationale,
+    is_primary,
+    created_at,
+    updated_at
+FROM weave.thread_facets
+WHERE organization_id = $1
+  AND thread_id = ANY($2::uuid[])
+ORDER BY thread_id ASC, is_primary DESC, score DESC, knowledge_type ASC`;
+async function listThreadFacetsByThreadIDs(client, args) {
+    const result = await client.query({
+        text: exports.listThreadFacetsByThreadIDsQuery,
+        values: [args.organizationId, args.threadIds],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            threadId: row[0],
+            organizationId: row[1],
+            knowledgeType: row[2],
+            score: row[3],
+            facetFields: row[4],
+            rationale: row[5],
+            isPrimary: row[6],
+            createdAt: row[7],
+            updatedAt: row[8]
+        };
+    });
+}
+exports.upsertThreadFacetQuery = `-- name: UpsertThreadFacet :one
+WITH scoped_thread AS (
+    SELECT t.id
+    FROM weave.threads t
+    WHERE t.organization_id = $1
+      AND t.id = $2
+), clear_primary AS (
+    UPDATE weave.thread_facets tf
+    SET is_primary = FALSE,
+        updated_at = now()
+    WHERE tf.organization_id = $1
+      AND tf.thread_id = $2
+      AND $3::boolean
+    RETURNING tf.thread_id
+), upsert AS (
+    INSERT INTO weave.thread_facets (
+        thread_id,
+        organization_id,
+        knowledge_type,
+        score,
+        facet_fields,
+        rationale,
+        is_primary
+    )
+    SELECT
+        scoped_thread.id,
+        $1,
+        $4,
+        $5,
+        $6,
+        $7,
+        $3
+    FROM scoped_thread
+    ON CONFLICT (thread_id, knowledge_type) DO UPDATE
+    SET
+        score = EXCLUDED.score,
+        facet_fields = EXCLUDED.facet_fields,
+        rationale = EXCLUDED.rationale,
+        is_primary = EXCLUDED.is_primary,
+        updated_at = now()
+    RETURNING
+        thread_id,
+        organization_id,
+        knowledge_type,
+        score,
+        facet_fields,
+        rationale,
+        is_primary,
+        created_at,
+        updated_at
+)
+SELECT
+    thread_id,
+    organization_id,
+    knowledge_type,
+    score,
+    facet_fields,
+    rationale,
+    is_primary,
+    created_at,
+    updated_at
+FROM upsert`;
+async function upsertThreadFacet(client, args) {
+    const result = await client.query({
+        text: exports.upsertThreadFacetQuery,
+        values: [args.organizationId, args.threadId, args.isPrimary, args.knowledgeType, args.score, args.facetFields, args.rationale],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        threadId: row[0],
+        organizationId: row[1],
+        knowledgeType: row[2],
+        score: row[3],
+        facetFields: row[4],
+        rationale: row[5],
+        isPrimary: row[6],
+        createdAt: row[7],
+        updatedAt: row[8]
     };
 }
+exports.deleteThreadFacetsByThreadQuery = `-- name: DeleteThreadFacetsByThread :execrows
+DELETE FROM weave.thread_facets
+WHERE organization_id = $1
+  AND thread_id = $2`;
 exports.listFacetConfigsQuery = `-- name: ListFacetConfigs :many
 SELECT
     organization_id,