@eucoder/rag 0.2.0

package/dist/hybrid-search.js

@@ -0,0 +1,70 @@
+/**
+ * Hybrid Search RAG - Combina semantic search e keyword search usando RRF (Reciprocal Rank Fusion)
+ *
+ * RRF Formula: score(d) = Σ 1/(k + rank_i(d))
+ * dove k è una costante (tipicamente 60) e rank_i è la posizione del documento nella lista i
+ */
+export class HybridSearchRag {
+    options;
+    constructor(options) {
+        this.options = {
+            semanticWeight: options.semanticWeight ?? 0.5,
+            keywordWeight: options.keywordWeight ?? 0.5,
+            rrfK: options.rrfK ?? 60,
+            topK: options.topK ?? 10,
+            ...options,
+        };
+    }
+    async search(query, topK) {
+        const limit = topK ?? this.options.topK;
+        // Esegui entrambe le ricerche in parallelo
+        const [semanticHits, keywordHits] = await Promise.all([
+            this.options.semanticRetriever.search(query, limit * 2),
+            this.options.keywordRetriever.search(query, limit * 2),
+        ]);
+        // Applica RRF fusion
+        const fusedHits = this.reciprocalRankFusion(semanticHits, keywordHits);
+        // Prendi i top K risultati
+        const finalHits = fusedHits.slice(0, limit);
+        return {
+            hits: finalHits,
+            fusionMethod: "rrf",
+            semanticCount: semanticHits.length,
+            keywordCount: keywordHits.length,
+        };
+    }
+    reciprocalRankFusion(semanticHits, keywordHits) {
+        const scoreMap = new Map();
+        const k = this.options.rrfK;
+        // Calcola RRF score per semantic hits
+        semanticHits.forEach((hit, index) => {
+            const key = `${hit.path}:${hit.startLine}`;
+            const rrfScore = 1 / (k + index + 1);
+            const existing = scoreMap.get(key);
+            if (existing) {
+                existing.score += rrfScore * this.options.semanticWeight;
+            }
+            else {
+                scoreMap.set(key, { hit, score: rrfScore * this.options.semanticWeight });
+            }
+        });
+        // Calcola RRF score per keyword hits
+        keywordHits.forEach((hit, index) => {
+            const key = `${hit.path}:${hit.startLine}`;
+            const rrfScore = 1 / (k + index + 1);
+            const existing = scoreMap.get(key);
+            if (existing) {
+                existing.score += rrfScore * this.options.keywordWeight;
+            }
+            else {
+                scoreMap.set(key, { hit, score: rrfScore * this.options.keywordWeight });
+            }
+        });
+        // Ordina per score decrescente
+        const sorted = Array.from(scoreMap.values())
+            .sort((a, b) => b.score - a.score)
+            .map(({ hit, score }) => ({ ...hit, score }));
+        return sorted;
+    }
+}
+//# sourceMappingURL=hybrid-search.js.map

package/dist/hybrid-search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hybrid-search.js","sourceRoot":"","sources":["../src/hybrid-search.ts"],"names":[],"mappings":"AAGA;;;;;GAKG;AACH,MAAM,OAAO,eAAe;IAClB,OAAO,CAAgC;IAE/C,YAAY,OAA4B;QACtC,IAAI,CAAC,OAAO,GAAG;YACb,cAAc,EAAE,OAAO,CAAC,cAAc,IAAI,GAAG;YAC7C,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,GAAG;YAC3C,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE;YACxB,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE;YACxB,GAAG,OAAO;SACX,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,IAAa;QACvC,MAAM,KAAK,GAAG,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;QAExC,2CAA2C;QAC3C,MAAM,CAAC,YAAY,EAAE,WAAW,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACpD,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,GAAG,CAAC,CAAC;YACvD,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,GAAG,CAAC,CAAC;SACvD,CAAC,CAAC;QAEH,qBAAqB;QACrB,MAAM,SAAS,GAAG,IAAI,CAAC,oBAAoB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;QAEvE,2BAA2B;QAC3B,MAAM,SAAS,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QAE5C,OAAO;YACL,IAAI,EAAE,SAAS;YACf,YAAY,EAAE,KAAK;YACnB,aAAa,EAAE,YAAY,CAAC,MAAM;YAClC,YAAY,EAAE,WAAW,CAAC,MAAM;SACjC,CAAC;IACJ,CAAC;IAEO,oBAAoB,CAC1B,YAAyB,EACzB,WAAwB;QAExB,MAAM,QAAQ,GAAG,IAAI,GAAG,EAA6C,CAAC;QACtE,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;QAE5B,sCAAsC;QACtC,YAAY,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;YAClC,MAAM,GAAG,GAAG,GAAG,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;YAC3C,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC;YAErC,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACnC,IAAI,QAAQ,EAAE,CAAC;gBACb,QAAQ,CAAC,KAAK,IAAI,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC;YAC3D,CAAC;iBAAM,CAAC;gBACN,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;YAC5E,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,qCAAqC;QACrC,WAAW,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;YACjC,MAAM,GAAG,GAAG,GAAG,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;YAC3C,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC;YAErC,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACnC,IAAI,QAAQ,EAAE,CAAC;gBACb,QAAQ,CAAC,KAAK,IAAI,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;YAC1D,CAAC;iBAAM,CAAC;gBACN,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;YAC3E,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,+BAA+B;QAC/B,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;aACzC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;aACjC,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QAEhD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF"}

package/dist/hybrid-search.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=hybrid-search.test.d.ts.map

package/dist/hybrid-search.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"hybrid-search.test.d.ts","sourceRoot":"","sources":["../src/hybrid-search.test.ts"],"names":[],"mappings":""}

package/dist/hybrid-search.test.js

@@ -0,0 +1,93 @@
+import { describe, it, expect } from "vitest";
+import { HybridSearchRag } from "./hybrid-search.js";
+class MockSemanticRetriever {
+    async search(_query, topK) {
+        return [
+            {
+                id: "hit-1",
+                path: "src/auth.ts",
+                startLine: 10,
+                endLine: 20,
+                text: "Authentication logic",
+                score: 0.9,
+            },
+            {
+                id: "hit-2",
+                path: "src/login.ts",
+                startLine: 5,
+                endLine: 15,
+                text: "Login implementation",
+                score: 0.8,
+            },
+        ].slice(0, topK);
+    }
+}
+class MockKeywordRetriever {
+    async search(_query, topK) {
+        return [
+            {
+                id: "hit-2",
+                path: "src/login.ts",
+                startLine: 5,
+                endLine: 15,
+                text: "Login implementation",
+                score: 0.85,
+            },
+            {
+                id: "hit-3",
+                path: "src/session.ts",
+                startLine: 1,
+                endLine: 10,
+                text: "Session management",
+                score: 0.7,
+            },
+        ].slice(0, topK);
+    }
+}
+describe("HybridSearchRag", () => {
+    it("should combine semantic and keyword results using RRF", async () => {
+        const hybrid = new HybridSearchRag({
+            semanticRetriever: new MockSemanticRetriever(),
+            keywordRetriever: new MockKeywordRetriever(),
+            topK: 5,
+        });
+        const result = await hybrid.search("authentication");
+        expect(result.hits.length).toBeGreaterThan(0);
+        expect(result.fusionMethod).toBe("rrf");
+        expect(result.semanticCount).toBe(2);
+        expect(result.keywordCount).toBe(2);
+    });
+    it("should rank overlapping results higher", async () => {
+        const hybrid = new HybridSearchRag({
+            semanticRetriever: new MockSemanticRetriever(),
+            keywordRetriever: new MockKeywordRetriever(),
+            topK: 5,
+        });
+        const result = await hybrid.search("login");
+        // src/login.ts appears in both, should be ranked higher
+        const loginHit = result.hits.find((h) => h.path === "src/login.ts");
+        expect(loginHit).toBeDefined();
+        expect(result.hits[0]?.path).toBe("src/login.ts");
+    });
+    it("should respect custom weights", async () => {
+        const hybridSemantic = new HybridSearchRag({
+            semanticRetriever: new MockSemanticRetriever(),
+            keywordRetriever: new MockKeywordRetriever(),
+            semanticWeight: 0.9,
+            keywordWeight: 0.1,
+            topK: 5,
+        });
+        const resultSemantic = await hybridSemantic.search("auth");
+        const hybridKeyword = new HybridSearchRag({
+            semanticRetriever: new MockSemanticRetriever(),
+            keywordRetriever: new MockKeywordRetriever(),
+            semanticWeight: 0.1,
+            keywordWeight: 0.9,
+            topK: 5,
+        });
+        const resultKeyword = await hybridKeyword.search("auth");
+        // Results should differ based on weights
+        expect(resultSemantic.hits[0]?.score).not.toBe(resultKeyword.hits[0]?.score);
+    });
+});
+//# sourceMappingURL=hybrid-search.test.js.map

package/dist/hybrid-search.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hybrid-search.test.js","sourceRoot":"","sources":["../src/hybrid-search.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAIrD,MAAM,qBAAqB;IACzB,KAAK,CAAC,MAAM,CAAC,MAAc,EAAE,IAAY;QACvC,OAAO;YACL;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,aAAa;gBACnB,SAAS,EAAE,EAAE;gBACb,OAAO,EAAE,EAAE;gBACX,IAAI,EAAE,sBAAsB;gBAC5B,KAAK,EAAE,GAAG;aACX;YACD;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,cAAc;gBACpB,SAAS,EAAE,CAAC;gBACZ,OAAO,EAAE,EAAE;gBACX,IAAI,EAAE,sBAAsB;gBAC5B,KAAK,EAAE,GAAG;aACX;SACF,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACnB,CAAC;CACF;AAED,MAAM,oBAAoB;IACxB,KAAK,CAAC,MAAM,CAAC,MAAc,EAAE,IAAY;QACvC,OAAO;YACL;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,cAAc;gBACpB,SAAS,EAAE,CAAC;gBACZ,OAAO,EAAE,EAAE;gBACX,IAAI,EAAE,sBAAsB;gBAC5B,KAAK,EAAE,IAAI;aACZ;YACD;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,gBAAgB;gBACtB,SAAS,EAAE,CAAC;gBACZ,OAAO,EAAE,EAAE;gBACX,IAAI,EAAE,oBAAoB;gBAC1B,KAAK,EAAE,GAAG;aACX;SACF,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACnB,CAAC;CACF;AAED,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC;YACjC,iBAAiB,EAAE,IAAI,qBAAqB,EAAE;YAC9C,gBAAgB,EAAE,IAAI,oBAAoB,EAAE;YAC5C,IAAI,EAAE,CAAC;SACR,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAErD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC;YACjC,iBAAiB,EAAE,IAAI,qBAAqB,EAAE;YAC9C,gBAAgB,EAAE,IAAI,oBAAoB,EAAE;YAC5C,IAAI,EAAE,CAAC;SACR,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAE5C,wDAAwD;QACxD,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC;QACpE,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,cAAc,GAAG,IAAI,eAAe,CAAC;YACzC,iBAAiB,EAAE,IAAI,qBAAqB,EAAE;YAC9C,gBAAgB,EAAE,IAAI,oBAAoB,EAAE;YAC5C,cAAc,EAAE,GAAG;YACnB,aAAa,EAAE,GAAG;YAClB,IAAI,EAAE,CAAC;SACR,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG,MAAM,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAE3D,MAAM,aAAa,GAAG,IAAI,eAAe,CAAC;YACxC,iBAAiB,EAAE,IAAI,qBAAqB,EAAE;YAC9C,gBAAgB,EAAE,IAAI,oBAAoB,EAAE;YAC5C,cAAc,EAAE,GAAG;YACnB,aAAa,EAAE,GAAG;YAClB,IAAI,EAAE,CAAC;SACR,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAEzD,yCAAyC;QACzC,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAC/E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,17 @@
+export { CorrectiveRag, createCorrectiveRag } from "./corrective-rag.js";
+export { LlmGrader, LlmQueryRewriter } from "./llm-grader.js";
+export { HybridSearchRag } from "./hybrid-search.js";
+export { KnowledgeGraphRag, InMemoryKnowledgeGraph } from "./knowledge-graph.js";
+export { AgenticRag } from "./agentic-rag.js";
+export { InMemoryFeedbackStorage } from "./feedback.js";
+export { MetricsCalculator } from "./metrics.js";
+export { ABTestFramework } from "./ab-testing.js";
+export { ParameterOptimizer } from "./optimizer.js";
+export { SelfImprovingRag } from "./self-improving.js";
+export type { LlmProvider } from "./llm-grader.js";
+export type { Retriever, KeywordRetriever, Grader, GraderResult, QueryRewriter, CorrectiveRagOptions, RagAnswer, RagCitation, HybridSearchOptions, HybridSearchResult, Entity, Relation, KnowledgeGraph, EntityExtractor, RelationExtractor, KnowledgeGraphRagOptions, KnowledgeGraphRagResult, ReasoningStep, AgenticRagOptions, AgenticRagResult, } from "./types.js";
+export type { RagFeedback, RagMetrics, ABTestResult, OptimizationSuggestion, FeedbackStorage, } from "./feedback.js";
+export type { RagStrategyConfig, } from "./ab-testing.js";
+export type { OptimizableParams, } from "./optimizer.js";
+export type { SelfImprovingConfig, } from "./self-improving.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AACzE,OAAO,EAAE,SAAS,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AACjF,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAG9C,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACjD,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAEvD,YAAY,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACnD,YAAY,EACV,SAAS,EACT,gBAAgB,EAChB,MAAM,EACN,YAAY,EACZ,aAAa,EACb,oBAAoB,EACpB,SAAS,EACT,WAAW,EACX,mBAAmB,EACnB,kBAAkB,EAClB,MAAM,EACN,QAAQ,EACR,cAAc,EACd,eAAe,EACf,iBAAiB,EACjB,wBAAwB,EACxB,uBAAuB,EACvB,aAAa,EACb,iBAAiB,EACjB,gBAAgB,GACjB,MAAM,YAAY,CAAC;AAGpB,YAAY,EACV,WAAW,EACX,UAAU,EACV,YAAY,EACZ,sBAAsB,EACtB,eAAe,GAChB,MAAM,eAAe,CAAC;AAEvB,YAAY,EACV,iBAAiB,GAClB,MAAM,iBAAiB,CAAC;AAEzB,YAAY,EACV,iBAAiB,GAClB,MAAM,gBAAgB,CAAC;AAExB,YAAY,EACV,mBAAmB,GACpB,MAAM,qBAAqB,CAAC"}

package/dist/index.js

@@ -0,0 +1,12 @@
+export { CorrectiveRag, createCorrectiveRag } from "./corrective-rag.js";
+export { LlmGrader, LlmQueryRewriter } from "./llm-grader.js";
+export { HybridSearchRag } from "./hybrid-search.js";
+export { KnowledgeGraphRag, InMemoryKnowledgeGraph } from "./knowledge-graph.js";
+export { AgenticRag } from "./agentic-rag.js";
+// Fase 4: Self-Improving
+export { InMemoryFeedbackStorage } from "./feedback.js";
+export { MetricsCalculator } from "./metrics.js";
+export { ABTestFramework } from "./ab-testing.js";
+export { ParameterOptimizer } from "./optimizer.js";
+export { SelfImprovingRag } from "./self-improving.js";
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AACzE,OAAO,EAAE,SAAS,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AACjF,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C,yBAAyB;AACzB,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACjD,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC"}

package/dist/knowledge-graph.d.ts

@@ -0,0 +1,24 @@
+import type { Entity, Relation, KnowledgeGraph, KnowledgeGraphRagOptions, KnowledgeGraphRagResult } from "./types.js";
+/**
+ * Implementazione base di Knowledge Graph in memoria
+ */
+export declare class InMemoryKnowledgeGraph implements KnowledgeGraph {
+    entities: Map<string, Entity>;
+    relations: Relation[];
+    addEntity(entity: Entity): void;
+    addRelation(relation: Relation): void;
+    getEntity(id: string): Entity | undefined;
+    getRelations(entityId: string): Relation[];
+    getConnectedEntities(entityId: string, depth?: number): Entity[];
+}
+/**
+ * Knowledge Graph RAG - Arricchisce i risultati con contesto da un grafo di conoscenza
+ */
+export declare class KnowledgeGraphRag {
+    private options;
+    constructor(options: KnowledgeGraphRagOptions);
+    search(query: string): Promise<KnowledgeGraphRagResult>;
+    private buildGraphContext;
+    private buildAnswer;
+}
+//# sourceMappingURL=knowledge-graph.d.ts.map

package/dist/knowledge-graph.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"knowledge-graph.d.ts","sourceRoot":"","sources":["../src/knowledge-graph.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,MAAM,EACN,QAAQ,EACR,cAAc,EACd,wBAAwB,EACxB,uBAAuB,EAExB,MAAM,YAAY,CAAC;AAEpB;;GAEG;AACH,qBAAa,sBAAuB,YAAW,cAAc;IAC3D,QAAQ,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAa;IAC1C,SAAS,EAAE,QAAQ,EAAE,CAAM;IAE3B,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAI/B,WAAW,CAAC,QAAQ,EAAE,QAAQ,GAAG,IAAI;IAIrC,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAIzC,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,QAAQ,EAAE;IAM1C,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,GAAE,MAAU,GAAG,MAAM,EAAE;CAyBpE;AAED;;GAEG;AACH,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,OAAO,CAAqC;gBAExC,OAAO,EAAE,wBAAwB;IAQvC,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,uBAAuB,CAAC;IAsC7D,OAAO,CAAC,iBAAiB;IAkCzB,OAAO,CAAC,WAAW;CAiCpB"}

package/dist/knowledge-graph.js

@@ -0,0 +1,131 @@
+/**
+ * Implementazione base di Knowledge Graph in memoria
+ */
+export class InMemoryKnowledgeGraph {
+    entities = new Map();
+    relations = [];
+    addEntity(entity) {
+        this.entities.set(entity.id, entity);
+    }
+    addRelation(relation) {
+        this.relations.push(relation);
+    }
+    getEntity(id) {
+        return this.entities.get(id);
+    }
+    getRelations(entityId) {
+        return this.relations.filter((r) => r.source === entityId || r.target === entityId);
+    }
+    getConnectedEntities(entityId, depth = 1) {
+        const visited = new Set();
+        const result = [];
+        const traverse = (currentId, currentDepth) => {
+            if (currentDepth > depth || visited.has(currentId))
+                return;
+            visited.add(currentId);
+            const entity = this.getEntity(currentId);
+            if (entity && currentId !== entityId) {
+                result.push(entity);
+            }
+            if (currentDepth < depth) {
+                const relations = this.getRelations(currentId);
+                for (const rel of relations) {
+                    const nextId = rel.source === currentId ? rel.target : rel.source;
+                    traverse(nextId, currentDepth + 1);
+                }
+            }
+        };
+        traverse(entityId, 0);
+        return result;
+    }
+}
+/**
+ * Knowledge Graph RAG - Arricchisce i risultati con contesto da un grafo di conoscenza
+ */
+export class KnowledgeGraphRag {
+    options;
+    constructor(options) {
+        this.options = {
+            topK: options.topK ?? 10,
+            maxDepth: options.maxDepth ?? 2,
+            ...options,
+        };
+    }
+    async search(query) {
+        // 1. Recupera risultati dal retriever
+        const hits = await this.options.retriever.search(query, this.options.topK);
+        // 2. Estrai entità dai risultati
+        const allText = hits.map((h) => h.text).join("\n");
+        const entities = await this.options.entityExtractor.extract(allText);
+        // 3. Aggiungi entità al grafo
+        entities.forEach((e) => this.options.graph.addEntity(e));
+        // 4. Estrai relazioni
+        const relations = await this.options.relationExtractor.extract(allText, entities);
+        relations.forEach((r) => this.options.graph.addRelation(r));
+        // 5. Arricchisci con contesto del grafo
+        const graphContext = this.buildGraphContext(entities);
+        // 6. Genera citazioni
+        const citations = hits.map((hit) => ({
+            path: hit.path,
+            startLine: hit.startLine,
+            endLine: hit.endLine,
+            text: hit.text,
+        }));
+        // 7. Costruisci risposta
+        const answer = this.buildAnswer(query, hits, entities, relations, graphContext);
+        return {
+            answer,
+            citations,
+            entities,
+            relations,
+            graphContext,
+        };
+    }
+    buildGraphContext(entities) {
+        const contextParts = [];
+        for (const entity of entities) {
+            const connected = this.options.graph.getConnectedEntities(entity.id, this.options.maxDepth);
+            const relations = this.options.graph.getRelations(entity.id);
+            if (connected.length > 0 || relations.length > 0) {
+                let part = `**${entity.name}** (${entity.type})`;
+                if (relations.length > 0) {
+                    const relDescs = relations.map((r) => {
+                        const otherId = r.source === entity.id ? r.target : r.source;
+                        const other = this.options.graph.getEntity(otherId);
+                        return `${r.type} → ${other?.name ?? otherId}`;
+                    });
+                    part += `\n  Relazioni: ${relDescs.join(", ")}`;
+                }
+                if (connected.length > 0) {
+                    const connNames = connected.map((e) => e.name).join(", ");
+                    part += `\n  Connesso a: ${connNames}`;
+                }
+                contextParts.push(part);
+            }
+        }
+        return contextParts.join("\n\n");
+    }
+    buildAnswer(_query, hits, entities, _relations, graphContext) {
+        const parts = [];
+        parts.push(`## Risposta\n`);
+        parts.push(`Basato sull'analisi di ${hits.length} documenti e ${entities.length} entità identificate:\n`);
+        if (graphContext) {
+            parts.push(`### Contesto dal Knowledge Graph\n${graphContext}\n`);
+        }
+        if (hits.length > 0) {
+            parts.push(`### Fonti Principali\n`);
+            hits.slice(0, 3).forEach((hit, i) => {
+                parts.push(`${i + 1}. **${hit.path}** (righe ${hit.startLine}-${hit.endLine})`);
+                parts.push(`   ${hit.text.slice(0, 200)}...\n`);
+            });
+        }
+        if (entities.length > 0) {
+            parts.push(`### Entità Identificate\n`);
+            entities.slice(0, 5).forEach((e) => {
+                parts.push(`- **${e.name}** (${e.type})`);
+            });
+        }
+        return parts.join("\n");
+    }
+}
+//# sourceMappingURL=knowledge-graph.js.map

package/dist/knowledge-graph.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"knowledge-graph.js","sourceRoot":"","sources":["../src/knowledge-graph.ts"],"names":[],"mappings":"AAUA;;GAEG;AACH,MAAM,OAAO,sBAAsB;IACjC,QAAQ,GAAwB,IAAI,GAAG,EAAE,CAAC;IAC1C,SAAS,GAAe,EAAE,CAAC;IAE3B,SAAS,CAAC,MAAc;QACtB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;IACvC,CAAC;IAED,WAAW,CAAC,QAAkB;QAC5B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC;IAED,SAAS,CAAC,EAAU;QAClB,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC/B,CAAC;IAED,YAAY,CAAC,QAAgB;QAC3B,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAC1B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,QAAQ,IAAI,CAAC,CAAC,MAAM,KAAK,QAAQ,CACtD,CAAC;IACJ,CAAC;IAED,oBAAoB,CAAC,QAAgB,EAAE,QAAgB,CAAC;QACtD,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;QAClC,MAAM,MAAM,GAAa,EAAE,CAAC;QAE5B,MAAM,QAAQ,GAAG,CAAC,SAAiB,EAAE,YAAoB,EAAE,EAAE;YAC3D,IAAI,YAAY,GAAG,KAAK,IAAI,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC;gBAAE,OAAO;YAC3D,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAEvB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YACzC,IAAI,MAAM,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;gBACrC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtB,CAAC;YAED,IAAI,YAAY,GAAG,KAAK,EAAE,CAAC;gBACzB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;gBAC/C,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;oBAC5B,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC;oBAClE,QAAQ,CAAC,MAAM,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC;gBACrC,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QAEF,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QACtB,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,iBAAiB;IACpB,OAAO,CAAqC;IAEpD,YAAY,OAAiC;QAC3C,IAAI,CAAC,OAAO,GAAG;YACb,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE;YACxB,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,CAAC;YAC/B,GAAG,OAAO;SACX,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa;QACxB,sCAAsC;QACtC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAE3E,iCAAiC;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAErE,8BAA8B;QAC9B,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;QAEzD,sBAAsB;QACtB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAClF,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;QAE5D,wCAAwC;QACxC,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAEtD,sBAAsB;QACtB,MAAM,SAAS,GAAkB,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAClD,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,IAAI,EAAE,GAAG,CAAC,IAAI;SACf,CAAC,CAAC,CAAC;QAEJ,yBAAyB;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,YAAY,CAAC,CAAC;QAEhF,OAAO;YACL,MAAM;YACN,SAAS;YACT,QAAQ;YACR,SAAS;YACT,YAAY;SACb,CAAC;IACJ,CAAC;IAEO,iBAAiB,CAAC,QAAkB;QAC1C,MAAM,YAAY,GAAa,EAAE,CAAC;QAElC,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,oBAAoB,CACvD,MAAM,CAAC,EAAE,EACT,IAAI,CAAC,OAAO,CAAC,QAAQ,CACtB,CAAC;YACF,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAE7D,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACjD,IAAI,IAAI,GAAG,KAAK,MAAM,CAAC,IAAI,OAAO,MAAM,CAAC,IAAI,GAAG,CAAC;gBAEjD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACzB,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;wBACnC,MAAM,OAAO,GAAG,CAAC,CAAC,MAAM,KAAK,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;wBAC7D,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;wBACpD,OAAO,GAAG,CAAC,CAAC,IAAI,MAAM,KAAK,EAAE,IAAI,IAAI,OAAO,EAAE,CAAC;oBACjD,CAAC,CAAC,CAAC;oBACH,IAAI,IAAI,kBAAkB,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAClD,CAAC;gBAED,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACzB,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBAC1D,IAAI,IAAI,mBAAmB,SAAS,EAAE,CAAC;gBACzC,CAAC;gBAED,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QAED,OAAO,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAEO,WAAW,CACjB,MAAc,EACd,IAAiB,EACjB,QAAkB,EAClB,UAAsB,EACtB,YAAoB;QAEpB,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC5B,KAAK,CAAC,IAAI,CAAC,0BAA0B,IAAI,CAAC,MAAM,gBAAgB,QAAQ,CAAC,MAAM,yBAAyB,CAAC,CAAC;QAE1G,IAAI,YAAY,EAAE,CAAC;YACjB,KAAK,CAAC,IAAI,CAAC,qCAAqC,YAAY,IAAI,CAAC,CAAC;QACpE,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpB,KAAK,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACrC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE;gBAClC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,GAAG,CAAC,IAAI,aAAa,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,OAAO,GAAG,CAAC,CAAC;gBAChF,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;YAClD,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,KAAK,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;YACxC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;gBACjC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC;YAC5C,CAAC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;CACF"}

package/dist/knowledge-graph.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=knowledge-graph.test.d.ts.map

package/dist/knowledge-graph.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"knowledge-graph.test.d.ts","sourceRoot":"","sources":["../src/knowledge-graph.test.ts"],"names":[],"mappings":""}

package/dist/knowledge-graph.test.js

@@ -0,0 +1,140 @@
+import { describe, it, expect } from "vitest";
+import { KnowledgeGraphRag, InMemoryKnowledgeGraph } from "./knowledge-graph.js";
+class MockRetriever {
+    async search(_query, topK) {
+        return [
+            {
+                id: "hit-1",
+                path: "src/user.ts",
+                startLine: 1,
+                endLine: 10,
+                text: "User class with authentication",
+                score: 0.9,
+            },
+            {
+                id: "hit-2",
+                path: "src/auth.ts",
+                startLine: 5,
+                endLine: 15,
+                text: "Authentication service",
+                score: 0.85,
+            },
+        ].slice(0, topK);
+    }
+}
+class MockEntityExtractor {
+    async extract(_text) {
+        return [
+            {
+                id: "user",
+                name: "User",
+                type: "class",
+                properties: {},
+            },
+            {
+                id: "auth",
+                name: "Authentication",
+                type: "service",
+                properties: {},
+            },
+        ];
+    }
+}
+class MockRelationExtractor {
+    async extract(_text, _entities) {
+        return [
+            {
+                id: "user-auth",
+                source: "user",
+                target: "auth",
+                type: "uses",
+                properties: {},
+            },
+        ];
+    }
+}
+describe("KnowledgeGraphRag", () => {
+    it("should extract entities and relations from search results", async () => {
+        const graph = new InMemoryKnowledgeGraph();
+        const kgRag = new KnowledgeGraphRag({
+            retriever: new MockRetriever(),
+            entityExtractor: new MockEntityExtractor(),
+            relationExtractor: new MockRelationExtractor(),
+            graph,
+        });
+        const result = await kgRag.search("authentication");
+        expect(result.entities.length).toBe(2);
+        expect(result.relations.length).toBe(1);
+        expect(result.citations.length).toBe(2);
+        expect(result.answer).toContain("Knowledge Graph");
+    });
+    it("should build graph context with connected entities", async () => {
+        const graph = new InMemoryKnowledgeGraph();
+        const kgRag = new KnowledgeGraphRag({
+            retriever: new MockRetriever(),
+            entityExtractor: new MockEntityExtractor(),
+            relationExtractor: new MockRelationExtractor(),
+            graph,
+            maxDepth: 2,
+        });
+        const result = await kgRag.search("user authentication");
+        expect(result.graphContext).toContain("User");
+        expect(result.graphContext).toContain("Authentication");
+        expect(result.graphContext).toContain("uses");
+    });
+    it("should persist entities in the graph across searches", async () => {
+        const graph = new InMemoryKnowledgeGraph();
+        const kgRag = new KnowledgeGraphRag({
+            retriever: new MockRetriever(),
+            entityExtractor: new MockEntityExtractor(),
+            relationExtractor: new MockRelationExtractor(),
+            graph,
+        });
+        await kgRag.search("first query");
+        await kgRag.search("second query");
+        expect(graph.entities.size).toBe(2);
+        expect(graph.relations.length).toBe(2);
+    });
+});
+describe("InMemoryKnowledgeGraph", () => {
+    it("should add and retrieve entities", () => {
+        const graph = new InMemoryKnowledgeGraph();
+        const entity = {
+            id: "test",
+            name: "Test",
+            type: "class",
+            properties: {},
+        };
+        graph.addEntity(entity);
+        const retrieved = graph.getEntity("test");
+        expect(retrieved).toEqual(entity);
+    });
+    it("should add and retrieve relations", () => {
+        const graph = new InMemoryKnowledgeGraph();
+        const relation = {
+            id: "rel1",
+            source: "a",
+            target: "b",
+            type: "uses",
+            properties: {},
+        };
+        graph.addRelation(relation);
+        const relations = graph.getRelations("a");
+        expect(relations.length).toBe(1);
+        expect(relations[0]).toEqual(relation);
+    });
+    it("should find connected entities", () => {
+        const graph = new InMemoryKnowledgeGraph();
+        graph.addEntity({ id: "a", name: "A", type: "class", properties: {} });
+        graph.addEntity({ id: "b", name: "B", type: "class", properties: {} });
+        graph.addEntity({ id: "c", name: "C", type: "class", properties: {} });
+        graph.addRelation({ id: "r1", source: "a", target: "b", type: "uses", properties: {} });
+        graph.addRelation({ id: "r2", source: "b", target: "c", type: "uses", properties: {} });
+        const connected = graph.getConnectedEntities("a", 1);
+        expect(connected.length).toBe(1);
+        expect(connected[0]?.id).toBe("b");
+        const connectedDeep = graph.getConnectedEntities("a", 2);
+        expect(connectedDeep.length).toBe(2);
+    });
+});
+//# sourceMappingURL=knowledge-graph.test.js.map

package/dist/knowledge-graph.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"knowledge-graph.test.js","sourceRoot":"","sources":["../src/knowledge-graph.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAUjF,MAAM,aAAa;IACjB,KAAK,CAAC,MAAM,CAAC,MAAc,EAAE,IAAY;QACvC,OAAO;YACL;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,aAAa;gBACnB,SAAS,EAAE,CAAC;gBACZ,OAAO,EAAE,EAAE;gBACX,IAAI,EAAE,gCAAgC;gBACtC,KAAK,EAAE,GAAG;aACX;YACD;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,aAAa;gBACnB,SAAS,EAAE,CAAC;gBACZ,OAAO,EAAE,EAAE;gBACX,IAAI,EAAE,wBAAwB;gBAC9B,KAAK,EAAE,IAAI;aACZ;SACF,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACnB,CAAC;CACF;AAED,MAAM,mBAAmB;IACvB,KAAK,CAAC,OAAO,CAAC,KAAa;QACzB,OAAO;YACL;gBACE,EAAE,EAAE,MAAM;gBACV,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,OAAO;gBACb,UAAU,EAAE,EAAE;aACf;YACD;gBACE,EAAE,EAAE,MAAM;gBACV,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,SAAS;gBACf,UAAU,EAAE,EAAE;aACf;SACF,CAAC;IACJ,CAAC;CACF;AAED,MAAM,qBAAqB;IACzB,KAAK,CAAC,OAAO,CAAC,KAAa,EAAE,SAAmB;QAC9C,OAAO;YACL;gBACE,EAAE,EAAE,WAAW;gBACf,MAAM,EAAE,MAAM;gBACd,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,MAAM;gBACZ,UAAU,EAAE,EAAE;aACf;SACF,CAAC;IACJ,CAAC;CACF;AAED,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,KAAK,GAAG,IAAI,sBAAsB,EAAE,CAAC;QAC3C,MAAM,KAAK,GAAG,IAAI,iBAAiB,CAAC;YAClC,SAAS,EAAE,IAAI,aAAa,EAAE;YAC9B,eAAe,EAAE,IAAI,mBAAmB,EAAE;YAC1C,iBAAiB,EAAE,IAAI,qBAAqB,EAAE;YAC9C,KAAK;SACN,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAEpD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,KAAK,GAAG,IAAI,sBAAsB,EAAE,CAAC;QAC3C,MAAM,KAAK,GAAG,IAAI,iBAAiB,CAAC;YAClC,SAAS,EAAE,IAAI,aAAa,EAAE;YAC9B,eAAe,EAAE,IAAI,mBAAmB,EAAE;YAC1C,iBAAiB,EAAE,IAAI,qBAAqB,EAAE;YAC9C,KAAK;YACL,QAAQ,EAAE,CAAC;SACZ,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC;QAEzD,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,KAAK,GAAG,IAAI,sBAAsB,EAAE,CAAC;QAC3C,MAAM,KAAK,GAAG,IAAI,iBAAiB,CAAC;YAClC,SAAS,EAAE,IAAI,aAAa,EAAE;YAC9B,eAAe,EAAE,IAAI,mBAAmB,EAAE;YAC1C,iBAAiB,EAAE,IAAI,qBAAqB,EAAE;YAC9C,KAAK;SACN,CAAC,CAAC;QAEH,MAAM,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAClC,MAAM,KAAK,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QAEnC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,KAAK,GAAG,IAAI,sBAAsB,EAAE,CAAC;QAC3C,MAAM,MAAM,GAAW;YACrB,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,OAAO;YACb,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACxB,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAE1C,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,KAAK,GAAG,IAAI,sBAAsB,EAAE,CAAC;QAC3C,MAAM,QAAQ,GAAa;YACzB,EAAE,EAAE,MAAM;YACV,MAAM,EAAE,GAAG;YACX,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,MAAM;YACZ,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,KAAK,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC5B,MAAM,SAAS,GAAG,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;QAE1C,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,KAAK,GAAG,IAAI,sBAAsB,EAAE,CAAC;QAE3C,KAAK,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;QACvE,KAAK,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;QACvE,KAAK,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;QAEvE,KAAK,CAAC,WAAW,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;QACxF,KAAK,CAAC,WAAW,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;QAExF,MAAM,SAAS,GAAG,KAAK,CAAC,oBAAoB,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QACrD,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAEnC,MAAM,aAAa,GAAG,KAAK,CAAC,oBAAoB,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QACzD,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/llm-grader.d.ts

@@ -0,0 +1,19 @@
+import type { SearchHit } from "@eucode/indexer";
+import type { Grader, GraderResult, QueryRewriter } from "./types.js";
+export interface LlmProvider {
+    chat(messages: Array<{
+        role: string;
+        content: string;
+    }>): Promise<string>;
+}
+export declare class LlmGrader implements Grader {
+    private llm;
+    constructor(llm: LlmProvider);
+    grade(query: string, hit: SearchHit): Promise<GraderResult>;
+}
+export declare class LlmQueryRewriter implements QueryRewriter {
+    private llm;
+    constructor(llm: LlmProvider);
+    rewrite(query: string, context: string): Promise<string>;
+}
+//# sourceMappingURL=llm-grader.d.ts.map

package/dist/llm-grader.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm-grader.d.ts","sourceRoot":"","sources":["../src/llm-grader.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,MAAM,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAEtE,MAAM,WAAW,WAAW;IAC1B,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;CAC3E;AAED,qBAAa,SAAU,YAAW,MAAM;IAC1B,OAAO,CAAC,GAAG;gBAAH,GAAG,EAAE,WAAW;IAE9B,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,YAAY,CAAC;CAgClE;AAED,qBAAa,gBAAiB,YAAW,aAAa;IACxC,OAAO,CAAC,GAAG;gBAAH,GAAG,EAAE,WAAW;IAE9B,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAqB/D"}

package/dist/llm-grader.js

@@ -0,0 +1,63 @@
+export class LlmGrader {
+    llm;
+    constructor(llm) {
+        this.llm = llm;
+    }
+    async grade(query, hit) {
+        const prompt = `Rate the relevance of this code snippet to the query.
+
+Query: ${query}
+
+Code snippet from ${hit.path}:${hit.startLine}-${hit.endLine}:
+\`\`\`
+${hit.text.slice(0, 500)}
+\`\`\`
+
+Respond with JSON only: {"relevant": true/false, "score": 0.0-1.0}`;
+        try {
+            const response = await this.llm.chat([
+                { role: "system", content: "You are a code relevance grader. Respond only with valid JSON." },
+                { role: "user", content: prompt },
+            ]);
+            const match = response.match(/\{[\s\S]*\}/);
+            if (!match) {
+                return { relevant: false, score: 0 };
+            }
+            const parsed = JSON.parse(match[0]);
+            return {
+                relevant: Boolean(parsed.relevant),
+                score: Number(parsed.score) || 0,
+            };
+        }
+        catch {
+            return { relevant: false, score: 0 };
+        }
+    }
+}
+export class LlmQueryRewriter {
+    llm;
+    constructor(llm) {
+        this.llm = llm;
+    }
+    async rewrite(query, context) {
+        const prompt = `The original query didn't find enough relevant results. Rewrite it to be more specific and effective.
+
+Original query: ${query}
+
+Context from partial results:
+${context.slice(0, 300)}
+
+Provide a rewritten query that will find better results. Respond with the query only, no explanation.`;
+        try {
+            const response = await this.llm.chat([
+                { role: "system", content: "You are a search query rewriter. Respond with the rewritten query only." },
+                { role: "user", content: prompt },
+            ]);
+            return response.trim();
+        }
+        catch {
+            return query;
+        }
+    }
+}
+//# sourceMappingURL=llm-grader.js.map

package/dist/llm-grader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm-grader.js","sourceRoot":"","sources":["../src/llm-grader.ts"],"names":[],"mappings":"AAOA,MAAM,OAAO,SAAS;IACA;IAApB,YAAoB,GAAgB;QAAhB,QAAG,GAAH,GAAG,CAAa;IAAG,CAAC;IAExC,KAAK,CAAC,KAAK,CAAC,KAAa,EAAE,GAAc;QACvC,MAAM,MAAM,GAAG;;SAEV,KAAK;;oBAEM,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,OAAO;;EAE1D,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;;;mEAG2C,CAAC;QAEhE,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC;gBACnC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,gEAAgE,EAAE;gBAC7F,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;aAClC,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAC5C,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;YACvC,CAAC;YAED,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YACpC,OAAO;gBACL,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC;gBAClC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC;aACjC,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QACvC,CAAC;IACH,CAAC;CACF;AAED,MAAM,OAAO,gBAAgB;IACP;IAApB,YAAoB,GAAgB;QAAhB,QAAG,GAAH,GAAG,CAAa;IAAG,CAAC;IAExC,KAAK,CAAC,OAAO,CAAC,KAAa,EAAE,OAAe;QAC1C,MAAM,MAAM,GAAG;;kBAED,KAAK;;;EAGrB,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;;sGAE+E,CAAC;QAEnG,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC;gBACnC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,yEAAyE,EAAE;gBACtG,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;aAClC,CAAC,CAAC;YAEH,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAC;QACzB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF"}