@inkeep/agents-core 0.64.2 → 0.64.6

package/dist/db/manage/manage-schema.js

@@ -1,6 +1,7 @@
 import { __exportAll } from "../../_virtual/rolldown_runtime.js";
+import { jsonb as jsonb$1 } from "./dolt-safe-jsonb.js";
 import { relations } from "drizzle-orm";
-import { boolean, doublePrecision, foreignKey, index, jsonb, numeric, pgTable, primaryKey, text, timestamp, unique, varchar } from "drizzle-orm/pg-core";
+import { boolean, doublePrecision, foreignKey, index, numeric, pgTable, primaryKey, text, timestamp, unique, varchar } from "drizzle-orm/pg-core";
 
 //#region src/db/manage/manage-schema.ts
 var manage_schema_exports = /* @__PURE__ */ __exportAll({
@@ -90,8 +91,8 @@ const timestamps = {
 const projects = pgTable("projects", {
 	...tenantScoped,
 	...uiProperties,
-	models: jsonb("models").$type(),
-	stopWhen: jsonb("stop_when").$type(),
+	models: jsonb$1("models").$type(),
+	stopWhen: jsonb$1("stop_when").$type(),
 	...timestamps
 }, (table) => [primaryKey({ columns: [table.tenantId, table.id] })]);
 const agents = pgTable("agent", {
@@ -99,10 +100,10 @@ const agents = pgTable("agent", {
 	...uiProperties,
 	defaultSubAgentId: varchar("default_sub_agent_id", { length: 256 }),
 	contextConfigId: varchar("context_config_id", { length: 256 }),
-	models: jsonb("models").$type(),
-	statusUpdates: jsonb("status_updates").$type(),
+	models: jsonb$1("models").$type(),
+	statusUpdates: jsonb$1("status_updates").$type(),
 	prompt: text("prompt"),
-	stopWhen: jsonb("stop_when").$type(),
+	stopWhen: jsonb$1("stop_when").$type(),
 	executionMode: varchar("execution_mode", { length: 50 }).$type().notNull().default("classic"),
 	...timestamps
 }, (table) => [primaryKey({ columns: [
@@ -116,8 +117,8 @@ const agents = pgTable("agent", {
 }).onDelete("cascade")]);
 const contextConfigs = pgTable("context_configs", {
 	...agentScoped,
-	headersSchema: jsonb("headers_schema").$type(),
-	contextVariables: jsonb("context_variables").$type(),
+	headersSchema: jsonb$1("headers_schema").$type(),
+	contextVariables: jsonb$1("context_variables").$type(),
 	...timestamps
 }, (table) => [primaryKey({ columns: [
 	table.tenantId,
@@ -141,12 +142,12 @@ const triggers = pgTable("triggers", {
 	...agentScoped,
 	...uiProperties,
 	enabled: boolean("enabled").notNull().default(true),
-	inputSchema: jsonb("input_schema").$type(),
-	outputTransform: jsonb("output_transform").$type(),
+	inputSchema: jsonb$1("input_schema").$type(),
+	outputTransform: jsonb$1("output_transform").$type(),
 	messageTemplate: text("message_template"),
-	authentication: jsonb("authentication").$type(),
+	authentication: jsonb$1("authentication").$type(),
 	signingSecretCredentialReferenceId: varchar("signing_secret_credential_reference_id", { length: 256 }),
-	signatureVerification: jsonb("signature_verification").$type().default(null),
+	signatureVerification: jsonb$1("signature_verification").$type().default(null),
 	runAsUserId: varchar("run_as_user_id", { length: 256 }),
 	createdBy: varchar("created_by", { length: 256 }),
 	...timestamps
@@ -180,15 +181,15 @@ const subAgents = pgTable("sub_agents", {
 	...agentScoped,
 	...uiProperties,
 	prompt: text("prompt"),
-	conversationHistoryConfig: jsonb("conversation_history_config").$type().default({
+	conversationHistoryConfig: jsonb$1("conversation_history_config").$type().default({
 		mode: "full",
 		limit: 50,
 		maxOutputTokens: 4e3,
 		includeInternal: false,
 		messageTypes: ["chat", "tool-result"]
 	}),
-	models: jsonb("models").$type(),
-	stopWhen: jsonb("stop_when").$type(),
+	models: jsonb$1("models").$type(),
+	stopWhen: jsonb$1("stop_when").$type(),
 	...timestamps
 }, (table) => [primaryKey({ columns: [
 	table.tenantId,
@@ -214,7 +215,7 @@ const skills = pgTable("skills", {
 	name: varchar("name", { length: 64 }).notNull(),
 	description: text("description").notNull(),
 	content: text("content").notNull(),
-	metadata: jsonb("metadata").$type(),
+	metadata: jsonb$1("metadata").$type(),
 	...timestamps
 }, (table) => [primaryKey({ columns: [
 	table.tenantId,
@@ -347,8 +348,8 @@ const externalAgents = pgTable("external_agents", {
 const dataComponents = pgTable("data_components", {
 	...projectScoped,
 	...uiProperties,
-	props: jsonb("props").$type().notNull(),
-	render: jsonb("render").$type(),
+	props: jsonb$1("props").$type().notNull(),
+	render: jsonb$1("render").$type(),
 	...timestamps
 }, (table) => [primaryKey({ columns: [
 	table.tenantId,
@@ -401,8 +402,8 @@ const subAgentDataComponents = pgTable("sub_agent_data_components", {
 const artifactComponents = pgTable("artifact_components", {
 	...projectScoped,
 	...uiProperties,
-	props: jsonb("props").$type(),
-	render: jsonb("render").$type(),
+	props: jsonb$1("props").$type(),
+	render: jsonb$1("render").$type(),
 	...timestamps
 }, (table) => [primaryKey({ columns: [
 	table.tenantId,
@@ -460,12 +461,12 @@ const subAgentArtifactComponents = pgTable("sub_agent_artifact_components", {
 const tools = pgTable("tools", {
 	...projectScoped,
 	...uiProperties,
-	config: jsonb("config").$type().notNull(),
+	config: jsonb$1("config").$type().notNull(),
 	credentialReferenceId: varchar("credential_reference_id", { length: 256 }),
 	credentialScope: varchar("credential_scope", { length: 50 }).notNull().default("project"),
-	headers: jsonb("headers").$type(),
+	headers: jsonb$1("headers").$type(),
 	imageUrl: text("image_url"),
-	capabilities: jsonb("capabilities").$type(),
+	capabilities: jsonb$1("capabilities").$type(),
 	lastError: text("last_error"),
 	isWorkApp: boolean("is_work_app").notNull().default(false),
 	...timestamps
@@ -527,9 +528,9 @@ const functionTools = pgTable("function_tools", {
 ]);
 const functions = pgTable("functions", {
 	...projectScoped,
-	inputSchema: jsonb("input_schema").$type(),
+	inputSchema: jsonb$1("input_schema").$type(),
 	executeCode: text("execute_code").notNull(),
-	dependencies: jsonb("dependencies").$type(),
+	dependencies: jsonb$1("dependencies").$type(),
 	...timestamps
 }, (table) => [primaryKey({ columns: [
 	table.tenantId,
@@ -543,9 +544,9 @@ const functions = pgTable("functions", {
 const subAgentToolRelations = pgTable("sub_agent_tool_relations", {
 	...subAgentScoped,
 	toolId: varchar("tool_id", { length: 256 }).notNull(),
-	selectedTools: jsonb("selected_tools").$type(),
-	headers: jsonb("headers").$type(),
-	toolPolicies: jsonb("tool_policies").$type(),
+	selectedTools: jsonb$1("selected_tools").$type(),
+	headers: jsonb$1("headers").$type(),
+	toolPolicies: jsonb$1("tool_policies").$type(),
 	...timestamps
 }, (table) => [
 	primaryKey({ columns: [
@@ -586,7 +587,7 @@ const subAgentToolRelations = pgTable("sub_agent_tool_relations", {
 const subAgentExternalAgentRelations = pgTable("sub_agent_external_agent_relations", {
 	...subAgentScoped,
 	externalAgentId: varchar("external_agent_id", { length: 256 }).notNull(),
-	headers: jsonb("headers").$type(),
+	headers: jsonb$1("headers").$type(),
 	...timestamps
 }, (table) => [
 	primaryKey({
@@ -630,7 +631,7 @@ const subAgentExternalAgentRelations = pgTable("sub_agent_external_agent_relatio
 const subAgentTeamAgentRelations = pgTable("sub_agent_team_agent_relations", {
 	...subAgentScoped,
 	targetAgentId: varchar("target_agent_id", { length: 256 }).notNull(),
-	headers: jsonb("headers").$type(),
+	headers: jsonb$1("headers").$type(),
 	...timestamps
 }, (table) => [
 	primaryKey({
@@ -674,7 +675,7 @@ const subAgentTeamAgentRelations = pgTable("sub_agent_team_agent_relations", {
 const subAgentFunctionToolRelations = pgTable("sub_agent_function_tool_relations", {
 	...subAgentScoped,
 	functionToolId: varchar("function_tool_id", { length: 256 }).notNull(),
-	toolPolicies: jsonb("tool_policies").$type(),
+	toolPolicies: jsonb$1("tool_policies").$type(),
 	...timestamps
 }, (table) => [
 	primaryKey({
@@ -722,7 +723,7 @@ const credentialReferences = pgTable("credential_references", {
 	name: uiProperties.name,
 	type: varchar("type", { length: 256 }).notNull(),
 	credentialStoreId: varchar("credential_store_id", { length: 256 }).notNull(),
-	retrievalParams: jsonb("retrieval_params").$type(),
+	retrievalParams: jsonb$1("retrieval_params").$type(),
 	toolId: varchar("tool_id", { length: 256 }),
 	userId: varchar("user_id", { length: 256 }),
 	createdBy: varchar("created_by", { length: 256 }),
@@ -775,8 +776,8 @@ const dataset = pgTable("dataset", {
 const datasetItem = pgTable("dataset_item", {
 	...projectScoped,
 	datasetId: varchar("dataset_id", { length: 256 }).notNull(),
-	input: jsonb("input").$type().notNull(),
-	expectedOutput: jsonb("expected_output").$type(),
+	input: jsonb$1("input").$type().notNull(),
+	expectedOutput: jsonb$1("expected_output").$type(),
 	...timestamps
 }, (table) => [primaryKey({ columns: [
 	table.tenantId,
@@ -799,9 +800,9 @@ const evaluator = pgTable("evaluator", {
 	...projectScoped,
 	...uiProperties,
 	prompt: text("prompt").notNull(),
-	schema: jsonb("schema").$type().notNull(),
-	model: jsonb("model").$type().notNull(),
-	passCriteria: jsonb("pass_criteria").$type(),
+	schema: jsonb$1("schema").$type().notNull(),
+	model: jsonb$1("model").$type().notNull(),
+	passCriteria: jsonb$1("pass_criteria").$type(),
 	...timestamps
 }, (table) => [primaryKey({ columns: [
 	table.tenantId,
@@ -859,7 +860,7 @@ const datasetRunConfig = pgTable("dataset_run_config", {
 */
 const evaluationSuiteConfig = pgTable("evaluation_suite_config", {
 	...projectScoped,
-	filters: jsonb("filters").$type(),
+	filters: jsonb$1("filters").$type(),
 	sampleRate: doublePrecision("sample_rate"),
 	...timestamps
 }, (table) => [primaryKey({ columns: [
@@ -1010,7 +1011,7 @@ const evaluationRunConfig = pgTable("evaluation_run_config", {
 */
 const evaluationJobConfig = pgTable("evaluation_job_config", {
 	...projectScoped,
-	jobFilters: jsonb("job_filters").$type(),
+	jobFilters: jsonb$1("job_filters").$type(),
 	...timestamps
 }, (table) => [primaryKey({ columns: [
 	table.tenantId,

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

@@ -7,7 +7,7 @@ import * as drizzle_orm_pg_core1458 from "drizzle-orm/pg-core";
 
 //#region src/db/runtime/runtime-schema.d.ts
 declare namespace runtime_schema_d_exports {
-  export { GenerationType, USAGE_GENERATION_TYPES, account, apiKeys, apps, contextCache, conversations, conversationsRelations, datasetRun, datasetRunConversationRelations, deviceCode, evaluationResult, evaluationRun, invitation, ledgerArtifacts, ledgerArtifactsRelations, member, messages, messagesRelations, orgEntitlement, organization, projectMetadata, scheduledTriggerInvocations, scheduledTriggers, schedulerState, session, ssoProvider, taskRelations, taskRelationsRelations, tasks, tasksRelations, triggerInvocations, user, userProfile, userProfileRelations, verification, workAppGitHubInstallations, workAppGitHubInstallationsRelations, workAppGitHubMcpToolAccessMode, workAppGitHubMcpToolRepositoryAccess, workAppGitHubMcpToolRepositoryAccessRelations, workAppGitHubProjectAccessMode, workAppGitHubProjectRepositoryAccess, workAppGitHubProjectRepositoryAccessRelations, workAppGitHubRepositories, workAppGitHubRepositoriesRelations, workAppSlackChannelAgentConfigs, workAppSlackMcpToolAccessConfig, workAppSlackUserMappings, workAppSlackWorkspaces, workflowExecutions };
+  export { GenerationType, USAGE_GENERATION_TYPES, account, apiKeys, apps, contextCache, conversations, conversationsRelations, datasetRun, datasetRunConversationRelations, deviceCode, evaluationResult, evaluationRun, invitation, ledgerArtifacts, ledgerArtifactsRelations, member, messages, messagesRelations, orgEntitlement, organization, projectMetadata, scheduledTriggerInvocations, scheduledTriggers, schedulerState, session, ssoProvider, streamChunks, taskRelations, taskRelationsRelations, tasks, tasksRelations, triggerInvocations, user, userProfile, userProfileRelations, verification, workAppGitHubInstallations, workAppGitHubInstallationsRelations, workAppGitHubMcpToolAccessMode, workAppGitHubMcpToolRepositoryAccess, workAppGitHubMcpToolRepositoryAccessRelations, workAppGitHubProjectAccessMode, workAppGitHubProjectRepositoryAccess, workAppGitHubProjectRepositoryAccessRelations, workAppGitHubRepositories, workAppGitHubRepositoriesRelations, workAppSlackChannelAgentConfigs, workAppSlackMcpToolAccessConfig, workAppSlackUserMappings, workAppSlackWorkspaces, workflowExecutions };
 }
 /**
  * Runtime projects table - source of truth for which projects exist in a tenant.
@@ -764,6 +764,138 @@ declare const workflowExecutions: drizzle_orm_pg_core1458.PgTableWithColumns<{
   };
   dialect: "pg";
 }>;
+declare const streamChunks: drizzle_orm_pg_core1458.PgTableWithColumns<{
+  name: "stream_chunks";
+  schema: undefined;
+  columns: {
+    tenantId: drizzle_orm_pg_core1458.PgColumn<{
+      name: "tenant_id";
+      tableName: "stream_chunks";
+      dataType: "string";
+      columnType: "PgVarchar";
+      data: string;
+      driverParam: string;
+      notNull: true;
+      hasDefault: false;
+      isPrimaryKey: false;
+      isAutoincrement: false;
+      hasRuntimeDefault: false;
+      enumValues: [string, ...string[]];
+      baseColumn: never;
+      identity: undefined;
+      generated: undefined;
+    }, {}, {
+      length: 256;
+    }>;
+    projectId: drizzle_orm_pg_core1458.PgColumn<{
+      name: "project_id";
+      tableName: "stream_chunks";
+      dataType: "string";
+      columnType: "PgVarchar";
+      data: string;
+      driverParam: string;
+      notNull: true;
+      hasDefault: false;
+      isPrimaryKey: false;
+      isAutoincrement: false;
+      hasRuntimeDefault: false;
+      enumValues: [string, ...string[]];
+      baseColumn: never;
+      identity: undefined;
+      generated: undefined;
+    }, {}, {
+      length: 256;
+    }>;
+    conversationId: drizzle_orm_pg_core1458.PgColumn<{
+      name: "conversation_id";
+      tableName: "stream_chunks";
+      dataType: "string";
+      columnType: "PgVarchar";
+      data: string;
+      driverParam: string;
+      notNull: true;
+      hasDefault: false;
+      isPrimaryKey: false;
+      isAutoincrement: false;
+      hasRuntimeDefault: false;
+      enumValues: [string, ...string[]];
+      baseColumn: never;
+      identity: undefined;
+      generated: undefined;
+    }, {}, {
+      length: 256;
+    }>;
+    idx: drizzle_orm_pg_core1458.PgColumn<{
+      name: "idx";
+      tableName: "stream_chunks";
+      dataType: "number";
+      columnType: "PgInteger";
+      data: number;
+      driverParam: string | number;
+      notNull: true;
+      hasDefault: false;
+      isPrimaryKey: false;
+      isAutoincrement: false;
+      hasRuntimeDefault: false;
+      enumValues: undefined;
+      baseColumn: never;
+      identity: undefined;
+      generated: undefined;
+    }, {}, {}>;
+    data: drizzle_orm_pg_core1458.PgColumn<{
+      name: "data";
+      tableName: "stream_chunks";
+      dataType: "string";
+      columnType: "PgText";
+      data: string;
+      driverParam: string;
+      notNull: true;
+      hasDefault: false;
+      isPrimaryKey: false;
+      isAutoincrement: false;
+      hasRuntimeDefault: false;
+      enumValues: [string, ...string[]];
+      baseColumn: never;
+      identity: undefined;
+      generated: undefined;
+    }, {}, {}>;
+    isFinal: drizzle_orm_pg_core1458.PgColumn<{
+      name: "is_final";
+      tableName: "stream_chunks";
+      dataType: "boolean";
+      columnType: "PgBoolean";
+      data: boolean;
+      driverParam: boolean;
+      notNull: true;
+      hasDefault: true;
+      isPrimaryKey: false;
+      isAutoincrement: false;
+      hasRuntimeDefault: false;
+      enumValues: undefined;
+      baseColumn: never;
+      identity: undefined;
+      generated: undefined;
+    }, {}, {}>;
+    createdAt: drizzle_orm_pg_core1458.PgColumn<{
+      name: "created_at";
+      tableName: "stream_chunks";
+      dataType: "string";
+      columnType: "PgTimestampString";
+      data: string;
+      driverParam: string;
+      notNull: true;
+      hasDefault: true;
+      isPrimaryKey: false;
+      isAutoincrement: false;
+      hasRuntimeDefault: false;
+      enumValues: undefined;
+      baseColumn: never;
+      identity: undefined;
+      generated: undefined;
+    }, {}, {}>;
+  };
+  dialect: "pg";
+}>;
 declare const apiKeys: drizzle_orm_pg_core1458.PgTableWithColumns<{
   name: "api_keys";
   schema: undefined;
@@ -2772,7 +2904,7 @@ declare const scheduledTriggerInvocations: drizzle_orm_pg_core1458.PgTableWithCo
       tableName: "scheduled_trigger_invocations";
       dataType: "string";
       columnType: "PgVarchar";
-      data: "pending" | "running" | "completed" | "failed" | "cancelled";
+      data: "pending" | "failed" | "running" | "completed" | "cancelled";
       driverParam: string;
       notNull: true;
       hasDefault: false;
@@ -2785,7 +2917,7 @@ declare const scheduledTriggerInvocations: drizzle_orm_pg_core1458.PgTableWithCo
       generated: undefined;
     }, {}, {
       length: 50;
-      $type: "pending" | "running" | "completed" | "failed" | "cancelled";
+      $type: "pending" | "failed" | "running" | "completed" | "cancelled";
     }>;
     scheduledFor: drizzle_orm_pg_core1458.PgColumn<{
       name: "scheduled_for";
@@ -6186,4 +6318,4 @@ declare const orgEntitlement: drizzle_orm_pg_core1458.PgTableWithColumns<{
   dialect: "pg";
 }>;
 //#endregion
-export { GenerationType, USAGE_GENERATION_TYPES, account, apiKeys, apps, contextCache, conversations, conversationsRelations, datasetRun, datasetRunConversationRelations, deviceCode, evaluationResult, evaluationRun, invitation, ledgerArtifacts, ledgerArtifactsRelations, member, messages, messagesRelations, orgEntitlement, organization, projectMetadata, runtime_schema_d_exports, scheduledTriggerInvocations, scheduledTriggers, schedulerState, session, ssoProvider, taskRelations, taskRelationsRelations, tasks, tasksRelations, triggerInvocations, user, userProfile, userProfileRelations, verification, workAppGitHubInstallations, workAppGitHubInstallationsRelations, workAppGitHubMcpToolAccessMode, workAppGitHubMcpToolRepositoryAccess, workAppGitHubMcpToolRepositoryAccessRelations, workAppGitHubProjectAccessMode, workAppGitHubProjectRepositoryAccess, workAppGitHubProjectRepositoryAccessRelations, workAppGitHubRepositories, workAppGitHubRepositoriesRelations, workAppSlackChannelAgentConfigs, workAppSlackMcpToolAccessConfig, workAppSlackUserMappings, workAppSlackWorkspaces, workflowExecutions };
+export { GenerationType, USAGE_GENERATION_TYPES, account, apiKeys, apps, contextCache, conversations, conversationsRelations, datasetRun, datasetRunConversationRelations, deviceCode, evaluationResult, evaluationRun, invitation, ledgerArtifacts, ledgerArtifactsRelations, member, messages, messagesRelations, orgEntitlement, organization, projectMetadata, runtime_schema_d_exports, scheduledTriggerInvocations, scheduledTriggers, schedulerState, session, ssoProvider, streamChunks, taskRelations, taskRelationsRelations, tasks, tasksRelations, triggerInvocations, user, userProfile, userProfileRelations, verification, workAppGitHubInstallations, workAppGitHubInstallationsRelations, workAppGitHubMcpToolAccessMode, workAppGitHubMcpToolRepositoryAccess, workAppGitHubMcpToolRepositoryAccessRelations, workAppGitHubProjectAccessMode, workAppGitHubProjectRepositoryAccess, workAppGitHubProjectRepositoryAccessRelations, workAppGitHubRepositories, workAppGitHubRepositoriesRelations, workAppSlackChannelAgentConfigs, workAppSlackMcpToolAccessConfig, workAppSlackUserMappings, workAppSlackWorkspaces, workflowExecutions };

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

@@ -32,6 +32,7 @@ var runtime_schema_exports = /* @__PURE__ */ __exportAll({
 	schedulerState: () => schedulerState,
 	session: () => session,
 	ssoProvider: () => ssoProvider,
+	streamChunks: () => streamChunks,
 	taskRelations: () => taskRelations,
 	taskRelationsRelations: () => taskRelationsRelations,
 	tasks: () => tasks,
@@ -140,6 +141,24 @@ const workflowExecutions = pgTable("workflow_executions", {
 	table.projectId,
 	table.id
 ] }), index("workflow_executions_conversation_idx").on(table.tenantId, table.projectId, table.conversationId)]);
+const streamChunks = pgTable("stream_chunks", {
+	tenantId: varchar("tenant_id", { length: 256 }).notNull(),
+	projectId: varchar("project_id", { length: 256 }).notNull(),
+	conversationId: varchar("conversation_id", { length: 256 }).notNull(),
+	idx: integer("idx").notNull(),
+	data: text("data").notNull(),
+	isFinal: boolean("is_final").notNull().default(false),
+	createdAt: timestamp("created_at", { mode: "string" }).notNull().defaultNow()
+}, (table) => [
+	primaryKey({ columns: [
+		table.tenantId,
+		table.projectId,
+		table.conversationId,
+		table.idx
+	] }),
+	index("stream_chunks_cleanup_idx").on(table.createdAt),
+	index("stream_chunks_conversation_idx").on(table.tenantId, table.projectId, table.conversationId, table.idx)
+]);
 const apiKeys = pgTable("api_keys", {
 	...projectScoped,
 	agentId: varchar("agent_id", { length: 256 }).notNull(),
@@ -889,4 +908,4 @@ const orgEntitlement = pgTable("org_entitlement", {
 ]);
 
 //#endregion
-export { USAGE_GENERATION_TYPES, account, apiKeys, apps, contextCache, conversations, conversationsRelations, datasetRun, datasetRunConversationRelations, deviceCode, evaluationResult, evaluationRun, invitation, ledgerArtifacts, ledgerArtifactsRelations, member, messages, messagesRelations, orgEntitlement, organization, projectMetadata, runtime_schema_exports, scheduledTriggerInvocations, scheduledTriggers, schedulerState, session, ssoProvider, taskRelations, taskRelationsRelations, tasks, tasksRelations, triggerInvocations, user, userProfile, userProfileRelations, verification, workAppGitHubInstallations, workAppGitHubInstallationsRelations, workAppGitHubMcpToolAccessMode, workAppGitHubMcpToolRepositoryAccess, workAppGitHubMcpToolRepositoryAccessRelations, workAppGitHubProjectAccessMode, workAppGitHubProjectRepositoryAccess, workAppGitHubProjectRepositoryAccessRelations, workAppGitHubRepositories, workAppGitHubRepositoriesRelations, workAppSlackChannelAgentConfigs, workAppSlackMcpToolAccessConfig, workAppSlackUserMappings, workAppSlackWorkspaces, workflowExecutions };
+export { USAGE_GENERATION_TYPES, account, apiKeys, apps, contextCache, conversations, conversationsRelations, datasetRun, datasetRunConversationRelations, deviceCode, evaluationResult, evaluationRun, invitation, ledgerArtifacts, ledgerArtifactsRelations, member, messages, messagesRelations, orgEntitlement, organization, projectMetadata, runtime_schema_exports, scheduledTriggerInvocations, scheduledTriggers, schedulerState, session, ssoProvider, streamChunks, taskRelations, taskRelationsRelations, tasks, tasksRelations, triggerInvocations, user, userProfile, userProfileRelations, verification, workAppGitHubInstallations, workAppGitHubInstallationsRelations, workAppGitHubMcpToolAccessMode, workAppGitHubMcpToolRepositoryAccess, workAppGitHubMcpToolRepositoryAccessRelations, workAppGitHubProjectAccessMode, workAppGitHubProjectRepositoryAccess, workAppGitHubProjectRepositoryAccessRelations, workAppGitHubRepositories, workAppGitHubRepositoriesRelations, workAppSlackChannelAgentConfigs, workAppSlackMcpToolAccessConfig, workAppSlackUserMappings, workAppSlackWorkspaces, workflowExecutions };

package/dist/dolt/ref-helpers.js

@@ -16,6 +16,7 @@ const getTenantScopedRef = (tenantId, ref) => {
 	return `${tenantId}_${ref}`;
 };
 const resolveRef = (db) => async (ref) => {
+	const startTime = Date.now();
 	if (isValidCommitHash(ref)) return {
 		type: "commit",
 		name: ref,
@@ -27,7 +28,20 @@ const resolveRef = (db) => async (ref) => {
 		name: ref,
 		hash: tag.tag_hash
 	};
-	const branch = (await doltListBranches(db)()).find((b) => b.name === ref);
+	const branchQueryStart = Date.now();
+	const branches = await doltListBranches(db)();
+	const branchQueryMs = Date.now() - branchQueryStart;
+	const branch = branches.find((b) => b.name === ref);
+	if (!branch) {
+		const totalMs = Date.now() - startTime;
+		logger.info({
+			ref,
+			branchCount: branches.length,
+			branchQueryMs,
+			totalMs,
+			branchNames: branches.map((b) => b.name)
+		}, "resolveRef: branch not found in dolt_branches result");
+	}
 	if (branch) return {
 		type: "branch",
 		name: ref,

package/dist/dolt/ref-scope.js

@@ -95,7 +95,14 @@ async function withRef(pool, resolvedRef, dataAccessFn, options) {
 		refType: resolvedRef.type,
 		connectionId
 	}, "Acquiring connection for ref scope");
+	const connectStart = Date.now();
 	const connection = await pool.connect();
+	const connectMs = Date.now() - connectStart;
+	if (connectMs > 5e3) logger.info({
+		ref: resolvedRef.name,
+		connectMs,
+		connectionId
+	}, "Slow pool.connect in withRef");
 	let tempBranch = null;
 	try {
 		const db = drizzle(connection, { schema: manage_schema_exports });
@@ -104,10 +111,17 @@ async function withRef(pool, resolvedRef, dataAccessFn, options) {
 				branch: resolvedRef.name,
 				connectionId
 			}, "Checking out branch");
+			const checkoutStart = Date.now();
 			await checkoutBranch(db)({
 				branchName: resolvedRef.name,
 				syncSchema: false
 			});
+			const checkoutMs = Date.now() - checkoutStart;
+			if (checkoutMs > 5e3) logger.info({
+				ref: resolvedRef.name,
+				checkoutMs,
+				connectionId
+			}, "Slow checkoutBranch in withRef");
 		} else {
 			tempBranch = `temp_${resolvedRef.type}_${Date.now()}_${generateId()}`;
 			logger.debug({
@@ -180,6 +194,7 @@ async function withRef(pool, resolvedRef, dataAccessFn, options) {
 		}, "Ref scope failed");
 		throw error;
 	} finally {
+		const cleanupStart = Date.now();
 		try {
 			await connection.query(`SELECT DOLT_CHECKOUT('main')`);
 			if (tempBranch) {
@@ -190,16 +205,29 @@ async function withRef(pool, resolvedRef, dataAccessFn, options) {
 				await connection.query(`SELECT DOLT_BRANCH('-D', $1)`, [tempBranch]);
 			}
 		} catch (cleanupError) {
+			logger.info({
+				ref: resolvedRef.name,
+				connectionId,
+				cleanupMs: Date.now() - cleanupStart,
+				error: cleanupError instanceof Error ? cleanupError.message : String(cleanupError)
+			}, "withRef cleanup failed");
 			logger.error({
 				error: cleanupError,
 				tempBranch,
 				connectionId
 			}, "Error during ref scope cleanup");
 		} finally {
+			const totalMs = Date.now() - startTime;
+			if (totalMs > 5e3) logger.info({
+				ref: resolvedRef.name,
+				totalMs,
+				connectMs,
+				connectionId
+			}, "Slow withRef total duration");
 			connection.release();
 			logger.debug({
 				ref: resolvedRef.name,
-				duration: Date.now() - startTime,
+				duration: totalMs,
 				connectionId
 			}, "Connection released");
 		}