@databricks/appkit 0.75.0 → 0.75.1

package/CLAUDE.md

@@ -97,7 +97,7 @@ npx @databricks/appkit docs <query>
 - [Function: createLakebasePool()](./docs/api/appkit/Function.createLakebasePool.md): Create a Lakebase pool with appkit's logger integration.
 - [Function: createLakebasePoolManager()](./docs/api/appkit/Function.createLakebasePoolManager.md): Create a pool manager that maintains per-key Lakebase connection pools.
 - [Function: createWorkspaceClient()](./docs/api/appkit/Function.createWorkspaceClient.md): Construct an AppKit workspace client.
-- [Function: database()](./docs/api/appkit/Function.database.md): Create a typed database plugin registration for a finalized schema.
+- [Function: database()](./docs/api/appkit/Function.database.md): Call Signature
 - [Function: defineEval()](./docs/api/appkit/Function.defineEval.md): Define an agent eval. Default-export the result from a
 - [Function: defineEvalConfig()](./docs/api/appkit/Function.defineEvalConfig.md): Define per-directory eval config. Default-export from evals.config.ts.
 - [Function: defineManifest()](./docs/api/appkit/Function.defineManifest.md): Validates a raw manifest (typically a manifest.json import) against the

package/dist/appkit/package.js

@@ -1,6 +1,6 @@
 //#region package.json
 var name = "@databricks/appkit";
-var version = "0.75.0";
+var version = "0.75.1";
 
 //#endregion
 export { name, version };

package/dist/beta.d.ts

@@ -34,10 +34,10 @@ import { agents } from "./plugins/agents/agents.js";
 import "./plugins/agents/index.js";
 import { IAiSearchConfig, IndexConfig, RerankerConfig, SearchFilters, SearchRequest, SearchResponse, SearchResult } from "./plugins/ai-search/types.js";
 import { aiSearch } from "./plugins/ai-search/ai-search.js";
+import "./plugins/beta-exports.generated.js";
 import { DatabaseExports, TransactionClient } from "./plugins/database/entity-types.js";
 import { EntityMutationHooks, HookApp, HookContext } from "./plugins/database/hooks.js";
 import { DatabaseApiConfig, DatabaseApiWriteOperation, DatabaseApiWritesConfig, EntityHooks, IDatabaseConfig, ReadSerializer, ReadSerializerContext } from "./plugins/database/types.js";
 import { database } from "./plugins/database/database.js";
 import "./plugins/database/index.js";
-import "./plugins/beta-exports.generated.js";
 export { type AgentAdapter, type AgentDefinition, type AgentEvent, type AgentInput, type AgentRunContext, type AgentTool, type AgentToolDefinition, type AgentTools, type AgentToolsFn, type AgentsPluginConfig, AppKitMcpClient, AssertionHandle, AssertionResult, Assessment, type AutoInheritToolsConfig, type BaseSystemPromptOption, CustomJudgeSpec, type DatabaseApiConfig, type DatabaseApiWriteOperation, type DatabaseApiWritesConfig, type DatabaseExports, DatabricksAdapter, DatabricksAuth, DatasetRow, DiscoveredEval, DiscoveredEvalConfig, DriveResult, type EntityHooks, type EntityMutationHooks, EvalDefinition, EvalDriver, EvalProgress, EvalResult, EvalRunSummary, EvalSummary, EvalWebServer, type FunctionTool, type GenerationParams, type HookApp, type HookContext, type HostedSupervisorTool, type HostedTool, HttpDriverOptions, type IAiSearchConfig, type IDatabaseConfig, type IndexConfig, JudgeConfig, JudgeScore, MatchResult, Matcher, type McpConnectAllResult, type Message, MlflowClient, type PluginToolkitProvider, type Plugins, PostResult, type PromptContext, ReadEvalDatasetOptions, type ReadSerializer, type ReadSerializerContext, type RegisteredAgent, ReportOutcome, type RerankerConfig, ResolveDatabricksAuthOptions, type ResolvedToolEntry, type RunAgentInput, type RunAgentResult, RunEvalOptions, RunEvalsOptions, SUPERVISOR_EXTENSION_KEY, type Schema, type SearchFilters, type SearchRequest, type SearchResponse, type SearchResult, Severity, SupervisorApiAdapter, type SupervisorApiAdapterOptions, type SupervisorExtension, type SupervisorTool, TestContext, type Thread, type ThreadStore, type ToolAnnotations, type ToolConfig, type ToolEntry, type ToolProvider, type ToolRegistry, type ToolkitEntry, type ToolkitOptions, type TransactionClient, type WorkspaceClientLike, agentIdFromMarkdownPath, agents, aiSearch, bigid, bigint, boolean, buildAssessments, configureJudge, createAgent, createHttpDriver, database, defineEval, defineEvalConfig, defineSchema, defineTool, discoverEvalConfigs, discoverEvalFiles, enumColumn, equals, evalGlyph, executeFromRegistry, findRootEvalConfig, fk, formatEvalDetail, formatEvalHeadline, formatEvalResults, formatResultsJUnit, formatResultsJson, formatSummaryLine, fromSupervisorApi, functionToolToDefinition, id, includes, integer, isFunctionTool, isHostedTool, isJudgeConfigured, isSupervisorTool, isToolkitEntry, jsonb, loadAgentFromFile, loadAgentsFromDir, loadRootEvalConfig, matches, mcpServer, normalizeHost, parseTextToolCalls, readEvalDataset, reportToMlflow, resolveDatabricksAuth, resolveHostedTools, resolveWorkspaceClient, runAgent, runEval, runEvalsInDir, runWithRetries, summarize, supervisorTools, text, timestamp, tool, toolsFromRegistry, userTurns, uuid, varchar };

package/dist/beta.js

@@ -30,7 +30,8 @@ import { agentIdFromMarkdownPath, loadAgentFromFile, loadAgentsFromDir } from ".
 import { agents } from "./plugins/agents/agents.js";
 import "./plugins/agents/index.js";
 import { aiSearch } from "./plugins/ai-search/ai-search.js";
-import { database } from "./plugins/database/database.js";
 import "./plugins/beta-exports.generated.js";
+import { database } from "./plugins/database/database.js";
+import "./plugins/database/index.js";
 
 export { AppKitMcpClient, DatabricksAdapter, MlflowClient, SUPERVISOR_EXTENSION_KEY, SupervisorApiAdapter, agentIdFromMarkdownPath, agents, aiSearch, bigid, bigint, boolean, buildAssessments, configureJudge, createAgent, createHttpDriver, database, defineEval, defineEvalConfig, defineSchema, defineTool, discoverEvalConfigs, discoverEvalFiles, enumColumn, equals, evalGlyph, executeFromRegistry, findRootEvalConfig, fk, formatEvalDetail, formatEvalHeadline, formatEvalResults, formatResultsJUnit, formatResultsJson, formatSummaryLine, fromSupervisorApi, functionToolToDefinition, id, includes, integer, isFunctionTool, isHostedTool, isJudgeConfigured, isSupervisorTool, isToolkitEntry, jsonb, loadAgentFromFile, loadAgentsFromDir, loadRootEvalConfig, matches, mcpServer, normalizeHost, parseTextToolCalls, readEvalDataset, reportToMlflow, resolveDatabricksAuth, resolveHostedTools, resolveWorkspaceClient, runAgent, runEval, runEvalsInDir, runWithRetries, summarize, supervisorTools, text, timestamp, tool, toolsFromRegistry, userTurns, uuid, varchar };

package/dist/cache/index.js

@@ -1,11 +1,11 @@
 import { ApiError } from "../shared/src/workspace-client/errors.js";
 import { createWorkspaceClient } from "../shared/src/workspace-client/factory.js";
-import { createLogger } from "../logging/logger.js";
+import { getClientOptions } from "../context/client-options.js";
 import { AppKitError } from "../errors/base.js";
 import { ExecutionError } from "../errors/execution.js";
 import { InitializationError } from "../errors/initialization.js";
 import "../errors/index.js";
-import { getClientOptions } from "../context/client-options.js";
+import { createLogger } from "../logging/logger.js";
 import { createLakebasePool } from "../connectors/lakebase/index.js";
 import { TelemetryManager } from "../telemetry/telemetry-manager.js";
 import { SpanStatusCode } from "../telemetry/index.js";

package/dist/cache/storage/persistent.js

@@ -1,7 +1,7 @@
-import { createLogger } from "../../logging/logger.js";
 import { InitializationError } from "../../errors/initialization.js";
 import { ValidationError } from "../../errors/validation.js";
 import "../../errors/index.js";
+import { createLogger } from "../../logging/logger.js";
 import { lakebaseStorageDefaults } from "./defaults.js";
 import { createHash } from "node:crypto";
 

package/dist/connectors/index.js

@@ -1,6 +1,6 @@
 import { createLakebasePoolManager } from "./lakebase/pool-manager.js";
 import { RoutingPool } from "./lakebase/routing-pool.js";
-import { RequestedClaimsPermissionSet, createLakebasePool, generateDatabaseCredential, getLakebaseOrmConfig, getLakebasePgConfig, getUsernameWithApiLookup, getWorkspaceClient } from "./lakebase/index.js";
+import { RequestedClaimsPermissionSet, createLakebasePool, generateDatabaseCredential, getLakebaseOrmConfig, getLakebasePgConfig, getUsernameWithApiLookup, getWorkspaceClient, initializeLakebasePool } from "./lakebase/index.js";
 import { AiSearchConnector } from "./ai-search/client.js";
 import "./ai-search/index.js";
 import { FILES_MAX_READ_SIZE, SAFE_INLINE_CONTENT_TYPES, contentTypeFromPath, isSafeInlineContentType, isTextContentType, validateCustomContentTypes } from "./files/defaults.js";

package/dist/connectors/jobs/client.js

@@ -1,8 +1,8 @@
 import { Context } from "../../shared/src/workspace-client/legacy.js";
-import { createLogger } from "../../logging/logger.js";
 import { AppKitError } from "../../errors/base.js";
 import { ExecutionError } from "../../errors/execution.js";
 import "../../errors/index.js";
+import { createLogger } from "../../logging/logger.js";
 import { TelemetryManager } from "../../telemetry/telemetry-manager.js";
 import { SpanKind, SpanStatusCode } from "../../telemetry/index.js";
 

package/dist/connectors/lakebase/index.d.ts

@@ -1,6 +1,6 @@
 import { LakebasePoolManager, createLakebasePoolManager } from "./pool-manager.js";
 import { LakebasePool } from "./routing-pool.js";
-import { DatabaseCredential, GenerateDatabaseCredentialRequest, LakebasePoolConfig, LakebasePoolConfig as LakebasePoolConfig$1, RequestedClaims, RequestedClaimsPermissionSet, RequestedResource, generateDatabaseCredential, getLakebaseOrmConfig, getLakebasePgConfig, getUsernameWithApiLookup, getWorkspaceClient } from "@databricks/lakebase";
+import { DatabaseCredential, GenerateDatabaseCredentialRequest, LakebasePoolConfig, LakebasePoolConfig as LakebasePoolConfig$1, RequestedClaims, RequestedClaimsPermissionSet, RequestedResource, generateDatabaseCredential, getLakebaseOrmConfig, getLakebasePgConfig, getUsernameWithApiLookup as getUsernameWithApiLookup$1, getWorkspaceClient } from "@databricks/lakebase";
 import { Pool } from "pg";
 
 //#region src/connectors/lakebase/index.d.ts
@@ -13,5 +13,5 @@ import { Pool } from "pg";
  */
 declare function createLakebasePool$1(config?: Partial<LakebasePoolConfig>): Pool;
 //#endregion
-export { type DatabaseCredential, type GenerateDatabaseCredentialRequest, type LakebasePoolConfig$1 as LakebasePoolConfig, type RequestedClaims, RequestedClaimsPermissionSet, type RequestedResource, createLakebasePool$1 as createLakebasePool, generateDatabaseCredential, getLakebaseOrmConfig, getLakebasePgConfig, getUsernameWithApiLookup, getWorkspaceClient };
+export { type DatabaseCredential, type GenerateDatabaseCredentialRequest, type LakebasePoolConfig$1 as LakebasePoolConfig, type RequestedClaims, RequestedClaimsPermissionSet, type RequestedResource, createLakebasePool$1 as createLakebasePool, generateDatabaseCredential, getLakebaseOrmConfig, getLakebasePgConfig, getUsernameWithApiLookup$1 as getUsernameWithApiLookup, getWorkspaceClient };
 //# sourceMappingURL=index.d.ts.map

package/dist/connectors/lakebase/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","names":[],"sources":["../../../src/connectors/lakebase/index.ts"],"mappings":";;;;;;;;;;AAeA;;;iBAAgB,oBAAA,CAAmB,MAAA,GAAS,OAAA,CAAQ,kBAAA,IAAsB,IAAA"}
+{"version":3,"file":"index.d.ts","names":[],"sources":["../../../src/connectors/lakebase/index.ts"],"mappings":";;;;;;;;;;AAoBA;;;iBAAgB,oBAAA,CAAmB,MAAA,GAAS,OAAA,CAAQ,kBAAA,IAAsB,IAAA"}

package/dist/connectors/lakebase/index.js

@@ -1,7 +1,12 @@
+import { createWorkspaceClient } from "../../shared/src/workspace-client/factory.js";
+import { getClientOptions } from "../../context/client-options.js";
+import { ConfigurationError } from "../../errors/configuration.js";
+import "../../errors/index.js";
+import { ServiceContext } from "../../context/service-context.js";
 import { createLogger } from "../../logging/logger.js";
 import { createLakebasePoolManager } from "./pool-manager.js";
 import { RoutingPool } from "./routing-pool.js";
-import { RequestedClaimsPermissionSet, createLakebasePool, generateDatabaseCredential, getLakebaseOrmConfig, getLakebasePgConfig, getUsernameWithApiLookup, getWorkspaceClient } from "@databricks/lakebase";
+import { RequestedClaimsPermissionSet, createLakebasePool, generateDatabaseCredential, getLakebaseOrmConfig, getLakebasePgConfig, getUsernameWithApiLookup, getUsernameWithApiLookup as getUsernameWithApiLookup$1, getWorkspaceClient } from "@databricks/lakebase";
 
 //#region src/connectors/lakebase/index.ts
 /**
@@ -17,7 +22,25 @@ function createLakebasePool$1(config) {
 		...config
 	});
 }
+/**
+* Resolve the startup identity and create a pool through the existing Lakebase
+* connector. Explicit clients win over the app's service client; request/OBO
+* context is never selected implicitly. The pool connects lazily on first use.
+*
+* Keep createLakebasePool synchronous for existing callers and OBO pool caches.
+*/
+async function initializeLakebasePool(config = {}) {
+	const resolved = { ...config };
+	const needsClient = config.password === void 0 || !(config.user || process.env.PGUSER || process.env.DATABRICKS_CLIENT_ID);
+	if (!resolved.workspaceClient && needsClient) resolved.workspaceClient = (ServiceContext.isInitialized() ? ServiceContext.get().client : createWorkspaceClient({ clientOptions: getClientOptions() })).toLegacyWorkspaceClient();
+	const user = await getUsernameWithApiLookup(resolved);
+	if (!user) throw ConfigurationError.invalidConnection("Lakebase", "Could not determine the PostgreSQL user from the current Databricks credentials. Check your authentication or set PGUSER explicitly.");
+	return createLakebasePool$1({
+		...resolved,
+		user
+	});
+}
 
 //#endregion
-export { RequestedClaimsPermissionSet, createLakebasePool$1 as createLakebasePool, generateDatabaseCredential, getLakebaseOrmConfig, getLakebasePgConfig, getUsernameWithApiLookup, getWorkspaceClient };
+export { RequestedClaimsPermissionSet, createLakebasePool$1 as createLakebasePool, generateDatabaseCredential, getLakebaseOrmConfig, getLakebasePgConfig, getUsernameWithApiLookup$1 as getUsernameWithApiLookup, getWorkspaceClient, initializeLakebasePool };
 //# sourceMappingURL=index.js.map

package/dist/connectors/lakebase/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":["createLakebasePool","createLakebasePoolBase"],"sources":["../../../src/connectors/lakebase/index.ts"],"sourcesContent":["import {\n  createLakebasePool as createLakebasePoolBase,\n  type LakebasePoolConfig,\n} from \"@databricks/lakebase\";\nimport type { Pool } from \"pg\";\n\nimport { createLogger } from \"../../logging/logger\";\n\n/**\n * Create a Lakebase pool with appkit's logger integration.\n * Telemetry automatically uses appkit's OpenTelemetry configuration via global registry.\n *\n * @param config - Lakebase pool configuration\n * @returns PostgreSQL pool with appkit integration\n */\nexport function createLakebasePool(config?: Partial<LakebasePoolConfig>): Pool {\n  return createLakebasePoolBase({\n    logger: createLogger(\"connectors:lakebase\"),\n    ...config,\n  });\n}\n\n// Re-export everything else from lakebase\nexport {\n  type DatabaseCredential,\n  type GenerateDatabaseCredentialRequest,\n  generateDatabaseCredential,\n  getLakebaseOrmConfig,\n  getLakebasePgConfig,\n  getUsernameWithApiLookup,\n  getWorkspaceClient,\n  type LakebasePoolConfig,\n  type RequestedClaims,\n  RequestedClaimsPermissionSet,\n  type RequestedResource,\n} from \"@databricks/lakebase\";\n\nexport {\n  createLakebasePoolManager,\n  type LakebasePoolManager,\n} from \"./pool-manager\";\n\nexport { type LakebasePool, RoutingPool } from \"./routing-pool\";\n"],"mappings":";;;;;;;;;;;;;AAeA,SAAgBA,qBAAmB,QAA4C;AAC7E,QAAOC,mBAAuB;EAC5B,QAAQ,aAAa,sBAAsB;EAC3C,GAAG;EACJ,CAAC"}
+{"version":3,"file":"index.js","names":["createLakebasePool","createLakebasePoolBase"],"sources":["../../../src/connectors/lakebase/index.ts"],"sourcesContent":["import {\n  createLakebasePool as createLakebasePoolBase,\n  getUsernameWithApiLookup,\n  type LakebasePoolConfig,\n} from \"@databricks/lakebase\";\nimport type { Pool } from \"pg\";\n\nimport { getClientOptions } from \"../../context/client-options\";\nimport { ServiceContext } from \"../../context/service-context\";\nimport { ConfigurationError } from \"../../errors\";\nimport { createLogger } from \"../../logging/logger\";\nimport { createWorkspaceClient } from \"../../workspace-client\";\n\n/**\n * Create a Lakebase pool with appkit's logger integration.\n * Telemetry automatically uses appkit's OpenTelemetry configuration via global registry.\n *\n * @param config - Lakebase pool configuration\n * @returns PostgreSQL pool with appkit integration\n */\nexport function createLakebasePool(config?: Partial<LakebasePoolConfig>): Pool {\n  return createLakebasePoolBase({\n    logger: createLogger(\"connectors:lakebase\"),\n    ...config,\n  });\n}\n\n/**\n * Resolve the startup identity and create a pool through the existing Lakebase\n * connector. Explicit clients win over the app's service client; request/OBO\n * context is never selected implicitly. The pool connects lazily on first use.\n *\n * Keep createLakebasePool synchronous for existing callers and OBO pool caches.\n */\nexport async function initializeLakebasePool(\n  config: Partial<LakebasePoolConfig> = {},\n): Promise<Pool> {\n  const resolved = { ...config };\n  // Native password auth with an explicit/environment user needs no Databricks\n  // client. Otherwise use one client for both identity lookup and token refresh.\n  const needsClient =\n    config.password === undefined ||\n    !(config.user || process.env.PGUSER || process.env.DATABRICKS_CLIENT_ID);\n  if (!resolved.workspaceClient && needsClient) {\n    const client = ServiceContext.isInitialized()\n      ? ServiceContext.get().client\n      : createWorkspaceClient({ clientOptions: getClientOptions() });\n    resolved.workspaceClient = client.toLegacyWorkspaceClient();\n  }\n  const user = await getUsernameWithApiLookup(resolved);\n  if (!user) {\n    throw ConfigurationError.invalidConnection(\n      \"Lakebase\",\n      \"Could not determine the PostgreSQL user from the current Databricks credentials. Check your authentication or set PGUSER explicitly.\",\n    );\n  }\n  return createLakebasePool({ ...resolved, user });\n}\n\n// Re-export everything else from lakebase\nexport {\n  type DatabaseCredential,\n  type GenerateDatabaseCredentialRequest,\n  generateDatabaseCredential,\n  getLakebaseOrmConfig,\n  getLakebasePgConfig,\n  getUsernameWithApiLookup,\n  getWorkspaceClient,\n  type LakebasePoolConfig,\n  type RequestedClaims,\n  RequestedClaimsPermissionSet,\n  type RequestedResource,\n} from \"@databricks/lakebase\";\n\nexport {\n  createLakebasePoolManager,\n  type LakebasePoolManager,\n} from \"./pool-manager\";\n\nexport { type LakebasePool, RoutingPool } from \"./routing-pool\";\n"],"mappings":";;;;;;;;;;;;;;;;;;AAoBA,SAAgBA,qBAAmB,QAA4C;AAC7E,QAAOC,mBAAuB;EAC5B,QAAQ,aAAa,sBAAsB;EAC3C,GAAG;EACJ,CAAC;;;;;;;;;AAUJ,eAAsB,uBACpB,SAAsC,EAAE,EACzB;CACf,MAAM,WAAW,EAAE,GAAG,QAAQ;CAG9B,MAAM,cACJ,OAAO,aAAa,UACpB,EAAE,OAAO,QAAQ,QAAQ,IAAI,UAAU,QAAQ,IAAI;AACrD,KAAI,CAAC,SAAS,mBAAmB,YAI/B,UAAS,mBAHM,eAAe,eAAe,GACzC,eAAe,KAAK,CAAC,SACrB,sBAAsB,EAAE,eAAe,kBAAkB,EAAE,CAAC,EAC9B,yBAAyB;CAE7D,MAAM,OAAO,MAAM,yBAAyB,SAAS;AACrD,KAAI,CAAC,KACH,OAAM,mBAAmB,kBACvB,YACA,uIACD;AAEH,QAAOD,qBAAmB;EAAE,GAAG;EAAU;EAAM,CAAC"}

package/dist/connectors/mcp/client.js

@@ -1,5 +1,5 @@
-import { createLogger } from "../../logging/logger.js";
 import { APPKIT_USER_AGENT } from "../../context/client-options.js";
+import { createLogger } from "../../logging/logger.js";
 import { assertResolvedHostSafe, checkMcpUrl } from "./host-policy.js";
 
 //#region src/connectors/mcp/client.ts

package/dist/connectors/sql-warehouse/client.js

@@ -1,11 +1,11 @@
 import { Context } from "../../shared/src/workspace-client/legacy.js";
-import { createLogger } from "../../logging/logger.js";
 import { AppKitError } from "../../errors/base.js";
 import { ConfigurationError } from "../../errors/configuration.js";
 import { ConnectionError } from "../../errors/connection.js";
 import { ExecutionError } from "../../errors/execution.js";
 import { ValidationError } from "../../errors/validation.js";
 import "../../errors/index.js";
+import { createLogger } from "../../logging/logger.js";
 import { TelemetryManager } from "../../telemetry/telemetry-manager.js";
 import { SpanKind, SpanStatusCode } from "../../telemetry/index.js";
 import { ArrowStreamProcessor } from "../../stream/arrow-stream-processor.js";

package/dist/context/service-context.js

@@ -1,10 +1,10 @@
 import { ConfigError } from "../shared/src/workspace-client/legacy.js";
 import { createWorkspaceClient } from "../shared/src/workspace-client/factory.js";
+import { getClientOptions } from "./client-options.js";
 import { AuthenticationError } from "../errors/authentication.js";
 import { ConfigurationError } from "../errors/configuration.js";
 import { InitializationError } from "../errors/initialization.js";
 import "../errors/index.js";
-import { getClientOptions } from "./client-options.js";
 import { createHash } from "node:crypto";
 
 //#region src/context/service-context.ts

package/dist/core/appkit.js

@@ -1,6 +1,6 @@
-import { createLogger } from "../logging/logger.js";
 import { version } from "../appkit/package.js";
 import { ServiceContext } from "../context/service-context.js";
+import { createLogger } from "../logging/logger.js";
 import { TelemetryManager } from "../telemetry/telemetry-manager.js";
 import "../telemetry/index.js";
 import { CacheManager } from "../cache/index.js";

package/dist/database/errors.js

@@ -1,6 +1,6 @@
-import { createLogger } from "../logging/logger.js";
 import { AppKitError } from "../errors/base.js";
 import "../errors/index.js";
+import { createLogger } from "../logging/logger.js";
 
 //#region src/database/errors.ts
 const logger = createLogger("database");

package/dist/index.js

@@ -1,7 +1,6 @@
 import { isSQLTypeMarker, sql } from "./shared/src/sql/helpers.js";
 import { ApiError } from "./shared/src/workspace-client/errors.js";
 import { createWorkspaceClient } from "./shared/src/workspace-client/factory.js";
-import { createLakebasePoolManager } from "./connectors/lakebase/pool-manager.js";
 import { AppKitError } from "./errors/base.js";
 import { AuthenticationError } from "./errors/authentication.js";
 import { ConfigurationError } from "./errors/configuration.js";
@@ -13,6 +12,7 @@ import { ServerError } from "./errors/server.js";
 import { TunnelError } from "./errors/tunnel.js";
 import { ValidationError } from "./errors/validation.js";
 import "./errors/index.js";
+import { createLakebasePoolManager } from "./connectors/lakebase/pool-manager.js";
 import { getExecutionContext } from "./context/execution-context.js";
 import { RequestedClaimsPermissionSet, createLakebasePool, generateDatabaseCredential, getLakebaseOrmConfig, getLakebasePgConfig, getUsernameWithApiLookup, getWorkspaceClient } from "./connectors/lakebase/index.js";
 import { SeverityNumber, SpanStatusCode } from "./telemetry/index.js";

package/dist/plugin/dev-reader.js

@@ -1,6 +1,6 @@
-import { createLogger } from "../logging/logger.js";
 import { TunnelError } from "../errors/tunnel.js";
 import "../errors/index.js";
+import { createLogger } from "../logging/logger.js";
 import { isRemoteTunnelAllowedByEnv } from "../plugins/server/remote-tunnel/gate.js";
 import { randomUUID } from "node:crypto";
 

package/dist/plugin/interceptors/retry.js

@@ -1,5 +1,5 @@
-import { createLogger } from "../../logging/logger.js";
 import { AppKitError } from "../../errors/base.js";
+import { createLogger } from "../../logging/logger.js";
 
 //#region src/plugin/interceptors/retry.ts
 const logger = createLogger("interceptors:retry");

package/dist/plugin/plugin.js

@@ -1,9 +1,9 @@
 import { camelToKebab } from "../shared/src/naming.js";
-import { createLogger } from "../logging/logger.js";
 import { AppKitError } from "../errors/base.js";
 import { AuthenticationError } from "../errors/authentication.js";
 import "../errors/index.js";
 import { ServiceContext } from "../context/service-context.js";
+import { createLogger } from "../logging/logger.js";
 import { getCurrentUserId, runInUserContext } from "../context/execution-context.js";
 import { normalizeTelemetryOptions } from "../telemetry/config.js";
 import { TelemetryManager } from "../telemetry/telemetry-manager.js";

package/dist/plugins/analytics/analytics.js

@@ -1,8 +1,8 @@
 import { makeResultMessage } from "../../shared/src/sse/analytics.js";
-import { createLogger } from "../../logging/logger.js";
 import { AppKitError } from "../../errors/base.js";
 import { ExecutionError } from "../../errors/execution.js";
 import "../../errors/index.js";
+import { createLogger } from "../../logging/logger.js";
 import { getWarehouseId, getWorkspaceClient } from "../../context/execution-context.js";
 import "../../context/index.js";
 import { Plugin } from "../../plugin/plugin.js";

package/dist/plugins/analytics/result-delivery.js

@@ -1,6 +1,6 @@
-import { createLogger } from "../../logging/logger.js";
 import { ExecutionError } from "../../errors/execution.js";
 import "../../errors/index.js";
+import { createLogger } from "../../logging/logger.js";
 import { Type, tableFromIPC } from "apache-arrow";
 
 //#region src/plugins/analytics/result-delivery.ts

package/dist/plugins/beta-exports.generated.d.ts

@@ -1,6 +1,4 @@
 import { agents } from "./agents/agents.js";
 import "./agents/index.js";
 import { aiSearch } from "./ai-search/ai-search.js";
-import "./ai-search/index.js";
-import { database } from "./database/database.js";
-import "./database/index.js";
+import "./ai-search/index.js";

package/dist/plugins/beta-exports.generated.js

@@ -2,7 +2,5 @@ import { agents } from "./agents/agents.js";
 import "./agents/index.js";
 import { aiSearch } from "./ai-search/ai-search.js";
 import "./ai-search/index.js";
-import { database } from "./database/database.js";
-import "./database/index.js";
 
 export {  };

package/dist/plugins/database/config.js

@@ -0,0 +1,14 @@
+import { databaseSetupFailed } from "../../database/errors.js";
+
+//#region src/plugins/database/config.ts
+/** Reject invalid arguments instead of turning them into the default API. */
+function assertDatabaseConfig(config) {
+	if (typeof config !== "object" || config === null || Array.isArray(config)) throw databaseSetupFailed("Expected a database configuration object or no argument.");
+	const prototype = Object.getPrototypeOf(config);
+	if (prototype !== Object.prototype && prototype !== null) throw databaseSetupFailed("Expected a plain database configuration object.");
+	if ("crudRoutes" in config) throw databaseSetupFailed("\"crudRoutes\" was renamed to \"api\". Use api: false to disable generated routes or api: { writes: false } for reads only.");
+}
+
+//#endregion
+export { assertDatabaseConfig };
+//# sourceMappingURL=config.js.map

package/dist/plugins/database/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","names":[],"sources":["../../../src/plugins/database/config.ts"],"sourcesContent":["import { databaseSetupFailed } from \"../../database/errors\";\n\n/** Reject invalid arguments instead of turning them into the default API. */\nexport function assertDatabaseConfig(config: unknown): void {\n  if (typeof config !== \"object\" || config === null || Array.isArray(config)) {\n    throw databaseSetupFailed(\n      \"Expected a database configuration object or no argument.\",\n    );\n  }\n  const prototype = Object.getPrototypeOf(config);\n  if (prototype !== Object.prototype && prototype !== null) {\n    throw databaseSetupFailed(\n      \"Expected a plain database configuration object.\",\n    );\n  }\n  // Do not silently turn a previous opt-out into the default full API.\n  if (\"crudRoutes\" in config) {\n    throw databaseSetupFailed(\n      '\"crudRoutes\" was renamed to \"api\". Use api: false to disable generated routes or api: { writes: false } for reads only.',\n    );\n  }\n}\n"],"mappings":";;;;AAGA,SAAgB,qBAAqB,QAAuB;AAC1D,KAAI,OAAO,WAAW,YAAY,WAAW,QAAQ,MAAM,QAAQ,OAAO,CACxE,OAAM,oBACJ,2DACD;CAEH,MAAM,YAAY,OAAO,eAAe,OAAO;AAC/C,KAAI,cAAc,OAAO,aAAa,cAAc,KAClD,OAAM,oBACJ,kDACD;AAGH,KAAI,gBAAgB,OAClB,OAAM,oBACJ,8HACD"}

package/dist/plugins/database/database.d.ts

@@ -6,12 +6,12 @@ import "../../plugin/index.js";
 import { PluginManifest } from "../../registry/types.js";
 import "../../registry/index.js";
 import { DatabaseExports } from "./entity-types.js";
-import { IDatabaseConfig } from "./types.js";
+import { DefaultDatabaseSchema, IDatabaseConfig } from "./types.js";
 import express from "express";
 
 //#region src/plugins/database/database.d.ts
 /** Schema-driven database plugin */
-declare class DatabasePlugin<TSchema extends Schema> extends Plugin<IDatabaseConfig<TSchema>> {
+declare class DatabasePlugin<TSchema extends Schema = DefaultDatabaseSchema> extends Plugin<IDatabaseConfig<TSchema>> {
   /** Plugin metadata and required PostgreSQL resource. */
   static manifest: PluginManifest<"database">;
   protected config: IDatabaseConfig<TSchema>;
@@ -20,7 +20,8 @@ declare class DatabasePlugin<TSchema extends Schema> extends Plugin<IDatabaseCon
   private draining;
   private shutdownPromise;
   private exposure;
-  constructor(config: IDatabaseConfig<TSchema>);
+  private resolvedSchema;
+  constructor(config?: IDatabaseConfig<TSchema>);
   /** Build and verify one candidate state before publishing its exports. */
   setup(): Promise<void>;
   /** Register generated CRUD, subject to the configured table and write restrictions. */
@@ -34,12 +35,28 @@ declare class DatabasePlugin<TSchema extends Schema> extends Plugin<IDatabaseCon
   /** Trace one generated route with allowlisted, low-cardinality attributes. */
   private runRouteSpan;
 }
-/** Create a typed database plugin registration for a finalized schema. */
-declare function database<TSchema extends Schema>(config: IDatabaseConfig<TSchema>): {
+type DatabaseRegistration<TSchema extends Schema> = {
   plugin: PluginConstructor<BasePluginConfig, DatabasePlugin<TSchema>>;
   config: IDatabaseConfig<TSchema>;
   name: "database";
 };
+/**
+ * Create the database plugin. Omit configuration to load
+ * `config/database/schema.ts`, or supply a typed schema override.
+ */
+declare function database<TSchema extends Schema>(config: IDatabaseConfig<TSchema> & {
+  readonly schema: TSchema;
+}): DatabaseRegistration<TSchema> & {
+  config: IDatabaseConfig<TSchema> & {
+    readonly schema: TSchema;
+  };
+};
+/**
+ * Register the database plugin with opinionated defaults and full HTTP CRUD.
+ * By default, setup loads the named `schema` export from the application's
+ * `config/database/schema.ts`. Registration itself performs no file or database I/O.
+ */
+declare function database<TSchema extends Schema = DefaultDatabaseSchema>(config?: IDatabaseConfig<TSchema>): DatabaseRegistration<TSchema>;
 //#endregion
 export { database };
 //# sourceMappingURL=database.d.ts.map

package/dist/plugins/database/database.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"database.d.ts","names":[],"sources":["../../../src/plugins/database/database.ts"],"mappings":";;;;;;;;;;;;;cA6Ba,cAAA,iBAA+B,MAAA,UAAgB,MAAA,CAC1D,eAAA,CAAgB,OAAA;;SAGT,QAAA,EAAuB,cAAA;EAAA,UACZ,MAAA,EAAQ,eAAA,CAAgB,OAAA;EAAA,QAClC,KAAA;EAAA,QACA,YAAA;EAAA,QACA,QAAA;EAAA,QACA,eAAA;EAAA,QACA,QAAA;cAEI,MAAA,EAAQ,eAAA,CAAgB,OAAA;EARN;EAwBxB,KAAA,CAAA,GAAS,OAAA;EAvBW;EA0D1B,YAAA,CAAa,MAAA,EAAQ,OAAA,CAAQ,MAAA;EAnDT;EAAA,QAqHZ,KAAA;EAlEa;EAuErB,OAAA,CAAA,GAOO,eAAA;EAIW;EAAZ,QAAA,CAAA,GAAY,OAAA;EAjJ8C;EAAA,QAqKxD,YAAA;AAAA;;iBA+BM,QAAA,iBAAyB,MAAA,CAAA,CACvC,MAAA,EAAQ,eAAA,CAAgB,OAAA;UAGe,iBAAA,CACnC,gBAAA,EACA,cAAA,CAAe,OAAA"}
+{"version":3,"file":"database.d.ts","names":[],"sources":["../../../src/plugins/database/database.ts"],"mappings":";;;;;;;;;;;;;cAoCa,cAAA,iBACK,MAAA,GAAS,qBAAA,UACjB,MAAA,CAAO,eAAA,CAAgB,OAAA;;SAExB,QAAA,EAAuB,cAAA;EAAA,UACZ,MAAA,EAAQ,eAAA,CAAgB,OAAA;EAAA,QAClC,KAAA;EAAA,QACA,YAAA;EAAA,QACA,QAAA;EAAA,QACA,eAAA;EAAA,QACA,QAAA;EAAA,QACA,cAAA;cAEI,MAAA,GAAQ,eAAA,CAAgB,OAAA;EATN;EAoBxB,KAAA,CAAA,GAAS,OAAA;EAnBW;EAoE1B,YAAA,CAAa,MAAA,EAAQ,OAAA,CAAQ,MAAA;EA5DT;EAAA,QA6HZ,KAAA;EAjEa;EAsErB,OAAA,CAAA,GAOO,eAAA;EAIW;EAAZ,QAAA,CAAA,GAAY,OAAA;EAxJJ;EAAA,QA4KN,YAAA;AAAA;AAAA,KA8BL,oBAAA,iBAAqC,MAAA;EACxC,MAAA,EAAQ,iBAAA,CAAkB,gBAAA,EAAkB,cAAA,CAAe,OAAA;EAC3D,MAAA,EAAQ,eAAA,CAAgB,OAAA;EACxB,IAAA;AAAA;;;;;iBAOc,QAAA,iBAAyB,MAAA,CAAA,CACvC,MAAA,EAAQ,eAAA,CAAgB,OAAA;EAAA,SAAsB,MAAA,EAAQ,OAAA;AAAA,IACrD,oBAAA,CAAqB,OAAA;EACtB,MAAA,EAAQ,eAAA,CAAgB,OAAA;IAAA,SAAsB,MAAA,EAAQ,OAAA;EAAA;AAAA;;;;;;iBAOxC,QAAA,iBAAyB,MAAA,GAAS,qBAAA,CAAA,CAChD,MAAA,GAAS,eAAA,CAAgB,OAAA,IACxB,oBAAA,CAAqB,OAAA"}

package/dist/plugins/database/database.js

@@ -1,11 +1,14 @@
 import { Plugin } from "../../plugin/plugin.js";
 import "../../plugin/index.js";
+import { assertFinalizedSchema } from "../../database/schema-builder/define-schema.js";
 import { DatabasePluginError, databaseSetupFailed } from "../../database/errors.js";
+import { assertDatabaseConfig } from "./config.js";
 import { compileCrudTables } from "./crud/contract.js";
 import { resolveCrudExposure } from "./crud/exposure.js";
 import { routeOutcome } from "./crud/response.js";
 import { createCreateHandler, createDeleteHandler, createDetailHandler, createListHandler, createUpdateHandler } from "./crud/routes.js";
 import { createDatabaseState } from "./lifecycle.js";
+import { loadDefaultDatabaseSchema } from "./load-schema.js";
 import manifest_default from "./manifest.js";
 
 //#region src/plugins/database/database.ts
@@ -21,9 +24,10 @@ var DatabasePlugin = class extends Plugin {
 		tables: [],
 		writes: /* @__PURE__ */ new Map()
 	};
-	constructor(config) {
+	resolvedSchema = null;
+	constructor(config = {}) {
+		assertDatabaseConfig(config);
 		super({ schema: config.schema });
-		if ("crudRoutes" in config) throw databaseSetupFailed("\"crudRoutes\" was renamed to \"api\". Use api: false to disable generated routes or api: { writes: false } for reads only.");
 		this.config = {
 			schema: config.schema,
 			api: config.api,
@@ -34,14 +38,23 @@ var DatabasePlugin = class extends Plugin {
 	async setup() {
 		if (this.draining || this.state) throw databaseSetupFailed();
 		if (!this.setupPromise) this.setupPromise = (async () => {
-			this.exposure = resolveCrudExposure(this.config.api, Object.keys(this.config.schema.$tables));
-			for (const name of Object.keys(this.hooks() ?? {})) if (!Object.hasOwn(this.config.schema.$tables, name)) throw databaseSetupFailed(`hooks names undeclared table ${JSON.stringify(name)}. Use a table declared in schema.`);
-			const candidate = await createDatabaseState(this.config.schema, (operation, options) => this.execute(operation, options), this.hooks());
+			const schema = this.config.schema === void 0 ? await loadDefaultDatabaseSchema() : this.config.schema;
+			try {
+				assertFinalizedSchema(schema);
+			} catch {
+				throw databaseSetupFailed("schema must be a finalized AppKit schema created with defineSchema().");
+			}
+			if (this.draining) throw databaseSetupFailed();
+			const exposure = resolveCrudExposure(this.config.api, Object.keys(schema.$tables));
+			for (const name of Object.keys(this.hooks() ?? {})) if (!Object.hasOwn(schema.$tables, name)) throw databaseSetupFailed(`hooks names undeclared table ${JSON.stringify(name)}. Use a table declared in schema.`);
+			const candidate = await createDatabaseState(schema, (operation, options) => this.execute(operation, options), this.hooks());
 			if (this.draining) {
 				candidate.deactivate();
 				await candidate.pool.end().catch(() => void 0);
 				throw databaseSetupFailed();
 			}
+			this.resolvedSchema = schema;
+			this.exposure = exposure;
 			this.state = candidate;
 		})();
 		return this.setupPromise;
@@ -49,7 +62,9 @@ var DatabasePlugin = class extends Plugin {
 	/** Register generated CRUD, subject to the configured table and write restrictions. */
 	injectRoutes(router) {
 		if (this.exposure.tables.length === 0) return;
-		const tables = compileCrudTables(Object.fromEntries(this.exposure.tables.map((name) => [name, this.config.schema.$tables[name]])));
+		const schema = this.resolvedSchema;
+		if (!schema) throw databaseSetupFailed();
+		const tables = compileCrudTables(Object.fromEntries(this.exposure.tables.map((name) => [name, schema.$tables[name]])));
 		const hooks = this.hooks();
 		const entities = () => this.exports();
 		for (const table of tables.values()) {
@@ -138,8 +153,8 @@ var DatabasePlugin = class extends Plugin {
 		});
 	}
 };
-/** Create a typed database plugin registration for a finalized schema. */
-function database(config) {
+function database(config = {}) {
+	assertDatabaseConfig(config);
 	return {
 		plugin: DatabasePlugin,
 		config,

package/dist/plugins/database/database.js.map

@@ -1 +1 @@
-{"version":3,"file":"database.js","names":["manifest"],"sources":["../../../src/plugins/database/database.ts"],"sourcesContent":["import type express from \"express\";\nimport type { BasePluginConfig, PluginConstructor } from \"shared\";\n\nimport {\n  DatabasePluginError,\n  databaseSetupFailed,\n} from \"../../database/errors\";\nimport type { Schema } from \"../../database/schema-builder\";\nimport { Plugin } from \"../../plugin\";\nimport type { PluginManifest } from \"../../registry\";\nimport { compileCrudTables } from \"./crud/contract\";\nimport { type CrudExposure, resolveCrudExposure } from \"./crud/exposure\";\nimport { routeOutcome } from \"./crud/response\";\nimport {\n  type CrudEntity,\n  type CrudOperation,\n  type CrudRouteDeps,\n  createCreateHandler,\n  createDeleteHandler,\n  createDetailHandler,\n  createListHandler,\n  createUpdateHandler,\n} from \"./crud/routes\";\nimport type { DatabaseExports } from \"./entity-types\";\nimport { createDatabaseState, type DatabaseState } from \"./lifecycle\";\nimport manifest from \"./manifest.json\";\nimport type { DatabaseHooks, IDatabaseConfig } from \"./types\";\n\n/** Schema-driven database plugin */\nexport class DatabasePlugin<TSchema extends Schema> extends Plugin<\n  IDatabaseConfig<TSchema>\n> {\n  /** Plugin metadata and required PostgreSQL resource. */\n  static manifest = manifest as PluginManifest<\"database\">;\n  declare protected config: IDatabaseConfig<TSchema>;\n  private state: DatabaseState | null = null;\n  private setupPromise: Promise<void> | null = null;\n  private draining = false;\n  private shutdownPromise: Promise<void> | null = null;\n  private exposure: CrudExposure = { tables: [], writes: new Map() };\n\n  constructor(config: IDatabaseConfig<TSchema>) {\n    super({ schema: config.schema });\n    // Do not silently turn a previous opt-out into the default full API.\n    if (\"crudRoutes\" in config) {\n      throw databaseSetupFailed(\n        '\"crudRoutes\" was renamed to \"api\". Use api: false to disable generated routes or api: { writes: false } for reads only.',\n      );\n    }\n    this.config = {\n      schema: config.schema,\n      api: config.api,\n      hooks: config.hooks,\n    };\n  }\n\n  /** Build and verify one candidate state before publishing its exports. */\n  async setup(): Promise<void> {\n    if (this.draining || this.state) throw databaseSetupFailed();\n    if (!this.setupPromise) {\n      const attempt = (async () => {\n        this.exposure = resolveCrudExposure(\n          this.config.api,\n          Object.keys(this.config.schema.$tables),\n        );\n        // A hook key naming no declared table would silently never run.\n        for (const name of Object.keys(this.hooks() ?? {})) {\n          if (!Object.hasOwn(this.config.schema.$tables, name)) {\n            throw databaseSetupFailed(\n              `hooks names undeclared table ${JSON.stringify(name)}. Use a table declared in schema.`,\n            );\n          }\n        }\n        const candidate = await createDatabaseState(\n          this.config.schema,\n          (operation, options) => this.execute(operation, options),\n          this.hooks(),\n        );\n        if (this.draining) {\n          // Setup may finish while shutdown is waiting; never publish that state.\n          candidate.deactivate();\n          await candidate.pool.end().catch(() => undefined);\n          throw databaseSetupFailed();\n        }\n        this.state = candidate;\n      })();\n      this.setupPromise = attempt;\n    }\n    return this.setupPromise;\n  }\n\n  /** Register generated CRUD, subject to the configured table and write restrictions. */\n  injectRoutes(router: express.Router): void {\n    if (this.exposure.tables.length === 0) return;\n    const tables = compileCrudTables(\n      Object.fromEntries(\n        this.exposure.tables.map((name) => [\n          name,\n          this.config.schema.$tables[name],\n        ]),\n      ),\n    );\n    const hooks = this.hooks();\n    // Every exposed name is a declared table, so its export is an entity client.\n    const entities = () =>\n      this.exports() as unknown as Record<string, CrudEntity>;\n\n    for (const table of tables.values()) {\n      const deps: CrudRouteDeps = {\n        table,\n        entity: () => entities()[table.name],\n        serialize: hooks?.[table.name]?.serialize,\n        runRouteSpan: (operation, route, run) =>\n          this.runRouteSpan(table.name, operation, route, run),\n      };\n      this.route(router, {\n        name: `${table.name}.list`,\n        method: \"get\",\n        path: `/${table.name}`,\n        handler: createListHandler(deps),\n      });\n      const writes = this.exposure.writes.get(table.name);\n      if (writes?.has(\"create\")) {\n        this.route(router, {\n          name: `${table.name}.create`,\n          method: \"post\",\n          path: `/${table.name}`,\n          handler: createCreateHandler(deps),\n        });\n      }\n      // Addressing one row needs a public key.\n      if (!table.primaryKey) continue;\n      this.route(router, {\n        name: `${table.name}.detail`,\n        method: \"get\",\n        path: `/${table.name}/:id`,\n        handler: createDetailHandler(deps),\n      });\n      if (writes?.has(\"update\")) {\n        this.route(router, {\n          name: `${table.name}.update`,\n          method: \"patch\",\n          path: `/${table.name}/:id`,\n          handler: createUpdateHandler(deps),\n        });\n      }\n      if (writes?.has(\"delete\")) {\n        this.route(router, {\n          name: `${table.name}.delete`,\n          method: \"delete\",\n          path: `/${table.name}/:id`,\n          handler: createDeleteHandler(deps),\n        });\n      }\n    }\n  }\n\n  /** Typed hook keys are schema table names, which routing addresses at runtime. */\n  private hooks(): DatabaseHooks | undefined {\n    return this.config.hooks as DatabaseHooks | undefined;\n  }\n\n  /** Return the typed database API only while the plugin is active. */\n  exports() {\n    if (!this.state || this.draining)\n      throw new DatabasePluginError(\"INTERNAL\", \"read\");\n    // AppKit binds exported functions onto this object on every access.\n    return Object.assign(\n      Object.create(null),\n      this.state.exports,\n    ) as DatabaseExports;\n  }\n\n  /** Stop new work, wait for setup, and close the owned pool exactly once. */\n  async shutdown(): Promise<void> {\n    if (this.shutdownPromise) return this.shutdownPromise;\n    this.draining = true;\n    this.shutdownPromise = (async () => {\n      await this.setupPromise?.catch(() => undefined);\n      const state = this.state;\n      state?.deactivate();\n      this.state = null;\n      if (state) {\n        try {\n          await state.pool.end();\n        } catch {\n          throw new DatabasePluginError(\"INTERNAL\", \"shutdown\");\n        }\n      }\n    })();\n    return this.shutdownPromise;\n  }\n\n  /** Trace one generated route with allowlisted, low-cardinality attributes. */\n  private runRouteSpan(\n    table: string,\n    operation: CrudOperation,\n    route: string,\n    run: () => Promise<void>,\n  ): Promise<void> {\n    return this.telemetry.startActiveSpan(\n      \"database.crud.route\",\n      {\n        attributes: {\n          table_name: table,\n          operation,\n          \"http.route\": `/api/${this.name}${route}`,\n        },\n      },\n      async (span) => {\n        try {\n          await run();\n          span.setAttribute(\"outcome\", \"success\");\n        } catch (error) {\n          span.setAttribute(\"outcome\", routeOutcome(error));\n          throw error;\n        } finally {\n          span.end();\n        }\n      },\n    );\n  }\n}\n\n/** Create a typed database plugin registration for a finalized schema. */\nexport function database<TSchema extends Schema>(\n  config: IDatabaseConfig<TSchema>,\n) {\n  return {\n    plugin: DatabasePlugin as unknown as PluginConstructor<\n      BasePluginConfig,\n      DatabasePlugin<TSchema>\n    >,\n    config,\n    name: \"database\" as const,\n  };\n}\n"],"mappings":";;;;;;;;;;;;AA6BA,IAAa,iBAAb,cAA4D,OAE1D;;CAEA,OAAO,WAAWA;CAElB,AAAQ,QAA8B;CACtC,AAAQ,eAAqC;CAC7C,AAAQ,WAAW;CACnB,AAAQ,kBAAwC;CAChD,AAAQ,WAAyB;EAAE,QAAQ,EAAE;EAAE,wBAAQ,IAAI,KAAK;EAAE;CAElE,YAAY,QAAkC;AAC5C,QAAM,EAAE,QAAQ,OAAO,QAAQ,CAAC;AAEhC,MAAI,gBAAgB,OAClB,OAAM,oBACJ,8HACD;AAEH,OAAK,SAAS;GACZ,QAAQ,OAAO;GACf,KAAK,OAAO;GACZ,OAAO,OAAO;GACf;;;CAIH,MAAM,QAAuB;AAC3B,MAAI,KAAK,YAAY,KAAK,MAAO,OAAM,qBAAqB;AAC5D,MAAI,CAAC,KAAK,aA2BR,MAAK,gBA1BY,YAAY;AAC3B,QAAK,WAAW,oBACd,KAAK,OAAO,KACZ,OAAO,KAAK,KAAK,OAAO,OAAO,QAAQ,CACxC;AAED,QAAK,MAAM,QAAQ,OAAO,KAAK,KAAK,OAAO,IAAI,EAAE,CAAC,CAChD,KAAI,CAAC,OAAO,OAAO,KAAK,OAAO,OAAO,SAAS,KAAK,CAClD,OAAM,oBACJ,gCAAgC,KAAK,UAAU,KAAK,CAAC,mCACtD;GAGL,MAAM,YAAY,MAAM,oBACtB,KAAK,OAAO,SACX,WAAW,YAAY,KAAK,QAAQ,WAAW,QAAQ,EACxD,KAAK,OAAO,CACb;AACD,OAAI,KAAK,UAAU;AAEjB,cAAU,YAAY;AACtB,UAAM,UAAU,KAAK,KAAK,CAAC,YAAY,OAAU;AACjD,UAAM,qBAAqB;;AAE7B,QAAK,QAAQ;MACX;AAGN,SAAO,KAAK;;;CAId,aAAa,QAA8B;AACzC,MAAI,KAAK,SAAS,OAAO,WAAW,EAAG;EACvC,MAAM,SAAS,kBACb,OAAO,YACL,KAAK,SAAS,OAAO,KAAK,SAAS,CACjC,MACA,KAAK,OAAO,OAAO,QAAQ,MAC5B,CAAC,CACH,CACF;EACD,MAAM,QAAQ,KAAK,OAAO;EAE1B,MAAM,iBACJ,KAAK,SAAS;AAEhB,OAAK,MAAM,SAAS,OAAO,QAAQ,EAAE;GACnC,MAAM,OAAsB;IAC1B;IACA,cAAc,UAAU,CAAC,MAAM;IAC/B,WAAW,QAAQ,MAAM,OAAO;IAChC,eAAe,WAAW,OAAO,QAC/B,KAAK,aAAa,MAAM,MAAM,WAAW,OAAO,IAAI;IACvD;AACD,QAAK,MAAM,QAAQ;IACjB,MAAM,GAAG,MAAM,KAAK;IACpB,QAAQ;IACR,MAAM,IAAI,MAAM;IAChB,SAAS,kBAAkB,KAAK;IACjC,CAAC;GACF,MAAM,SAAS,KAAK,SAAS,OAAO,IAAI,MAAM,KAAK;AACnD,OAAI,QAAQ,IAAI,SAAS,CACvB,MAAK,MAAM,QAAQ;IACjB,MAAM,GAAG,MAAM,KAAK;IACpB,QAAQ;IACR,MAAM,IAAI,MAAM;IAChB,SAAS,oBAAoB,KAAK;IACnC,CAAC;AAGJ,OAAI,CAAC,MAAM,WAAY;AACvB,QAAK,MAAM,QAAQ;IACjB,MAAM,GAAG,MAAM,KAAK;IACpB,QAAQ;IACR,MAAM,IAAI,MAAM,KAAK;IACrB,SAAS,oBAAoB,KAAK;IACnC,CAAC;AACF,OAAI,QAAQ,IAAI,SAAS,CACvB,MAAK,MAAM,QAAQ;IACjB,MAAM,GAAG,MAAM,KAAK;IACpB,QAAQ;IACR,MAAM,IAAI,MAAM,KAAK;IACrB,SAAS,oBAAoB,KAAK;IACnC,CAAC;AAEJ,OAAI,QAAQ,IAAI,SAAS,CACvB,MAAK,MAAM,QAAQ;IACjB,MAAM,GAAG,MAAM,KAAK;IACpB,QAAQ;IACR,MAAM,IAAI,MAAM,KAAK;IACrB,SAAS,oBAAoB,KAAK;IACnC,CAAC;;;;CAMR,AAAQ,QAAmC;AACzC,SAAO,KAAK,OAAO;;;CAIrB,UAAU;AACR,MAAI,CAAC,KAAK,SAAS,KAAK,SACtB,OAAM,IAAI,oBAAoB,YAAY,OAAO;AAEnD,SAAO,OAAO,OACZ,OAAO,OAAO,KAAK,EACnB,KAAK,MAAM,QACZ;;;CAIH,MAAM,WAA0B;AAC9B,MAAI,KAAK,gBAAiB,QAAO,KAAK;AACtC,OAAK,WAAW;AAChB,OAAK,mBAAmB,YAAY;AAClC,SAAM,KAAK,cAAc,YAAY,OAAU;GAC/C,MAAM,QAAQ,KAAK;AACnB,UAAO,YAAY;AACnB,QAAK,QAAQ;AACb,OAAI,MACF,KAAI;AACF,UAAM,MAAM,KAAK,KAAK;WAChB;AACN,UAAM,IAAI,oBAAoB,YAAY,WAAW;;MAGvD;AACJ,SAAO,KAAK;;;CAId,AAAQ,aACN,OACA,WACA,OACA,KACe;AACf,SAAO,KAAK,UAAU,gBACpB,uBACA,EACE,YAAY;GACV,YAAY;GACZ;GACA,cAAc,QAAQ,KAAK,OAAO;GACnC,EACF,EACD,OAAO,SAAS;AACd,OAAI;AACF,UAAM,KAAK;AACX,SAAK,aAAa,WAAW,UAAU;YAChC,OAAO;AACd,SAAK,aAAa,WAAW,aAAa,MAAM,CAAC;AACjD,UAAM;aACE;AACR,SAAK,KAAK;;IAGf;;;;AAKL,SAAgB,SACd,QACA;AACA,QAAO;EACL,QAAQ;EAIR;EACA,MAAM;EACP"}
+{"version":3,"file":"database.js","names":["manifest"],"sources":["../../../src/plugins/database/database.ts"],"sourcesContent":["import type express from \"express\";\nimport type { BasePluginConfig, PluginConstructor } from \"shared\";\n\nimport {\n  DatabasePluginError,\n  databaseSetupFailed,\n} from \"../../database/errors\";\nimport type { Schema } from \"../../database/schema-builder\";\nimport { assertFinalizedSchema } from \"../../database/schema-builder/define-schema\";\nimport { Plugin } from \"../../plugin\";\nimport type { PluginManifest } from \"../../registry\";\nimport { assertDatabaseConfig } from \"./config\";\nimport { compileCrudTables } from \"./crud/contract\";\nimport { type CrudExposure, resolveCrudExposure } from \"./crud/exposure\";\nimport { routeOutcome } from \"./crud/response\";\nimport {\n  type CrudEntity,\n  type CrudOperation,\n  type CrudRouteDeps,\n  createCreateHandler,\n  createDeleteHandler,\n  createDetailHandler,\n  createListHandler,\n  createUpdateHandler,\n} from \"./crud/routes\";\nimport type { DatabaseExports } from \"./entity-types\";\nimport { createDatabaseState, type DatabaseState } from \"./lifecycle\";\nimport { loadDefaultDatabaseSchema } from \"./load-schema\";\nimport manifest from \"./manifest.json\";\nimport type {\n  DatabaseHooks,\n  DefaultDatabaseSchema,\n  IDatabaseConfig,\n} from \"./types\";\n\n/** Schema-driven database plugin */\nexport class DatabasePlugin<\n  TSchema extends Schema = DefaultDatabaseSchema,\n> extends Plugin<IDatabaseConfig<TSchema>> {\n  /** Plugin metadata and required PostgreSQL resource. */\n  static manifest = manifest as PluginManifest<\"database\">;\n  declare protected config: IDatabaseConfig<TSchema>;\n  private state: DatabaseState | null = null;\n  private setupPromise: Promise<void> | null = null;\n  private draining = false;\n  private shutdownPromise: Promise<void> | null = null;\n  private exposure: CrudExposure = { tables: [], writes: new Map() };\n  private resolvedSchema: Schema | null = null;\n\n  constructor(config: IDatabaseConfig<TSchema> = {}) {\n    assertDatabaseConfig(config);\n    super({ schema: config.schema });\n    this.config = {\n      schema: config.schema,\n      api: config.api,\n      hooks: config.hooks,\n    };\n  }\n\n  /** Build and verify one candidate state before publishing its exports. */\n  async setup(): Promise<void> {\n    if (this.draining || this.state) throw databaseSetupFailed();\n    if (!this.setupPromise) {\n      const attempt = (async () => {\n        const schema =\n          this.config.schema === undefined\n            ? await loadDefaultDatabaseSchema()\n            : this.config.schema;\n        try {\n          assertFinalizedSchema(schema);\n        } catch {\n          throw databaseSetupFailed(\n            \"schema must be a finalized AppKit schema created with defineSchema().\",\n          );\n        }\n        if (this.draining) throw databaseSetupFailed();\n        const exposure = resolveCrudExposure(\n          this.config.api,\n          Object.keys(schema.$tables),\n        );\n        // A hook key naming no declared table would silently never run.\n        for (const name of Object.keys(this.hooks() ?? {})) {\n          if (!Object.hasOwn(schema.$tables, name)) {\n            throw databaseSetupFailed(\n              `hooks names undeclared table ${JSON.stringify(name)}. Use a table declared in schema.`,\n            );\n          }\n        }\n        const candidate = await createDatabaseState(\n          schema,\n          (operation, options) => this.execute(operation, options),\n          this.hooks(),\n        );\n        if (this.draining) {\n          // Setup may finish while shutdown is waiting; never publish that state.\n          candidate.deactivate();\n          await candidate.pool.end().catch(() => undefined);\n          throw databaseSetupFailed();\n        }\n        this.resolvedSchema = schema;\n        this.exposure = exposure;\n        this.state = candidate;\n      })();\n      this.setupPromise = attempt;\n    }\n    return this.setupPromise;\n  }\n\n  /** Register generated CRUD, subject to the configured table and write restrictions. */\n  injectRoutes(router: express.Router): void {\n    if (this.exposure.tables.length === 0) return;\n    const schema = this.resolvedSchema;\n    if (!schema) throw databaseSetupFailed();\n    const tables = compileCrudTables(\n      Object.fromEntries(\n        this.exposure.tables.map((name) => [name, schema.$tables[name]]),\n      ),\n    );\n    const hooks = this.hooks();\n    // Every exposed name is a declared table, so its export is an entity client.\n    const entities = () =>\n      this.exports() as unknown as Record<string, CrudEntity>;\n\n    for (const table of tables.values()) {\n      const deps: CrudRouteDeps = {\n        table,\n        entity: () => entities()[table.name],\n        serialize: hooks?.[table.name]?.serialize,\n        runRouteSpan: (operation, route, run) =>\n          this.runRouteSpan(table.name, operation, route, run),\n      };\n      this.route(router, {\n        name: `${table.name}.list`,\n        method: \"get\",\n        path: `/${table.name}`,\n        handler: createListHandler(deps),\n      });\n      const writes = this.exposure.writes.get(table.name);\n      if (writes?.has(\"create\")) {\n        this.route(router, {\n          name: `${table.name}.create`,\n          method: \"post\",\n          path: `/${table.name}`,\n          handler: createCreateHandler(deps),\n        });\n      }\n      // Addressing one row needs a public key.\n      if (!table.primaryKey) continue;\n      this.route(router, {\n        name: `${table.name}.detail`,\n        method: \"get\",\n        path: `/${table.name}/:id`,\n        handler: createDetailHandler(deps),\n      });\n      if (writes?.has(\"update\")) {\n        this.route(router, {\n          name: `${table.name}.update`,\n          method: \"patch\",\n          path: `/${table.name}/:id`,\n          handler: createUpdateHandler(deps),\n        });\n      }\n      if (writes?.has(\"delete\")) {\n        this.route(router, {\n          name: `${table.name}.delete`,\n          method: \"delete\",\n          path: `/${table.name}/:id`,\n          handler: createDeleteHandler(deps),\n        });\n      }\n    }\n  }\n\n  /** Typed hook keys are schema table names, which routing addresses at runtime. */\n  private hooks(): DatabaseHooks | undefined {\n    return this.config.hooks as DatabaseHooks | undefined;\n  }\n\n  /** Return the typed database API only while the plugin is active. */\n  exports() {\n    if (!this.state || this.draining)\n      throw new DatabasePluginError(\"INTERNAL\", \"read\");\n    // AppKit binds exported functions onto this object on every access.\n    return Object.assign(\n      Object.create(null),\n      this.state.exports,\n    ) as DatabaseExports;\n  }\n\n  /** Stop new work, wait for setup, and close the owned pool exactly once. */\n  async shutdown(): Promise<void> {\n    if (this.shutdownPromise) return this.shutdownPromise;\n    this.draining = true;\n    this.shutdownPromise = (async () => {\n      await this.setupPromise?.catch(() => undefined);\n      const state = this.state;\n      state?.deactivate();\n      this.state = null;\n      if (state) {\n        try {\n          await state.pool.end();\n        } catch {\n          throw new DatabasePluginError(\"INTERNAL\", \"shutdown\");\n        }\n      }\n    })();\n    return this.shutdownPromise;\n  }\n\n  /** Trace one generated route with allowlisted, low-cardinality attributes. */\n  private runRouteSpan(\n    table: string,\n    operation: CrudOperation,\n    route: string,\n    run: () => Promise<void>,\n  ): Promise<void> {\n    return this.telemetry.startActiveSpan(\n      \"database.crud.route\",\n      {\n        attributes: {\n          table_name: table,\n          operation,\n          \"http.route\": `/api/${this.name}${route}`,\n        },\n      },\n      async (span) => {\n        try {\n          await run();\n          span.setAttribute(\"outcome\", \"success\");\n        } catch (error) {\n          span.setAttribute(\"outcome\", routeOutcome(error));\n          throw error;\n        } finally {\n          span.end();\n        }\n      },\n    );\n  }\n}\n\ntype DatabaseRegistration<TSchema extends Schema> = {\n  plugin: PluginConstructor<BasePluginConfig, DatabasePlugin<TSchema>>;\n  config: IDatabaseConfig<TSchema>;\n  name: \"database\";\n};\n\n/**\n * Create the database plugin. Omit configuration to load\n * `config/database/schema.ts`, or supply a typed schema override.\n */\nexport function database<TSchema extends Schema>(\n  config: IDatabaseConfig<TSchema> & { readonly schema: TSchema },\n): DatabaseRegistration<TSchema> & {\n  config: IDatabaseConfig<TSchema> & { readonly schema: TSchema };\n};\n/**\n * Register the database plugin with opinionated defaults and full HTTP CRUD.\n * By default, setup loads the named `schema` export from the application's\n * `config/database/schema.ts`. Registration itself performs no file or database I/O.\n */\nexport function database<TSchema extends Schema = DefaultDatabaseSchema>(\n  config?: IDatabaseConfig<TSchema>,\n): DatabaseRegistration<TSchema>;\nexport function database<TSchema extends Schema = DefaultDatabaseSchema>(\n  config: IDatabaseConfig<TSchema> = {},\n): DatabaseRegistration<TSchema> {\n  assertDatabaseConfig(config);\n  return {\n    plugin: DatabasePlugin as unknown as PluginConstructor<\n      BasePluginConfig,\n      DatabasePlugin<TSchema>\n    >,\n    config,\n    name: \"database\" as const,\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;AAoCA,IAAa,iBAAb,cAEU,OAAiC;;CAEzC,OAAO,WAAWA;CAElB,AAAQ,QAA8B;CACtC,AAAQ,eAAqC;CAC7C,AAAQ,WAAW;CACnB,AAAQ,kBAAwC;CAChD,AAAQ,WAAyB;EAAE,QAAQ,EAAE;EAAE,wBAAQ,IAAI,KAAK;EAAE;CAClE,AAAQ,iBAAgC;CAExC,YAAY,SAAmC,EAAE,EAAE;AACjD,uBAAqB,OAAO;AAC5B,QAAM,EAAE,QAAQ,OAAO,QAAQ,CAAC;AAChC,OAAK,SAAS;GACZ,QAAQ,OAAO;GACf,KAAK,OAAO;GACZ,OAAO,OAAO;GACf;;;CAIH,MAAM,QAAuB;AAC3B,MAAI,KAAK,YAAY,KAAK,MAAO,OAAM,qBAAqB;AAC5D,MAAI,CAAC,KAAK,aAyCR,MAAK,gBAxCY,YAAY;GAC3B,MAAM,SACJ,KAAK,OAAO,WAAW,SACnB,MAAM,2BAA2B,GACjC,KAAK,OAAO;AAClB,OAAI;AACF,0BAAsB,OAAO;WACvB;AACN,UAAM,oBACJ,wEACD;;AAEH,OAAI,KAAK,SAAU,OAAM,qBAAqB;GAC9C,MAAM,WAAW,oBACf,KAAK,OAAO,KACZ,OAAO,KAAK,OAAO,QAAQ,CAC5B;AAED,QAAK,MAAM,QAAQ,OAAO,KAAK,KAAK,OAAO,IAAI,EAAE,CAAC,CAChD,KAAI,CAAC,OAAO,OAAO,OAAO,SAAS,KAAK,CACtC,OAAM,oBACJ,gCAAgC,KAAK,UAAU,KAAK,CAAC,mCACtD;GAGL,MAAM,YAAY,MAAM,oBACtB,SACC,WAAW,YAAY,KAAK,QAAQ,WAAW,QAAQ,EACxD,KAAK,OAAO,CACb;AACD,OAAI,KAAK,UAAU;AAEjB,cAAU,YAAY;AACtB,UAAM,UAAU,KAAK,KAAK,CAAC,YAAY,OAAU;AACjD,UAAM,qBAAqB;;AAE7B,QAAK,iBAAiB;AACtB,QAAK,WAAW;AAChB,QAAK,QAAQ;MACX;AAGN,SAAO,KAAK;;;CAId,aAAa,QAA8B;AACzC,MAAI,KAAK,SAAS,OAAO,WAAW,EAAG;EACvC,MAAM,SAAS,KAAK;AACpB,MAAI,CAAC,OAAQ,OAAM,qBAAqB;EACxC,MAAM,SAAS,kBACb,OAAO,YACL,KAAK,SAAS,OAAO,KAAK,SAAS,CAAC,MAAM,OAAO,QAAQ,MAAM,CAAC,CACjE,CACF;EACD,MAAM,QAAQ,KAAK,OAAO;EAE1B,MAAM,iBACJ,KAAK,SAAS;AAEhB,OAAK,MAAM,SAAS,OAAO,QAAQ,EAAE;GACnC,MAAM,OAAsB;IAC1B;IACA,cAAc,UAAU,CAAC,MAAM;IAC/B,WAAW,QAAQ,MAAM,OAAO;IAChC,eAAe,WAAW,OAAO,QAC/B,KAAK,aAAa,MAAM,MAAM,WAAW,OAAO,IAAI;IACvD;AACD,QAAK,MAAM,QAAQ;IACjB,MAAM,GAAG,MAAM,KAAK;IACpB,QAAQ;IACR,MAAM,IAAI,MAAM;IAChB,SAAS,kBAAkB,KAAK;IACjC,CAAC;GACF,MAAM,SAAS,KAAK,SAAS,OAAO,IAAI,MAAM,KAAK;AACnD,OAAI,QAAQ,IAAI,SAAS,CACvB,MAAK,MAAM,QAAQ;IACjB,MAAM,GAAG,MAAM,KAAK;IACpB,QAAQ;IACR,MAAM,IAAI,MAAM;IAChB,SAAS,oBAAoB,KAAK;IACnC,CAAC;AAGJ,OAAI,CAAC,MAAM,WAAY;AACvB,QAAK,MAAM,QAAQ;IACjB,MAAM,GAAG,MAAM,KAAK;IACpB,QAAQ;IACR,MAAM,IAAI,MAAM,KAAK;IACrB,SAAS,oBAAoB,KAAK;IACnC,CAAC;AACF,OAAI,QAAQ,IAAI,SAAS,CACvB,MAAK,MAAM,QAAQ;IACjB,MAAM,GAAG,MAAM,KAAK;IACpB,QAAQ;IACR,MAAM,IAAI,MAAM,KAAK;IACrB,SAAS,oBAAoB,KAAK;IACnC,CAAC;AAEJ,OAAI,QAAQ,IAAI,SAAS,CACvB,MAAK,MAAM,QAAQ;IACjB,MAAM,GAAG,MAAM,KAAK;IACpB,QAAQ;IACR,MAAM,IAAI,MAAM,KAAK;IACrB,SAAS,oBAAoB,KAAK;IACnC,CAAC;;;;CAMR,AAAQ,QAAmC;AACzC,SAAO,KAAK,OAAO;;;CAIrB,UAAU;AACR,MAAI,CAAC,KAAK,SAAS,KAAK,SACtB,OAAM,IAAI,oBAAoB,YAAY,OAAO;AAEnD,SAAO,OAAO,OACZ,OAAO,OAAO,KAAK,EACnB,KAAK,MAAM,QACZ;;;CAIH,MAAM,WAA0B;AAC9B,MAAI,KAAK,gBAAiB,QAAO,KAAK;AACtC,OAAK,WAAW;AAChB,OAAK,mBAAmB,YAAY;AAClC,SAAM,KAAK,cAAc,YAAY,OAAU;GAC/C,MAAM,QAAQ,KAAK;AACnB,UAAO,YAAY;AACnB,QAAK,QAAQ;AACb,OAAI,MACF,KAAI;AACF,UAAM,MAAM,KAAK,KAAK;WAChB;AACN,UAAM,IAAI,oBAAoB,YAAY,WAAW;;MAGvD;AACJ,SAAO,KAAK;;;CAId,AAAQ,aACN,OACA,WACA,OACA,KACe;AACf,SAAO,KAAK,UAAU,gBACpB,uBACA,EACE,YAAY;GACV,YAAY;GACZ;GACA,cAAc,QAAQ,KAAK,OAAO;GACnC,EACF,EACD,OAAO,SAAS;AACd,OAAI;AACF,UAAM,KAAK;AACX,SAAK,aAAa,WAAW,UAAU;YAChC,OAAO;AACd,SAAK,aAAa,WAAW,aAAa,MAAM,CAAC;AACjD,UAAM;aACE;AACR,SAAK,KAAK;;IAGf;;;AA2BL,SAAgB,SACd,SAAmC,EAAE,EACN;AAC/B,sBAAqB,OAAO;AAC5B,QAAO;EACL,QAAQ;EAIR;EACA,MAAM;EACP"}

package/dist/plugins/database/lifecycle.js

@@ -1,7 +1,7 @@
-import { createLogger } from "../../logging/logger.js";
 import { DatabaseValidationError } from "../../errors/database-validation.js";
 import "../../errors/index.js";
-import { createLakebasePool } from "../../connectors/lakebase/index.js";
+import { createLogger } from "../../logging/logger.js";
+import { initializeLakebasePool } from "../../connectors/lakebase/index.js";
 import { assertFinalizedSchema } from "../../database/schema-builder/define-schema.js";
 import { DatabasePluginError, classifyDatabaseError } from "../../database/errors.js";
 import { IDLE_IN_TRANSACTION_TIMEOUT_MS, STATEMENT_TIMEOUT_MS, TRANSACTION_TIMEOUT_MS } from "./defaults.js";
@@ -112,7 +112,7 @@ async function createDatabaseState(schema, execute, hooks) {
 		if (!active) throw new DatabasePluginError("INTERNAL", "runtime");
 	};
 	try {
-		pool = createLakebasePool({
+		pool = await initializeLakebasePool({
 			statement_timeout: STATEMENT_TIMEOUT_MS,
 			idle_in_transaction_session_timeout: IDLE_IN_TRANSACTION_TIMEOUT_MS
 		});