gitea/infio-copilot
1
0
Fork 0
mirror of https://github.com/EthanMarti/infio-copilot.git synced 2026-01-16 08:21:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

添加嵌入管理器的初始化和清理逻辑，更新嵌入模型加载和嵌入测试的实现。

Browse Source
This commit is contained in:
duanfuxiang 2025-07-04 09:33:08 +08:00
parent 65c5df3d22
commit 8b3babc28e
1 changed files with 20 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/main.ts
View File

@ -14,6 +14,7 @@ import { RAGEngine } from './core/rag/rag-engine'
import { TransEngine } from './core/transformations/trans-engine'
import { DBManager } from './database/database-manager'
import { migrateToJsonDatabase } from './database/json/migrateToJsonDatabase'
import { EmbeddingManager } from './embedworker/EmbeddingManager'
import EventListener from "./event-listener"
import { t } from './lang/helpers'
import { PreviewView } from './PreviewView'
@ -52,6 +53,7 @@ export default class InfioPlugin extends Plugin {
mcpHub: McpHub | null = null
ragEngine: RAGEngine | null = null
transEngine: TransEngine | null = null
embeddingManager: EmbeddingManager | null = null
inlineEdit: InlineEdit | null = null
diffStrategy?: DiffStrategy
dataviewManager: DataviewManager | null = null
@ -73,6 +75,10 @@ export default class InfioPlugin extends Plugin {
// initialize dataview manager
this.dataviewManager = createDataviewManager(this.app)
// initialize embedding manager
this.embeddingManager = new EmbeddingManager()
console.log('EmbeddingManager initialized')
// add icon to ribbon
this.addRibbonIcon('wand-sparkles', t('main.openInfioCopilot'), () =>
this.openChatView(),
@ -426,16 +432,18 @@ export default class InfioPlugin extends Plugin {
name: '测试本地嵌入模型',
callback: async () => {
try {
// 动态导入嵌入管理器
const { embeddingManager } = await import('./embedworker/index');
if (!this.embeddingManager) {
new Notice('EmbeddingManager 未初始化', 5000);
return;
}
// 加载模型
await embeddingManager.loadModel("Xenova/all-MiniLM-L6-v2", true);
await this.embeddingManager.loadModel("Xenova/all-MiniLM-L6-v2", true);
// 测试嵌入 "hello world"
const testText = "hello world";
const result = await embeddingManager.embed(testText);
const result = await this.embeddingManager.embed(testText);
// 显示结果
const resultMessage = `
@ -480,6 +488,9 @@ export default class InfioPlugin extends Plugin {
// MCP Hub cleanup
this.mcpHub?.dispose()
this.mcpHub = null
// EmbeddingManager cleanup
this.embeddingManager?.terminate()
this.embeddingManager = null
// Dataview cleanup
this.dataviewManager = null
}
@ -638,6 +649,10 @@ export default class InfioPlugin extends Plugin {
return this.transEngineInitPromise
}
getEmbeddingManager(): EmbeddingManager | null {
return this.embeddingManager
}
private async migrateToJsonStorage() {
try {
const dbManager = await this.getDbManager()