@elizaos/plugin-sql 1.0.11 → 1.0.13

package/dist/index.js

@@ -35,7 +35,7 @@ import { v4 } from "uuid";
 
 // src/schema/embedding.ts
 import { sql as sql5 } from "drizzle-orm";
-import { check as check2, foreignKey as foreignKey2, index as index2, pgTable as pgTable5, timestamp as timestamp5, uuid as uuid5, vector as vector2 } from "drizzle-orm/pg-core";
+import { check as check2, foreignKey as foreignKey2, index as index2, pgTable as pgTable5, timestamp as timestamp5, uuid as uuid5, vector } from "drizzle-orm/pg-core";
 import { VECTOR_DIMS } from "@elizaos/core";
 
 // src/schema/memory.ts
@@ -221,12 +221,12 @@ var embeddingTable = pgTable5(
     id: uuid5("id").primaryKey().defaultRandom().notNull(),
     memoryId: uuid5("memory_id").references(() => memoryTable.id, { onDelete: "cascade" }),
     createdAt: timestamp5("created_at").default(sql5`now()`).notNull(),
-    dim384: vector2("dim_384", { dimensions: VECTOR_DIMS.SMALL }),
-    dim512: vector2("dim_512", { dimensions: VECTOR_DIMS.MEDIUM }),
-    dim768: vector2("dim_768", { dimensions: VECTOR_DIMS.LARGE }),
-    dim1024: vector2("dim_1024", { dimensions: VECTOR_DIMS.XL }),
-    dim1536: vector2("dim_1536", { dimensions: VECTOR_DIMS.XXL }),
-    dim3072: vector2("dim_3072", { dimensions: VECTOR_DIMS.XXXL })
+    dim384: vector("dim_384", { dimensions: VECTOR_DIMS.SMALL }),
+    dim512: vector("dim_512", { dimensions: VECTOR_DIMS.MEDIUM }),
+    dim768: vector("dim_768", { dimensions: VECTOR_DIMS.LARGE }),
+    dim1024: vector("dim_1024", { dimensions: VECTOR_DIMS.XL }),
+    dim1536: vector("dim_1536", { dimensions: VECTOR_DIMS.XXL }),
+    dim3072: vector("dim_3072", { dimensions: VECTOR_DIMS.XXXL })
   },
   (table) => [
     check2("embedding_source_check", sql5`"memory_id" IS NOT NULL`),
@@ -382,7 +382,7 @@ import {
   pgTable as pgTable11,
   text as text10,
   timestamp as timestamp11,
-  unique as unique5,
+  unique as unique3,
   uuid as uuid11
 } from "drizzle-orm/pg-core";
 var relationshipTable = pgTable11(
@@ -398,7 +398,7 @@ var relationshipTable = pgTable11(
   },
   (table) => [
     index4("idx_relationships_users").on(table.sourceEntityId, table.targetEntityId),
-    unique5("unique_relationship").on(table.sourceEntityId, table.targetEntityId, table.agentId),
+    unique3("unique_relationship").on(table.sourceEntityId, table.targetEntityId, table.agentId),
     foreignKey5({
       name: "fk_user_a",
       columns: [table.sourceEntityId],
@@ -583,7 +583,7 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
     return this.withDatabase(async () => {
       const existingMemory = await this.db.select().from(memoryTable).innerJoin(embeddingTable, eq(embeddingTable.memoryId, memoryTable.id)).where(eq(memoryTable.agentId, this.agentId)).limit(1);
       if (existingMemory.length > 0) {
-        const usedDimension = Object.entries(DIMENSION_MAP).find(
+        Object.entries(DIMENSION_MAP).find(
           ([_, colName]) => existingMemory[0].embeddings[colName] !== null
         );
       }
@@ -744,7 +744,6 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
         return source;
       }
       const output = typeof target === "object" && target !== null && !Array.isArray(target) ? { ...target } : {};
-      let isEmpty = true;
       for (const key of Object.keys(source)) {
         const sourceValue = source[key];
         if (sourceValue === null) {
@@ -755,11 +754,9 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
             delete output[key];
           } else {
             output[key] = nestedMergeResult;
-            isEmpty = false;
           }
         } else {
           output[key] = sourceValue;
-          isEmpty = false;
         }
       }
       if (Object.keys(output).length === 0) {
@@ -1170,7 +1167,7 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
    * @returns {Promise<Memory[]>} A Promise that resolves to an array of memories.
    */
   async getMemories(params) {
-    const { entityId, agentId, roomId, worldId, tableName, count: count2, unique: unique6, start, end } = params;
+    const { entityId, agentId, roomId, worldId, tableName, unique: unique4, start, end } = params;
     if (!tableName) throw new Error("tableName is required");
     return this.withDatabase(async () => {
       const conditions = [eq(memoryTable.type, tableName)];
@@ -1189,7 +1186,7 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
       if (end) {
         conditions.push(lte(memoryTable.createdAt, new Date(end)));
       }
-      if (unique6) {
+      if (unique4) {
         conditions.push(eq(memoryTable.unique, true));
       }
       if (agentId) {
@@ -1681,8 +1678,7 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
             } else {
               const embeddingValues = {
                 id: v4(),
-                memoryId: memory.id,
-                createdAt: (/* @__PURE__ */ new Date()).getTime()
+                memoryId: memory.id
               };
               embeddingValues[this.embeddingDimension] = cleanVector;
               await tx.insert(embeddingTable).values([embeddingValues]);
@@ -1814,11 +1810,11 @@ var BaseDrizzleAdapter = class extends DatabaseAdapter {
    * @param {string} [tableName] - The name of the table to count memories in.
    * @returns {Promise<number>} A Promise that resolves to the number of memories.
    */
-  async countMemories(roomId, unique6 = true, tableName = "") {
+  async countMemories(roomId, unique4 = true, tableName = "") {
     if (!tableName) throw new Error("tableName is required");
     return this.withDatabase(async () => {
       const conditions = [eq(memoryTable.roomId, roomId), eq(memoryTable.type, tableName)];
-      if (unique6) {
+      if (unique4) {
         conditions.push(eq(memoryTable.unique, true));
       }
       const result = await this.db.select({ count: sql16`count(*)` }).from(memoryTable).where(and(...conditions));
@@ -2950,14 +2946,13 @@ var PgliteDatabaseAdapter = class extends BaseDrizzleAdapter {
 // src/pglite/manager.ts
 import { PGlite } from "@electric-sql/pglite";
 import { fuzzystrmatch } from "@electric-sql/pglite/contrib/fuzzystrmatch";
-import { vector as vector3 } from "@electric-sql/pglite/vector";
+import { vector as vector2 } from "@electric-sql/pglite/vector";
 var PGliteClientManager = class {
   static {
     __name(this, "PGliteClientManager");
   }
   client;
   shuttingDown = false;
-  shutdownTimeout = 500;
   /**
    * Constructor for creating a new instance of PGlite with the provided options.
    * Initializes the PGlite client with additional extensions.
@@ -2967,7 +2962,7 @@ var PGliteClientManager = class {
     this.client = new PGlite({
       ...options,
       extensions: {
-        vector: vector3,
+        vector: vector2,
         fuzzystrmatch
       }
     });
@@ -2997,7 +2992,7 @@ var PgDatabaseAdapter = class extends BaseDrizzleAdapter {
   }
   embeddingDimension = DIMENSION_MAP[384];
   manager;
-  constructor(agentId, manager, schema) {
+  constructor(agentId, manager, _schema) {
     super(agentId);
     this.manager = manager;
     this.db = manager.getDatabase();
@@ -3209,6 +3204,31 @@ import { logger as logger6 } from "@elizaos/core";
 // src/custom-migrator.ts
 import { sql as sql17 } from "drizzle-orm";
 import { logger as logger5 } from "@elizaos/core";
+function extractErrorMessage(error) {
+  if (error instanceof Error && "cause" in error && error.cause) {
+    return error.cause.message;
+  } else if (error instanceof Error) {
+    return error.message;
+  }
+  return "Unknown error";
+}
+__name(extractErrorMessage, "extractErrorMessage");
+function extractErrorDetails(error) {
+  if (error instanceof Error && "cause" in error && error.cause) {
+    const cause = error.cause;
+    return {
+      message: cause.message,
+      stack: cause.stack || error.stack
+    };
+  } else if (error instanceof Error) {
+    return {
+      message: error.message,
+      stack: error.stack
+    };
+  }
+  return { message: "Unknown error" };
+}
+__name(extractErrorDetails, "extractErrorDetails");
 var KNOWN_COMPOSITE_PRIMARY_KEYS = {
   cache: { columns: ["key", "agent_id"] }
   // Add other tables with composite primary keys here if needed
@@ -3650,7 +3670,7 @@ var DrizzleSchemaIntrospector = class {
             }
           }
         } else if (extraConfig && typeof extraConfig === "object") {
-          for (const [key, value] of Object.entries(extraConfig)) {
+          for (const [_key, value] of Object.entries(extraConfig)) {
             if (value && typeof value === "object" && value._) {
               const config = value._;
               if (config.name && config.columns) {
@@ -3945,11 +3965,12 @@ var PluginNamespaceManager = class {
           await this.db.execute(sql17.raw(constraintSQL));
           logger5.debug(`[CUSTOM MIGRATOR] Successfully added foreign key constraint: ${fk.name}`);
         } catch (error) {
-          if (error.message?.includes("already exists")) {
+          const errorMessage = extractErrorMessage(error);
+          if (errorMessage.includes("already exists")) {
             logger5.debug(`[CUSTOM MIGRATOR] Foreign key constraint already exists: ${fk.name}`);
           } else {
             logger5.warn(
-              `[CUSTOM MIGRATOR] Could not add foreign key constraint (may already exist): ${error.message}`
+              `[CUSTOM MIGRATOR] Could not add foreign key constraint (may already exist): ${errorMessage}`
             );
           }
         }
@@ -3975,13 +3996,14 @@ var PluginNamespaceManager = class {
             `[CUSTOM MIGRATOR] Successfully added check constraint: ${checkConstraint.name}`
           );
         } catch (error) {
-          if (error.message?.includes("already exists")) {
+          const errorMessage = extractErrorMessage(error);
+          if (errorMessage.includes("already exists")) {
             logger5.debug(
               `[CUSTOM MIGRATOR] Check constraint already exists: ${checkConstraint.name}`
             );
           } else {
             logger5.warn(
-              `[CUSTOM MIGRATOR] Could not add check constraint ${checkConstraint.name} (may already exist): ${error.message}`
+              `[CUSTOM MIGRATOR] Could not add check constraint ${checkConstraint.name} (may already exist): ${errorMessage}`
             );
           }
         }
@@ -4001,10 +4023,13 @@ var ExtensionManager = class {
       try {
         await this.db.execute(sql17.raw(`CREATE EXTENSION IF NOT EXISTS "${extension}"`));
       } catch (error) {
-        logger5.warn(`Could not install extension ${extension}:`, {
-          message: error.message,
-          stack: error.stack
-        });
+        const errorDetails = extractErrorDetails(error);
+        logger5.warn(`Could not install extension ${extension}: ${errorDetails.message}`);
+        if (errorDetails.stack) {
+          logger5.debug(
+            `[CUSTOM MIGRATOR] Extension installation stack trace: ${errorDetails.stack}`
+          );
+        }
       }
     }
   }
@@ -4043,6 +4068,17 @@ function topologicalSort(tables) {
 __name(topologicalSort, "topologicalSort");
 async function runPluginMigrations(db, pluginName, schema) {
   logger5.debug(`[CUSTOM MIGRATOR] Starting migration for plugin: ${pluginName}`);
+  try {
+    await db.execute(sql17.raw("SELECT 1"));
+    logger5.debug("[CUSTOM MIGRATOR] Database connection verified");
+  } catch (error) {
+    const errorDetails = extractErrorDetails(error);
+    logger5.error(`[CUSTOM MIGRATOR] Database connection failed: ${errorDetails.message}`);
+    if (errorDetails.stack) {
+      logger5.error(`[CUSTOM MIGRATOR] Stack trace: ${errorDetails.stack}`);
+    }
+    throw new Error(`Database connection failed: ${errorDetails.message}`);
+  }
   const namespaceManager = new PluginNamespaceManager(db);
   const introspector = new DrizzleSchemaIntrospector();
   const extensionManager = new ExtensionManager(db);
@@ -4060,32 +4096,54 @@ async function runPluginMigrations(db, pluginName, schema) {
     tableDefinitions.set(tableDef.name, tableDef);
   }
   const sortedTableNames = topologicalSort(tableDefinitions);
-  logger5.debug(`[CUSTOM MIGRATOR] Phase 1: Creating tables...`);
-  for (const tableName of sortedTableNames) {
-    const tableDef = tableDefinitions.get(tableName);
-    if (!tableDef) continue;
-    const tableExists = existingTables.includes(tableDef.name);
-    logger5.debug(`[CUSTOM MIGRATOR] Table ${tableDef.name} exists: ${tableExists}`);
-    if (!tableExists) {
-      logger5.debug(`[CUSTOM MIGRATOR] Creating table: ${tableDef.name}`);
-      await namespaceManager.createTable(tableDef, schemaName);
-    } else {
-      logger5.debug(`[CUSTOM MIGRATOR] Table ${tableDef.name} already exists, skipping creation`);
+  try {
+    logger5.debug(`[CUSTOM MIGRATOR] Phase 1: Creating tables...`);
+    for (const tableName of sortedTableNames) {
+      const tableDef = tableDefinitions.get(tableName);
+      if (!tableDef) continue;
+      const tableExists = existingTables.includes(tableDef.name);
+      logger5.debug(`[CUSTOM MIGRATOR] Table ${tableDef.name} exists: ${tableExists}`);
+      if (!tableExists) {
+        logger5.debug(`[CUSTOM MIGRATOR] Creating table: ${tableDef.name}`);
+        try {
+          await namespaceManager.createTable(tableDef, schemaName);
+        } catch (error) {
+          const errorDetails = extractErrorDetails(error);
+          logger5.error(
+            `[CUSTOM MIGRATOR] Failed to create table ${tableDef.name}: ${errorDetails.message}`
+          );
+          if (errorDetails.stack) {
+            logger5.error(`[CUSTOM MIGRATOR] Table creation stack trace: ${errorDetails.stack}`);
+          }
+          throw new Error(`Failed to create table ${tableDef.name}: ${errorDetails.message}`);
+        }
+      } else {
+        logger5.debug(`[CUSTOM MIGRATOR] Table ${tableDef.name} already exists, skipping creation`);
+      }
     }
-  }
-  logger5.debug(`[CUSTOM MIGRATOR] Phase 2: Adding constraints...`);
-  for (const tableName of sortedTableNames) {
-    const tableDef = tableDefinitions.get(tableName);
-    if (!tableDef) continue;
-    if (tableDef.foreignKeys.length > 0 || tableDef.checkConstraints.length > 0) {
-      logger5.debug(`[CUSTOM MIGRATOR] Adding constraints for table: ${tableDef.name}`, {
-        foreignKeys: tableDef.foreignKeys.length,
-        checkConstraints: tableDef.checkConstraints.length
-      });
-      await namespaceManager.addConstraints(tableDef, schemaName);
+    logger5.debug(`[CUSTOM MIGRATOR] Phase 2: Adding constraints...`);
+    for (const tableName of sortedTableNames) {
+      const tableDef = tableDefinitions.get(tableName);
+      if (!tableDef) continue;
+      if (tableDef.foreignKeys.length > 0 || tableDef.checkConstraints.length > 0) {
+        logger5.debug(`[CUSTOM MIGRATOR] Adding constraints for table: ${tableDef.name}`, {
+          foreignKeys: tableDef.foreignKeys.length,
+          checkConstraints: tableDef.checkConstraints.length
+        });
+        await namespaceManager.addConstraints(tableDef, schemaName);
+      }
     }
+    logger5.debug(`[CUSTOM MIGRATOR] Completed migration for plugin: ${pluginName}`);
+  } catch (error) {
+    const errorDetails = extractErrorDetails(error);
+    logger5.error(
+      `[CUSTOM MIGRATOR] Migration failed for plugin ${pluginName}: ${errorDetails.message}`
+    );
+    if (errorDetails.stack) {
+      logger5.error(`[CUSTOM MIGRATOR] Migration stack trace: ${errorDetails.stack}`);
+    }
+    throw new Error(`Migration failed for plugin ${pluginName}: ${errorDetails.message}`);
   }
-  logger5.debug(`[CUSTOM MIGRATOR] Completed migration for plugin: ${pluginName}`);
 }
 __name(runPluginMigrations, "runPluginMigrations");
 
@@ -4134,7 +4192,6 @@ if (!globalSymbols[GLOBAL_SINGLETONS]) {
 }
 var globalSingletons = globalSymbols[GLOBAL_SINGLETONS];
 function createDatabaseAdapter(config, agentId) {
-  const dataDir = resolvePgliteDir(config.dataDir);
   if (config.postgresUrl) {
     if (!globalSingletons.postgresConnectionManager) {
       globalSingletons.postgresConnectionManager = new PostgresConnectionManager(
@@ -4143,6 +4200,7 @@ function createDatabaseAdapter(config, agentId) {
     }
     return new PgDatabaseAdapter(agentId, globalSingletons.postgresConnectionManager);
   }
+  const dataDir = resolvePgliteDir(config.dataDir);
   if (!globalSingletons.pgLiteClientManager) {
     globalSingletons.pgLiteClientManager = new PGliteClientManager({ dataDir });
   }