Tools/Notes
1
0
Fork 0
mirror of https://github.com/TriliumNext/Notes.git synced 2025-07-30 19:52:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove isEnabled from embedding providers

Browse Source
This commit is contained in:
perf3ct 2025-03-24 20:35:46 +00:00
parent 0707266dc1
commit 150b0f0977
No known key found for this signature in database
GPG Key ID: 569C4EEC436F5232
2 changed files with 10 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
db/migrations/0229__vector_embeddings.sql
View File

@ -37,7 +37,6 @@ CREATE TABLE IF NOT EXISTS "embedding_queue" (
CREATE TABLE IF NOT EXISTS "embedding_providers" ( CREATE TABLE IF NOT EXISTS "embedding_providers" (
"providerId" TEXT NOT NULL PRIMARY KEY, "providerId" TEXT NOT NULL PRIMARY KEY,
"name" TEXT NOT NULL, "name" TEXT NOT NULL,
"isEnabled" INTEGER NOT NULL DEFAULT 0,
"priority" INTEGER NOT NULL DEFAULT 0, "priority" INTEGER NOT NULL DEFAULT 0,
"config" TEXT NOT NULL, -- JSON config object "config" TEXT NOT NULL, -- JSON config object
"dateCreated" TEXT NOT NULL, "dateCreated" TEXT NOT NULL,

28
src/services/llm/embeddings/providers.ts
View File

@ -96,17 +96,16 @@ export async function getEnabledEmbeddingProviders(): Promise<EmbeddingProvider[
return []; return [];
} }
// Get enabled providers from database // Get providers from database ordered by priority
const enabledProviders = await sql.getRows(` const dbProviders = await sql.getRows(`
SELECT providerId, name, config SELECT providerId, name, config
FROM embedding_providers FROM embedding_providers
WHERE isEnabled = 1
ORDER BY priority DESC` ORDER BY priority DESC`
); );
const result: EmbeddingProvider[] = []; const result: EmbeddingProvider[] = [];
for (const row of enabledProviders) { for (const row of dbProviders) {
const rowData = row as any; const rowData = row as any;
const provider = providers.get(rowData.name); const provider = providers.get(rowData.name);
@ -127,7 +126,6 @@ export async function getEnabledEmbeddingProviders(): Promise<EmbeddingProvider[
export async function createEmbeddingProviderConfig( export async function createEmbeddingProviderConfig(
name: string, name: string,
config: EmbeddingConfig, config: EmbeddingConfig,
isEnabled = false,
priority = 0 priority = 0
): Promise<string> { ): Promise<string> {
const providerId = randomString(16); const providerId = randomString(16);
@ -136,10 +134,10 @@ export async function createEmbeddingProviderConfig(
await sql.execute(` await sql.execute(`
INSERT INTO embedding_providers INSERT INTO embedding_providers
(providerId, name, isEnabled, priority, config, (providerId, name, priority, config,
dateCreated, utcDateCreated, dateModified, utcDateModified) dateCreated, utcDateCreated, dateModified, utcDateModified)
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`, VALUES (?, ?, ?, ?, ?, ?, ?, ?)`,
[providerId, name, isEnabled ? 1 : 0, priority, JSON.stringify(config), [providerId, name, priority, JSON.stringify(config),
now, utcNow, now, utcNow] now, utcNow, now, utcNow]
); );
@ -151,7 +149,6 @@ export async function createEmbeddingProviderConfig(
*/ */
export async function updateEmbeddingProviderConfig( export async function updateEmbeddingProviderConfig(
providerId: string, providerId: string,
isEnabled?: boolean,
priority?: number, priority?: number,
config?: EmbeddingConfig config?: EmbeddingConfig
): Promise<boolean> { ): Promise<boolean> {
@ -172,11 +169,6 @@ export async function updateEmbeddingProviderConfig(
const updates = []; const updates = [];
const params: any[] = []; const params: any[] = [];
if (isEnabled !== undefined) {
updates.push("isEnabled = ?");
params.push(isEnabled ? 1 : 0);
}
if (priority !== undefined) { if (priority !== undefined) {
updates.push("priority = ?"); updates.push("priority = ?");
params.push(priority); params.push(priority);
@ -256,7 +248,7 @@ export async function initializeDefaultProviders() {
model: openaiModel, model: openaiModel,
dimension: 1536, dimension: 1536,
type: 'float32' type: 'float32'
}, true, 100); }, 100);
} }
} }
@ -285,7 +277,7 @@ export async function initializeDefaultProviders() {
model: voyageModel, model: voyageModel,
dimension: 1024, dimension: 1024,
type: 'float32' type: 'float32'
}, true, 75); }, 75);
} }
} }
@ -322,7 +314,7 @@ export async function initializeDefaultProviders() {
model: embeddingModel, model: embeddingModel,
dimension: ollamaProvider.getDimension(), dimension: ollamaProvider.getDimension(),
type: 'float32' type: 'float32'
}, true, 50); }, 50);
} }
} catch (error: any) { } catch (error: any) {
log.error(`Error initializing Ollama embedding provider: ${error.message || 'Unknown error'}`); log.error(`Error initializing Ollama embedding provider: ${error.message || 'Unknown error'}`);
@ -347,7 +339,7 @@ export async function initializeDefaultProviders() {
model: 'local', model: 'local',
dimension: 384, dimension: 384,
type: 'float32' type: 'float32'
}, true, 10); }, 10);
} }
} catch (error: any) { } catch (error: any) {
log.error(`Error initializing default embedding providers: ${error.message || 'Unknown error'}`); log.error(`Error initializing default embedding providers: ${error.message || 'Unknown error'}`);