opencode-rag-plugin 1.2.10 → 1.3.1
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/ReadMe.md +30 -57
- package/dist/cli.js +27 -3
- package/dist/cli.js.map +1 -1
- package/dist/core/config.d.ts +13 -1
- package/dist/core/config.js +20 -1
- package/dist/core/config.js.map +1 -1
- package/dist/core/interfaces.d.ts +8 -0
- package/dist/indexer.d.ts +2 -1
- package/dist/indexer.js +7 -0
- package/dist/indexer.js.map +1 -1
- package/dist/opencode/create-read-tool.d.ts +3 -1
- package/dist/opencode/create-read-tool.js +3 -3
- package/dist/opencode/create-read-tool.js.map +1 -1
- package/dist/plugin.d.ts +2 -1
- package/dist/plugin.js +104 -9
- package/dist/plugin.js.map +1 -1
- package/dist/retriever/keyword-index.d.ts +16 -0
- package/dist/retriever/keyword-index.js +172 -0
- package/dist/retriever/keyword-index.js.map +1 -0
- package/dist/retriever/retriever.d.ts +4 -1
- package/dist/retriever/retriever.js +42 -2
- package/dist/retriever/retriever.js.map +1 -1
- package/dist/watcher.d.ts +2 -1
- package/dist/watcher.js +3 -1
- package/dist/watcher.js.map +1 -1
- package/package.json +1 -1
|
@@ -1,14 +1,54 @@
|
|
|
1
1
|
export async function retrieve(query, embedder, store, options = {}) {
|
|
2
2
|
const topK = options.topK ?? 10;
|
|
3
|
+
const minScore = options.minScore ?? 0;
|
|
3
4
|
const embeddings = await embedder.embed([query]);
|
|
4
5
|
const embedding = embeddings[0];
|
|
5
6
|
if (!embedding || embedding.length === 0) {
|
|
6
7
|
return [];
|
|
7
8
|
}
|
|
8
|
-
// Guard against malformed provider output before searching.
|
|
9
9
|
if (typeof embedding[0] !== "number") {
|
|
10
10
|
return [];
|
|
11
11
|
}
|
|
12
|
-
|
|
12
|
+
const vectorFactor = 3;
|
|
13
|
+
const vectorResults = await store.search(embedding, topK * vectorFactor);
|
|
14
|
+
let keywordResults = [];
|
|
15
|
+
if (options.keywordIndex) {
|
|
16
|
+
keywordResults = options.keywordIndex.search(query, topK * vectorFactor);
|
|
17
|
+
}
|
|
18
|
+
if (keywordResults.length === 0) {
|
|
19
|
+
return vectorResults.filter((r) => r.score >= minScore);
|
|
20
|
+
}
|
|
21
|
+
const kwTopScore = keywordResults.length > 0 ? keywordResults[0].score : 1;
|
|
22
|
+
const combined = new Map();
|
|
23
|
+
for (const r of vectorResults) {
|
|
24
|
+
combined.set(r.chunk.id, {
|
|
25
|
+
chunk: r.chunk,
|
|
26
|
+
vScore: r.score,
|
|
27
|
+
kScore: 0,
|
|
28
|
+
});
|
|
29
|
+
}
|
|
30
|
+
for (const r of keywordResults) {
|
|
31
|
+
const existing = combined.get(r.chunk.id);
|
|
32
|
+
if (existing) {
|
|
33
|
+
existing.kScore = kwTopScore > 0 ? r.score / kwTopScore : 0;
|
|
34
|
+
}
|
|
35
|
+
else {
|
|
36
|
+
combined.set(r.chunk.id, {
|
|
37
|
+
chunk: r.chunk,
|
|
38
|
+
vScore: 0,
|
|
39
|
+
kScore: kwTopScore > 0 ? r.score / kwTopScore : 0,
|
|
40
|
+
});
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
const kw = options.keywordWeight ?? 0.4;
|
|
44
|
+
const combinedResults = [...combined.values()]
|
|
45
|
+
.map((entry) => ({
|
|
46
|
+
chunk: entry.chunk,
|
|
47
|
+
score: (1 - kw) * entry.vScore + kw * entry.kScore,
|
|
48
|
+
}))
|
|
49
|
+
.filter((r) => r.score >= minScore)
|
|
50
|
+
.sort((a, b) => b.score - a.score)
|
|
51
|
+
.slice(0, topK);
|
|
52
|
+
return combinedResults;
|
|
13
53
|
}
|
|
14
54
|
//# sourceMappingURL=retriever.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"retriever.js","sourceRoot":"","sources":["../../src/retriever/retriever.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"retriever.js","sourceRoot":"","sources":["../../src/retriever/retriever.ts"],"names":[],"mappings":"AASA,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,KAAa,EACb,QAA2B,EAC3B,KAAkB,EAClB,UAA2B,EAAE;IAE7B,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC;IAChC,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,CAAC,CAAC;IAEvC,MAAM,UAAU,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IACjD,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;IAChC,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzC,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,OAAO,SAAS,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;QACrC,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,YAAY,GAAG,CAAC,CAAC;IACvB,MAAM,aAAa,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,SAAqB,EAAE,IAAI,GAAG,YAAY,CAAC,CAAC;IAErF,IAAI,cAAc,GAAmB,EAAE,CAAC;IACxC,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;QACzB,cAAc,GAAG,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,GAAG,YAAY,CAAC,CAAC;IAC3E,CAAC;IAED,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAChC,OAAO,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,IAAI,QAAQ,CAAC,CAAC;IAC1D,CAAC;IAED,MAAM,UAAU,GAAG,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAE5E,MAAM,QAAQ,GAAG,IAAI,GAAG,EAA4E,CAAC;IAErG,KAAK,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;QAC9B,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,EAAE;YACvB,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,MAAM,EAAE,CAAC,CAAC,KAAK;YACf,MAAM,EAAE,CAAC;SACV,CAAC,CAAC;IACL,CAAC;IAED,KAAK,MAAM,CAAC,IAAI,cAAc,EAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC1C,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,MAAM,GAAG,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9D,CAAC;aAAM,CAAC;YACN,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,EAAE;gBACvB,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,MAAM,EAAE,CAAC;gBACT,MAAM,EAAE,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;aAClD,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,MAAM,EAAE,GAAG,OAAO,CAAC,aAAa,IAAI,GAAG,CAAC;IACxC,MAAM,eAAe,GAAmB,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC;SAC3D,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACf,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,KAAK,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,GAAG,EAAE,GAAG,KAAK,CAAC,MAAM;KACnD,CAAC,CAAC;SACF,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,IAAI,QAAQ,CAAC;SAClC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;SACjC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IAElB,OAAO,eAAe,CAAC;AACzB,CAAC"}
|
package/dist/watcher.d.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import type { RagConfig } from "./core/config.js";
|
|
2
|
-
import type { EmbeddingProvider, VectorStore } from "./core/interfaces.js";
|
|
2
|
+
import type { EmbeddingProvider, KeywordIndex, VectorStore } from "./core/interfaces.js";
|
|
3
3
|
export interface BackgroundIndexer {
|
|
4
4
|
close(): Promise<void>;
|
|
5
5
|
}
|
|
@@ -10,5 +10,6 @@ export interface CreateBackgroundIndexerOptions {
|
|
|
10
10
|
store: VectorStore;
|
|
11
11
|
embedder: EmbeddingProvider;
|
|
12
12
|
logFilePath: string;
|
|
13
|
+
keywordIndex?: KeywordIndex;
|
|
13
14
|
}
|
|
14
15
|
export declare function createBackgroundIndexer(options: CreateBackgroundIndexerOptions): BackgroundIndexer;
|
package/dist/watcher.js
CHANGED
|
@@ -2,7 +2,7 @@ import chokidar from "chokidar";
|
|
|
2
2
|
import { appendDebugLog } from "./core/fileLogger.js";
|
|
3
3
|
import { createWatchPassScheduler, createWatchIgnore, runIndexPass, } from "./indexer.js";
|
|
4
4
|
export function createBackgroundIndexer(options) {
|
|
5
|
-
const { cwd, storePath, config, store, embedder, logFilePath } = options;
|
|
5
|
+
const { cwd, storePath, config, store, embedder, logFilePath, keywordIndex } = options;
|
|
6
6
|
// Fire-and-forget initial index pass
|
|
7
7
|
runIndexPass({
|
|
8
8
|
cwd,
|
|
@@ -10,6 +10,7 @@ export function createBackgroundIndexer(options) {
|
|
|
10
10
|
config,
|
|
11
11
|
store,
|
|
12
12
|
embedder,
|
|
13
|
+
keywordIndex,
|
|
13
14
|
logger: {
|
|
14
15
|
//info: (message) => appendDebugLog(logFilePath, { scope: "autoIndex", message }),
|
|
15
16
|
warn: (message) => appendDebugLog(logFilePath, { scope: "autoIndex", message }),
|
|
@@ -29,6 +30,7 @@ export function createBackgroundIndexer(options) {
|
|
|
29
30
|
config,
|
|
30
31
|
store,
|
|
31
32
|
embedder,
|
|
33
|
+
keywordIndex,
|
|
32
34
|
logger: {
|
|
33
35
|
//info: (message) => appendDebugLog(logFilePath, { scope: "autoIndex", message }),
|
|
34
36
|
warn: (message) => appendDebugLog(logFilePath, { scope: "autoIndex", message }),
|
package/dist/watcher.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"watcher.js","sourceRoot":"","sources":["../src/watcher.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,MAAM,UAAU,CAAC;AAEhC,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAGtD,OAAO,EACL,wBAAwB,EACxB,iBAAiB,EACjB,YAAY,GACb,MAAM,cAAc,CAAC;
|
|
1
|
+
{"version":3,"file":"watcher.js","sourceRoot":"","sources":["../src/watcher.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,MAAM,UAAU,CAAC;AAEhC,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAGtD,OAAO,EACL,wBAAwB,EACxB,iBAAiB,EACjB,YAAY,GACb,MAAM,cAAc,CAAC;AAgBtB,MAAM,UAAU,uBAAuB,CAAC,OAAuC;IAC7E,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC;IAEvF,qCAAqC;IACrC,YAAY,CAAC;QACX,GAAG;QACH,SAAS;QACT,MAAM;QACN,KAAK;QACL,QAAQ;QACR,YAAY;QACZ,MAAM,EAAE;YACN,kFAAkF;YAClF,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,cAAc,CAAC,WAAW,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;SAChF;KACF,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;QACf,cAAc,CAAC,WAAW,EAAE;YAC1B,KAAK,EAAE,WAAW;YAClB,OAAO,EAAE,2BAA2B;YACpC,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,KAAK,IAAmB,EAAE;QACxC,IAAI,CAAC;YACH,MAAM,YAAY,CAAC;gBACjB,GAAG;gBACH,SAAS;gBACT,MAAM;gBACN,KAAK;gBACL,QAAQ;gBACR,YAAY;gBACZ,MAAM,EAAE;oBACN,kFAAkF;oBAClF,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,cAAc,CAAC,WAAW,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;iBAChF;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,cAAc,CAAC,WAAW,EAAE;gBAC1B,KAAK,EAAE,WAAW;gBAClB,OAAO,EAAE,2BAA2B;gBACpC,KAAK,EAAE,GAAG;aACX,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,YAAY,GAAG,MAAM,CAAC,QAAQ,CAAC,SAAS,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;IAC1G,MAAM,SAAS,GAAG,wBAAwB,CACxC,OAAO,EACP,CAAC,KAAK,EAAE,EAAE;QACR,MAAM,OAAO,GAAI,KAAe,CAAC,OAAO,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1D,cAAc,CAAC,WAAW,EAAE;YAC1B,KAAK,EAAE,WAAW;YAClB,OAAO,EAAE,yBAAyB,OAAO,EAAE;YAC3C,KAAK;SACN,CAAC,CAAC;IACL,CAAC,EACD,YAAY,CAAC,UAAU,CACxB,CAAC;IAEF,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE;QAClC,OAAO,EAAE,iBAAiB,CAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC;QAClD,aAAa,EAAE,IAAI;QACnB,UAAU,EAAE,IAAI;KACjB,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,GAAG,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,CAAC;IACpD,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;IAChC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IACnC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IACnC,OAAO,CAAC,EAAE,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;IACtC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IACnC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;QAC5B,cAAc,CAAC,WAAW,EAAE;YAC1B,KAAK,EAAE,WAAW;YAClB,OAAO,EAAE,kBAAmB,KAAe,CAAC,OAAO,EAAE;YACrD,KAAK;SACN,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,EAAE;QACrC,SAAS,CAAC,YAAY,EAAE,CAAC;IAC3B,CAAC,EAAE,YAAY,CAAC,UAAU,CAAC,CAAC;IAE5B,OAAO;QACL,KAAK,CAAC,KAAK;YACT,aAAa,CAAC,aAAa,CAAC,CAAC;YAC7B,SAAS,CAAC,KAAK,EAAE,CAAC;YAClB,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;YACtB,cAAc,CAAC,WAAW,EAAE;gBAC1B,KAAK,EAAE,WAAW;gBAClB,OAAO,EAAE,8BAA8B;aACxC,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC"}
|