@sunub/obsidian-mcp-server 0.3.4 → 0.3.9
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/build/config.d.ts +2 -27
- package/build/config.js +11 -31
- package/build/config.js.map +1 -1
- package/build/db.js +3 -0
- package/build/db.js.map +1 -1
- package/build/index.js +28 -6
- package/build/index.js.map +1 -1
- package/build/server.js +16 -13
- package/build/server.js.map +1 -1
- package/build/setup.d.ts +4 -0
- package/build/setup.js +114 -0
- package/build/setup.js.map +1 -0
- package/build/tools/create_document_with_properties/index.d.ts +5 -5
- package/build/tools/create_document_with_properties/index.js +2 -2
- package/build/tools/create_document_with_properties/index.js.map +1 -1
- package/build/tools/create_document_with_properties/params.d.ts +8 -8
- package/build/tools/create_document_with_properties/params.js +1 -1
- package/build/tools/create_document_with_properties/params.js.map +1 -1
- package/build/tools/generate_property/index.d.ts +1 -1
- package/build/tools/generate_property/index.js +1 -1
- package/build/tools/generate_property/index.js.map +1 -1
- package/build/tools/generate_property/params.d.ts +4 -4
- package/build/tools/index.d.ts +8 -8
- package/build/tools/index.js +5 -5
- package/build/tools/index.js.map +1 -1
- package/build/tools/organize_attachments/index.d.ts +1 -1
- package/build/tools/organize_attachments/index.js +2 -2
- package/build/tools/organize_attachments/index.js.map +1 -1
- package/build/tools/vault/index.d.ts +2 -2
- package/build/tools/vault/index.js +7 -8
- package/build/tools/vault/index.js.map +1 -1
- package/build/tools/vault/params.d.ts +4 -4
- package/build/tools/vault/types/collect_context.d.ts +58 -58
- package/build/tools/vault/types/collect_context.js +1 -1
- package/build/tools/vault/types/collect_context.js.map +1 -1
- package/build/tools/vault/types/list_all.d.ts +42 -42
- package/build/tools/vault/types/list_all.js +2 -2
- package/build/tools/vault/types/list_all.js.map +1 -1
- package/build/tools/vault/types/read_specific.d.ts +6 -6
- package/build/tools/vault/types/read_specific.js +1 -1
- package/build/tools/vault/types/read_specific.js.map +1 -1
- package/build/tools/vault/types/search.d.ts +70 -70
- package/build/tools/vault/utils/actions/collect_context.d.ts +1 -1
- package/build/tools/vault/utils/actions/collect_context.js +4 -4
- package/build/tools/vault/utils/actions/collect_context.js.map +1 -1
- package/build/tools/vault/utils/actions/list_all.d.ts +1 -1
- package/build/tools/vault/utils/actions/list_all.js +1 -1
- package/build/tools/vault/utils/actions/list_all.js.map +1 -1
- package/build/tools/vault/utils/actions/load_memory.d.ts +1 -1
- package/build/tools/vault/utils/actions/load_memory.js +3 -3
- package/build/tools/vault/utils/actions/load_memory.js.map +1 -1
- package/build/tools/vault/utils/actions/read.d.ts +1 -1
- package/build/tools/vault/utils/actions/read.js +1 -1
- package/build/tools/vault/utils/actions/read.js.map +1 -1
- package/build/tools/vault/utils/actions/search.d.ts +1 -1
- package/build/tools/vault/utils/actions/search.js +25 -15
- package/build/tools/vault/utils/actions/search.js.map +1 -1
- package/build/tools/vault/utils/shared.d.ts +1 -1
- package/build/tools/vault/utils.d.ts +7 -8
- package/build/tools/vault/utils.js +7 -8
- package/build/tools/vault/utils.js.map +1 -1
- package/build/tools/write_property/index.d.ts +5 -5
- package/build/tools/write_property/index.js +1 -1
- package/build/tools/write_property/index.js.map +1 -1
- package/build/tools/write_property/params.d.ts +8 -8
- package/build/utils/DirectoryWalker.d.ts +1 -1
- package/build/utils/Embedder.d.ts +11 -0
- package/build/utils/Embedder.js +46 -0
- package/build/utils/Embedder.js.map +1 -0
- package/build/utils/Indexer.d.ts +4 -2
- package/build/utils/Indexer.js +19 -2
- package/build/utils/Indexer.js.map +1 -1
- package/build/utils/LLMClient.js +16 -0
- package/build/utils/LLMClient.js.map +1 -1
- package/build/utils/LocalReranker.d.ts +17 -0
- package/build/utils/LocalReranker.js +71 -0
- package/build/utils/LocalReranker.js.map +1 -0
- package/build/utils/RAGIndexer.d.ts +10 -1
- package/build/utils/RAGIndexer.js +59 -27
- package/build/utils/RAGIndexer.js.map +1 -1
- package/build/utils/RerankerClient.d.ts +6 -6
- package/build/utils/RerankerClient.js +20 -1
- package/build/utils/RerankerClient.js.map +1 -1
- package/build/utils/VaultManger/VaultManager.d.ts +38 -2
- package/build/utils/VaultManger/VaultManager.js +268 -13
- package/build/utils/VaultManger/VaultManager.js.map +1 -1
- package/build/utils/VaultManger/index.d.ts +3 -3
- package/build/utils/VaultManger/index.js +2 -2
- package/build/utils/VaultManger/index.js.map +1 -1
- package/build/utils/VaultManger/types.d.ts +1 -1
- package/build/utils/VaultWatcher.d.ts +1 -1
- package/build/utils/VaultWatcher.js +12 -55
- package/build/utils/VaultWatcher.js.map +1 -1
- package/build/utils/VectorDB.js +31 -6
- package/build/utils/VectorDB.js.map +1 -1
- package/build/utils/constants.d.ts +3 -0
- package/build/utils/constants.js +14 -0
- package/build/utils/constants.js.map +1 -0
- package/build/utils/getVaultManager.d.ts +1 -1
- package/build/utils/getVaultManager.js +1 -1
- package/build/utils/getVaultManager.js.map +1 -1
- package/build/utils/processor/MatterParser.d.ts +1 -1
- package/build/utils/processor/MatterParser.js +1 -1
- package/build/utils/processor/MatterParser.js.map +1 -1
- package/build/utils/processor/types.d.ts +4 -0
- package/build/utils/processor/types.js +1 -0
- package/build/utils/processor/types.js.map +1 -1
- package/package.json +16 -49
- package/LICENSE +0 -15
- package/README.md +0 -512
- package/build/cli/AppContainer.d.ts +0 -1
- package/build/cli/AppContainer.js +0 -197
- package/build/cli/AppContainer.js.map +0 -1
- package/build/cli/app.d.ts +0 -1
- package/build/cli/app.js +0 -7
- package/build/cli/app.js.map +0 -1
- package/build/cli/colors.d.ts +0 -15
- package/build/cli/colors.js +0 -15
- package/build/cli/colors.js.map +0 -1
- package/build/cli/commands/types.d.ts +0 -19
- package/build/cli/commands/types.js +0 -11
- package/build/cli/commands/types.js.map +0 -1
- package/build/cli/config/SettingContext.d.ts +0 -1
- package/build/cli/config/SettingContext.js +0 -2
- package/build/cli/config/SettingContext.js.map +0 -1
- package/build/cli/config/mcpServersConfig.d.ts +0 -30
- package/build/cli/config/mcpServersConfig.js +0 -119
- package/build/cli/config/mcpServersConfig.js.map +0 -1
- package/build/cli/config/settingSchema.d.ts +0 -180
- package/build/cli/config/settingSchema.js +0 -128
- package/build/cli/config/settingSchema.js.map +0 -1
- package/build/cli/constants.d.ts +0 -47
- package/build/cli/constants.js +0 -101
- package/build/cli/constants.js.map +0 -1
- package/build/cli/context/InputContext.d.ts +0 -12
- package/build/cli/context/InputContext.js +0 -10
- package/build/cli/context/InputContext.js.map +0 -1
- package/build/cli/context/KeypressContext.d.ts +0 -30
- package/build/cli/context/KeypressContext.js +0 -200
- package/build/cli/context/KeypressContext.js.map +0 -1
- package/build/cli/context/KeypressContext.util.d.ts +0 -7
- package/build/cli/context/KeypressContext.util.js +0 -474
- package/build/cli/context/KeypressContext.util.js.map +0 -1
- package/build/cli/hooks/useAtCompletion.d.ts +0 -12
- package/build/cli/hooks/useAtCompletion.js +0 -5
- package/build/cli/hooks/useAtCompletion.js.map +0 -1
- package/build/cli/hooks/useCommandCompletion.d.ts +0 -59
- package/build/cli/hooks/useCommandCompletion.js +0 -344
- package/build/cli/hooks/useCommandCompletion.js.map +0 -1
- package/build/cli/hooks/useCompletion.d.ts +0 -22
- package/build/cli/hooks/useCompletion.js +0 -84
- package/build/cli/hooks/useCompletion.js.map +0 -1
- package/build/cli/hooks/useDispatcher.d.ts +0 -6
- package/build/cli/hooks/useDispatcher.js +0 -286
- package/build/cli/hooks/useDispatcher.js.map +0 -1
- package/build/cli/hooks/useFocus.d.ts +0 -2
- package/build/cli/hooks/useFocus.js +0 -3
- package/build/cli/hooks/useFocus.js.map +0 -1
- package/build/cli/hooks/useInputHistory.d.ts +0 -24
- package/build/cli/hooks/useInputHistory.js +0 -159
- package/build/cli/hooks/useInputHistory.js.map +0 -1
- package/build/cli/hooks/useKeyMatchers.d.ts +0 -17
- package/build/cli/hooks/useKeyMatchers.js +0 -13
- package/build/cli/hooks/useKeyMatchers.js.map +0 -1
- package/build/cli/hooks/useKeypress.d.ts +0 -19
- package/build/cli/hooks/useKeypress.js +0 -28
- package/build/cli/hooks/useKeypress.js.map +0 -1
- package/build/cli/hooks/useLlmStream.d.ts +0 -12
- package/build/cli/hooks/useLlmStream.js +0 -365
- package/build/cli/hooks/useLlmStream.js.map +0 -1
- package/build/cli/hooks/useMcpClient.d.ts +0 -10
- package/build/cli/hooks/useMcpClient.js +0 -100
- package/build/cli/hooks/useMcpClient.js.map +0 -1
- package/build/cli/hooks/useMcpManager.d.ts +0 -22
- package/build/cli/hooks/useMcpManager.js +0 -86
- package/build/cli/hooks/useMcpManager.js.map +0 -1
- package/build/cli/hooks/usePromptCompletion.d.ts +0 -19
- package/build/cli/hooks/usePromptCompletion.js +0 -13
- package/build/cli/hooks/usePromptCompletion.js.map +0 -1
- package/build/cli/hooks/useRagContext.d.ts +0 -6
- package/build/cli/hooks/useRagContext.js +0 -83
- package/build/cli/hooks/useRagContext.js.map +0 -1
- package/build/cli/hooks/useShellCompletion.d.ts +0 -18
- package/build/cli/hooks/useShellCompletion.js +0 -10
- package/build/cli/hooks/useShellCompletion.js.map +0 -1
- package/build/cli/hooks/useSlashCompletion.d.ts +0 -21
- package/build/cli/hooks/useSlashCompletion.js +0 -11
- package/build/cli/hooks/useSlashCompletion.js.map +0 -1
- package/build/cli/hooks/useTerminalSize.d.ts +0 -9
- package/build/cli/hooks/useTerminalSize.js +0 -26
- package/build/cli/hooks/useTerminalSize.js.map +0 -1
- package/build/cli/index.d.ts +0 -1
- package/build/cli/index.js +0 -39
- package/build/cli/index.js.map +0 -1
- package/build/cli/key/input.d.ts +0 -5
- package/build/cli/key/input.js +0 -9
- package/build/cli/key/input.js.map +0 -1
- package/build/cli/key/keyBindings.d.ts +0 -152
- package/build/cli/key/keyBindings.js +0 -630
- package/build/cli/key/keyBindings.js.map +0 -1
- package/build/cli/key/keyMatchers.d.ts +0 -27
- package/build/cli/key/keyMatchers.js +0 -33
- package/build/cli/key/keyMatchers.js.map +0 -1
- package/build/cli/key/mouse.d.ts +0 -23
- package/build/cli/key/mouse.js +0 -155
- package/build/cli/key/mouse.js.map +0 -1
- package/build/cli/key/text-buffer-bestcase.d.ts +0 -867
- package/build/cli/key/text-buffer-bestcase.js +0 -2754
- package/build/cli/key/text-buffer-bestcase.js.map +0 -1
- package/build/cli/key/text-buffer.d.ts +0 -509
- package/build/cli/key/text-buffer.js +0 -918
- package/build/cli/key/text-buffer.js.map +0 -1
- package/build/cli/main.d.ts +0 -1
- package/build/cli/main.js +0 -20
- package/build/cli/main.js.map +0 -1
- package/build/cli/reference/HistoryItemDisplay.d.ts +0 -23
- package/build/cli/reference/HistoryItemDisplay.js +0 -63
- package/build/cli/reference/HistoryItemDisplay.js.map +0 -1
- package/build/cli/reference/InputContext.d.ts +0 -17
- package/build/cli/reference/InputContext.js +0 -16
- package/build/cli/reference/InputContext.js.map +0 -1
- package/build/cli/reference/KerypressContext.d.ts +0 -37
- package/build/cli/reference/KerypressContext.js +0 -791
- package/build/cli/reference/KerypressContext.js.map +0 -1
- package/build/cli/reference/MainContent.d.ts +0 -6
- package/build/cli/reference/MainContent.js +0 -207
- package/build/cli/reference/MainContent.js.map +0 -1
- package/build/cli/reference/UIStateContext.d.ts +0 -188
- package/build/cli/reference/UIStateContext.js +0 -16
- package/build/cli/reference/UIStateContext.js.map +0 -1
- package/build/cli/reference/text-buffer.d.ts +0 -867
- package/build/cli/reference/text-buffer.js +0 -2754
- package/build/cli/reference/text-buffer.js.map +0 -1
- package/build/cli/reference/useGeminiStream.d.ts +0 -42
- package/build/cli/reference/useGeminiStream.js +0 -1404
- package/build/cli/reference/useGeminiStream.js.map +0 -1
- package/build/cli/services/McpClientService.d.ts +0 -32
- package/build/cli/services/McpClientService.js +0 -123
- package/build/cli/services/McpClientService.js.map +0 -1
- package/build/cli/services/McpManager.d.ts +0 -28
- package/build/cli/services/McpManager.js +0 -146
- package/build/cli/services/McpManager.js.map +0 -1
- package/build/cli/theme/builtin/default-dark.d.ts +0 -7
- package/build/cli/theme/builtin/default-dark.js +0 -143
- package/build/cli/theme/builtin/default-dark.js.map +0 -1
- package/build/cli/theme/builtin/no-color.d.ts +0 -7
- package/build/cli/theme/builtin/no-color.js +0 -125
- package/build/cli/theme/builtin/no-color.js.map +0 -1
- package/build/cli/theme/builtin/tokyonight-dark.d.ts +0 -7
- package/build/cli/theme/builtin/tokyonight-dark.js +0 -146
- package/build/cli/theme/builtin/tokyonight-dark.js.map +0 -1
- package/build/cli/theme/semantic-colors.d.ts +0 -2
- package/build/cli/theme/semantic-colors.js +0 -19
- package/build/cli/theme/semantic-colors.js.map +0 -1
- package/build/cli/theme/semantic-tokens.d.ts +0 -42
- package/build/cli/theme/semantic-tokens.js +0 -77
- package/build/cli/theme/semantic-tokens.js.map +0 -1
- package/build/cli/theme/theme-manager.d.ts +0 -32
- package/build/cli/theme/theme-manager.js +0 -150
- package/build/cli/theme/theme-manager.js.map +0 -1
- package/build/cli/theme/theme.d.ts +0 -107
- package/build/cli/theme/theme.js +0 -333
- package/build/cli/theme/theme.js.map +0 -1
- package/build/cli/tmp.d.ts +0 -1
- package/build/cli/tmp.js +0 -27
- package/build/cli/tmp.js.map +0 -1
- package/build/cli/types.d.ts +0 -72
- package/build/cli/types.js +0 -8
- package/build/cli/types.js.map +0 -1
- package/build/cli/ui/ConnectionStatus.d.ts +0 -13
- package/build/cli/ui/ConnectionStatus.js +0 -24
- package/build/cli/ui/ConnectionStatus.js.map +0 -1
- package/build/cli/ui/HelpCommands.d.ts +0 -5
- package/build/cli/ui/HelpCommands.js +0 -11
- package/build/cli/ui/HelpCommands.js.map +0 -1
- package/build/cli/ui/HistoryItemDisplay.d.ts +0 -9
- package/build/cli/ui/HistoryItemDisplay.js +0 -38
- package/build/cli/ui/HistoryItemDisplay.js.map +0 -1
- package/build/cli/ui/InputPrompt-bestcase.d.ts +0 -63
- package/build/cli/ui/InputPrompt-bestcase.js +0 -1301
- package/build/cli/ui/InputPrompt-bestcase.js.map +0 -1
- package/build/cli/ui/InputPrompt.d.ts +0 -12
- package/build/cli/ui/InputPrompt.js +0 -145
- package/build/cli/ui/InputPrompt.js.map +0 -1
- package/build/cli/ui/InputPrompt.tmp.d.ts +0 -7
- package/build/cli/ui/InputPrompt.tmp.js +0 -215
- package/build/cli/ui/InputPrompt.tmp.js.map +0 -1
- package/build/cli/ui/MainContent.d.ts +0 -10
- package/build/cli/ui/MainContent.js +0 -24
- package/build/cli/ui/MainContent.js.map +0 -1
- package/build/cli/ui/MouseContext.d.ts +0 -1
- package/build/cli/ui/MouseContext.js +0 -2
- package/build/cli/ui/MouseContext.js.map +0 -1
- package/build/cli/ui/SuggestionsDisplay.d.ts +0 -25
- package/build/cli/ui/SuggestionsDisplay.js +0 -45
- package/build/cli/ui/SuggestionsDisplay.js.map +0 -1
- package/build/cli/ui/ThinkingIndicator.d.ts +0 -2
- package/build/cli/ui/ThinkingIndicator.js +0 -16
- package/build/cli/ui/ThinkingIndicator.js.map +0 -1
- package/build/cli/ui/layout/DefaultAppLayout.d.ts +0 -2
- package/build/cli/ui/layout/DefaultAppLayout.js +0 -6
- package/build/cli/ui/layout/DefaultAppLayout.js.map +0 -1
- package/build/cli/ui/shared/ExpandableText.d.ts +0 -14
- package/build/cli/ui/shared/ExpandableText.js +0 -19
- package/build/cli/ui/shared/ExpandableText.js.map +0 -1
- package/build/cli/utils/cache.d.ts +0 -21
- package/build/cli/utils/cache.js +0 -56
- package/build/cli/utils/cache.js.map +0 -1
- package/build/cli/utils/commandUtils.d.ts +0 -11
- package/build/cli/utils/commandUtils.js +0 -16
- package/build/cli/utils/commandUtils.js.map +0 -1
- package/build/cli/utils/debugLogger.d.ts +0 -12
- package/build/cli/utils/debugLogger.js +0 -54
- package/build/cli/utils/debugLogger.js.map +0 -1
- package/build/cli/utils/historyStorage.d.ts +0 -13
- package/build/cli/utils/historyStorage.js +0 -64
- package/build/cli/utils/historyStorage.js.map +0 -1
- package/build/cli/utils/textUtil.d.ts +0 -68
- package/build/cli/utils/textUtil.js +0 -236
- package/build/cli/utils/textUtil.js.map +0 -1
- package/build/tools/vault/utils/actions/search_semantic.d.ts +0 -3
- package/build/tools/vault/utils/actions/search_semantic.js +0 -71
- package/build/tools/vault/utils/actions/search_semantic.js.map +0 -1
package/build/utils/Indexer.d.ts
CHANGED
|
@@ -1,11 +1,13 @@
|
|
|
1
|
-
import type { DocumentIndex } from "
|
|
2
|
-
import type { Semaphore } from "
|
|
1
|
+
import type { DocumentIndex } from "../utils/processor/types.js";
|
|
2
|
+
import type { Semaphore } from "../utils/semaphore.js";
|
|
3
3
|
export declare class Indexer {
|
|
4
4
|
private documentMap;
|
|
5
5
|
private backlinkIndex;
|
|
6
6
|
private invertedIndex;
|
|
7
7
|
get totalFiles(): number;
|
|
8
8
|
build(filePaths: string[], ioSemaphore: Semaphore): Promise<void>;
|
|
9
|
+
upsertFile(filePath: string, ioSemaphore: Semaphore): Promise<void>;
|
|
10
|
+
removeFileEntries(filePath: string): void;
|
|
9
11
|
search(keyword: string): DocumentIndex[];
|
|
10
12
|
getDocument(filePath: string): DocumentIndex | null;
|
|
11
13
|
getAllDocuments(): DocumentIndex[];
|
package/build/utils/Indexer.js
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import { readFile, stat } from "node:fs/promises";
|
|
2
2
|
import { basename, extname, join } from "node:path";
|
|
3
|
-
import * as LinkExtractor from "
|
|
4
|
-
import * as MatterParser from "
|
|
3
|
+
import * as LinkExtractor from "../utils/processor/LinkExtractor.js";
|
|
4
|
+
import * as MatterParser from "../utils/processor/MatterParser.js";
|
|
5
5
|
export class Indexer {
|
|
6
6
|
documentMap = new Map();
|
|
7
7
|
backlinkIndex = new Map();
|
|
@@ -15,6 +15,22 @@ export class Indexer {
|
|
|
15
15
|
await Promise.all(tasks);
|
|
16
16
|
this.buildBacklinkIndex();
|
|
17
17
|
}
|
|
18
|
+
async upsertFile(filePath, ioSemaphore) {
|
|
19
|
+
this.removeFileEntries(filePath);
|
|
20
|
+
await this.processFile(filePath, ioSemaphore);
|
|
21
|
+
this.buildBacklinkIndex();
|
|
22
|
+
}
|
|
23
|
+
removeFileEntries(filePath) {
|
|
24
|
+
for (const [token, fileSet] of this.invertedIndex.entries()) {
|
|
25
|
+
if (fileSet.has(filePath)) {
|
|
26
|
+
fileSet.delete(filePath);
|
|
27
|
+
if (fileSet.size === 0) {
|
|
28
|
+
this.invertedIndex.delete(token);
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
this.documentMap.delete(filePath);
|
|
33
|
+
}
|
|
18
34
|
search(keyword) {
|
|
19
35
|
const lowerKeyword = keyword.toLowerCase().trim();
|
|
20
36
|
if (!lowerKeyword) {
|
|
@@ -76,6 +92,7 @@ export class Indexer {
|
|
|
76
92
|
filePath,
|
|
77
93
|
frontmatter,
|
|
78
94
|
contentLength: content.length,
|
|
95
|
+
mtime: fileStat.mtime.getTime(),
|
|
79
96
|
imageLinks,
|
|
80
97
|
documentLinks,
|
|
81
98
|
};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"Indexer.js","sourceRoot":"","sources":["../../src/utils/Indexer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACpD,OAAO,KAAK,aAAa,MAAM,
|
|
1
|
+
{"version":3,"file":"Indexer.js","sourceRoot":"","sources":["../../src/utils/Indexer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACpD,OAAO,KAAK,aAAa,MAAM,oCAAoC,CAAC;AACpE,OAAO,KAAK,YAAY,MAAM,mCAAmC,CAAC;AAIlE,MAAM,OAAO,OAAO;IACX,WAAW,GAA+B,IAAI,GAAG,EAAE,CAAC;IACpD,aAAa,GAA6B,IAAI,GAAG,EAAE,CAAC;IACpD,aAAa,GAA6B,IAAI,GAAG,EAAE,CAAC;IAE5D,IAAW,UAAU;QACpB,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;IAC9B,CAAC;IAEM,KAAK,CAAC,KAAK,CACjB,SAAmB,EACnB,WAAsB;QAEtB,IAAI,CAAC,KAAK,EAAE,CAAC;QACb,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CACxC,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,WAAW,CAAC,CACvC,CAAC;QACF,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACzB,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC3B,CAAC;IAEM,KAAK,CAAC,UAAU,CACtB,QAAgB,EAChB,WAAsB;QAEtB,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QACjC,MAAM,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAC9C,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC3B,CAAC;IAEM,iBAAiB,CAAC,QAAgB;QACxC,KAAK,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,EAAE,CAAC;YAC7D,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC3B,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACzB,IAAI,OAAO,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;oBACxB,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAClC,CAAC;YACF,CAAC;QACF,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IAEM,MAAM,CAAC,OAAe;QAC5B,MAAM,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;QAClD,IAAI,CAAC,YAAY,EAAE,CAAC;YACnB,OAAO,EAAE,CAAC;QACX,CAAC;QAED,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACrE,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,CAAC;QACX,CAAC;QAED,8BAA8B;QAC9B,6BAA6B;QAC7B,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QACrD,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO,EAAE,CAAC;QACX,CAAC;QAED,IAAI,WAAW,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC;QAEtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YACrD,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBAC1C,OAAO,EAAE,CAAC;YACX,CAAC;YACD,WAAW,GAAG,IAAI,GAAG,CACpB,CAAC,GAAG,WAAW,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CACvD,CAAC;YACF,IAAI,WAAW,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBAC5B,OAAO,EAAE,CAAC;YACX,CAAC;QACF,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC;aAC5B,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;aACjD,MAAM,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,KAAK,SAAS,CAAC,CAAC;IAC1D,CAAC;IAEM,WAAW,CAAC,QAAgB;QAClC,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC;IAC/C,CAAC;IAEM,eAAe;QACrB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;IAC9C,CAAC;IAEM,YAAY,CAAC,QAAgB;QACnC,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CACpC,IAAI,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,CACvC,CAAC;QACF,IAAI,CAAC,UAAU,EAAE,CAAC;YACjB,OAAO,EAAE,CAAC;QACX,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACrD,OAAO,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC/C,CAAC;IAEM,KAAK;QACX,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;QAC3B,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;IAC5B,CAAC;IAEO,KAAK,CAAC,WAAW,CACxB,QAAgB,EAChB,WAAsB;QAEtB,MAAM,WAAW,CAAC,OAAO,EAAE,CAAC;QAC5B,IAAI,CAAC;YACJ,MAAM,WAAW,GAAG,MAAM,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACtD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC;YACtC,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,GAAG,YAAY,CAAC,KAAK,CAClD,QAAQ,EACR,QAAQ,CAAC,SAAS,CAAC,WAAW,EAAE,EAChC,WAAW,CACX,CAAC;YAEF,MAAM,UAAU,GAAG,aAAa,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAC5D,MAAM,aAAa,GAAG,aAAa,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;YAElE,MAAM,KAAK,GAAkB;gBAC5B,QAAQ;gBACR,WAAW;gBACX,aAAa,EAAE,OAAO,CAAC,MAAM;gBAC7B,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC/B,UAAU;gBACV,aAAa;aACb,CAAC;YAEF,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YACtC,IAAI,CAAC,yBAAyB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAChD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,KAAK,CAAC,mBAAmB,QAAQ,EAAE,EAAE,KAAK,CAAC,CAAC;QACrD,CAAC;gBAAS,CAAC;YACV,WAAW,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;IACF,CAAC;IAEO,yBAAyB,CAChC,KAAoB,EACpB,OAAe;QAEf,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC;QAEjC,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QACzD,MAAM,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAEzB,IAAI,KAAK,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;YAC7B,KAAK,CAAC,WAAW,CAAC,KAAK;iBACrB,WAAW,EAAE;iBACb,KAAK,CAAC,KAAK,CAAC;iBACZ,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;gBACd,CAAC,IAAI,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YACpB,CAAC,CAAC,CAAC;QACL,CAAC;QACD,IAAI,KAAK,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC;YAC5B,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;gBACtC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC;YAC/B,CAAC,CAAC,CAAC;QACJ,CAAC;QAED,OAAO;aACL,WAAW,EAAE;aACb,KAAK,CAAC,eAAe,CAAC;YACvB,EAAE,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;YACnB,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACnB,CAAC,CAAC,CAAC;QAEJ,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;QACzD,IAAI,QAAQ,EAAE,CAAC;YACd,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,WAAW,GAAG,cAAc,CAAC;QACnC,IAAI,KAAK,GAA2B,IAAI,CAAC;QACzC,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClC,OAAO,KAAK,KAAK,IAAI,EAAE,CAAC;YACvB,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gBACd,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;YAC3C,CAAC;YACD,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACnC,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC5B,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBACpC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;YAC1C,CAAC;YACD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACpD,CAAC;IACF,CAAC;IAEO,kBAAkB;QACzB,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;QAC3B,KAAK,MAAM,CAAC,UAAU,EAAE,SAAS,CAAC,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,EAAE,CAAC;YAClE,KAAK,MAAM,UAAU,IAAI,SAAS,CAAC,aAAa,EAAE,CAAC;gBAClD,MAAM,gBAAgB,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;gBACxD,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,CAAC;oBAC/C,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,gBAAgB,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;gBACrD,CAAC;gBACD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC;YAC3D,CAAC;QACF,CAAC;IACF,CAAC;IAEO,aAAa,CAAC,IAAY;QACjC,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;IAClD,CAAC;CACD"}
|
package/build/utils/LLMClient.js
CHANGED
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import state from "../config.js";
|
|
2
|
+
import { localEmbedder } from "../utils/Embedder.js";
|
|
2
3
|
export class LLMClient {
|
|
3
4
|
// 설정값을 실시간으로 가져오기 위한 Getter들
|
|
4
5
|
get apiUrl() {
|
|
@@ -14,6 +15,10 @@ export class LLMClient {
|
|
|
14
15
|
return state.llmChatModel;
|
|
15
16
|
}
|
|
16
17
|
async isEmbeddingServerHealthy() {
|
|
18
|
+
// 로컬 임베더 모델이 존재하는지 먼저 확인
|
|
19
|
+
if (await localEmbedder.checkModelPresence()) {
|
|
20
|
+
return true;
|
|
21
|
+
}
|
|
17
22
|
const url = `${this.embeddingApiUrl}/v1/models`;
|
|
18
23
|
try {
|
|
19
24
|
const response = await fetch(url, { signal: AbortSignal.timeout(2000) });
|
|
@@ -65,6 +70,17 @@ ${chunkContent}
|
|
|
65
70
|
}
|
|
66
71
|
}
|
|
67
72
|
async generateEmbedding(text) {
|
|
73
|
+
// 1. 로컬 임베더 사용 시도 (우선순위)
|
|
74
|
+
try {
|
|
75
|
+
const isLocalAvailable = await localEmbedder.checkModelPresence();
|
|
76
|
+
if (isLocalAvailable) {
|
|
77
|
+
return await localEmbedder.embed(text);
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
catch (error) {
|
|
81
|
+
console.warn("[LLMClient] Local embedder failed, falling back to server:", error);
|
|
82
|
+
}
|
|
83
|
+
// 2. 외부 API 서버 사용 (폴백)
|
|
68
84
|
const url = `${this.embeddingApiUrl}/v1/embeddings`;
|
|
69
85
|
try {
|
|
70
86
|
const response = await fetch(url, {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"LLMClient.js","sourceRoot":"","sources":["../../src/utils/LLMClient.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,
|
|
1
|
+
{"version":3,"file":"LLMClient.js","sourceRoot":"","sources":["../../src/utils/LLMClient.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,aAAa,CAAC;AAChC,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AA6BpD,MAAM,OAAO,SAAS;IACrB,6BAA6B;IAC7B,IAAY,MAAM;QACjB,OAAO,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAC3C,CAAC;IACD,IAAY,eAAe;QAC1B,OAAO,KAAK,CAAC,kBAAkB,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACpD,CAAC;IACD,IAAY,cAAc;QACzB,OAAO,KAAK,CAAC,iBAAiB,CAAC;IAChC,CAAC;IACD,IAAY,SAAS;QACpB,OAAO,KAAK,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,wBAAwB;QAC7B,yBAAyB;QACzB,IAAI,MAAM,aAAa,CAAC,kBAAkB,EAAE,EAAE,CAAC;YAC9C,OAAO,IAAI,CAAC;QACb,CAAC;QAED,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,eAAe,YAAY,CAAC;QAChD,IAAI,CAAC;YACJ,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACzE,OAAO,QAAQ,CAAC,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,MAAM,EAAE,CAAC;YACjB,OAAO,KAAK,CAAC;QACd,CAAC;IACF,CAAC;IAED,KAAK,CAAC,eAAe,CACpB,eAAuB,EACvB,YAAoB;QAEpB,MAAM,MAAM,GAAG;EACf,eAAe;;;;;;EAMf,YAAY;SACL,CAAC;QAER,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,sBAAsB,CAAC;QACjD,IAAI,CAAC;YACJ,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBACjC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACpB,KAAK,EAAE,IAAI,CAAC,SAAS;oBACrB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;oBAC7C,UAAU,EAAE,EAAE;oBACd,WAAW,EAAE,CAAC;oBACd,MAAM,EAAE,KAAK;iBACb,CAAC;gBACF,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,WAAW;aAC/C,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBAClB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,MAAM,IAAI,KAAK,CACd,kBAAkB,QAAQ,CAAC,MAAM,QAAQ,GAAG,KAAK,SAAS,IAAI,QAAQ,CAAC,UAAU,EAAE,CACnF,CAAC;YACH,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAoB,CAAC;YACxD,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QAC/C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC3D,OAAO,CAAC,KAAK,CAAC,sBAAsB,GAAG,EAAE,CAAC,CAAC;YAC5C,CAAC;iBAAM,CAAC;gBACP,OAAO,CAAC,KAAK,CAAC,wCAAwC,GAAG,GAAG,EAAE,KAAK,CAAC,CAAC;YACtE,CAAC;YACD,OAAO,EAAE,CAAC;QACX,CAAC;IACF,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,IAAY;QACnC,yBAAyB;QACzB,IAAI,CAAC;YACJ,MAAM,gBAAgB,GAAG,MAAM,aAAa,CAAC,kBAAkB,EAAE,CAAC;YAClE,IAAI,gBAAgB,EAAE,CAAC;gBACtB,OAAO,MAAM,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACxC,CAAC;QACF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,IAAI,CACX,4DAA4D,EAC5D,KAAK,CACL,CAAC;QACH,CAAC;QAED,uBAAuB;QACvB,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,eAAe,gBAAgB,CAAC;QACpD,IAAI,CAAC;YACJ,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBACjC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACpB,KAAK,EAAE,IAAI,CAAC,cAAc;oBAC1B,KAAK,EAAE,IAAI;iBACX,CAAC;gBACF,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,oBAAoB;aACvD,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBAClB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,MAAM,IAAI,KAAK,CACd,wBAAwB,QAAQ,CAAC,MAAM,QAAQ,GAAG,KAAK,SAAS,IAAI,QAAQ,CAAC,UAAU,EAAE,CACzF,CAAC;YACH,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAyB,CAAC;YAC7D,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;gBAC5D,MAAM,IAAI,KAAK,CAAC,6CAA6C,GAAG,EAAE,CAAC,CAAC;YACrE,CAAC;YACD,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,KAAK,CAAC,0CAA0C,GAAG,GAAG,EAAE,KAAK,CAAC,CAAC;YACvE,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;CACD;AAED,MAAM,CAAC,MAAM,SAAS,GAAG,IAAI,SAAS,EAAE,CAAC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
declare class Reranker {
|
|
2
|
+
private modelName;
|
|
3
|
+
private tokenizer;
|
|
4
|
+
private model;
|
|
5
|
+
private initPromise;
|
|
6
|
+
constructor();
|
|
7
|
+
checkModelPresence(): Promise<boolean>;
|
|
8
|
+
get isReady(): boolean;
|
|
9
|
+
private loadModel;
|
|
10
|
+
init(): Promise<void>;
|
|
11
|
+
rerank(query: string, documents: string[]): Promise<{
|
|
12
|
+
document: string;
|
|
13
|
+
score: number;
|
|
14
|
+
}[]>;
|
|
15
|
+
}
|
|
16
|
+
export declare const localReranker: Reranker;
|
|
17
|
+
export {};
|
|
@@ -0,0 +1,71 @@
|
|
|
1
|
+
import { AutoModelForSequenceClassification, AutoTokenizer, env, } from "@huggingface/transformers";
|
|
2
|
+
import z from "zod";
|
|
3
|
+
import { MODELS_DIR } from "../utils/constants.js";
|
|
4
|
+
const rerankedResultsSchema = z.array(z.object({
|
|
5
|
+
document: z.string(),
|
|
6
|
+
score: z.number(),
|
|
7
|
+
}));
|
|
8
|
+
class Reranker {
|
|
9
|
+
modelName = "Xenova/bge-reranker-base";
|
|
10
|
+
tokenizer = null;
|
|
11
|
+
model = null;
|
|
12
|
+
initPromise = null;
|
|
13
|
+
constructor() {
|
|
14
|
+
env.allowRemoteModels = false;
|
|
15
|
+
env.localModelPath = MODELS_DIR;
|
|
16
|
+
env.cacheDir = MODELS_DIR;
|
|
17
|
+
}
|
|
18
|
+
async checkModelPresence() {
|
|
19
|
+
try {
|
|
20
|
+
await AutoTokenizer.from_pretrained(this.modelName, {
|
|
21
|
+
local_files_only: true,
|
|
22
|
+
});
|
|
23
|
+
return true;
|
|
24
|
+
}
|
|
25
|
+
catch (_e) {
|
|
26
|
+
return false;
|
|
27
|
+
}
|
|
28
|
+
}
|
|
29
|
+
get isReady() {
|
|
30
|
+
return this.tokenizer !== null && this.model !== null;
|
|
31
|
+
}
|
|
32
|
+
async loadModel() {
|
|
33
|
+
if (!this.tokenizer) {
|
|
34
|
+
this.tokenizer = await AutoTokenizer.from_pretrained(this.modelName);
|
|
35
|
+
}
|
|
36
|
+
if (!this.model) {
|
|
37
|
+
this.model = await AutoModelForSequenceClassification.from_pretrained(this.modelName);
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
async init() {
|
|
41
|
+
if (!this.initPromise) {
|
|
42
|
+
this.initPromise = this.loadModel();
|
|
43
|
+
}
|
|
44
|
+
return this.initPromise;
|
|
45
|
+
}
|
|
46
|
+
async rerank(query, documents) {
|
|
47
|
+
if (!this.isReady) {
|
|
48
|
+
await this.init();
|
|
49
|
+
}
|
|
50
|
+
if (!this.tokenizer || !this.model) {
|
|
51
|
+
throw new Error("Model or tokenizer failed to load.");
|
|
52
|
+
}
|
|
53
|
+
const queries = Array(documents.length).fill(query);
|
|
54
|
+
const inputs = this.tokenizer(queries, {
|
|
55
|
+
text_pair: documents,
|
|
56
|
+
padding: true,
|
|
57
|
+
truncation: true,
|
|
58
|
+
});
|
|
59
|
+
const { logits } = await this.model(inputs);
|
|
60
|
+
const rawScores = Array.from(logits.data);
|
|
61
|
+
const scores = rawScores.map((score) => 1 / (1 + Math.exp(-score)));
|
|
62
|
+
const rankedDocuments = documents.map((doc, index) => ({
|
|
63
|
+
document: doc,
|
|
64
|
+
score: scores[index],
|
|
65
|
+
}));
|
|
66
|
+
const parsedResults = rerankedResultsSchema.parse(rankedDocuments);
|
|
67
|
+
return parsedResults.sort((a, b) => b.score - a.score);
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
export const localReranker = new Reranker();
|
|
71
|
+
//# sourceMappingURL=LocalReranker.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"LocalReranker.js","sourceRoot":"","sources":["../../src/utils/LocalReranker.ts"],"names":[],"mappings":"AAIA,OAAO,EACN,kCAAkC,EAClC,aAAa,EACb,GAAG,GACH,MAAM,2BAA2B,CAAC;AACnC,OAAO,CAAC,MAAM,KAAK,CAAC;AACpB,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAElD,MAAM,qBAAqB,GAAG,CAAC,CAAC,KAAK,CACpC,CAAC,CAAC,MAAM,CAAC;IACR,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;CACjB,CAAC,CACF,CAAC;AAEF,MAAM,QAAQ;IACL,SAAS,GAAG,0BAA0B,CAAC;IACvC,SAAS,GAA+B,IAAI,CAAC;IAC7C,KAAK,GAA2B,IAAI,CAAC;IACrC,WAAW,GAAyB,IAAI,CAAC;IAEjD;QACC,GAAG,CAAC,iBAAiB,GAAG,KAAK,CAAC;QAC9B,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;QAChC,GAAG,CAAC,QAAQ,GAAG,UAAU,CAAC;IAC3B,CAAC;IAEM,KAAK,CAAC,kBAAkB;QAC9B,IAAI,CAAC;YACJ,MAAM,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,EAAE;gBACnD,gBAAgB,EAAE,IAAI;aACtB,CAAC,CAAC;YACH,OAAO,IAAI,CAAC;QACb,CAAC;QAAC,OAAO,EAAE,EAAE,CAAC;YACb,OAAO,KAAK,CAAC;QACd,CAAC;IACF,CAAC;IAED,IAAI,OAAO;QACV,OAAO,IAAI,CAAC,SAAS,KAAK,IAAI,IAAI,IAAI,CAAC,KAAK,KAAK,IAAI,CAAC;IACvD,CAAC;IAEO,KAAK,CAAC,SAAS;QACtB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACrB,IAAI,CAAC,SAAS,GAAG,MAAM,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YACjB,IAAI,CAAC,KAAK,GAAG,MAAM,kCAAkC,CAAC,eAAe,CACpE,IAAI,CAAC,SAAS,CACd,CAAC;QACH,CAAC;IACF,CAAC;IAEM,KAAK,CAAC,IAAI;QAChB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACvB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QACrC,CAAC;QACD,OAAO,IAAI,CAAC,WAAW,CAAC;IACzB,CAAC;IAEM,KAAK,CAAC,MAAM,CAClB,KAAa,EACb,SAAmB;QAEnB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACnB,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACvD,CAAC;QAED,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpD,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE;YACtC,SAAS,EAAE,SAAS;YACpB,OAAO,EAAE,IAAI;YACb,UAAU,EAAE,IAAI;SAChB,CAAC,CAAC;QAEH,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAC5C,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAa,CAAC;QAEtD,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACpE,MAAM,eAAe,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;YACtD,QAAQ,EAAE,GAAG;YACb,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;SACpB,CAAC,CAAC,CAAC;QAEJ,MAAM,aAAa,GAAG,qBAAqB,CAAC,KAAK,CAAC,eAAe,CAG9D,CAAC;QAEJ,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;IACxD,CAAC;CACD;AAED,MAAM,CAAC,MAAM,aAAa,GAAG,IAAI,QAAQ,EAAE,CAAC"}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { type VectorRecord } from "
|
|
1
|
+
import { type VectorRecord } from "../utils/VectorDB.js";
|
|
2
2
|
export declare class RAGIndexer {
|
|
3
3
|
private splitter;
|
|
4
4
|
private embeddingSemaphore;
|
|
@@ -6,10 +6,19 @@ export declare class RAGIndexer {
|
|
|
6
6
|
private spinner;
|
|
7
7
|
private totalFiles;
|
|
8
8
|
private processedFiles;
|
|
9
|
+
private _isIndexing;
|
|
9
10
|
private enc;
|
|
10
11
|
constructor();
|
|
11
12
|
setSpinner(total: number): void;
|
|
12
13
|
private updateProgress;
|
|
14
|
+
getStatus(): {
|
|
15
|
+
isIndexing: boolean;
|
|
16
|
+
processed: number;
|
|
17
|
+
total: number;
|
|
18
|
+
progress: number;
|
|
19
|
+
};
|
|
20
|
+
startIndexing(total: number): void;
|
|
21
|
+
stopIndexing(): void;
|
|
13
22
|
private buildFileRecords;
|
|
14
23
|
processFile(filePath: string, content?: string): Promise<void>;
|
|
15
24
|
processFileInMemory(filePath: string, content?: string): Promise<{
|
|
@@ -3,11 +3,12 @@ import path from "node:path";
|
|
|
3
3
|
import { RecursiveCharacterTextSplitter } from "@langchain/textsplitters";
|
|
4
4
|
import { encodingForModel } from "js-tiktoken";
|
|
5
5
|
import ora, {} from "ora";
|
|
6
|
-
import { DirectoryWalker } from "
|
|
7
|
-
import {
|
|
8
|
-
import {
|
|
9
|
-
import {
|
|
10
|
-
import {
|
|
6
|
+
import { DirectoryWalker } from "../utils/DirectoryWalker.js";
|
|
7
|
+
import { localEmbedder } from "../utils/Embedder.js";
|
|
8
|
+
import { llmClient } from "../utils/LLMClient.js";
|
|
9
|
+
import { parse as parseMatter } from "../utils/processor/MatterParser.js";
|
|
10
|
+
import { Semaphore } from "../utils/semaphore.js";
|
|
11
|
+
import { vectorDB } from "../utils/VectorDB.js";
|
|
11
12
|
function extractHeadingsWithPositions(body) {
|
|
12
13
|
const headingRegex = /^(#{1,3})\s+(.+)$/gm;
|
|
13
14
|
const results = [];
|
|
@@ -41,6 +42,7 @@ export class RAGIndexer {
|
|
|
41
42
|
spinner = null;
|
|
42
43
|
totalFiles = 0;
|
|
43
44
|
processedFiles = 0;
|
|
45
|
+
_isIndexing = false;
|
|
44
46
|
enc = encodingForModel("gpt-3.5-turbo");
|
|
45
47
|
constructor() {
|
|
46
48
|
this.splitter = new RecursiveCharacterTextSplitter({
|
|
@@ -71,6 +73,24 @@ export class RAGIndexer {
|
|
|
71
73
|
this.spinner = null;
|
|
72
74
|
}
|
|
73
75
|
}
|
|
76
|
+
getStatus() {
|
|
77
|
+
return {
|
|
78
|
+
isIndexing: this._isIndexing,
|
|
79
|
+
processed: this.processedFiles,
|
|
80
|
+
total: this.totalFiles,
|
|
81
|
+
progress: this.totalFiles > 0
|
|
82
|
+
? Math.round((this.processedFiles / this.totalFiles) * 100)
|
|
83
|
+
: 0,
|
|
84
|
+
};
|
|
85
|
+
}
|
|
86
|
+
startIndexing(total) {
|
|
87
|
+
this._isIndexing = true;
|
|
88
|
+
this.totalFiles = total;
|
|
89
|
+
this.processedFiles = 0;
|
|
90
|
+
}
|
|
91
|
+
stopIndexing() {
|
|
92
|
+
this._isIndexing = false;
|
|
93
|
+
}
|
|
74
94
|
async buildFileRecords(filePath, content) {
|
|
75
95
|
const fileStat = await fs.stat(filePath);
|
|
76
96
|
const fileContent = content ?? (await fs.readFile(filePath, "utf-8"));
|
|
@@ -116,7 +136,13 @@ export class RAGIndexer {
|
|
|
116
136
|
await this.embeddingSemaphore.acquire();
|
|
117
137
|
let vector;
|
|
118
138
|
try {
|
|
119
|
-
|
|
139
|
+
const isLocalReady = await localEmbedder.checkModelPresence();
|
|
140
|
+
if (isLocalReady) {
|
|
141
|
+
vector = await localEmbedder.embed(`search_document: ${safeText}`);
|
|
142
|
+
}
|
|
143
|
+
else {
|
|
144
|
+
vector = await llmClient.generateEmbedding(`search_document: ${safeText}`);
|
|
145
|
+
}
|
|
120
146
|
}
|
|
121
147
|
finally {
|
|
122
148
|
this.embeddingSemaphore.release();
|
|
@@ -173,29 +199,35 @@ export class RAGIndexer {
|
|
|
173
199
|
}
|
|
174
200
|
}
|
|
175
201
|
async indexAll(vaultPath) {
|
|
176
|
-
|
|
177
|
-
|
|
178
|
-
|
|
179
|
-
|
|
180
|
-
|
|
181
|
-
|
|
182
|
-
|
|
183
|
-
|
|
184
|
-
|
|
185
|
-
|
|
186
|
-
|
|
187
|
-
|
|
188
|
-
|
|
189
|
-
|
|
202
|
+
this._isIndexing = true;
|
|
203
|
+
try {
|
|
204
|
+
const walker = new DirectoryWalker();
|
|
205
|
+
const filePaths = await walker.walk(vaultPath, this.ioSemaphore);
|
|
206
|
+
const markdownFiles = filePaths.filter((f) => f.endsWith(".md") || f.endsWith(".mdx"));
|
|
207
|
+
this.setSpinner(markdownFiles.length);
|
|
208
|
+
const allRecords = [];
|
|
209
|
+
const allMeta = [];
|
|
210
|
+
const fileSemaphore = new Semaphore(8);
|
|
211
|
+
await Promise.all(markdownFiles.map(async (filePath) => {
|
|
212
|
+
await fileSemaphore.acquire();
|
|
213
|
+
try {
|
|
214
|
+
const result = await this.processFileInMemory(filePath);
|
|
215
|
+
if (result && result.records.length > 0) {
|
|
216
|
+
allRecords.push(...result.records);
|
|
217
|
+
allMeta.push({ filePath, mtime: result.mtime });
|
|
218
|
+
}
|
|
190
219
|
}
|
|
220
|
+
finally {
|
|
221
|
+
fileSemaphore.release();
|
|
222
|
+
}
|
|
223
|
+
}));
|
|
224
|
+
if (allRecords.length > 0) {
|
|
225
|
+
await vectorDB.upsertChunks(allRecords);
|
|
226
|
+
await vectorDB.updateFileMetaBatch(allMeta);
|
|
191
227
|
}
|
|
192
|
-
|
|
193
|
-
|
|
194
|
-
|
|
195
|
-
}));
|
|
196
|
-
if (allRecords.length > 0) {
|
|
197
|
-
await vectorDB.upsertChunks(allRecords);
|
|
198
|
-
await vectorDB.updateFileMetaBatch(allMeta);
|
|
228
|
+
}
|
|
229
|
+
finally {
|
|
230
|
+
this._isIndexing = false;
|
|
199
231
|
}
|
|
200
232
|
}
|
|
201
233
|
async deleteFile(filePath) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"RAGIndexer.js","sourceRoot":"","sources":["../../src/utils/RAGIndexer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,8BAA8B,EAAE,MAAM,0BAA0B,CAAC;AAC1E,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,GAAG,EAAE,EAAY,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,MAAM,
|
|
1
|
+
{"version":3,"file":"RAGIndexer.js","sourceRoot":"","sources":["../../src/utils/RAGIndexer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,8BAA8B,EAAE,MAAM,0BAA0B,CAAC;AAC1E,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,GAAG,EAAE,EAAY,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,KAAK,IAAI,WAAW,EAAE,MAAM,mCAAmC,CAAC;AACzE,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAqB,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAIlE,SAAS,4BAA4B,CAAC,IAAY;IACjD,MAAM,YAAY,GAAG,qBAAqB,CAAC;IAC3C,MAAM,OAAO,GAAmB,EAAE,CAAC;IACnC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IAC5C,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC7B,OAAO,CAAC,IAAI,CAAC;YACZ,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE;YACxB,GAAG,EAAE,KAAK,CAAC,KAAK;YAChB,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM;SACtB,CAAC,CAAC;IACJ,CAAC;IACD,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,SAAS,mBAAmB,CAC3B,IAAY,EACZ,KAAa,EACb,QAAwB;IAExB,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IACvC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAC7C,IAAI,GAAG,KAAK,CAAC,CAAC;QAAE,OAAO,IAAI,CAAC;IAC5B,IAAI,OAAO,GAAkB,IAAI,CAAC;IAClC,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;QAC1B,IAAI,CAAC,CAAC,GAAG,IAAI,GAAG;YAAE,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;IACvC,CAAC;IACD,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,MAAM,OAAO,UAAU;IACd,QAAQ,CAAiC;IACzC,kBAAkB,CAAY;IAC9B,WAAW,CAAY;IACvB,OAAO,GAAe,IAAI,CAAC;IAC3B,UAAU,GAAG,CAAC,CAAC;IACf,cAAc,GAAG,CAAC,CAAC;IACnB,WAAW,GAAG,KAAK,CAAC;IACpB,GAAG,GAAG,gBAAgB,CAAC,eAAe,CAAC,CAAC;IAEhD;QACC,IAAI,CAAC,QAAQ,GAAG,IAAI,8BAA8B,CAAC;YAClD,SAAS,EAAE,GAAG;YACd,YAAY,EAAE,EAAE;YAChB,cAAc,EAAE,CAAC,IAAY,EAAE,EAAE;gBAChC,OAAO,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC;YACrC,CAAC;SACD,CAAC,CAAC;QAEH,IAAI,CAAC,kBAAkB,GAAG,IAAI,SAAS,CAAC,CAAC,CAAC,CAAC;QAC3C,IAAI,CAAC,WAAW,GAAG,IAAI,SAAS,CAAC,EAAE,CAAC,CAAC;IACtC,CAAC;IAEM,UAAU,CAAC,KAAa;QAC9B,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;QACxB,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC;YAClB,IAAI,EAAE,gCAAgC,KAAK,MAAM;YACjD,MAAM,EAAE,OAAO,CAAC,MAAM;SACtB,CAAC,CAAC,KAAK,EAAE,CAAC;IACZ,CAAC;IAEO,cAAc,CAAC,QAAgB;QACtC,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,IAAI,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,UAAU,eAAe,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QACxG,CAAC;QACD,IAAI,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5D,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,wBAAwB,IAAI,CAAC,UAAU,SAAS,CAAC,CAAC;YACvE,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACrB,CAAC;IACF,CAAC;IAEM,SAAS;QACf,OAAO;YACN,UAAU,EAAE,IAAI,CAAC,WAAW;YAC5B,SAAS,EAAE,IAAI,CAAC,cAAc;YAC9B,KAAK,EAAE,IAAI,CAAC,UAAU;YACtB,QAAQ,EACP,IAAI,CAAC,UAAU,GAAG,CAAC;gBAClB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,GAAG,CAAC;gBAC3D,CAAC,CAAC,CAAC;SACL,CAAC;IACH,CAAC;IAEM,aAAa,CAAC,KAAa;QACjC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;IACzB,CAAC;IAEM,YAAY;QAClB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC7B,QAAgB,EAChB,OAAgB;QAEhB,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,WAAW,GAAG,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;QACtE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,WAAW,CACjD,QAAQ,EACR,QAAQ,CAAC,SAAS,CAAC,WAAW,EAAE,EAChC,WAAW,CACX,CAAC;QACF,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAEzC,MAAM,QAAQ,GAAG,4BAA4B,CAAC,IAAI,CAAC,CAAC;QACpD,MAAM,aAAa,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,EAAE,OAAO,IAAI,IAAI,CAAC;QAC3E,MAAM,QAAQ,GAAG,WAAW,CAAC,KAAK,IAAI,aAAa,IAAI,QAAQ,CAAC;QAChE,MAAM,YAAY,GAAG,QAAQ;aAC3B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;aAC3B,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;aACX,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QAErC,MAAM,cAAc,GAAG;YACtB,UAAU,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE;YACjC,WAAW,CAAC,IAAI,EAAE,MAAM;gBACvB,CAAC,CAAC,SAAS,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;gBACpD,CAAC,CAAC,EAAE;YACL,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE;SACzE;aACC,MAAM,CAAC,OAAO,CAAC;aACf,IAAI,CAAC,IAAI,CAAC,CAAC;QAEb,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACnD,MAAM,OAAO,GAAmB,EAAE,CAAC;QAEnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YAExB,MAAM,cAAc,GAAG,mBAAmB,CAAC,IAAI,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;YAClE,MAAM,OAAO,GAAG;gBACf,cAAc,CAAC,CAAC,CAAC,YAAY,cAAc,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE;gBAC/D,YAAY,CAAC,MAAM,GAAG,CAAC;oBACtB,CAAC,CAAC,YAAY,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE;oBACpD,CAAC,CAAC,EAAE;aACL;iBACC,MAAM,CAAC,OAAO,CAAC;iBACf,IAAI,CAAC,IAAI,CAAC,CAAC;YAEb,MAAM,QAAQ,GAAG,CAAC,cAAc,EAAE,OAAO,EAAE,KAAK,CAAC;iBAC/C,MAAM,CAAC,OAAO,CAAC;iBACf,IAAI,CAAC,MAAM,CAAC,CAAC;YAEf,MAAM,gBAAgB,GAAG,GAAG,CAAC;YAC7B,MAAM,kBAAkB,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC;YAC5D,MAAM,QAAQ,GACb,kBAAkB,GAAG,gBAAgB;gBACpC,CAAC,CAAC,QAAQ,CAAC,KAAK,CACd,CAAC,EACD,IAAI,CAAC,KAAK,CACT,QAAQ,CAAC,MAAM,GAAG,CAAC,gBAAgB,GAAG,kBAAkB,CAAC,CACzD,CACD;gBACF,CAAC,CAAC,QAAQ,CAAC;YAEb,MAAM,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,CAAC;YACxC,IAAI,MAAgB,CAAC;YACrB,IAAI,CAAC;gBACJ,MAAM,YAAY,GAAG,MAAM,aAAa,CAAC,kBAAkB,EAAE,CAAC;gBAC9D,IAAI,YAAY,EAAE,CAAC;oBAClB,MAAM,GAAG,MAAM,aAAa,CAAC,KAAK,CAAC,oBAAoB,QAAQ,EAAE,CAAC,CAAC;gBACpE,CAAC;qBAAM,CAAC;oBACP,MAAM,GAAG,MAAM,SAAS,CAAC,iBAAiB,CACzC,oBAAoB,QAAQ,EAAE,CAC9B,CAAC;gBACH,CAAC;YACF,CAAC;oBAAS,CAAC;gBACV,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,CAAC;YACnC,CAAC;YAED,OAAO,CAAC,IAAI,CAAC;gBACZ,EAAE,EAAE,GAAG,QAAQ,UAAU,CAAC,EAAE;gBAC5B,QAAQ;gBACR,QAAQ;gBACR,UAAU,EAAE,CAAC;gBACb,OAAO,EAAE,KAAK;gBACd,OAAO;gBACP,MAAM;gBACN,QAAQ,EAAE;oBACT,KAAK,EAAE,QAAQ;oBACf,IAAI,EACH,WAAW,CAAC,IAAI,YAAY,IAAI;wBAC/B,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,EAAE;wBAChC,CAAC,CAAC,WAAW,CAAC,IAAI,IAAI,QAAQ,CAAC,SAAS,CAAC,WAAW,EAAE;oBACxD,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;oBACxC,OAAO,EAAE,WAAW,CAAC,OAAO,IAAI,EAAE;oBAClC,IAAI,EAAE,WAAW,CAAC,IAAI,IAAI,EAAE;oBAC5B,QAAQ,EAAE,WAAW,CAAC,QAAQ,IAAI,KAAK;oBACvC,SAAS,EAAE,WAAW,CAAC,SAAS,IAAI,KAAK;iBACzC;aACD,CAAC,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,QAAgB,EAAE,OAAgB;QACnD,IAAI,CAAC;YACJ,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC1E,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,MAAM,QAAQ,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACtC,CAAC;YACD,0CAA0C;YAC1C,MAAM,QAAQ,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAChD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,KAAK,CAAC,oCAAoC,QAAQ,EAAE,EAAE,KAAK,CAAC,CAAC;QACtE,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;IACF,CAAC;IAED,KAAK,CAAC,mBAAmB,CACxB,QAAgB,EAChB,OAAgB;QAEhB,IAAI,CAAC;YACJ,OAAO,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACvD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,KAAK,CAAC,oCAAoC,QAAQ,EAAE,EAAE,KAAK,CAAC,CAAC;YACrE,OAAO,IAAI,CAAC;QACb,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;IACF,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,SAAiB;QAC/B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC;YACJ,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;YACrC,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;YACjE,MAAM,aAAa,GAAG,SAAS,CAAC,MAAM,CACrC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAC9C,CAAC;YAEF,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;YAEtC,MAAM,UAAU,GAAmB,EAAE,CAAC;YACtC,MAAM,OAAO,GAA0C,EAAE,CAAC;YAC1D,MAAM,aAAa,GAAG,IAAI,SAAS,CAAC,CAAC,CAAC,CAAC;YAEvC,MAAM,OAAO,CAAC,GAAG,CAChB,aAAa,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE;gBACpC,MAAM,aAAa,CAAC,OAAO,EAAE,CAAC;gBAC9B,IAAI,CAAC;oBACJ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;oBACxD,IAAI,MAAM,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACzC,UAAU,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;wBACnC,OAAO,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;oBACjD,CAAC;gBACF,CAAC;wBAAS,CAAC;oBACV,aAAa,CAAC,OAAO,EAAE,CAAC;gBACzB,CAAC;YACF,CAAC,CAAC,CACF,CAAC;YAEF,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,MAAM,QAAQ,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;gBACxC,MAAM,QAAQ,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;YAC7C,CAAC;QACF,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QAC1B,CAAC;IACF,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,QAAgB;QAChC,MAAM,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC3C,CAAC;CACD;AAED,MAAM,CAAC,MAAM,UAAU,GAAG,IAAI,UAAU,EAAE,CAAC"}
|
|
@@ -24,26 +24,26 @@ export declare const RerankResponseSchema: z.ZodObject<{
|
|
|
24
24
|
}>, "many">;
|
|
25
25
|
}, "strip", z.ZodTypeAny, {
|
|
26
26
|
object: string;
|
|
27
|
+
results: {
|
|
28
|
+
index: number;
|
|
29
|
+
relevance_score: number;
|
|
30
|
+
}[];
|
|
27
31
|
usage: {
|
|
28
32
|
prompt_tokens: number;
|
|
29
33
|
total_tokens: number;
|
|
30
34
|
};
|
|
31
35
|
model: string;
|
|
36
|
+
}, {
|
|
37
|
+
object: string;
|
|
32
38
|
results: {
|
|
33
39
|
index: number;
|
|
34
40
|
relevance_score: number;
|
|
35
41
|
}[];
|
|
36
|
-
}, {
|
|
37
|
-
object: string;
|
|
38
42
|
usage: {
|
|
39
43
|
prompt_tokens: number;
|
|
40
44
|
total_tokens: number;
|
|
41
45
|
};
|
|
42
46
|
model: string;
|
|
43
|
-
results: {
|
|
44
|
-
index: number;
|
|
45
|
-
relevance_score: number;
|
|
46
|
-
}[];
|
|
47
47
|
}>;
|
|
48
48
|
export type RerankResult = {
|
|
49
49
|
index: number;
|
|
@@ -1,6 +1,7 @@
|
|
|
1
|
+
import { debugLogger } from "@sunub/core";
|
|
1
2
|
import { z } from "zod";
|
|
2
|
-
import { debugLogger } from "../cli/utils/debugLogger.js";
|
|
3
3
|
import state from "../config.js";
|
|
4
|
+
import { localReranker } from "../utils/LocalReranker.js";
|
|
4
5
|
export const RerankResponseSchema = z.object({
|
|
5
6
|
model: z.string(),
|
|
6
7
|
object: z.string(),
|
|
@@ -25,6 +26,24 @@ class RerankerClient {
|
|
|
25
26
|
debugLogger.warn("No documents to rerank, returning empty results.");
|
|
26
27
|
return [];
|
|
27
28
|
}
|
|
29
|
+
// 1. 로컬 리랭커 사용 시도 (우선순위)
|
|
30
|
+
try {
|
|
31
|
+
const isLocalAvailable = await localReranker.checkModelPresence();
|
|
32
|
+
if (isLocalAvailable) {
|
|
33
|
+
const results = await localReranker.rerank(query, documents);
|
|
34
|
+
return results
|
|
35
|
+
.map((r) => ({
|
|
36
|
+
index: documents.indexOf(r.document),
|
|
37
|
+
relevance_score: r.score,
|
|
38
|
+
document: r.document,
|
|
39
|
+
}))
|
|
40
|
+
.slice(0, topN);
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
catch (error) {
|
|
44
|
+
console.warn("[RerankerClient] Local reranker failed, falling back to server:", error);
|
|
45
|
+
}
|
|
46
|
+
// 2. 외부 API 서버 사용 (폴백)
|
|
28
47
|
try {
|
|
29
48
|
const response = await fetch(`${this.endpoint}/v1/rerank`, {
|
|
30
49
|
method: "POST",
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"RerankerClient.js","sourceRoot":"","sources":["../../src/utils/RerankerClient.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"RerankerClient.js","sourceRoot":"","sources":["../../src/utils/RerankerClient.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,MAAM,aAAa,CAAC;AAChC,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAEzD,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5C,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;IACjB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;IAClB,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC;QACf,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;QACzB,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;KACxB,CAAC;IACF,OAAO,EAAE,CAAC,CAAC,KAAK,CACf,CAAC,CAAC,MAAM,CAAC;QACR,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;QACjB,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE;KAC3B,CAAC,CACF;CACD,CAAC,CAAC;AAQH,MAAM,cAAc;IACX,QAAQ,CAAS;IACjB,KAAK,CAAS;IAEtB;QACC,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,iBAAiB,CAAC;QACxC,IAAI,CAAC,KAAK,GAAG,yBAAyB,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,MAAM,CACX,KAAa,EACb,SAAmB,EACnB,OAAe,CAAC;QAEhB,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,WAAW,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;YACrE,OAAO,EAAE,CAAC;QACX,CAAC;QAED,yBAAyB;QACzB,IAAI,CAAC;YACJ,MAAM,gBAAgB,GAAG,MAAM,aAAa,CAAC,kBAAkB,EAAE,CAAC;YAClE,IAAI,gBAAgB,EAAE,CAAC;gBACtB,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;gBAC7D,OAAO,OAAO;qBACZ,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBACZ,KAAK,EAAE,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC;oBACpC,eAAe,EAAE,CAAC,CAAC,KAAK;oBACxB,QAAQ,EAAE,CAAC,CAAC,QAAQ;iBACpB,CAAC,CAAC;qBACF,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;YAClB,CAAC;QACF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,IAAI,CACX,iEAAiE,EACjE,KAAK,CACL,CAAC;QACH,CAAC;QAED,uBAAuB;QACvB,IAAI,CAAC;YACJ,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,QAAQ,YAAY,EAAE;gBAC1D,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACR,cAAc,EAAE,kBAAkB;iBAClC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACpB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,KAAK;oBACL,SAAS;oBACT,KAAK,EAAE,IAAI;iBACX,CAAC;gBACF,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,WAAW;aAC/C,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBAClB,MAAM,IAAI,KAAK,CACd,uBAAuB,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAC/D,CAAC;YACH,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,MAAM,MAAM,GAAG,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAChD,MAAM,gBAAgB,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACxD,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,eAAe,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC;gBACrD,QAAQ,EAAE,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC;aACjC,CAAC,CAAC,CAAC;YAEJ,OAAO,gBAAgB;iBACrB,MAAM,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,eAAe,GAAG,GAAG,CAAC;iBACtD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,eAAe,GAAG,CAAC,CAAC,eAAe,CAAC,CAAC;QACzD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,WAAW,CAAC,KAAK,CAChB,mDAAmD,EACnD,KAAK,CACL,CAAC;YACF,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;gBACrC,KAAK;gBACL,eAAe,EAAE,CAAC;gBAClB,QAAQ,EAAE,GAAG;aACb,CAAC,CAAC,CAAC;QACL,CAAC;IACF,CAAC;IAEO,OAAO,CAAC,KAAa;QAC5B,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IACnC,CAAC;CACD;AAED,MAAM,CAAC,MAAM,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC"}
|
|
@@ -1,15 +1,39 @@
|
|
|
1
|
-
import type { DocumentIndex } from "
|
|
2
|
-
import type { EnrichedDocument } from "
|
|
1
|
+
import type { DocumentIndex } from "../../utils/processor/types.js";
|
|
2
|
+
import type { EnrichedDocument } from "../../utils/VaultManger/types.js";
|
|
3
|
+
import { type ChunkMetadata } from "../../utils/VectorDB.js";
|
|
4
|
+
export interface HybridSearchResult {
|
|
5
|
+
score: number;
|
|
6
|
+
document: DocumentIndex;
|
|
7
|
+
matchedChunks: ChunkMetadata[];
|
|
8
|
+
finalScore: number;
|
|
9
|
+
}
|
|
3
10
|
export declare class VaultManager {
|
|
4
11
|
private vaultPath;
|
|
5
12
|
private isInitialized;
|
|
13
|
+
private initPromise;
|
|
14
|
+
private isLocalAIReady;
|
|
6
15
|
private walker;
|
|
7
16
|
private indexer;
|
|
8
17
|
private ioSemaphore;
|
|
18
|
+
private hasNotifiedMissingModels;
|
|
9
19
|
constructor(vaultPath: string, maxConcurrentIO?: number);
|
|
10
20
|
initialize(): Promise<void>;
|
|
21
|
+
private checkAIAvailability;
|
|
22
|
+
/**
|
|
23
|
+
* 키워드 검색과 벡터 검색을 병합하고 Reranker로 최적화하는 하이브리드 검색
|
|
24
|
+
*/
|
|
25
|
+
hybridSearch(query: string, limit?: number): Promise<{
|
|
26
|
+
results: HybridSearchResult[];
|
|
27
|
+
diagnostic_message?: string;
|
|
28
|
+
}>;
|
|
29
|
+
private safeSemanticSearch;
|
|
30
|
+
syncMissingRagIndices(): Promise<void>;
|
|
31
|
+
private executeSync;
|
|
32
|
+
upsertDocument(filePath: string): Promise<void>;
|
|
33
|
+
removeDocument(filePath: string): Promise<void>;
|
|
11
34
|
getAllDocuments(): Promise<DocumentIndex[]>;
|
|
12
35
|
searchDocuments(keyword: string): Promise<DocumentIndex[]>;
|
|
36
|
+
private waitUntilIndexed;
|
|
13
37
|
getDocumentInfo(filename: string, options?: {
|
|
14
38
|
includeStats?: boolean;
|
|
15
39
|
includeBacklinks?: boolean;
|
|
@@ -23,6 +47,18 @@ export declare class VaultManager {
|
|
|
23
47
|
totalFiles: number;
|
|
24
48
|
isInitialized: boolean;
|
|
25
49
|
vaultPath: string;
|
|
50
|
+
ragStatus: {
|
|
51
|
+
isIndexing: boolean;
|
|
52
|
+
processed: number;
|
|
53
|
+
total: number;
|
|
54
|
+
progress: number;
|
|
55
|
+
};
|
|
56
|
+
};
|
|
57
|
+
getRagIndexingStatus(): {
|
|
58
|
+
isIndexing: boolean;
|
|
59
|
+
processed: number;
|
|
60
|
+
total: number;
|
|
61
|
+
progress: number;
|
|
26
62
|
};
|
|
27
63
|
private parseFilenameToFullPath;
|
|
28
64
|
private resolvePathForWrite;
|