@inkeep/agents-core 0.78.3 → 0.78.4

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

@@ -4,7 +4,13 @@ import { and, count, desc, eq, isNotNull, lte } from "drizzle-orm";
 
 //#region src/data-access/runtime/scheduledTriggers.ts
 const getScheduledTriggerById = (db) => async (params) => {
-	return await db.query.scheduledTriggers.findFirst({ where: and(eq(scheduledTriggers.tenantId, params.scopes.tenantId), eq(scheduledTriggers.projectId, params.scopes.projectId), eq(scheduledTriggers.agentId, params.scopes.agentId), eq(scheduledTriggers.id, params.scheduledTriggerId)) });
+	const conditions = [
+		eq(scheduledTriggers.tenantId, params.scopes.tenantId),
+		eq(scheduledTriggers.projectId, params.scopes.projectId),
+		eq(scheduledTriggers.id, params.scheduledTriggerId)
+	];
+	if (params.scopes.agentId) conditions.push(eq(scheduledTriggers.agentId, params.scopes.agentId));
+	return await db.query.scheduledTriggers.findFirst({ where: and(...conditions) });
 };
 const listScheduledTriggersPaginated = (db) => async (params) => {
 	const page = params.pagination?.page || 1;
@@ -29,16 +35,27 @@ const createScheduledTrigger = (db) => async (params) => {
 	return created;
 };
 const updateScheduledTrigger = (db) => async (params) => {
-	const updateData = {
+	const conditions = [
+		eq(scheduledTriggers.tenantId, params.scopes.tenantId),
+		eq(scheduledTriggers.projectId, params.scopes.projectId),
+		eq(scheduledTriggers.id, params.scheduledTriggerId)
+	];
+	if (params.scopes.agentId) conditions.push(eq(scheduledTriggers.agentId, params.scopes.agentId));
+	const updated = (await db.update(scheduledTriggers).set({
 		...params.data,
 		updatedAt: (/* @__PURE__ */ new Date()).toISOString()
-	};
-	const updated = (await db.update(scheduledTriggers).set(updateData).where(and(eq(scheduledTriggers.tenantId, params.scopes.tenantId), eq(scheduledTriggers.projectId, params.scopes.projectId), eq(scheduledTriggers.agentId, params.scopes.agentId), eq(scheduledTriggers.id, params.scheduledTriggerId))).returning())[0];
+	}).where(and(...conditions)).returning())[0];
 	if (!updated) throw new Error(`Scheduled trigger ${params.scheduledTriggerId} not found for update`);
 	return updated;
 };
 const deleteScheduledTrigger = (db) => async (params) => {
-	await db.delete(scheduledTriggers).where(and(eq(scheduledTriggers.tenantId, params.scopes.tenantId), eq(scheduledTriggers.projectId, params.scopes.projectId), eq(scheduledTriggers.agentId, params.scopes.agentId), eq(scheduledTriggers.id, params.scheduledTriggerId)));
+	const conditions = [
+		eq(scheduledTriggers.tenantId, params.scopes.tenantId),
+		eq(scheduledTriggers.projectId, params.scopes.projectId),
+		eq(scheduledTriggers.id, params.scheduledTriggerId)
+	];
+	if (params.scopes.agentId) conditions.push(eq(scheduledTriggers.agentId, params.scopes.agentId));
+	await db.delete(scheduledTriggers).where(and(...conditions));
 };
 const upsertScheduledTrigger = (db) => async (params) => {
 	const nextRunAt = params.data.enabled ?? true ? computeNextRunAt({
@@ -84,8 +101,18 @@ const advanceScheduledTriggerNextRunAt = (db) => async (params) => {
 		updatedAt: (/* @__PURE__ */ new Date()).toISOString()
 	};
 	if (params.enabled !== void 0) set.enabled = params.enabled;
-	await db.update(scheduledTriggers).set(set).where(and(eq(scheduledTriggers.tenantId, params.scopes.tenantId), eq(scheduledTriggers.projectId, params.scopes.projectId), eq(scheduledTriggers.agentId, params.scopes.agentId), eq(scheduledTriggers.id, params.scheduledTriggerId)));
+	const conditions = [
+		eq(scheduledTriggers.tenantId, params.scopes.tenantId),
+		eq(scheduledTriggers.projectId, params.scopes.projectId),
+		eq(scheduledTriggers.id, params.scheduledTriggerId)
+	];
+	if (params.scopes.agentId) conditions.push(eq(scheduledTriggers.agentId, params.scopes.agentId));
+	await db.update(scheduledTriggers).set(set).where(and(...conditions));
+};
+const findScheduledTriggerByDatasetRunConfigId = (db) => async (params) => {
+	const [result] = await db.select().from(scheduledTriggers).where(and(eq(scheduledTriggers.tenantId, params.tenantId), eq(scheduledTriggers.projectId, params.projectId), eq(scheduledTriggers.datasetRunConfigId, params.datasetRunConfigId)));
+	return result;
 };
 
 //#endregion
-export { advanceScheduledTriggerNextRunAt, createScheduledTrigger, deleteScheduledTrigger, deleteScheduledTriggersByRunAsUserId, findDueScheduledTriggersAcrossProjects, getScheduledTriggerById, listScheduledTriggers, listScheduledTriggersPaginated, updateScheduledTrigger, upsertScheduledTrigger };
+export { advanceScheduledTriggerNextRunAt, createScheduledTrigger, deleteScheduledTrigger, deleteScheduledTriggersByRunAsUserId, findDueScheduledTriggersAcrossProjects, findScheduledTriggerByDatasetRunConfigId, getScheduledTriggerById, listScheduledTriggers, listScheduledTriggersPaginated, updateScheduledTrigger, upsertScheduledTrigger };

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

@@ -8,19 +8,19 @@ 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;
   createdAt: string;
-  updatedAt: string;
+  metadata: TaskMetadataConfig | null;
   ref: {
     type: "commit" | "tag" | "branch";
     name: string;
     hash: string;
   } | null;
-  metadata: TaskMetadataConfig | null;
   status: string;
-  tenantId: string;
-  projectId: string;
   agentId: string;
+  projectId: string;
   subAgentId: string;
+  updatedAt: string;
   contextId: string;
 }>;
 declare const getTask: (db: AgentsRunDatabaseClient) => (params: {