agentic-qe 2.5.10 → 2.6.0

package/dist/code-intelligence/storage/CodeChunkStore.d.ts

@@ -0,0 +1,213 @@
+/**
+ * CodeChunkStore - RuVector-based storage for code intelligence chunks
+ *
+ * Extends RuVectorPostgresAdapter to provide specialized storage for:
+ * - Code chunks with embeddings
+ * - Entity relationships
+ * - File-based filtering and search
+ *
+ * Uses the existing RuVector PostgreSQL infrastructure instead of
+ * reinventing vector storage.
+ *
+ * @module code-intelligence/storage/CodeChunkStore
+ */
+import type { EntityRelationship, Language } from '../config/database-schema.js';
+/**
+ * Configuration for CodeChunkStore
+ */
+export interface CodeChunkStoreConfig {
+    /** PostgreSQL connection string */
+    connectionString?: string;
+    /** PostgreSQL host (default: localhost) */
+    host?: string;
+    /** PostgreSQL port (default: 5432) */
+    port?: number;
+    /** PostgreSQL database name (default: ruvector_db) */
+    database?: string;
+    /** PostgreSQL user (default: ruvector) */
+    user?: string;
+    /** PostgreSQL password (default: ruvector) */
+    password?: string;
+    /** Embedding dimension (default: 768 for nomic-embed-text) */
+    embeddingDimension?: number;
+    /** Connection pool size (default: 10) */
+    poolSize?: number;
+    /** Enable debug logging (default: false) */
+    debug?: boolean;
+}
+/**
+ * Search options for code chunks
+ */
+export interface CodeSearchOptions {
+    /** Maximum number of results */
+    topK?: number;
+    /** Minimum similarity score (0-1) */
+    minScore?: number;
+    /** Filter by language */
+    language?: Language;
+    /** Filter by file path pattern */
+    filePattern?: string;
+    /** Filter by entity type */
+    entityType?: string;
+    /** Include full content in results */
+    includeContent?: boolean;
+}
+/**
+ * Code search result
+ */
+export interface CodeSearchResult {
+    id: string;
+    filePath: string;
+    content: string;
+    startLine: number;
+    endLine: number;
+    score: number;
+    chunkType?: string;
+    name?: string;
+    language?: string;
+    metadata?: Record<string, unknown>;
+}
+/**
+ * CodeChunkStore - RuVector-backed code chunk storage
+ *
+ * Uses PostgreSQL with RuVector extension for:
+ * - Semantic vector search via ruvector_cosine_distance()
+ * - Full-text search via tsvector/GIN
+ * - Hybrid search combining both
+ */
+export declare class CodeChunkStore {
+    private pool;
+    private readonly config;
+    private initialized;
+    constructor(config?: CodeChunkStoreConfig);
+    /**
+     * Initialize the store and create tables if needed
+     */
+    initialize(): Promise<void>;
+    /**
+     * Store a code chunk with its embedding
+     */
+    storeChunk(chunk: {
+        id: string;
+        filePath: string;
+        content: string;
+        embedding: number[];
+        chunkType?: string;
+        name?: string;
+        startLine?: number;
+        endLine?: number;
+        language?: string;
+        metadata?: Record<string, unknown>;
+    }): Promise<void>;
+    /**
+     * Store multiple chunks in batch
+     */
+    storeChunks(chunks: Array<{
+        id: string;
+        filePath: string;
+        content: string;
+        embedding: number[];
+        chunkType?: string;
+        name?: string;
+        startLine?: number;
+        endLine?: number;
+        language?: string;
+        metadata?: Record<string, unknown>;
+    }>): Promise<void>;
+    /**
+     * Search for similar code chunks using vector similarity
+     */
+    search(queryEmbedding: number[], options?: CodeSearchOptions): Promise<CodeSearchResult[]>;
+    /**
+     * Hybrid search combining vector similarity and keyword matching
+     */
+    hybridSearch(queryEmbedding: number[], queryText: string, options?: CodeSearchOptions & {
+        semanticWeight?: number;
+    }): Promise<CodeSearchResult[]>;
+    /**
+     * Store a code entity
+     */
+    storeEntity(entity: {
+        id: string;
+        filePath: string;
+        entityType: string;
+        name: string;
+        signature?: string;
+        startLine?: number;
+        endLine?: number;
+        language?: string;
+        parentId?: string;
+        metadata?: Record<string, unknown>;
+    }): Promise<void>;
+    /**
+     * Store a relationship between entities
+     */
+    storeRelationship(relationship: {
+        sourceId: string;
+        targetId: string;
+        relationshipType: string;
+        confidence?: number;
+        metadata?: Record<string, unknown>;
+    }): Promise<void>;
+    /**
+     * Get relationships for an entity
+     */
+    getRelationships(entityId: string, direction?: 'outgoing' | 'incoming' | 'both'): Promise<EntityRelationship[]>;
+    /**
+     * Delete chunks for a file (for re-indexing)
+     */
+    deleteChunksForFile(filePath: string): Promise<number>;
+    /**
+     * Delete entities for a file
+     */
+    deleteEntitiesForFile(filePath: string): Promise<number>;
+    /**
+     * Delete chunks by file path (alias for deleteChunksForFile)
+     */
+    deleteByFilePath(filePath: string): Promise<number>;
+    /**
+     * Get statistics about stored data
+     */
+    getStats(): Promise<{
+        chunkCount: number;
+        entityCount: number;
+        relationshipCount: number;
+    }>;
+    /**
+     * Get chunk count
+     */
+    getChunkCount(): Promise<number>;
+    /**
+     * Get entity count
+     */
+    getEntityCount(): Promise<number>;
+    /**
+     * Health check
+     */
+    healthCheck(): Promise<{
+        healthy: boolean;
+        chunkCount: number;
+        entityCount: number;
+        ruvectorVersion?: string;
+        error?: string;
+    }>;
+    /**
+     * Close the connection pool
+     */
+    close(): Promise<void>;
+    /**
+     * Clear all data (use with caution)
+     */
+    clear(): Promise<void>;
+    private ensureInitialized;
+    private log;
+}
+/**
+ * Factory function to create CodeChunkStore with Docker defaults
+ */
+export declare function createDockerCodeChunkStore(options?: Partial<CodeChunkStoreConfig>): CodeChunkStore;
+/**
+ * Factory function to create CodeChunkStore from environment
+ */
+export declare function createCodeChunkStoreFromEnv(): CodeChunkStore;
+//# sourceMappingURL=CodeChunkStore.d.ts.map

package/dist/code-intelligence/storage/CodeChunkStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"CodeChunkStore.d.ts","sourceRoot":"","sources":["../../../src/code-intelligence/storage/CodeChunkStore.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAGH,OAAO,KAAK,EAAyB,kBAAkB,EAAE,QAAQ,EAAE,MAAM,8BAA8B,CAAC;AAExG;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,mCAAmC;IACnC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,2CAA2C;IAC3C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,sCAAsC;IACtC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,sDAAsD;IACtD,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,0CAA0C;IAC1C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,8CAA8C;IAC9C,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,8DAA8D;IAC9D,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,yCAAyC;IACzC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,4CAA4C;IAC5C,KAAK,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,gCAAgC;IAChC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,qCAAqC;IACrC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,yBAAyB;IACzB,QAAQ,CAAC,EAAE,QAAQ,CAAC;IACpB,kCAAkC;IAClC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,4BAA4B;IAC5B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,sCAAsC;IACtC,cAAc,CAAC,EAAE,OAAO,CAAC;CAC1B;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED;;;;;;;GAOG;AACH,qBAAa,cAAc;IACzB,OAAO,CAAC,IAAI,CAAO;IACnB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAiC;IACxD,OAAO,CAAC,WAAW,CAAS;gBAEhB,MAAM,GAAE,oBAAyB;IAmC7C;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAmFjC;;OAEG;IACG,UAAU,CAAC,KAAK,EAAE;QACtB,EAAE,EAAE,MAAM,CAAC;QACX,QAAQ,EAAE,MAAM,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,SAAS,EAAE,MAAM,EAAE,CAAC;QACpB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACpC,GAAG,OAAO,CAAC,IAAI,CAAC;IAoCjB;;OAEG;IACG,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC;QAC9B,EAAE,EAAE,MAAM,CAAC;QACX,QAAQ,EAAE,MAAM,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,SAAS,EAAE,MAAM,EAAE,CAAC;QACpB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACpC,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAgDlB;;OAEG;IACG,MAAM,CACV,cAAc,EAAE,MAAM,EAAE,EACxB,OAAO,GAAE,iBAAsB,GAC9B,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAoF9B;;OAEG;IACG,YAAY,CAChB,cAAc,EAAE,MAAM,EAAE,EACxB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE,iBAAiB,GAAG;QAAE,cAAc,CAAC,EAAE,MAAM,CAAA;KAAO,GAC5D,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAuF9B;;OAEG;IACG,WAAW,CAAC,MAAM,EAAE;QACxB,EAAE,EAAE,MAAM,CAAC;QACX,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACpC,GAAG,OAAO,CAAC,IAAI,CAAC;IAgCjB;;OAEG;IACG,iBAAiB,CAAC,YAAY,EAAE;QACpC,QAAQ,EAAE,MAAM,CAAC;QACjB,QAAQ,EAAE,MAAM,CAAC;QACjB,gBAAgB,EAAE,MAAM,CAAC;QACzB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACpC,GAAG,OAAO,CAAC,IAAI,CAAC;IAoBjB;;OAEG;IACG,gBAAgB,CACpB,QAAQ,EAAE,MAAM,EAChB,SAAS,GAAE,UAAU,GAAG,UAAU,GAAG,MAAe,GACnD,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAkBhC;;OAEG;IACG,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAW5D;;OAEG;IACG,qBAAqB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAW9D;;OAEG;IACG,gBAAgB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAIzD;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC;QACxB,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,EAAE,MAAM,CAAC;QACpB,iBAAiB,EAAE,MAAM,CAAC;KAC3B,CAAC;IAgBF;;OAEG;IACG,aAAa,IAAI,OAAO,CAAC,MAAM,CAAC;IAUtC;;OAEG;IACG,cAAc,IAAI,OAAO,CAAC,MAAM,CAAC;IAUvC;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC;QAC3B,OAAO,EAAE,OAAO,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,EAAE,MAAM,CAAC;QACpB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;IA0BF;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAM5B;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;YAYd,iBAAiB;IAM/B,OAAO,CAAC,GAAG;CAMZ;AAED;;GAEG;AACH,wBAAgB,0BAA0B,CACxC,OAAO,CAAC,EAAE,OAAO,CAAC,oBAAoB,CAAC,GACtC,cAAc,CAUhB;AAED;;GAEG;AACH,wBAAgB,2BAA2B,IAAI,cAAc,CAU5D"}