@code-rag/core 0.1.0

package/dist/enrichment/nl-enricher.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/enrichment/nl-enricher.test.js

@@ -0,0 +1,154 @@
+import { describe, it, expect, vi } from 'vitest';
+import { ok, err } from 'neverthrow';
+import { OllamaError } from './ollama-client.js';
+import { NLEnricher, EnrichmentError } from './nl-enricher.js';
+function makeChunk(overrides = {}) {
+    return {
+        id: 'chunk-1',
+        content: 'function add(a: number, b: number): number { return a + b; }',
+        nlSummary: '',
+        filePath: '/src/math.ts',
+        startLine: 1,
+        endLine: 1,
+        language: 'typescript',
+        metadata: {
+            chunkType: 'function',
+            name: 'add',
+            declarations: ['add'],
+            imports: [],
+            exports: ['add'],
+        },
+        ...overrides,
+    };
+}
+function createMockClient(generateFn) {
+    return { generate: generateFn };
+}
+describe('NLEnricher', () => {
+    describe('enrichChunk', () => {
+        it('should add nlSummary to the chunk on success', async () => {
+            const mockClient = createMockClient(vi.fn().mockResolvedValue(ok('Adds two numbers together.')));
+            const enricher = new NLEnricher(mockClient);
+            const result = await enricher.enrichChunk(makeChunk());
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value.nlSummary).toBe('Adds two numbers together.');
+                expect(result.value.id).toBe('chunk-1');
+                expect(result.value.content).toBe('function add(a: number, b: number): number { return a + b; }');
+            }
+        });
+        it('should trim whitespace from the summary', async () => {
+            const mockClient = createMockClient(vi.fn().mockResolvedValue(ok('  Adds two numbers.  \n')));
+            const enricher = new NLEnricher(mockClient);
+            const result = await enricher.enrichChunk(makeChunk());
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value.nlSummary).toBe('Adds two numbers.');
+            }
+        });
+        it('should build the correct prompt with the chunk language', async () => {
+            const generateFn = vi.fn().mockResolvedValue(ok('Summary.'));
+            const mockClient = createMockClient(generateFn);
+            const enricher = new NLEnricher(mockClient);
+            const chunk = makeChunk({ language: 'python' });
+            await enricher.enrichChunk(chunk);
+            expect(generateFn).toHaveBeenCalledWith(expect.stringContaining('Summarize this python code'));
+            expect(generateFn).toHaveBeenCalledWith(expect.stringContaining(chunk.content));
+        });
+        it('should return EnrichmentError when Ollama fails', async () => {
+            const mockClient = createMockClient(vi.fn().mockResolvedValue(err(new OllamaError('Connection refused'))));
+            const enricher = new NLEnricher(mockClient);
+            const result = await enricher.enrichChunk(makeChunk());
+            expect(result.isErr()).toBe(true);
+            if (result.isErr()) {
+                expect(result.error).toBeInstanceOf(EnrichmentError);
+                expect(result.error.message).toContain('chunk-1');
+                expect(result.error.message).toContain('Connection refused');
+            }
+        });
+    });
+    describe('enrichBatch', () => {
+        it('should process all chunks successfully', async () => {
+            let callCount = 0;
+            const mockClient = createMockClient(vi.fn().mockImplementation(() => {
+                callCount++;
+                return Promise.resolve(ok(`Summary ${callCount}.`));
+            }));
+            const enricher = new NLEnricher(mockClient);
+            const chunks = [
+                makeChunk({ id: 'c1' }),
+                makeChunk({ id: 'c2' }),
+                makeChunk({ id: 'c3' }),
+            ];
+            const result = await enricher.enrichBatch(chunks);
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value.enriched).toHaveLength(3);
+                expect(result.value.failedCount).toBe(0);
+                expect(result.value.enriched[0].nlSummary).toBe('Summary 1.');
+                expect(result.value.enriched[1].nlSummary).toBe('Summary 2.');
+                expect(result.value.enriched[2].nlSummary).toBe('Summary 3.');
+            }
+        });
+        it('should respect concurrency limit', async () => {
+            let maxConcurrent = 0;
+            let currentConcurrent = 0;
+            const mockClient = createMockClient(vi.fn().mockImplementation(() => {
+                currentConcurrent++;
+                if (currentConcurrent > maxConcurrent) {
+                    maxConcurrent = currentConcurrent;
+                }
+                return new Promise((resolve) => {
+                    setTimeout(() => {
+                        currentConcurrent--;
+                        resolve(ok('Summary.'));
+                    }, 10);
+                });
+            }));
+            const enricher = new NLEnricher(mockClient);
+            const chunks = Array.from({ length: 6 }, (_, i) => makeChunk({ id: `c${i}` }));
+            const result = await enricher.enrichBatch(chunks, 2);
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value.enriched).toHaveLength(6);
+                expect(result.value.failedCount).toBe(0);
+            }
+            expect(maxConcurrent).toBeLessThanOrEqual(2);
+        });
+        it('should return ok with empty array when input is empty', async () => {
+            const mockClient = createMockClient(vi.fn());
+            const enricher = new NLEnricher(mockClient);
+            const result = await enricher.enrichBatch([]);
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value.enriched).toEqual([]);
+                expect(result.value.failedCount).toBe(0);
+            }
+        });
+        it('should return partial results when some chunks fail', async () => {
+            let callCount = 0;
+            const mockClient = createMockClient(vi.fn().mockImplementation(() => {
+                callCount++;
+                if (callCount === 2) {
+                    return Promise.resolve(err(new OllamaError('Model not loaded')));
+                }
+                return Promise.resolve(ok('Summary.'));
+            }));
+            const enricher = new NLEnricher(mockClient);
+            const chunks = [
+                makeChunk({ id: 'c1' }),
+                makeChunk({ id: 'c2' }),
+                makeChunk({ id: 'c3' }),
+            ];
+            const result = await enricher.enrichBatch(chunks);
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value.enriched).toHaveLength(2);
+                expect(result.value.failedCount).toBe(1);
+                expect(result.value.enriched[0].id).toBe('c1');
+                expect(result.value.enriched[1].id).toBe('c3');
+            }
+        });
+    });
+});
+//# sourceMappingURL=nl-enricher.test.js.map

package/dist/enrichment/nl-enricher.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"nl-enricher.test.js","sourceRoot":"","sources":["../../src/enrichment/nl-enricher.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,YAAY,CAAC;AAErC,OAAO,EAAgB,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAE/D,SAAS,SAAS,CAAC,YAA4B,EAAE;IAC/C,OAAO;QACL,EAAE,EAAE,SAAS;QACb,OAAO,EAAE,8DAA8D;QACvE,SAAS,EAAE,EAAE;QACb,QAAQ,EAAE,cAAc;QACxB,SAAS,EAAE,CAAC;QACZ,OAAO,EAAE,CAAC;QACV,QAAQ,EAAE,YAAY;QACtB,QAAQ,EAAE;YACR,SAAS,EAAE,UAAU;YACrB,IAAI,EAAE,KAAK;YACX,YAAY,EAAE,CAAC,KAAK,CAAC;YACrB,OAAO,EAAE,EAAE;YACX,OAAO,EAAE,CAAC,KAAK,CAAC;SACjB;QACD,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CACvB,UAAoC;IAEpC,OAAO,EAAE,QAAQ,EAAE,UAAU,EAA6B,CAAC;AAC7D,CAAC;AAED,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,UAAU,GAAG,gBAAgB,CACjC,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,4BAA4B,CAAC,CAAC,CAC5D,CAAC;YACF,MAAM,QAAQ,GAAG,IAAI,UAAU,CAAC,UAAU,CAAC,CAAC;YAE5C,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,SAAS,EAAE,CAAC,CAAC;YAEvD,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;gBAClE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACxC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAC/B,8DAA8D,CAC/D,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,UAAU,GAAG,gBAAgB,CACjC,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,yBAAyB,CAAC,CAAC,CACzD,CAAC;YACF,MAAM,QAAQ,GAAG,IAAI,UAAU,CAAC,UAAU,CAAC,CAAC;YAE5C,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,SAAS,EAAE,CAAC,CAAC;YAEvD,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YAC7D,MAAM,UAAU,GAAG,gBAAgB,CAAC,UAAU,CAAC,CAAC;YAChD,MAAM,QAAQ,GAAG,IAAI,UAAU,CAAC,UAAU,CAAC,CAAC;YAE5C,MAAM,KAAK,GAAG,SAAS,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;YAChD,MAAM,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAElC,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,MAAM,CAAC,gBAAgB,CAAC,4BAA4B,CAAC,CACtD,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,MAAM,CAAC,gBAAgB,CAAC,KAAK,CAAC,OAAO,CAAC,CACvC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,UAAU,GAAG,gBAAgB,CACjC,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,WAAW,CAAC,oBAAoB,CAAC,CAAC,CAAC,CACtE,CAAC;YACF,MAAM,QAAQ,GAAG,IAAI,UAAU,CAAC,UAAU,CAAC,CAAC;YAE5C,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,SAAS,EAAE,CAAC,CAAC;YAEvD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;gBACrD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;gBAClD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,MAAM,UAAU,GAAG,gBAAgB,CACjC,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBAC9B,SAAS,EAAE,CAAC;gBACZ,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,WAAW,SAAS,GAAG,CAAC,CAAC,CAAC;YACtD,CAAC,CAAC,CACH,CAAC;YACF,MAAM,QAAQ,GAAG,IAAI,UAAU,CAAC,UAAU,CAAC,CAAC;YAE5C,MAAM,MAAM,GAAG;gBACb,SAAS,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;gBACvB,SAAS,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;gBACvB,SAAS,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;aACxB,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAElD,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;gBAC9C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBACzC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBAC/D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBAC/D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACjE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,IAAI,aAAa,GAAG,CAAC,CAAC;YACtB,IAAI,iBAAiB,GAAG,CAAC,CAAC;YAE1B,MAAM,UAAU,GAAG,gBAAgB,CACjC,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBAC9B,iBAAiB,EAAE,CAAC;gBACpB,IAAI,iBAAiB,GAAG,aAAa,EAAE,CAAC;oBACtC,aAAa,GAAG,iBAAiB,CAAC;gBACpC,CAAC;gBACD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;oBAC7B,UAAU,CAAC,GAAG,EAAE;wBACd,iBAAiB,EAAE,CAAC;wBACpB,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;oBAC1B,CAAC,EAAE,EAAE,CAAC,CAAC;gBACT,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CACH,CAAC;YACF,MAAM,QAAQ,GAAG,IAAI,UAAU,CAAC,UAAU,CAAC,CAAC;YAE5C,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAChD,SAAS,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAC3B,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAErD,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;gBAC9C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3C,CAAC;YACD,MAAM,CAAC,aAAa,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,UAAU,GAAG,gBAAgB,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,IAAI,UAAU,CAAC,UAAU,CAAC,CAAC;YAE5C,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YAE9C,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC1C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,MAAM,UAAU,GAAG,gBAAgB,CACjC,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBAC9B,SAAS,EAAE,CAAC;gBACZ,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;oBACpB,OAAO,OAAO,CAAC,OAAO,CACpB,GAAG,CAAC,IAAI,WAAW,CAAC,kBAAkB,CAAC,CAAC,CACzC,CAAC;gBACJ,CAAC;gBACD,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YACzC,CAAC,CAAC,CACH,CAAC;YACF,MAAM,QAAQ,GAAG,IAAI,UAAU,CAAC,UAAU,CAAC,CAAC;YAE5C,MAAM,MAAM,GAAG;gBACb,SAAS,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;gBACvB,SAAS,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;gBACvB,SAAS,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;aACxB,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAElD,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;gBAC9C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBACzC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAE,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAChD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAE,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/enrichment/ollama-client.d.ts

@@ -0,0 +1,18 @@
+import { type Result } from 'neverthrow';
+export interface OllamaConfig {
+    baseUrl: string;
+    model: string;
+    timeout: number;
+    /** Maximum tokens to generate per request (Ollama num_predict). 0 = unlimited. */
+    maxTokens: number;
+}
+export declare class OllamaError extends Error {
+    constructor(message: string);
+}
+export declare class OllamaClient {
+    private readonly config;
+    constructor(config?: Partial<OllamaConfig>);
+    get currentConfig(): OllamaConfig;
+    generate(prompt: string): Promise<Result<string, OllamaError>>;
+    isAvailable(): Promise<boolean>;
+}

package/dist/enrichment/ollama-client.js

@@ -0,0 +1,55 @@
+import { ok, err } from 'neverthrow';
+const DEFAULT_CONFIG = {
+    baseUrl: 'http://localhost:11434',
+    model: 'qwen2.5-coder:7b',
+    timeout: 60_000,
+    maxTokens: 100,
+};
+export class OllamaError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'OllamaError';
+    }
+}
+export class OllamaClient {
+    config;
+    constructor(config) {
+        this.config = { ...DEFAULT_CONFIG, ...config };
+    }
+    get currentConfig() {
+        return { ...this.config };
+    }
+    async generate(prompt) {
+        try {
+            const response = await globalThis.fetch(`${this.config.baseUrl}/api/generate`, {
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({
+                    model: this.config.model,
+                    prompt,
+                    stream: false,
+                    ...(this.config.maxTokens > 0 ? { options: { num_predict: this.config.maxTokens } } : {}),
+                }),
+                signal: AbortSignal.timeout(this.config.timeout),
+            });
+            if (!response.ok) {
+                return err(new OllamaError(`Ollama API returned status ${response.status}: ${response.statusText}`));
+            }
+            const data = (await response.json());
+            return ok(data.response);
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : 'Unknown error';
+            return err(new OllamaError(`Ollama request failed: ${message}`));
+        }
+    }
+    async isAvailable() {
+        try {
+            const response = await globalThis.fetch(`${this.config.baseUrl}/api/tags`, { signal: AbortSignal.timeout(this.config.timeout) });
+            return response.ok;
+        }
+        catch {
+            return false;
+        }
+    }
+}

package/dist/enrichment/ollama-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama-client.js","sourceRoot":"","sources":["../../src/enrichment/ollama-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,GAAG,EAAe,MAAM,YAAY,CAAC;AAUlD,MAAM,cAAc,GAAiB;IACnC,OAAO,EAAE,wBAAwB;IACjC,KAAK,EAAE,kBAAkB;IACzB,OAAO,EAAE,MAAM;IACf,SAAS,EAAE,GAAG;CACf,CAAC;AAEF,MAAM,OAAO,WAAY,SAAQ,KAAK;IACpC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,aAAa,CAAC;IAC5B,CAAC;CACF;AAMD,MAAM,OAAO,YAAY;IACN,MAAM,CAAe;IAEtC,YAAY,MAA8B;QACxC,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;IACjD,CAAC;IAED,IAAI,aAAa;QACf,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAc;QAC3B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,UAAU,CAAC,KAAK,CACrC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,eAAe,EACrC;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;oBACxB,MAAM;oBACN,MAAM,EAAE,KAAK;oBACb,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC1F,CAAC;gBACF,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;aACjD,CACF,CAAC;YAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,OAAO,GAAG,CACR,IAAI,WAAW,CACb,8BAA8B,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,EAAE,CACxE,CACF,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA2B,CAAC;YAC/D,OAAO,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YAC3D,OAAO,GAAG,CAAC,IAAI,WAAW,CAAC,0BAA0B,OAAO,EAAE,CAAC,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW;QACf,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,UAAU,CAAC,KAAK,CACrC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,WAAW,EACjC,EAAE,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CACrD,CAAC;YACF,OAAO,QAAQ,CAAC,EAAE,CAAC;QACrB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF"}

package/dist/enrichment/ollama-client.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/enrichment/ollama-client.test.js

@@ -0,0 +1,129 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { OllamaClient, OllamaError } from './ollama-client.js';
+describe('OllamaClient', () => {
+    const originalFetch = globalThis.fetch;
+    beforeEach(() => {
+        globalThis.fetch = vi.fn();
+    });
+    afterEach(() => {
+        globalThis.fetch = originalFetch;
+    });
+    describe('constructor / config', () => {
+        it('should use default config values when none provided', () => {
+            const client = new OllamaClient();
+            const config = client.currentConfig;
+            expect(config.baseUrl).toBe('http://localhost:11434');
+            expect(config.model).toBe('qwen2.5-coder:7b');
+            expect(config.timeout).toBe(60_000);
+            expect(config.maxTokens).toBe(100);
+        });
+        it('should accept custom config values', () => {
+            const client = new OllamaClient({
+                baseUrl: 'http://remote:9999',
+                model: 'llama3.2',
+                timeout: 60_000,
+                maxTokens: 200,
+            });
+            const config = client.currentConfig;
+            expect(config.baseUrl).toBe('http://remote:9999');
+            expect(config.model).toBe('llama3.2');
+            expect(config.timeout).toBe(60_000);
+            expect(config.maxTokens).toBe(200);
+        });
+        it('should merge partial config with defaults', () => {
+            const client = new OllamaClient({ model: 'codellama' });
+            const config = client.currentConfig;
+            expect(config.baseUrl).toBe('http://localhost:11434');
+            expect(config.model).toBe('codellama');
+            expect(config.timeout).toBe(60_000);
+        });
+    });
+    describe('generate', () => {
+        it('should return the response on success', async () => {
+            const mockResponse = {
+                ok: true,
+                status: 200,
+                statusText: 'OK',
+                json: vi.fn().mockResolvedValue({ response: 'This function adds two numbers.' }),
+            };
+            vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
+            const client = new OllamaClient();
+            const result = await client.generate('Summarize this code');
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value).toBe('This function adds two numbers.');
+            }
+            expect(globalThis.fetch).toHaveBeenCalledWith('http://localhost:11434/api/generate', expect.objectContaining({
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({
+                    model: 'qwen2.5-coder:7b',
+                    prompt: 'Summarize this code',
+                    stream: false,
+                    options: { num_predict: 100 },
+                }),
+            }));
+        });
+        it('should return OllamaError on non-200 status', async () => {
+            const mockResponse = {
+                ok: false,
+                status: 500,
+                statusText: 'Internal Server Error',
+            };
+            vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
+            const client = new OllamaClient();
+            const result = await client.generate('Summarize this code');
+            expect(result.isErr()).toBe(true);
+            if (result.isErr()) {
+                expect(result.error).toBeInstanceOf(OllamaError);
+                expect(result.error.message).toContain('status 500');
+                expect(result.error.message).toContain('Internal Server Error');
+            }
+        });
+        it('should return OllamaError on network error', async () => {
+            vi.mocked(globalThis.fetch).mockRejectedValue(new Error('ECONNREFUSED'));
+            const client = new OllamaClient();
+            const result = await client.generate('Summarize this code');
+            expect(result.isErr()).toBe(true);
+            if (result.isErr()) {
+                expect(result.error).toBeInstanceOf(OllamaError);
+                expect(result.error.message).toContain('Ollama request failed');
+                expect(result.error.message).toContain('ECONNREFUSED');
+            }
+        });
+        it('should handle non-Error throw gracefully', async () => {
+            vi.mocked(globalThis.fetch).mockRejectedValue('string error');
+            const client = new OllamaClient();
+            const result = await client.generate('Summarize this code');
+            expect(result.isErr()).toBe(true);
+            if (result.isErr()) {
+                expect(result.error).toBeInstanceOf(OllamaError);
+                expect(result.error.message).toContain('Unknown error');
+            }
+        });
+    });
+    describe('isAvailable', () => {
+        it('should return true on 200 response', async () => {
+            const mockResponse = { ok: true, status: 200 };
+            vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
+            const client = new OllamaClient();
+            const available = await client.isAvailable();
+            expect(available).toBe(true);
+            expect(globalThis.fetch).toHaveBeenCalledWith('http://localhost:11434/api/tags', expect.objectContaining({ signal: expect.any(AbortSignal) }));
+        });
+        it('should return false on network error', async () => {
+            vi.mocked(globalThis.fetch).mockRejectedValue(new Error('ECONNREFUSED'));
+            const client = new OllamaClient();
+            const available = await client.isAvailable();
+            expect(available).toBe(false);
+        });
+        it('should return false on non-200 response', async () => {
+            const mockResponse = { ok: false, status: 503 };
+            vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
+            const client = new OllamaClient();
+            const available = await client.isAvailable();
+            expect(available).toBe(false);
+        });
+    });
+});
+//# sourceMappingURL=ollama-client.test.js.map

package/dist/enrichment/ollama-client.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama-client.test.js","sourceRoot":"","sources":["../../src/enrichment/ollama-client.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,YAAY,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAE/D,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,MAAM,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC;IAEvC,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,CAAC,KAAK,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,UAAU,CAAC,KAAK,GAAG,aAAa,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,MAAM,CAAC,aAAa,CAAC;YAEpC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACtD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC;gBAC9B,OAAO,EAAE,oBAAoB;gBAC7B,KAAK,EAAE,UAAU;gBACjB,OAAO,EAAE,MAAM;gBACf,SAAS,EAAE,GAAG;aACf,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,CAAC,aAAa,CAAC;YAEpC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;YACxD,MAAM,MAAM,GAAG,MAAM,CAAC,aAAa,CAAC;YAEpC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACtD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,IAAI;gBACR,MAAM,EAAE,GAAG;gBACX,UAAU,EAAE,IAAI;gBAChB,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,iCAAiC,EAAE,CAAC;aACjF,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAAC,YAAmC,CAAC,CAAC;YAEnF,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;YAC/D,CAAC;YAED,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAC3C,qCAAqC,EACrC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,kBAAkB;oBACzB,MAAM,EAAE,qBAAqB;oBAC7B,MAAM,EAAE,KAAK;oBACb,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE;iBAC9B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,GAAG;gBACX,UAAU,EAAE,uBAAuB;aACpC,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAAC,YAAmC,CAAC,CAAC;YAEnF,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;gBACjD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;gBACrD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;YAClE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;YAEzE,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;gBACjD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;gBAChE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YACzD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAE9D,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;gBACjD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YAC1D,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,YAAY,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;YAC/C,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAAC,YAAmC,CAAC,CAAC;YAEnF,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAClC,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;YAE7C,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAC3C,iCAAiC,EACjC,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC,CAC7D,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;YAEzE,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAClC,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;YAE7C,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,YAAY,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;YAChD,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAAC,YAAmC,CAAC,CAAC;YAEnF,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAClC,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;YAE7C,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/git/git-client.d.ts

@@ -0,0 +1,22 @@
+import type { Result } from 'neverthrow';
+export type FileChangeStatus = 'added' | 'modified' | 'deleted' | 'renamed';
+export interface FileChange {
+    filePath: string;
+    status: FileChangeStatus;
+    oldPath?: string;
+}
+export interface FileMetadata {
+    filePath: string;
+    lastModified: Date;
+    author: string;
+    commitHash: string;
+}
+export declare class GitError extends Error {
+    constructor(message: string);
+}
+export interface GitClient {
+    getChangedFiles(since?: string): Promise<Result<FileChange[], GitError>>;
+    getFileMetadata(filePath: string): Promise<Result<FileMetadata, GitError>>;
+    getCurrentCommit(): Promise<Result<string, GitError>>;
+    isGitRepo(dir: string): Promise<Result<boolean, GitError>>;
+}

package/dist/git/git-client.js

@@ -0,0 +1,6 @@
+export class GitError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'GitError';
+    }
+}

package/dist/git/git-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"git-client.js","sourceRoot":"","sources":["../../src/git/git-client.ts"],"names":[],"mappings":"AAiBA,MAAM,OAAO,QAAS,SAAQ,KAAK;IACjC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC;IACzB,CAAC;CACF"}

package/dist/git/git-client.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/git/git-client.test.js

@@ -0,0 +1,200 @@
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { mkdtempSync, writeFileSync, rmSync } from 'node:fs';
+import { join } from 'node:path';
+import { tmpdir } from 'node:os';
+import { simpleGit } from 'simple-git';
+import { SimpleGitClient } from './simple-git-client.js';
+import { GitError } from './git-client.js';
+describe('SimpleGitClient', () => {
+    let tempDir;
+    let client;
+    beforeEach(async () => {
+        tempDir = mkdtempSync(join(tmpdir(), 'coderag-git-test-'));
+        const git = simpleGit(tempDir);
+        await git.init();
+        await git.addConfig('user.email', 'test@coderag.dev');
+        await git.addConfig('user.name', 'Test Author');
+        // Disable GPG/SSH signing — avoids 1Password timeouts in CI/test
+        await git.addConfig('commit.gpgSign', 'false');
+        writeFileSync(join(tempDir, 'initial.txt'), 'initial content');
+        await git.add('initial.txt');
+        await git.commit('Initial commit');
+        client = new SimpleGitClient(tempDir);
+    });
+    afterEach(() => {
+        rmSync(tempDir, { recursive: true, force: true });
+    });
+    describe('isGitRepo', () => {
+        it('should return true for a git repository', async () => {
+            const result = await client.isGitRepo(tempDir);
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value).toBe(true);
+            }
+        });
+        it('should return false for a non-git directory', async () => {
+            const nonGitDir = mkdtempSync(join(tmpdir(), 'coderag-nogit-'));
+            try {
+                const result = await client.isGitRepo(nonGitDir);
+                expect(result.isOk()).toBe(true);
+                if (result.isOk()) {
+                    expect(result.value).toBe(false);
+                }
+            }
+            finally {
+                rmSync(nonGitDir, { recursive: true, force: true });
+            }
+        });
+    });
+    describe('getCurrentCommit', () => {
+        it('should return the current commit hash', async () => {
+            const result = await client.getCurrentCommit();
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value).toMatch(/^[0-9a-f]{40}$/);
+            }
+        });
+        it('should return an error for a directory without commits', async () => {
+            const emptyDir = mkdtempSync(join(tmpdir(), 'coderag-empty-'));
+            try {
+                const git = simpleGit(emptyDir);
+                await git.init();
+                const emptyClient = new SimpleGitClient(emptyDir);
+                const result = await emptyClient.getCurrentCommit();
+                expect(result.isErr()).toBe(true);
+                if (result.isErr()) {
+                    expect(result.error).toBeInstanceOf(GitError);
+                }
+            }
+            finally {
+                rmSync(emptyDir, { recursive: true, force: true });
+            }
+        });
+    });
+    describe('getChangedFiles', () => {
+        it('should list all tracked files when since is undefined', async () => {
+            const result = await client.getChangedFiles();
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                const filePaths = result.value.map((f) => f.filePath);
+                expect(filePaths).toContain('initial.txt');
+            }
+        });
+        it('should detect added files since a commit', async () => {
+            const git = simpleGit(tempDir);
+            const beforeCommit = (await git.revparse(['HEAD'])).trim();
+            writeFileSync(join(tempDir, 'newfile.ts'), 'export const x = 1;');
+            await git.add('newfile.ts');
+            await git.commit('Add new file');
+            const result = await client.getChangedFiles(beforeCommit);
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value).toHaveLength(1);
+                expect(result.value[0]?.filePath).toBe('newfile.ts');
+                expect(result.value[0]?.status).toBe('added');
+            }
+        });
+        it('should detect modified files since a commit', async () => {
+            const git = simpleGit(tempDir);
+            const beforeCommit = (await git.revparse(['HEAD'])).trim();
+            writeFileSync(join(tempDir, 'initial.txt'), 'modified content');
+            await git.add('initial.txt');
+            await git.commit('Modify file');
+            const result = await client.getChangedFiles(beforeCommit);
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value).toHaveLength(1);
+                expect(result.value[0]?.filePath).toBe('initial.txt');
+                expect(result.value[0]?.status).toBe('modified');
+            }
+        });
+        it('should detect deleted files since a commit', async () => {
+            const git = simpleGit(tempDir);
+            const beforeCommit = (await git.revparse(['HEAD'])).trim();
+            await git.rm('initial.txt');
+            await git.commit('Delete file');
+            const result = await client.getChangedFiles(beforeCommit);
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value).toHaveLength(1);
+                expect(result.value[0]?.filePath).toBe('initial.txt');
+                expect(result.value[0]?.status).toBe('deleted');
+            }
+        });
+        it('should return empty array when no changes since commit', async () => {
+            const git = simpleGit(tempDir);
+            const currentCommit = (await git.revparse(['HEAD'])).trim();
+            const result = await client.getChangedFiles(currentCommit);
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value).toHaveLength(0);
+            }
+        });
+        it('should return error for invalid since reference', async () => {
+            const result = await client.getChangedFiles('invalid-ref-abc123');
+            expect(result.isErr()).toBe(true);
+            if (result.isErr()) {
+                expect(result.error).toBeInstanceOf(GitError);
+                expect(result.error.message).toContain('Failed to get changed files');
+            }
+        });
+    });
+    describe('getFileMetadata', () => {
+        it('should return metadata for a committed file', async () => {
+            const result = await client.getFileMetadata('initial.txt');
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value.filePath).toBe('initial.txt');
+                expect(result.value.author).toBe('Test Author');
+                expect(result.value.commitHash).toMatch(/^[0-9a-f]{40}$/);
+                expect(result.value.lastModified).toBeInstanceOf(Date);
+                expect(result.value.lastModified.getTime()).not.toBeNaN();
+            }
+        });
+        it('should return error for a file with no git history', async () => {
+            const result = await client.getFileMetadata('nonexistent-file.ts');
+            expect(result.isErr()).toBe(true);
+            if (result.isErr()) {
+                expect(result.error).toBeInstanceOf(GitError);
+                expect(result.error.message).toContain('No git history found');
+            }
+        });
+        it('should return metadata from the latest commit for a file', async () => {
+            const git = simpleGit(tempDir);
+            writeFileSync(join(tempDir, 'initial.txt'), 'updated content');
+            await git.add('initial.txt');
+            await git.commit('Update initial file');
+            const currentHash = (await git.revparse(['HEAD'])).trim();
+            const result = await client.getFileMetadata('initial.txt');
+            expect(result.isOk()).toBe(true);
+            if (result.isOk()) {
+                expect(result.value.commitHash).toBe(currentHash);
+            }
+        });
+        it('should reject path traversal attempts', async () => {
+            const result = await client.getFileMetadata('../../etc/passwd');
+            expect(result.isErr()).toBe(true);
+            if (result.isErr()) {
+                expect(result.error).toBeInstanceOf(GitError);
+                expect(result.error.message).toContain('Path escapes working directory');
+            }
+        });
+    });
+    describe('getChangedFiles security', () => {
+        it('should reject invalid git references', async () => {
+            const result = await client.getChangedFiles('--option-injection');
+            expect(result.isErr()).toBe(true);
+            if (result.isErr()) {
+                expect(result.error).toBeInstanceOf(GitError);
+                expect(result.error.message).toContain('Invalid git reference');
+            }
+        });
+        it('should accept valid hex commit hashes', async () => {
+            const git = simpleGit(tempDir);
+            const hash = (await git.revparse(['HEAD'])).trim();
+            const result = await client.getChangedFiles(hash);
+            expect(result.isOk()).toBe(true);
+        });
+    });
+});
+//# sourceMappingURL=git-client.test.js.map