@inkeep/agents-core 0.41.2 → 0.42.0

package/dist/db/manage/test-manage-client.d.ts

@@ -0,0 +1,27 @@
+import { AgentsManageDatabaseClient } from "./manage-client.js";
+
+//#region src/db/manage/test-manage-client.d.ts
+
+/**
+ * Creates a test database client using an in-memory PostgreSQL database (PGlite)
+ * This provides real database operations for integration testing with perfect isolation
+ * Each call creates a fresh database with all migrations applied
+ */
+declare function createTestManageDatabaseClient(drizzleDir: string): Promise<AgentsManageDatabaseClient>;
+declare function createTestManageDatabaseClientNoMigrations(): AgentsManageDatabaseClient;
+/**
+ * Cleans up test database by removing all data but keeping schema
+ * Dynamically gets all tables from the public schema and truncates them
+ */
+declare function cleanupTestManageDatabase(db: AgentsManageDatabaseClient): Promise<void>;
+/**
+ * Closes the test database and removes the file
+ */
+declare function closeTestManageDatabase(db: AgentsManageDatabaseClient): Promise<void>;
+/**
+ * Creates a test project in the database
+ * NOTE: original implementation ensured the organization exists first but that is a runtime db entity now
+ */
+declare function createTestProject(db: AgentsManageDatabaseClient, tenantId: string, projectId?: string): Promise<void>;
+//#endregion
+export { cleanupTestManageDatabase, closeTestManageDatabase, createTestManageDatabaseClient, createTestManageDatabaseClientNoMigrations, createTestProject };

package/dist/db/manage/test-manage-client.js

@@ -0,0 +1,68 @@
+import { manage_schema_exports, projects } from "./manage-schema.js";
+import { sql } from "drizzle-orm";
+import { PGlite } from "@electric-sql/pglite";
+import { drizzle } from "drizzle-orm/pglite";
+import { migrate } from "drizzle-orm/pglite/migrator";
+
+//#region src/db/manage/test-manage-client.ts
+/**
+* Creates a test database client using an in-memory PostgreSQL database (PGlite)
+* This provides real database operations for integration testing with perfect isolation
+* Each call creates a fresh database with all migrations applied
+*/
+async function createTestManageDatabaseClient(drizzleDir) {
+	const db = drizzle(new PGlite(), { schema: manage_schema_exports });
+	try {
+		await migrate(db, { migrationsFolder: drizzleDir });
+	} catch (error) {
+		console.error("Failed to initialize test database schema:", error);
+		throw error;
+	}
+	return db;
+}
+function createTestManageDatabaseClientNoMigrations() {
+	return drizzle(new PGlite(), { schema: manage_schema_exports });
+}
+/**
+* Cleans up test database by removing all data but keeping schema
+* Dynamically gets all tables from the public schema and truncates them
+*/
+async function cleanupTestManageDatabase(db) {
+	try {
+		const tables = (await db.execute(sql.raw(`
+      SELECT tablename 
+      FROM pg_tables 
+      WHERE schemaname = 'public'
+    `))).rows.map((row) => row.tablename);
+		if (tables.length === 0) return;
+		const tableList = tables.map((t) => `"${t}"`).join(", ");
+		await db.execute(sql.raw(`TRUNCATE TABLE ${tableList} RESTART IDENTITY CASCADE`));
+	} catch (error) {
+		console.debug("Could not clean test database:", error);
+	}
+}
+/**
+* Closes the test database and removes the file
+*/
+async function closeTestManageDatabase(db) {
+	try {
+		if ("close" in db && typeof db.close === "function") db.close();
+	} catch (error) {
+		console.debug("Error closing database:", error);
+	}
+}
+/**
+* Creates a test project in the database
+* NOTE: original implementation ensured the organization exists first but that is a runtime db entity now
+*/
+async function createTestProject(db, tenantId, projectId = "default") {
+	await db.insert(projects).values({
+		tenantId,
+		id: projectId,
+		name: `Test Project ${projectId}`,
+		description: `Test project for ${projectId}`
+	}).onConflictDoNothing();
+}
+
+//#endregion
+export { cleanupTestManageDatabase, closeTestManageDatabase, createTestManageDatabaseClient, createTestManageDatabaseClientNoMigrations, createTestProject };

package/dist/db/runtime/runtime-client.d.ts

@@ -0,0 +1,20 @@
+import { runtime_schema_d_exports } from "./runtime-schema.js";
+import { NodePgDatabase } from "drizzle-orm/node-postgres";
+import { PgliteDatabase } from "drizzle-orm/pglite";
+
+//#region src/db/runtime/runtime-client.d.ts
+type AgentsRunDatabaseClient = NodePgDatabase<typeof runtime_schema_d_exports> | PgliteDatabase<typeof runtime_schema_d_exports>;
+interface AgentsRunDatabaseConfig {
+  connectionString?: string;
+  poolSize?: number;
+  ssl?: boolean;
+  logger?: {
+    logQuery: (query: string, params: unknown[]) => void;
+  };
+}
+/**
+ * Creates a PostgreSQL database client with connection pooling
+ */
+declare function createAgentsRunDatabaseClient(config?: AgentsRunDatabaseConfig): AgentsRunDatabaseClient;
+//#endregion
+export { AgentsRunDatabaseClient, AgentsRunDatabaseConfig, createAgentsRunDatabaseClient };

package/dist/db/runtime/runtime-client.js

@@ -0,0 +1,30 @@
+import { runtime_schema_exports } from "./runtime-schema.js";
+import { env, loadEnvironmentFiles } from "../../env.js";
+import { createTestRuntimeDatabaseClientNoMigrations } from "./test-runtime-client.js";
+import { drizzle } from "drizzle-orm/node-postgres";
+import { Pool } from "pg";
+
+//#region src/db/runtime/runtime-client.ts
+loadEnvironmentFiles();
+/**
+* Creates a PostgreSQL database client with connection pooling
+*/
+function createAgentsRunDatabaseClient(config) {
+	const connectionString = config?.connectionString || env.INKEEP_AGENTS_RUN_DATABASE_URL;
+	if (env.ENVIRONMENT === "test") return createTestRuntimeDatabaseClientNoMigrations();
+	if (!connectionString) throw new Error("INKEEP_AGENTS_RUN_DATABASE_URL environment variable is required. Please set it to your PostgreSQL connection string.");
+	const pool = new Pool({
+		connectionString,
+		max: config?.poolSize || Number(env.POSTGRES_POOL_SIZE) || 100
+	});
+	pool.on("error", (err) => {
+		console.error("Unexpected PostgreSQL pool error:", err);
+	});
+	return drizzle(pool, {
+		schema: runtime_schema_exports,
+		logger: config?.logger
+	});
+}
+
+//#endregion
+export { createAgentsRunDatabaseClient };