Dockery/Trilium
1
0
Fork 0
mirror of https://github.com/zadam/trilium.git synced 2025-11-07 05:46:10 +01:00
Code Issues Packages Projects Releases Activity

feat(llm): create endpoints for starting/stopping embeddings

Browse Source
This commit is contained in:
perf3ct
2025-06-05 18:47:25 +00:00
parent a084805762
commit 49e123f399
7 changed files with 197 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
apps/server/src/services/llm/ai_service_manager.ts
View File

@@ -605,6 +605,7 @@ export class AIServiceManager implements IAIServiceManager {
private setupProviderChangeListener(): void {
// List of AI-related options that should trigger service recreation
const aiRelatedOptions = [
'aiEnabled',
'aiSelectedProvider',
'embeddingSelectedProvider',
'openaiApiKey',
@@ -618,10 +619,29 @@ export class AIServiceManager implements IAIServiceManager {
'voyageApiKey'
];
eventService.subscribe(['entityChanged'], ({ entityName, entity }) => {
eventService.subscribe(['entityChanged'], async ({ entityName, entity }) => {
if (entityName === 'options' && entity && aiRelatedOptions.includes(entity.name)) {
log.info(`AI-related option '${entity.name}' changed, recreating LLM services`);
this.recreateServices();
// Special handling for aiEnabled toggle
if (entity.name === 'aiEnabled') {
const isEnabled = entity.value === 'true';
if (isEnabled) {
log.info('AI features enabled, initializing AI service and embeddings');
// Initialize the AI service
await this.initialize();
// Initialize embeddings through index service
await indexService.startEmbeddingGeneration();
} else {
log.info('AI features disabled, stopping embeddings');
// Stop embeddings through index service
await indexService.stopEmbeddingGeneration();
}
} else {
// For other AI-related options, just recreate services
this.recreateServices();
}
}
});
}