@elizaos/plugin-sql 1.0.0-alpha.20 → 1.0.0-alpha.21

package/README.md

@@ -9,19 +9,6 @@ A PostgreSQL database adapter built with Drizzle ORM for the ElizaOS ecosystem.
 bun add @elizaos/plugin-sql
 ```
 
-## Database Schema
-
-The adapter includes the following tables:
-- accounts
-- cache
-- embeddings
-- goals
-- logs
-- memories
-- participants
-- relationships
-- rooms
-
 ## Vector Dimensions
 
 The adapter supports the following vector dimensions:
@@ -55,7 +42,7 @@ The adapter is typically used as part of the ElizaOS runtime:
 
 ```typescript
 async function findDatabaseAdapter(runtime: IAgentRuntime) {
-    let adapter = runtime.getDatabaseAdapter();
+    let adapter = runtime;
     
     if (!adapter) {
         const drizzleAdapterPlugin = await import('@elizaos/plugin-sql');

package/dist/{chunk-LLEN3JTK.js → chunk-2FOOMI3E.js}

@@ -93,6 +93,21 @@ var PGliteClientManager = class {
       await this.gracefulShutdown();
     });
   }
+  /**
+   * Asynchronously checks if migrations exist in the database.
+   * @returns {Promise<boolean>} A Promise that resolves to true if migrations exist, otherwise false.
+   */
+  async hasMigrations() {
+    try {
+      const result = await this.client.query(
+        "SELECT EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = '__drizzle_migrations')"
+      );
+      return result.rows[0].exists;
+    } catch (error) {
+      logger.error("Failed to check migrations:", error);
+      return false;
+    }
+  }
   /**
    * Initializes the client for PGlite.
    *
@@ -134,12 +149,16 @@ var PGliteClientManager = class {
   async runMigrations() {
     try {
       const db = drizzle(this.client);
+      if (await this.hasMigrations()) {
+        logger.info("Migrations already exist, skipping...");
+        return;
+      }
       await migrate(db, {
         migrationsFolder: path.resolve(__dirname, "../drizzle/migrations")
       });
       logger.info("Migrations completed successfully!");
     } catch (error) {
-      logger.error("Failed to run database migrations:", error);
+      logger.error("Failed to run database migrations (pglite):", error);
     }
   }
 };
@@ -296,6 +315,21 @@ var PostgresConnectionManager = class {
   async close() {
     await this.cleanup();
   }
+  /**
+   * Asynchronously checks if migrations exist in the database.
+   * @returns {Promise<boolean>} A Promise that resolves to true if migrations exist, otherwise false.
+   */
+  async hasMigrations() {
+    try {
+      const result = await this.pool.query(
+        "SELECT EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = '__drizzle_migrations')"
+      );
+      return result.rows[0].exists;
+    } catch (error) {
+      logger2.error("Failed to check migrations:", error);
+      return false;
+    }
+  }
   /**
    * Cleans up and closes the database pool.
    * @returns {Promise<void>} A Promise that resolves when the database pool is closed.
@@ -316,12 +350,16 @@ var PostgresConnectionManager = class {
   async runMigrations() {
     try {
       const db = drizzle2(this.pool);
+      if (await this.hasMigrations()) {
+        logger2.info("Migrations already exist, skipping...");
+        return;
+      }
       await migrate2(db, {
         migrationsFolder: path2.resolve(__dirname2, "../drizzle/migrations")
       });
       logger2.info("Migrations completed successfully!");
     } catch (error) {
-      logger2.error("Failed to run database migrations:", error);
+      logger2.error("Failed to run database migrations (pg):", error);
     }
   }
 };
@@ -331,4 +369,4 @@ export {
   PGliteClientManager,
   PostgresConnectionManager
 };
-//# sourceMappingURL=chunk-LLEN3JTK.js.map
+//# sourceMappingURL=chunk-2FOOMI3E.js.map

package/dist/chunk-2FOOMI3E.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/pg-lite/manager.ts","../src/pg/manager.ts"],"sourceRoot":"./","sourcesContent":["import path from \"node:path\";\nimport { fileURLToPath } from \"node:url\";\nimport { PGlite, type PGliteOptions } from \"@electric-sql/pglite\";\nimport { fuzzystrmatch } from \"@electric-sql/pglite/contrib/fuzzystrmatch\";\nimport { vector } from \"@electric-sql/pglite/vector\";\nimport { logger } from \"@elizaos/core\";\nimport { drizzle } from \"drizzle-orm/pglite\";\nimport { migrate } from \"drizzle-orm/pglite/migrator\";\nimport type { IDatabaseClientManager } from \"../types\";\n\nconst __filename = fileURLToPath(import.meta.url);\nconst __dirname = path.dirname(__filename);\n\n/**\n * Class representing a database client manager for PGlite.\n * @implements { IDatabaseClientManager }\n */\nexport class PGliteClientManager implements IDatabaseClientManager<PGlite> {\n\tprivate client: PGlite;\n\tprivate shuttingDown = false;\n\tprivate readonly shutdownTimeout = 500;\n\n\t/**\n\t * Constructor for creating a new instance of PGlite with the provided options.\n\t * Initializes the PGlite client with additional extensions.\n\t * @param {PGliteOptions} options - The options to configure the PGlite client.\n\t */\n\tconstructor(options: PGliteOptions) {\n\t\tthis.client = new PGlite({\n\t\t\t...options,\n\t\t\textensions: {\n\t\t\t\tvector,\n\t\t\t\tfuzzystrmatch,\n\t\t\t},\n\t\t});\n\t\tthis.setupShutdownHandlers();\n\t}\n\n\t/**\n\t * Retrieves the PostgreSQL lite connection.\n\t *\n\t * @returns {PGlite} The PostgreSQL lite connection.\n\t * @throws {Error} If the client manager is currently shutting down.\n\t */\n\tpublic getConnection(): PGlite {\n\t\tif (this.shuttingDown) {\n\t\t\tthrow new Error(\"Client manager is shutting down\");\n\t\t}\n\t\treturn this.client;\n\t}\n\n\t/**\n\t * Initiates a graceful shutdown of the PGlite client.\n\t * Checks if the client is already in the process of shutting down.\n\t * Logs the start of shutdown process and sets shuttingDown flag to true.\n\t * Sets a timeout for the shutdown process and forces closure of database connection if timeout is reached.\n\t * Handles the shutdown process, closes the client connection, clears the timeout, and logs the completion of shutdown.\n\t * Logs any errors that occur during the shutdown process.\n\t */\n\tprivate async gracefulShutdown() {\n\t\tif (this.shuttingDown) {\n\t\t\treturn;\n\t\t}\n\n\t\tthis.shuttingDown = true;\n\t\tlogger.info(\"Starting graceful shutdown of PGlite client...\");\n\n\t\tconst timeout = setTimeout(() => {\n\t\t\tlogger.warn(\n\t\t\t\t\"Shutdown timeout reached, forcing database connection closure...\",\n\t\t\t);\n\t\t\tthis.client.close().finally(() => {\n\t\t\t\tprocess.exit(1);\n\t\t\t});\n\t\t}, this.shutdownTimeout);\n\n\t\ttry {\n\t\t\tawait this.client.close();\n\t\t\tclearTimeout(timeout);\n\t\t\tlogger.info(\"PGlite client shutdown completed successfully\");\n\t\t\tprocess.exit(0);\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Error during graceful shutdown:\", error);\n\t\t\tprocess.exit(1);\n\t\t}\n\t}\n\n\t/**\n\t * Sets up shutdown handlers for SIGINT, SIGTERM, and beforeExit events to gracefully shutdown the application.\n\t * @private\n\t */\n\tprivate setupShutdownHandlers() {\n\t\tprocess.on(\"SIGINT\", async () => {\n\t\t\tawait this.gracefulShutdown();\n\t\t});\n\n\t\tprocess.on(\"SIGTERM\", async () => {\n\t\t\tawait this.gracefulShutdown();\n\t\t});\n\n\t\tprocess.on(\"beforeExit\", async () => {\n\t\t\tawait this.gracefulShutdown();\n\t\t});\n\t}\n\n\t/**\n\t * Asynchronously checks if migrations exist in the database.\n\t * @returns {Promise<boolean>} A Promise that resolves to true if migrations exist, otherwise false.\n\t */\n\tprivate async hasMigrations(): Promise<boolean> {\n\t\ttry {\n\t\t\tconst result = await this.client.query(\n\t\t\t\t\"SELECT EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = '__drizzle_migrations')\"\n\t\t\t);\n\t\t\treturn (result.rows[0] as any).exists;\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Failed to check migrations:\", error);\n\t\t\treturn false;\n\t\t}\n\t}\n\n\n\t/**\n\t * Initializes the client for PGlite.\n\t *\n\t * @returns {Promise<void>} A Promise that resolves when the client is initialized successfully\n\t */\n\tpublic async initialize(): Promise<void> {\n\t\ttry {\n\t\t\tawait this.client.waitReady;\n\t\t\tlogger.info(\"PGlite client initialized successfully\");\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Failed to initialize PGlite client:\", error);\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\t/**\n\t * Asynchronously closes the resource. If the resource is not already shutting down,\n\t * it performs a graceful shutdown before closing.\n\t *\n\t * @returns A promise that resolves once the resource has been closed.\n\t */\n\tpublic async close(): Promise<void> {\n\t\tif (!this.shuttingDown) {\n\t\t\tawait this.gracefulShutdown();\n\t\t}\n\t}\n\n\t/**\n\t * Check if the system is currently shutting down.\n\t *\n\t * @returns {boolean} True if the system is shutting down, false otherwise.\n\t */\n\tpublic isShuttingDown(): boolean {\n\t\treturn this.shuttingDown;\n\t}\n\n\t/**\n\t * Asynchronously runs database migrations using Drizzle.\n\t *\n\t * @returns {Promise<void>} A Promise that resolves once the migrations are completed successfully.\n\t */\n\tasync runMigrations(): Promise<void> {\n\t\ttry {\n\t\t\tconst db = drizzle(this.client);\n\t\t\tif (await this.hasMigrations()) {\n\t\t\t\tlogger.info(\"Migrations already exist, skipping...\");\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tawait migrate(db, {\n\t\t\t\tmigrationsFolder: path.resolve(__dirname, \"../drizzle/migrations\"),\n\t\t\t});\n\t\t\tlogger.info(\"Migrations completed successfully!\");\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Failed to run database migrations (pglite):\", error);\n\t\t\t// throw error;\n\t\t}\n\t}\n}\n","import path from \"node:path\";\nimport { fileURLToPath } from \"node:url\";\nimport { logger } from \"@elizaos/core\";\nimport { drizzle } from \"drizzle-orm/node-postgres\";\nimport { migrate } from \"drizzle-orm/node-postgres/migrator\";\nimport pkg, { type Pool as PgPool } from \"pg\";\nimport type { IDatabaseClientManager } from \"../types\";\n\nconst { Pool } = pkg;\n\nconst __filename = fileURLToPath(import.meta.url);\nconst __dirname = path.dirname(__filename);\n\n/**\n * Manages connections to a PostgreSQL database using a connection pool.\n * Implements IDatabaseClientManager interface.\n */\n\nexport class PostgresConnectionManager\n\timplements IDatabaseClientManager<PgPool>\n{\n\tprivate pool: PgPool;\n\tprivate isShuttingDown = false;\n\tprivate readonly connectionTimeout: number = 5000;\n\n\t/**\n\t * Constructor for creating a connection pool.\n\t * @param {string} connectionString - The connection string used to connect to the database.\n\t */\n\tconstructor(connectionString: string) {\n\t\tconst defaultConfig = {\n\t\t\tmax: 20,\n\t\t\tidleTimeoutMillis: 30000,\n\t\t\tconnectionTimeoutMillis: this.connectionTimeout,\n\t\t};\n\n\t\tthis.pool = new Pool({\n\t\t\t...defaultConfig,\n\t\t\tconnectionString,\n\t\t});\n\n\t\tthis.pool.on(\"error\", (err) => {\n\t\t\tlogger.error(\"Unexpected pool error\", err);\n\t\t\tthis.handlePoolError(err);\n\t\t});\n\n\t\tthis.setupPoolErrorHandling();\n\t\tthis.testConnection();\n\t}\n\n\t/**\n\t * Handles a pool error by attempting to reconnect the pool.\n\t *\n\t * @param {Error} error The error that occurred in the pool.\n\t * @throws {Error} If failed to reconnect the pool.\n\t */\n\tprivate async handlePoolError(error: Error) {\n\t\tlogger.error(\"Pool error occurred, attempting to reconnect\", {\n\t\t\terror: error.message,\n\t\t});\n\n\t\ttry {\n\t\t\tawait this.pool.end();\n\n\t\t\tthis.pool = new Pool({\n\t\t\t\t...this.pool.options,\n\t\t\t\tconnectionTimeoutMillis: this.connectionTimeout,\n\t\t\t});\n\n\t\t\tawait this.testConnection();\n\t\t\tlogger.success(\"Pool reconnection successful\");\n\t\t} catch (reconnectError) {\n\t\t\tlogger.error(\"Failed to reconnect pool\", {\n\t\t\t\terror:\n\t\t\t\t\treconnectError instanceof Error\n\t\t\t\t\t\t? reconnectError.message\n\t\t\t\t\t\t: String(reconnectError),\n\t\t\t});\n\t\t\tthrow reconnectError;\n\t\t}\n\t}\n\n\t/**\n\t * Asynchronously tests the database connection by executing a query to get the current timestamp.\n\t *\n\t * @returns {Promise<boolean>} - A Promise that resolves to true if the database connection test is successful.\n\t */\n\tasync testConnection(): Promise<boolean> {\n\t\tlet client: pkg.PoolClient | null = null;\n\t\ttry {\n\t\t\tclient = await this.pool.connect();\n\t\t\tconst result = await client.query(\"SELECT NOW()\");\n\t\t\tlogger.success(\"Database connection test successful:\", result.rows[0]);\n\t\t\treturn true;\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Database connection test failed:\", error);\n\t\t\tthrow new Error(\n\t\t\t\t`Failed to connect to database: ${(error as Error).message}`,\n\t\t\t);\n\t\t} finally {\n\t\t\tif (client) client.release();\n\t\t}\n\t}\n\n\t/**\n\t * Sets up event listeners to handle pool cleanup on SIGINT, SIGTERM, and beforeExit events.\n\t */\n\tprivate setupPoolErrorHandling() {\n\t\tprocess.on(\"SIGINT\", async () => {\n\t\t\tawait this.cleanup();\n\t\t\tprocess.exit(0);\n\t\t});\n\n\t\tprocess.on(\"SIGTERM\", async () => {\n\t\t\tawait this.cleanup();\n\t\t\tprocess.exit(0);\n\t\t});\n\n\t\tprocess.on(\"beforeExit\", async () => {\n\t\t\tawait this.cleanup();\n\t\t});\n\t}\n\n\t/**\n\t * Get the connection pool.\n\t * @returns {PgPool} The connection pool\n\t * @throws {Error} If the connection manager is shutting down or an error occurs when trying to get the connection from the pool\n\t */\n\tpublic getConnection(): PgPool {\n\t\tif (this.isShuttingDown) {\n\t\t\tthrow new Error(\"Connection manager is shutting down\");\n\t\t}\n\n\t\ttry {\n\t\t\treturn this.pool;\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Failed to get connection from pool:\", error);\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\t/**\n\t * Asynchronously acquires a database client from the connection pool.\n\t *\n\t * @returns {Promise<pkg.PoolClient>} A Promise that resolves with the acquired database client.\n\t * @throws {Error} If an error occurs while acquiring the database client.\n\t */\n\tpublic async getClient(): Promise<pkg.PoolClient> {\n\t\ttry {\n\t\t\treturn await this.pool.connect();\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Failed to acquire a database client:\", error);\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\t/**\n\t * Initializes the PostgreSQL connection manager by testing the connection and logging the result.\n\t *\n\t * @returns {Promise<void>} A Promise that resolves once the manager is successfully initialized\n\t * @throws {Error} If there is an error initializing the connection manager\n\t */\n\tpublic async initialize(): Promise<void> {\n\t\ttry {\n\t\t\tawait this.testConnection();\n\t\t\tlogger.info(\"PostgreSQL connection manager initialized successfully\");\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Failed to initialize connection manager:\", error);\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\t/**\n\t * Asynchronously close the current process by executing a cleanup function.\n\t * @returns A promise that resolves once the cleanup is complete.\n\t */\n\tpublic async close(): Promise<void> {\n\t\tawait this.cleanup();\n\t}\n\n\t/**\n\t * Asynchronously checks if migrations exist in the database.\n\t * @returns {Promise<boolean>} A Promise that resolves to true if migrations exist, otherwise false.\n\t */\n\tprivate async hasMigrations(): Promise<boolean> {\n\t\ttry {\n\t\t\tconst result = await this.pool.query(\n\t\t\t\t\"SELECT EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = '__drizzle_migrations')\"\n\t\t\t);\n\t\t\treturn result.rows[0].exists;\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Failed to check migrations:\", error);\n\t\t\treturn false;\n\t\t}\n\t}\n\n\t/**\n\t * Cleans up and closes the database pool.\n\t * @returns {Promise<void>} A Promise that resolves when the database pool is closed.\n\t */\n\tasync cleanup(): Promise<void> {\n\t\ttry {\n\t\t\tawait this.pool.end();\n\t\t\tlogger.info(\"Database pool closed\");\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Error closing database pool:\", error);\n\t\t}\n\t}\n\n\t/**\n\t * Asynchronously runs database migrations using the Drizzle library.\n\t *\n\t * @returns {Promise<void>} A Promise that resolves once the migrations are completed successfully.\n\t */\n\tasync runMigrations(): Promise<void> {\n\t\ttry {\n\t\t\tconst db = drizzle(this.pool);\n\t\t\tif (await this.hasMigrations()) {\n\t\t\t\tlogger.info(\"Migrations already exist, skipping...\");\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tawait migrate(db, {\n\t\t\t\tmigrationsFolder: path.resolve(__dirname, \"../drizzle/migrations\"),\n\t\t\t});\n\t\t\tlogger.info(\"Migrations completed successfully!\");\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Failed to run database migrations (pg):\", error);\n\t\t\t// throw error;\n\t\t}\n\t}\n}\n"],"mappings":";;;;AAAA,OAAO,UAAU;AACjB,SAAS,qBAAqB;AAC9B,SAAS,cAAkC;AAC3C,SAAS,qBAAqB;AAC9B,SAAS,cAAc;AACvB,SAAS,cAAc;AACvB,SAAS,eAAe;AACxB,SAAS,eAAe;AAGxB,IAAM,aAAa,cAAc,YAAY,GAAG;AAChD,IAAM,YAAY,KAAK,QAAQ,UAAU;AAMlC,IAAM,sBAAN,MAAoE;AAAA,EAjB3E,OAiB2E;AAAA;AAAA;AAAA,EAClE;AAAA,EACA,eAAe;AAAA,EACN,kBAAkB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOnC,YAAY,SAAwB;AACnC,SAAK,SAAS,IAAI,OAAO;AAAA,MACxB,GAAG;AAAA,MACH,YAAY;AAAA,QACX;AAAA,QACA;AAAA,MACD;AAAA,IACD,CAAC;AACD,SAAK,sBAAsB;AAAA,EAC5B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQO,gBAAwB;AAC9B,QAAI,KAAK,cAAc;AACtB,YAAM,IAAI,MAAM,iCAAiC;AAAA,IAClD;AACA,WAAO,KAAK;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAc,mBAAmB;AAChC,QAAI,KAAK,cAAc;AACtB;AAAA,IACD;AAEA,SAAK,eAAe;AACpB,WAAO,KAAK,gDAAgD;AAE5D,UAAM,UAAU,WAAW,MAAM;AAChC,aAAO;AAAA,QACN;AAAA,MACD;AACA,WAAK,OAAO,MAAM,EAAE,QAAQ,MAAM;AACjC,gBAAQ,KAAK,CAAC;AAAA,MACf,CAAC;AAAA,IACF,GAAG,KAAK,eAAe;AAEvB,QAAI;AACH,YAAM,KAAK,OAAO,MAAM;AACxB,mBAAa,OAAO;AACpB,aAAO,KAAK,+CAA+C;AAC3D,cAAQ,KAAK,CAAC;AAAA,IACf,SAAS,OAAO;AACf,aAAO,MAAM,mCAAmC,KAAK;AACrD,cAAQ,KAAK,CAAC;AAAA,IACf;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA,EAMQ,wBAAwB;AAC/B,YAAQ,GAAG,UAAU,YAAY;AAChC,YAAM,KAAK,iBAAiB;AAAA,IAC7B,CAAC;AAED,YAAQ,GAAG,WAAW,YAAY;AACjC,YAAM,KAAK,iBAAiB;AAAA,IAC7B,CAAC;AAED,YAAQ,GAAG,cAAc,YAAY;AACpC,YAAM,KAAK,iBAAiB;AAAA,IAC7B,CAAC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAc,gBAAkC;AAC/C,QAAI;AACH,YAAM,SAAS,MAAM,KAAK,OAAO;AAAA,QAChC;AAAA,MACD;AACA,aAAQ,OAAO,KAAK,CAAC,EAAU;AAAA,IAChC,SAAS,OAAO;AACf,aAAO,MAAM,+BAA+B,KAAK;AACjD,aAAO;AAAA,IACR;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAa,aAA4B;AACxC,QAAI;AACH,YAAM,KAAK,OAAO;AAClB,aAAO,KAAK,wCAAwC;AAAA,IACrD,SAAS,OAAO;AACf,aAAO,MAAM,uCAAuC,KAAK;AACzD,YAAM;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAa,QAAuB;AACnC,QAAI,CAAC,KAAK,cAAc;AACvB,YAAM,KAAK,iBAAiB;AAAA,IAC7B;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOO,iBAA0B;AAChC,WAAO,KAAK;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,gBAA+B;AACpC,QAAI;AACH,YAAM,KAAK,QAAQ,KAAK,MAAM;AAC9B,UAAI,MAAM,KAAK,cAAc,GAAG;AAC/B,eAAO,KAAK,uCAAuC;AACnD;AAAA,MACD;AACA,YAAM,QAAQ,IAAI;AAAA,QACjB,kBAAkB,KAAK,QAAQ,WAAW,uBAAuB;AAAA,MAClE,CAAC;AACD,aAAO,KAAK,oCAAoC;AAAA,IACjD,SAAS,OAAO;AACf,aAAO,MAAM,+CAA+C,KAAK;AAAA,IAElE;AAAA,EACD;AACD;;;ACnLA,OAAOA,WAAU;AACjB,SAAS,iBAAAC,sBAAqB;AAC9B,SAAS,UAAAC,eAAc;AACvB,SAAS,WAAAC,gBAAe;AACxB,SAAS,WAAAC,gBAAe;AACxB,OAAO,SAAkC;AAGzC,IAAM,EAAE,KAAK,IAAI;AAEjB,IAAMC,cAAaC,eAAc,YAAY,GAAG;AAChD,IAAMC,aAAYC,MAAK,QAAQH,WAAU;AAOlC,IAAM,4BAAN,MAEP;AAAA,EApBA,OAoBA;AAAA;AAAA;AAAA,EACS;AAAA,EACA,iBAAiB;AAAA,EACR,oBAA4B;AAAA;AAAA;AAAA;AAAA;AAAA,EAM7C,YAAY,kBAA0B;AACrC,UAAM,gBAAgB;AAAA,MACrB,KAAK;AAAA,MACL,mBAAmB;AAAA,MACnB,yBAAyB,KAAK;AAAA,IAC/B;AAEA,SAAK,OAAO,IAAI,KAAK;AAAA,MACpB,GAAG;AAAA,MACH;AAAA,IACD,CAAC;AAED,SAAK,KAAK,GAAG,SAAS,CAAC,QAAQ;AAC9B,MAAAI,QAAO,MAAM,yBAAyB,GAAG;AACzC,WAAK,gBAAgB,GAAG;AAAA,IACzB,CAAC;AAED,SAAK,uBAAuB;AAC5B,SAAK,eAAe;AAAA,EACrB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAc,gBAAgB,OAAc;AAC3C,IAAAA,QAAO,MAAM,gDAAgD;AAAA,MAC5D,OAAO,MAAM;AAAA,IACd,CAAC;AAED,QAAI;AACH,YAAM,KAAK,KAAK,IAAI;AAEpB,WAAK,OAAO,IAAI,KAAK;AAAA,QACpB,GAAG,KAAK,KAAK;AAAA,QACb,yBAAyB,KAAK;AAAA,MAC/B,CAAC;AAED,YAAM,KAAK,eAAe;AAC1B,MAAAA,QAAO,QAAQ,8BAA8B;AAAA,IAC9C,SAAS,gBAAgB;AACxB,MAAAA,QAAO,MAAM,4BAA4B;AAAA,QACxC,OACC,0BAA0B,QACvB,eAAe,UACf,OAAO,cAAc;AAAA,MAC1B,CAAC;AACD,YAAM;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,iBAAmC;AACxC,QAAI,SAAgC;AACpC,QAAI;AACH,eAAS,MAAM,KAAK,KAAK,QAAQ;AACjC,YAAM,SAAS,MAAM,OAAO,MAAM,cAAc;AAChD,MAAAA,QAAO,QAAQ,wCAAwC,OAAO,KAAK,CAAC,CAAC;AACrE,aAAO;AAAA,IACR,SAAS,OAAO;AACf,MAAAA,QAAO,MAAM,oCAAoC,KAAK;AACtD,YAAM,IAAI;AAAA,QACT,kCAAmC,MAAgB,OAAO;AAAA,MAC3D;AAAA,IACD,UAAE;AACD,UAAI,OAAQ,QAAO,QAAQ;AAAA,IAC5B;AAAA,EACD;AAAA;AAAA;AAAA;AAAA,EAKQ,yBAAyB;AAChC,YAAQ,GAAG,UAAU,YAAY;AAChC,YAAM,KAAK,QAAQ;AACnB,cAAQ,KAAK,CAAC;AAAA,IACf,CAAC;AAED,YAAQ,GAAG,WAAW,YAAY;AACjC,YAAM,KAAK,QAAQ;AACnB,cAAQ,KAAK,CAAC;AAAA,IACf,CAAC;AAED,YAAQ,GAAG,cAAc,YAAY;AACpC,YAAM,KAAK,QAAQ;AAAA,IACpB,CAAC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOO,gBAAwB;AAC9B,QAAI,KAAK,gBAAgB;AACxB,YAAM,IAAI,MAAM,qCAAqC;AAAA,IACtD;AAEA,QAAI;AACH,aAAO,KAAK;AAAA,IACb,SAAS,OAAO;AACf,MAAAA,QAAO,MAAM,uCAAuC,KAAK;AACzD,YAAM;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAa,YAAqC;AACjD,QAAI;AACH,aAAO,MAAM,KAAK,KAAK,QAAQ;AAAA,IAChC,SAAS,OAAO;AACf,MAAAA,QAAO,MAAM,wCAAwC,KAAK;AAC1D,YAAM;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAa,aAA4B;AACxC,QAAI;AACH,YAAM,KAAK,eAAe;AAC1B,MAAAA,QAAO,KAAK,wDAAwD;AAAA,IACrE,SAAS,OAAO;AACf,MAAAA,QAAO,MAAM,4CAA4C,KAAK;AAC9D,YAAM;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAa,QAAuB;AACnC,UAAM,KAAK,QAAQ;AAAA,EACpB;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAc,gBAAkC;AAC/C,QAAI;AACH,YAAM,SAAS,MAAM,KAAK,KAAK;AAAA,QAC9B;AAAA,MACD;AACA,aAAO,OAAO,KAAK,CAAC,EAAE;AAAA,IACvB,SAAS,OAAO;AACf,MAAAA,QAAO,MAAM,+BAA+B,KAAK;AACjD,aAAO;AAAA,IACR;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,UAAyB;AAC9B,QAAI;AACH,YAAM,KAAK,KAAK,IAAI;AACpB,MAAAA,QAAO,KAAK,sBAAsB;AAAA,IACnC,SAAS,OAAO;AACf,MAAAA,QAAO,MAAM,gCAAgC,KAAK;AAAA,IACnD;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,gBAA+B;AACpC,QAAI;AACH,YAAM,KAAKC,SAAQ,KAAK,IAAI;AAC5B,UAAI,MAAM,KAAK,cAAc,GAAG;AAC/B,QAAAD,QAAO,KAAK,uCAAuC;AACnD;AAAA,MACD;AACA,YAAME,SAAQ,IAAI;AAAA,QACjB,kBAAkBH,MAAK,QAAQD,YAAW,uBAAuB;AAAA,MAClE,CAAC;AACD,MAAAE,QAAO,KAAK,oCAAoC;AAAA,IACjD,SAAS,OAAO;AACf,MAAAA,QAAO,MAAM,2CAA2C,KAAK;AAAA,IAE9D;AAAA,EACD;AACD;","names":["path","fileURLToPath","logger","drizzle","migrate","__filename","fileURLToPath","__dirname","path","logger","drizzle","migrate"]}

package/dist/index.js

@@ -2,7 +2,7 @@ import {
   PGliteClientManager,
   PostgresConnectionManager,
   __name
-} from "./chunk-LLEN3JTK.js";
+} from "./chunk-2FOOMI3E.js";
 
 // src/index.ts
 import * as os from "node:os";
@@ -29,7 +29,7 @@ import {
   inArray,
   lte,
   or,
-  sql as sql13
+  sql as sql12
 } from "drizzle-orm";
 import { v4 } from "uuid";
 
@@ -325,27 +325,18 @@ var componentTable = pgTable8("components", {
   createdAt: numberTimestamp("createdAt").default(sql8`now()`).notNull()
 });
 
-// src/schema/goal.ts
+// src/schema/log.ts
 import { sql as sql9 } from "drizzle-orm";
 import { foreignKey as foreignKey3, jsonb as jsonb8, pgTable as pgTable9, text as text8, uuid as uuid9 } from "drizzle-orm/pg-core";
-var goalTable = pgTable9(
-  "goals",
+var logTable = pgTable9(
+  "logs",
   {
-    id: uuid9("id").notNull().primaryKey().default(sql9`gen_random_uuid()`),
+    id: uuid9("id").defaultRandom().notNull(),
     createdAt: numberTimestamp("createdAt").default(sql9`now()`).notNull(),
-    entityId: uuid9("entityId").references(() => entityTable.id, {
-      onDelete: "cascade"
-    }),
-    agentId: uuid9("agentId").references(() => agentTable.id, {
-      onDelete: "cascade"
-    }),
-    name: text8("name"),
-    status: text8("status"),
-    description: text8("description"),
-    roomId: uuid9("roomId").references(() => roomTable.id, {
-      onDelete: "cascade"
-    }),
-    objectives: jsonb8("objectives").default("[]").notNull()
+    entityId: uuid9("entityId").notNull().references(() => entityTable.id),
+    body: jsonb8("body").notNull(),
+    type: text8("type").notNull(),
+    roomId: uuid9("roomId").notNull().references(() => roomTable.id)
   },
   (table) => [
     foreignKey3({
@@ -361,68 +352,41 @@ var goalTable = pgTable9(
   ]
 );
 
-// src/schema/log.ts
-import { sql as sql10 } from "drizzle-orm";
-import { foreignKey as foreignKey4, jsonb as jsonb9, pgTable as pgTable10, text as text9, uuid as uuid10 } from "drizzle-orm/pg-core";
-var logTable = pgTable10(
-  "logs",
-  {
-    id: uuid10("id").defaultRandom().notNull(),
-    createdAt: numberTimestamp("createdAt").default(sql10`now()`).notNull(),
-    entityId: uuid10("entityId").notNull().references(() => entityTable.id),
-    body: jsonb9("body").notNull(),
-    type: text9("type").notNull(),
-    roomId: uuid10("roomId").notNull().references(() => roomTable.id)
-  },
-  (table) => [
-    foreignKey4({
-      name: "fk_room",
-      columns: [table.roomId],
-      foreignColumns: [roomTable.id]
-    }).onDelete("cascade"),
-    foreignKey4({
-      name: "fk_user",
-      columns: [table.entityId],
-      foreignColumns: [entityTable.id]
-    }).onDelete("cascade")
-  ]
-);
-
 // src/schema/participant.ts
-import { sql as sql11 } from "drizzle-orm";
+import { sql as sql10 } from "drizzle-orm";
 import {
-  foreignKey as foreignKey5,
+  foreignKey as foreignKey4,
   index as index3,
-  pgTable as pgTable11,
-  text as text10,
-  uuid as uuid11
+  pgTable as pgTable10,
+  text as text9,
+  uuid as uuid10
 } from "drizzle-orm/pg-core";
-var participantTable = pgTable11(
+var participantTable = pgTable10(
   "participants",
   {
-    id: uuid11("id").notNull().primaryKey().default(sql11`gen_random_uuid()`),
-    createdAt: numberTimestamp("createdAt").default(sql11`now()`).notNull(),
-    entityId: uuid11("entityId").references(() => entityTable.id, {
+    id: uuid10("id").notNull().primaryKey().default(sql10`gen_random_uuid()`),
+    createdAt: numberTimestamp("createdAt").default(sql10`now()`).notNull(),
+    entityId: uuid10("entityId").references(() => entityTable.id, {
       onDelete: "cascade"
     }),
-    roomId: uuid11("roomId").references(() => roomTable.id, {
+    roomId: uuid10("roomId").references(() => roomTable.id, {
       onDelete: "cascade"
     }),
-    agentId: uuid11("agentId").references(() => agentTable.id, {
+    agentId: uuid10("agentId").references(() => agentTable.id, {
       onDelete: "cascade"
     }),
-    roomState: text10("roomState")
+    roomState: text9("roomState")
   },
   (table) => [
     // unique("participants_user_room_agent_unique").on(table.entityId, table.roomId, table.agentId),
     index3("idx_participants_user").on(table.entityId),
     index3("idx_participants_room").on(table.roomId),
-    foreignKey5({
+    foreignKey4({
       name: "fk_room",
       columns: [table.roomId],
       foreignColumns: [roomTable.id]
     }).onDelete("cascade"),
-    foreignKey5({
+    foreignKey4({
       name: "fk_user",
       columns: [table.entityId],
       foreignColumns: [entityTable.id]
@@ -431,26 +395,26 @@ var participantTable = pgTable11(
 );
 
 // src/schema/relationship.ts
-import { sql as sql12 } from "drizzle-orm";
+import { sql as sql11 } from "drizzle-orm";
 import {
-  foreignKey as foreignKey6,
+  foreignKey as foreignKey5,
   index as index4,
-  jsonb as jsonb10,
-  pgTable as pgTable12,
-  text as text11,
+  jsonb as jsonb9,
+  pgTable as pgTable11,
+  text as text10,
   unique as unique6,
-  uuid as uuid12
+  uuid as uuid11
 } from "drizzle-orm/pg-core";
-var relationshipTable = pgTable12(
+var relationshipTable = pgTable11(
   "relationships",
   {
-    id: uuid12("id").notNull().primaryKey().default(sql12`gen_random_uuid()`),
-    createdAt: numberTimestamp("createdAt").default(sql12`now()`).notNull(),
-    sourceEntityId: uuid12("sourceEntityId").notNull().references(() => entityTable.id, { onDelete: "cascade" }),
-    targetEntityId: uuid12("targetEntityId").notNull().references(() => entityTable.id, { onDelete: "cascade" }),
-    agentId: uuid12("agentId").notNull().references(() => agentTable.id, { onDelete: "cascade" }),
-    tags: text11("tags").array(),
-    metadata: jsonb10("metadata")
+    id: uuid11("id").notNull().primaryKey().default(sql11`gen_random_uuid()`),
+    createdAt: numberTimestamp("createdAt").default(sql11`now()`).notNull(),
+    sourceEntityId: uuid11("sourceEntityId").notNull().references(() => entityTable.id, { onDelete: "cascade" }),
+    targetEntityId: uuid11("targetEntityId").notNull().references(() => entityTable.id, { onDelete: "cascade" }),
+    agentId: uuid11("agentId").notNull().references(() => agentTable.id, { onDelete: "cascade" }),
+    tags: text10("tags").array(),
+    metadata: jsonb9("metadata")
   },
   (table) => [
     index4("idx_relationships_users").on(
@@ -462,12 +426,12 @@ var relationshipTable = pgTable12(
       table.targetEntityId,
       table.agentId
     ),
-    foreignKey6({
+    foreignKey5({
       name: "fk_user_a",
       columns: [table.sourceEntityId],
       foreignColumns: [entityTable.id]
     }).onDelete("cascade"),
-    foreignKey6({
+    foreignKey5({
       name: "fk_user_b",
       columns: [table.targetEntityId],
       foreignColumns: [entityTable.id]
@@ -476,16 +440,16 @@ var relationshipTable = pgTable12(
 );
 
 // src/schema/tasks.ts
-import { jsonb as jsonb11, pgTable as pgTable13, text as text12, timestamp, uuid as uuid13 } from "drizzle-orm/pg-core";
-var taskTable = pgTable13("tasks", {
-  id: uuid13("id").primaryKey().defaultRandom(),
-  name: text12("name").notNull(),
-  description: text12("description").notNull(),
-  roomId: uuid13("room_id"),
-  worldId: uuid13("world_id"),
-  agentId: uuid13("agent_id").notNull(),
-  tags: text12("tags").array(),
-  metadata: jsonb11("metadata"),
+import { jsonb as jsonb10, pgTable as pgTable12, text as text11, timestamp, uuid as uuid12 } from "drizzle-orm/pg-core";
+var taskTable = pgTable12("tasks", {
+  id: uuid12("id").primaryKey().defaultRandom(),
+  name: text11("name").notNull(),
+  description: text11("description").notNull(),
+  roomId: uuid12("room_id"),
+  worldId: uuid12("world_id"),
+  agentId: uuid12("agent_id").notNull(),
+  tags: text11("tags").array(),
+  metadata: jsonb10("metadata"),
   createdAt: timestamp("created_at").defaultNow(),
   updatedAt: timestamp("updated_at").defaultNow()
 });
@@ -583,9 +547,6 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
       const usedDimension = Object.entries(DIMENSION_MAP).find(
         ([_, colName]) => existingMemory[0].embedding[colName] !== null
       );
-      if (usedDimension && usedDimension[1] !== DIMENSION_MAP[dimension]) {
-        throw new Error("Cannot change embedding dimension for agent");
-      }
     }
     this.embeddingDimension = DIMENSION_MAP[dimension];
   }
@@ -1014,7 +975,7 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
   async getCachedEmbeddings(opts) {
     return this.withDatabase(async () => {
       try {
-        const results = await this.db.execute(sql13`
+        const results = await this.db.execute(sql12`
                     WITH content_text AS (
                         SELECT
                             m.id,
@@ -1071,7 +1032,7 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
       try {
         await this.db.transaction(async (tx) => {
           await tx.insert(logTable).values({
-            body: sql13`${params.body}::jsonb`,
+            body: sql12`${params.body}::jsonb`,
             entityId: params.entityId,
             roomId: params.roomId,
             type: params.type
@@ -1097,30 +1058,12 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
       tableName: params.tableName
     });
   }
-  async updateGoalStatus(params) {
-    return this.withDatabase(async () => {
-      try {
-        await this.db.transaction(async (tx) => {
-          await tx.update(goalTable).set({
-            status: params.status
-          }).where(eq(goalTable.id, params.goalId));
-        });
-      } catch (error) {
-        logger.error("Failed to update goal status:", {
-          goalId: params.goalId,
-          status: params.status,
-          error: error instanceof Error ? error.message : String(error)
-        });
-        throw error;
-      }
-    });
-  }
   async searchMemoriesByEmbedding(embedding, params) {
     return this.withDatabase(async () => {
       const cleanVector = embedding.map(
         (n) => Number.isFinite(n) ? Number(n.toFixed(6)) : 0
       );
-      const similarity = sql13`1 - (${cosineDistance(
+      const similarity = sql12`1 - (${cosineDistance(
         embeddingTable[this.embeddingDimension],
         cleanVector
       )})`;
@@ -1180,8 +1123,8 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
         {
           id: memoryId,
           type: tableName,
-          content: sql13`${contentToInsert}::jsonb`,
-          metadata: sql13`${memory.metadata || {}}::jsonb`,
+          content: sql12`${contentToInsert}::jsonb`,
+          metadata: sql12`${memory.metadata || {}}::jsonb`,
           entityId: memory.entityId,
           roomId: memory.roomId,
           agentId: memory.agentId,
@@ -1258,103 +1201,10 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
       if (unique7) {
         conditions.push(eq(memoryTable.unique, true));
       }
-      const result = await this.db.select({ count: sql13`count(*)` }).from(memoryTable).where(and(...conditions));
+      const result = await this.db.select({ count: sql12`count(*)` }).from(memoryTable).where(and(...conditions));
       return Number(result[0]?.count ?? 0);
     });
   }
-  async getGoals(params) {
-    return this.withDatabase(async () => {
-      const conditions = [eq(goalTable.roomId, params.roomId)];
-      if (params.entityId) {
-        conditions.push(eq(goalTable.entityId, params.entityId));
-      }
-      if (params.onlyInProgress) {
-        conditions.push(eq(goalTable.status, "IN_PROGRESS"));
-      }
-      const query = this.db.select().from(goalTable).where(and(...conditions)).orderBy(desc(goalTable.createdAt));
-      const result = await (params.count ? query.limit(params.count) : query);
-      return result.map((row) => ({
-        id: row.id,
-        roomId: row.roomId,
-        entityId: row.entityId,
-        name: row.name ?? "",
-        status: row.status ?? "NOT_STARTED",
-        description: row.description ?? "",
-        objectives: row.objectives,
-        createdAt: row.createdAt
-      }));
-    });
-  }
-  async updateGoal(goal) {
-    return this.withDatabase(async () => {
-      try {
-        await this.db.transaction(async (tx) => {
-          await tx.update(goalTable).set({
-            name: goal.name,
-            status: goal.status,
-            objectives: goal.objectives
-          }).where(eq(goalTable.id, goal.id));
-        });
-      } catch (error) {
-        logger.error("Failed to update goal:", {
-          error: error instanceof Error ? error.message : String(error),
-          goalId: goal.id,
-          status: goal.status
-        });
-        throw error;
-      }
-    });
-  }
-  async createGoal(goal) {
-    return this.withDatabase(async () => {
-      try {
-        await this.db.transaction(async (tx) => {
-          await tx.insert(goalTable).values({
-            id: goal.id ?? v4(),
-            roomId: goal.roomId,
-            entityId: goal.entityId,
-            name: goal.name,
-            status: goal.status,
-            objectives: sql13`${goal.objectives}::jsonb`
-          });
-        });
-      } catch (error) {
-        logger.error("Failed to update goal:", {
-          goalId: goal.id,
-          error: error instanceof Error ? error.message : String(error),
-          status: goal.status
-        });
-        throw error;
-      }
-    });
-  }
-  async removeGoal(goalId) {
-    if (!goalId) throw new Error("Goal ID is required");
-    return this.withDatabase(async () => {
-      try {
-        await this.db.transaction(async (tx) => {
-          await tx.delete(goalTable).where(eq(goalTable.id, goalId));
-        });
-        logger.debug("Goal removal attempt:", {
-          goalId,
-          removed: true
-        });
-      } catch (error) {
-        logger.error("Failed to remove goal:", {
-          error: error instanceof Error ? error.message : String(error),
-          goalId
-        });
-        throw error;
-      }
-    });
-  }
-  async removeAllGoals(roomId) {
-    return this.withDatabase(async () => {
-      await this.db.transaction(async (tx) => {
-        await tx.delete(goalTable).where(eq(goalTable.roomId, roomId));
-      });
-    });
-  }
   async getRoom(roomId) {
     return this.withDatabase(async () => {
       const result = await this.db.select({
@@ -1633,7 +1483,7 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
         if (params.tags && params.tags.length > 0) {
           const tagParams = params.tags.map((tag) => `'${tag.replace(/'/g, "''")}'`).join(", ");
           query = query.where(
-            sql13`${relationshipTable.tags} @> ARRAY[${sql13.raw(
+            sql12`${relationshipTable.tags} @> ARRAY[${sql12.raw(
               tagParams
             )}]::text[]`
           );
@@ -1796,7 +1646,7 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
         if (params.tags && params.tags.length > 0) {
           const tagParams = params.tags.map((tag) => `'${tag.replace(/'/g, "''")}'`).join(", ");
           query = query.where(
-            sql13`${taskTable.tags} @> ARRAY[${sql13.raw(tagParams)}]::text[]`
+            sql12`${taskTable.tags} @> ARRAY[${sql12.raw(tagParams)}]::text[]`
           );
         }
         const result = await query;