@traqr/memory 0.1.3 → 0.1.4

package/dist/vectordb/index.d.ts

@@ -2,12 +2,16 @@
  * VectorDB Provider Factory
  *
  * Abstraction layer for vector database operations.
- * Currently supports Supabase pgvector.
+ * Supports Supabase (PostgREST) and raw Postgres (pg wire protocol).
+ * Auto-detects from DATABASE_URL vs SUPABASE_URL.
  */
 import type { VectorDBProvider, ProviderConfig } from './types.js';
 export * from './types.js';
 /**
- * Get the configured vector database provider
+ * Get the configured vector database provider.
+ * Auto-detects from config or env vars:
+ *   DATABASE_URL → PostgresVectorProvider
+ *   SUPABASE_URL → SupabaseVectorProvider
  */
 export declare function getVectorDB(config?: ProviderConfig): VectorDBProvider;
 /**

package/dist/vectordb/index.js

@@ -2,30 +2,46 @@
  * VectorDB Provider Factory
  *
  * Abstraction layer for vector database operations.
- * Currently supports Supabase pgvector.
+ * Supports Supabase (PostgREST) and raw Postgres (pg wire protocol).
+ * Auto-detects from DATABASE_URL vs SUPABASE_URL.
  */
 import { SupabaseVectorProvider } from './supabase.js';
+import { PostgresVectorProvider, resetPostgresPool } from './postgres.js';
+import { getMemoryConfig } from '../lib/client.js';
 // Re-export types for convenience
 export * from './types.js';
 // Singleton instance
 let providerInstance = null;
 /**
- * Get the configured vector database provider
+ * Get the configured vector database provider.
+ * Auto-detects from config or env vars:
+ *   DATABASE_URL → PostgresVectorProvider
+ *   SUPABASE_URL → SupabaseVectorProvider
  */
 export function getVectorDB(config) {
     if (providerInstance)
         return providerInstance;
-    const providerType = config?.type || 'supabase';
-    switch (providerType) {
-        case 'supabase':
-            providerInstance = new SupabaseVectorProvider();
-            break;
-        case 'pinecone':
-            throw new Error('Pinecone provider not yet implemented');
-        case 'qdrant':
-            throw new Error('Qdrant provider not yet implemented');
-        default:
-            throw new Error(`Unknown provider type: ${providerType}`);
+    // Explicit type takes priority
+    if (config?.type) {
+        switch (config.type) {
+            case 'postgres':
+                providerInstance = new PostgresVectorProvider();
+                break;
+            case 'supabase':
+                providerInstance = new SupabaseVectorProvider();
+                break;
+            default:
+                throw new Error(`Unknown provider type: ${config.type}`);
+        }
+        return providerInstance;
+    }
+    // Auto-detect from stored config / env vars
+    const memConfig = getMemoryConfig();
+    if (memConfig.databaseUrl) {
+        providerInstance = new PostgresVectorProvider();
+    }
+    else {
+        providerInstance = new SupabaseVectorProvider();
     }
     return providerInstance;
 }
@@ -34,6 +50,7 @@ export function getVectorDB(config) {
  */
 export function resetVectorDB() {
     providerInstance = null;
+    resetPostgresPool();
 }
 export default getVectorDB;
 //# sourceMappingURL=index.js.map

package/dist/vectordb/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/vectordb/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAA;AAGtD,kCAAkC;AAClC,cAAc,YAAY,CAAA;AAE1B,qBAAqB;AACrB,IAAI,gBAAgB,GAA4B,IAAI,CAAA;AAEpD;;GAEG;AACH,MAAM,UAAU,WAAW,CAAC,MAAuB;IACjD,IAAI,gBAAgB;QAAE,OAAO,gBAAgB,CAAA;IAE7C,MAAM,YAAY,GAAG,MAAM,EAAE,IAAI,IAAI,UAAU,CAAA;IAE/C,QAAQ,YAAY,EAAE,CAAC;QACrB,KAAK,UAAU;YACb,gBAAgB,GAAG,IAAI,sBAAsB,EAAE,CAAA;YAC/C,MAAK;QACP,KAAK,UAAU;YACb,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAA;QAC1D,KAAK,QAAQ;YACX,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;QACxD;YACE,MAAM,IAAI,KAAK,CAAC,0BAA0B,YAAY,EAAE,CAAC,CAAA;IAC7D,CAAC;IAED,OAAO,gBAAgB,CAAA;AACzB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa;IAC3B,gBAAgB,GAAG,IAAI,CAAA;AACzB,CAAC;AAED,eAAe,WAAW,CAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/vectordb/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAA;AACtD,OAAO,EAAE,sBAAsB,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAA;AACzE,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAA;AAGlD,kCAAkC;AAClC,cAAc,YAAY,CAAA;AAE1B,qBAAqB;AACrB,IAAI,gBAAgB,GAA4B,IAAI,CAAA;AAEpD;;;;;GAKG;AACH,MAAM,UAAU,WAAW,CAAC,MAAuB;IACjD,IAAI,gBAAgB;QAAE,OAAO,gBAAgB,CAAA;IAE7C,+BAA+B;IAC/B,IAAI,MAAM,EAAE,IAAI,EAAE,CAAC;QACjB,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;YACpB,KAAK,UAAU;gBACb,gBAAgB,GAAG,IAAI,sBAAsB,EAAE,CAAA;gBAC/C,MAAK;YACP,KAAK,UAAU;gBACb,gBAAgB,GAAG,IAAI,sBAAsB,EAAE,CAAA;gBAC/C,MAAK;YACP;gBACE,MAAM,IAAI,KAAK,CAAC,0BAA0B,MAAM,CAAC,IAAI,EAAE,CAAC,CAAA;QAC5D,CAAC;QACD,OAAO,gBAAgB,CAAA;IACzB,CAAC;IAED,4CAA4C;IAC5C,MAAM,SAAS,GAAG,eAAe,EAAE,CAAA;IACnC,IAAI,SAAS,CAAC,WAAW,EAAE,CAAC;QAC1B,gBAAgB,GAAG,IAAI,sBAAsB,EAAE,CAAA;IACjD,CAAC;SAAM,CAAC;QACN,gBAAgB,GAAG,IAAI,sBAAsB,EAAE,CAAA;IACjD,CAAC;IAED,OAAO,gBAAgB,CAAA;AACzB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa;IAC3B,gBAAgB,GAAG,IAAI,CAAA;IACvB,iBAAiB,EAAE,CAAA;AACrB,CAAC;AAED,eAAe,WAAW,CAAA"}

package/dist/vectordb/postgres.d.ts

@@ -0,0 +1,76 @@
+/**
+ * Postgres VectorDB Provider
+ *
+ * Implementation of VectorDBProvider using raw pg wire protocol.
+ * Calls the same SQL functions as SupabaseVectorProvider, but via
+ * parameterized queries instead of PostgREST RPC.
+ *
+ * Requires: npm install pg (dynamic import — not a hard dependency)
+ */
+import type { VectorDBProvider, Memory, MemoryInput, MemorySearchResult, MemoryUpdate, MemoryExport, MemoryDomain, SearchOptions, BrowseResult, BM25SearchResult, TemporalSearchResult, GraphSearchResult } from './types.js';
+/** Reset pool (for testing or reconfiguration) */
+export declare function resetPostgresPool(): void;
+export declare class PostgresVectorProvider implements VectorDBProvider {
+    store(input: MemoryInput, domainId?: string): Promise<Memory>;
+    search(queryText: string, options?: SearchOptions & {
+        precomputedEmbedding?: string;
+    }): Promise<MemorySearchResult[]>;
+    getById(id: string): Promise<Memory | null>;
+    update(id: string, updates: MemoryUpdate): Promise<Memory>;
+    delete(id: string): Promise<void>;
+    validate(id: string): Promise<Memory>;
+    archive(id: string, reason?: string): Promise<Memory>;
+    unarchive(id: string): Promise<Memory>;
+    exportAll(domainId?: string): Promise<MemoryExport[]>;
+    importBulk(memories: MemoryExport[], domainId: string): Promise<number>;
+    createDomain(name: string, description?: string, userId?: string): Promise<MemoryDomain>;
+    getDomain(name: string): Promise<MemoryDomain | null>;
+    getDefaultDomain(): Promise<MemoryDomain>;
+    bm25Search(queryText: string, options?: {
+        projectId?: string;
+        domain?: string;
+        category?: string;
+        limit?: number;
+        minScore?: number;
+    }): Promise<BM25SearchResult[]>;
+    temporalSearch(queryText: string, dateStart: Date, dateEnd: Date, options?: {
+        projectId?: string;
+        similarityThreshold?: number;
+        limit?: number;
+        precomputedEmbedding?: string;
+    }): Promise<TemporalSearchResult[]>;
+    graphSearch(seedIds: string[], options?: {
+        edgeTypes?: string[];
+        maxDepth?: number;
+        limit?: number;
+    }): Promise<GraphSearchResult[]>;
+    invalidate(id: string): Promise<void>;
+    supersede(id: string): Promise<void>;
+    findEntityByName(name: string, entityType: string): Promise<any | null>;
+    findEntityByNameFuzzy(name: string, entityType: string): Promise<any | null>;
+    findEntityByEmbedding(embeddingStr: string, entityType: string, threshold?: number): Promise<any | null>;
+    createEntity(entity: {
+        name: string;
+        entityType: string;
+        embedding?: string;
+        userId?: string;
+    }): Promise<any>;
+    incrementEntityMentions(entityId: string): Promise<void>;
+    linkMemoryToEntity(memoryId: string, entityId: string, role?: string): Promise<void>;
+    findEntitiesByNames(names: string[]): Promise<{
+        id: string;
+        name: string;
+    }[]>;
+    findOrphanedEntities(): Promise<string[]>;
+    archiveEntities(ids: string[]): Promise<number>;
+    browse(options?: {
+        domain?: string;
+        category?: string;
+        limit?: number;
+    }): Promise<BrowseResult[]>;
+    forget(id: string): Promise<void>;
+    createRelationship(sourceId: string, targetId: string, edgeType: string, metadata?: Record<string, unknown>): Promise<string | null>;
+    countEntityMentions(name: string, userId: string): Promise<number>;
+    schemaVersion(): Promise<number | null>;
+    ping(): Promise<boolean>;
+}