@inkeep/agents-core 0.78.5 → 0.79.1

package/dist/data-access/runtime/scheduledTriggerInvocations.d.ts

@@ -40,7 +40,7 @@ declare const listScheduledTriggerInvocationsPaginated: (db: AgentsRunDatabaseCl
       name: string;
       hash: string;
     } | null;
-    status: "pending" | "failed" | "running" | "completed" | "cancelled";
+    status: "pending" | "running" | "completed" | "failed" | "cancelled";
     scheduledFor: string;
     startedAt: string | null;
     completedAt: string | null;
@@ -199,7 +199,7 @@ declare const listUpcomingInvocationsForAgentPaginated: (db: AgentsRunDatabaseCl
       name: string;
       hash: string;
     } | null;
-    status: "pending" | "failed" | "running" | "completed" | "cancelled";
+    status: "pending" | "running" | "completed" | "failed" | "cancelled";
     scheduledFor: string;
     startedAt: string | null;
     completedAt: string | null;
@@ -239,7 +239,7 @@ declare const listProjectScheduledTriggerInvocationsPaginated: (db: AgentsRunDat
       name: string;
       hash: string;
     } | null;
-    status: "pending" | "failed" | "running" | "completed" | "cancelled";
+    status: "pending" | "running" | "completed" | "failed" | "cancelled";
     scheduledFor: string;
     startedAt: string | null;
     completedAt: string | null;
@@ -292,7 +292,7 @@ declare const listScheduledTriggerInvocationsByTriggerId: (db: AgentsRunDatabase
     name: string;
     hash: string;
   } | null;
-  status: "pending" | "failed" | "running" | "completed" | "cancelled";
+  status: "pending" | "running" | "completed" | "failed" | "cancelled";
   scheduledFor: string;
   startedAt: string | null;
   completedAt: string | null;
@@ -320,7 +320,7 @@ declare const listScheduledTriggerInvocationsByDatasetRunId: (db: AgentsRunDatab
     name: string;
     hash: string;
   } | null;
-  status: "pending" | "failed" | "running" | "completed" | "cancelled";
+  status: "pending" | "running" | "completed" | "failed" | "cancelled";
   scheduledFor: string;
   startedAt: string | null;
   completedAt: string | null;

package/dist/data-access/runtime/tasks.d.ts

@@ -17,9 +17,9 @@ declare const createTask: (db: AgentsRunDatabaseClient) => (params: TaskInsert)
   } | null;
   metadata: TaskMetadataConfig | null;
   status: string;
+  agentId: string;
   projectId: string;
   tenantId: string;
-  agentId: string;
   subAgentId: string;
   contextId: string;
 }>;

package/dist/data-access/runtime/toolApprovalDecisions.d.ts

@@ -0,0 +1,37 @@
+import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
+
+//#region src/data-access/runtime/toolApprovalDecisions.d.ts
+interface ToolApprovalScope {
+  tenantId: string;
+  projectId: string;
+  conversationId: string;
+}
+/**
+ * Persist an approval/denial so the instance running the suspended agent can
+ * pick it up. Used when the approval request lands on a different instance than
+ * the one holding the in-memory pending approval. Upserts so a retry/re-submit
+ * overwrites rather than conflicts.
+ */
+declare const recordToolApprovalDecision: (db: AgentsRunDatabaseClient) => (params: ToolApprovalScope & {
+  toolCallId: string;
+  approved: boolean;
+  reason?: string;
+}) => Promise<void>;
+/**
+ * Atomically read-and-delete a pending decision for a tool call. Returns null
+ * when none is present. The delete-on-read (DELETE ... RETURNING) guarantees a
+ * single waiter consumes any given decision exactly once.
+ */
+declare const consumeToolApprovalDecision: (db: AgentsRunDatabaseClient) => (params: ToolApprovalScope & {
+  toolCallId: string;
+}) => Promise<{
+  approved: boolean;
+  reason: string | null;
+} | null>;
+/**
+ * Delete decisions older than the timeout window. Batched to bound statement
+ * size, mirroring cleanupExpiredStreamChunks.
+ */
+declare const cleanupExpiredToolApprovalDecisions: (db: AgentsRunDatabaseClient) => (olderThanMinutes?: number, batchSize?: number) => Promise<void>;
+//#endregion
+export { cleanupExpiredToolApprovalDecisions, consumeToolApprovalDecision, recordToolApprovalDecision };

package/dist/data-access/runtime/toolApprovalDecisions.js

@@ -0,0 +1,76 @@
+import { toolApprovalDecisions } from "../../db/runtime/runtime-schema.js";
+import { and, eq, sql } from "drizzle-orm";
+
+//#region src/data-access/runtime/toolApprovalDecisions.ts
+const DEFAULT_CLEANUP_AGE_MINUTES = 15;
+const DEFAULT_CLEANUP_BATCH_SIZE = 1e3;
+const decisionConditions = (scope, toolCallId) => [
+	eq(toolApprovalDecisions.tenantId, scope.tenantId),
+	eq(toolApprovalDecisions.projectId, scope.projectId),
+	eq(toolApprovalDecisions.conversationId, scope.conversationId),
+	eq(toolApprovalDecisions.toolCallId, toolCallId)
+];
+/**
+* Persist an approval/denial so the instance running the suspended agent can
+* pick it up. Used when the approval request lands on a different instance than
+* the one holding the in-memory pending approval. Upserts so a retry/re-submit
+* overwrites rather than conflicts.
+*/
+const recordToolApprovalDecision = (db) => async (params) => {
+	await db.insert(toolApprovalDecisions).values({
+		tenantId: params.tenantId,
+		projectId: params.projectId,
+		conversationId: params.conversationId,
+		toolCallId: params.toolCallId,
+		approved: params.approved,
+		reason: params.reason ?? null
+	}).onConflictDoUpdate({
+		target: [
+			toolApprovalDecisions.tenantId,
+			toolApprovalDecisions.projectId,
+			toolApprovalDecisions.conversationId,
+			toolApprovalDecisions.toolCallId
+		],
+		set: {
+			approved: params.approved,
+			reason: params.reason ?? null
+		}
+	});
+};
+/**
+* Atomically read-and-delete a pending decision for a tool call. Returns null
+* when none is present. The delete-on-read (DELETE ... RETURNING) guarantees a
+* single waiter consumes any given decision exactly once.
+*/
+const consumeToolApprovalDecision = (db) => async (params) => {
+	const [row] = await db.delete(toolApprovalDecisions).where(and(...decisionConditions(params, params.toolCallId))).returning();
+	if (!row) return null;
+	return {
+		approved: row.approved,
+		reason: row.reason ?? null
+	};
+};
+/**
+* Delete decisions older than the timeout window. Batched to bound statement
+* size, mirroring cleanupExpiredStreamChunks.
+*/
+const cleanupExpiredToolApprovalDecisions = (db) => async (olderThanMinutes = DEFAULT_CLEANUP_AGE_MINUTES, batchSize = DEFAULT_CLEANUP_BATCH_SIZE) => {
+	const cutoff = sql`now() - make_interval(mins => ${olderThanMinutes})`;
+	let deleted;
+	do {
+		const batch = await db.select({
+			tenantId: toolApprovalDecisions.tenantId,
+			projectId: toolApprovalDecisions.projectId,
+			conversationId: toolApprovalDecisions.conversationId,
+			toolCallId: toolApprovalDecisions.toolCallId
+		}).from(toolApprovalDecisions).where(sql`${toolApprovalDecisions.createdAt} < ${cutoff}`).limit(batchSize);
+		deleted = batch.length;
+		if (deleted > 0) {
+			const pks = batch.map((row) => sql`(${row.tenantId}, ${row.projectId}, ${row.conversationId}, ${row.toolCallId})`);
+			await db.delete(toolApprovalDecisions).where(sql`(${toolApprovalDecisions.tenantId}, ${toolApprovalDecisions.projectId}, ${toolApprovalDecisions.conversationId}, ${toolApprovalDecisions.toolCallId}) IN (${sql.join(pks, sql`, `)})`);
+		}
+	} while (deleted >= batchSize);
+};
+
+//#endregion
+export { cleanupExpiredToolApprovalDecisions, consumeToolApprovalDecision, recordToolApprovalDecision };