@inkeep/agents-core 0.0.0-dev-20260212154015 → 0.0.0-dev-20260212170401

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

@@ -0,0 +1,226 @@
+import { scheduledTriggerInvocations } from "../../db/runtime/runtime-schema.js";
+import { and, asc, count, desc, eq, gte, inArray, lte, ne, sql } from "drizzle-orm";
+
+//#region src/data-access/runtime/scheduledTriggerInvocations.ts
+/**
+* Get a scheduled trigger invocation by ID (agent-scoped)
+*/
+const getScheduledTriggerInvocationById = (db) => async (params) => {
+	return await db.query.scheduledTriggerInvocations.findFirst({ where: and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.id, params.invocationId)) });
+};
+/**
+* Get a scheduled trigger invocation by idempotency key
+* Used to check if an invocation already exists for a given schedule
+*/
+const getScheduledTriggerInvocationByIdempotencyKey = (db) => async (params) => {
+	return await db.query.scheduledTriggerInvocations.findFirst({ where: eq(scheduledTriggerInvocations.idempotencyKey, params.idempotencyKey) });
+};
+/**
+* List scheduled trigger invocations with optional filtering (agent-scoped)
+*/
+const listScheduledTriggerInvocationsPaginated = (db) => async (params) => {
+	const page = params.pagination?.page || 1;
+	const limit = Math.min(params.pagination?.limit || 10, 100);
+	const offset = (page - 1) * limit;
+	const conditions = [
+		eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId),
+		eq(scheduledTriggerInvocations.projectId, params.scopes.projectId),
+		eq(scheduledTriggerInvocations.agentId, params.scopes.agentId),
+		eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId)
+	];
+	if (params.filters?.status) conditions.push(eq(scheduledTriggerInvocations.status, params.filters.status));
+	if (params.filters?.from) conditions.push(gte(scheduledTriggerInvocations.scheduledFor, params.filters.from));
+	if (params.filters?.to) conditions.push(lte(scheduledTriggerInvocations.scheduledFor, params.filters.to));
+	const whereClause = and(...conditions);
+	const [data, totalResult] = await Promise.all([db.select().from(scheduledTriggerInvocations).where(whereClause).limit(limit).offset(offset).orderBy(desc(scheduledTriggerInvocations.scheduledFor)), db.select({ count: count() }).from(scheduledTriggerInvocations).where(whereClause)]);
+	const total = totalResult[0]?.count || 0;
+	return {
+		data,
+		pagination: {
+			page,
+			limit,
+			total,
+			pages: Math.ceil(total / limit)
+		}
+	};
+};
+/**
+* List pending invocations for a trigger, ordered by scheduledFor (earliest first)
+* Used by workflow to get the next invocation to execute
+*/
+const listPendingScheduledTriggerInvocations = (db) => async (params) => {
+	const maxLimit = Math.min(params.limit || 10, 100);
+	return await db.select().from(scheduledTriggerInvocations).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.status, "pending"))).orderBy(asc(scheduledTriggerInvocations.scheduledFor)).limit(maxLimit);
+};
+/**
+* Delete all pending invocations for a trigger
+* Used when cron expression changes or trigger is disabled
+*/
+const deletePendingInvocationsForTrigger = (db) => async (params) => {
+	return (await db.delete(scheduledTriggerInvocations).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.status, "pending"))).returning()).length;
+};
+/**
+* Create a new scheduled trigger invocation (agent-scoped)
+*/
+const createScheduledTriggerInvocation = (db) => async (params) => {
+	return (await db.insert(scheduledTriggerInvocations).values(params).returning())[0];
+};
+/**
+* Update scheduled trigger invocation status (agent-scoped)
+*/
+const updateScheduledTriggerInvocationStatus = (db) => async (params) => {
+	return (await db.update(scheduledTriggerInvocations).set(params.data).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.id, params.invocationId))).returning())[0];
+};
+/**
+* Mark invocation as running
+*/
+const markScheduledTriggerInvocationRunning = (db) => async (params) => {
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	return (await db.update(scheduledTriggerInvocations).set({
+		status: "running",
+		startedAt: sql`COALESCE(${scheduledTriggerInvocations.startedAt}, ${now})`
+	}).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.id, params.invocationId))).returning())[0];
+};
+/**
+* Mark invocation as completed
+* Note: Will not update if status is already 'cancelled' to respect user cancellation
+*/
+const markScheduledTriggerInvocationCompleted = (db) => async (params) => {
+	return (await db.update(scheduledTriggerInvocations).set({
+		status: "completed",
+		completedAt: (/* @__PURE__ */ new Date()).toISOString()
+	}).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.id, params.invocationId), ne(scheduledTriggerInvocations.status, "cancelled"))).returning())[0];
+};
+/**
+* Mark invocation as failed
+* Note: Will not update if status is already 'cancelled' to respect user cancellation
+*/
+const markScheduledTriggerInvocationFailed = (db) => async (params) => {
+	return (await db.update(scheduledTriggerInvocations).set({
+		status: "failed",
+		completedAt: (/* @__PURE__ */ new Date()).toISOString()
+	}).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.id, params.invocationId), ne(scheduledTriggerInvocations.status, "cancelled"))).returning())[0];
+};
+/**
+* Add a conversation ID to the invocation's conversationIds array
+* Used to track all conversations created during retries
+*/
+const addConversationIdToInvocation = (db) => async (params) => {
+	return (await db.update(scheduledTriggerInvocations).set({ conversationIds: sql`${scheduledTriggerInvocations.conversationIds} || ${JSON.stringify([params.conversationId])}::jsonb` }).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.id, params.invocationId))).returning())[0];
+};
+/**
+* Mark invocation as cancelled
+*/
+const markScheduledTriggerInvocationCancelled = (db) => async (params) => {
+	return (await db.update(scheduledTriggerInvocations).set({
+		status: "cancelled",
+		completedAt: (/* @__PURE__ */ new Date()).toISOString()
+	}).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.id, params.invocationId))).returning())[0];
+};
+/**
+* Cancel all pending invocations for a trigger
+* Used when a trigger is deleted
+*/
+const cancelPendingInvocationsForTrigger = (db) => async (params) => {
+	return (await db.update(scheduledTriggerInvocations).set({
+		status: "cancelled",
+		completedAt: (/* @__PURE__ */ new Date()).toISOString()
+	}).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), inArray(scheduledTriggerInvocations.status, ["pending", "running"]))).returning()).length;
+};
+/**
+* Cancel only PAST pending invocations for a trigger (scheduledFor <= now)
+* Used when a trigger is disabled - keeps future invocations pending
+*/
+const cancelPastPendingInvocationsForTrigger = (db) => async (params) => {
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	return (await db.update(scheduledTriggerInvocations).set({
+		status: "cancelled",
+		completedAt: now
+	}).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), inArray(scheduledTriggerInvocations.status, ["pending", "running"]), lte(scheduledTriggerInvocations.scheduledFor, now))).returning()).length;
+};
+/**
+* Get run info for multiple scheduled triggers in a single query
+* Returns last run (completed/failed) and next pending run for each trigger
+*/
+const getScheduledTriggerRunInfoBatch = (db) => async (params) => {
+	if (params.triggerIds.length === 0) return /* @__PURE__ */ new Map();
+	const { tenantId, projectId } = params.scopes;
+	const allInvocations = await db.select({
+		scheduledTriggerId: scheduledTriggerInvocations.scheduledTriggerId,
+		status: scheduledTriggerInvocations.status,
+		scheduledFor: scheduledTriggerInvocations.scheduledFor,
+		completedAt: scheduledTriggerInvocations.completedAt,
+		conversationIds: scheduledTriggerInvocations.conversationIds
+	}).from(scheduledTriggerInvocations).where(and(eq(scheduledTriggerInvocations.tenantId, tenantId), eq(scheduledTriggerInvocations.projectId, projectId), inArray(scheduledTriggerInvocations.scheduledTriggerId, params.triggerIds.map((t) => t.triggerId)))).orderBy(desc(scheduledTriggerInvocations.completedAt));
+	const result = /* @__PURE__ */ new Map();
+	for (const trigger of params.triggerIds) result.set(trigger.triggerId, {
+		lastRunAt: null,
+		lastRunStatus: null,
+		lastRunConversationIds: [],
+		nextRunAt: null
+	});
+	for (const inv of allInvocations) {
+		const triggerInfo = result.get(inv.scheduledTriggerId);
+		if (!triggerInfo) continue;
+		if (inv.status === "pending" && !triggerInfo.nextRunAt) triggerInfo.nextRunAt = inv.scheduledFor;
+		if ((inv.status === "completed" || inv.status === "failed") && !triggerInfo.lastRunAt) {
+			triggerInfo.lastRunAt = inv.completedAt;
+			triggerInfo.lastRunStatus = inv.status;
+			triggerInfo.lastRunConversationIds = inv.conversationIds || [];
+		}
+	}
+	return result;
+};
+/**
+* List upcoming invocations across ALL triggers for an agent with pagination
+* Used for the upcoming runs dashboard with full pagination support
+*/
+const listUpcomingInvocationsForAgentPaginated = (db) => async (params) => {
+	const page = params.pagination?.page || 1;
+	const limit = Math.min(params.pagination?.limit || 20, 100);
+	const offset = (page - 1) * limit;
+	const statusCondition = params.includeRunning ? inArray(scheduledTriggerInvocations.status, ["pending", "running"]) : eq(scheduledTriggerInvocations.status, "pending");
+	const whereClause = and(...[
+		eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId),
+		eq(scheduledTriggerInvocations.projectId, params.scopes.projectId),
+		eq(scheduledTriggerInvocations.agentId, params.scopes.agentId),
+		statusCondition
+	]);
+	const [data, totalResult] = await Promise.all([db.select().from(scheduledTriggerInvocations).where(whereClause).limit(limit).offset(offset).orderBy(asc(scheduledTriggerInvocations.scheduledFor)), db.select({ count: count() }).from(scheduledTriggerInvocations).where(whereClause)]);
+	const total = totalResult[0]?.count || 0;
+	return {
+		data,
+		pagination: {
+			page,
+			limit,
+			total,
+			pages: Math.ceil(total / limit)
+		}
+	};
+};
+/**
+* List all invocations across ALL triggers for a PROJECT with pagination
+* Used for the project-level invocations dashboard
+*/
+const listProjectScheduledTriggerInvocationsPaginated = (db) => async (params) => {
+	const page = params.pagination?.page || 1;
+	const limit = Math.min(params.pagination?.limit || 20, 100);
+	const offset = (page - 1) * limit;
+	const conditions = [eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId)];
+	if (params.filters?.status) conditions.push(eq(scheduledTriggerInvocations.status, params.filters.status));
+	const whereClause = and(...conditions);
+	const [data, totalResult] = await Promise.all([db.select().from(scheduledTriggerInvocations).where(whereClause).limit(limit).offset(offset).orderBy(desc(scheduledTriggerInvocations.createdAt)), db.select({ count: count() }).from(scheduledTriggerInvocations).where(whereClause)]);
+	const total = totalResult[0]?.count || 0;
+	return {
+		data,
+		pagination: {
+			page,
+			limit,
+			total,
+			pages: Math.ceil(total / limit)
+		}
+	};
+};
+
+//#endregion
+export { addConversationIdToInvocation, cancelPastPendingInvocationsForTrigger, cancelPendingInvocationsForTrigger, createScheduledTriggerInvocation, deletePendingInvocationsForTrigger, getScheduledTriggerInvocationById, getScheduledTriggerInvocationByIdempotencyKey, getScheduledTriggerRunInfoBatch, listPendingScheduledTriggerInvocations, listProjectScheduledTriggerInvocationsPaginated, listScheduledTriggerInvocationsPaginated, listUpcomingInvocationsForAgentPaginated, markScheduledTriggerInvocationCancelled, markScheduledTriggerInvocationCompleted, markScheduledTriggerInvocationFailed, markScheduledTriggerInvocationRunning, updateScheduledTriggerInvocationStatus };

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

@@ -7,20 +7,20 @@ import { TaskInsert, TaskSelect } from "../../types/entities.js";
 //#region src/data-access/runtime/tasks.d.ts
 declare const createTask: (db: AgentsRunDatabaseClient) => (params: TaskInsert) => Promise<{
   id: string;
-  tenantId: string;
-  projectId: string;
-  agentId: string;
   createdAt: string;
   updatedAt: string;
-  metadata: TaskMetadataConfig | null;
-  status: string;
   ref: {
-    type: "commit" | "tag" | "branch";
+    type: "tag" | "commit" | "branch";
     name: string;
     hash: string;
   } | null;
-  subAgentId: string;
+  metadata: TaskMetadataConfig | null;
+  status: string;
+  agentId: string;
+  projectId: string;
+  tenantId: string;
   contextId: string;
+  subAgentId: string;
 }>;
 declare const getTask: (db: AgentsRunDatabaseClient) => (params: {
   id: string;
@@ -36,7 +36,7 @@ declare const updateTask: (db: AgentsRunDatabaseClient) => (params: {
   updatedAt: string;
   contextId: string;
   ref: {
-    type: "commit" | "tag" | "branch";
+    type: "tag" | "commit" | "branch";
     name: string;
     hash: string;
   } | null;