lance-context 1.25.0 → 1.26.0
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/dist/__tests__/embeddings/factory.test.js +48 -9
- package/dist/__tests__/embeddings/factory.test.js.map +1 -1
- package/dist/__tests__/embeddings/gemini.test.d.ts +2 -0
- package/dist/__tests__/embeddings/gemini.test.d.ts.map +1 -0
- package/dist/__tests__/embeddings/gemini.test.js +178 -0
- package/dist/__tests__/embeddings/gemini.test.js.map +1 -0
- package/dist/__tests__/embeddings/jina.test.js +5 -12
- package/dist/__tests__/embeddings/jina.test.js.map +1 -1
- package/dist/__tests__/mocks/fetch.mock.d.ts +8 -0
- package/dist/__tests__/mocks/fetch.mock.d.ts.map +1 -1
- package/dist/__tests__/mocks/fetch.mock.js +24 -0
- package/dist/__tests__/mocks/fetch.mock.js.map +1 -1
- package/dist/config.d.ts +4 -2
- package/dist/config.d.ts.map +1 -1
- package/dist/config.js +19 -5
- package/dist/config.js.map +1 -1
- package/dist/dashboard/ui.d.ts.map +1 -1
- package/dist/dashboard/ui.js +55 -21
- package/dist/dashboard/ui.js.map +1 -1
- package/dist/embeddings/gemini.d.ts +28 -0
- package/dist/embeddings/gemini.d.ts.map +1 -0
- package/dist/embeddings/gemini.js +124 -0
- package/dist/embeddings/gemini.js.map +1 -0
- package/dist/embeddings/index.d.ts +8 -9
- package/dist/embeddings/index.d.ts.map +1 -1
- package/dist/embeddings/index.js +45 -34
- package/dist/embeddings/index.js.map +1 -1
- package/dist/embeddings/jina.d.ts +0 -12
- package/dist/embeddings/jina.d.ts.map +1 -1
- package/dist/embeddings/jina.js +4 -16
- package/dist/embeddings/jina.js.map +1 -1
- package/dist/embeddings/types.d.ts +1 -1
- package/dist/embeddings/types.d.ts.map +1 -1
- package/dist/index.js +19 -2
- package/dist/index.js.map +1 -1
- package/package.json +1 -1
|
@@ -1,9 +1,14 @@
|
|
|
1
1
|
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
+
import { createSuccessFetch } from '../mocks/fetch.mock.js';
|
|
2
3
|
import { DEFAULT_OLLAMA_MODEL } from '../../embeddings/ollama.js';
|
|
3
4
|
/** Helper to create a mock /api/tags response with the default model available */
|
|
4
5
|
function createTagsResponseWithDefaultModel() {
|
|
5
6
|
return { models: [{ name: DEFAULT_OLLAMA_MODEL }] };
|
|
6
7
|
}
|
|
8
|
+
/** Helper to create a mock Gemini embedding response */
|
|
9
|
+
function createGeminiEmbeddingResponse() {
|
|
10
|
+
return { embedding: { values: [0.1, 0.2] } };
|
|
11
|
+
}
|
|
7
12
|
// We need to dynamically import createEmbeddingBackend after setting env vars
|
|
8
13
|
async function getCreateEmbeddingBackend() {
|
|
9
14
|
// Reset module cache to pick up new env vars
|
|
@@ -16,6 +21,7 @@ describe('createEmbeddingBackend', () => {
|
|
|
16
21
|
beforeEach(() => {
|
|
17
22
|
// Clear relevant env vars
|
|
18
23
|
delete process.env.JINA_API_KEY;
|
|
24
|
+
delete process.env.GEMINI_API_KEY;
|
|
19
25
|
delete process.env.OLLAMA_URL;
|
|
20
26
|
});
|
|
21
27
|
afterEach(() => {
|
|
@@ -25,8 +31,40 @@ describe('createEmbeddingBackend', () => {
|
|
|
25
31
|
vi.resetModules();
|
|
26
32
|
});
|
|
27
33
|
describe('priority order', () => {
|
|
28
|
-
it('should
|
|
34
|
+
it('should prefer Gemini when GEMINI_API_KEY is set', async () => {
|
|
35
|
+
process.env.GEMINI_API_KEY = 'test-gemini-key';
|
|
36
|
+
const mockFetch = vi.fn().mockResolvedValue({
|
|
37
|
+
ok: true,
|
|
38
|
+
status: 200,
|
|
39
|
+
json: async () => createGeminiEmbeddingResponse(),
|
|
40
|
+
});
|
|
41
|
+
vi.stubGlobal('fetch', mockFetch);
|
|
42
|
+
const createEmbeddingBackend = await getCreateEmbeddingBackend();
|
|
43
|
+
const backend = await createEmbeddingBackend();
|
|
44
|
+
expect(backend.name).toBe('gemini');
|
|
45
|
+
});
|
|
46
|
+
it('should prefer Gemini over Jina when both keys are set', async () => {
|
|
47
|
+
process.env.GEMINI_API_KEY = 'test-gemini-key';
|
|
48
|
+
process.env.JINA_API_KEY = 'test-jina-key';
|
|
29
49
|
const mockFetch = vi.fn().mockResolvedValue({
|
|
50
|
+
ok: true,
|
|
51
|
+
status: 200,
|
|
52
|
+
json: async () => createGeminiEmbeddingResponse(),
|
|
53
|
+
});
|
|
54
|
+
vi.stubGlobal('fetch', mockFetch);
|
|
55
|
+
const createEmbeddingBackend = await getCreateEmbeddingBackend();
|
|
56
|
+
const backend = await createEmbeddingBackend();
|
|
57
|
+
expect(backend.name).toBe('gemini');
|
|
58
|
+
});
|
|
59
|
+
it('should fallback to Jina when Gemini fails', async () => {
|
|
60
|
+
process.env.GEMINI_API_KEY = 'invalid-gemini-key';
|
|
61
|
+
process.env.JINA_API_KEY = 'test-jina-key';
|
|
62
|
+
const mockFetch = vi
|
|
63
|
+
.fn()
|
|
64
|
+
// Gemini fails
|
|
65
|
+
.mockResolvedValueOnce({ ok: false, status: 401, text: async () => 'Unauthorized' })
|
|
66
|
+
// Jina succeeds
|
|
67
|
+
.mockResolvedValue({
|
|
30
68
|
ok: true,
|
|
31
69
|
status: 200,
|
|
32
70
|
json: async () => ({ data: [{ embedding: [0.1] }] }),
|
|
@@ -36,8 +74,8 @@ describe('createEmbeddingBackend', () => {
|
|
|
36
74
|
const backend = await createEmbeddingBackend();
|
|
37
75
|
expect(backend.name).toBe('jina');
|
|
38
76
|
});
|
|
39
|
-
it('should prefer
|
|
40
|
-
process.env.JINA_API_KEY = '
|
|
77
|
+
it('should prefer Jina when only JINA_API_KEY is set', async () => {
|
|
78
|
+
process.env.JINA_API_KEY = 'test-jina-key';
|
|
41
79
|
const mockFetch = vi.fn().mockResolvedValue({
|
|
42
80
|
ok: true,
|
|
43
81
|
status: 200,
|
|
@@ -47,14 +85,9 @@ describe('createEmbeddingBackend', () => {
|
|
|
47
85
|
const createEmbeddingBackend = await getCreateEmbeddingBackend();
|
|
48
86
|
const backend = await createEmbeddingBackend();
|
|
49
87
|
expect(backend.name).toBe('jina');
|
|
50
|
-
// Check that custom key was used in request
|
|
51
|
-
expect(mockFetch).toHaveBeenCalledWith(expect.any(String), expect.objectContaining({
|
|
52
|
-
headers: expect.objectContaining({
|
|
53
|
-
Authorization: 'Bearer custom-jina-key',
|
|
54
|
-
}),
|
|
55
|
-
}));
|
|
56
88
|
});
|
|
57
89
|
it('should fallback to Ollama when Jina fails', async () => {
|
|
90
|
+
process.env.JINA_API_KEY = 'invalid-key';
|
|
58
91
|
const mockFetch = vi
|
|
59
92
|
.fn()
|
|
60
93
|
// Jina fails (called during initialize embed test)
|
|
@@ -70,6 +103,12 @@ describe('createEmbeddingBackend', () => {
|
|
|
70
103
|
const backend = await createEmbeddingBackend();
|
|
71
104
|
expect(backend.name).toBe('ollama');
|
|
72
105
|
});
|
|
106
|
+
it('should use Ollama when no API keys are set', async () => {
|
|
107
|
+
vi.stubGlobal('fetch', createSuccessFetch(createTagsResponseWithDefaultModel()));
|
|
108
|
+
const createEmbeddingBackend = await getCreateEmbeddingBackend();
|
|
109
|
+
const backend = await createEmbeddingBackend();
|
|
110
|
+
expect(backend.name).toBe('ollama');
|
|
111
|
+
});
|
|
73
112
|
});
|
|
74
113
|
describe('error when all backends fail', () => {
|
|
75
114
|
it('should throw when no backend is available', async () => {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"factory.test.js","sourceRoot":"","sources":["../../../src/__tests__/embeddings/factory.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAElE,kFAAkF;AAClF,SAAS,kCAAkC;IACzC,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC,EAAE,CAAC;AACtD,CAAC;AAED,8EAA8E;AAC9E,KAAK,UAAU,yBAAyB;IACtC,6CAA6C;IAC7C,EAAE,CAAC,YAAY,EAAE,CAAC;IAClB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;IACzD,OAAO,MAAM,CAAC,sBAAsB,CAAC;AACvC,CAAC;AAED,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,MAAM,WAAW,GAAG,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAEvC,UAAU,CAAC,GAAG,EAAE;QACd,0BAA0B;QAC1B,OAAO,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;QAChC,OAAO,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,uBAAuB;QACvB,OAAO,CAAC,GAAG,GAAG,EAAE,GAAG,WAAW,EAAE,CAAC;QACjC,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,EAAE,CAAC,YAAY,EAAE,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,
|
|
1
|
+
{"version":3,"file":"factory.test.js","sourceRoot":"","sources":["../../../src/__tests__/embeddings/factory.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAElE,kFAAkF;AAClF,SAAS,kCAAkC;IACzC,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC,EAAE,CAAC;AACtD,CAAC;AAED,wDAAwD;AACxD,SAAS,6BAA6B;IACpC,OAAO,EAAE,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC;AAC/C,CAAC;AAED,8EAA8E;AAC9E,KAAK,UAAU,yBAAyB;IACtC,6CAA6C;IAC7C,EAAE,CAAC,YAAY,EAAE,CAAC;IAClB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;IACzD,OAAO,MAAM,CAAC,sBAAsB,CAAC;AACvC,CAAC;AAED,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,MAAM,WAAW,GAAG,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAEvC,UAAU,CAAC,GAAG,EAAE;QACd,0BAA0B;QAC1B,OAAO,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;QAChC,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAClC,OAAO,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,uBAAuB;QACvB,OAAO,CAAC,GAAG,GAAG,EAAE,GAAG,WAAW,EAAE,CAAC;QACjC,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,EAAE,CAAC,YAAY,EAAE,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,iBAAiB,CAAC;YAE/C,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAC1C,EAAE,EAAE,IAAI;gBACR,MAAM,EAAE,GAAG;gBACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,6BAA6B,EAAE;aAClD,CAAC,CAAC;YACH,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,sBAAsB,GAAG,MAAM,yBAAyB,EAAE,CAAC;YACjE,MAAM,OAAO,GAAG,MAAM,sBAAsB,EAAE,CAAC;YAE/C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,iBAAiB,CAAC;YAC/C,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,eAAe,CAAC;YAE3C,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAC1C,EAAE,EAAE,IAAI;gBACR,MAAM,EAAE,GAAG;gBACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,6BAA6B,EAAE;aAClD,CAAC,CAAC;YACH,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,sBAAsB,GAAG,MAAM,yBAAyB,EAAE,CAAC;YACjE,MAAM,OAAO,GAAG,MAAM,sBAAsB,EAAE,CAAC;YAE/C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,oBAAoB,CAAC;YAClD,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,eAAe,CAAC;YAE3C,MAAM,SAAS,GAAG,EAAE;iBACjB,EAAE,EAAE;gBACL,eAAe;iBACd,qBAAqB,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,cAAc,EAAE,CAAC;gBACpF,gBAAgB;iBACf,iBAAiB,CAAC;gBACjB,EAAE,EAAE,IAAI;gBACR,MAAM,EAAE,GAAG;gBACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;aACrD,CAAC,CAAC;YACL,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,sBAAsB,GAAG,MAAM,yBAAyB,EAAE,CAAC;YACjE,MAAM,OAAO,GAAG,MAAM,sBAAsB,EAAE,CAAC;YAE/C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,eAAe,CAAC;YAE3C,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAC1C,EAAE,EAAE,IAAI;gBACR,MAAM,EAAE,GAAG;gBACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;aACrD,CAAC,CAAC;YACH,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,sBAAsB,GAAG,MAAM,yBAAyB,EAAE,CAAC;YACjE,MAAM,OAAO,GAAG,MAAM,sBAAsB,EAAE,CAAC;YAE/C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,aAAa,CAAC;YAEzC,MAAM,SAAS,GAAG,EAAE;iBACjB,EAAE,EAAE;gBACL,mDAAmD;iBAClD,qBAAqB,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,cAAc,EAAE,CAAC;gBACpF,iDAAiD;iBAChD,iBAAiB,CAAC;gBACjB,EAAE,EAAE,IAAI;gBACR,MAAM,EAAE,GAAG;gBACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,kCAAkC,EAAE;aACvD,CAAC,CAAC;YACL,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,sBAAsB,GAAG,MAAM,yBAAyB,EAAE,CAAC;YACjE,MAAM,OAAO,GAAG,MAAM,sBAAsB,EAAE,CAAC;YAE/C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,kBAAkB,CAAC,kCAAkC,EAAE,CAAC,CAAC,CAAC;YAEjF,MAAM,sBAAsB,GAAG,MAAM,yBAAyB,EAAE,CAAC;YACjE,MAAM,OAAO,GAAG,MAAM,sBAAsB,EAAE,CAAC;YAE/C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;QAC5C,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC,CAAC;YAC7E,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,sBAAsB,GAAG,MAAM,yBAAyB,EAAE,CAAC;YACjE,MAAM,MAAM,CAAC,sBAAsB,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,gCAAgC,CAAC,CAAC;QAC3F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAC1C,EAAE,EAAE,IAAI;gBACR,MAAM,EAAE,GAAG;gBACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;aACrD,CAAC,CAAC;YACH,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,sBAAsB,GAAG,MAAM,yBAAyB,EAAE,CAAC;YACjE,MAAM,OAAO,GAAG,MAAM,sBAAsB,CAAC,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC,CAAC;YAE5E,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,OAAO,CAAC,GAAG,CAAC,UAAU,GAAG,4BAA4B,CAAC;YAEtD,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,EAAE,GAAW,EAAE,EAAE;gBACjE,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;gBACvC,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,kCAAkC,EAAE,EAAE,CAAC;YAC3F,CAAC,CAAC,CAAC;YACH,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,sBAAsB,GAAG,MAAM,yBAAyB,EAAE,CAAC;YACjE,MAAM,OAAO,GAAG,MAAM,sBAAsB,EAAE,CAAC;YAE/C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,OAAO,CAAC,GAAG,CAAC,UAAU,GAAG,yBAAyB,CAAC;YAEnD,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,EAAE,GAAW,EAAE,EAAE;gBACjE,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;gBACvC,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,kCAAkC,EAAE,EAAE,CAAC;YAC3F,CAAC,CAAC,CAAC;YACH,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,sBAAsB,GAAG,MAAM,yBAAyB,EAAE,CAAC;YACjE,MAAM,OAAO,GAAG,MAAM,sBAAsB,CAAC,EAAE,OAAO,EAAE,4BAA4B,EAAE,CAAC,CAAC;YAExF,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/embeddings/gemini.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,178 @@
|
|
|
1
|
+
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
+
import { GeminiBackend } from '../../embeddings/gemini.js';
|
|
3
|
+
import { createGeminiEmbeddingResponse, createGeminiBatchEmbeddingResponse, createErrorFetch, } from '../mocks/fetch.mock.js';
|
|
4
|
+
describe('GeminiBackend', () => {
|
|
5
|
+
beforeEach(() => {
|
|
6
|
+
vi.stubGlobal('fetch', vi.fn());
|
|
7
|
+
});
|
|
8
|
+
afterEach(() => {
|
|
9
|
+
vi.unstubAllGlobals();
|
|
10
|
+
});
|
|
11
|
+
describe('constructor', () => {
|
|
12
|
+
it('should throw if API key is not provided', () => {
|
|
13
|
+
expect(() => new GeminiBackend({ backend: 'gemini' })).toThrow('Gemini API key is required');
|
|
14
|
+
});
|
|
15
|
+
it('should accept API key in config', () => {
|
|
16
|
+
const backend = new GeminiBackend({ backend: 'gemini', apiKey: 'test-key' });
|
|
17
|
+
expect(backend.name).toBe('gemini');
|
|
18
|
+
});
|
|
19
|
+
it('should use default model gemini-embedding-001', () => {
|
|
20
|
+
const backend = new GeminiBackend({ backend: 'gemini', apiKey: 'test-key' });
|
|
21
|
+
expect(backend.getModel()).toBe('gemini-embedding-001');
|
|
22
|
+
});
|
|
23
|
+
it('should return 768 dimensions by default', () => {
|
|
24
|
+
const backend = new GeminiBackend({ backend: 'gemini', apiKey: 'test-key' });
|
|
25
|
+
expect(backend.getDimensions()).toBe(768);
|
|
26
|
+
});
|
|
27
|
+
it('should accept custom model', () => {
|
|
28
|
+
const backend = new GeminiBackend({
|
|
29
|
+
backend: 'gemini',
|
|
30
|
+
apiKey: 'test-key',
|
|
31
|
+
model: 'custom-model',
|
|
32
|
+
});
|
|
33
|
+
expect(backend.getModel()).toBe('custom-model');
|
|
34
|
+
});
|
|
35
|
+
});
|
|
36
|
+
describe('initialize', () => {
|
|
37
|
+
it('should test API key by calling embed', async () => {
|
|
38
|
+
const mockFetch = vi.fn().mockResolvedValue(createGeminiEmbeddingResponse([0.1, 0.2]));
|
|
39
|
+
vi.stubGlobal('fetch', mockFetch);
|
|
40
|
+
const backend = new GeminiBackend({ backend: 'gemini', apiKey: 'test-key' });
|
|
41
|
+
await backend.initialize();
|
|
42
|
+
expect(mockFetch).toHaveBeenCalledWith('https://generativelanguage.googleapis.com/v1beta/models/gemini-embedding-001:embedContent', expect.objectContaining({
|
|
43
|
+
method: 'POST',
|
|
44
|
+
headers: expect.objectContaining({
|
|
45
|
+
'x-goog-api-key': 'test-key',
|
|
46
|
+
}),
|
|
47
|
+
}));
|
|
48
|
+
});
|
|
49
|
+
it('should throw on initialization failure', async () => {
|
|
50
|
+
const mockFetch = createErrorFetch(401, 'Unauthorized');
|
|
51
|
+
vi.stubGlobal('fetch', mockFetch);
|
|
52
|
+
const backend = new GeminiBackend({ backend: 'gemini', apiKey: 'invalid-key' });
|
|
53
|
+
await expect(backend.initialize()).rejects.toThrow('Failed to initialize Gemini backend');
|
|
54
|
+
});
|
|
55
|
+
});
|
|
56
|
+
describe('embed', () => {
|
|
57
|
+
it('should send correct request format', async () => {
|
|
58
|
+
const mockFetch = vi.fn().mockResolvedValue(createGeminiEmbeddingResponse([0.1, 0.2, 0.3]));
|
|
59
|
+
vi.stubGlobal('fetch', mockFetch);
|
|
60
|
+
const backend = new GeminiBackend({ backend: 'gemini', apiKey: 'test-key' });
|
|
61
|
+
await backend.embed('test text');
|
|
62
|
+
expect(mockFetch).toHaveBeenCalledWith('https://generativelanguage.googleapis.com/v1beta/models/gemini-embedding-001:embedContent', expect.objectContaining({
|
|
63
|
+
method: 'POST',
|
|
64
|
+
headers: {
|
|
65
|
+
'Content-Type': 'application/json',
|
|
66
|
+
'x-goog-api-key': 'test-key',
|
|
67
|
+
},
|
|
68
|
+
body: JSON.stringify({
|
|
69
|
+
model: 'models/gemini-embedding-001',
|
|
70
|
+
content: {
|
|
71
|
+
parts: [{ text: 'test text' }],
|
|
72
|
+
},
|
|
73
|
+
outputDimensionality: 768,
|
|
74
|
+
}),
|
|
75
|
+
}));
|
|
76
|
+
});
|
|
77
|
+
it('should return embedding from response', async () => {
|
|
78
|
+
const embedding = [0.1, 0.2, 0.3];
|
|
79
|
+
const mockFetch = vi.fn().mockResolvedValue(createGeminiEmbeddingResponse(embedding));
|
|
80
|
+
vi.stubGlobal('fetch', mockFetch);
|
|
81
|
+
const backend = new GeminiBackend({ backend: 'gemini', apiKey: 'test-key' });
|
|
82
|
+
const result = await backend.embed('test text');
|
|
83
|
+
expect(result).toEqual(embedding);
|
|
84
|
+
});
|
|
85
|
+
it('should throw on API error', async () => {
|
|
86
|
+
const mockFetch = vi.fn().mockResolvedValue({
|
|
87
|
+
ok: false,
|
|
88
|
+
status: 400,
|
|
89
|
+
text: async () => 'Bad request',
|
|
90
|
+
});
|
|
91
|
+
vi.stubGlobal('fetch', mockFetch);
|
|
92
|
+
const backend = new GeminiBackend({ backend: 'gemini', apiKey: 'test-key' });
|
|
93
|
+
await expect(backend.embed('test')).rejects.toThrow('Gemini API error: 400 - Bad request');
|
|
94
|
+
});
|
|
95
|
+
});
|
|
96
|
+
describe('embedBatch', () => {
|
|
97
|
+
it('should use batchEmbedContents endpoint', async () => {
|
|
98
|
+
const mockFetch = vi
|
|
99
|
+
.fn()
|
|
100
|
+
.mockResolvedValue(createGeminiBatchEmbeddingResponse([[0.1], [0.2], [0.3]]));
|
|
101
|
+
vi.stubGlobal('fetch', mockFetch);
|
|
102
|
+
const backend = new GeminiBackend({ backend: 'gemini', apiKey: 'test-key' });
|
|
103
|
+
await backend.embedBatch(['text1', 'text2', 'text3']);
|
|
104
|
+
expect(mockFetch).toHaveBeenCalledTimes(1);
|
|
105
|
+
expect(mockFetch).toHaveBeenCalledWith('https://generativelanguage.googleapis.com/v1beta/models/gemini-embedding-001:batchEmbedContents', expect.objectContaining({
|
|
106
|
+
body: JSON.stringify({
|
|
107
|
+
requests: [
|
|
108
|
+
{
|
|
109
|
+
model: 'models/gemini-embedding-001',
|
|
110
|
+
content: { parts: [{ text: 'text1' }] },
|
|
111
|
+
outputDimensionality: 768,
|
|
112
|
+
},
|
|
113
|
+
{
|
|
114
|
+
model: 'models/gemini-embedding-001',
|
|
115
|
+
content: { parts: [{ text: 'text2' }] },
|
|
116
|
+
outputDimensionality: 768,
|
|
117
|
+
},
|
|
118
|
+
{
|
|
119
|
+
model: 'models/gemini-embedding-001',
|
|
120
|
+
content: { parts: [{ text: 'text3' }] },
|
|
121
|
+
outputDimensionality: 768,
|
|
122
|
+
},
|
|
123
|
+
],
|
|
124
|
+
}),
|
|
125
|
+
}));
|
|
126
|
+
});
|
|
127
|
+
it('should return embeddings in order', async () => {
|
|
128
|
+
const embeddings = [[0.1], [0.2], [0.3]];
|
|
129
|
+
const mockFetch = vi.fn().mockResolvedValue(createGeminiBatchEmbeddingResponse(embeddings));
|
|
130
|
+
vi.stubGlobal('fetch', mockFetch);
|
|
131
|
+
const backend = new GeminiBackend({ backend: 'gemini', apiKey: 'test-key' });
|
|
132
|
+
const result = await backend.embedBatch(['text1', 'text2', 'text3']);
|
|
133
|
+
expect(result).toEqual(embeddings);
|
|
134
|
+
});
|
|
135
|
+
it('should throw on API error', async () => {
|
|
136
|
+
// Use 400 (non-retryable) instead of 500 (retryable)
|
|
137
|
+
const mockFetch = vi.fn().mockResolvedValue({
|
|
138
|
+
ok: false,
|
|
139
|
+
status: 400,
|
|
140
|
+
text: async () => 'Bad request',
|
|
141
|
+
});
|
|
142
|
+
vi.stubGlobal('fetch', mockFetch);
|
|
143
|
+
const backend = new GeminiBackend({ backend: 'gemini', apiKey: 'test-key' });
|
|
144
|
+
await expect(backend.embedBatch(['text'])).rejects.toThrow('Gemini API error: 400 - Bad request');
|
|
145
|
+
});
|
|
146
|
+
it('should chunk large batches and make multiple requests', async () => {
|
|
147
|
+
const mockFetch = vi
|
|
148
|
+
.fn()
|
|
149
|
+
.mockImplementation(() => createGeminiBatchEmbeddingResponse([[0.1], [0.2]]));
|
|
150
|
+
vi.stubGlobal('fetch', mockFetch);
|
|
151
|
+
// Create backend with batch size of 2
|
|
152
|
+
const backend = new GeminiBackend({ backend: 'gemini', apiKey: 'test-key', batchSize: 2 });
|
|
153
|
+
const result = await backend.embedBatch(['text1', 'text2', 'text3', 'text4', 'text5']);
|
|
154
|
+
// Should make 3 requests: [text1, text2], [text3, text4], [text5]
|
|
155
|
+
expect(mockFetch).toHaveBeenCalledTimes(3);
|
|
156
|
+
// Should return all 5 embeddings (though mock returns 2 per call, so we get 6)
|
|
157
|
+
expect(result).toHaveLength(6);
|
|
158
|
+
});
|
|
159
|
+
it('should not chunk when batch is smaller than batch size', async () => {
|
|
160
|
+
const mockFetch = vi
|
|
161
|
+
.fn()
|
|
162
|
+
.mockResolvedValue(createGeminiBatchEmbeddingResponse([[0.1], [0.2], [0.3]]));
|
|
163
|
+
vi.stubGlobal('fetch', mockFetch);
|
|
164
|
+
// Create backend with batch size of 100 (larger than input)
|
|
165
|
+
const backend = new GeminiBackend({ backend: 'gemini', apiKey: 'test-key', batchSize: 100 });
|
|
166
|
+
await backend.embedBatch(['text1', 'text2', 'text3']);
|
|
167
|
+
// Should make only 1 request
|
|
168
|
+
expect(mockFetch).toHaveBeenCalledTimes(1);
|
|
169
|
+
});
|
|
170
|
+
});
|
|
171
|
+
describe('getDimensions', () => {
|
|
172
|
+
it('should return 768 for default configuration', () => {
|
|
173
|
+
const backend = new GeminiBackend({ backend: 'gemini', apiKey: 'test-key' });
|
|
174
|
+
expect(backend.getDimensions()).toBe(768);
|
|
175
|
+
});
|
|
176
|
+
});
|
|
177
|
+
});
|
|
178
|
+
//# sourceMappingURL=gemini.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini.test.js","sourceRoot":"","sources":["../../../src/__tests__/embeddings/gemini.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAC3D,OAAO,EACL,6BAA6B,EAC7B,kCAAkC,EAClC,gBAAgB,GACjB,MAAM,wBAAwB,CAAC;AAEhC,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,4BAA4B,CAAC,CAAC;QAC/F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAC7E,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAC7E,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAC7E,MAAM,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC;gBAChC,OAAO,EAAE,QAAQ;gBACjB,MAAM,EAAE,UAAU;gBAClB,KAAK,EAAE,cAAc;aACtB,CAAC,CAAC;YACH,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,6BAA6B,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;YACvF,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAC7E,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;YAE3B,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,2FAA2F,EAC3F,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,gBAAgB,EAAE,UAAU;iBAC7B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,SAAS,GAAG,gBAAgB,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;YACxD,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;YAChF,MAAM,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAC;QAC5F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,6BAA6B,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;YAC5F,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAC7E,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEjC,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,2FAA2F,EAC3F,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,gBAAgB,EAAE,UAAU;iBAC7B;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,6BAA6B;oBACpC,OAAO,EAAE;wBACP,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;qBAC/B;oBACD,oBAAoB,EAAE,GAAG;iBAC1B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,SAAS,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAClC,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,6BAA6B,CAAC,SAAS,CAAC,CAAC,CAAC;YACtF,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAC7E,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEhD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAC1C,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,GAAG;gBACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,aAAa;aAChC,CAAC,CAAC;YACH,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAC7E,MAAM,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAC;QAC7F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,SAAS,GAAG,EAAE;iBACjB,EAAE,EAAE;iBACJ,iBAAiB,CAAC,kCAAkC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAChF,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAC7E,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAEtD,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3C,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,iGAAiG,EACjG,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,QAAQ,EAAE;wBACR;4BACE,KAAK,EAAE,6BAA6B;4BACpC,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE;4BACvC,oBAAoB,EAAE,GAAG;yBAC1B;wBACD;4BACE,KAAK,EAAE,6BAA6B;4BACpC,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE;4BACvC,oBAAoB,EAAE,GAAG;yBAC1B;wBACD;4BACE,KAAK,EAAE,6BAA6B;4BACpC,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE;4BACvC,oBAAoB,EAAE,GAAG;yBAC1B;qBACF;iBACF,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,UAAU,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;YACzC,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,kCAAkC,CAAC,UAAU,CAAC,CAAC,CAAC;YAC5F,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAC7E,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAErE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,qDAAqD;YACrD,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAC1C,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,GAAG;gBACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,aAAa;aAChC,CAAC,CAAC;YACH,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAC7E,MAAM,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACxD,qCAAqC,CACtC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,SAAS,GAAG,EAAE;iBACjB,EAAE,EAAE;iBACJ,kBAAkB,CAAC,GAAG,EAAE,CAAC,kCAAkC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAChF,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,sCAAsC;YACtC,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;YAC3F,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAEvF,kEAAkE;YAClE,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3C,+EAA+E;YAC/E,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,SAAS,GAAG,EAAE;iBACjB,EAAE,EAAE;iBACJ,iBAAiB,CAAC,kCAAkC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAChF,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,4DAA4D;YAC5D,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;YAC7F,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAEtD,6BAA6B;YAC7B,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAC7E,MAAM,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
-
import { JinaBackend
|
|
2
|
+
import { JinaBackend } from '../../embeddings/jina.js';
|
|
3
3
|
import { createJinaEmbeddingResponse, createErrorFetch } from '../mocks/fetch.mock.js';
|
|
4
4
|
describe('JinaBackend', () => {
|
|
5
5
|
beforeEach(() => {
|
|
@@ -9,15 +9,12 @@ describe('JinaBackend', () => {
|
|
|
9
9
|
vi.unstubAllGlobals();
|
|
10
10
|
});
|
|
11
11
|
describe('constructor', () => {
|
|
12
|
-
it('should
|
|
13
|
-
|
|
14
|
-
expect(backend.name).toBe('jina');
|
|
15
|
-
expect(backend.isUsingDefaultKey()).toBe(true);
|
|
12
|
+
it('should throw if API key is not provided', () => {
|
|
13
|
+
expect(() => new JinaBackend({ backend: 'jina' })).toThrow('Jina API key is required');
|
|
16
14
|
});
|
|
17
|
-
it('should accept
|
|
18
|
-
const backend = new JinaBackend({ backend: 'jina', apiKey: '
|
|
15
|
+
it('should accept API key in config', () => {
|
|
16
|
+
const backend = new JinaBackend({ backend: 'jina', apiKey: 'test-key' });
|
|
19
17
|
expect(backend.name).toBe('jina');
|
|
20
|
-
expect(backend.isUsingDefaultKey()).toBe(false);
|
|
21
18
|
});
|
|
22
19
|
it('should use default model jina-embeddings-v3', () => {
|
|
23
20
|
const backend = new JinaBackend({ backend: 'jina', apiKey: 'test-key' });
|
|
@@ -31,10 +28,6 @@ describe('JinaBackend', () => {
|
|
|
31
28
|
});
|
|
32
29
|
expect(backend.name).toBe('jina');
|
|
33
30
|
});
|
|
34
|
-
it('should export the default API key constant', () => {
|
|
35
|
-
expect(DEFAULT_JINA_API_KEY).toBeDefined();
|
|
36
|
-
expect(typeof DEFAULT_JINA_API_KEY).toBe('string');
|
|
37
|
-
});
|
|
38
31
|
});
|
|
39
32
|
describe('initialize', () => {
|
|
40
33
|
it('should test API key by calling embed', async () => {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"jina.test.js","sourceRoot":"","sources":["../../../src/__tests__/embeddings/jina.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,WAAW,EAAE,
|
|
1
|
+
{"version":3,"file":"jina.test.js","sourceRoot":"","sources":["../../../src/__tests__/embeddings/jina.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AACvD,OAAO,EAAE,2BAA2B,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAEvF,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,0BAA0B,CAAC,CAAC;QACzF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YACzE,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YACzE,MAAM,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC;gBAC9B,OAAO,EAAE,MAAM;gBACf,MAAM,EAAE,UAAU;gBAClB,KAAK,EAAE,cAAc;aACtB,CAAC,CAAC;YACH,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACvF,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YACzE,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;YAE3B,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,mCAAmC,EACnC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,aAAa,EAAE,iBAAiB;iBACjC,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,SAAS,GAAG,gBAAgB,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;YACxD,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;YAC5E,MAAM,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,mCAAmC,CAAC,CAAC;QAC1F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5F,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YACzE,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEjC,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,mCAAmC,EACnC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,iBAAiB;iBACjC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,oBAAoB;oBAC3B,KAAK,EAAE,CAAC,WAAW,CAAC;iBACrB,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,SAAS,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAClC,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YACtF,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YACzE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEhD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAC1C,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,GAAG;gBACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,aAAa;aAChC,CAAC,CAAC;YACH,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YACzE,MAAM,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,mCAAmC,CAAC,CAAC;QAC3F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,SAAS,GAAG,EAAE;iBACjB,EAAE,EAAE;iBACJ,iBAAiB,CAAC,2BAA2B,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACzE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YACzE,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAEtD,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3C,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,oBAAoB;oBAC3B,KAAK,EAAE,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC;iBACnC,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,UAAU,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;YACzC,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,UAAU,CAAC,CAAC,CAAC;YACrF,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YACzE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAErE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,qDAAqD;YACrD,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAC1C,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,GAAG;gBACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,aAAa;aAChC,CAAC,CAAC;YACH,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YACzE,MAAM,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACxD,mCAAmC,CACpC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,SAAS,GAAG,EAAE;iBACjB,EAAE,EAAE;iBACJ,kBAAkB,CAAC,GAAG,EAAE,CAAC,2BAA2B,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACzE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,sCAAsC;YACtC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;YACvF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAEvF,kEAAkE;YAClE,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3C,+EAA+E;YAC/E,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,SAAS,GAAG,EAAE;iBACjB,EAAE,EAAE;iBACJ,iBAAiB,CAAC,2BAA2B,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACzE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,4DAA4D;YAC5D,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;YACzF,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAEtD,6BAA6B;YAC7B,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YACzE,MAAM,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -41,5 +41,13 @@ export declare function createOllamaEmbeddingResponse(embedding: number[]): Mock
|
|
|
41
41
|
* Creates Ollama-style batch embedding response (/api/embed)
|
|
42
42
|
*/
|
|
43
43
|
export declare function createOllamaBatchEmbeddingResponse(embeddings: number[][]): MockFetchResponse;
|
|
44
|
+
/**
|
|
45
|
+
* Creates Gemini-style single embedding response (embedContent)
|
|
46
|
+
*/
|
|
47
|
+
export declare function createGeminiEmbeddingResponse(embedding: number[]): MockFetchResponse;
|
|
48
|
+
/**
|
|
49
|
+
* Creates Gemini-style batch embedding response (batchEmbedContents)
|
|
50
|
+
*/
|
|
51
|
+
export declare function createGeminiBatchEmbeddingResponse(embeddings: number[][]): MockFetchResponse;
|
|
44
52
|
export {};
|
|
45
53
|
//# sourceMappingURL=fetch.mock.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"fetch.mock.d.ts","sourceRoot":"","sources":["../../../src/__tests__/mocks/fetch.mock.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,OAAO,CAAC;IACZ,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE;QAAE,GAAG,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,MAAM,GAAG,IAAI,CAAA;KAAE,CAAC;IACnD,IAAI,CAAC,EAAE,MAAM,OAAO,CAAC,OAAO,CAAC,CAAC;IAC9B,IAAI,CAAC,EAAE,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC;CAC9B;AAED,KAAK,YAAY,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,iBAAiB,CAAC,CAAC;AAEvF;;GAEG;AACH,wBAAgB,eAAe,CAAC,OAAO,EAAE,YAAY,0DAEpD;AAOD;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,OAAO,GAAG,UAAU,CAAC,OAAO,eAAe,CAAC,CAQpF;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAC9B,MAAM,EAAE,MAAM,EACd,UAAU,GAAE,MAAgB,EAC5B,IAAI,GAAE,MAAW,GAChB,UAAU,CAAC,OAAO,eAAe,CAAC,CASpC;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CACrC,OAAO,GAAE,MAAuB,GAC/B,UAAU,CAAC,OAAO,eAAe,CAAC,CAEpC;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CACnC,SAAS,EAAE,iBAAiB,EAAE,GAC7B,UAAU,CAAC,OAAO,eAAe,CAAC,CAOpC;AAED;;GAEG;AACH,wBAAgB,2BAA2B,CAAC,UAAU,EAAE,MAAM,EAAE,EAAE,GAAG,iBAAiB,CAarF;AAED;;GAEG;AACH,wBAAgB,6BAA6B,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,iBAAiB,CAQpF;AAED;;GAEG;AACH,wBAAgB,kCAAkC,CAAC,UAAU,EAAE,MAAM,EAAE,EAAE,GAAG,iBAAiB,CAQ5F"}
|
|
1
|
+
{"version":3,"file":"fetch.mock.d.ts","sourceRoot":"","sources":["../../../src/__tests__/mocks/fetch.mock.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,OAAO,CAAC;IACZ,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE;QAAE,GAAG,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,MAAM,GAAG,IAAI,CAAA;KAAE,CAAC;IACnD,IAAI,CAAC,EAAE,MAAM,OAAO,CAAC,OAAO,CAAC,CAAC;IAC9B,IAAI,CAAC,EAAE,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC;CAC9B;AAED,KAAK,YAAY,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,iBAAiB,CAAC,CAAC;AAEvF;;GAEG;AACH,wBAAgB,eAAe,CAAC,OAAO,EAAE,YAAY,0DAEpD;AAOD;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,OAAO,GAAG,UAAU,CAAC,OAAO,eAAe,CAAC,CAQpF;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAC9B,MAAM,EAAE,MAAM,EACd,UAAU,GAAE,MAAgB,EAC5B,IAAI,GAAE,MAAW,GAChB,UAAU,CAAC,OAAO,eAAe,CAAC,CASpC;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CACrC,OAAO,GAAE,MAAuB,GAC/B,UAAU,CAAC,OAAO,eAAe,CAAC,CAEpC;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CACnC,SAAS,EAAE,iBAAiB,EAAE,GAC7B,UAAU,CAAC,OAAO,eAAe,CAAC,CAOpC;AAED;;GAEG;AACH,wBAAgB,2BAA2B,CAAC,UAAU,EAAE,MAAM,EAAE,EAAE,GAAG,iBAAiB,CAarF;AAED;;GAEG;AACH,wBAAgB,6BAA6B,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,iBAAiB,CAQpF;AAED;;GAEG;AACH,wBAAgB,kCAAkC,CAAC,UAAU,EAAE,MAAM,EAAE,EAAE,GAAG,iBAAiB,CAQ5F;AAED;;GAEG;AACH,wBAAgB,6BAA6B,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,iBAAiB,CAQpF;AAED;;GAEG;AACH,wBAAgB,kCAAkC,CAAC,UAAU,EAAE,MAAM,EAAE,EAAE,GAAG,iBAAiB,CAQ5F"}
|
|
@@ -91,4 +91,28 @@ export function createOllamaBatchEmbeddingResponse(embeddings) {
|
|
|
91
91
|
text: async () => JSON.stringify({ embeddings }),
|
|
92
92
|
};
|
|
93
93
|
}
|
|
94
|
+
/**
|
|
95
|
+
* Creates Gemini-style single embedding response (embedContent)
|
|
96
|
+
*/
|
|
97
|
+
export function createGeminiEmbeddingResponse(embedding) {
|
|
98
|
+
return {
|
|
99
|
+
ok: true,
|
|
100
|
+
status: 200,
|
|
101
|
+
headers: createMockHeaders(),
|
|
102
|
+
json: async () => ({ embedding: { values: embedding } }),
|
|
103
|
+
text: async () => JSON.stringify({ embedding: { values: embedding } }),
|
|
104
|
+
};
|
|
105
|
+
}
|
|
106
|
+
/**
|
|
107
|
+
* Creates Gemini-style batch embedding response (batchEmbedContents)
|
|
108
|
+
*/
|
|
109
|
+
export function createGeminiBatchEmbeddingResponse(embeddings) {
|
|
110
|
+
return {
|
|
111
|
+
ok: true,
|
|
112
|
+
status: 200,
|
|
113
|
+
headers: createMockHeaders(),
|
|
114
|
+
json: async () => ({ embeddings: embeddings.map((values) => ({ values })) }),
|
|
115
|
+
text: async () => JSON.stringify({ embeddings: embeddings.map((values) => ({ values })) }),
|
|
116
|
+
};
|
|
117
|
+
}
|
|
94
118
|
//# sourceMappingURL=fetch.mock.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"fetch.mock.js","sourceRoot":"","sources":["../../../src/__tests__/mocks/fetch.mock.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAa5B;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,OAAqB;IACnD,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;AAC7C,CAAC;AAED,oCAAoC;AACpC,SAAS,iBAAiB;IACxB,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC;AAC7B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,IAAa;IAC9C,OAAO,eAAe,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;QAClC,EAAE,EAAE,IAAI;QACR,MAAM,EAAE,GAAG;QACX,OAAO,EAAE,iBAAiB,EAAE;QAC5B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI;QACtB,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;KACvC,CAAC,CAAC,CAAC;AACN,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAC9B,MAAc,EACd,aAAqB,OAAO,EAC5B,OAAe,EAAE;IAEjB,OAAO,eAAe,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;QAClC,EAAE,EAAE,KAAK;QACT,MAAM;QACN,UAAU;QACV,OAAO,EAAE,iBAAiB,EAAE;QAC5B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QACnC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI;KACvB,CAAC,CAAC,CAAC;AACN,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB,CACrC,UAAkB,cAAc;IAEhC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;AACvD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CACnC,SAA8B;IAE9B,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;QAC3C,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QACtE,SAAS,EAAE,CAAC;QACZ,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,2BAA2B,CAAC,UAAsB;IAChE,OAAO;QACL,EAAE,EAAE,IAAI;QACR,MAAM,EAAE,GAAG;QACX,OAAO,EAAE,iBAAiB,EAAE;QAC5B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;YACjB,IAAI,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;SACrD,CAAC;QACF,IAAI,EAAE,KAAK,IAAI,EAAE,CACf,IAAI,CAAC,SAAS,CAAC;YACb,IAAI,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;SACrD,CAAC;KACL,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,6BAA6B,CAAC,SAAmB;IAC/D,OAAO;QACL,EAAE,EAAE,IAAI;QACR,MAAM,EAAE,GAAG;QACX,OAAO,EAAE,iBAAiB,EAAE;QAC5B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC;QACjC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC;KAChD,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kCAAkC,CAAC,UAAsB;IACvE,OAAO;QACL,EAAE,EAAE,IAAI;QACR,MAAM,EAAE,GAAG;QACX,OAAO,EAAE,iBAAiB,EAAE;QAC5B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC;QAClC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC;KACjD,CAAC;AACJ,CAAC"}
|
|
1
|
+
{"version":3,"file":"fetch.mock.js","sourceRoot":"","sources":["../../../src/__tests__/mocks/fetch.mock.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAa5B;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,OAAqB;IACnD,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;AAC7C,CAAC;AAED,oCAAoC;AACpC,SAAS,iBAAiB;IACxB,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC;AAC7B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,IAAa;IAC9C,OAAO,eAAe,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;QAClC,EAAE,EAAE,IAAI;QACR,MAAM,EAAE,GAAG;QACX,OAAO,EAAE,iBAAiB,EAAE;QAC5B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI;QACtB,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;KACvC,CAAC,CAAC,CAAC;AACN,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAC9B,MAAc,EACd,aAAqB,OAAO,EAC5B,OAAe,EAAE;IAEjB,OAAO,eAAe,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;QAClC,EAAE,EAAE,KAAK;QACT,MAAM;QACN,UAAU;QACV,OAAO,EAAE,iBAAiB,EAAE;QAC5B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QACnC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI;KACvB,CAAC,CAAC,CAAC;AACN,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB,CACrC,UAAkB,cAAc;IAEhC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;AACvD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CACnC,SAA8B;IAE9B,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;QAC3C,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QACtE,SAAS,EAAE,CAAC;QACZ,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,2BAA2B,CAAC,UAAsB;IAChE,OAAO;QACL,EAAE,EAAE,IAAI;QACR,MAAM,EAAE,GAAG;QACX,OAAO,EAAE,iBAAiB,EAAE;QAC5B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;YACjB,IAAI,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;SACrD,CAAC;QACF,IAAI,EAAE,KAAK,IAAI,EAAE,CACf,IAAI,CAAC,SAAS,CAAC;YACb,IAAI,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;SACrD,CAAC;KACL,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,6BAA6B,CAAC,SAAmB;IAC/D,OAAO;QACL,EAAE,EAAE,IAAI;QACR,MAAM,EAAE,GAAG;QACX,OAAO,EAAE,iBAAiB,EAAE;QAC5B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC;QACjC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC;KAChD,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kCAAkC,CAAC,UAAsB;IACvE,OAAO;QACL,EAAE,EAAE,IAAI;QACR,MAAM,EAAE,GAAG;QACX,OAAO,EAAE,iBAAiB,EAAE;QAC5B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC;QAClC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC;KACjD,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,6BAA6B,CAAC,SAAmB;IAC/D,OAAO;QACL,EAAE,EAAE,IAAI;QACR,MAAM,EAAE,GAAG;QACX,OAAO,EAAE,iBAAiB,EAAE;QAC5B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;QACxD,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;KACvE,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kCAAkC,CAAC,UAAsB;IACvE,OAAO;QACL,EAAE,EAAE,IAAI;QACR,MAAM,EAAE,GAAG;QACX,OAAO,EAAE,iBAAiB,EAAE;QAC5B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC;QAC5E,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC;KAC3F,CAAC;AACJ,CAAC"}
|
package/dist/config.d.ts
CHANGED
|
@@ -27,6 +27,7 @@ declare const ConfigSchema: z.ZodObject<{
|
|
|
27
27
|
backend: z.ZodOptional<z.ZodEnum<{
|
|
28
28
|
jina: "jina";
|
|
29
29
|
ollama: "ollama";
|
|
30
|
+
gemini: "gemini";
|
|
30
31
|
}>>;
|
|
31
32
|
model: z.ZodOptional<z.ZodString>;
|
|
32
33
|
ollamaConcurrency: z.ZodOptional<z.ZodNumber>;
|
|
@@ -107,6 +108,7 @@ export declare function getInstructions(config: LanceContextConfig): string | un
|
|
|
107
108
|
*/
|
|
108
109
|
export interface LanceContextSecrets {
|
|
109
110
|
jinaApiKey?: string;
|
|
111
|
+
geminiApiKey?: string;
|
|
110
112
|
}
|
|
111
113
|
/**
|
|
112
114
|
* Load secrets from .lance-context/secrets.json
|
|
@@ -120,7 +122,7 @@ export declare function saveSecrets(projectPath: string, secrets: LanceContextSe
|
|
|
120
122
|
* Embedding settings for dashboard configuration
|
|
121
123
|
*/
|
|
122
124
|
export interface EmbeddingSettings {
|
|
123
|
-
backend: 'jina' | 'ollama';
|
|
125
|
+
backend: 'jina' | 'ollama' | 'gemini';
|
|
124
126
|
apiKey?: string;
|
|
125
127
|
ollamaUrl?: string;
|
|
126
128
|
/** Number of concurrent requests to Ollama */
|
|
@@ -138,7 +140,7 @@ export declare function saveEmbeddingSettings(projectPath: string, settings: Emb
|
|
|
138
140
|
* Get current embedding settings including secrets
|
|
139
141
|
*/
|
|
140
142
|
export declare function getEmbeddingSettings(projectPath: string): Promise<{
|
|
141
|
-
backend: 'jina' | 'ollama';
|
|
143
|
+
backend: 'jina' | 'ollama' | 'gemini';
|
|
142
144
|
hasApiKey: boolean;
|
|
143
145
|
ollamaUrl?: string;
|
|
144
146
|
ollamaConcurrency: number;
|
package/dist/config.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB;;GAEG;AACH,QAAA,MAAM,oBAAoB;;;iBAGxB,CAAC;AAEH,QAAA,MAAM,kBAAkB;;;;iBAKtB,CAAC;AASH,QAAA,MAAM,qBAAqB;;;;iBAIzB,CAAC;AAEH,QAAA,MAAM,oBAAoB;;;iBAKxB,CAAC;AAEH,QAAA,MAAM,YAAY
|
|
1
|
+
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB;;GAEG;AACH,QAAA,MAAM,oBAAoB;;;iBAGxB,CAAC;AAEH,QAAA,MAAM,kBAAkB;;;;iBAKtB,CAAC;AASH,QAAA,MAAM,qBAAqB;;;;iBAIzB,CAAC;AAEH,QAAA,MAAM,oBAAoB;;;iBAKxB,CAAC;AAEH,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAShB,CAAC;AAEH,MAAM,MAAM,kBAAkB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,YAAY,CAAC,CAAC;AAC9D,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAClE,MAAM,MAAM,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAC9D,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC;AACpE,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAoClE;;GAEG;AACH,eAAO,MAAM,gBAAgB,EAAE,QAAQ,CAAC,cAAc,CAGrD,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,cAAc,EAAE,QAAQ,CAAC,YAAY,CAIjD,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,iBAAiB,EAAE,QAAQ,CAAC,eAAe,CAIvD,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,gBAAgB,EAAE,QAAQ,CAAC,cAAc,CAGrD,CAAC;AAEF,eAAO,MAAM,cAAc,EAAE,kBAO5B,CAAC;AAyUF,wBAAsB,UAAU,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,CAAC,CA+EjF;AAED;;GAEG;AACH,wBAAgB,kBAAkB,IAAI,MAAM,EAAE,CAE7C;AAED;;GAEG;AACH,wBAAgB,yBAAyB,IAAI,MAAM,EAAE,CAEpD;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,kBAAkB,GAAG,QAAQ,CAAC,cAAc,CAAC,CAKtF;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,kBAAkB,GAAG,QAAQ,CAAC,YAAY,CAAC,CAKlF;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,kBAAkB,GAAG,QAAQ,CAAC,eAAe,CAAC,CAKxF;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,kBAAkB,GAAG,QAAQ,CAAC,cAAc,CAAC,CAKtF;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,kBAAkB,GAAG,MAAM,GAAG,SAAS,CAE9E;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED;;GAEG;AACH,wBAAsB,WAAW,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAQnF;AAED;;GAEG;AACH,wBAAsB,WAAW,CAC/B,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,mBAAmB,GAC3B,OAAO,CAAC,IAAI,CAAC,CAYf;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,MAAM,GAAG,QAAQ,GAAG,QAAQ,CAAC;IACtC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,8CAA8C;IAC9C,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,2CAA2C;IAC3C,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED;;;;GAIG;AACH,wBAAsB,qBAAqB,CACzC,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,iBAAiB,GAC1B,OAAO,CAAC,IAAI,CAAC,CA0Cf;AAED;;GAEG;AACH,wBAAsB,oBAAoB,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC;IACvE,OAAO,EAAE,MAAM,GAAG,QAAQ,GAAG,QAAQ,CAAC;IACtC,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC,CAoBD;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB;AAED;;GAEG;AACH,wBAAsB,qBAAqB,CACzC,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,iBAAiB,GAC1B,OAAO,CAAC,IAAI,CAAC,CA4Bf;AAED;;GAEG;AACH,wBAAsB,oBAAoB,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC;IACvE,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,OAAO,CAAC;CACtB,CAAC,CAQD"}
|
package/dist/config.js
CHANGED
|
@@ -15,7 +15,7 @@ const SearchConfigSchema = z.object({
|
|
|
15
15
|
autoReindex: z.boolean().optional(),
|
|
16
16
|
});
|
|
17
17
|
const EmbeddingConfigSchema = z.object({
|
|
18
|
-
backend: z.enum(['jina', 'ollama']).optional(),
|
|
18
|
+
backend: z.enum(['jina', 'ollama', 'gemini']).optional(),
|
|
19
19
|
model: z.string().optional(),
|
|
20
20
|
/** Number of concurrent requests to Ollama (default: 10). Increase if your system has capacity. */
|
|
21
21
|
ollamaConcurrency: z.number().min(1).max(200).optional(),
|
|
@@ -101,7 +101,7 @@ export const DEFAULT_DASHBOARD = {
|
|
|
101
101
|
*/
|
|
102
102
|
export const DEFAULT_INDEXING = {
|
|
103
103
|
batchDelayMs: 0,
|
|
104
|
-
batchSize:
|
|
104
|
+
batchSize: 256,
|
|
105
105
|
};
|
|
106
106
|
export const DEFAULT_CONFIG = {
|
|
107
107
|
patterns: DEFAULT_PATTERNS,
|
|
@@ -574,7 +574,12 @@ export async function saveEmbeddingSettings(projectPath, settings) {
|
|
|
574
574
|
await fs.writeFile(configPath, JSON.stringify(existingConfig, null, 2));
|
|
575
575
|
// Save API key to secrets if provided
|
|
576
576
|
if (settings.apiKey) {
|
|
577
|
-
|
|
577
|
+
if (settings.backend === 'gemini') {
|
|
578
|
+
await saveSecrets(projectPath, { geminiApiKey: settings.apiKey });
|
|
579
|
+
}
|
|
580
|
+
else if (settings.backend === 'jina') {
|
|
581
|
+
await saveSecrets(projectPath, { jinaApiKey: settings.apiKey });
|
|
582
|
+
}
|
|
578
583
|
}
|
|
579
584
|
}
|
|
580
585
|
/**
|
|
@@ -583,9 +588,18 @@ export async function saveEmbeddingSettings(projectPath, settings) {
|
|
|
583
588
|
export async function getEmbeddingSettings(projectPath) {
|
|
584
589
|
const config = await loadConfig(projectPath);
|
|
585
590
|
const secrets = await loadSecrets(projectPath);
|
|
591
|
+
const backend = config.embedding?.backend || 'ollama';
|
|
592
|
+
// Check for API key based on backend
|
|
593
|
+
let hasApiKey = false;
|
|
594
|
+
if (backend === 'jina') {
|
|
595
|
+
hasApiKey = !!(secrets.jinaApiKey || process.env.JINA_API_KEY);
|
|
596
|
+
}
|
|
597
|
+
else if (backend === 'gemini') {
|
|
598
|
+
hasApiKey = !!(secrets.geminiApiKey || process.env.GEMINI_API_KEY);
|
|
599
|
+
}
|
|
586
600
|
return {
|
|
587
|
-
backend
|
|
588
|
-
hasApiKey
|
|
601
|
+
backend,
|
|
602
|
+
hasApiKey,
|
|
589
603
|
ollamaUrl: process.env.OLLAMA_URL || 'http://localhost:11434',
|
|
590
604
|
ollamaConcurrency: config.embedding?.ollamaConcurrency || 1,
|
|
591
605
|
batchSize: config.indexing?.batchSize || DEFAULT_INDEXING.batchSize,
|