@elizaos/plugin-sql 1.5.9-alpha.7 → 1.5.9-alpha.9
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
|
@@ -2,7 +2,7 @@
|
|
|
2
2
|
"version": 3,
|
|
3
3
|
"sources": ["../../src/index.node.ts", "../../src/pglite/adapter.ts", "../../../../node_modules/drizzle-orm/pglite/driver.js", "../../../../node_modules/drizzle-orm/entity.js", "../../../../node_modules/drizzle-orm/logger.js", "../../../../node_modules/drizzle-orm/query-promise.js", "../../../../node_modules/drizzle-orm/column.js", "../../../../node_modules/drizzle-orm/column-builder.js", "../../../../node_modules/drizzle-orm/table.utils.js", "../../../../node_modules/drizzle-orm/pg-core/foreign-keys.js", "../../../../node_modules/drizzle-orm/tracing-utils.js", "../../../../node_modules/drizzle-orm/pg-core/unique-constraint.js", "../../../../node_modules/drizzle-orm/pg-core/utils/array.js", "../../../../node_modules/drizzle-orm/pg-core/columns/common.js", "../../../../node_modules/drizzle-orm/pg-core/columns/enum.js", "../../../../node_modules/drizzle-orm/subquery.js", "../../../../node_modules/drizzle-orm/version.js", "../../../../node_modules/drizzle-orm/tracing.js", "../../../../node_modules/drizzle-orm/view-common.js", "../../../../node_modules/drizzle-orm/table.js", "../../../../node_modules/drizzle-orm/sql/sql.js", "../../../../node_modules/drizzle-orm/alias.js", "../../../../node_modules/drizzle-orm/selection-proxy.js", "../../../../node_modules/drizzle-orm/utils.js", "../../../../node_modules/drizzle-orm/pg-core/columns/int.common.js", "../../../../node_modules/drizzle-orm/pg-core/columns/bigint.js", "../../../../node_modules/drizzle-orm/pg-core/columns/bigserial.js", "../../../../node_modules/drizzle-orm/pg-core/columns/boolean.js", "../../../../node_modules/drizzle-orm/pg-core/columns/char.js", "../../../../node_modules/drizzle-orm/pg-core/columns/cidr.js", "../../../../node_modules/drizzle-orm/pg-core/columns/custom.js", "../../../../node_modules/drizzle-orm/pg-core/columns/date.common.js", "../../../../node_modules/drizzle-orm/pg-core/columns/date.js", "../../../../node_modules/drizzle-orm/pg-core/columns/double-precision.js", "../../../../node_modules/drizzle-orm/pg-core/columns/inet.js", "../../../../node_modules/drizzle-orm/pg-core/columns/integer.js", "../../../../node_modules/drizzle-orm/pg-core/columns/interval.js", "../../../../node_modules/drizzle-orm/pg-core/columns/json.js", "../../../../node_modules/drizzle-orm/pg-core/columns/jsonb.js", "../../../../node_modules/drizzle-orm/pg-core/columns/line.js", "../../../../node_modules/drizzle-orm/pg-core/columns/macaddr.js", "../../../../node_modules/drizzle-orm/pg-core/columns/macaddr8.js", "../../../../node_modules/drizzle-orm/pg-core/columns/numeric.js", "../../../../node_modules/drizzle-orm/pg-core/columns/point.js", "../../../../node_modules/drizzle-orm/pg-core/columns/postgis_extension/utils.js", "../../../../node_modules/drizzle-orm/pg-core/columns/postgis_extension/geometry.js", "../../../../node_modules/drizzle-orm/pg-core/columns/real.js", "../../../../node_modules/drizzle-orm/pg-core/columns/serial.js", "../../../../node_modules/drizzle-orm/pg-core/columns/smallint.js", "../../../../node_modules/drizzle-orm/pg-core/columns/smallserial.js", "../../../../node_modules/drizzle-orm/pg-core/columns/text.js", "../../../../node_modules/drizzle-orm/pg-core/columns/time.js", "../../../../node_modules/drizzle-orm/pg-core/columns/timestamp.js", "../../../../node_modules/drizzle-orm/pg-core/columns/uuid.js", "../../../../node_modules/drizzle-orm/pg-core/columns/varchar.js", "../../../../node_modules/drizzle-orm/pg-core/columns/vector_extension/bit.js", "../../../../node_modules/drizzle-orm/pg-core/columns/vector_extension/halfvec.js", "../../../../node_modules/drizzle-orm/pg-core/columns/vector_extension/sparsevec.js", "../../../../node_modules/drizzle-orm/pg-core/columns/vector_extension/vector.js", "../../../../node_modules/drizzle-orm/pg-core/columns/all.js", "../../../../node_modules/drizzle-orm/pg-core/table.js", "../../../../node_modules/drizzle-orm/pg-core/checks.js", "../../../../node_modules/drizzle-orm/pg-core/indexes.js", "../../../../node_modules/drizzle-orm/pg-core/primary-keys.js", "../../../../node_modules/drizzle-orm/casing.js", "../../../../node_modules/drizzle-orm/errors.js", "../../../../node_modules/drizzle-orm/sql/expressions/conditions.js", "../../../../node_modules/drizzle-orm/sql/expressions/select.js", "../../../../node_modules/drizzle-orm/relations.js", "../../../../node_modules/drizzle-orm/sql/functions/aggregate.js", "../../../../node_modules/drizzle-orm/sql/functions/vector.js", "../../../../node_modules/drizzle-orm/pg-core/view-base.js", "../../../../node_modules/drizzle-orm/pg-core/dialect.js", "../../../../node_modules/drizzle-orm/query-builders/query-builder.js", "../../../../node_modules/drizzle-orm/pg-core/query-builders/select.js", "../../../../node_modules/drizzle-orm/pg-core/query-builders/query-builder.js", "../../../../node_modules/drizzle-orm/pg-core/utils.js", "../../../../node_modules/drizzle-orm/pg-core/query-builders/delete.js", "../../../../node_modules/drizzle-orm/pg-core/query-builders/insert.js", "../../../../node_modules/drizzle-orm/pg-core/query-builders/refresh-materialized-view.js", "../../../../node_modules/drizzle-orm/pg-core/query-builders/update.js", "../../../../node_modules/drizzle-orm/pg-core/query-builders/count.js", "../../../../node_modules/drizzle-orm/pg-core/query-builders/query.js", "../../../../node_modules/drizzle-orm/pg-core/query-builders/raw.js", "../../../../node_modules/drizzle-orm/pg-core/db.js", "../../../../node_modules/drizzle-orm/cache/core/cache.js", "../../../../node_modules/drizzle-orm/pg-core/session.js", "../../../../node_modules/drizzle-orm/pglite/session.js", "../../src/base.ts", "../../src/schema/embedding.ts", "../../src/schema/agent.ts", "../../src/schema/entity.ts", "../../src/schema/room.ts", "../../src/schema/memory.ts", "../../src/schema/cache.ts", "../../src/schema/world.ts", "../../src/schema/component.ts", "../../src/schema/log.ts", "../../src/schema/participant.ts", "../../src/schema/relationship.ts", "../../src/schema/tasks.ts", "../../src/schema/messageServer.ts", "../../src/schema/channel.ts", "../../src/schema/message.ts", "../../src/schema/channelParticipant.ts", "../../src/schema/serverAgent.ts", "../../src/pglite/manager.ts", "../../src/pg/adapter.ts", "../../../../node_modules/drizzle-orm/node-postgres/driver.js", "../../../../node_modules/drizzle-orm/node-postgres/session.js", "../../src/pg/manager.ts", "../../src/utils.node.ts", "../../src/migration-service.ts", "../../src/custom-migrator.ts"],
|
|
4
4
|
"sourcesContent": [
|
|
5
|
-
"import type { IDatabaseAdapter, UUID } from '@elizaos/core';\nimport { type IAgentRuntime, type Plugin, logger } from '@elizaos/core';\nimport { PgliteDatabaseAdapter } from './pglite/adapter';\nimport { PGliteClientManager } from './pglite/manager';\nimport { PgDatabaseAdapter } from './pg/adapter';\nimport { PostgresConnectionManager } from './pg/manager';\nimport { resolvePgliteDir } from './utils.node';\nimport * as schema from './schema';\n\nconst GLOBAL_SINGLETONS = Symbol.for('@elizaos/plugin-sql/global-singletons');\n\ninterface GlobalSingletons {\n pgLiteClientManager?: PGliteClientManager;\n postgresConnectionManager?: PostgresConnectionManager;\n}\n\nconst globalSymbols = globalThis as unknown as Record<symbol, GlobalSingletons>;\nif (!globalSymbols[GLOBAL_SINGLETONS]) {\n globalSymbols[GLOBAL_SINGLETONS] = {};\n}\nconst globalSingletons = globalSymbols[GLOBAL_SINGLETONS];\n\nexport function createDatabaseAdapter(\n config: {\n dataDir?: string;\n postgresUrl?: string;\n },\n agentId: UUID\n): IDatabaseAdapter {\n if (config.postgresUrl) {\n if (!globalSingletons.postgresConnectionManager) {\n globalSingletons.postgresConnectionManager = new PostgresConnectionManager(\n config.postgresUrl\n );\n }\n return new PgDatabaseAdapter(agentId, globalSingletons.postgresConnectionManager);\n }\n\n const dataDir = resolvePgliteDir(config.dataDir);\n if (!globalSingletons.pgLiteClientManager) {\n globalSingletons.pgLiteClientManager = new PGliteClientManager({ dataDir });\n }\n return new PgliteDatabaseAdapter(agentId, globalSingletons.pgLiteClientManager);\n}\n\nexport const plugin: Plugin = {\n name: '@elizaos/plugin-sql',\n description: 'A plugin for SQL database access with dynamic schema migrations',\n priority: 0,\n schema: schema,\n init: async (_config, runtime: IAgentRuntime) => {\n logger.info('plugin-sql (node) init starting...');\n\n const adapterRegistered = await runtime\n .isReady()\n .then(() => true)\n .catch((error) => {\n logger.warn(\n 'Database adapter readiness check failed; proceeding to register adapter',\n error\n );\n return false;\n });\n if (adapterRegistered) {\n logger.info('Database adapter already registered, skipping creation');\n return;\n }\n\n const postgresUrl = runtime.getSetting('POSTGRES_URL');\n const dataDir =\n runtime.getSetting('PGLITE_PATH') ||\n runtime.getSetting('DATABASE_PATH') ||\n './.eliza/.elizadb';\n\n const dbAdapter = createDatabaseAdapter(\n {\n dataDir,\n postgresUrl,\n },\n runtime.agentId\n );\n\n runtime.registerDatabaseAdapter(dbAdapter);\n logger.info('Database adapter created and registered');\n },\n};\n\nexport default plugin;\n\
|
|
5
|
+
"import type { IDatabaseAdapter, UUID } from '@elizaos/core';\nimport { type IAgentRuntime, type Plugin, logger } from '@elizaos/core';\nimport { PgliteDatabaseAdapter } from './pglite/adapter';\nimport { PGliteClientManager } from './pglite/manager';\nimport { PgDatabaseAdapter } from './pg/adapter';\nimport { PostgresConnectionManager } from './pg/manager';\nimport { resolvePgliteDir } from './utils.node';\nimport * as schema from './schema';\n\nconst GLOBAL_SINGLETONS = Symbol.for('@elizaos/plugin-sql/global-singletons');\n\ninterface GlobalSingletons {\n pgLiteClientManager?: PGliteClientManager;\n postgresConnectionManager?: PostgresConnectionManager;\n}\n\nconst globalSymbols = globalThis as unknown as Record<symbol, GlobalSingletons>;\nif (!globalSymbols[GLOBAL_SINGLETONS]) {\n globalSymbols[GLOBAL_SINGLETONS] = {};\n}\nconst globalSingletons = globalSymbols[GLOBAL_SINGLETONS];\n\nexport function createDatabaseAdapter(\n config: {\n dataDir?: string;\n postgresUrl?: string;\n },\n agentId: UUID\n): IDatabaseAdapter {\n if (config.postgresUrl) {\n if (!globalSingletons.postgresConnectionManager) {\n globalSingletons.postgresConnectionManager = new PostgresConnectionManager(\n config.postgresUrl\n );\n }\n return new PgDatabaseAdapter(agentId, globalSingletons.postgresConnectionManager);\n }\n\n const dataDir = resolvePgliteDir(config.dataDir);\n if (!globalSingletons.pgLiteClientManager) {\n globalSingletons.pgLiteClientManager = new PGliteClientManager({ dataDir });\n }\n return new PgliteDatabaseAdapter(agentId, globalSingletons.pgLiteClientManager);\n}\n\nexport const plugin: Plugin = {\n name: '@elizaos/plugin-sql',\n description: 'A plugin for SQL database access with dynamic schema migrations',\n priority: 0,\n schema: schema,\n init: async (_config, runtime: IAgentRuntime) => {\n logger.info('plugin-sql (node) init starting...');\n\n const adapterRegistered = await runtime\n .isReady()\n .then(() => true)\n .catch((error) => {\n logger.warn(\n 'Database adapter readiness check failed; proceeding to register adapter',\n error\n );\n return false;\n });\n if (adapterRegistered) {\n logger.info('Database adapter already registered, skipping creation');\n return;\n }\n\n const postgresUrl = runtime.getSetting('POSTGRES_URL');\n const dataDir =\n runtime.getSetting('PGLITE_PATH') ||\n runtime.getSetting('DATABASE_PATH') ||\n './.eliza/.elizadb';\n\n const dbAdapter = createDatabaseAdapter(\n {\n dataDir,\n postgresUrl,\n },\n runtime.agentId\n );\n\n runtime.registerDatabaseAdapter(dbAdapter);\n logger.info('Database adapter created and registered');\n },\n};\n\nexport default plugin;\n\nexport { DatabaseMigrationService } from './migration-service';\n\n\n",
|
|
6
6
|
"import { type UUID, logger, type Agent, type Entity, type Memory } from '@elizaos/core';\nimport { drizzle } from 'drizzle-orm/pglite';\nimport { BaseDrizzleAdapter } from '../base';\nimport { DIMENSION_MAP, type EmbeddingDimensionColumn } from '../schema/embedding';\nimport type { PGliteClientManager } from './manager';\n\n/**\n * PgliteDatabaseAdapter class represents an adapter for interacting with a PgliteDatabase.\n * Extends BaseDrizzleAdapter.\n *\n * @constructor\n * @param {UUID} agentId - The ID of the agent.\n * @param {PGliteClientManager} manager - The manager for the Pglite client.\n *\n * @method withDatabase\n * @param {() => Promise<T>} operation - The operation to perform on the database.\n * @return {Promise<T>} - The result of the operation.\n *\n * @method init\n * @return {Promise<void>} - A Promise that resolves when the initialization is complete.\n *\n * @method close\n * @return {void} - A Promise that resolves when the database is closed.\n */\nexport class PgliteDatabaseAdapter extends BaseDrizzleAdapter {\n private manager: PGliteClientManager;\n protected embeddingDimension: EmbeddingDimensionColumn = DIMENSION_MAP[384];\n\n /**\n * Constructor for creating an instance of a class.\n * @param {UUID} agentId - The unique identifier for the agent.\n * @param {PGliteClientManager} manager - The manager for the Pglite client.\n */\n constructor(agentId: UUID, manager: PGliteClientManager) {\n super(agentId);\n this.manager = manager;\n this.db = drizzle(this.manager.getConnection() as any);\n }\n\n // Methods required by TypeScript but not in base class\n async getEntityByIds(entityIds: UUID[]): Promise<Entity[] | null> {\n // Delegate to the correct method name\n return this.getEntitiesByIds(entityIds);\n }\n\n async getMemoriesByServerId(_params: { serverId: UUID; count?: number }): Promise<Memory[]> {\n // This method doesn't seem to exist in the base implementation\n // Provide a basic implementation that returns empty array\n logger.warn('getMemoriesByServerId called but not implemented - returning empty array');\n return [];\n }\n\n async ensureAgentExists(agent: Partial<Agent>): Promise<Agent> {\n // Check if agent exists, create if not\n const existingAgent = await this.getAgent(this.agentId);\n if (existingAgent) {\n return existingAgent;\n }\n\n // Create the agent with required fields\n const newAgent: Agent = {\n id: this.agentId,\n name: agent.name || 'Unknown Agent',\n username: agent.username,\n bio: agent.bio || 'An AI agent',\n createdAt: agent.createdAt || Date.now(),\n updatedAt: agent.updatedAt || Date.now(),\n };\n\n await this.createAgent(newAgent);\n const createdAgent = await this.getAgent(this.agentId);\n if (!createdAgent) {\n throw new Error('Failed to create agent');\n }\n return createdAgent;\n }\n\n /**\n * Runs database migrations. For PGLite, migrations are handled by the\n * migration service, not the adapter itself.\n * @returns {Promise<void>}\n */\n async runMigrations(): Promise<void> {\n logger.debug('PgliteDatabaseAdapter: Migrations are handled by the migration service');\n // Migrations are handled by the migration service, not the adapter\n }\n\n /**\n * Asynchronously runs the provided database operation while checking if the database is currently shutting down.\n * If the database is shutting down, a warning is logged and null is returned.\n *\n * @param {Function} operation - The database operation to be performed.\n * @returns {Promise<T>} A promise that resolves with the result of the database operation.\n */\n protected async withDatabase<T>(operation: () => Promise<T>): Promise<T> {\n if (this.manager.isShuttingDown()) {\n logger.warn('Database is shutting down');\n return null as unknown as T;\n }\n return operation();\n }\n\n /**\n * Asynchronously initializes the database by running migrations.\n *\n * @returns {Promise<void>} A Promise that resolves when the database initialization is complete.\n */\n async init(): Promise<void> {\n logger.debug('PGliteDatabaseAdapter initialized, skipping automatic migrations.');\n }\n\n /**\n * Checks if the database connection is ready and active.\n * For PGLite, this checks if the client is not in a shutting down state.\n * @returns {Promise<boolean>} A Promise that resolves to true if the connection is healthy.\n */\n async isReady(): Promise<boolean> {\n return !this.manager.isShuttingDown();\n }\n\n /**\n * Asynchronously closes the database.\n */\n async close() {\n await this.manager.close();\n }\n\n /**\n * Asynchronously retrieves the connection from the client.\n *\n * @returns {Promise<PGlite>} A Promise that resolves with the connection.\n */\n async getConnection() {\n return this.manager.getConnection();\n }\n}\n",
|
|
7
7
|
"import { PGlite } from \"@electric-sql/pglite\";\nimport { entityKind } from \"../entity.js\";\nimport { DefaultLogger } from \"../logger.js\";\nimport { PgDatabase } from \"../pg-core/db.js\";\nimport { PgDialect } from \"../pg-core/dialect.js\";\nimport {\n createTableRelationsHelpers,\n extractTablesRelationalConfig\n} from \"../relations.js\";\nimport { isConfig } from \"../utils.js\";\nimport { PgliteSession } from \"./session.js\";\nclass PgliteDriver {\n constructor(client, dialect, options = {}) {\n this.client = client;\n this.dialect = dialect;\n this.options = options;\n }\n static [entityKind] = \"PgliteDriver\";\n createSession(schema) {\n return new PgliteSession(this.client, this.dialect, schema, {\n logger: this.options.logger,\n cache: this.options.cache\n });\n }\n}\nclass PgliteDatabase extends PgDatabase {\n static [entityKind] = \"PgliteDatabase\";\n}\nfunction construct(client, config = {}) {\n const dialect = new PgDialect({ casing: config.casing });\n let logger;\n if (config.logger === true) {\n logger = new DefaultLogger();\n } else if (config.logger !== false) {\n logger = config.logger;\n }\n let schema;\n if (config.schema) {\n const tablesConfig = extractTablesRelationalConfig(\n config.schema,\n createTableRelationsHelpers\n );\n schema = {\n fullSchema: config.schema,\n schema: tablesConfig.tables,\n tableNamesMap: tablesConfig.tableNamesMap\n };\n }\n const driver = new PgliteDriver(client, dialect, { logger, cache: config.cache });\n const session = driver.createSession(schema);\n const db = new PgliteDatabase(dialect, session, schema);\n db.$client = client;\n db.$cache = config.cache;\n if (db.$cache) {\n db.$cache[\"invalidate\"] = config.cache?.onMutate;\n }\n return db;\n}\nfunction drizzle(...params) {\n if (params[0] === void 0 || typeof params[0] === \"string\") {\n const instance = new PGlite(params[0]);\n return construct(instance, params[1]);\n }\n if (isConfig(params[0])) {\n const { connection, client, ...drizzleConfig } = params[0];\n if (client) return construct(client, drizzleConfig);\n if (typeof connection === \"object\") {\n const { dataDir, ...options } = connection;\n const instance2 = new PGlite(dataDir, options);\n return construct(instance2, drizzleConfig);\n }\n const instance = new PGlite(connection);\n return construct(instance, drizzleConfig);\n }\n return construct(params[0], params[1]);\n}\n((drizzle2) => {\n function mock(config) {\n return construct({}, config);\n }\n drizzle2.mock = mock;\n})(drizzle || (drizzle = {}));\nexport {\n PgliteDatabase,\n PgliteDriver,\n drizzle\n};\n//# sourceMappingURL=driver.js.map",
|
|
8
8
|
"const entityKind = Symbol.for(\"drizzle:entityKind\");\nconst hasOwnEntityKind = Symbol.for(\"drizzle:hasOwnEntityKind\");\nfunction is(value, type) {\n if (!value || typeof value !== \"object\") {\n return false;\n }\n if (value instanceof type) {\n return true;\n }\n if (!Object.prototype.hasOwnProperty.call(type, entityKind)) {\n throw new Error(\n `Class \"${type.name ?? \"<unknown>\"}\" doesn't look like a Drizzle entity. If this is incorrect and the class is provided by Drizzle, please report this as a bug.`\n );\n }\n let cls = Object.getPrototypeOf(value).constructor;\n if (cls) {\n while (cls) {\n if (entityKind in cls && cls[entityKind] === type[entityKind]) {\n return true;\n }\n cls = Object.getPrototypeOf(cls);\n }\n }\n return false;\n}\nexport {\n entityKind,\n hasOwnEntityKind,\n is\n};\n//# sourceMappingURL=entity.js.map",
|