opencode-rag-plugin 1.2.4 → 1.2.10
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 +491 -463
- package/dist/cli.js +55 -1
- package/dist/cli.js.map +1 -1
- package/dist/plugin-entry.d.ts +1 -1
- package/dist/plugin-entry.js +2 -2
- package/dist/plugin-entry.js.map +1 -1
- package/package.json +90 -90
- package/dist/__tests__/chunker/c.test.d.ts +0 -1
- package/dist/__tests__/chunker/c.test.js +0 -82
- package/dist/__tests__/chunker/c.test.js.map +0 -1
- package/dist/__tests__/chunker/cpp.test.d.ts +0 -1
- package/dist/__tests__/chunker/cpp.test.js +0 -96
- package/dist/__tests__/chunker/cpp.test.js.map +0 -1
- package/dist/__tests__/chunker/csharp.test.d.ts +0 -1
- package/dist/__tests__/chunker/csharp.test.js +0 -85
- package/dist/__tests__/chunker/csharp.test.js.map +0 -1
- package/dist/__tests__/chunker/css.test.d.ts +0 -1
- package/dist/__tests__/chunker/css.test.js +0 -64
- package/dist/__tests__/chunker/css.test.js.map +0 -1
- package/dist/__tests__/chunker/factory.test.d.ts +0 -1
- package/dist/__tests__/chunker/factory.test.js +0 -202
- package/dist/__tests__/chunker/factory.test.js.map +0 -1
- package/dist/__tests__/chunker/fallback.test.d.ts +0 -1
- package/dist/__tests__/chunker/fallback.test.js +0 -67
- package/dist/__tests__/chunker/fallback.test.js.map +0 -1
- package/dist/__tests__/chunker/go.test.d.ts +0 -1
- package/dist/__tests__/chunker/go.test.js +0 -90
- package/dist/__tests__/chunker/go.test.js.map +0 -1
- package/dist/__tests__/chunker/grammar.test.d.ts +0 -1
- package/dist/__tests__/chunker/grammar.test.js +0 -108
- package/dist/__tests__/chunker/grammar.test.js.map +0 -1
- package/dist/__tests__/chunker/html.test.d.ts +0 -1
- package/dist/__tests__/chunker/html.test.js +0 -64
- package/dist/__tests__/chunker/html.test.js.map +0 -1
- package/dist/__tests__/chunker/java.test.d.ts +0 -1
- package/dist/__tests__/chunker/java.test.js +0 -83
- package/dist/__tests__/chunker/java.test.js.map +0 -1
- package/dist/__tests__/chunker/javascript.test.d.ts +0 -1
- package/dist/__tests__/chunker/javascript.test.js +0 -88
- package/dist/__tests__/chunker/javascript.test.js.map +0 -1
- package/dist/__tests__/chunker/json.test.d.ts +0 -1
- package/dist/__tests__/chunker/json.test.js +0 -54
- package/dist/__tests__/chunker/json.test.js.map +0 -1
- package/dist/__tests__/chunker/kotlin.test.d.ts +0 -1
- package/dist/__tests__/chunker/kotlin.test.js +0 -42
- package/dist/__tests__/chunker/kotlin.test.js.map +0 -1
- package/dist/__tests__/chunker/markdown.test.d.ts +0 -1
- package/dist/__tests__/chunker/markdown.test.js +0 -76
- package/dist/__tests__/chunker/markdown.test.js.map +0 -1
- package/dist/__tests__/chunker/python.test.d.ts +0 -1
- package/dist/__tests__/chunker/python.test.js +0 -81
- package/dist/__tests__/chunker/python.test.js.map +0 -1
- package/dist/__tests__/chunker/razor.test.d.ts +0 -1
- package/dist/__tests__/chunker/razor.test.js +0 -73
- package/dist/__tests__/chunker/razor.test.js.map +0 -1
- package/dist/__tests__/chunker/register.test.d.ts +0 -1
- package/dist/__tests__/chunker/register.test.js +0 -93
- package/dist/__tests__/chunker/register.test.js.map +0 -1
- package/dist/__tests__/chunker/ruby.test.d.ts +0 -1
- package/dist/__tests__/chunker/ruby.test.js +0 -40
- package/dist/__tests__/chunker/ruby.test.js.map +0 -1
- package/dist/__tests__/chunker/rust.test.d.ts +0 -1
- package/dist/__tests__/chunker/rust.test.js +0 -42
- package/dist/__tests__/chunker/rust.test.js.map +0 -1
- package/dist/__tests__/chunker/sln.test.d.ts +0 -1
- package/dist/__tests__/chunker/sln.test.js +0 -45
- package/dist/__tests__/chunker/sln.test.js.map +0 -1
- package/dist/__tests__/chunker/swift.test.d.ts +0 -1
- package/dist/__tests__/chunker/swift.test.js +0 -49
- package/dist/__tests__/chunker/swift.test.js.map +0 -1
- package/dist/__tests__/chunker/typescript.test.d.ts +0 -1
- package/dist/__tests__/chunker/typescript.test.js +0 -92
- package/dist/__tests__/chunker/typescript.test.js.map +0 -1
- package/dist/__tests__/chunker/uuid.test.d.ts +0 -1
- package/dist/__tests__/chunker/uuid.test.js +0 -19
- package/dist/__tests__/chunker/uuid.test.js.map +0 -1
- package/dist/__tests__/chunker/xml.test.d.ts +0 -1
- package/dist/__tests__/chunker/xml.test.js +0 -50
- package/dist/__tests__/chunker/xml.test.js.map +0 -1
- package/dist/__tests__/core/config.test.d.ts +0 -1
- package/dist/__tests__/core/config.test.js +0 -75
- package/dist/__tests__/core/config.test.js.map +0 -1
- package/dist/__tests__/core/fileLogger.test.d.ts +0 -1
- package/dist/__tests__/core/fileLogger.test.js +0 -34
- package/dist/__tests__/core/fileLogger.test.js.map +0 -1
- package/dist/__tests__/core/manifest.test.d.ts +0 -1
- package/dist/__tests__/core/manifest.test.js +0 -56
- package/dist/__tests__/core/manifest.test.js.map +0 -1
- package/dist/__tests__/embedder/embedBatch.test.d.ts +0 -1
- package/dist/__tests__/embedder/embedBatch.test.js +0 -88
- package/dist/__tests__/embedder/embedBatch.test.js.map +0 -1
- package/dist/__tests__/embedder/factory.test.d.ts +0 -1
- package/dist/__tests__/embedder/factory.test.js +0 -71
- package/dist/__tests__/embedder/factory.test.js.map +0 -1
- package/dist/__tests__/embedder/ollama.test.d.ts +0 -1
- package/dist/__tests__/embedder/ollama.test.js +0 -106
- package/dist/__tests__/embedder/ollama.test.js.map +0 -1
- package/dist/__tests__/embedder/openai.test.d.ts +0 -1
- package/dist/__tests__/embedder/openai.test.js +0 -94
- package/dist/__tests__/embedder/openai.test.js.map +0 -1
- package/dist/__tests__/indexer/indexer.test.d.ts +0 -1
- package/dist/__tests__/indexer/indexer.test.js +0 -176
- package/dist/__tests__/indexer/indexer.test.js.map +0 -1
- package/dist/__tests__/plugin.test.d.ts +0 -1
- package/dist/__tests__/plugin.test.js +0 -77
- package/dist/__tests__/plugin.test.js.map +0 -1
- package/dist/__tests__/retriever/retriever.test.d.ts +0 -1
- package/dist/__tests__/retriever/retriever.test.js +0 -97
- package/dist/__tests__/retriever/retriever.test.js.map +0 -1
- package/dist/__tests__/vectorstore/lancedb.test.d.ts +0 -1
- package/dist/__tests__/vectorstore/lancedb.test.js +0 -159
- package/dist/__tests__/vectorstore/lancedb.test.js.map +0 -1
|
@@ -1,19 +0,0 @@
|
|
|
1
|
-
import { describe, it } from "node:test";
|
|
2
|
-
import assert from "node:assert/strict";
|
|
3
|
-
import { uuid } from "../../chunker/uuid.js";
|
|
4
|
-
describe("uuid", () => {
|
|
5
|
-
it("generates a valid UUID v4 format", () => {
|
|
6
|
-
const id = uuid();
|
|
7
|
-
assert.match(id, /^[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/);
|
|
8
|
-
});
|
|
9
|
-
it("generates unique IDs", () => {
|
|
10
|
-
const ids = new Set(Array.from({ length: 100 }, () => uuid()));
|
|
11
|
-
assert.equal(ids.size, 100);
|
|
12
|
-
});
|
|
13
|
-
it("generates IDs as strings", () => {
|
|
14
|
-
const id = uuid();
|
|
15
|
-
assert.equal(typeof id, "string");
|
|
16
|
-
assert.equal(id.length, 36);
|
|
17
|
-
});
|
|
18
|
-
});
|
|
19
|
-
//# sourceMappingURL=uuid.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"uuid.test.js","sourceRoot":"","sources":["../../../src/__tests__/chunker/uuid.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,IAAI,EAAE,MAAM,uBAAuB,CAAC;AAE7C,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;IACpB,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,EAAE,GAAG,IAAI,EAAE,CAAC;QAClB,MAAM,CAAC,KAAK,CACV,EAAE,EACF,uEAAuE,CACxE,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAC/D,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,EAAE,GAAG,IAAI,EAAE,CAAC;QAClB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,QAAQ,CAAC,CAAC;QAClC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAC9B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|
|
@@ -1,50 +0,0 @@
|
|
|
1
|
-
import { describe, it } from "node:test";
|
|
2
|
-
import assert from "node:assert/strict";
|
|
3
|
-
import { xmlChunker } from "../../chunker/xml.js";
|
|
4
|
-
describe("XmlChunker (csproj)", () => {
|
|
5
|
-
it("language is 'xml'", () => {
|
|
6
|
-
assert.equal(xmlChunker.language, "xml");
|
|
7
|
-
});
|
|
8
|
-
it("fileExtensions includes .xml and .csproj", () => {
|
|
9
|
-
assert.ok(xmlChunker.fileExtensions.includes(".xml"));
|
|
10
|
-
assert.ok(xmlChunker.fileExtensions.includes(".csproj"));
|
|
11
|
-
});
|
|
12
|
-
it("grammarName is 'xml'", () => {
|
|
13
|
-
assert.equal(xmlChunker.grammarName, "xml");
|
|
14
|
-
});
|
|
15
|
-
it("nodeTypes contains element", () => {
|
|
16
|
-
assert.ok(xmlChunker.nodeTypes.has("element"));
|
|
17
|
-
});
|
|
18
|
-
it("chunk returns empty array for empty content", async () => {
|
|
19
|
-
const chunks = await xmlChunker.chunk("test.csproj", "");
|
|
20
|
-
assert.deepStrictEqual(chunks, []);
|
|
21
|
-
});
|
|
22
|
-
it("chunk parses the root element of a csproj file", async () => {
|
|
23
|
-
const csproj = `<Project Sdk="Microsoft.NET.Sdk">
|
|
24
|
-
<PropertyGroup>
|
|
25
|
-
<TargetFramework>net8.0</TargetFramework>
|
|
26
|
-
</PropertyGroup>
|
|
27
|
-
<ItemGroup>
|
|
28
|
-
<PackageReference Include="Newtonsoft.Json" Version="13.0.3" />
|
|
29
|
-
</ItemGroup>
|
|
30
|
-
</Project>`;
|
|
31
|
-
const chunks = await xmlChunker.chunk("test.csproj", csproj);
|
|
32
|
-
assert.equal(chunks.length, 1);
|
|
33
|
-
assert.ok(chunks[0].content.includes("Project"));
|
|
34
|
-
assert.ok(chunks[0].content.includes("PropertyGroup"));
|
|
35
|
-
assert.ok(chunks[0].content.includes("ItemGroup"));
|
|
36
|
-
});
|
|
37
|
-
it("chunk generates unique IDs", async () => {
|
|
38
|
-
const csproj = `<Project><PropertyGroup><A>1</A></PropertyGroup></Project>`;
|
|
39
|
-
const chunks = await xmlChunker.chunk("test.csproj", csproj);
|
|
40
|
-
const ids = chunks.map((c) => c.id);
|
|
41
|
-
assert.equal(new Set(ids).size, ids.length);
|
|
42
|
-
});
|
|
43
|
-
it("chunk sets correct startLine metadata", async () => {
|
|
44
|
-
const csproj = `\n<Project Sdk="X"><PropertyGroup><A>1</A></PropertyGroup></Project>`;
|
|
45
|
-
const chunks = await xmlChunker.chunk("test.csproj", csproj);
|
|
46
|
-
assert.equal(chunks.length, 1);
|
|
47
|
-
assert.equal(chunks[0].metadata.startLine, 2);
|
|
48
|
-
});
|
|
49
|
-
});
|
|
50
|
-
//# sourceMappingURL=xml.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"xml.test.js","sourceRoot":"","sources":["../../../src/__tests__/chunker/xml.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAElD,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,mBAAmB,EAAE,GAAG,EAAE;QAC3B,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;QACtD,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;QACzD,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,MAAM,GAAG;;;;;;;WAOR,CAAC;QACR,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;QAC7D,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC;QACxD,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC1C,MAAM,MAAM,GAAG,4DAA4D,CAAC;QAC5E,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;QAC7D,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,MAAM,GAAG,sEAAsE,CAAC;QACtF,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;QAC7D,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|
|
@@ -1,75 +0,0 @@
|
|
|
1
|
-
import { describe, it, before, after } from "node:test";
|
|
2
|
-
import assert from "node:assert/strict";
|
|
3
|
-
import { writeFileSync, unlinkSync } from "node:fs";
|
|
4
|
-
import { tmpdir } from "node:os";
|
|
5
|
-
import { join } from "node:path";
|
|
6
|
-
import { loadConfig, DEFAULT_CONFIG } from "../../core/config.js";
|
|
7
|
-
describe("loadConfig", () => {
|
|
8
|
-
let tmpFile;
|
|
9
|
-
before(() => {
|
|
10
|
-
tmpFile = join(tmpdir(), `opencode-rag-test-${Date.now()}.json`);
|
|
11
|
-
});
|
|
12
|
-
after(() => {
|
|
13
|
-
try {
|
|
14
|
-
unlinkSync(tmpFile);
|
|
15
|
-
}
|
|
16
|
-
catch {
|
|
17
|
-
// ignore
|
|
18
|
-
}
|
|
19
|
-
});
|
|
20
|
-
it("returns default config for empty file", () => {
|
|
21
|
-
writeFileSync(tmpFile, "{}", "utf-8");
|
|
22
|
-
const config = loadConfig(tmpFile);
|
|
23
|
-
assert.deepStrictEqual(config, { ...DEFAULT_CONFIG, chunkers: undefined });
|
|
24
|
-
});
|
|
25
|
-
it("allows partial override of embedding settings", () => {
|
|
26
|
-
writeFileSync(tmpFile, JSON.stringify({ embedding: { provider: "openai", model: "custom-model" } }), "utf-8");
|
|
27
|
-
const config = loadConfig(tmpFile);
|
|
28
|
-
assert.equal(config.embedding.provider, "openai");
|
|
29
|
-
assert.equal(config.embedding.model, "custom-model");
|
|
30
|
-
assert.equal(config.embedding.baseUrl, DEFAULT_CONFIG.embedding.baseUrl);
|
|
31
|
-
});
|
|
32
|
-
it("allows partial override of indexing settings", () => {
|
|
33
|
-
writeFileSync(tmpFile, JSON.stringify({ indexing: { chunkOverlap: 5 } }), "utf-8");
|
|
34
|
-
const config = loadConfig(tmpFile);
|
|
35
|
-
assert.equal(config.indexing.chunkOverlap, 5);
|
|
36
|
-
assert.deepStrictEqual(config.indexing.includeExtensions, DEFAULT_CONFIG.indexing.includeExtensions);
|
|
37
|
-
});
|
|
38
|
-
it("allows partial override of retrieval settings", () => {
|
|
39
|
-
writeFileSync(tmpFile, JSON.stringify({ retrieval: { topK: 20 } }), "utf-8");
|
|
40
|
-
const config = loadConfig(tmpFile);
|
|
41
|
-
assert.equal(config.retrieval.topK, 20);
|
|
42
|
-
});
|
|
43
|
-
it("allows partial override of openCode settings", () => {
|
|
44
|
-
writeFileSync(tmpFile, JSON.stringify({ openCode: { maxContextChunks: 10 } }), "utf-8");
|
|
45
|
-
const config = loadConfig(tmpFile);
|
|
46
|
-
assert.equal(config.openCode.maxContextChunks, 10);
|
|
47
|
-
assert.equal(config.openCode.enabled, DEFAULT_CONFIG.openCode.enabled);
|
|
48
|
-
});
|
|
49
|
-
it("allows partial override of vectorStore path", () => {
|
|
50
|
-
writeFileSync(tmpFile, JSON.stringify({ vectorStore: { path: "/custom/path" } }), "utf-8");
|
|
51
|
-
const config = loadConfig(tmpFile);
|
|
52
|
-
assert.equal(config.vectorStore.path, "/custom/path");
|
|
53
|
-
});
|
|
54
|
-
});
|
|
55
|
-
describe("DEFAULT_CONFIG", () => {
|
|
56
|
-
it("has ollama as default embedding provider", () => {
|
|
57
|
-
assert.equal(DEFAULT_CONFIG.embedding.provider, "ollama");
|
|
58
|
-
});
|
|
59
|
-
it("includes TypeScript extensions", () => {
|
|
60
|
-
assert.ok(DEFAULT_CONFIG.indexing.includeExtensions.includes(".ts"));
|
|
61
|
-
assert.ok(DEFAULT_CONFIG.indexing.includeExtensions.includes(".tsx"));
|
|
62
|
-
});
|
|
63
|
-
it("excludes node_modules, .git, and .opencode", () => {
|
|
64
|
-
assert.ok(DEFAULT_CONFIG.indexing.excludeDirs.includes("node_modules"));
|
|
65
|
-
assert.ok(DEFAULT_CONFIG.indexing.excludeDirs.includes(".git"));
|
|
66
|
-
assert.ok(DEFAULT_CONFIG.indexing.excludeDirs.includes(".opencode"));
|
|
67
|
-
});
|
|
68
|
-
it("has openCode enabled by default", () => {
|
|
69
|
-
assert.equal(DEFAULT_CONFIG.openCode.enabled, true);
|
|
70
|
-
});
|
|
71
|
-
it("has topK of 10", () => {
|
|
72
|
-
assert.equal(DEFAULT_CONFIG.retrieval.topK, 10);
|
|
73
|
-
});
|
|
74
|
-
});
|
|
75
|
-
//# sourceMappingURL=config.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"config.test.js","sourceRoot":"","sources":["../../../src/__tests__/core/config.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,WAAW,CAAC;AACxD,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACpD,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAElE,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,IAAI,OAAe,CAAC;IAEpB,MAAM,CAAC,GAAG,EAAE;QACV,OAAO,GAAG,IAAI,CAAC,MAAM,EAAE,EAAE,qBAAqB,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,GAAG,EAAE;QACT,IAAI,CAAC;YACH,UAAU,CAAC,OAAO,CAAC,CAAC;QACtB,CAAC;QAAC,MAAM,CAAC;YACP,SAAS;QACX,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,aAAa,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QACtC,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC;QACnC,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE,GAAG,cAAc,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,aAAa,CACX,OAAO,EACP,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE,CAAC,EAC5E,OAAO,CACR,CAAC;QACF,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;QACrD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,cAAc,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,aAAa,CACX,OAAO,EACP,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC,EACjD,OAAO,CACR,CAAC;QACF,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,eAAe,CACpB,MAAM,CAAC,QAAQ,CAAC,iBAAiB,EACjC,cAAc,CAAC,QAAQ,CAAC,iBAAiB,CAC1C,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,aAAa,CACX,OAAO,EACP,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC,EAC3C,OAAO,CACR,CAAC;QACF,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,aAAa,CACX,OAAO,EACP,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,EAAE,gBAAgB,EAAE,EAAE,EAAE,EAAE,CAAC,EACtD,OAAO,CACR,CAAC;QACF,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,cAAc,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,aAAa,CACX,OAAO,EACP,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,EAAE,CAAC,EACzD,OAAO,CACR,CAAC;QACF,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,CAAC,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,iBAAiB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;QACrE,MAAM,CAAC,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,iBAAiB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,CAAC,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,WAAW,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC;QACxE,MAAM,CAAC,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,WAAW,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;QAChE,MAAM,CAAC,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,WAAW,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gBAAgB,EAAE,GAAG,EAAE;QACxB,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|
|
@@ -1,34 +0,0 @@
|
|
|
1
|
-
import { describe, it } from "node:test";
|
|
2
|
-
import assert from "node:assert/strict";
|
|
3
|
-
import { mkdtempSync, readFileSync, rmSync } from "node:fs";
|
|
4
|
-
import { tmpdir } from "node:os";
|
|
5
|
-
import path from "node:path";
|
|
6
|
-
import { appendDebugLog } from "../../core/fileLogger.js";
|
|
7
|
-
describe("appendDebugLog", () => {
|
|
8
|
-
it("writes error entries to a file and appends subsequent entries", () => {
|
|
9
|
-
const dir = mkdtempSync(path.join(tmpdir(), "opencode-rag-log-test-"));
|
|
10
|
-
try {
|
|
11
|
-
const logFilePath = path.join(dir, ".opencode", "opencode-rag.log");
|
|
12
|
-
appendDebugLog(logFilePath, {
|
|
13
|
-
scope: "chat.message",
|
|
14
|
-
message: "first failure",
|
|
15
|
-
error: new Error("boom"),
|
|
16
|
-
});
|
|
17
|
-
appendDebugLog(logFilePath, {
|
|
18
|
-
scope: "tool.execute.after",
|
|
19
|
-
message: "second failure",
|
|
20
|
-
error: "plain string error",
|
|
21
|
-
});
|
|
22
|
-
const content = readFileSync(logFilePath, "utf8");
|
|
23
|
-
assert.match(content, /\[chat\.message\] first failure/);
|
|
24
|
-
assert.match(content, /Error: boom/);
|
|
25
|
-
assert.match(content, /\[tool\.execute\.after\] second failure/);
|
|
26
|
-
assert.match(content, /plain string error/);
|
|
27
|
-
assert.ok(content.includes("\n\n"));
|
|
28
|
-
}
|
|
29
|
-
finally {
|
|
30
|
-
rmSync(dir, { recursive: true, force: true });
|
|
31
|
-
}
|
|
32
|
-
});
|
|
33
|
-
});
|
|
34
|
-
//# sourceMappingURL=fileLogger.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"fileLogger.test.js","sourceRoot":"","sources":["../../../src/__tests__/core/fileLogger.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAC5D,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAE1D,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,wBAAwB,CAAC,CAAC,CAAC;QAEvE,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,WAAW,EAAE,kBAAkB,CAAC,CAAC;YAEpE,cAAc,CAAC,WAAW,EAAE;gBAC1B,KAAK,EAAE,cAAc;gBACrB,OAAO,EAAE,eAAe;gBACxB,KAAK,EAAE,IAAI,KAAK,CAAC,MAAM,CAAC;aACzB,CAAC,CAAC;YAEH,cAAc,CAAC,WAAW,EAAE;gBAC1B,KAAK,EAAE,oBAAoB;gBAC3B,OAAO,EAAE,gBAAgB;gBACzB,KAAK,EAAE,oBAAoB;aAC5B,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,YAAY,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,iCAAiC,CAAC,CAAC;YACzD,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;YACrC,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,yCAAyC,CAAC,CAAC;YACjE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,oBAAoB,CAAC,CAAC;YAC5C,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;QACtC,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAChD,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|
|
@@ -1,56 +0,0 @@
|
|
|
1
|
-
import { describe, it } from "node:test";
|
|
2
|
-
import assert from "node:assert/strict";
|
|
3
|
-
import fs from "node:fs/promises";
|
|
4
|
-
import os from "node:os";
|
|
5
|
-
import path from "node:path";
|
|
6
|
-
import { computeFileHash, createEmptyManifest, loadManifest, manifestPathFor, normalizeFilePath, saveManifest, } from "../../core/manifest.js";
|
|
7
|
-
async function makeTempDir(name) {
|
|
8
|
-
return fs.mkdtemp(path.join(os.tmpdir(), `${name}-`));
|
|
9
|
-
}
|
|
10
|
-
describe("manifest", () => {
|
|
11
|
-
it("creates an empty manifest", () => {
|
|
12
|
-
assert.deepStrictEqual(createEmptyManifest(), { files: {} });
|
|
13
|
-
});
|
|
14
|
-
it("normalizes file paths to absolute forward-slash paths", () => {
|
|
15
|
-
const normalized = normalizeFilePath("src\\test.ts");
|
|
16
|
-
assert.match(normalized, /^([A-Za-z]:)?\//);
|
|
17
|
-
assert.ok(!normalized.includes("\\"));
|
|
18
|
-
});
|
|
19
|
-
it("computes stable hashes", () => {
|
|
20
|
-
assert.equal(computeFileHash("abc"), computeFileHash("abc"));
|
|
21
|
-
assert.notEqual(computeFileHash("abc"), computeFileHash("abcd"));
|
|
22
|
-
});
|
|
23
|
-
it("returns missing status when manifest file does not exist", async () => {
|
|
24
|
-
const dir = await makeTempDir("manifest-missing");
|
|
25
|
-
const result = await loadManifest(dir);
|
|
26
|
-
assert.equal(result.status, "missing");
|
|
27
|
-
assert.deepStrictEqual(result.manifest, { files: {} });
|
|
28
|
-
assert.equal(result.path, manifestPathFor(dir));
|
|
29
|
-
});
|
|
30
|
-
it("saves and loads manifest data", async () => {
|
|
31
|
-
const dir = await makeTempDir("manifest-save");
|
|
32
|
-
const manifest = {
|
|
33
|
-
lastIndexedAt: 123,
|
|
34
|
-
files: {
|
|
35
|
-
"/tmp/example.ts": {
|
|
36
|
-
hash: "hash-1",
|
|
37
|
-
chunkCount: 2,
|
|
38
|
-
indexedAt: 123,
|
|
39
|
-
},
|
|
40
|
-
},
|
|
41
|
-
};
|
|
42
|
-
await saveManifest(dir, manifest);
|
|
43
|
-
const result = await loadManifest(dir);
|
|
44
|
-
assert.equal(result.status, "ok");
|
|
45
|
-
assert.deepStrictEqual(result.manifest, manifest);
|
|
46
|
-
});
|
|
47
|
-
it("returns corrupt status for invalid JSON", async () => {
|
|
48
|
-
const dir = await makeTempDir("manifest-corrupt");
|
|
49
|
-
await fs.mkdir(dir, { recursive: true });
|
|
50
|
-
await fs.writeFile(manifestPathFor(dir), "{not-json", "utf-8");
|
|
51
|
-
const result = await loadManifest(dir);
|
|
52
|
-
assert.equal(result.status, "corrupt");
|
|
53
|
-
assert.deepStrictEqual(result.manifest, { files: {} });
|
|
54
|
-
});
|
|
55
|
-
});
|
|
56
|
-
//# sourceMappingURL=manifest.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"manifest.test.js","sourceRoot":"","sources":["../../../src/__tests__/core/manifest.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EACL,eAAe,EACf,mBAAmB,EACnB,YAAY,EACZ,eAAe,EACf,iBAAiB,EACjB,YAAY,GACb,MAAM,wBAAwB,CAAC;AAEhC,KAAK,UAAU,WAAW,CAAC,IAAY;IACrC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC;AACxD,CAAC;AAED,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACxB,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,CAAC,eAAe,CAAC,mBAAmB,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,UAAU,GAAG,iBAAiB,CAAC,cAAc,CAAC,CAAC;QACrD,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,iBAAiB,CAAC,CAAC;QAC5C,MAAM,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;QAChC,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC;QAC7D,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,GAAG,GAAG,MAAM,WAAW,CAAC,kBAAkB,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,GAAG,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QACvC,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACvD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,GAAG,GAAG,MAAM,WAAW,CAAC,eAAe,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG;YACf,aAAa,EAAE,GAAG;YAClB,KAAK,EAAE;gBACL,iBAAiB,EAAE;oBACjB,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE,CAAC;oBACb,SAAS,EAAE,GAAG;iBACf;aACF;SACF,CAAC;QAEF,MAAM,YAAY,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;QAClC,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,GAAG,CAAC,CAAC;QAEvC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,GAAG,GAAG,MAAM,WAAW,CAAC,kBAAkB,CAAC,CAAC;QAClD,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACzC,MAAM,EAAE,CAAC,SAAS,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;QAE/D,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,GAAG,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QACvC,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|
|
@@ -1,88 +0,0 @@
|
|
|
1
|
-
import { describe, it } from "node:test";
|
|
2
|
-
import assert from "node:assert/strict";
|
|
3
|
-
import { embedBatch } from "../../embedder/factory.js";
|
|
4
|
-
function mockEmbedder() {
|
|
5
|
-
const state = { calls: 0, lastBatchSizes: [] };
|
|
6
|
-
return {
|
|
7
|
-
name: "mock",
|
|
8
|
-
get calls() { return state.calls; },
|
|
9
|
-
get lastBatchSizes() { return [...state.lastBatchSizes]; },
|
|
10
|
-
async embed(texts) {
|
|
11
|
-
state.calls++;
|
|
12
|
-
state.lastBatchSizes.push(texts.length);
|
|
13
|
-
// Return embedding where first element encodes the call number and position
|
|
14
|
-
return texts.map((_, i) => [state.calls * 100 + i]);
|
|
15
|
-
},
|
|
16
|
-
};
|
|
17
|
-
}
|
|
18
|
-
describe("embedBatch", () => {
|
|
19
|
-
it("returns empty array for empty input", async () => {
|
|
20
|
-
const m = mockEmbedder();
|
|
21
|
-
const result = await embedBatch(m, []);
|
|
22
|
-
assert.deepStrictEqual(result, []);
|
|
23
|
-
assert.equal(m.calls, 0);
|
|
24
|
-
});
|
|
25
|
-
it("returns single embedding for single text", async () => {
|
|
26
|
-
const m = mockEmbedder();
|
|
27
|
-
const result = await embedBatch(m, ["hello"]);
|
|
28
|
-
assert.equal(result.length, 1);
|
|
29
|
-
assert.deepStrictEqual(result[0], [100]);
|
|
30
|
-
assert.equal(m.calls, 1);
|
|
31
|
-
});
|
|
32
|
-
it("returns multiple embeddings for multiple texts (single batch)", async () => {
|
|
33
|
-
const m = mockEmbedder();
|
|
34
|
-
const result = await embedBatch(m, ["a", "b", "c"]);
|
|
35
|
-
assert.equal(result.length, 3);
|
|
36
|
-
assert.equal(m.calls, 1);
|
|
37
|
-
assert.equal(m.lastBatchSizes[0], 3);
|
|
38
|
-
});
|
|
39
|
-
it("respects custom batchSize by calling embed multiple times", async () => {
|
|
40
|
-
const m = mockEmbedder();
|
|
41
|
-
const result = await embedBatch(m, ["a", "b", "c", "d", "e"], 2);
|
|
42
|
-
assert.equal(result.length, 5);
|
|
43
|
-
// 5 items / 2 = 3 batches (2, 2, 1)
|
|
44
|
-
assert.equal(m.calls, 3);
|
|
45
|
-
assert.deepStrictEqual(m.lastBatchSizes, [2, 2, 1]);
|
|
46
|
-
});
|
|
47
|
-
it("batch of exactly batchSize makes one call", async () => {
|
|
48
|
-
const m = mockEmbedder();
|
|
49
|
-
const result = await embedBatch(m, ["a", "b", "c"], 3);
|
|
50
|
-
assert.equal(result.length, 3);
|
|
51
|
-
assert.equal(m.calls, 1);
|
|
52
|
-
assert.equal(m.lastBatchSizes[0], 3);
|
|
53
|
-
});
|
|
54
|
-
it("flattens results correctly across batches", async () => {
|
|
55
|
-
const m = mockEmbedder();
|
|
56
|
-
const result = await embedBatch(m, ["a", "b", "c", "d"], 2);
|
|
57
|
-
assert.equal(result.length, 4);
|
|
58
|
-
// Batch 1: [[100], [101]] Batch 2: [[200], [201]]
|
|
59
|
-
assert.deepStrictEqual(result, [[100], [101], [200], [201]]);
|
|
60
|
-
});
|
|
61
|
-
it("default batchSize is 10", async () => {
|
|
62
|
-
const m = mockEmbedder();
|
|
63
|
-
const texts = Array.from({ length: 25 }, (_, i) => `text-${i}`);
|
|
64
|
-
const result = await embedBatch(m, texts);
|
|
65
|
-
assert.equal(result.length, 25);
|
|
66
|
-
// 25 items with default batchSize 10: 3 batches (10, 10, 5)
|
|
67
|
-
assert.equal(m.calls, 3);
|
|
68
|
-
assert.deepStrictEqual(m.lastBatchSizes, [10, 10, 5]);
|
|
69
|
-
});
|
|
70
|
-
it("handles single item larger than batchSize", async () => {
|
|
71
|
-
const m = mockEmbedder();
|
|
72
|
-
const result = await embedBatch(m, ["single"], 5);
|
|
73
|
-
assert.equal(result.length, 1);
|
|
74
|
-
assert.equal(m.calls, 1);
|
|
75
|
-
assert.equal(m.lastBatchSizes[0], 1);
|
|
76
|
-
});
|
|
77
|
-
it("passes all texts in last batch even when smaller than batchSize", async () => {
|
|
78
|
-
const m = mockEmbedder();
|
|
79
|
-
const result = await embedBatch(m, ["a", "b", "c", "d"], 3);
|
|
80
|
-
// Batch 1: 3 items, Batch 2: 1 item
|
|
81
|
-
assert.equal(result.length, 4);
|
|
82
|
-
assert.equal(m.calls, 2);
|
|
83
|
-
assert.deepStrictEqual(m.lastBatchSizes, [3, 1]);
|
|
84
|
-
// Verify last batch content: last item gets call=2 embedding
|
|
85
|
-
assert.deepStrictEqual(result[3], [200]);
|
|
86
|
-
});
|
|
87
|
-
});
|
|
88
|
-
//# sourceMappingURL=embedBatch.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"embedBatch.test.js","sourceRoot":"","sources":["../../../src/__tests__/embedder/embedBatch.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAGvD,SAAS,YAAY;IAInB,MAAM,KAAK,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,cAAc,EAAE,EAAc,EAAE,CAAC;IAC3D,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,IAAI,KAAK,KAAK,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;QACnC,IAAI,cAAc,KAAK,OAAO,CAAC,GAAG,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;QAC1D,KAAK,CAAC,KAAK,CAAC,KAAe;YACzB,KAAK,CAAC,KAAK,EAAE,CAAC;YACd,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACxC,4EAA4E;YAC5E,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,GAAG,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;QACtD,CAAC;KACF,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,CAAC,GAAG,YAAY,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACvC,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,CAAC,GAAG,YAAY,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,CAAC,GAAG,YAAY,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,CAAC,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QACpD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACzB,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,CAAC,GAAG,YAAY,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,CAAC,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QACjE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC/B,oCAAoC;QACpC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACzB,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,CAAC,GAAG,YAAY,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,CAAC,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACzB,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,CAAC,GAAG,YAAY,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,CAAC,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5D,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC/B,mDAAmD;QACnD,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,CAAC,GAAG,YAAY,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAChE,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QAC1C,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAChC,4DAA4D;QAC5D,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACzB,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,CAAC,GAAG,YAAY,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACzB,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,MAAM,CAAC,GAAG,YAAY,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,CAAC,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5D,oCAAoC;QACpC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACzB,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACjD,6DAA6D;QAC7D,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|
|
@@ -1,71 +0,0 @@
|
|
|
1
|
-
import { describe, it } from "node:test";
|
|
2
|
-
import assert from "node:assert/strict";
|
|
3
|
-
import { createEmbedder } from "../../embedder/factory.js";
|
|
4
|
-
import { DEFAULT_CONFIG } from "../../core/config.js";
|
|
5
|
-
function makeConfig(overrides) {
|
|
6
|
-
return {
|
|
7
|
-
...DEFAULT_CONFIG,
|
|
8
|
-
...overrides,
|
|
9
|
-
embedding: {
|
|
10
|
-
...DEFAULT_CONFIG.embedding,
|
|
11
|
-
...overrides.embedding,
|
|
12
|
-
},
|
|
13
|
-
};
|
|
14
|
-
}
|
|
15
|
-
describe("createEmbedder", () => {
|
|
16
|
-
it("creates Ollama provider with default config", () => {
|
|
17
|
-
const embedder = createEmbedder(DEFAULT_CONFIG);
|
|
18
|
-
assert.equal(embedder.name, "ollama");
|
|
19
|
-
});
|
|
20
|
-
it("creates Ollama provider explicitly", () => {
|
|
21
|
-
const config = makeConfig({
|
|
22
|
-
embedding: {
|
|
23
|
-
provider: "ollama",
|
|
24
|
-
baseUrl: "http://localhost:11434/v1",
|
|
25
|
-
model: "custom-model",
|
|
26
|
-
apiKey: undefined,
|
|
27
|
-
},
|
|
28
|
-
});
|
|
29
|
-
const embedder = createEmbedder(config);
|
|
30
|
-
assert.equal(embedder.name, "ollama");
|
|
31
|
-
});
|
|
32
|
-
it("creates OpenAI provider with apiKey", () => {
|
|
33
|
-
const config = makeConfig({
|
|
34
|
-
embedding: {
|
|
35
|
-
provider: "openai",
|
|
36
|
-
baseUrl: "https://api.openai.com/v1",
|
|
37
|
-
model: "text-embedding-3-small",
|
|
38
|
-
apiKey: "sk-test-key",
|
|
39
|
-
},
|
|
40
|
-
});
|
|
41
|
-
const embedder = createEmbedder(config);
|
|
42
|
-
assert.equal(embedder.name, "openai");
|
|
43
|
-
});
|
|
44
|
-
it("throws for OpenAI provider without apiKey", () => {
|
|
45
|
-
const config = makeConfig({
|
|
46
|
-
embedding: {
|
|
47
|
-
provider: "openai",
|
|
48
|
-
baseUrl: "https://api.openai.com/v1",
|
|
49
|
-
model: "text-embedding-3-small",
|
|
50
|
-
apiKey: undefined,
|
|
51
|
-
},
|
|
52
|
-
});
|
|
53
|
-
assert.throws(() => createEmbedder(config), {
|
|
54
|
-
message: "OpenAI provider requires an apiKey",
|
|
55
|
-
});
|
|
56
|
-
});
|
|
57
|
-
it("throws for unknown provider", () => {
|
|
58
|
-
const config = makeConfig({
|
|
59
|
-
embedding: {
|
|
60
|
-
provider: "unknown",
|
|
61
|
-
baseUrl: "",
|
|
62
|
-
model: "",
|
|
63
|
-
apiKey: undefined,
|
|
64
|
-
},
|
|
65
|
-
});
|
|
66
|
-
assert.throws(() => createEmbedder(config), {
|
|
67
|
-
message: /Unknown embedding provider/,
|
|
68
|
-
});
|
|
69
|
-
});
|
|
70
|
-
});
|
|
71
|
-
//# sourceMappingURL=factory.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"factory.test.js","sourceRoot":"","sources":["../../../src/__tests__/embedder/factory.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAE3D,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEtD,SAAS,UAAU,CAAC,SAA6B;IAC/C,OAAO;QACL,GAAG,cAAc;QACjB,GAAG,SAAS;QACZ,SAAS,EAAE;YACT,GAAG,cAAc,CAAC,SAAS;YAC3B,GAAG,SAAS,CAAC,SAAS;SACvB;KACF,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,QAAQ,GAAG,cAAc,CAAC,cAAc,CAAC,CAAC;QAChD,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,MAAM,GAAG,UAAU,CAAC;YACxB,SAAS,EAAE;gBACT,QAAQ,EAAE,QAAQ;gBAClB,OAAO,EAAE,2BAA2B;gBACpC,KAAK,EAAE,cAAc;gBACrB,MAAM,EAAE,SAAS;aAClB;SACF,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,MAAM,GAAG,UAAU,CAAC;YACxB,SAAS,EAAE;gBACT,QAAQ,EAAE,QAAQ;gBAClB,OAAO,EAAE,2BAA2B;gBACpC,KAAK,EAAE,wBAAwB;gBAC/B,MAAM,EAAE,aAAa;aACtB;SACF,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,MAAM,GAAG,UAAU,CAAC;YACxB,SAAS,EAAE;gBACT,QAAQ,EAAE,QAAQ;gBAClB,OAAO,EAAE,2BAA2B;gBACpC,KAAK,EAAE,wBAAwB;gBAC/B,MAAM,EAAE,SAAS;aAClB;SACF,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE;YAC1C,OAAO,EAAE,oCAAoC;SAC9C,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,MAAM,GAAG,UAAU,CAAC;YACxB,SAAS,EAAE;gBACT,QAAQ,EAAE,SAAqB;gBAC/B,OAAO,EAAE,EAAE;gBACX,KAAK,EAAE,EAAE;gBACT,MAAM,EAAE,SAAS;aAClB;SACF,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE;YAC1C,OAAO,EAAE,4BAA4B;SACtC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|
|
@@ -1,106 +0,0 @@
|
|
|
1
|
-
import { describe, it } from "node:test";
|
|
2
|
-
import assert from "node:assert/strict";
|
|
3
|
-
import { createServer } from "node:http";
|
|
4
|
-
import { OllamaProvider } from "../../embedder/ollama.js";
|
|
5
|
-
describe("OllamaProvider", () => {
|
|
6
|
-
it("name is 'ollama'", () => {
|
|
7
|
-
const p = new OllamaProvider("http://localhost:11434/api", "nomic-embed-text");
|
|
8
|
-
assert.equal(p.name, "ollama");
|
|
9
|
-
});
|
|
10
|
-
it("strips single trailing slash from baseUrl", () => {
|
|
11
|
-
const p = new OllamaProvider("http://localhost:11434/api/", "model");
|
|
12
|
-
assert.ok(p);
|
|
13
|
-
});
|
|
14
|
-
it("strips multiple trailing slashes from baseUrl", () => {
|
|
15
|
-
const p = new OllamaProvider("http://localhost:11434/api///", "model");
|
|
16
|
-
assert.ok(p);
|
|
17
|
-
});
|
|
18
|
-
it("preserves baseUrl without trailing slash", () => {
|
|
19
|
-
const p = new OllamaProvider("http://localhost:11434/api", "model");
|
|
20
|
-
assert.ok(p);
|
|
21
|
-
});
|
|
22
|
-
it("stores apiKey when provided", () => {
|
|
23
|
-
const p = new OllamaProvider("http://localhost:11434/api", "model", "my-api-key");
|
|
24
|
-
assert.ok(p);
|
|
25
|
-
});
|
|
26
|
-
it("apiKey defaults to undefined when not provided", () => {
|
|
27
|
-
const p = new OllamaProvider("http://localhost:11434/api", "model");
|
|
28
|
-
assert.ok(p);
|
|
29
|
-
});
|
|
30
|
-
it("embeds single text with correct API format", async () => {
|
|
31
|
-
// Test that the embed method exists and is callable.
|
|
32
|
-
// Actual fetch calls are not made in unit tests.
|
|
33
|
-
const p = new OllamaProvider("http://localhost:11434/api", "embeddinggemma");
|
|
34
|
-
assert.equal(typeof p.embed, "function");
|
|
35
|
-
assert.equal(p.embed.length, 1); // expects texts array parameter
|
|
36
|
-
});
|
|
37
|
-
it("does not require apiKey for construction", () => {
|
|
38
|
-
const p = new OllamaProvider("http://localhost:11434/api", "embeddinggemma");
|
|
39
|
-
assert.equal(p.name, "ollama");
|
|
40
|
-
});
|
|
41
|
-
it("handles custom baseUrl with port and path", () => {
|
|
42
|
-
const p = new OllamaProvider("http://192.168.1.100:8080/api", "model");
|
|
43
|
-
assert.ok(p);
|
|
44
|
-
});
|
|
45
|
-
it("handles https baseUrl", () => {
|
|
46
|
-
const p = new OllamaProvider("https://ollama.example.com/api", "model");
|
|
47
|
-
assert.ok(p);
|
|
48
|
-
});
|
|
49
|
-
it("aborts slow embedding requests instead of hanging forever", async () => {
|
|
50
|
-
const originalFetch = globalThis.fetch;
|
|
51
|
-
globalThis.fetch = (async (_url, init) => {
|
|
52
|
-
const signal = init?.signal;
|
|
53
|
-
return await new Promise((_resolve, reject) => {
|
|
54
|
-
if (signal) {
|
|
55
|
-
signal.addEventListener("abort", () => {
|
|
56
|
-
const error = new Error("Aborted");
|
|
57
|
-
error.name = "AbortError";
|
|
58
|
-
reject(error);
|
|
59
|
-
}, { once: true });
|
|
60
|
-
}
|
|
61
|
-
});
|
|
62
|
-
});
|
|
63
|
-
try {
|
|
64
|
-
const p = new OllamaProvider("http://localhost:11434/api", "embeddinggemma", undefined, 1);
|
|
65
|
-
await assert.rejects(() => p.embed(["test"]), /timed out after 1ms/);
|
|
66
|
-
}
|
|
67
|
-
finally {
|
|
68
|
-
globalThis.fetch = originalFetch;
|
|
69
|
-
}
|
|
70
|
-
});
|
|
71
|
-
it("uses a direct localhost request instead of fetch", async () => {
|
|
72
|
-
const originalFetch = globalThis.fetch;
|
|
73
|
-
globalThis.fetch = (async () => {
|
|
74
|
-
throw new Error("fetch should not be called for localhost");
|
|
75
|
-
});
|
|
76
|
-
const server = createServer((req, res) => {
|
|
77
|
-
let body = "";
|
|
78
|
-
req.on("data", (chunk) => {
|
|
79
|
-
body += String(chunk);
|
|
80
|
-
});
|
|
81
|
-
req.on("end", () => {
|
|
82
|
-
assert.equal(req.method, "POST");
|
|
83
|
-
assert.equal(req.url, "/api/embed");
|
|
84
|
-
assert.equal(req.headers["content-type"], "application/json");
|
|
85
|
-
assert.match(body, /"input":"hello"/);
|
|
86
|
-
res.writeHead(200, { "Content-Type": "application/json" });
|
|
87
|
-
res.end(JSON.stringify({ embeddings: [[1, 2, 3]] }));
|
|
88
|
-
});
|
|
89
|
-
});
|
|
90
|
-
try {
|
|
91
|
-
await new Promise((resolve) => server.listen(0, "127.0.0.1", resolve));
|
|
92
|
-
const address = server.address();
|
|
93
|
-
if (typeof address !== "object" || address === null) {
|
|
94
|
-
throw new Error("failed to start test server");
|
|
95
|
-
}
|
|
96
|
-
const p = new OllamaProvider(`http://localhost:${address.port}/api`, "embeddinggemma");
|
|
97
|
-
const embeddings = await p.embed(["hello"]);
|
|
98
|
-
assert.deepEqual(embeddings, [[1, 2, 3]]);
|
|
99
|
-
}
|
|
100
|
-
finally {
|
|
101
|
-
await new Promise((resolve) => server.close(() => resolve()));
|
|
102
|
-
globalThis.fetch = originalFetch;
|
|
103
|
-
}
|
|
104
|
-
});
|
|
105
|
-
});
|
|
106
|
-
//# sourceMappingURL=ollama.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"ollama.test.js","sourceRoot":"","sources":["../../../src/__tests__/embedder/ollama.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAE1D,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAC1B,MAAM,CAAC,GAAG,IAAI,cAAc,CAAC,4BAA4B,EAAE,kBAAkB,CAAC,CAAC;QAC/E,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,CAAC,GAAG,IAAI,cAAc,CAAC,6BAA6B,EAAE,OAAO,CAAC,CAAC;QACrE,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,CAAC,GAAG,IAAI,cAAc,CAAC,+BAA+B,EAAE,OAAO,CAAC,CAAC;QACvE,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,CAAC,GAAG,IAAI,cAAc,CAAC,4BAA4B,EAAE,OAAO,CAAC,CAAC;QACpE,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,CAAC,GAAG,IAAI,cAAc,CAAC,4BAA4B,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;QAClF,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,CAAC,GAAG,IAAI,cAAc,CAAC,4BAA4B,EAAE,OAAO,CAAC,CAAC;QACpE,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,qDAAqD;QACrD,iDAAiD;QACjD,MAAM,CAAC,GAAG,IAAI,cAAc,CAAC,4BAA4B,EAAE,gBAAgB,CAAC,CAAC;QAC7E,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,gCAAgC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,CAAC,GAAG,IAAI,cAAc,CAAC,4BAA4B,EAAE,gBAAgB,CAAC,CAAC;QAC7E,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,CAAC,GAAG,IAAI,cAAc,CAAC,+BAA+B,EAAE,OAAO,CAAC,CAAC;QACvE,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;QAC/B,MAAM,CAAC,GAAG,IAAI,cAAc,CAAC,gCAAgC,EAAE,OAAO,CAAC,CAAC;QACxE,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC;QAEvC,UAAU,CAAC,KAAK,GAAG,CAAC,KAAK,EAAE,IAAkB,EAAE,IAAkB,EAAE,EAAE;YACnE,MAAM,MAAM,GAAG,IAAI,EAAE,MAAM,CAAC;YAC5B,OAAO,MAAM,IAAI,OAAO,CAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,EAAE;gBACtD,IAAI,MAAM,EAAE,CAAC;oBACX,MAAM,CAAC,gBAAgB,CACrB,OAAO,EACP,GAAG,EAAE;wBACH,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;wBACnC,KAAK,CAAC,IAAI,GAAG,YAAY,CAAC;wBAC1B,MAAM,CAAC,KAAK,CAAC,CAAC;oBAChB,CAAC,EACD,EAAE,IAAI,EAAE,IAAI,EAAE,CACf,CAAC;gBACJ,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAiB,CAAC;QAEnB,IAAI,CAAC;YACH,MAAM,CAAC,GAAG,IAAI,cAAc,CAC1B,4BAA4B,EAC5B,gBAAgB,EAChB,SAAS,EACT,CAAC,CACF,CAAC;YAEF,MAAM,MAAM,CAAC,OAAO,CAClB,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,EACvB,qBAAqB,CACtB,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,UAAU,CAAC,KAAK,GAAG,aAAa,CAAC;QACnC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC;QACvC,UAAU,CAAC,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;YAC7B,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC9D,CAAC,CAAiB,CAAC;QAEnB,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACvC,IAAI,IAAI,GAAG,EAAE,CAAC;YACd,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;gBACvB,IAAI,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC;YACxB,CAAC,CAAC,CAAC;YACH,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBACjB,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBACjC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;gBACpC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,kBAAkB,CAAC,CAAC;gBAC9D,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC;gBAEtC,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;gBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACvD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC;YAC7E,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YACjC,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;gBACpD,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;YACjD,CAAC;YAED,MAAM,CAAC,GAAG,IAAI,cAAc,CAAC,oBAAoB,OAAO,CAAC,IAAI,MAAM,EAAE,gBAAgB,CAAC,CAAC;YACvF,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YAE5C,MAAM,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC;gBAAS,CAAC;YACT,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACpE,UAAU,CAAC,KAAK,GAAG,aAAa,CAAC;QACnC,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|