weave-typescript 0.44.0 → 0.45.0

package/dist/weavesql/weavedb/ingestion_sql.d.ts

@@ -231,7 +231,7 @@ export interface GetDatabaseConfigRow {
     updatedAt: Date;
 }
 export declare function getDatabaseConfig(client: Client): Promise<GetDatabaseConfigRow | null>;
-export declare const createIngestionRunQuery = "-- name: CreateIngestionRun :one\nINSERT INTO weave.ingestion_runs (\n    id,\n    organization_id,\n    document_id,\n    status,\n    reasoning_model_id,\n    reasoning_provider_configuration_id,\n    embedding_model_id,\n    embedding_provider_configuration_id,\n    embedding_dimensions\n) VALUES (\n    $1,\n    $2,\n    $3,\n    $4,\n    $5,\n    $6,\n    $7,\n    $8,\n    $9\n)\nRETURNING\n    id,\n    organization_id,\n    document_id,\n    status,\n    reasoning_model_id,\n    reasoning_provider_configuration_id,\n    embedding_model_id,\n    embedding_provider_configuration_id,\n    embedding_dimensions,\n    thread_count,\n    pending_count,\n    approved_count,\n    rejected_count,\n    error,\n    token_usage,\n    started_at,\n    completed_at,\n    committed_at";
+export declare const createIngestionRunQuery = "-- name: CreateIngestionRun :one\nINSERT INTO weave.ingestion_runs (\n    id,\n    organization_id,\n    document_id,\n    status,\n    reasoning_model_id,\n    reasoning_provider_configuration_id,\n    embedding_model_id,\n    embedding_provider_configuration_id,\n    embedding_dimensions,\n    stage_model_settings,\n    refine_loop_count\n) VALUES (\n    $1,\n    $2,\n    $3,\n    $4,\n    $5,\n    $6,\n    $7,\n    $8,\n    $9,\n    $10,\n    $11\n)\nRETURNING\n    id,\n    organization_id,\n    document_id,\n    status,\n    reasoning_model_id,\n    reasoning_provider_configuration_id,\n    embedding_model_id,\n    embedding_provider_configuration_id,\n    embedding_dimensions,\n    stage_model_settings,\n    refine_loop_count,\n    thread_count,\n    pending_count,\n    approved_count,\n    rejected_count,\n    error,\n    token_usage,\n    started_at,\n    completed_at,\n    committed_at";
 export interface CreateIngestionRunArgs {
     id: string;
     organizationId: string;
@@ -242,6 +242,8 @@ export interface CreateIngestionRunArgs {
     embeddingModelId: string | null;
     embeddingProviderConfigurationId: string | null;
     embeddingDimensions: number | null;
+    stageModelSettings: any;
+    refineLoopCount: number;
 }
 export interface CreateIngestionRunRow {
     id: string;
@@ -253,6 +255,8 @@ export interface CreateIngestionRunRow {
     embeddingModelId: string | null;
     embeddingProviderConfigurationId: string | null;
     embeddingDimensions: number | null;
+    stageModelSettings: any;
+    refineLoopCount: number;
     threadCount: number;
     pendingCount: number;
     approvedCount: number;
@@ -264,7 +268,7 @@ export interface CreateIngestionRunRow {
     committedAt: Date | null;
 }
 export declare function createIngestionRun(client: Client, args: CreateIngestionRunArgs): Promise<CreateIngestionRunRow | null>;
-export declare const getIngestionRunByOrganizationAndIDQuery = "-- name: GetIngestionRunByOrganizationAndID :one\nSELECT\n    id,\n    organization_id,\n    document_id,\n    status,\n    reasoning_model_id,\n    reasoning_provider_configuration_id,\n    embedding_model_id,\n    embedding_provider_configuration_id,\n    embedding_dimensions,\n    thread_count,\n    pending_count,\n    approved_count,\n    rejected_count,\n    error,\n    token_usage,\n    started_at,\n    completed_at,\n    committed_at\nFROM weave.ingestion_runs\nWHERE organization_id = $1\n  AND id = $2";
+export declare const getIngestionRunByOrganizationAndIDQuery = "-- name: GetIngestionRunByOrganizationAndID :one\nSELECT\n    id,\n    organization_id,\n    document_id,\n    status,\n    reasoning_model_id,\n    reasoning_provider_configuration_id,\n    embedding_model_id,\n    embedding_provider_configuration_id,\n    embedding_dimensions,\n    stage_model_settings,\n    refine_loop_count,\n    thread_count,\n    pending_count,\n    approved_count,\n    rejected_count,\n    error,\n    token_usage,\n    started_at,\n    completed_at,\n    committed_at\nFROM weave.ingestion_runs\nWHERE organization_id = $1\n  AND id = $2";
 export interface GetIngestionRunByOrganizationAndIDArgs {
     organizationId: string;
     id: string;
@@ -279,6 +283,8 @@ export interface GetIngestionRunByOrganizationAndIDRow {
     embeddingModelId: string | null;
     embeddingProviderConfigurationId: string | null;
     embeddingDimensions: number | null;
+    stageModelSettings: any;
+    refineLoopCount: number;
     threadCount: number;
     pendingCount: number;
     approvedCount: number;
@@ -290,7 +296,7 @@ export interface GetIngestionRunByOrganizationAndIDRow {
     committedAt: Date | null;
 }
 export declare function getIngestionRunByOrganizationAndID(client: Client, args: GetIngestionRunByOrganizationAndIDArgs): Promise<GetIngestionRunByOrganizationAndIDRow | null>;
-export declare const updateIngestionRunStateQuery = "-- name: UpdateIngestionRunState :one\nUPDATE weave.ingestion_runs\nSET\n    status = $1,\n    thread_count = $2,\n    pending_count = $3,\n    approved_count = $4,\n    rejected_count = $5,\n    error = $6,\n    token_usage = $7,\n    completed_at = $8,\n    committed_at = $9\nWHERE id = $10\n  AND organization_id = $11\nRETURNING\n    id,\n    organization_id,\n    document_id,\n    status,\n    reasoning_model_id,\n    reasoning_provider_configuration_id,\n    embedding_model_id,\n    embedding_provider_configuration_id,\n    embedding_dimensions,\n    thread_count,\n    pending_count,\n    approved_count,\n    rejected_count,\n    error,\n    token_usage,\n    started_at,\n    completed_at,\n    committed_at";
+export declare const updateIngestionRunStateQuery = "-- name: UpdateIngestionRunState :one\nUPDATE weave.ingestion_runs\nSET\n    status = $1,\n    thread_count = $2,\n    pending_count = $3,\n    approved_count = $4,\n    rejected_count = $5,\n    error = $6,\n    token_usage = $7,\n    completed_at = $8,\n    committed_at = $9\nWHERE id = $10\n  AND organization_id = $11\nRETURNING\n    id,\n    organization_id,\n    document_id,\n    status,\n    reasoning_model_id,\n    reasoning_provider_configuration_id,\n    embedding_model_id,\n    embedding_provider_configuration_id,\n    embedding_dimensions,\n    stage_model_settings,\n    refine_loop_count,\n    thread_count,\n    pending_count,\n    approved_count,\n    rejected_count,\n    error,\n    token_usage,\n    started_at,\n    completed_at,\n    committed_at";
 export interface UpdateIngestionRunStateArgs {
     status: string;
     threadCount: number;
@@ -314,6 +320,8 @@ export interface UpdateIngestionRunStateRow {
     embeddingModelId: string | null;
     embeddingProviderConfigurationId: string | null;
     embeddingDimensions: number | null;
+    stageModelSettings: any;
+    refineLoopCount: number;
     threadCount: number;
     pendingCount: number;
     approvedCount: number;
@@ -325,7 +333,7 @@ export interface UpdateIngestionRunStateRow {
     committedAt: Date | null;
 }
 export declare function updateIngestionRunState(client: Client, args: UpdateIngestionRunStateArgs): Promise<UpdateIngestionRunStateRow | null>;
-export declare const listIngestionRunsByOrganizationQuery = "-- name: ListIngestionRunsByOrganization :many\nSELECT\n    id,\n    organization_id,\n    document_id,\n    status,\n    reasoning_model_id,\n    reasoning_provider_configuration_id,\n    embedding_model_id,\n    embedding_provider_configuration_id,\n    embedding_dimensions,\n    thread_count,\n    pending_count,\n    approved_count,\n    rejected_count,\n    error,\n    token_usage,\n    started_at,\n    completed_at,\n    committed_at\nFROM weave.ingestion_runs\nWHERE organization_id = $1\nORDER BY started_at DESC\nLIMIT $3 OFFSET $2";
+export declare const listIngestionRunsByOrganizationQuery = "-- name: ListIngestionRunsByOrganization :many\nSELECT\n    id,\n    organization_id,\n    document_id,\n    status,\n    reasoning_model_id,\n    reasoning_provider_configuration_id,\n    embedding_model_id,\n    embedding_provider_configuration_id,\n    embedding_dimensions,\n    stage_model_settings,\n    refine_loop_count,\n    thread_count,\n    pending_count,\n    approved_count,\n    rejected_count,\n    error,\n    token_usage,\n    started_at,\n    completed_at,\n    committed_at\nFROM weave.ingestion_runs\nWHERE organization_id = $1\nORDER BY started_at DESC\nLIMIT $3 OFFSET $2";
 export interface ListIngestionRunsByOrganizationArgs {
     organizationId: string;
     pageOffset: string;
@@ -341,6 +349,8 @@ export interface ListIngestionRunsByOrganizationRow {
     embeddingModelId: string | null;
     embeddingProviderConfigurationId: string | null;
     embeddingDimensions: number | null;
+    stageModelSettings: any;
+    refineLoopCount: number;
     threadCount: number;
     pendingCount: number;
     approvedCount: number;
@@ -352,7 +362,7 @@ export interface ListIngestionRunsByOrganizationRow {
     committedAt: Date | null;
 }
 export declare function listIngestionRunsByOrganization(client: Client, args: ListIngestionRunsByOrganizationArgs): Promise<ListIngestionRunsByOrganizationRow[]>;
-export declare const listIngestionRunsByDocumentQuery = "-- name: ListIngestionRunsByDocument :many\nSELECT\n    id,\n    organization_id,\n    document_id,\n    status,\n    reasoning_model_id,\n    reasoning_provider_configuration_id,\n    embedding_model_id,\n    embedding_provider_configuration_id,\n    embedding_dimensions,\n    thread_count,\n    pending_count,\n    approved_count,\n    rejected_count,\n    error,\n    token_usage,\n    started_at,\n    completed_at,\n    committed_at\nFROM weave.ingestion_runs\nWHERE organization_id = $1\n  AND document_id = $2\nORDER BY started_at DESC";
+export declare const listIngestionRunsByDocumentQuery = "-- name: ListIngestionRunsByDocument :many\nSELECT\n    id,\n    organization_id,\n    document_id,\n    status,\n    reasoning_model_id,\n    reasoning_provider_configuration_id,\n    embedding_model_id,\n    embedding_provider_configuration_id,\n    embedding_dimensions,\n    stage_model_settings,\n    refine_loop_count,\n    thread_count,\n    pending_count,\n    approved_count,\n    rejected_count,\n    error,\n    token_usage,\n    started_at,\n    completed_at,\n    committed_at\nFROM weave.ingestion_runs\nWHERE organization_id = $1\n  AND document_id = $2\nORDER BY started_at DESC";
 export interface ListIngestionRunsByDocumentArgs {
     organizationId: string;
     documentId: string;
@@ -367,6 +377,8 @@ export interface ListIngestionRunsByDocumentRow {
     embeddingModelId: string | null;
     embeddingProviderConfigurationId: string | null;
     embeddingDimensions: number | null;
+    stageModelSettings: any;
+    refineLoopCount: number;
     threadCount: number;
     pendingCount: number;
     approvedCount: number;
@@ -1128,7 +1140,7 @@ export interface ListEntitiesByThreadRow {
     relationshipConfidence: number;
 }
 export declare function listEntitiesByThread(client: Client, args: ListEntitiesByThreadArgs): Promise<ListEntitiesByThreadRow[]>;
-export declare const ensureOrganizationIngestionSettingsQuery = "-- name: EnsureOrganizationIngestionSettings :one\nWITH upsert AS (\n    INSERT INTO weave.org_ingestion_settings (organization_id)\n    VALUES ($1)\n    ON CONFLICT (organization_id) DO UPDATE\n    SET organization_id = EXCLUDED.organization_id\n    RETURNING\n        organization_id,\n        reasoning_model_id,\n        reasoning_provider_configuration_id,\n        embedding_model_id,\n        embedding_provider_configuration_id,\n        updated_at\n)\nSELECT\n    upsert.organization_id,\n    upsert.reasoning_model_id,\n    upsert.reasoning_provider_configuration_id,\n    upsert.embedding_model_id,\n    upsert.embedding_provider_configuration_id,\n    cfg.embedding_dimension AS embedding_dimensions,\n    upsert.updated_at\nFROM upsert\nJOIN weave.database_config cfg\n  ON cfg.id = 1";
+export declare const ensureOrganizationIngestionSettingsQuery = "-- name: EnsureOrganizationIngestionSettings :one\nWITH upsert AS (\n    INSERT INTO weave.org_ingestion_settings (organization_id)\n    VALUES ($1)\n    ON CONFLICT (organization_id) DO UPDATE\n    SET organization_id = EXCLUDED.organization_id\n    RETURNING\n        organization_id,\n        reasoning_model_id,\n        reasoning_provider_configuration_id,\n        embedding_model_id,\n        embedding_provider_configuration_id,\n        classify_model_id,\n        classify_provider_configuration_id,\n        refine_model_id,\n        refine_provider_configuration_id,\n        extract_model_id,\n        extract_provider_configuration_id,\n        embed_model_id,\n        embed_provider_configuration_id,\n        reconcile_model_id,\n        reconcile_provider_configuration_id,\n        diff_model_id,\n        diff_provider_configuration_id,\n        refine_loop_count,\n        updated_at\n)\nSELECT\n    upsert.organization_id,\n    upsert.reasoning_model_id,\n    upsert.reasoning_provider_configuration_id,\n    upsert.embedding_model_id,\n    upsert.embedding_provider_configuration_id,\n    upsert.classify_model_id,\n    upsert.classify_provider_configuration_id,\n    upsert.refine_model_id,\n    upsert.refine_provider_configuration_id,\n    upsert.extract_model_id,\n    upsert.extract_provider_configuration_id,\n    upsert.embed_model_id,\n    upsert.embed_provider_configuration_id,\n    upsert.reconcile_model_id,\n    upsert.reconcile_provider_configuration_id,\n    upsert.diff_model_id,\n    upsert.diff_provider_configuration_id,\n    cfg.embedding_dimension AS embedding_dimensions,\n    cfg.embedding_dimension AS embed_dimensions,\n    upsert.refine_loop_count,\n    upsert.updated_at\nFROM upsert\nJOIN weave.database_config cfg\n  ON cfg.id = 1";
 export interface EnsureOrganizationIngestionSettingsArgs {
     organizationId: string;
 }
@@ -1138,11 +1150,25 @@ export interface EnsureOrganizationIngestionSettingsRow {
     reasoningProviderConfigurationId: string | null;
     embeddingModelId: string | null;
     embeddingProviderConfigurationId: string | null;
+    classifyModelId: string | null;
+    classifyProviderConfigurationId: string | null;
+    refineModelId: string | null;
+    refineProviderConfigurationId: string | null;
+    extractModelId: string | null;
+    extractProviderConfigurationId: string | null;
+    embedModelId: string | null;
+    embedProviderConfigurationId: string | null;
+    reconcileModelId: string | null;
+    reconcileProviderConfigurationId: string | null;
+    diffModelId: string | null;
+    diffProviderConfigurationId: string | null;
     embeddingDimensions: number;
+    embedDimensions: number;
+    refineLoopCount: number;
     updatedAt: Date;
 }
 export declare function ensureOrganizationIngestionSettings(client: Client, args: EnsureOrganizationIngestionSettingsArgs): Promise<EnsureOrganizationIngestionSettingsRow | null>;
-export declare const getOrganizationIngestionSettingsQuery = "-- name: GetOrganizationIngestionSettings :one\nSELECT\n    settings.organization_id,\n    settings.reasoning_model_id,\n    settings.reasoning_provider_configuration_id,\n    settings.embedding_model_id,\n    settings.embedding_provider_configuration_id,\n    cfg.embedding_dimension AS embedding_dimensions,\n    settings.updated_at\nFROM weave.org_ingestion_settings settings\nJOIN weave.database_config cfg\n  ON cfg.id = 1\nWHERE settings.organization_id = $1";
+export declare const getOrganizationIngestionSettingsQuery = "-- name: GetOrganizationIngestionSettings :one\nSELECT\n    settings.organization_id,\n    settings.reasoning_model_id,\n    settings.reasoning_provider_configuration_id,\n    settings.embedding_model_id,\n    settings.embedding_provider_configuration_id,\n    settings.classify_model_id,\n    settings.classify_provider_configuration_id,\n    settings.refine_model_id,\n    settings.refine_provider_configuration_id,\n    settings.extract_model_id,\n    settings.extract_provider_configuration_id,\n    settings.embed_model_id,\n    settings.embed_provider_configuration_id,\n    settings.reconcile_model_id,\n    settings.reconcile_provider_configuration_id,\n    settings.diff_model_id,\n    settings.diff_provider_configuration_id,\n    cfg.embedding_dimension AS embedding_dimensions,\n    cfg.embedding_dimension AS embed_dimensions,\n    settings.refine_loop_count,\n    settings.updated_at\nFROM weave.org_ingestion_settings settings\nJOIN weave.database_config cfg\n  ON cfg.id = 1\nWHERE settings.organization_id = $1";
 export interface GetOrganizationIngestionSettingsArgs {
     organizationId: string;
 }
@@ -1152,11 +1178,25 @@ export interface GetOrganizationIngestionSettingsRow {
     reasoningProviderConfigurationId: string | null;
     embeddingModelId: string | null;
     embeddingProviderConfigurationId: string | null;
+    classifyModelId: string | null;
+    classifyProviderConfigurationId: string | null;
+    refineModelId: string | null;
+    refineProviderConfigurationId: string | null;
+    extractModelId: string | null;
+    extractProviderConfigurationId: string | null;
+    embedModelId: string | null;
+    embedProviderConfigurationId: string | null;
+    reconcileModelId: string | null;
+    reconcileProviderConfigurationId: string | null;
+    diffModelId: string | null;
+    diffProviderConfigurationId: string | null;
     embeddingDimensions: number;
+    embedDimensions: number;
+    refineLoopCount: number;
     updatedAt: Date;
 }
 export declare function getOrganizationIngestionSettings(client: Client, args: GetOrganizationIngestionSettingsArgs): Promise<GetOrganizationIngestionSettingsRow | null>;
-export declare const upsertOrganizationIngestionSettingsQuery = "-- name: UpsertOrganizationIngestionSettings :one\nWITH cfg AS (\n    SELECT embedding_dimension\n    FROM weave.database_config\n    WHERE id = 1\n      AND ($1 = 0 OR $1 = embedding_dimension)\n), upsert AS (\n    INSERT INTO weave.org_ingestion_settings (\n        organization_id,\n        reasoning_model_id,\n        reasoning_provider_configuration_id,\n        embedding_model_id,\n        embedding_provider_configuration_id\n    )\n    SELECT\n        $2,\n        $3,\n        $4,\n        $5,\n        $6\n    FROM cfg\n    ON CONFLICT (organization_id) DO UPDATE\n    SET\n        reasoning_model_id = EXCLUDED.reasoning_model_id,\n        reasoning_provider_configuration_id = EXCLUDED.reasoning_provider_configuration_id,\n        embedding_model_id = EXCLUDED.embedding_model_id,\n        embedding_provider_configuration_id = EXCLUDED.embedding_provider_configuration_id,\n        updated_at = now()\n    RETURNING\n        organization_id,\n        reasoning_model_id,\n        reasoning_provider_configuration_id,\n        embedding_model_id,\n        embedding_provider_configuration_id,\n        updated_at\n)\nSELECT\n    upsert.organization_id,\n    upsert.reasoning_model_id,\n    upsert.reasoning_provider_configuration_id,\n    upsert.embedding_model_id,\n    upsert.embedding_provider_configuration_id,\n    cfg.embedding_dimension AS embedding_dimensions,\n    upsert.updated_at\nFROM upsert\nJOIN cfg\n  ON TRUE";
+export declare const upsertOrganizationIngestionSettingsQuery = "-- name: UpsertOrganizationIngestionSettings :one\nWITH cfg AS (\n    SELECT embedding_dimension\n    FROM weave.database_config\n    WHERE id = 1\n      AND ($1 = 0 OR $1 = embedding_dimension)\n), upsert AS (\n    INSERT INTO weave.org_ingestion_settings (\n        organization_id,\n        reasoning_model_id,\n        reasoning_provider_configuration_id,\n        embedding_model_id,\n        embedding_provider_configuration_id,\n        classify_model_id,\n        classify_provider_configuration_id,\n        refine_model_id,\n        refine_provider_configuration_id,\n        extract_model_id,\n        extract_provider_configuration_id,\n        embed_model_id,\n        embed_provider_configuration_id,\n        reconcile_model_id,\n        reconcile_provider_configuration_id,\n        diff_model_id,\n        diff_provider_configuration_id\n    )\n    SELECT\n        $2,\n        $3,\n        $4,\n        $5,\n        $6,\n        $3,\n        $4,\n        $3,\n        $4,\n        $3,\n        $4,\n        $5,\n        $6,\n        $3,\n        $4,\n        $3,\n        $4\n    FROM cfg\n    ON CONFLICT (organization_id) DO UPDATE\n    SET\n        reasoning_model_id = EXCLUDED.reasoning_model_id,\n        reasoning_provider_configuration_id = EXCLUDED.reasoning_provider_configuration_id,\n        embedding_model_id = EXCLUDED.embedding_model_id,\n        embedding_provider_configuration_id = EXCLUDED.embedding_provider_configuration_id,\n        classify_model_id = EXCLUDED.classify_model_id,\n        classify_provider_configuration_id = EXCLUDED.classify_provider_configuration_id,\n        refine_model_id = EXCLUDED.refine_model_id,\n        refine_provider_configuration_id = EXCLUDED.refine_provider_configuration_id,\n        extract_model_id = EXCLUDED.extract_model_id,\n        extract_provider_configuration_id = EXCLUDED.extract_provider_configuration_id,\n        embed_model_id = EXCLUDED.embed_model_id,\n        embed_provider_configuration_id = EXCLUDED.embed_provider_configuration_id,\n        reconcile_model_id = EXCLUDED.reconcile_model_id,\n        reconcile_provider_configuration_id = EXCLUDED.reconcile_provider_configuration_id,\n        diff_model_id = EXCLUDED.diff_model_id,\n        diff_provider_configuration_id = EXCLUDED.diff_provider_configuration_id,\n        updated_at = now()\n    RETURNING\n        organization_id,\n        reasoning_model_id,\n        reasoning_provider_configuration_id,\n        embedding_model_id,\n        embedding_provider_configuration_id,\n        classify_model_id,\n        classify_provider_configuration_id,\n        refine_model_id,\n        refine_provider_configuration_id,\n        extract_model_id,\n        extract_provider_configuration_id,\n        embed_model_id,\n        embed_provider_configuration_id,\n        reconcile_model_id,\n        reconcile_provider_configuration_id,\n        diff_model_id,\n        diff_provider_configuration_id,\n        refine_loop_count,\n        updated_at\n)\nSELECT\n    upsert.organization_id,\n    upsert.reasoning_model_id,\n    upsert.reasoning_provider_configuration_id,\n    upsert.embedding_model_id,\n    upsert.embedding_provider_configuration_id,\n    upsert.classify_model_id,\n    upsert.classify_provider_configuration_id,\n    upsert.refine_model_id,\n    upsert.refine_provider_configuration_id,\n    upsert.extract_model_id,\n    upsert.extract_provider_configuration_id,\n    upsert.embed_model_id,\n    upsert.embed_provider_configuration_id,\n    upsert.reconcile_model_id,\n    upsert.reconcile_provider_configuration_id,\n    upsert.diff_model_id,\n    upsert.diff_provider_configuration_id,\n    cfg.embedding_dimension AS embedding_dimensions,\n    cfg.embedding_dimension AS embed_dimensions,\n    upsert.refine_loop_count,\n    upsert.updated_at\nFROM upsert\nJOIN cfg\n  ON TRUE";
 export interface UpsertOrganizationIngestionSettingsArgs {
     embeddingDimensions: string | null;
     organizationId: string;
@@ -1171,11 +1211,63 @@ export interface UpsertOrganizationIngestionSettingsRow {
     reasoningProviderConfigurationId: string | null;
     embeddingModelId: string | null;
     embeddingProviderConfigurationId: string | null;
+    classifyModelId: string | null;
+    classifyProviderConfigurationId: string | null;
+    refineModelId: string | null;
+    refineProviderConfigurationId: string | null;
+    extractModelId: string | null;
+    extractProviderConfigurationId: string | null;
+    embedModelId: string | null;
+    embedProviderConfigurationId: string | null;
+    reconcileModelId: string | null;
+    reconcileProviderConfigurationId: string | null;
+    diffModelId: string | null;
+    diffProviderConfigurationId: string | null;
     embeddingDimensions: number;
+    embedDimensions: number;
+    refineLoopCount: number;
     updatedAt: Date;
 }
 export declare function upsertOrganizationIngestionSettings(client: Client, args: UpsertOrganizationIngestionSettingsArgs): Promise<UpsertOrganizationIngestionSettingsRow | null>;
-export declare const syncOrganizationIngestionReasoningDefaultsQuery = "-- name: SyncOrganizationIngestionReasoningDefaults :one\nWITH cfg AS (\n    SELECT embedding_dimension\n    FROM weave.database_config\n    WHERE id = 1\n), upsert AS (\n    INSERT INTO weave.org_ingestion_settings (\n        organization_id,\n        reasoning_model_id,\n        reasoning_provider_configuration_id\n    )\n    SELECT\n        cs.organization_id,\n        cs.default_chat_model_id,\n        cs.default_provider_configuration_id\n    FROM weave.organization_chat_settings cs\n    WHERE cs.organization_id = $1\n    ON CONFLICT (organization_id) DO UPDATE\n    SET\n        reasoning_model_id = COALESCE(weave.org_ingestion_settings.reasoning_model_id, EXCLUDED.reasoning_model_id),\n        reasoning_provider_configuration_id = COALESCE(\n            weave.org_ingestion_settings.reasoning_provider_configuration_id,\n            EXCLUDED.reasoning_provider_configuration_id\n        ),\n        updated_at = now()\n    RETURNING\n        organization_id,\n        reasoning_model_id,\n        reasoning_provider_configuration_id,\n        embedding_model_id,\n        embedding_provider_configuration_id,\n        updated_at\n)\nSELECT\n    upsert.organization_id,\n    upsert.reasoning_model_id,\n    upsert.reasoning_provider_configuration_id,\n    upsert.embedding_model_id,\n    upsert.embedding_provider_configuration_id,\n    cfg.embedding_dimension AS embedding_dimensions,\n    upsert.updated_at\nFROM upsert\nJOIN cfg\n  ON TRUE";
+export declare const upsertOrganizationIngestionStageSettingsQuery = "-- name: UpsertOrganizationIngestionStageSettings :one\nWITH cfg AS (\n    SELECT embedding_dimension\n    FROM weave.database_config\n    WHERE id = 1\n      AND ($1 = 0 OR $1 = embedding_dimension)\n), upsert AS (\n    INSERT INTO weave.org_ingestion_settings (\n        organization_id,\n        classify_model_id,\n        classify_provider_configuration_id,\n        refine_model_id,\n        refine_provider_configuration_id,\n        extract_model_id,\n        extract_provider_configuration_id,\n        embed_model_id,\n        embed_provider_configuration_id,\n        reconcile_model_id,\n        reconcile_provider_configuration_id,\n        diff_model_id,\n        diff_provider_configuration_id,\n        refine_loop_count\n    )\n    SELECT\n        $2,\n        $3,\n        $4,\n        $5,\n        $6,\n        $7,\n        $8,\n        $9,\n        $10,\n        $11,\n        $12,\n        $13,\n        $14,\n        $15\n    FROM cfg\n    ON CONFLICT (organization_id) DO UPDATE\n    SET\n        classify_model_id = EXCLUDED.classify_model_id,\n        classify_provider_configuration_id = EXCLUDED.classify_provider_configuration_id,\n        refine_model_id = EXCLUDED.refine_model_id,\n        refine_provider_configuration_id = EXCLUDED.refine_provider_configuration_id,\n        extract_model_id = EXCLUDED.extract_model_id,\n        extract_provider_configuration_id = EXCLUDED.extract_provider_configuration_id,\n        embed_model_id = EXCLUDED.embed_model_id,\n        embed_provider_configuration_id = EXCLUDED.embed_provider_configuration_id,\n        reconcile_model_id = EXCLUDED.reconcile_model_id,\n        reconcile_provider_configuration_id = EXCLUDED.reconcile_provider_configuration_id,\n        diff_model_id = EXCLUDED.diff_model_id,\n        diff_provider_configuration_id = EXCLUDED.diff_provider_configuration_id,\n        refine_loop_count = EXCLUDED.refine_loop_count,\n        updated_at = now()\n    RETURNING\n        organization_id,\n        classify_model_id,\n        classify_provider_configuration_id,\n        refine_model_id,\n        refine_provider_configuration_id,\n        extract_model_id,\n        extract_provider_configuration_id,\n        embed_model_id,\n        embed_provider_configuration_id,\n        reconcile_model_id,\n        reconcile_provider_configuration_id,\n        diff_model_id,\n        diff_provider_configuration_id,\n        refine_loop_count,\n        updated_at\n)\nSELECT\n    upsert.organization_id,\n    upsert.classify_model_id,\n    upsert.classify_provider_configuration_id,\n    upsert.refine_model_id,\n    upsert.refine_provider_configuration_id,\n    upsert.extract_model_id,\n    upsert.extract_provider_configuration_id,\n    upsert.embed_model_id,\n    upsert.embed_provider_configuration_id,\n    upsert.reconcile_model_id,\n    upsert.reconcile_provider_configuration_id,\n    upsert.diff_model_id,\n    upsert.diff_provider_configuration_id,\n    cfg.embedding_dimension AS embedding_dimensions,\n    cfg.embedding_dimension AS embed_dimensions,\n    upsert.refine_loop_count,\n    upsert.updated_at\nFROM upsert\nJOIN cfg\n  ON TRUE";
+export interface UpsertOrganizationIngestionStageSettingsArgs {
+    embedDimensions: string | null;
+    organizationId: string;
+    classifyModelId: string | null;
+    classifyProviderConfigurationId: string | null;
+    refineModelId: string | null;
+    refineProviderConfigurationId: string | null;
+    extractModelId: string | null;
+    extractProviderConfigurationId: string | null;
+    embedModelId: string | null;
+    embedProviderConfigurationId: string | null;
+    reconcileModelId: string | null;
+    reconcileProviderConfigurationId: string | null;
+    diffModelId: string | null;
+    diffProviderConfigurationId: string | null;
+    refineLoopCount: number;
+}
+export interface UpsertOrganizationIngestionStageSettingsRow {
+    organizationId: string;
+    classifyModelId: string | null;
+    classifyProviderConfigurationId: string | null;
+    refineModelId: string | null;
+    refineProviderConfigurationId: string | null;
+    extractModelId: string | null;
+    extractProviderConfigurationId: string | null;
+    embedModelId: string | null;
+    embedProviderConfigurationId: string | null;
+    reconcileModelId: string | null;
+    reconcileProviderConfigurationId: string | null;
+    diffModelId: string | null;
+    diffProviderConfigurationId: string | null;
+    embeddingDimensions: number;
+    embedDimensions: number;
+    refineLoopCount: number;
+    updatedAt: Date;
+}
+export declare function upsertOrganizationIngestionStageSettings(client: Client, args: UpsertOrganizationIngestionStageSettingsArgs): Promise<UpsertOrganizationIngestionStageSettingsRow | null>;
+export declare const syncOrganizationIngestionReasoningDefaultsQuery = "-- name: SyncOrganizationIngestionReasoningDefaults :one\nWITH cfg AS (\n    SELECT embedding_dimension\n    FROM weave.database_config\n    WHERE id = 1\n), upsert AS (\n    INSERT INTO weave.org_ingestion_settings (\n        organization_id,\n        reasoning_model_id,\n        reasoning_provider_configuration_id,\n        classify_model_id,\n        classify_provider_configuration_id,\n        refine_model_id,\n        refine_provider_configuration_id,\n        extract_model_id,\n        extract_provider_configuration_id,\n        reconcile_model_id,\n        reconcile_provider_configuration_id,\n        diff_model_id,\n        diff_provider_configuration_id\n    )\n    SELECT\n        cs.organization_id,\n        cs.default_chat_model_id,\n        cs.default_provider_configuration_id,\n        cs.default_chat_model_id,\n        cs.default_provider_configuration_id,\n        cs.default_chat_model_id,\n        cs.default_provider_configuration_id,\n        cs.default_chat_model_id,\n        cs.default_provider_configuration_id,\n        cs.default_chat_model_id,\n        cs.default_provider_configuration_id,\n        cs.default_chat_model_id,\n        cs.default_provider_configuration_id\n    FROM weave.organization_chat_settings cs\n    WHERE cs.organization_id = $1\n    ON CONFLICT (organization_id) DO UPDATE\n    SET\n        reasoning_model_id = COALESCE(weave.org_ingestion_settings.reasoning_model_id, EXCLUDED.reasoning_model_id),\n        reasoning_provider_configuration_id = COALESCE(\n            weave.org_ingestion_settings.reasoning_provider_configuration_id,\n            EXCLUDED.reasoning_provider_configuration_id\n        ),\n        classify_model_id = COALESCE(weave.org_ingestion_settings.classify_model_id, EXCLUDED.classify_model_id),\n        classify_provider_configuration_id = COALESCE(\n            weave.org_ingestion_settings.classify_provider_configuration_id,\n            EXCLUDED.classify_provider_configuration_id\n        ),\n        refine_model_id = COALESCE(weave.org_ingestion_settings.refine_model_id, EXCLUDED.refine_model_id),\n        refine_provider_configuration_id = COALESCE(\n            weave.org_ingestion_settings.refine_provider_configuration_id,\n            EXCLUDED.refine_provider_configuration_id\n        ),\n        extract_model_id = COALESCE(weave.org_ingestion_settings.extract_model_id, EXCLUDED.extract_model_id),\n        extract_provider_configuration_id = COALESCE(\n            weave.org_ingestion_settings.extract_provider_configuration_id,\n            EXCLUDED.extract_provider_configuration_id\n        ),\n        reconcile_model_id = COALESCE(weave.org_ingestion_settings.reconcile_model_id, EXCLUDED.reconcile_model_id),\n        reconcile_provider_configuration_id = COALESCE(\n            weave.org_ingestion_settings.reconcile_provider_configuration_id,\n            EXCLUDED.reconcile_provider_configuration_id\n        ),\n        diff_model_id = COALESCE(weave.org_ingestion_settings.diff_model_id, EXCLUDED.diff_model_id),\n        diff_provider_configuration_id = COALESCE(\n            weave.org_ingestion_settings.diff_provider_configuration_id,\n            EXCLUDED.diff_provider_configuration_id\n        ),\n        updated_at = now()\n    RETURNING\n        organization_id,\n        reasoning_model_id,\n        reasoning_provider_configuration_id,\n        embedding_model_id,\n        embedding_provider_configuration_id,\n        classify_model_id,\n        classify_provider_configuration_id,\n        refine_model_id,\n        refine_provider_configuration_id,\n        extract_model_id,\n        extract_provider_configuration_id,\n        embed_model_id,\n        embed_provider_configuration_id,\n        reconcile_model_id,\n        reconcile_provider_configuration_id,\n        diff_model_id,\n        diff_provider_configuration_id,\n        refine_loop_count,\n        updated_at\n)\nSELECT\n    upsert.organization_id,\n    upsert.reasoning_model_id,\n    upsert.reasoning_provider_configuration_id,\n    upsert.embedding_model_id,\n    upsert.embedding_provider_configuration_id,\n    upsert.classify_model_id,\n    upsert.classify_provider_configuration_id,\n    upsert.refine_model_id,\n    upsert.refine_provider_configuration_id,\n    upsert.extract_model_id,\n    upsert.extract_provider_configuration_id,\n    upsert.embed_model_id,\n    upsert.embed_provider_configuration_id,\n    upsert.reconcile_model_id,\n    upsert.reconcile_provider_configuration_id,\n    upsert.diff_model_id,\n    upsert.diff_provider_configuration_id,\n    cfg.embedding_dimension AS embedding_dimensions,\n    cfg.embedding_dimension AS embed_dimensions,\n    upsert.refine_loop_count,\n    upsert.updated_at\nFROM upsert\nJOIN cfg\n  ON TRUE";
 export interface SyncOrganizationIngestionReasoningDefaultsArgs {
     organizationId: string;
 }
@@ -1185,10 +1277,68 @@ export interface SyncOrganizationIngestionReasoningDefaultsRow {
     reasoningProviderConfigurationId: string | null;
     embeddingModelId: string | null;
     embeddingProviderConfigurationId: string | null;
+    classifyModelId: string | null;
+    classifyProviderConfigurationId: string | null;
+    refineModelId: string | null;
+    refineProviderConfigurationId: string | null;
+    extractModelId: string | null;
+    extractProviderConfigurationId: string | null;
+    embedModelId: string | null;
+    embedProviderConfigurationId: string | null;
+    reconcileModelId: string | null;
+    reconcileProviderConfigurationId: string | null;
+    diffModelId: string | null;
+    diffProviderConfigurationId: string | null;
     embeddingDimensions: number;
+    embedDimensions: number;
+    refineLoopCount: number;
     updatedAt: Date;
 }
 export declare function syncOrganizationIngestionReasoningDefaults(client: Client, args: SyncOrganizationIngestionReasoningDefaultsArgs): Promise<SyncOrganizationIngestionReasoningDefaultsRow | null>;
+export declare const listThreadFacetsByThreadIDsQuery = "-- name: ListThreadFacetsByThreadIDs :many\nSELECT\n    thread_id,\n    organization_id,\n    knowledge_type,\n    score,\n    facet_fields,\n    rationale,\n    is_primary,\n    created_at,\n    updated_at\nFROM weave.thread_facets\nWHERE organization_id = $1\n  AND thread_id = ANY($2::uuid[])\nORDER BY thread_id ASC, is_primary DESC, score DESC, knowledge_type ASC";
+export interface ListThreadFacetsByThreadIDsArgs {
+    organizationId: string;
+    threadIds: string[];
+}
+export interface ListThreadFacetsByThreadIDsRow {
+    threadId: string;
+    organizationId: string;
+    knowledgeType: string;
+    score: number;
+    facetFields: any;
+    rationale: string;
+    isPrimary: boolean;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function listThreadFacetsByThreadIDs(client: Client, args: ListThreadFacetsByThreadIDsArgs): Promise<ListThreadFacetsByThreadIDsRow[]>;
+export declare const upsertThreadFacetQuery = "-- name: UpsertThreadFacet :one\nWITH scoped_thread AS (\n    SELECT t.id\n    FROM weave.threads t\n    WHERE t.organization_id = $1\n      AND t.id = $2\n), clear_primary AS (\n    UPDATE weave.thread_facets tf\n    SET is_primary = FALSE,\n        updated_at = now()\n    WHERE tf.organization_id = $1\n      AND tf.thread_id = $2\n      AND $3::boolean\n    RETURNING tf.thread_id\n), upsert AS (\n    INSERT INTO weave.thread_facets (\n        thread_id,\n        organization_id,\n        knowledge_type,\n        score,\n        facet_fields,\n        rationale,\n        is_primary\n    )\n    SELECT\n        scoped_thread.id,\n        $1,\n        $4,\n        $5,\n        $6,\n        $7,\n        $3\n    FROM scoped_thread\n    ON CONFLICT (thread_id, knowledge_type) DO UPDATE\n    SET\n        score = EXCLUDED.score,\n        facet_fields = EXCLUDED.facet_fields,\n        rationale = EXCLUDED.rationale,\n        is_primary = EXCLUDED.is_primary,\n        updated_at = now()\n    RETURNING\n        thread_id,\n        organization_id,\n        knowledge_type,\n        score,\n        facet_fields,\n        rationale,\n        is_primary,\n        created_at,\n        updated_at\n)\nSELECT\n    thread_id,\n    organization_id,\n    knowledge_type,\n    score,\n    facet_fields,\n    rationale,\n    is_primary,\n    created_at,\n    updated_at\nFROM upsert";
+export interface UpsertThreadFacetArgs {
+    organizationId: string;
+    threadId: string;
+    isPrimary: boolean;
+    knowledgeType: string;
+    score: number;
+    facetFields: any;
+    rationale: string;
+}
+export interface UpsertThreadFacetRow {
+    threadId: string;
+    organizationId: string;
+    knowledgeType: string;
+    score: number;
+    facetFields: any;
+    rationale: string;
+    isPrimary: boolean;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function upsertThreadFacet(client: Client, args: UpsertThreadFacetArgs): Promise<UpsertThreadFacetRow | null>;
+export declare const deleteThreadFacetsByThreadQuery = "-- name: DeleteThreadFacetsByThread :execrows\nDELETE FROM weave.thread_facets\nWHERE organization_id = $1\n  AND thread_id = $2";
+export interface DeleteThreadFacetsByThreadArgs {
+    organizationId: string;
+    threadId: string;
+}
 export declare const listFacetConfigsQuery = "-- name: ListFacetConfigs :many\nSELECT\n    organization_id,\n    knowledge_type,\n    label,\n    extraction_hint,\n    required_fields,\n    active\nFROM weave.org_facet_config\nWHERE organization_id = $1\nORDER BY knowledge_type ASC";
 export interface ListFacetConfigsArgs {
     organizationId: string;