@elizaos/plugin-sql 1.0.0-beta.4 → 1.0.0-beta.41

package/dist/migrate.js

@@ -7,6 +7,8 @@ import {
 // src/migrate.ts
 import { logger } from "@elizaos/core";
 import { config } from "dotenv";
+import os from "node:os";
+import path from "node:path";
 config({ path: "../../.env" });
 async function runMigrations() {
   if (process.env.POSTGRES_URL) {
@@ -21,9 +23,12 @@ async function runMigrations() {
       process.exit(1);
     }
   } else {
+    const homeDir = os.homedir();
+    const elizaDir = path.join(homeDir, ".eliza");
+    const elizaDbDir = path.join(elizaDir, "db");
     logger.info("Using PGlite database");
     const clientManager = new PGliteClientManager({
-      dataDir: "../../pglite"
+      dataDir: elizaDbDir
     });
     try {
       await clientManager.initialize();

package/dist/migrate.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/migrate.ts"],"sourceRoot":"./","sourcesContent":["import { logger } from '@elizaos/core';\nimport { config } from 'dotenv';\nimport { PGliteClientManager } from './pglite/manager.js';\nimport { PostgresConnectionManager } from './pg/manager.js';\n\nconfig({ path: '../../.env' });\n\n/**\n * Runs the database migrations based on the environment variable POSTGRES_URL.\n * If the POSTGRES_URL is provided, it indicates the use of a PostgreSQL database and the corresponding migration logic needs to be implemented.\n * If POSTGRES_URL is not provided, it uses a PGlite database and runs the migrations on it.\n * @returns {Promise<void>} A promise that resolves once the migrations are completed successfully or rejects if an error occurs.\n */\nasync function runMigrations() {\n  if (process.env.POSTGRES_URL) {\n    try {\n      const connectionManager = new PostgresConnectionManager(process.env.POSTGRES_URL);\n      await connectionManager.initialize();\n      await connectionManager.runMigrations();\n      // await connectionManager.close();\n      logger.success('PostgreSQL migrations completed successfully');\n      process.exit(0);\n    } catch (error) {\n      logger.warn('PostgreSQL migration failed:', error);\n      process.exit(1);\n    }\n  } else {\n    logger.info('Using PGlite database');\n    const clientManager = new PGliteClientManager({\n      dataDir: '../../pglite',\n    });\n\n    try {\n      await clientManager.initialize();\n      await clientManager.runMigrations();\n      logger.success('PGlite migrations completed successfully');\n      await clientManager.close();\n      process.exit(0);\n    } catch (error) {\n      logger.error('PGlite migration failed:', error);\n      try {\n        await clientManager.close();\n      } catch (closeError) {\n        logger.error('Failed to close PGlite connection:', closeError);\n      }\n      process.exit(1);\n    }\n  }\n}\n\nrunMigrations().catch((error) => {\n  logger.error('Unhandled error in migrations:', error);\n  process.exit(1);\n});\n"],"mappings":";;;;;;;AAAA,SAAS,cAAc;AACvB,SAAS,cAAc;AAIvB,OAAO,EAAE,MAAM,aAAa,CAAC;AAQ7B,eAAe,gBAAgB;AAC7B,MAAI,QAAQ,IAAI,cAAc;AAC5B,QAAI;AACF,YAAM,oBAAoB,IAAI,0BAA0B,QAAQ,IAAI,YAAY;AAChF,YAAM,kBAAkB,WAAW;AACnC,YAAM,kBAAkB,cAAc;AAEtC,aAAO,QAAQ,8CAA8C;AAC7D,cAAQ,KAAK,CAAC;AAAA,IAChB,SAAS,OAAO;AACd,aAAO,KAAK,gCAAgC,KAAK;AACjD,cAAQ,KAAK,CAAC;AAAA,IAChB;AAAA,EACF,OAAO;AACL,WAAO,KAAK,uBAAuB;AACnC,UAAM,gBAAgB,IAAI,oBAAoB;AAAA,MAC5C,SAAS;AAAA,IACX,CAAC;AAED,QAAI;AACF,YAAM,cAAc,WAAW;AAC/B,YAAM,cAAc,cAAc;AAClC,aAAO,QAAQ,0CAA0C;AACzD,YAAM,cAAc,MAAM;AAC1B,cAAQ,KAAK,CAAC;AAAA,IAChB,SAAS,OAAO;AACd,aAAO,MAAM,4BAA4B,KAAK;AAC9C,UAAI;AACF,cAAM,cAAc,MAAM;AAAA,MAC5B,SAAS,YAAY;AACnB,eAAO,MAAM,sCAAsC,UAAU;AAAA,MAC/D;AACA,cAAQ,KAAK,CAAC;AAAA,IAChB;AAAA,EACF;AACF;AAnCe;AAqCf,cAAc,EAAE,MAAM,CAAC,UAAU;AAC/B,SAAO,MAAM,kCAAkC,KAAK;AACpD,UAAQ,KAAK,CAAC;AAChB,CAAC;","names":[]}
+{"version":3,"sources":["../src/migrate.ts"],"sourceRoot":"./","sourcesContent":["import { logger } from '@elizaos/core';\nimport { config } from 'dotenv';\nimport { PGliteClientManager } from './pglite/manager.js';\nimport { PostgresConnectionManager } from './pg/manager.js';\nimport os from 'node:os';\nimport path from 'node:path';\n\nconfig({ path: '../../.env' });\n\n/**\n * Runs the database migrations based on the environment variable POSTGRES_URL.\n * If the POSTGRES_URL is provided, it indicates the use of a PostgreSQL database and the corresponding migration logic needs to be implemented.\n * If POSTGRES_URL is not provided, it uses a PGlite database and runs the migrations on it.\n * @returns {Promise<void>} A promise that resolves once the migrations are completed successfully or rejects if an error occurs.\n */\nasync function runMigrations() {\n  if (process.env.POSTGRES_URL) {\n    try {\n      const connectionManager = new PostgresConnectionManager(process.env.POSTGRES_URL);\n      await connectionManager.initialize();\n      await connectionManager.runMigrations();\n      // await connectionManager.close();\n      logger.success('PostgreSQL migrations completed successfully');\n      process.exit(0);\n    } catch (error) {\n      logger.warn('PostgreSQL migration failed:', error);\n      process.exit(1);\n    }\n  } else {\n    const homeDir = os.homedir();\n    const elizaDir = path.join(homeDir, '.eliza');\n    const elizaDbDir = path.join(elizaDir, 'db');\n\n    logger.info('Using PGlite database');\n    const clientManager = new PGliteClientManager({\n      dataDir: elizaDbDir,\n    });\n\n    try {\n      await clientManager.initialize();\n      await clientManager.runMigrations();\n      logger.success('PGlite migrations completed successfully');\n      await clientManager.close();\n      process.exit(0);\n    } catch (error) {\n      logger.error('PGlite migration failed:', error);\n      try {\n        await clientManager.close();\n      } catch (closeError) {\n        logger.error('Failed to close PGlite connection:', closeError);\n      }\n      process.exit(1);\n    }\n  }\n}\n\nrunMigrations().catch((error) => {\n  logger.error('Unhandled error in migrations:', error);\n  process.exit(1);\n});\n"],"mappings":";;;;;;;AAAA,SAAS,cAAc;AACvB,SAAS,cAAc;AAGvB,OAAO,QAAQ;AACf,OAAO,UAAU;AAEjB,OAAO,EAAE,MAAM,aAAa,CAAC;AAQ7B,eAAe,gBAAgB;AAC7B,MAAI,QAAQ,IAAI,cAAc;AAC5B,QAAI;AACF,YAAM,oBAAoB,IAAI,0BAA0B,QAAQ,IAAI,YAAY;AAChF,YAAM,kBAAkB,WAAW;AACnC,YAAM,kBAAkB,cAAc;AAEtC,aAAO,QAAQ,8CAA8C;AAC7D,cAAQ,KAAK,CAAC;AAAA,IAChB,SAAS,OAAO;AACd,aAAO,KAAK,gCAAgC,KAAK;AACjD,cAAQ,KAAK,CAAC;AAAA,IAChB;AAAA,EACF,OAAO;AACL,UAAM,UAAU,GAAG,QAAQ;AAC3B,UAAM,WAAW,KAAK,KAAK,SAAS,QAAQ;AAC5C,UAAM,aAAa,KAAK,KAAK,UAAU,IAAI;AAE3C,WAAO,KAAK,uBAAuB;AACnC,UAAM,gBAAgB,IAAI,oBAAoB;AAAA,MAC5C,SAAS;AAAA,IACX,CAAC;AAED,QAAI;AACF,YAAM,cAAc,WAAW;AAC/B,YAAM,cAAc,cAAc;AAClC,aAAO,QAAQ,0CAA0C;AACzD,YAAM,cAAc,MAAM;AAC1B,cAAQ,KAAK,CAAC;AAAA,IAChB,SAAS,OAAO;AACd,aAAO,MAAM,4BAA4B,KAAK;AAC9C,UAAI;AACF,cAAM,cAAc,MAAM;AAAA,MAC5B,SAAS,YAAY;AACnB,eAAO,MAAM,sCAAsC,UAAU;AAAA,MAC/D;AACA,cAAQ,KAAK,CAAC;AAAA,IAChB;AAAA,EACF;AACF;AAvCe;AAyCf,cAAc,EAAE,MAAM,CAAC,UAAU;AAC/B,SAAO,MAAM,kCAAkC,KAAK;AACpD,UAAQ,KAAK,CAAC;AAChB,CAAC;","names":[]}

package/drizzle/extension.sql

@@ -0,0 +1,3 @@
+CREATE EXTENSION IF NOT EXISTS vector;
+--> statement-breakpoint				
+CREATE EXTENSION IF NOT EXISTS fuzzystrmatch;

package/drizzle/migrations/{20250302132443_init.sql → 0000_low_anita_blake.sql}

@@ -1,7 +1,6 @@
-CREATE EXTENSION IF NOT EXISTS vector;
---> statement-breakpoint				
-CREATE EXTENSION IF NOT EXISTS fuzzystrmatch;
---> statement-breakpoint
+CREATE EXTENSION IF NOT EXISTS vector;--> statement-breakpoint				
+CREATE EXTENSION IF NOT EXISTS fuzzystrmatch;--> statement-breakpoint		
+
 CREATE TABLE "agents" (
 	"id" uuid PRIMARY KEY DEFAULT gen_random_uuid() NOT NULL,
 	"enabled" boolean DEFAULT true NOT NULL,
@@ -83,6 +82,7 @@ CREATE TABLE "memories" (
 	"entityId" uuid,
 	"agentId" uuid,
 	"roomId" uuid,
+	"worldId" uuid,
 	"unique" boolean DEFAULT true NOT NULL,
 	"metadata" jsonb DEFAULT '{}'::jsonb NOT NULL,
 	CONSTRAINT "fragment_metadata_check" CHECK (
@@ -173,9 +173,11 @@ ALTER TABLE "logs" ADD CONSTRAINT "fk_user" FOREIGN KEY ("entityId") REFERENCES
 ALTER TABLE "memories" ADD CONSTRAINT "memories_entityId_entities_id_fk" FOREIGN KEY ("entityId") REFERENCES "public"."entities"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
 ALTER TABLE "memories" ADD CONSTRAINT "memories_agentId_agents_id_fk" FOREIGN KEY ("agentId") REFERENCES "public"."agents"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
 ALTER TABLE "memories" ADD CONSTRAINT "memories_roomId_rooms_id_fk" FOREIGN KEY ("roomId") REFERENCES "public"."rooms"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "memories" ADD CONSTRAINT "memories_worldId_worlds_id_fk" FOREIGN KEY ("worldId") REFERENCES "public"."worlds"("id") ON DELETE set null ON UPDATE no action;--> statement-breakpoint
 ALTER TABLE "memories" ADD CONSTRAINT "fk_room" FOREIGN KEY ("roomId") REFERENCES "public"."rooms"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
 ALTER TABLE "memories" ADD CONSTRAINT "fk_user" FOREIGN KEY ("entityId") REFERENCES "public"."entities"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
-ALTER TABLE "memories" ADD CONSTRAINT "fk_agent" FOREIGN KEY ("agentId") REFERENCES "public"."entities"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "memories" ADD CONSTRAINT "fk_agent" FOREIGN KEY ("agentId") REFERENCES "public"."agents"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "memories" ADD CONSTRAINT "fk_world" FOREIGN KEY ("worldId") REFERENCES "public"."worlds"("id") ON DELETE set null ON UPDATE no action;--> statement-breakpoint
 ALTER TABLE "participants" ADD CONSTRAINT "participants_entityId_entities_id_fk" FOREIGN KEY ("entityId") REFERENCES "public"."entities"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
 ALTER TABLE "participants" ADD CONSTRAINT "participants_roomId_rooms_id_fk" FOREIGN KEY ("roomId") REFERENCES "public"."rooms"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
 ALTER TABLE "participants" ADD CONSTRAINT "participants_agentId_agents_id_fk" FOREIGN KEY ("agentId") REFERENCES "public"."agents"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
@@ -191,6 +193,7 @@ ALTER TABLE "rooms" ADD CONSTRAINT "rooms_worldId_worlds_id_fk" FOREIGN KEY ("wo
 ALTER TABLE "worlds" ADD CONSTRAINT "worlds_agentId_agents_id_fk" FOREIGN KEY ("agentId") REFERENCES "public"."agents"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
 CREATE INDEX "idx_embedding_memory" ON "embeddings" USING btree ("memory_id");--> statement-breakpoint
 CREATE INDEX "idx_memories_type_room" ON "memories" USING btree ("type","roomId");--> statement-breakpoint
+CREATE INDEX "idx_memories_world_id" ON "memories" USING btree ("worldId");--> statement-breakpoint
 CREATE INDEX "idx_memories_metadata_type" ON "memories" USING btree (((metadata->>'type')));--> statement-breakpoint
 CREATE INDEX "idx_memories_document_id" ON "memories" USING btree (((metadata->>'documentId')));--> statement-breakpoint
 CREATE INDEX "idx_fragments_order" ON "memories" USING btree (((metadata->>'documentId')),((metadata->>'position')));--> statement-breakpoint

package/drizzle/migrations/meta/{20250302132443_snapshot.json → 0000_snapshot.json}

@@ -1,5 +1,5 @@
 {
-  "id": "e868cf2d-10a2-406f-a31e-9342a0f29c35",
+  "id": "8d7ae0aa-76b7-4e80-9133-db45fb9b6d1e",
   "prevId": "00000000-0000-0000-0000-000000000000",
   "version": "7",
   "dialect": "postgresql",
@@ -20,7 +20,7 @@
           "type": "boolean",
           "primaryKey": false,
           "notNull": true,
-          "default": "true"
+          "default": true
         },
         "createdAt": {
           "name": "createdAt",
@@ -620,6 +620,12 @@
           "primaryKey": false,
           "notNull": false
         },
+        "worldId": {
+          "name": "worldId",
+          "type": "uuid",
+          "primaryKey": false,
+          "notNull": false
+        },
         "unique": {
           "name": "unique",
           "type": "boolean",
@@ -657,6 +663,21 @@
           "method": "btree",
           "with": {}
         },
+        "idx_memories_world_id": {
+          "name": "idx_memories_world_id",
+          "columns": [
+            {
+              "expression": "worldId",
+              "isExpression": false,
+              "asc": true,
+              "nulls": "last"
+            }
+          ],
+          "isUnique": false,
+          "concurrently": false,
+          "method": "btree",
+          "with": {}
+        },
         "idx_memories_metadata_type": {
           "name": "idx_memories_metadata_type",
           "columns": [
@@ -737,6 +758,15 @@
           "onDelete": "cascade",
           "onUpdate": "no action"
         },
+        "memories_worldId_worlds_id_fk": {
+          "name": "memories_worldId_worlds_id_fk",
+          "tableFrom": "memories",
+          "tableTo": "worlds",
+          "columnsFrom": ["worldId"],
+          "columnsTo": ["id"],
+          "onDelete": "set null",
+          "onUpdate": "no action"
+        },
         "fk_room": {
           "name": "fk_room",
           "tableFrom": "memories",
@@ -758,11 +788,20 @@
         "fk_agent": {
           "name": "fk_agent",
           "tableFrom": "memories",
-          "tableTo": "entities",
+          "tableTo": "agents",
           "columnsFrom": ["agentId"],
           "columnsTo": ["id"],
           "onDelete": "cascade",
           "onUpdate": "no action"
+        },
+        "fk_world": {
+          "name": "fk_world",
+          "tableFrom": "memories",
+          "tableTo": "worlds",
+          "columnsFrom": ["worldId"],
+          "columnsTo": ["id"],
+          "onDelete": "set null",
+          "onUpdate": "no action"
         }
       },
       "compositePrimaryKeys": {},

package/drizzle/migrations/meta/_journal.json

@@ -5,8 +5,8 @@
     {
       "idx": 0,
       "version": "7",
-      "when": 1740921883121,
-      "tag": "20250302132443_init",
+      "when": 1745909982134,
+      "tag": "0000_low_anita_blake",
       "breakpoints": true
     }
   ]

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@elizaos/plugin-sql",
-  "version": "1.0.0-beta.4",
+  "version": "1.0.0-beta.41",
   "type": "module",
   "main": "dist/index.js",
   "module": "dist/index.js",
@@ -27,29 +27,36 @@
   ],
   "dependencies": {
     "@electric-sql/pglite": "^0.2.17",
-    "@elizaos/core": "^1.0.0-beta.4",
+    "@elizaos/core": "^1.0.0-beta.41",
     "@types/pg": "8.11.10",
     "drizzle-kit": "^0.30.4",
     "drizzle-orm": "^0.39.1",
     "pg": "^8.13.3"
   },
   "devDependencies": {
+    "@types/node": "22.8.4",
     "dockerode": "^4.0.4",
     "prettier": "3.5.3",
-    "tsup": "8.4.0"
+    "tsup": "8.4.0",
+    "vitest": "^1.6.1",
+    "zod": "^3.22.4"
   },
   "scripts": {
     "build": "tsup",
     "dev": "tsup --watch",
-    "migrate:generate": "drizzle-kit generate:pg",
+    "migrate:generate": "drizzle-kit generate",
     "migrate": "tsup src/migrate.ts --format esm --clean=false --onSuccess=\"node dist/migrate.js\"",
     "lint": "prettier --write ./src",
     "clean": "rm -rf dist .turbo node_modules .turbo-tsconfig.json tsconfig.tsbuildinfo",
     "format": "prettier --write ./src",
-    "format:check": "prettier --check ./src"
+    "format:check": "prettier --check ./src",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "test:coverage": "vitest run --coverage"
   },
   "peerDependencies": {
-    "typescript": "5.8.2"
+    "typescript": "5.8.2",
+    "whatwg-url": "7.1.0"
   },
-  "gitHead": "55e5f31e3f9920fce14c57f149e185aa128e335b"
+  "gitHead": "62132d275b78fa5ae4d604586dffbbaf1be5a4c5"
 }