family-ai-agent 1.0.5 → 1.0.7

package/dist/database/adapters/postgres-adapter.js

@@ -0,0 +1,225 @@
+/**
+ * PostgreSQL Database Adapter
+ *
+ * Adapter for PostgreSQL with connection pooling and pgvector support.
+ */
+import pg from 'pg';
+import { drizzle } from 'drizzle-orm/node-postgres';
+import { BaseAdapter } from './base-adapter.js';
+import { createLogger } from '../../utils/logger.js';
+import { MemoryError } from '../../utils/errors.js';
+import * as schema from '../drizzle/schema/index.js';
+const { Pool } = pg;
+const logger = createLogger('PostgresAdapter');
+/**
+ * PostgreSQL database adapter with Drizzle ORM
+ */
+export class PostgresAdapter extends BaseAdapter {
+    type = 'postgresql';
+    pool = null;
+    drizzleInstance = null;
+    connectionString;
+    config;
+    constructor(connectionString, config) {
+        super();
+        this.connectionString = connectionString;
+        this.config = config;
+    }
+    /**
+     * Get Drizzle ORM instance
+     */
+    get drizzle() {
+        this.ensureInitialized();
+        return this.drizzleInstance;
+    }
+    /**
+     * Get raw pool (for advanced operations)
+     */
+    get rawPool() {
+        this.ensureInitialized();
+        return this.pool;
+    }
+    /**
+     * Initialize PostgreSQL connection pool
+     */
+    async initialize() {
+        if (this._isInitialized)
+            return;
+        try {
+            this.pool = new Pool({
+                connectionString: this.connectionString,
+                max: this.config.postgresPoolMax,
+                idleTimeoutMillis: this.config.postgresIdleTimeoutMs,
+                connectionTimeoutMillis: this.config.postgresConnectionTimeoutMs,
+            });
+            // Set up event handlers
+            this.pool.on('error', (err) => {
+                logger.error('Pool error', { error: err.message });
+            });
+            this.pool.on('connect', () => {
+                logger.debug('New client connected to pool');
+            });
+            // Create Drizzle instance
+            this.drizzleInstance = drizzle(this.pool, {
+                schema,
+                logger: process.env.NODE_ENV === 'development',
+            });
+            // Test connection and enable pgvector
+            await this.testConnection();
+            await this.enablePgVector();
+            this._isInitialized = true;
+            logger.info('PostgreSQL adapter initialized', {
+                host: this.config.postgresHost,
+                database: this.config.postgresDatabase,
+            });
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : 'Unknown error';
+            logger.error('Failed to initialize PostgreSQL', { error: message });
+            throw new MemoryError(`PostgreSQL initialization failed: ${message}`);
+        }
+    }
+    /**
+     * Test database connection
+     */
+    async testConnection() {
+        const client = await this.pool.connect();
+        try {
+            await client.query('SELECT 1');
+        }
+        finally {
+            client.release();
+        }
+    }
+    /**
+     * Enable pgvector extension if not already enabled
+     */
+    async enablePgVector() {
+        try {
+            await this.pool.query('CREATE EXTENSION IF NOT EXISTS vector');
+            logger.debug('pgvector extension enabled');
+        }
+        catch (error) {
+            logger.warn('Could not enable pgvector extension', {
+                error: error instanceof Error ? error.message : 'Unknown'
+            });
+            // Don't fail - extension might already exist or user might not have permissions
+        }
+    }
+    /**
+     * Execute a raw SQL query
+     */
+    async query(sqlText, params) {
+        this.ensureInitialized();
+        const start = Date.now();
+        try {
+            const result = await this.pool.query(sqlText, params);
+            const duration = Date.now() - start;
+            this.logQuery(sqlText, duration, result.rowCount);
+            return {
+                rows: result.rows,
+                rowCount: result.rowCount,
+            };
+        }
+        catch (error) {
+            this.logError(sqlText, error);
+            const message = error instanceof Error ? error.message : 'Unknown error';
+            throw new MemoryError(`Query failed: ${message}`);
+        }
+    }
+    /**
+     * Execute a transaction
+     */
+    async transaction(callback) {
+        this.ensureInitialized();
+        const client = await this.pool.connect();
+        try {
+            await client.query('BEGIN');
+            const transactionClient = {
+                query: async (sqlText, params) => {
+                    const result = await client.query(sqlText, params);
+                    return {
+                        rows: result.rows,
+                        rowCount: result.rowCount,
+                    };
+                },
+            };
+            const result = await callback(transactionClient);
+            await client.query('COMMIT');
+            return result;
+        }
+        catch (error) {
+            await client.query('ROLLBACK');
+            throw error;
+        }
+        finally {
+            client.release();
+        }
+    }
+    /**
+     * Check database health
+     */
+    async healthCheck() {
+        try {
+            const result = await this.query('SELECT 1 as ok');
+            return result.rows.length > 0 && result.rows[0].ok === 1;
+        }
+        catch {
+            return false;
+        }
+    }
+    /**
+     * Get connection pool statistics
+     */
+    getPoolStats() {
+        if (!this.pool) {
+            return { total: 0, idle: 0, waiting: 0 };
+        }
+        return {
+            total: this.pool.totalCount,
+            idle: this.pool.idleCount,
+            waiting: this.pool.waitingCount,
+        };
+    }
+    /**
+     * Close all connections
+     */
+    async close() {
+        if (this.pool) {
+            await this.pool.end();
+            this.pool = null;
+            this.drizzleInstance = null;
+            this._isInitialized = false;
+            logger.info('PostgreSQL pool closed');
+        }
+    }
+    /**
+     * Execute vector similarity search
+     */
+    async vectorSearch(table, embeddingColumn, queryEmbedding, options = {}) {
+        const { limit = 5, minSimilarity = 0.7, filters = {}, selectColumns = ['*'], } = options;
+        const embeddingStr = `[${queryEmbedding.join(',')}]`;
+        // Build WHERE clause
+        const whereConditions = [`1 - (${embeddingColumn} <=> $1::vector) >= $2`];
+        const params = [embeddingStr, minSimilarity];
+        let paramIndex = 3;
+        for (const [key, value] of Object.entries(filters)) {
+            whereConditions.push(`${key} = $${paramIndex}`);
+            params.push(value);
+            paramIndex++;
+        }
+        params.push(limit);
+        const columns = selectColumns.join(', ');
+        const sql = `
+      SELECT ${columns}, 1 - (${embeddingColumn} <=> $1::vector) as similarity
+      FROM ${table}
+      WHERE ${whereConditions.join(' AND ')}
+      ORDER BY similarity DESC
+      LIMIT $${paramIndex}
+    `;
+        const result = await this.query(sql, params);
+        return result.rows;
+    }
+}
+export default PostgresAdapter;
+//# sourceMappingURL=postgres-adapter.js.map

package/dist/database/adapters/postgres-adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"postgres-adapter.js","sourceRoot":"","sources":["../../../src/database/adapters/postgres-adapter.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,IAAI,CAAC;AAEpB,OAAO,EAAE,OAAO,EAAE,MAAM,2BAA2B,CAAC;AAIpD,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAOhD,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAEpD,OAAO,KAAK,MAAM,MAAM,4BAA4B,CAAC;AAErD,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;AACpB,MAAM,MAAM,GAAG,YAAY,CAAC,iBAAiB,CAAC,CAAC;AAE/C;;GAEG;AACH,MAAM,OAAO,eAAgB,SAAQ,WAAW;IACrC,IAAI,GAAG,YAAqB,CAAC;IAE9B,IAAI,GAAmB,IAAI,CAAC;IAC5B,eAAe,GAAyC,IAAI,CAAC;IAC7D,gBAAgB,CAAS;IACzB,MAAM,CAAiB;IAE/B,YAAY,gBAAwB,EAAE,MAAsB;QAC1D,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,IAAI,OAAO;QACT,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC,eAAgB,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,IAAI,OAAO;QACT,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC,IAAK,CAAC;IACpB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,cAAc;YAAE,OAAO;QAEhC,IAAI,CAAC;YACH,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,CAAC;gBACnB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;gBACvC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;gBAChC,iBAAiB,EAAE,IAAI,CAAC,MAAM,CAAC,qBAAqB;gBACpD,uBAAuB,EAAE,IAAI,CAAC,MAAM,CAAC,2BAA2B;aACjE,CAAC,CAAC;YAEH,wBAAwB;YACxB,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBAC5B,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;gBAC3B,MAAM,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;YAEH,0BAA0B;YAC1B,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE;gBACxC,MAAM;gBACN,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,aAAa;aAC/C,CAAC,CAAC;YAEH,sCAAsC;YACtC,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;YAC5B,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;YAE5B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC3B,MAAM,CAAC,IAAI,CAAC,gCAAgC,EAAE;gBAC5C,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;gBAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;aACvC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;YACpE,MAAM,IAAI,WAAW,CAAC,qCAAqC,OAAO,EAAE,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,cAAc;QAC1B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAK,CAAC,OAAO,EAAE,CAAC;QAC1C,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACjC,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,cAAc;QAC1B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,IAAK,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAC;YAChE,MAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,IAAI,CAAC,qCAAqC,EAAE;gBACjD,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;aAC1D,CAAC,CAAC;YACH,gFAAgF;QAClF,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CACT,OAAe,EACf,MAAkB;QAElB,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEzB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAK,CAAC,KAAK,CAAI,OAAO,EAAE,MAAM,CAAC,CAAC;YAC1D,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;YACpC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;YAElD,OAAO;gBACL,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;aAC1B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAC9B,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,MAAM,IAAI,WAAW,CAAC,iBAAiB,OAAO,EAAE,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CACf,QAAmD;QAEnD,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAK,CAAC,OAAO,EAAE,CAAC;QAE1C,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAE5B,MAAM,iBAAiB,GAAsB;gBAC3C,KAAK,EAAE,KAAK,EACV,OAAe,EACf,MAAkB,EACO,EAAE;oBAC3B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAI,OAAO,EAAE,MAAM,CAAC,CAAC;oBACtD,OAAO;wBACL,IAAI,EAAE,MAAM,CAAC,IAAI;wBACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;qBAC1B,CAAC;gBACJ,CAAC;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,iBAAiB,CAAC,CAAC;YACjD,MAAM,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC7B,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC/B,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW;QACf,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAiB,gBAAgB,CAAC,CAAC;YAClE,OAAO,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,EAAE,KAAK,CAAC,CAAC;QAC5D,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,YAAY;QACV,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACf,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;QAC3C,CAAC;QAED,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;YAC3B,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS;YACzB,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY;SAChC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;YACjB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;YAC5B,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;YAC5B,MAAM,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAChB,KAAa,EACb,eAAuB,EACvB,cAAwB,EACxB,UAKI,EAAE;QAEN,MAAM,EACJ,KAAK,GAAG,CAAC,EACT,aAAa,GAAG,GAAG,EACnB,OAAO,GAAG,EAAE,EACZ,aAAa,GAAG,CAAC,GAAG,CAAC,GACtB,GAAG,OAAO,CAAC;QAEZ,MAAM,YAAY,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;QAErD,qBAAqB;QACrB,MAAM,eAAe,GAAG,CAAC,QAAQ,eAAe,wBAAwB,CAAC,CAAC;QAC1E,MAAM,MAAM,GAAc,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;QACxD,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACnD,eAAe,CAAC,IAAI,CAAC,GAAG,GAAG,OAAO,UAAU,EAAE,CAAC,CAAC;YAChD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnB,UAAU,EAAE,CAAC;QACf,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEnB,MAAM,OAAO,GAAG,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,GAAG,GAAG;eACD,OAAO,UAAU,eAAe;aAClC,KAAK;cACJ,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC;;eAE5B,UAAU;KACpB,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAA6B,GAAG,EAAE,MAAM,CAAC,CAAC;QACzE,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC;CACF;AAED,eAAe,eAAe,CAAC"}

package/dist/database/adapters/sqlite-adapter.d.ts

@@ -0,0 +1,72 @@
+/**
+ * SQLite Database Adapter
+ *
+ * Adapter for SQLite using better-sqlite3 for embedded database support.
+ * Provides zero-config database that works out of the box.
+ */
+import Database from 'better-sqlite3';
+import type { BetterSQLite3Database } from 'drizzle-orm/better-sqlite3';
+import { BaseAdapter } from './base-adapter.js';
+import type { QueryResult, TransactionClient, DatabaseConfig } from '../types.js';
+import * as schema from '../drizzle/schema/index.js';
+/**
+ * SQLite database adapter with Drizzle ORM
+ */
+export declare class SQLiteAdapter extends BaseAdapter {
+    readonly type: "sqlite";
+    private db;
+    private drizzleInstance;
+    private dbPath;
+    private config;
+    constructor(dbPath: string, config: DatabaseConfig);
+    /**
+     * Get Drizzle ORM instance
+     */
+    get drizzle(): BetterSQLite3Database<typeof schema>;
+    /**
+     * Get raw database (for advanced operations)
+     */
+    get rawDb(): Database.Database;
+    /**
+     * Initialize SQLite database
+     */
+    initialize(): Promise<void>;
+    /**
+     * Initialize database schema
+     */
+    private initializeSchema;
+    /**
+     * Execute a raw SQL query
+     */
+    query<T extends Record<string, any> = Record<string, any>>(sqlText: string, params?: unknown[]): Promise<QueryResult<T>>;
+    /**
+     * Execute a transaction
+     */
+    transaction<T>(callback: (client: TransactionClient) => Promise<T>): Promise<T>;
+    /**
+     * Check database health
+     */
+    healthCheck(): Promise<boolean>;
+    /**
+     * Close database connection
+     */
+    close(): Promise<void>;
+    /**
+     * Get database file path
+     */
+    getPath(): string;
+    /**
+     * Get database file size in bytes
+     */
+    getFileSize(): number;
+    /**
+     * Vacuum database to reclaim space
+     */
+    vacuum(): void;
+    /**
+     * Checkpoint WAL to main database
+     */
+    checkpoint(): void;
+}
+export default SQLiteAdapter;
+//# sourceMappingURL=sqlite-adapter.d.ts.map

package/dist/database/adapters/sqlite-adapter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sqlite-adapter.d.ts","sourceRoot":"","sources":["../../../src/database/adapters/sqlite-adapter.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AAEtC,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAIxE,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,EACV,WAAW,EACX,iBAAiB,EACjB,cAAc,EACf,MAAM,aAAa,CAAC;AAIrB,OAAO,KAAK,MAAM,MAAM,4BAA4B,CAAC;AAIrD;;GAEG;AACH,qBAAa,aAAc,SAAQ,WAAW;IAC5C,QAAQ,CAAC,IAAI,EAAG,QAAQ,CAAU;IAElC,OAAO,CAAC,EAAE,CAAkC;IAC5C,OAAO,CAAC,eAAe,CAAqD;IAC5E,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAAiB;gBAEnB,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc;IAMlD;;OAEG;IACH,IAAI,OAAO,IAAI,qBAAqB,CAAC,OAAO,MAAM,CAAC,CAGlD;IAED;;OAEG;IACH,IAAI,KAAK,IAAI,QAAQ,CAAC,QAAQ,CAG7B;IAED;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAuCjC;;OAEG;YACW,gBAAgB;IA+J9B;;OAEG;IACG,KAAK,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAC7D,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,OAAO,EAAE,GACjB,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;IAyD1B;;OAEG;IACG,WAAW,CAAC,CAAC,EACjB,QAAQ,EAAE,CAAC,MAAM,EAAE,iBAAiB,KAAK,OAAO,CAAC,CAAC,CAAC,GAClD,OAAO,CAAC,CAAC,CAAC;IAuBb;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;IASrC;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAU5B;;OAEG;IACH,OAAO,IAAI,MAAM;IAIjB;;OAEG;IACH,WAAW,IAAI,MAAM;IAWrB;;OAEG;IACH,MAAM,IAAI,IAAI;IAMd;;OAEG;IACH,UAAU,IAAI,IAAI;CAKnB;AAED,eAAe,aAAa,CAAC"}

package/dist/database/adapters/sqlite-adapter.js

@@ -0,0 +1,368 @@
+/**
+ * SQLite Database Adapter
+ *
+ * Adapter for SQLite using better-sqlite3 for embedded database support.
+ * Provides zero-config database that works out of the box.
+ */
+import Database from 'better-sqlite3';
+import { drizzle } from 'drizzle-orm/better-sqlite3';
+import { existsSync, mkdirSync } from 'fs';
+import { dirname } from 'path';
+import { BaseAdapter } from './base-adapter.js';
+import { createLogger } from '../../utils/logger.js';
+import { MemoryError } from '../../utils/errors.js';
+import * as schema from '../drizzle/schema/index.js';
+const logger = createLogger('SQLiteAdapter');
+/**
+ * SQLite database adapter with Drizzle ORM
+ */
+export class SQLiteAdapter extends BaseAdapter {
+    type = 'sqlite';
+    db = null;
+    drizzleInstance = null;
+    dbPath;
+    config;
+    constructor(dbPath, config) {
+        super();
+        this.dbPath = dbPath;
+        this.config = config;
+    }
+    /**
+     * Get Drizzle ORM instance
+     */
+    get drizzle() {
+        this.ensureInitialized();
+        return this.drizzleInstance;
+    }
+    /**
+     * Get raw database (for advanced operations)
+     */
+    get rawDb() {
+        this.ensureInitialized();
+        return this.db;
+    }
+    /**
+     * Initialize SQLite database
+     */
+    async initialize() {
+        if (this._isInitialized)
+            return;
+        try {
+            // Ensure directory exists
+            const dir = dirname(this.dbPath);
+            if (!existsSync(dir)) {
+                mkdirSync(dir, { recursive: true });
+                logger.debug('Created database directory', { path: dir });
+            }
+            // Open database
+            this.db = new Database(this.dbPath);
+            // Configure for better performance
+            this.db.pragma('journal_mode = WAL');
+            this.db.pragma('synchronous = NORMAL');
+            this.db.pragma('cache_size = -64000'); // 64MB cache
+            this.db.pragma('foreign_keys = ON');
+            this.db.pragma('temp_store = MEMORY');
+            // Create Drizzle instance
+            this.drizzleInstance = drizzle(this.db, {
+                schema,
+                logger: process.env.NODE_ENV === 'development',
+            });
+            // Initialize schema
+            await this.initializeSchema();
+            this._isInitialized = true;
+            logger.info('SQLite adapter initialized', { path: this.dbPath });
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : 'Unknown error';
+            logger.error('Failed to initialize SQLite', { error: message });
+            throw new MemoryError(`SQLite initialization failed: ${message}`);
+        }
+    }
+    /**
+     * Initialize database schema
+     */
+    async initializeSchema() {
+        const statements = [
+            // Conversations
+            `CREATE TABLE IF NOT EXISTS conversations (
+        id TEXT PRIMARY KEY,
+        thread_id TEXT NOT NULL,
+        user_id TEXT,
+        created_at INTEGER NOT NULL,
+        updated_at INTEGER NOT NULL
+      )`,
+            `CREATE INDEX IF NOT EXISTS idx_conversations_thread ON conversations(thread_id)`,
+            `CREATE INDEX IF NOT EXISTS idx_conversations_user ON conversations(user_id)`,
+            // Messages
+            `CREATE TABLE IF NOT EXISTS messages (
+        id TEXT PRIMARY KEY,
+        conversation_id TEXT NOT NULL REFERENCES conversations(id) ON DELETE CASCADE,
+        role TEXT NOT NULL,
+        content TEXT NOT NULL,
+        metadata TEXT NOT NULL DEFAULT '{}',
+        created_at INTEGER NOT NULL
+      )`,
+            `CREATE INDEX IF NOT EXISTS idx_messages_conversation ON messages(conversation_id)`,
+            // Long-term memories
+            `CREATE TABLE IF NOT EXISTS long_term_memories (
+        id TEXT PRIMARY KEY,
+        user_id TEXT,
+        memory_type TEXT NOT NULL,
+        content TEXT NOT NULL,
+        embedding TEXT NOT NULL,
+        importance REAL NOT NULL DEFAULT 0.5,
+        access_count INTEGER NOT NULL DEFAULT 0,
+        last_accessed INTEGER,
+        metadata TEXT NOT NULL DEFAULT '{}',
+        created_at INTEGER NOT NULL,
+        updated_at INTEGER NOT NULL
+      )`,
+            `CREATE INDEX IF NOT EXISTS idx_memories_user ON long_term_memories(user_id)`,
+            `CREATE INDEX IF NOT EXISTS idx_memories_type ON long_term_memories(memory_type)`,
+            // Documents
+            `CREATE TABLE IF NOT EXISTS documents (
+        id TEXT PRIMARY KEY,
+        user_id TEXT,
+        filename TEXT NOT NULL,
+        file_type TEXT,
+        file_size INTEGER,
+        content TEXT,
+        metadata TEXT NOT NULL DEFAULT '{}',
+        created_at INTEGER NOT NULL
+      )`,
+            `CREATE INDEX IF NOT EXISTS idx_documents_user ON documents(user_id)`,
+            // Document chunks
+            `CREATE TABLE IF NOT EXISTS document_chunks (
+        id TEXT PRIMARY KEY,
+        document_id TEXT NOT NULL REFERENCES documents(id) ON DELETE CASCADE,
+        chunk_index INTEGER NOT NULL,
+        content TEXT NOT NULL,
+        embedding TEXT NOT NULL,
+        metadata TEXT NOT NULL DEFAULT '{}',
+        created_at INTEGER NOT NULL
+      )`,
+            `CREATE INDEX IF NOT EXISTS idx_chunks_document ON document_chunks(document_id)`,
+            // Checkpoints
+            `CREATE TABLE IF NOT EXISTS checkpoints (
+        thread_id TEXT NOT NULL,
+        checkpoint_ns TEXT NOT NULL DEFAULT '',
+        checkpoint_id TEXT NOT NULL,
+        parent_checkpoint_id TEXT,
+        type TEXT,
+        checkpoint TEXT NOT NULL,
+        metadata TEXT NOT NULL DEFAULT '{}',
+        created_at INTEGER NOT NULL,
+        PRIMARY KEY (thread_id, checkpoint_ns, checkpoint_id)
+      )`,
+            // Checkpoint writes
+            `CREATE TABLE IF NOT EXISTS checkpoint_writes (
+        thread_id TEXT NOT NULL,
+        checkpoint_ns TEXT NOT NULL DEFAULT '',
+        checkpoint_id TEXT NOT NULL,
+        task_id TEXT NOT NULL,
+        idx INTEGER NOT NULL,
+        channel TEXT NOT NULL,
+        type TEXT,
+        value TEXT,
+        created_at INTEGER NOT NULL,
+        PRIMARY KEY (thread_id, checkpoint_ns, checkpoint_id, task_id, idx)
+      )`,
+            // Audit logs
+            `CREATE TABLE IF NOT EXISTS audit_logs (
+        id TEXT PRIMARY KEY,
+        user_id TEXT,
+        agent_id TEXT,
+        action_type TEXT NOT NULL,
+        action_details TEXT NOT NULL DEFAULT '{}',
+        input_hash TEXT,
+        output_hash TEXT,
+        status TEXT NOT NULL DEFAULT 'success',
+        error_message TEXT,
+        execution_time_ms INTEGER,
+        ip_address TEXT,
+        user_agent TEXT,
+        success INTEGER NOT NULL DEFAULT 1,
+        created_at INTEGER NOT NULL
+      )`,
+            `CREATE INDEX IF NOT EXISTS idx_audit_user ON audit_logs(user_id)`,
+            `CREATE INDEX IF NOT EXISTS idx_audit_action ON audit_logs(action_type)`,
+            `CREATE INDEX IF NOT EXISTS idx_audit_created ON audit_logs(created_at)`,
+            // Tasks
+            `CREATE TABLE IF NOT EXISTS tasks (
+        id TEXT PRIMARY KEY,
+        user_id TEXT,
+        task_type TEXT NOT NULL,
+        priority INTEGER NOT NULL DEFAULT 0,
+        status TEXT NOT NULL DEFAULT 'pending',
+        payload TEXT NOT NULL DEFAULT '{}',
+        result TEXT,
+        error_message TEXT,
+        scheduled_at INTEGER,
+        started_at INTEGER,
+        completed_at INTEGER,
+        retry_count INTEGER NOT NULL DEFAULT 0,
+        max_retries INTEGER NOT NULL DEFAULT 3,
+        created_at INTEGER NOT NULL,
+        updated_at INTEGER NOT NULL
+      )`,
+            `CREATE INDEX IF NOT EXISTS idx_tasks_user ON tasks(user_id)`,
+            `CREATE INDEX IF NOT EXISTS idx_tasks_status ON tasks(status)`,
+            // Schema version tracking
+            `CREATE TABLE IF NOT EXISTS schema_versions (
+        version INTEGER PRIMARY KEY,
+        name TEXT NOT NULL,
+        applied_at INTEGER NOT NULL,
+        checksum TEXT
+      )`,
+        ];
+        for (const statement of statements) {
+            try {
+                this.db.exec(statement);
+            }
+            catch (error) {
+                // Ignore errors for existing objects
+                const message = error instanceof Error ? error.message : '';
+                if (!message.includes('already exists')) {
+                    throw error;
+                }
+            }
+        }
+        logger.debug('SQLite schema initialized');
+    }
+    /**
+     * Execute a raw SQL query
+     */
+    async query(sqlText, params) {
+        this.ensureInitialized();
+        const start = Date.now();
+        try {
+            // Convert PostgreSQL placeholders to SQLite
+            const sqliteSql = this.convertToSqlite(sqlText);
+            const sqliteParams = params ?? [];
+            const isSelect = this.isSelectQuery(sqlText);
+            const hasReturning = this.hasReturning(sqlText);
+            let rows = [];
+            let rowCount = 0;
+            if (isSelect) {
+                const stmt = this.db.prepare(sqliteSql);
+                rows = stmt.all(...sqliteParams);
+                rowCount = rows.length;
+            }
+            else if (hasReturning) {
+                // Handle INSERT/UPDATE/DELETE with RETURNING
+                const returning = this.extractReturning(sqlText);
+                const tableName = this.extractTableName(sqlText);
+                // Remove RETURNING clause for SQLite
+                const sqlWithoutReturning = sqliteSql.replace(/\s+RETURNING\s+.+$/i, '');
+                const stmt = this.db.prepare(sqlWithoutReturning);
+                const info = stmt.run(...sqliteParams);
+                rowCount = info.changes;
+                if (returning && tableName && info.lastInsertRowid) {
+                    // Fetch the inserted/updated row
+                    const selectStmt = this.db.prepare(`SELECT ${returning} FROM ${tableName} WHERE rowid = ?`);
+                    const row = selectStmt.get(info.lastInsertRowid);
+                    if (row) {
+                        rows = [row];
+                    }
+                }
+            }
+            else {
+                const stmt = this.db.prepare(sqliteSql);
+                const info = stmt.run(...sqliteParams);
+                rowCount = info.changes;
+            }
+            const duration = Date.now() - start;
+            this.logQuery(sqlText, duration, rowCount);
+            return { rows, rowCount };
+        }
+        catch (error) {
+            this.logError(sqlText, error);
+            const message = error instanceof Error ? error.message : 'Unknown error';
+            throw new MemoryError(`Query failed: ${message}`);
+        }
+    }
+    /**
+     * Execute a transaction
+     */
+    async transaction(callback) {
+        this.ensureInitialized();
+        const transactionClient = {
+            query: async (sqlText, params) => {
+                return this.query(sqlText, params);
+            },
+        };
+        try {
+            this.db.exec('BEGIN');
+            const result = await callback(transactionClient);
+            this.db.exec('COMMIT');
+            return result;
+        }
+        catch (error) {
+            this.db.exec('ROLLBACK');
+            throw error;
+        }
+    }
+    /**
+     * Check database health
+     */
+    async healthCheck() {
+        try {
+            const result = await this.query('SELECT 1 as ok');
+            return result.rows.length > 0 && result.rows[0].ok === 1;
+        }
+        catch {
+            return false;
+        }
+    }
+    /**
+     * Close database connection
+     */
+    async close() {
+        if (this.db) {
+            this.db.close();
+            this.db = null;
+            this.drizzleInstance = null;
+            this._isInitialized = false;
+            logger.info('SQLite database closed');
+        }
+    }
+    /**
+     * Get database file path
+     */
+    getPath() {
+        return this.dbPath;
+    }
+    /**
+     * Get database file size in bytes
+     */
+    getFileSize() {
+        if (!this.db)
+            return 0;
+        try {
+            const stats = this.db.pragma('page_count');
+            const pageSize = this.db.pragma('page_size');
+            return (stats[0]?.page_count ?? 0) * (pageSize[0]?.page_size ?? 4096);
+        }
+        catch {
+            return 0;
+        }
+    }
+    /**
+     * Vacuum database to reclaim space
+     */
+    vacuum() {
+        this.ensureInitialized();
+        this.db.exec('VACUUM');
+        logger.debug('Database vacuumed');
+    }
+    /**
+     * Checkpoint WAL to main database
+     */
+    checkpoint() {
+        this.ensureInitialized();
+        this.db.pragma('wal_checkpoint(TRUNCATE)');
+        logger.debug('WAL checkpoint completed');
+    }
+}
+export default SQLiteAdapter;
+//# sourceMappingURL=sqlite-adapter.js.map