diff --git a/db/migrations/0229__vector_embeddings.sql b/db/migrations/0229__vector_embeddings.sql index 742204a16..45f14fddf 100644 --- a/db/migrations/0229__vector_embeddings.sql +++ b/db/migrations/0229__vector_embeddings.sql @@ -37,7 +37,6 @@ CREATE TABLE IF NOT EXISTS "embedding_queue" ( CREATE TABLE IF NOT EXISTS "embedding_providers" ( "providerId" TEXT NOT NULL PRIMARY KEY, "name" TEXT NOT NULL, - "isEnabled" INTEGER NOT NULL DEFAULT 0, "priority" INTEGER NOT NULL DEFAULT 0, "config" TEXT NOT NULL, -- JSON config object "dateCreated" TEXT NOT NULL, diff --git a/src/services/llm/embeddings/providers.ts b/src/services/llm/embeddings/providers.ts index cb6bc6bac..5bb488967 100644 --- a/src/services/llm/embeddings/providers.ts +++ b/src/services/llm/embeddings/providers.ts @@ -96,17 +96,16 @@ export async function getEnabledEmbeddingProviders(): Promise { const providerId = randomString(16); @@ -136,10 +134,10 @@ export async function createEmbeddingProviderConfig( await sql.execute(` INSERT INTO embedding_providers - (providerId, name, isEnabled, priority, config, + (providerId, name, priority, config, dateCreated, utcDateCreated, dateModified, utcDateModified) - VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`, - [providerId, name, isEnabled ? 1 : 0, priority, JSON.stringify(config), + VALUES (?, ?, ?, ?, ?, ?, ?, ?)`, + [providerId, name, priority, JSON.stringify(config), now, utcNow, now, utcNow] ); @@ -151,7 +149,6 @@ export async function createEmbeddingProviderConfig( */ export async function updateEmbeddingProviderConfig( providerId: string, - isEnabled?: boolean, priority?: number, config?: EmbeddingConfig ): Promise { @@ -172,11 +169,6 @@ export async function updateEmbeddingProviderConfig( const updates = []; const params: any[] = []; - if (isEnabled !== undefined) { - updates.push("isEnabled = ?"); - params.push(isEnabled ? 1 : 0); - } - if (priority !== undefined) { updates.push("priority = ?"); params.push(priority); @@ -256,7 +248,7 @@ export async function initializeDefaultProviders() { model: openaiModel, dimension: 1536, type: 'float32' - }, true, 100); + }, 100); } } @@ -285,7 +277,7 @@ export async function initializeDefaultProviders() { model: voyageModel, dimension: 1024, type: 'float32' - }, true, 75); + }, 75); } } @@ -322,7 +314,7 @@ export async function initializeDefaultProviders() { model: embeddingModel, dimension: ollamaProvider.getDimension(), type: 'float32' - }, true, 50); + }, 50); } } catch (error: any) { log.error(`Error initializing Ollama embedding provider: ${error.message || 'Unknown error'}`); @@ -347,7 +339,7 @@ export async function initializeDefaultProviders() { model: 'local', dimension: 384, type: 'float32' - }, true, 10); + }, 10); } } catch (error: any) { log.error(`Error initializing default embedding providers: ${error.message || 'Unknown error'}`);