regen-koi-mcp 1.0.4 → 1.0.6

package/dist/unified_search.d.ts

@@ -0,0 +1,109 @@
+/**
+ * Unified Search - Single-Query Hybrid Search with RRF
+ *
+ * Combines graph-based entity search and vector semantic search
+ * in a SINGLE PostgreSQL query using Reciprocal Rank Fusion (RRF).
+ *
+ * Key Design Principles:
+ * 1. NO parallel database calls - Everything in one SQL query
+ * 2. Database does the work - RRF calculated in Postgres
+ * 3. Unified stack - AGE + pgvector in same database
+ *
+ * RRF Formula: score = 1.0 / (k + rank) where k=60
+ * This gives higher scores to higher-ranked results, with diminishing returns.
+ */
+export interface UnifiedSearchConfig {
+    host: string;
+    port: number;
+    database: string;
+    user?: string;
+    password?: string;
+    graphName?: string;
+    embeddingDimension?: 384 | 512 | 768 | 1024 | 1536 | 3072;
+    rrfConstant?: number;
+}
+export interface SearchHit {
+    id: string;
+    title: string;
+    content?: string;
+    source: 'graph' | 'vector' | 'module' | 'both';
+    graph_rank?: number;
+    vector_rank?: number;
+    module_rank?: number;
+    rrf_score: number;
+    final_score: number;
+    entity_type?: string;
+    file_path?: string;
+    line_number?: number;
+    docstring?: string;
+}
+export interface ModuleHit {
+    name: string;
+    repo: string;
+    path: string;
+    summary?: string;
+    entity_count: number;
+    score: number;
+}
+/**
+ * UnifiedSearch - Hybrid search combining graph and vector results with RRF
+ */
+export declare class UnifiedSearch {
+    private pool;
+    private graphName;
+    private embeddingDim;
+    private rrfK;
+    constructor(config: UnifiedSearchConfig);
+    /**
+     * Close the database connection pool
+     */
+    close(): Promise<void>;
+    /**
+     * Execute unified hybrid search - Single SQL query with RRF
+     *
+     * @param queryText - Natural language query
+     * @param queryEmbedding - Pre-computed query embedding vector
+     * @param entityNames - Entity names detected by router (optional, improves graph results)
+     * @param limit - Maximum results to return (default: 10)
+     */
+    search(queryText: string, queryEmbedding: number[], entityNames?: string[], limit?: number): Promise<SearchHit[]>;
+    /**
+     * Graph-only search (for entity lookup / relationship queries)
+     */
+    graphSearch(entityNames: string[], limit?: number): Promise<SearchHit[]>;
+    /**
+     * Vector-only search (for conceptual / semantic queries)
+     */
+    vectorSearch(queryEmbedding: number[], limit?: number): Promise<SearchHit[]>;
+    /**
+     * Module-level semantic search (RAPTOR)
+     * Searches module summaries by vector similarity
+     */
+    moduleSearch(queryEmbedding: number[], limit?: number): Promise<ModuleHit[]>;
+    /**
+     * Combined search including modules (RAPTOR-enhanced hybrid search)
+     * For high-level conceptual queries, searches module summaries first
+     */
+    searchWithModules(queryText: string, queryEmbedding: number[], entityNames?: string[], options?: {
+        includeModules?: boolean;
+        moduleLimit?: number;
+        limit?: number;
+    }): Promise<{
+        hits: SearchHit[];
+        moduleHits: ModuleHit[];
+    }>;
+    /**
+     * Get search statistics
+     */
+    getStats(): Promise<{
+        total_memories: number;
+        total_embeddings: number;
+        embedding_coverage: number;
+        graph_entities: number;
+    }>;
+}
+/**
+ * Create a default UnifiedSearch instance using environment variables
+ */
+export declare function createUnifiedSearch(): UnifiedSearch;
+//# sourceMappingURL=unified_search.d.ts.map

package/dist/unified_search.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"unified_search.d.ts","sourceRoot":"","sources":["../src/unified_search.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAOH,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,kBAAkB,CAAC,EAAE,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;IAC1D,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAGD,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,OAAO,GAAG,QAAQ,GAAG,QAAQ,GAAG,MAAM,CAAC;IAC/C,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAGD,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;CACf;AAED;;GAEG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,IAAI,CAAW;IACvB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,IAAI,CAAS;gBAET,MAAM,EAAE,mBAAmB;IAavC;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAI5B;;;;;;;OAOG;IACG,MAAM,CACV,SAAS,EAAE,MAAM,EACjB,cAAc,EAAE,MAAM,EAAE,EACxB,WAAW,GAAE,MAAM,EAAO,EAC1B,KAAK,GAAE,MAAW,GACjB,OAAO,CAAC,SAAS,EAAE,CAAC;IA6HvB;;OAEG;IACG,WAAW,CAAC,WAAW,EAAE,MAAM,EAAE,EAAE,KAAK,GAAE,MAAW,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAwClF;;OAEG;IACG,YAAY,CAAC,cAAc,EAAE,MAAM,EAAE,EAAE,KAAK,GAAE,MAAW,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IA4BtF;;;OAGG;IACG,YAAY,CAAC,cAAc,EAAE,MAAM,EAAE,EAAE,KAAK,GAAE,MAAU,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IA0CrF;;;OAGG;IACG,iBAAiB,CACrB,SAAS,EAAE,MAAM,EACjB,cAAc,EAAE,MAAM,EAAE,EACxB,WAAW,GAAE,MAAM,EAAO,EAC1B,OAAO,GAAE;QAAE,cAAc,CAAC,EAAE,OAAO,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAO,GAC/E,OAAO,CAAC;QAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QAAC,UAAU,EAAE,SAAS,EAAE,CAAA;KAAE,CAAC;IAoB1D;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC;QACxB,cAAc,EAAE,MAAM,CAAC;QACvB,gBAAgB,EAAE,MAAM,CAAC;QACzB,kBAAkB,EAAE,MAAM,CAAC;QAC3B,cAAc,EAAE,MAAM,CAAC;KACxB,CAAC;CAgCH;AAED;;GAEG;AACH,wBAAgB,mBAAmB,IAAI,aAAa,CAWnD"}

package/dist/unified_search.js

@@ -0,0 +1,352 @@
+/**
+ * Unified Search - Single-Query Hybrid Search with RRF
+ *
+ * Combines graph-based entity search and vector semantic search
+ * in a SINGLE PostgreSQL query using Reciprocal Rank Fusion (RRF).
+ *
+ * Key Design Principles:
+ * 1. NO parallel database calls - Everything in one SQL query
+ * 2. Database does the work - RRF calculated in Postgres
+ * 3. Unified stack - AGE + pgvector in same database
+ *
+ * RRF Formula: score = 1.0 / (k + rank) where k=60
+ * This gives higher scores to higher-ranked results, with diminishing returns.
+ */
+import pkg from 'pg';
+const { Pool } = pkg;
+/**
+ * UnifiedSearch - Hybrid search combining graph and vector results with RRF
+ */
+export class UnifiedSearch {
+    pool;
+    graphName;
+    embeddingDim;
+    rrfK;
+    constructor(config) {
+        this.pool = new Pool({
+            host: config.host,
+            port: config.port,
+            database: config.database,
+            user: config.user,
+            password: config.password,
+        });
+        this.graphName = config.graphName || 'regen_graph';
+        this.embeddingDim = `dim_${config.embeddingDimension || 1536}`;
+        this.rrfK = config.rrfConstant || 60;
+    }
+    /**
+     * Close the database connection pool
+     */
+    async close() {
+        await this.pool.end();
+    }
+    /**
+     * Execute unified hybrid search - Single SQL query with RRF
+     *
+     * @param queryText - Natural language query
+     * @param queryEmbedding - Pre-computed query embedding vector
+     * @param entityNames - Entity names detected by router (optional, improves graph results)
+     * @param limit - Maximum results to return (default: 10)
+     */
+    async search(queryText, queryEmbedding, entityNames = [], limit = 10) {
+        const client = await this.pool.connect();
+        try {
+            // Load AGE extension for this session
+            await client.query("LOAD 'age';");
+            await client.query(`SET search_path = ag_catalog, "$user", public;`);
+            // Build the unified hybrid search query
+            const query = `
+        WITH graph_hits AS (
+          -- Graph: Entity-based search using detected entities
+          SELECT
+            'graph_' || (e->>'name')::text || '_' || id(e)::text as id,
+            (e->>'name')::text as title,
+            (e->>'docstring')::text as content,
+            (e->>'file_path')::text as file_path,
+            (e->>'line_number')::int as line_number,
+            'graph' as source,
+            labels(e)[0]::text as entity_type,
+            ROW_NUMBER() OVER (ORDER BY id(e)) as rank
+          FROM cypher('${this.graphName}', $$
+            MATCH (e)
+            WHERE e.name = ANY($1::text[])
+            RETURN e
+          $$, $2) as (e agtype)
+          WHERE e IS NOT NULL
+          LIMIT 10
+        ),
+        vector_hits AS (
+          -- Vector: Semantic search on document embeddings
+          SELECT
+            m.id::text as id,
+            COALESCE(m.metadata->>'title', m.rid) as title,
+            m.content::text as content,
+            m.metadata->>'file_path' as file_path,
+            NULL::int as line_number,
+            'vector' as source,
+            NULL::text as entity_type,
+            ROW_NUMBER() OVER (ORDER BY e.${this.embeddingDim} <=> $3::vector) as rank
+          FROM koi_memories m
+          INNER JOIN koi_embeddings e ON e.memory_id = m.id
+          WHERE e.${this.embeddingDim} IS NOT NULL
+            AND m.superseded_at IS NULL
+          ORDER BY e.${this.embeddingDim} <=> $3::vector
+          LIMIT 10
+        ),
+        combined AS (
+          -- Combine graph and vector results
+          SELECT
+            id,
+            title,
+            content,
+            file_path,
+            line_number,
+            source,
+            entity_type,
+            rank,
+            1.0 / ($4::float + rank) as rrf_score
+          FROM graph_hits
+          UNION ALL
+          SELECT
+            id,
+            title,
+            content,
+            file_path,
+            line_number,
+            source,
+            entity_type,
+            rank,
+            1.0 / ($4::float + rank) as rrf_score
+          FROM vector_hits
+        ),
+        aggregated AS (
+          -- Aggregate scores for items appearing in both sources
+          SELECT
+            id,
+            MAX(title) as title,
+            MAX(content) as content,
+            MAX(file_path) as file_path,
+            MAX(line_number) as line_number,
+            MAX(entity_type) as entity_type,
+            CASE
+              WHEN COUNT(DISTINCT source) > 1 THEN 'both'
+              ELSE MAX(source)
+            END as source,
+            MAX(CASE WHEN source = 'graph' THEN rank END) as graph_rank,
+            MAX(CASE WHEN source = 'vector' THEN rank END) as vector_rank,
+            SUM(rrf_score) as final_score
+          FROM combined
+          GROUP BY id
+        )
+        SELECT
+          id,
+          title,
+          content,
+          file_path,
+          line_number,
+          entity_type,
+          source,
+          graph_rank,
+          vector_rank,
+          final_score,
+          final_score as rrf_score  -- For backwards compatibility
+        FROM aggregated
+        ORDER BY final_score DESC
+        LIMIT $5
+      `;
+            // Convert embedding to postgres array format
+            const embeddingStr = `[${queryEmbedding.join(',')}]`;
+            const result = await client.query(query, [
+                entityNames.length > 0 ? entityNames : ['__NONE__'], // Fallback if no entities
+                JSON.stringify(entityNames.length > 0 ? entityNames : ['__NONE__']),
+                embeddingStr,
+                this.rrfK,
+                limit
+            ]);
+            return result.rows;
+        }
+        finally {
+            client.release();
+        }
+    }
+    /**
+     * Graph-only search (for entity lookup / relationship queries)
+     */
+    async graphSearch(entityNames, limit = 10) {
+        const client = await this.pool.connect();
+        try {
+            await client.query("LOAD 'age';");
+            await client.query(`SET search_path = ag_catalog, "$user", public;`);
+            const query = `
+        SELECT
+          'graph_' || (e->>'name')::text || '_' || id(e)::text as id,
+          (e->>'name')::text as title,
+          (e->>'docstring')::text as content,
+          (e->>'file_path')::text as file_path,
+          (e->>'line_number')::int as line_number,
+          labels(e)[0]::text as entity_type,
+          'graph' as source,
+          ROW_NUMBER() OVER (ORDER BY id(e)) as graph_rank,
+          NULL::int as vector_rank,
+          1.0 as rrf_score,
+          1.0 as final_score
+        FROM cypher('${this.graphName}', $$
+          MATCH (e)
+          WHERE e.name = ANY($1::text[])
+          RETURN e
+        $$, $2) as (e agtype)
+        WHERE e IS NOT NULL
+        LIMIT $3
+      `;
+            const result = await client.query(query, [
+                entityNames,
+                JSON.stringify(entityNames),
+                limit
+            ]);
+            return result.rows;
+        }
+        finally {
+            client.release();
+        }
+    }
+    /**
+     * Vector-only search (for conceptual / semantic queries)
+     */
+    async vectorSearch(queryEmbedding, limit = 10) {
+        const embeddingStr = `[${queryEmbedding.join(',')}]`;
+        const query = `
+      SELECT
+        m.id::text as id,
+        COALESCE(m.metadata->>'title', m.rid) as title,
+        m.content::text as content,
+        m.metadata->>'file_path' as file_path,
+        NULL::int as line_number,
+        NULL::text as entity_type,
+        'vector' as source,
+        NULL::int as graph_rank,
+        ROW_NUMBER() OVER (ORDER BY e.${this.embeddingDim} <=> $1::vector) as vector_rank,
+        (1.0 - (e.${this.embeddingDim} <=> $1::vector)) as rrf_score,
+        (1.0 - (e.${this.embeddingDim} <=> $1::vector)) as final_score
+      FROM koi_memories m
+      INNER JOIN koi_embeddings e ON e.memory_id = m.id
+      WHERE e.${this.embeddingDim} IS NOT NULL
+        AND m.superseded_at IS NULL
+      ORDER BY e.${this.embeddingDim} <=> $1::vector
+      LIMIT $2
+    `;
+        const result = await this.pool.query(query, [embeddingStr, limit]);
+        return result.rows;
+    }
+    /**
+     * Module-level semantic search (RAPTOR)
+     * Searches module summaries by vector similarity
+     */
+    async moduleSearch(queryEmbedding, limit = 5) {
+        const client = await this.pool.connect();
+        try {
+            await client.query("LOAD 'age';");
+            await client.query(`SET search_path = ag_catalog, "$user", public;`);
+            const embeddingStr = `[${queryEmbedding.join(',')}]`;
+            // Search module summaries stored in koi_memories with type='module_summary'
+            const query = `
+        SELECT
+          m.rid,
+          m.content as summary,
+          m.metadata->>'repo' as repo,
+          m.metadata->>'path' as path,
+          m.metadata->>'title' as title,
+          COALESCE((m.metadata->>'entity_count')::int, 0) as entity_count,
+          (1.0 - (e.${this.embeddingDim} <=> $1::vector)) as score
+        FROM koi_memories m
+        INNER JOIN koi_embeddings e ON e.memory_id = m.id
+        WHERE e.${this.embeddingDim} IS NOT NULL
+          AND m.superseded_at IS NULL
+          AND m.metadata->>'type' = 'module_summary'
+        ORDER BY e.${this.embeddingDim} <=> $1::vector
+        LIMIT $2
+      `;
+            const result = await client.query(query, [embeddingStr, limit]);
+            return result.rows.map(row => ({
+                name: row.rid.replace('module:', '').split('/').pop() || row.title,
+                repo: row.repo || row.rid.split('/')[0]?.replace('module:', ''),
+                path: row.path || '',
+                summary: row.summary,
+                entity_count: row.entity_count,
+                score: parseFloat(row.score) || 0
+            }));
+        }
+        finally {
+            client.release();
+        }
+    }
+    /**
+     * Combined search including modules (RAPTOR-enhanced hybrid search)
+     * For high-level conceptual queries, searches module summaries first
+     */
+    async searchWithModules(queryText, queryEmbedding, entityNames = [], options = {}) {
+        const { includeModules = true, moduleLimit = 3, limit = 10 } = options;
+        // Run standard search
+        const hits = await this.search(queryText, queryEmbedding, entityNames, limit);
+        // Also search modules if enabled
+        let moduleHits = [];
+        if (includeModules) {
+            try {
+                moduleHits = await this.moduleSearch(queryEmbedding, moduleLimit);
+            }
+            catch (error) {
+                // Module search is optional - if it fails, just return empty
+                console.error('[UnifiedSearch] Module search failed:', error);
+            }
+        }
+        return { hits, moduleHits };
+    }
+    /**
+     * Get search statistics
+     */
+    async getStats() {
+        const client = await this.pool.connect();
+        try {
+            await client.query("LOAD 'age';");
+            await client.query(`SET search_path = ag_catalog, "$user", public;`);
+            const queries = await Promise.all([
+                this.pool.query('SELECT COUNT(*) as count FROM koi_memories WHERE superseded_at IS NULL'),
+                this.pool.query(`SELECT COUNT(*) as count FROM koi_embeddings WHERE ${this.embeddingDim} IS NOT NULL`),
+                client.query(`
+          SELECT COUNT(*) as count
+          FROM cypher('${this.graphName}', $$
+            MATCH (e)
+            RETURN count(e)
+          $$) as (count agtype)
+        `)
+            ]);
+            const total_memories = parseInt(queries[0].rows[0].count);
+            const total_embeddings = parseInt(queries[1].rows[0].count);
+            const graph_entities = parseInt(queries[2].rows[0].count);
+            return {
+                total_memories,
+                total_embeddings,
+                embedding_coverage: total_memories > 0 ? total_embeddings / total_memories : 0,
+                graph_entities
+            };
+        }
+        finally {
+            client.release();
+        }
+    }
+}
+/**
+ * Create a default UnifiedSearch instance using environment variables
+ */
+export function createUnifiedSearch() {
+    return new UnifiedSearch({
+        host: process.env.GRAPH_DB_HOST || 'localhost',
+        port: parseInt(process.env.GRAPH_DB_PORT || '5432'),
+        database: process.env.GRAPH_DB_NAME || 'eliza',
+        user: process.env.GRAPH_DB_USER,
+        password: process.env.GRAPH_DB_PASSWORD,
+        graphName: process.env.GRAPH_NAME || 'regen_graph',
+        embeddingDimension: parseInt(process.env.EMBEDDING_DIM || '1536'),
+        rrfConstant: parseInt(process.env.RRF_K || '60'),
+    });
+}
+//# sourceMappingURL=unified_search.js.map

package/dist/unified_search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"unified_search.js","sourceRoot":"","sources":["../src/unified_search.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,OAAO,GAAG,MAAM,IAAI,CAAC;AAErB,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,CAAC;AAyCrB;;GAEG;AACH,MAAM,OAAO,aAAa;IAChB,IAAI,CAAW;IACf,SAAS,CAAS;IAClB,YAAY,CAAS;IACrB,IAAI,CAAS;IAErB,YAAY,MAA2B;QACrC,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,CAAC;YACnB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;SAC1B,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,aAAa,CAAC;QACnD,IAAI,CAAC,YAAY,GAAG,OAAO,MAAM,CAAC,kBAAkB,IAAI,IAAI,EAAE,CAAC;QAC/D,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;IACxB,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,MAAM,CACV,SAAiB,EACjB,cAAwB,EACxB,cAAwB,EAAE,EAC1B,QAAgB,EAAE;QAElB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QACzC,IAAI,CAAC;YACH,sCAAsC;YACtC,MAAM,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAClC,MAAM,MAAM,CAAC,KAAK,CAAC,gDAAgD,CAAC,CAAC;YAErE,wCAAwC;YACxC,MAAM,KAAK,GAAG;;;;;;;;;;;;yBAYK,IAAI,CAAC,SAAS;;;;;;;;;;;;;;;;;;4CAkBK,IAAI,CAAC,YAAY;;;oBAGzC,IAAI,CAAC,YAAY;;uBAEd,IAAI,CAAC,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA+DjC,CAAC;YAEF,6CAA6C;YAC7C,MAAM,YAAY,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YAErD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE;gBACvC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,EAAG,0BAA0B;gBAChF,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;gBACnE,YAAY;gBACZ,IAAI,CAAC,IAAI;gBACT,KAAK;aACN,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC,IAAmB,CAAC;QACpC,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,WAAqB,EAAE,QAAgB,EAAE;QACzD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QACzC,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAClC,MAAM,MAAM,CAAC,KAAK,CAAC,gDAAgD,CAAC,CAAC;YAErE,MAAM,KAAK,GAAG;;;;;;;;;;;;;uBAaG,IAAI,CAAC,SAAS;;;;;;;OAO9B,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE;gBACvC,WAAW;gBACX,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;gBAC3B,KAAK;aACN,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC,IAAmB,CAAC;QACpC,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,cAAwB,EAAE,QAAgB,EAAE;QAC7D,MAAM,YAAY,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;QAErD,MAAM,KAAK,GAAG;;;;;;;;;;wCAUsB,IAAI,CAAC,YAAY;oBACrC,IAAI,CAAC,YAAY;oBACjB,IAAI,CAAC,YAAY;;;gBAGrB,IAAI,CAAC,YAAY;;mBAEd,IAAI,CAAC,YAAY;;KAE/B,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC,CAAC;QACnE,OAAO,MAAM,CAAC,IAAmB,CAAC;IACpC,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,YAAY,CAAC,cAAwB,EAAE,QAAgB,CAAC;QAC5D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QACzC,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAClC,MAAM,MAAM,CAAC,KAAK,CAAC,gDAAgD,CAAC,CAAC;YAErE,MAAM,YAAY,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YAErD,4EAA4E;YAC5E,MAAM,KAAK,GAAG;;;;;;;;sBAQE,IAAI,CAAC,YAAY;;;kBAGrB,IAAI,CAAC,YAAY;;;qBAGd,IAAI,CAAC,YAAY;;OAE/B,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC,CAAC;YAEhE,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC7B,IAAI,EAAE,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,GAAG,CAAC,KAAK;gBAClE,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;gBAC/D,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,EAAE;gBACpB,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,YAAY,EAAE,GAAG,CAAC,YAAY;gBAC9B,KAAK,EAAE,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;aAClC,CAAC,CAAC,CAAC;QACN,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,iBAAiB,CACrB,SAAiB,EACjB,cAAwB,EACxB,cAAwB,EAAE,EAC1B,UAA8E,EAAE;QAEhF,MAAM,EAAE,cAAc,GAAG,IAAI,EAAE,WAAW,GAAG,CAAC,EAAE,KAAK,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC;QAEvE,sBAAsB;QACtB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,cAAc,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC;QAE9E,iCAAiC;QACjC,IAAI,UAAU,GAAgB,EAAE,CAAC;QACjC,IAAI,cAAc,EAAE,CAAC;YACnB,IAAI,CAAC;gBACH,UAAU,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;YACpE,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,6DAA6D;gBAC7D,OAAO,CAAC,KAAK,CAAC,uCAAuC,EAAE,KAAK,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;QAED,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QAMZ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QACzC,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAClC,MAAM,MAAM,CAAC,KAAK,CAAC,gDAAgD,CAAC,CAAC;YAErE,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,wEAAwE,CAAC;gBACzF,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,sDAAsD,IAAI,CAAC,YAAY,cAAc,CAAC;gBACtG,MAAM,CAAC,KAAK,CAAC;;yBAEI,IAAI,CAAC,SAAS;;;;SAI9B,CAAC;aACH,CAAC,CAAC;YAEH,MAAM,cAAc,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YAC1D,MAAM,gBAAgB,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YAC5D,MAAM,cAAc,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YAE1D,OAAO;gBACL,cAAc;gBACd,gBAAgB;gBAChB,kBAAkB,EAAE,cAAc,GAAG,CAAC,CAAC,CAAC,CAAC,gBAAgB,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;gBAC9E,cAAc;aACf,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB;IACjC,OAAO,IAAI,aAAa,CAAC;QACvB,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,WAAW;QAC9C,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,MAAM,CAAC;QACnD,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,OAAO;QAC9C,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,aAAa;QAC/B,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;QACvC,SAAS,EAAE,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,aAAa;QAClD,kBAAkB,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,MAAM,CAAQ;QACxE,WAAW,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,IAAI,IAAI,CAAC;KACjD,CAAC,CAAC;AACL,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "regen-koi-mcp",
-  "version": "1.0.4",
+  "version": "1.0.6",
   "description": "MCP server for accessing Regen Network's KOI (Knowledge Organization Infrastructure) system",
   "main": "dist/index.js",
   "type": "module",
@@ -47,8 +47,10 @@
   "homepage": "https://github.com/gaiaaiagent/regen-koi-mcp#readme",
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.20.0",
+    "@types/pg": "^8.15.6",
     "axios": "^1.7.7",
     "dotenv": "^16.4.5",
+    "pg": "^8.16.3",
     "regen-koi-mcp": "^1.0.0"
   },
   "devDependencies": {