lance-context 1.11.0 → 1.12.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.
Files changed (45) hide show
  1. package/dist/__tests__/tools/clustering-handlers.test.d.ts +2 -0
  2. package/dist/__tests__/tools/clustering-handlers.test.d.ts.map +1 -0
  3. package/dist/__tests__/tools/clustering-handlers.test.js +286 -0
  4. package/dist/__tests__/tools/clustering-handlers.test.js.map +1 -0
  5. package/dist/__tests__/tools/index-handlers.test.d.ts +2 -0
  6. package/dist/__tests__/tools/index-handlers.test.d.ts.map +1 -0
  7. package/dist/__tests__/tools/index-handlers.test.js +184 -0
  8. package/dist/__tests__/tools/index-handlers.test.js.map +1 -0
  9. package/dist/__tests__/tools/memory-handlers.test.d.ts +2 -0
  10. package/dist/__tests__/tools/memory-handlers.test.d.ts.map +1 -0
  11. package/dist/__tests__/tools/memory-handlers.test.js +175 -0
  12. package/dist/__tests__/tools/memory-handlers.test.js.map +1 -0
  13. package/dist/__tests__/tools/search-handlers.test.d.ts +2 -0
  14. package/dist/__tests__/tools/search-handlers.test.d.ts.map +1 -0
  15. package/dist/__tests__/tools/search-handlers.test.js +229 -0
  16. package/dist/__tests__/tools/search-handlers.test.js.map +1 -0
  17. package/dist/__tests__/tools/worktree-handlers.test.d.ts +2 -0
  18. package/dist/__tests__/tools/worktree-handlers.test.d.ts.map +1 -0
  19. package/dist/__tests__/tools/worktree-handlers.test.js +297 -0
  20. package/dist/__tests__/tools/worktree-handlers.test.js.map +1 -0
  21. package/dist/tools/clustering-handlers.d.ts +70 -0
  22. package/dist/tools/clustering-handlers.d.ts.map +1 -0
  23. package/dist/tools/clustering-handlers.js +122 -0
  24. package/dist/tools/clustering-handlers.js.map +1 -0
  25. package/dist/tools/index-handlers.d.ts +31 -0
  26. package/dist/tools/index-handlers.d.ts.map +1 -0
  27. package/dist/tools/index-handlers.js +53 -0
  28. package/dist/tools/index-handlers.js.map +1 -0
  29. package/dist/tools/memory-handlers.d.ts +95 -0
  30. package/dist/tools/memory-handlers.d.ts.map +1 -0
  31. package/dist/tools/memory-handlers.js +124 -0
  32. package/dist/tools/memory-handlers.js.map +1 -0
  33. package/dist/tools/search-handlers.d.ts +47 -0
  34. package/dist/tools/search-handlers.d.ts.map +1 -0
  35. package/dist/tools/search-handlers.js +90 -0
  36. package/dist/tools/search-handlers.js.map +1 -0
  37. package/dist/tools/types.d.ts +32 -0
  38. package/dist/tools/types.d.ts.map +1 -0
  39. package/dist/tools/types.js +17 -0
  40. package/dist/tools/types.js.map +1 -0
  41. package/dist/tools/worktree-handlers.d.ts +96 -0
  42. package/dist/tools/worktree-handlers.d.ts.map +1 -0
  43. package/dist/tools/worktree-handlers.js +184 -0
  44. package/dist/tools/worktree-handlers.js.map +1 -0
  45. package/package.json +1 -1
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=clustering-handlers.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"clustering-handlers.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/tools/clustering-handlers.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,286 @@
1
+ import { describe, it, expect, vi, beforeEach } from 'vitest';
2
+ import { handleSummarizeCodebase, handleListConcepts, handleSearchByConcept, parseSummarizeCodebaseArgs, parseListConceptsArgs, parseSearchByConceptArgs, formatCodebaseSummary, formatConceptClusters, formatConceptSearchResults, } from '../../tools/clustering-handlers.js';
3
+ import { LanceContextError } from '../../utils/errors.js';
4
+ // Helper to create a valid ConceptCluster
5
+ function createConceptCluster(overrides = {}) {
6
+ return {
7
+ id: 0,
8
+ label: 'Test Cluster',
9
+ size: 10,
10
+ keywords: ['test', 'keyword'],
11
+ representativeChunks: ['test.ts:1-10'],
12
+ centroid: [0.1, 0.2, 0.3],
13
+ ...overrides,
14
+ };
15
+ }
16
+ describe('clustering-handlers', () => {
17
+ let mockIndexer;
18
+ let context;
19
+ beforeEach(() => {
20
+ mockIndexer = {
21
+ summarizeCodebase: vi.fn(),
22
+ listConcepts: vi.fn(),
23
+ searchByConcept: vi.fn(),
24
+ };
25
+ context = {
26
+ indexer: mockIndexer,
27
+ toolGuidance: '\n---\nGuidance',
28
+ };
29
+ });
30
+ describe('parseSummarizeCodebaseArgs', () => {
31
+ it('should return defaults when no args provided', () => {
32
+ const result = parseSummarizeCodebaseArgs(undefined);
33
+ expect(result.numClusters).toBeUndefined();
34
+ });
35
+ it('should parse numClusters', () => {
36
+ const result = parseSummarizeCodebaseArgs({ numClusters: 5 });
37
+ expect(result.numClusters).toBe(5);
38
+ });
39
+ it('should ignore invalid numClusters', () => {
40
+ const result = parseSummarizeCodebaseArgs({ numClusters: 'invalid' });
41
+ expect(result.numClusters).toBeUndefined();
42
+ });
43
+ });
44
+ describe('formatCodebaseSummary', () => {
45
+ it('should format summary correctly', () => {
46
+ const summary = {
47
+ totalFiles: 100,
48
+ totalChunks: 500,
49
+ languages: [
50
+ { language: 'typescript', fileCount: 80, chunkCount: 400 },
51
+ { language: 'javascript', fileCount: 20, chunkCount: 100 },
52
+ ],
53
+ concepts: [
54
+ createConceptCluster({
55
+ id: 0,
56
+ label: 'Authentication',
57
+ size: 50,
58
+ keywords: ['auth', 'login', 'token', 'jwt', 'session'],
59
+ representativeChunks: ['auth.ts:1-20'],
60
+ }),
61
+ ],
62
+ clusteringQuality: 0.75,
63
+ generatedAt: '2024-01-15T00:00:00.000Z',
64
+ };
65
+ const formatted = formatCodebaseSummary(summary);
66
+ expect(formatted).toContain('# Codebase Summary');
67
+ expect(formatted).toContain('**Total Files**: 100');
68
+ expect(formatted).toContain('**Total Chunks**: 500');
69
+ expect(formatted).toContain('75.0% (silhouette score)');
70
+ expect(formatted).toContain('**typescript**: 80 files');
71
+ expect(formatted).toContain('**Cluster 0: Authentication**');
72
+ expect(formatted).toContain('auth, login, token, jwt, session');
73
+ });
74
+ });
75
+ describe('handleSummarizeCodebase', () => {
76
+ it('should call indexer.summarizeCodebase', async () => {
77
+ const summary = {
78
+ totalFiles: 10,
79
+ totalChunks: 50,
80
+ languages: [],
81
+ concepts: [],
82
+ clusteringQuality: 0.8,
83
+ generatedAt: '2024-01-15T00:00:00.000Z',
84
+ };
85
+ vi.mocked(mockIndexer.summarizeCodebase).mockResolvedValue(summary);
86
+ await handleSummarizeCodebase({}, context);
87
+ expect(mockIndexer.summarizeCodebase).toHaveBeenCalledWith(undefined);
88
+ });
89
+ it('should pass numClusters option when provided', async () => {
90
+ const summary = {
91
+ totalFiles: 10,
92
+ totalChunks: 50,
93
+ languages: [],
94
+ concepts: [],
95
+ clusteringQuality: 0.8,
96
+ generatedAt: '2024-01-15T00:00:00.000Z',
97
+ };
98
+ vi.mocked(mockIndexer.summarizeCodebase).mockResolvedValue(summary);
99
+ await handleSummarizeCodebase({ numClusters: 5 }, context);
100
+ expect(mockIndexer.summarizeCodebase).toHaveBeenCalledWith({ numClusters: 5 });
101
+ });
102
+ it('should append tool guidance', async () => {
103
+ const summary = {
104
+ totalFiles: 10,
105
+ totalChunks: 50,
106
+ languages: [],
107
+ concepts: [],
108
+ clusteringQuality: 0.8,
109
+ generatedAt: '2024-01-15T00:00:00.000Z',
110
+ };
111
+ vi.mocked(mockIndexer.summarizeCodebase).mockResolvedValue(summary);
112
+ const result = await handleSummarizeCodebase({}, context);
113
+ expect(result.content[0].text).toContain('Guidance');
114
+ });
115
+ });
116
+ describe('parseListConceptsArgs', () => {
117
+ it('should return defaults when no args provided', () => {
118
+ const result = parseListConceptsArgs(undefined);
119
+ expect(result.forceRecluster).toBe(false);
120
+ });
121
+ it('should parse forceRecluster', () => {
122
+ const result = parseListConceptsArgs({ forceRecluster: true });
123
+ expect(result.forceRecluster).toBe(true);
124
+ });
125
+ it('should default forceRecluster to false for invalid value', () => {
126
+ const result = parseListConceptsArgs({ forceRecluster: 'true' });
127
+ expect(result.forceRecluster).toBe(false);
128
+ });
129
+ });
130
+ describe('formatConceptClusters', () => {
131
+ it('should return message for empty list', () => {
132
+ expect(formatConceptClusters([])).toContain('No concept clusters found');
133
+ });
134
+ it('should format concepts correctly', () => {
135
+ const concepts = [
136
+ createConceptCluster({
137
+ id: 0,
138
+ label: 'API Handlers',
139
+ size: 25,
140
+ keywords: ['api', 'handler', 'request', 'response', 'route'],
141
+ representativeChunks: ['api.ts:1-20', 'handler.ts:1-15', 'routes.ts:1-30'],
142
+ }),
143
+ createConceptCluster({
144
+ id: 1,
145
+ label: 'Database',
146
+ size: 15,
147
+ keywords: ['db', 'query', 'model'],
148
+ representativeChunks: ['db.ts:1-50'],
149
+ }),
150
+ ];
151
+ const formatted = formatConceptClusters(concepts);
152
+ expect(formatted).toContain('# Concept Clusters');
153
+ expect(formatted).toContain('## Cluster 0: API Handlers');
154
+ expect(formatted).toContain('**Size**: 25 code chunks');
155
+ expect(formatted).toContain('api, handler, request, response, route');
156
+ expect(formatted).toContain('api.ts:1-20, handler.ts:1-15, routes.ts:1-30');
157
+ expect(formatted).toContain('## Cluster 1: Database');
158
+ });
159
+ });
160
+ describe('handleListConcepts', () => {
161
+ it('should call indexer.listConcepts', async () => {
162
+ vi.mocked(mockIndexer.listConcepts).mockResolvedValue([]);
163
+ await handleListConcepts({ forceRecluster: true }, context);
164
+ expect(mockIndexer.listConcepts).toHaveBeenCalledWith(true);
165
+ });
166
+ it('should return formatted concepts', async () => {
167
+ const concepts = [
168
+ createConceptCluster({
169
+ id: 0,
170
+ label: 'Test',
171
+ size: 10,
172
+ keywords: ['test'],
173
+ representativeChunks: ['test.ts:1-10'],
174
+ }),
175
+ ];
176
+ vi.mocked(mockIndexer.listConcepts).mockResolvedValue(concepts);
177
+ const result = await handleListConcepts({ forceRecluster: false }, context);
178
+ expect(result.content[0].text).toContain('Cluster 0: Test');
179
+ });
180
+ it('should append tool guidance', async () => {
181
+ vi.mocked(mockIndexer.listConcepts).mockResolvedValue([]);
182
+ const result = await handleListConcepts({ forceRecluster: false }, context);
183
+ expect(result.content[0].text).toContain('Guidance');
184
+ });
185
+ });
186
+ describe('parseSearchByConceptArgs', () => {
187
+ it('should throw when conceptId is missing', () => {
188
+ expect(() => parseSearchByConceptArgs({})).toThrow(LanceContextError);
189
+ expect(() => parseSearchByConceptArgs({})).toThrow('conceptId is required');
190
+ });
191
+ it('should throw when conceptId is negative', () => {
192
+ expect(() => parseSearchByConceptArgs({ conceptId: -1 })).toThrow(LanceContextError);
193
+ });
194
+ it('should parse valid conceptId', () => {
195
+ const result = parseSearchByConceptArgs({ conceptId: 5 });
196
+ expect(result.conceptId).toBe(5);
197
+ });
198
+ it('should use default limit of 10', () => {
199
+ const result = parseSearchByConceptArgs({ conceptId: 0 });
200
+ expect(result.limit).toBe(10);
201
+ });
202
+ it('should parse custom limit', () => {
203
+ const result = parseSearchByConceptArgs({ conceptId: 0, limit: 20 });
204
+ expect(result.limit).toBe(20);
205
+ });
206
+ it('should parse optional query', () => {
207
+ const result = parseSearchByConceptArgs({ conceptId: 0, query: 'authentication' });
208
+ expect(result.query).toBe('authentication');
209
+ });
210
+ });
211
+ describe('formatConceptSearchResults', () => {
212
+ it('should return message for empty results', () => {
213
+ const formatted = formatConceptSearchResults([], 5);
214
+ expect(formatted).toContain('No code found in concept cluster 5');
215
+ expect(formatted).toContain('list_concepts');
216
+ });
217
+ it('should format results with symbol info', () => {
218
+ const results = [
219
+ {
220
+ id: 'test.ts:1-20',
221
+ filepath: 'test.ts',
222
+ content: 'function authenticate() {}',
223
+ startLine: 1,
224
+ endLine: 20,
225
+ language: 'typescript',
226
+ symbolName: 'authenticate',
227
+ symbolType: 'function',
228
+ },
229
+ ];
230
+ const formatted = formatConceptSearchResults(results, 0);
231
+ expect(formatted).toContain('## Result 1: test.ts:1-20');
232
+ expect(formatted).toContain('**Symbol:** `authenticate`');
233
+ expect(formatted).toContain('(function)');
234
+ expect(formatted).toContain('```typescript');
235
+ expect(formatted).toContain('function authenticate()');
236
+ });
237
+ it('should format results without symbol info', () => {
238
+ const results = [
239
+ {
240
+ id: 'config.json:1-10',
241
+ filepath: 'config.json',
242
+ content: '{ "key": "value" }',
243
+ startLine: 1,
244
+ endLine: 10,
245
+ language: 'json',
246
+ },
247
+ ];
248
+ const formatted = formatConceptSearchResults(results, 0);
249
+ expect(formatted).toContain('## Result 1: config.json:1-10');
250
+ expect(formatted).not.toContain('**Symbol:**');
251
+ });
252
+ });
253
+ describe('handleSearchByConcept', () => {
254
+ it('should call indexer.searchByConcept with correct options', async () => {
255
+ vi.mocked(mockIndexer.searchByConcept).mockResolvedValue([]);
256
+ await handleSearchByConcept({ conceptId: 3, query: 'auth', limit: 5 }, context);
257
+ expect(mockIndexer.searchByConcept).toHaveBeenCalledWith(3, 'auth', 5);
258
+ });
259
+ it('should use default limit when not provided', async () => {
260
+ vi.mocked(mockIndexer.searchByConcept).mockResolvedValue([]);
261
+ await handleSearchByConcept({ conceptId: 0 }, context);
262
+ expect(mockIndexer.searchByConcept).toHaveBeenCalledWith(0, undefined, 10);
263
+ });
264
+ it('should return formatted results', async () => {
265
+ const results = [
266
+ {
267
+ id: 'result.ts:1-10',
268
+ filepath: 'result.ts',
269
+ content: 'code here',
270
+ startLine: 1,
271
+ endLine: 10,
272
+ language: 'typescript',
273
+ },
274
+ ];
275
+ vi.mocked(mockIndexer.searchByConcept).mockResolvedValue(results);
276
+ const result = await handleSearchByConcept({ conceptId: 0 }, context);
277
+ expect(result.content[0].text).toContain('result.ts:1-10');
278
+ });
279
+ it('should append tool guidance', async () => {
280
+ vi.mocked(mockIndexer.searchByConcept).mockResolvedValue([]);
281
+ const result = await handleSearchByConcept({ conceptId: 0 }, context);
282
+ expect(result.content[0].text).toContain('Guidance');
283
+ });
284
+ });
285
+ });
286
+ //# sourceMappingURL=clustering-handlers.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"clustering-handlers.test.js","sourceRoot":"","sources":["../../../src/__tests__/tools/clustering-handlers.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EACL,uBAAuB,EACvB,kBAAkB,EAClB,qBAAqB,EACrB,0BAA0B,EAC1B,qBAAqB,EACrB,wBAAwB,EACxB,qBAAqB,EACrB,qBAAqB,EACrB,0BAA0B,GAE3B,MAAM,oCAAoC,CAAC;AAG5C,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAE1D,0CAA0C;AAC1C,SAAS,oBAAoB,CAAC,YAAqC,EAAE;IACnE,OAAO;QACL,EAAE,EAAE,CAAC;QACL,KAAK,EAAE,cAAc;QACrB,IAAI,EAAE,EAAE;QACR,QAAQ,EAAE,CAAC,MAAM,EAAE,SAAS,CAAC;QAC7B,oBAAoB,EAAE,CAAC,cAAc,CAAC;QACtC,QAAQ,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;QACzB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,IAAI,WAAiC,CAAC;IACtC,IAAI,OAA8B,CAAC;IAEnC,UAAU,CAAC,GAAG,EAAE;QACd,WAAW,GAAG;YACZ,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC1B,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;YACrB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;SACzB,CAAC;QAEF,OAAO,GAAG;YACR,OAAO,EAAE,WAA0B;YACnC,YAAY,EAAE,iBAAiB;SAChC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,MAAM,GAAG,0BAA0B,CAAC,SAAS,CAAC,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,aAAa,EAAE,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAClC,MAAM,MAAM,GAAG,0BAA0B,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,MAAM,GAAG,0BAA0B,CAAC,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC,CAAC;YACtE,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,aAAa,EAAE,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,OAAO,GAAoB;gBAC/B,UAAU,EAAE,GAAG;gBACf,WAAW,EAAE,GAAG;gBAChB,SAAS,EAAE;oBACT,EAAE,QAAQ,EAAE,YAAY,EAAE,SAAS,EAAE,EAAE,EAAE,UAAU,EAAE,GAAG,EAAE;oBAC1D,EAAE,QAAQ,EAAE,YAAY,EAAE,SAAS,EAAE,EAAE,EAAE,UAAU,EAAE,GAAG,EAAE;iBAC3D;gBACD,QAAQ,EAAE;oBACR,oBAAoB,CAAC;wBACnB,EAAE,EAAE,CAAC;wBACL,KAAK,EAAE,gBAAgB;wBACvB,IAAI,EAAE,EAAE;wBACR,QAAQ,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,CAAC;wBACtD,oBAAoB,EAAE,CAAC,cAAc,CAAC;qBACvC,CAAC;iBACH;gBACD,iBAAiB,EAAE,IAAI;gBACvB,WAAW,EAAE,0BAA0B;aACxC,CAAC;YAEF,MAAM,SAAS,GAAG,qBAAqB,CAAC,OAAO,CAAC,CAAC;YAEjD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;YAClD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;YACpD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;YACrD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;YACxD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;YACxD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,+BAA+B,CAAC,CAAC;YAC7D,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,kCAAkC,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,OAAO,GAAoB;gBAC/B,UAAU,EAAE,EAAE;gBACd,WAAW,EAAE,EAAE;gBACf,SAAS,EAAE,EAAE;gBACb,QAAQ,EAAE,EAAE;gBACZ,iBAAiB,EAAE,GAAG;gBACtB,WAAW,EAAE,0BAA0B;aACxC,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,iBAAkB,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAErE,MAAM,uBAAuB,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAE3C,MAAM,CAAC,WAAW,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,OAAO,GAAoB;gBAC/B,UAAU,EAAE,EAAE;gBACd,WAAW,EAAE,EAAE;gBACf,SAAS,EAAE,EAAE;gBACb,QAAQ,EAAE,EAAE;gBACZ,iBAAiB,EAAE,GAAG;gBACtB,WAAW,EAAE,0BAA0B;aACxC,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,iBAAkB,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAErE,MAAM,uBAAuB,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAE3D,MAAM,CAAC,WAAW,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;QACjF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,OAAO,GAAoB;gBAC/B,UAAU,EAAE,EAAE;gBACd,WAAW,EAAE,EAAE;gBACf,SAAS,EAAE,EAAE;gBACb,QAAQ,EAAE,EAAE;gBACZ,iBAAiB,EAAE,GAAG;gBACtB,WAAW,EAAE,0BAA0B;aACxC,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,iBAAkB,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAErE,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAE1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,MAAM,GAAG,qBAAqB,CAAC,SAAS,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,MAAM,GAAG,qBAAqB,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,MAAM,GAAG,qBAAqB,CAAC,EAAE,cAAc,EAAE,MAAM,EAAE,CAAC,CAAC;YACjE,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAC;QAC3E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,QAAQ,GAAqB;gBACjC,oBAAoB,CAAC;oBACnB,EAAE,EAAE,CAAC;oBACL,KAAK,EAAE,cAAc;oBACrB,IAAI,EAAE,EAAE;oBACR,QAAQ,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,CAAC;oBAC5D,oBAAoB,EAAE,CAAC,aAAa,EAAE,iBAAiB,EAAE,gBAAgB,CAAC;iBAC3E,CAAC;gBACF,oBAAoB,CAAC;oBACnB,EAAE,EAAE,CAAC;oBACL,KAAK,EAAE,UAAU;oBACjB,IAAI,EAAE,EAAE;oBACR,QAAQ,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC;oBAClC,oBAAoB,EAAE,CAAC,YAAY,CAAC;iBACrC,CAAC;aACH,CAAC;YAEF,MAAM,SAAS,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YAElD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;YAClD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAC;YAC1D,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;YACxD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,wCAAwC,CAAC,CAAC;YACtE,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,8CAA8C,CAAC,CAAC;YAC5E,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,YAAa,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAE3D,MAAM,kBAAkB,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;YAE5D,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,QAAQ,GAAqB;gBACjC,oBAAoB,CAAC;oBACnB,EAAE,EAAE,CAAC;oBACL,KAAK,EAAE,MAAM;oBACb,IAAI,EAAE,EAAE;oBACR,QAAQ,EAAE,CAAC,MAAM,CAAC;oBAClB,oBAAoB,EAAE,CAAC,cAAc,CAAC;iBACvC,CAAC;aACH,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,YAAa,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAEjE,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,EAAE,cAAc,EAAE,KAAK,EAAE,EAAE,OAAO,CAAC,CAAC;YAE5E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,YAAa,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,EAAE,cAAc,EAAE,KAAK,EAAE,EAAE,OAAO,CAAC,CAAC;YAE5E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;YACtE,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;QAC9E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QACvF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,MAAM,GAAG,wBAAwB,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,MAAM,GAAG,wBAAwB,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,MAAM,GAAG,wBAAwB,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,MAAM,GAAG,wBAAwB,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,gBAAgB,EAAE,CAAC,CAAC;YACnF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,SAAS,GAAG,0BAA0B,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YACpD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,oCAAoC,CAAC,CAAC;YAClE,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,OAAO,GAAgB;gBAC3B;oBACE,EAAE,EAAE,cAAc;oBAClB,QAAQ,EAAE,SAAS;oBACnB,OAAO,EAAE,4BAA4B;oBACrC,SAAS,EAAE,CAAC;oBACZ,OAAO,EAAE,EAAE;oBACX,QAAQ,EAAE,YAAY;oBACtB,UAAU,EAAE,cAAc;oBAC1B,UAAU,EAAE,UAAU;iBACvB;aACF,CAAC;YAEF,MAAM,SAAS,GAAG,0BAA0B,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAEzD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAC;YACzD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAC;YAC1D,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YAC1C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YAC7C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,OAAO,GAAgB;gBAC3B;oBACE,EAAE,EAAE,kBAAkB;oBACtB,QAAQ,EAAE,aAAa;oBACvB,OAAO,EAAE,oBAAoB;oBAC7B,SAAS,EAAE,CAAC;oBACZ,OAAO,EAAE,EAAE;oBACX,QAAQ,EAAE,MAAM;iBACjB;aACF,CAAC;YAEF,MAAM,SAAS,GAAG,0BAA0B,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAEzD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,+BAA+B,CAAC,CAAC;YAC7D,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,eAAgB,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAE9D,MAAM,qBAAqB,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAEhF,MAAM,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,eAAgB,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAE9D,MAAM,qBAAqB,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAEvD,MAAM,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,OAAO,GAAgB;gBAC3B;oBACE,EAAE,EAAE,gBAAgB;oBACpB,QAAQ,EAAE,WAAW;oBACrB,OAAO,EAAE,WAAW;oBACpB,SAAS,EAAE,CAAC;oBACZ,OAAO,EAAE,EAAE;oBACX,QAAQ,EAAE,YAAY;iBACvB;aACF,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,eAAgB,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAEnE,MAAM,MAAM,GAAG,MAAM,qBAAqB,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,eAAgB,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAE9D,MAAM,MAAM,GAAG,MAAM,qBAAqB,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=index-handlers.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index-handlers.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/tools/index-handlers.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,184 @@
1
+ import { describe, it, expect, vi, beforeEach } from 'vitest';
2
+ import { handleIndexCodebase, handleGetIndexStatus, handleClearIndex, parseIndexCodebaseArgs, } from '../../tools/index-handlers.js';
3
+ describe('index-handlers', () => {
4
+ let mockIndexer;
5
+ let context;
6
+ beforeEach(() => {
7
+ mockIndexer = {
8
+ indexCodebase: vi.fn(),
9
+ getStatus: vi.fn(),
10
+ clearIndex: vi.fn(),
11
+ };
12
+ context = {
13
+ indexer: mockIndexer,
14
+ projectPath: '/test/project',
15
+ toolGuidance: '\n---\nGuidance text',
16
+ };
17
+ });
18
+ describe('parseIndexCodebaseArgs', () => {
19
+ it('should return defaults when no args provided', () => {
20
+ const result = parseIndexCodebaseArgs(undefined);
21
+ expect(result).toEqual({
22
+ patterns: undefined,
23
+ excludePatterns: undefined,
24
+ forceReindex: false,
25
+ autoRepair: false,
26
+ });
27
+ });
28
+ it('should parse valid patterns array', () => {
29
+ const result = parseIndexCodebaseArgs({ patterns: ['**/*.ts', '**/*.js'] });
30
+ expect(result.patterns).toEqual(['**/*.ts', '**/*.js']);
31
+ });
32
+ it('should ignore invalid patterns', () => {
33
+ const result = parseIndexCodebaseArgs({ patterns: 'not-an-array' });
34
+ expect(result.patterns).toBeUndefined();
35
+ });
36
+ it('should parse forceReindex boolean', () => {
37
+ const result = parseIndexCodebaseArgs({ forceReindex: true });
38
+ expect(result.forceReindex).toBe(true);
39
+ });
40
+ it('should parse autoRepair boolean', () => {
41
+ const result = parseIndexCodebaseArgs({ autoRepair: true });
42
+ expect(result.autoRepair).toBe(true);
43
+ });
44
+ it('should default booleans to false for non-boolean values', () => {
45
+ const result = parseIndexCodebaseArgs({ forceReindex: 'true', autoRepair: 1 });
46
+ expect(result.forceReindex).toBe(false);
47
+ expect(result.autoRepair).toBe(false);
48
+ });
49
+ });
50
+ describe('handleIndexCodebase', () => {
51
+ it('should return full reindex message for non-incremental result', async () => {
52
+ vi.mocked(mockIndexer.indexCodebase).mockResolvedValue({
53
+ filesIndexed: 10,
54
+ chunksCreated: 50,
55
+ incremental: false,
56
+ repaired: false,
57
+ });
58
+ const result = await handleIndexCodebase({}, context);
59
+ expect(result.content[0].text).toContain('Full reindex');
60
+ expect(result.content[0].text).toContain('10 files');
61
+ expect(result.content[0].text).toContain('50 chunks');
62
+ });
63
+ it('should return incremental message for incremental result', async () => {
64
+ vi.mocked(mockIndexer.indexCodebase).mockResolvedValue({
65
+ filesIndexed: 2,
66
+ chunksCreated: 10,
67
+ incremental: true,
68
+ repaired: false,
69
+ });
70
+ const result = await handleIndexCodebase({}, context);
71
+ expect(result.content[0].text).toContain('Incremental update');
72
+ });
73
+ it('should return repaired message when corruption was fixed', async () => {
74
+ vi.mocked(mockIndexer.indexCodebase).mockResolvedValue({
75
+ filesIndexed: 10,
76
+ chunksCreated: 50,
77
+ incremental: false,
78
+ repaired: true,
79
+ });
80
+ const result = await handleIndexCodebase({}, context);
81
+ expect(result.content[0].text).toContain('Repaired');
82
+ expect(result.content[0].text).toContain('corruption detected');
83
+ });
84
+ it('should pass arguments to indexer', async () => {
85
+ vi.mocked(mockIndexer.indexCodebase).mockResolvedValue({
86
+ filesIndexed: 0,
87
+ chunksCreated: 0,
88
+ incremental: false,
89
+ repaired: false,
90
+ });
91
+ await handleIndexCodebase({
92
+ patterns: ['**/*.ts'],
93
+ excludePatterns: ['**/node_modules/**'],
94
+ forceReindex: true,
95
+ autoRepair: true,
96
+ }, context);
97
+ expect(mockIndexer.indexCodebase).toHaveBeenCalledWith(['**/*.ts'], ['**/node_modules/**'], true, undefined, true);
98
+ });
99
+ it('should call onProgress callback when provided', async () => {
100
+ vi.mocked(mockIndexer.indexCodebase).mockImplementation(async (_p, _e, _f, onProgress) => {
101
+ onProgress?.({ phase: 'scanning', current: 0, total: 0, message: 'test' });
102
+ return { filesIndexed: 0, chunksCreated: 0, incremental: false, repaired: false };
103
+ });
104
+ const onProgress = vi.fn();
105
+ await handleIndexCodebase({}, context, onProgress);
106
+ expect(onProgress).toHaveBeenCalledWith({
107
+ phase: 'scanning',
108
+ current: 0,
109
+ total: 0,
110
+ message: 'test',
111
+ });
112
+ });
113
+ it('should append tool guidance to response', async () => {
114
+ vi.mocked(mockIndexer.indexCodebase).mockResolvedValue({
115
+ filesIndexed: 0,
116
+ chunksCreated: 0,
117
+ incremental: false,
118
+ repaired: false,
119
+ });
120
+ const result = await handleIndexCodebase({}, context);
121
+ expect(result.content[0].text).toContain('Guidance text');
122
+ });
123
+ });
124
+ describe('handleGetIndexStatus', () => {
125
+ it('should return JSON status', async () => {
126
+ const status = {
127
+ indexed: true,
128
+ fileCount: 10,
129
+ chunkCount: 50,
130
+ lastUpdated: '2024-01-01T00:00:00.000Z',
131
+ indexPath: '/test/.lance-context',
132
+ };
133
+ vi.mocked(mockIndexer.getStatus).mockResolvedValue(status);
134
+ const result = await handleGetIndexStatus(context);
135
+ expect(result.content[0].text).toContain('"indexed": true');
136
+ expect(result.content[0].text).toContain('"fileCount": 10');
137
+ });
138
+ it('should include corruption warning when corrupted', async () => {
139
+ const status = {
140
+ indexed: true,
141
+ fileCount: 10,
142
+ chunkCount: 50,
143
+ lastUpdated: '2024-01-01T00:00:00.000Z',
144
+ indexPath: '/test/.lance-context',
145
+ corrupted: true,
146
+ corruptionReason: 'Checksum mismatch',
147
+ };
148
+ vi.mocked(mockIndexer.getStatus).mockResolvedValue(status);
149
+ const result = await handleGetIndexStatus(context);
150
+ expect(result.content[0].text).toContain('WARNING: Index corruption detected');
151
+ expect(result.content[0].text).toContain('Checksum mismatch');
152
+ expect(result.content[0].text).toContain('autoRepair: true');
153
+ });
154
+ it('should append tool guidance to response', async () => {
155
+ vi.mocked(mockIndexer.getStatus).mockResolvedValue({
156
+ indexed: false,
157
+ fileCount: 0,
158
+ chunkCount: 0,
159
+ lastUpdated: null,
160
+ indexPath: '/test/.lance-context',
161
+ });
162
+ const result = await handleGetIndexStatus(context);
163
+ expect(result.content[0].text).toContain('Guidance text');
164
+ });
165
+ });
166
+ describe('handleClearIndex', () => {
167
+ it('should call clearIndex on indexer', async () => {
168
+ vi.mocked(mockIndexer.clearIndex).mockResolvedValue(undefined);
169
+ await handleClearIndex(context);
170
+ expect(mockIndexer.clearIndex).toHaveBeenCalled();
171
+ });
172
+ it('should return success message', async () => {
173
+ vi.mocked(mockIndexer.clearIndex).mockResolvedValue(undefined);
174
+ const result = await handleClearIndex(context);
175
+ expect(result.content[0].text).toContain('Index cleared');
176
+ });
177
+ it('should append tool guidance to response', async () => {
178
+ vi.mocked(mockIndexer.clearIndex).mockResolvedValue(undefined);
179
+ const result = await handleClearIndex(context);
180
+ expect(result.content[0].text).toContain('Guidance text');
181
+ });
182
+ });
183
+ });
184
+ //# sourceMappingURL=index-handlers.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index-handlers.test.js","sourceRoot":"","sources":["../../../src/__tests__/tools/index-handlers.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EACL,mBAAmB,EACnB,oBAAoB,EACpB,gBAAgB,EAChB,sBAAsB,GACvB,MAAM,+BAA+B,CAAC;AAIvC,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,WAAiC,CAAC;IACtC,IAAI,OAAoB,CAAC;IAEzB,UAAU,CAAC,GAAG,EAAE;QACd,WAAW,GAAG;YACZ,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;YACtB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;YAClB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;SACpB,CAAC;QAEF,OAAO,GAAG;YACR,OAAO,EAAE,WAA0B;YACnC,WAAW,EAAE,eAAe;YAC5B,YAAY,EAAE,sBAAsB;SACrC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,MAAM,GAAG,sBAAsB,CAAC,SAAS,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,QAAQ,EAAE,SAAS;gBACnB,eAAe,EAAE,SAAS;gBAC1B,YAAY,EAAE,KAAK;gBACnB,UAAU,EAAE,KAAK;aAClB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,MAAM,GAAG,sBAAsB,CAAC,EAAE,QAAQ,EAAE,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;YAC5E,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,MAAM,GAAG,sBAAsB,CAAC,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC;YACpE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,MAAM,GAAG,sBAAsB,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,MAAM,GAAG,sBAAsB,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,MAAM,GAAG,sBAAsB,CAAC,EAAE,YAAY,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;YAC/E,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,aAAc,CAAC,CAAC,iBAAiB,CAAC;gBACtD,YAAY,EAAE,EAAE;gBAChB,aAAa,EAAE,EAAE;gBACjB,WAAW,EAAE,KAAK;gBAClB,QAAQ,EAAE,KAAK;aAChB,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAEtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,aAAc,CAAC,CAAC,iBAAiB,CAAC;gBACtD,YAAY,EAAE,CAAC;gBACf,aAAa,EAAE,EAAE;gBACjB,WAAW,EAAE,IAAI;gBACjB,QAAQ,EAAE,KAAK;aAChB,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAEtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,aAAc,CAAC,CAAC,iBAAiB,CAAC;gBACtD,YAAY,EAAE,EAAE;gBAChB,aAAa,EAAE,EAAE;gBACjB,WAAW,EAAE,KAAK;gBAClB,QAAQ,EAAE,IAAI;aACf,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAEtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,aAAc,CAAC,CAAC,iBAAiB,CAAC;gBACtD,YAAY,EAAE,CAAC;gBACf,aAAa,EAAE,CAAC;gBAChB,WAAW,EAAE,KAAK;gBAClB,QAAQ,EAAE,KAAK;aAChB,CAAC,CAAC;YAEH,MAAM,mBAAmB,CACvB;gBACE,QAAQ,EAAE,CAAC,SAAS,CAAC;gBACrB,eAAe,EAAE,CAAC,oBAAoB,CAAC;gBACvC,YAAY,EAAE,IAAI;gBAClB,UAAU,EAAE,IAAI;aACjB,EACD,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACpD,CAAC,SAAS,CAAC,EACX,CAAC,oBAAoB,CAAC,EACtB,IAAI,EACJ,SAAS,EACT,IAAI,CACL,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,aAAc,CAAC,CAAC,kBAAkB,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE;gBACxF,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC3E,OAAO,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;YACpF,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,MAAM,mBAAmB,CAAC,EAAE,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;YAEnD,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;gBACtC,KAAK,EAAE,UAAU;gBACjB,OAAO,EAAE,CAAC;gBACV,KAAK,EAAE,CAAC;gBACR,OAAO,EAAE,MAAM;aAChB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,aAAc,CAAC,CAAC,iBAAiB,CAAC;gBACtD,YAAY,EAAE,CAAC;gBACf,aAAa,EAAE,CAAC;gBAChB,WAAW,EAAE,KAAK;gBAClB,QAAQ,EAAE,KAAK;aAChB,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAEtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,MAAM,GAAgB;gBAC1B,OAAO,EAAE,IAAI;gBACb,SAAS,EAAE,EAAE;gBACb,UAAU,EAAE,EAAE;gBACd,WAAW,EAAE,0BAA0B;gBACvC,SAAS,EAAE,sBAAsB;aAClC,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,SAAU,CAAC,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAE5D,MAAM,MAAM,GAAG,MAAM,oBAAoB,CAAC,OAAO,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,MAAM,GAAgB;gBAC1B,OAAO,EAAE,IAAI;gBACb,SAAS,EAAE,EAAE;gBACb,UAAU,EAAE,EAAE;gBACd,WAAW,EAAE,0BAA0B;gBACvC,SAAS,EAAE,sBAAsB;gBACjC,SAAS,EAAE,IAAI;gBACf,gBAAgB,EAAE,mBAAmB;aACtC,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,SAAU,CAAC,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAE5D,MAAM,MAAM,GAAG,MAAM,oBAAoB,CAAC,OAAO,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,oCAAoC,CAAC,CAAC;YAC/E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,SAAU,CAAC,CAAC,iBAAiB,CAAC;gBAClD,OAAO,EAAE,KAAK;gBACd,SAAS,EAAE,CAAC;gBACZ,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,IAAI;gBACjB,SAAS,EAAE,sBAAsB;aAClC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,oBAAoB,CAAC,OAAO,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,UAAW,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEhE,MAAM,gBAAgB,CAAC,OAAO,CAAC,CAAC;YAEhC,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,UAAW,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEhE,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,OAAO,CAAC,CAAC;YAE/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,UAAW,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEhE,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,OAAO,CAAC,CAAC;YAE/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=memory-handlers.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"memory-handlers.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/tools/memory-handlers.test.ts"],"names":[],"mappings":""}