@inkeep/agents-core 0.0.0-dev-20260331192950 → 0.0.0-dev-20260331193342

package/dist/data-access/manage/evalConfig.d.ts

@@ -2,7 +2,7 @@ import { ProjectScopeConfig } from "../../db/manage/scope-definitions.js";
 import "../../types/utility.js";
 import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
 import { AgentsManageDatabaseClient } from "../../db/manage/manage-client.js";
-import { DatasetInsert, DatasetItemInsert, DatasetItemSelect, DatasetItemUpdate, DatasetRunConfigAgentRelationInsert, DatasetRunConfigAgentRelationSelect, DatasetRunConfigInsert, DatasetRunConfigSelect, DatasetRunConfigUpdate, DatasetSelect, DatasetUpdate, EvaluationJobConfigEvaluatorRelationInsert, EvaluationJobConfigEvaluatorRelationSelect, EvaluationJobConfigInsert, EvaluationJobConfigSelect, EvaluationRunConfigEvaluationSuiteConfigRelationInsert, EvaluationRunConfigEvaluationSuiteConfigRelationSelect, EvaluationRunConfigInsert, EvaluationRunConfigSelect, EvaluationRunConfigUpdate, EvaluationRunConfigWithSuiteConfigs, EvaluationSuiteConfigEvaluatorRelationInsert, EvaluationSuiteConfigEvaluatorRelationSelect, EvaluationSuiteConfigInsert, EvaluationSuiteConfigSelect, EvaluationSuiteConfigUpdate, EvaluatorInsert, EvaluatorSelect, EvaluatorUpdate } from "../../types/entities.js";
+import { AgentDatasetRelationInsert, AgentDatasetRelationSelect, AgentEvaluatorRelationInsert, AgentEvaluatorRelationSelect, DatasetInsert, DatasetItemInsert, DatasetItemSelect, DatasetItemUpdate, DatasetRunConfigAgentRelationInsert, DatasetRunConfigAgentRelationSelect, DatasetRunConfigInsert, DatasetRunConfigSelect, DatasetRunConfigUpdate, DatasetSelect, DatasetUpdate, EvaluationJobConfigEvaluatorRelationInsert, EvaluationJobConfigEvaluatorRelationSelect, EvaluationJobConfigInsert, EvaluationJobConfigSelect, EvaluationRunConfigEvaluationSuiteConfigRelationInsert, EvaluationRunConfigEvaluationSuiteConfigRelationSelect, EvaluationRunConfigInsert, EvaluationRunConfigSelect, EvaluationRunConfigUpdate, EvaluationRunConfigWithSuiteConfigs, EvaluationSuiteConfigEvaluatorRelationInsert, EvaluationSuiteConfigEvaluatorRelationSelect, EvaluationSuiteConfigInsert, EvaluationSuiteConfigSelect, EvaluationSuiteConfigUpdate, EvaluatorInsert, EvaluatorSelect, EvaluatorUpdate } from "../../types/entities.js";
 
 //#region src/data-access/manage/evalConfig.d.ts
 declare const getDatasetById: (db: AgentsManageDatabaseClient) => (params: {
@@ -218,5 +218,51 @@ declare const linkDatasetRunToEvaluationJobConfig: (db: AgentsRunDatabaseClient)
   };
   evaluationJobConfigId: string;
 }) => Promise<void>;
+declare const getAgentDatasetRelationsByAgent: (db: AgentsManageDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    agentId: string;
+  };
+}) => Promise<AgentDatasetRelationSelect[]>;
+declare const getAgentDatasetRelationsByDataset: (db: AgentsManageDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    datasetId: string;
+  };
+}) => Promise<AgentDatasetRelationSelect[]>;
+declare const createAgentDatasetRelation: (db: AgentsManageDatabaseClient) => (data: AgentDatasetRelationInsert) => Promise<AgentDatasetRelationSelect>;
+declare const deleteAgentDatasetRelation: (db: AgentsManageDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    agentId: string;
+    datasetId: string;
+  };
+}) => Promise<boolean>;
+declare const getAgentEvaluatorRelationsByAgent: (db: AgentsManageDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    agentId: string;
+  };
+}) => Promise<AgentEvaluatorRelationSelect[]>;
+declare const getAgentEvaluatorRelationsByEvaluator: (db: AgentsManageDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    evaluatorId: string;
+  };
+}) => Promise<AgentEvaluatorRelationSelect[]>;
+declare const createAgentEvaluatorRelation: (db: AgentsManageDatabaseClient) => (data: AgentEvaluatorRelationInsert) => Promise<AgentEvaluatorRelationSelect>;
+declare const deleteAgentEvaluatorRelation: (db: AgentsManageDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    agentId: string;
+    evaluatorId: string;
+  };
+}) => Promise<boolean>;
+declare const listDatasetsForAgent: (db: AgentsManageDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+  agentId: string;
+}) => Promise<DatasetSelect[]>;
+declare const listEvaluatorsForAgent: (db: AgentsManageDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+  agentId: string;
+}) => Promise<EvaluatorSelect[]>;
+declare const getAgentIdsForEvaluators: (db: AgentsManageDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+  evaluatorIds: string[];
+}) => Promise<Map<string, string[]>>;
 //#endregion
-export { createDataset, createDatasetItem, createDatasetItems, createDatasetRunConfig, createDatasetRunConfigAgentRelation, createEvaluationJobConfig, createEvaluationJobConfigEvaluatorRelation, createEvaluationRunConfig, createEvaluationRunConfigEvaluationSuiteConfigRelation, createEvaluationSuiteConfig, createEvaluationSuiteConfigEvaluatorRelation, createEvaluator, deleteDataset, deleteDatasetItem, deleteDatasetItemsByDataset, deleteDatasetRunConfig, deleteDatasetRunConfigAgentRelation, deleteEvaluationJobConfig, deleteEvaluationJobConfigEvaluatorRelation, deleteEvaluationJobConfigEvaluatorRelationsByEvaluator, deleteEvaluationRunConfig, deleteEvaluationRunConfigEvaluationSuiteConfigRelation, deleteEvaluationSuiteConfig, deleteEvaluationSuiteConfigEvaluatorRelation, deleteEvaluationSuiteConfigEvaluatorRelationsByEvaluator, deleteEvaluator, getDatasetById, getDatasetItemById, getDatasetRunConfigAgentRelations, getDatasetRunConfigById, getEvaluationJobConfigById, getEvaluationJobConfigEvaluatorRelations, getEvaluationRunConfigById, getEvaluationRunConfigEvaluationSuiteConfigRelations, getEvaluationSuiteConfigById, getEvaluationSuiteConfigEvaluatorRelations, getEvaluatorById, getEvaluatorsByIds, linkDatasetRunToEvaluationJobConfig, listDatasetItems, listDatasetRunConfigs, listDatasets, listEvaluationJobConfigs, listEvaluationRunConfigs, listEvaluationRunConfigsWithSuiteConfigs, listEvaluationSuiteConfigs, listEvaluators, updateDataset, updateDatasetItem, updateDatasetRunConfig, updateEvaluationRunConfig, updateEvaluationSuiteConfig, updateEvaluator };
+export { createAgentDatasetRelation, createAgentEvaluatorRelation, createDataset, createDatasetItem, createDatasetItems, createDatasetRunConfig, createDatasetRunConfigAgentRelation, createEvaluationJobConfig, createEvaluationJobConfigEvaluatorRelation, createEvaluationRunConfig, createEvaluationRunConfigEvaluationSuiteConfigRelation, createEvaluationSuiteConfig, createEvaluationSuiteConfigEvaluatorRelation, createEvaluator, deleteAgentDatasetRelation, deleteAgentEvaluatorRelation, deleteDataset, deleteDatasetItem, deleteDatasetItemsByDataset, deleteDatasetRunConfig, deleteDatasetRunConfigAgentRelation, deleteEvaluationJobConfig, deleteEvaluationJobConfigEvaluatorRelation, deleteEvaluationJobConfigEvaluatorRelationsByEvaluator, deleteEvaluationRunConfig, deleteEvaluationRunConfigEvaluationSuiteConfigRelation, deleteEvaluationSuiteConfig, deleteEvaluationSuiteConfigEvaluatorRelation, deleteEvaluationSuiteConfigEvaluatorRelationsByEvaluator, deleteEvaluator, getAgentDatasetRelationsByAgent, getAgentDatasetRelationsByDataset, getAgentEvaluatorRelationsByAgent, getAgentEvaluatorRelationsByEvaluator, getAgentIdsForEvaluators, getDatasetById, getDatasetItemById, getDatasetRunConfigAgentRelations, getDatasetRunConfigById, getEvaluationJobConfigById, getEvaluationJobConfigEvaluatorRelations, getEvaluationRunConfigById, getEvaluationRunConfigEvaluationSuiteConfigRelations, getEvaluationSuiteConfigById, getEvaluationSuiteConfigEvaluatorRelations, getEvaluatorById, getEvaluatorsByIds, linkDatasetRunToEvaluationJobConfig, listDatasetItems, listDatasetRunConfigs, listDatasets, listDatasetsForAgent, listEvaluationJobConfigs, listEvaluationRunConfigs, listEvaluationRunConfigsWithSuiteConfigs, listEvaluationSuiteConfigs, listEvaluators, listEvaluatorsForAgent, updateDataset, updateDatasetItem, updateDatasetRunConfig, updateEvaluationRunConfig, updateEvaluationSuiteConfig, updateEvaluator };

package/dist/data-access/manage/evalConfig.js

@@ -1,7 +1,7 @@
-import { dataset, datasetItem, datasetRunConfig, datasetRunConfigAgentRelations, evaluationJobConfig, evaluationJobConfigEvaluatorRelations, evaluationRunConfig, evaluationRunConfigEvaluationSuiteConfigRelations, evaluationSuiteConfig, evaluationSuiteConfigEvaluatorRelations, evaluator } from "../../db/manage/manage-schema.js";
+import { agentDatasetRelations, agentEvaluatorRelations, dataset, datasetItem, datasetRunConfig, datasetRunConfigAgentRelations, evaluationJobConfig, evaluationJobConfigEvaluatorRelations, evaluationRunConfig, evaluationRunConfigEvaluationSuiteConfigRelations, evaluationSuiteConfig, evaluationSuiteConfigEvaluatorRelations, evaluator } from "../../db/manage/manage-schema.js";
 import { datasetRun } from "../../db/runtime/runtime-schema.js";
 import { projectScopedWhere } from "./scope-helpers.js";
-import { and, eq, inArray } from "drizzle-orm";
+import { and, eq, inArray, isNull, or } from "drizzle-orm";
 
 //#region src/data-access/manage/evalConfig.ts
 const getDatasetById = (db) => async (params) => {
@@ -271,6 +271,64 @@ const deleteEvaluationJobConfigEvaluatorRelationsByEvaluator = (db) => async (pa
 const linkDatasetRunToEvaluationJobConfig = (db) => async (params) => {
 	await db.update(datasetRun).set({ evaluationJobConfigId: params.evaluationJobConfigId }).where(and(projectScopedWhere(datasetRun, params.scopes), eq(datasetRun.id, params.scopes.datasetRunId)));
 };
+const getAgentDatasetRelationsByAgent = (db) => async (params) => {
+	return await db.select().from(agentDatasetRelations).where(and(projectScopedWhere(agentDatasetRelations, params.scopes), eq(agentDatasetRelations.agentId, params.scopes.agentId)));
+};
+const getAgentDatasetRelationsByDataset = (db) => async (params) => {
+	return await db.select().from(agentDatasetRelations).where(and(projectScopedWhere(agentDatasetRelations, params.scopes), eq(agentDatasetRelations.datasetId, params.scopes.datasetId)));
+};
+const createAgentDatasetRelation = (db) => async (data) => {
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	const [created] = await db.insert(agentDatasetRelations).values({
+		...data,
+		createdAt: now,
+		updatedAt: now
+	}).returning();
+	return created;
+};
+const deleteAgentDatasetRelation = (db) => async (params) => {
+	return (await db.delete(agentDatasetRelations).where(and(projectScopedWhere(agentDatasetRelations, params.scopes), eq(agentDatasetRelations.agentId, params.scopes.agentId), eq(agentDatasetRelations.datasetId, params.scopes.datasetId))).returning()).length > 0;
+};
+const getAgentEvaluatorRelationsByAgent = (db) => async (params) => {
+	return await db.select().from(agentEvaluatorRelations).where(and(projectScopedWhere(agentEvaluatorRelations, params.scopes), eq(agentEvaluatorRelations.agentId, params.scopes.agentId)));
+};
+const getAgentEvaluatorRelationsByEvaluator = (db) => async (params) => {
+	return await db.select().from(agentEvaluatorRelations).where(and(projectScopedWhere(agentEvaluatorRelations, params.scopes), eq(agentEvaluatorRelations.evaluatorId, params.scopes.evaluatorId)));
+};
+const createAgentEvaluatorRelation = (db) => async (data) => {
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	const [created] = await db.insert(agentEvaluatorRelations).values({
+		...data,
+		createdAt: now,
+		updatedAt: now
+	}).returning();
+	return created;
+};
+const deleteAgentEvaluatorRelation = (db) => async (params) => {
+	return (await db.delete(agentEvaluatorRelations).where(and(projectScopedWhere(agentEvaluatorRelations, params.scopes), eq(agentEvaluatorRelations.agentId, params.scopes.agentId), eq(agentEvaluatorRelations.evaluatorId, params.scopes.evaluatorId))).returning()).length > 0;
+};
+const listDatasetsForAgent = (db) => async (params) => {
+	const { scopes, agentId } = params;
+	return (await db.select({ dataset }).from(dataset).leftJoin(agentDatasetRelations, and(eq(dataset.tenantId, agentDatasetRelations.tenantId), eq(dataset.projectId, agentDatasetRelations.projectId), eq(dataset.id, agentDatasetRelations.datasetId))).where(and(projectScopedWhere(dataset, scopes), or(isNull(agentDatasetRelations.agentId), eq(agentDatasetRelations.agentId, agentId))))).map((r) => r.dataset);
+};
+const listEvaluatorsForAgent = (db) => async (params) => {
+	const { scopes, agentId } = params;
+	return (await db.select({ evaluator }).from(evaluator).leftJoin(agentEvaluatorRelations, and(eq(evaluator.tenantId, agentEvaluatorRelations.tenantId), eq(evaluator.projectId, agentEvaluatorRelations.projectId), eq(evaluator.id, agentEvaluatorRelations.evaluatorId))).where(and(projectScopedWhere(evaluator, scopes), or(isNull(agentEvaluatorRelations.agentId), eq(agentEvaluatorRelations.agentId, agentId))))).map((r) => r.evaluator);
+};
+const getAgentIdsForEvaluators = (db) => async (params) => {
+	if (params.evaluatorIds.length === 0) return /* @__PURE__ */ new Map();
+	const rows = await db.select({
+		evaluatorId: agentEvaluatorRelations.evaluatorId,
+		agentId: agentEvaluatorRelations.agentId
+	}).from(agentEvaluatorRelations).where(and(projectScopedWhere(agentEvaluatorRelations, params.scopes), inArray(agentEvaluatorRelations.evaluatorId, params.evaluatorIds)));
+	const result = /* @__PURE__ */ new Map();
+	for (const row of rows) {
+		const existing = result.get(row.evaluatorId) ?? [];
+		existing.push(row.agentId);
+		result.set(row.evaluatorId, existing);
+	}
+	return result;
+};
 
 //#endregion
-export { createDataset, createDatasetItem, createDatasetItems, createDatasetRunConfig, createDatasetRunConfigAgentRelation, createEvaluationJobConfig, createEvaluationJobConfigEvaluatorRelation, createEvaluationRunConfig, createEvaluationRunConfigEvaluationSuiteConfigRelation, createEvaluationSuiteConfig, createEvaluationSuiteConfigEvaluatorRelation, createEvaluator, deleteDataset, deleteDatasetItem, deleteDatasetItemsByDataset, deleteDatasetRunConfig, deleteDatasetRunConfigAgentRelation, deleteEvaluationJobConfig, deleteEvaluationJobConfigEvaluatorRelation, deleteEvaluationJobConfigEvaluatorRelationsByEvaluator, deleteEvaluationRunConfig, deleteEvaluationRunConfigEvaluationSuiteConfigRelation, deleteEvaluationSuiteConfig, deleteEvaluationSuiteConfigEvaluatorRelation, deleteEvaluationSuiteConfigEvaluatorRelationsByEvaluator, deleteEvaluator, getDatasetById, getDatasetItemById, getDatasetRunConfigAgentRelations, getDatasetRunConfigById, getEvaluationJobConfigById, getEvaluationJobConfigEvaluatorRelations, getEvaluationRunConfigById, getEvaluationRunConfigEvaluationSuiteConfigRelations, getEvaluationSuiteConfigById, getEvaluationSuiteConfigEvaluatorRelations, getEvaluatorById, getEvaluatorsByIds, linkDatasetRunToEvaluationJobConfig, listDatasetItems, listDatasetRunConfigs, listDatasets, listEvaluationJobConfigs, listEvaluationRunConfigs, listEvaluationRunConfigsWithSuiteConfigs, listEvaluationSuiteConfigs, listEvaluators, updateDataset, updateDatasetItem, updateDatasetRunConfig, updateEvaluationRunConfig, updateEvaluationSuiteConfig, updateEvaluator };
+export { createAgentDatasetRelation, createAgentEvaluatorRelation, createDataset, createDatasetItem, createDatasetItems, createDatasetRunConfig, createDatasetRunConfigAgentRelation, createEvaluationJobConfig, createEvaluationJobConfigEvaluatorRelation, createEvaluationRunConfig, createEvaluationRunConfigEvaluationSuiteConfigRelation, createEvaluationSuiteConfig, createEvaluationSuiteConfigEvaluatorRelation, createEvaluator, deleteAgentDatasetRelation, deleteAgentEvaluatorRelation, deleteDataset, deleteDatasetItem, deleteDatasetItemsByDataset, deleteDatasetRunConfig, deleteDatasetRunConfigAgentRelation, deleteEvaluationJobConfig, deleteEvaluationJobConfigEvaluatorRelation, deleteEvaluationJobConfigEvaluatorRelationsByEvaluator, deleteEvaluationRunConfig, deleteEvaluationRunConfigEvaluationSuiteConfigRelation, deleteEvaluationSuiteConfig, deleteEvaluationSuiteConfigEvaluatorRelation, deleteEvaluationSuiteConfigEvaluatorRelationsByEvaluator, deleteEvaluator, getAgentDatasetRelationsByAgent, getAgentDatasetRelationsByDataset, getAgentEvaluatorRelationsByAgent, getAgentEvaluatorRelationsByEvaluator, getAgentIdsForEvaluators, getDatasetById, getDatasetItemById, getDatasetRunConfigAgentRelations, getDatasetRunConfigById, getEvaluationJobConfigById, getEvaluationJobConfigEvaluatorRelations, getEvaluationRunConfigById, getEvaluationRunConfigEvaluationSuiteConfigRelations, getEvaluationSuiteConfigById, getEvaluationSuiteConfigEvaluatorRelations, getEvaluatorById, getEvaluatorsByIds, linkDatasetRunToEvaluationJobConfig, listDatasetItems, listDatasetRunConfigs, listDatasets, listDatasetsForAgent, listEvaluationJobConfigs, listEvaluationRunConfigs, listEvaluationRunConfigsWithSuiteConfigs, listEvaluationSuiteConfigs, listEvaluators, listEvaluatorsForAgent, updateDataset, updateDatasetItem, updateDatasetRunConfig, updateEvaluationRunConfig, updateEvaluationSuiteConfig, updateEvaluator };

package/dist/data-access/manage/skills.d.ts

@@ -145,8 +145,8 @@ declare const upsertSubAgentSkill: (db: AgentsManageDatabaseClient) => (params:
   agentId: string;
   projectId: string;
   tenantId: string;
-  skillId: string;
   subAgentId: string;
+  skillId: string;
   index: number;
   alwaysLoaded: boolean;
 }>;

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

@@ -15,11 +15,11 @@ declare const getApiKeyById: (db: AgentsRunDatabaseClient) => (params: {
   agentId: string;
   projectId: string;
   tenantId: string;
-  expiresAt: string | null;
   publicId: string;
   keyHash: string;
   keyPrefix: string;
   lastUsedAt: string | null;
+  expiresAt: string | null;
 } | undefined>;
 declare const getApiKeyByPublicId: (db: AgentsRunDatabaseClient) => (publicId: string) => Promise<{
   id: string;
@@ -29,11 +29,11 @@ declare const getApiKeyByPublicId: (db: AgentsRunDatabaseClient) => (publicId: s
   agentId: string;
   projectId: string;
   tenantId: string;
-  expiresAt: string | null;
   publicId: string;
   keyHash: string;
   keyPrefix: string;
   lastUsedAt: string | null;
+  expiresAt: string | null;
 } | undefined>;
 declare const listApiKeys: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
@@ -46,11 +46,11 @@ declare const listApiKeys: (db: AgentsRunDatabaseClient) => (params: {
   agentId: string;
   projectId: string;
   tenantId: string;
-  expiresAt: string | null;
   publicId: string;
   keyHash: string;
   keyPrefix: string;
   lastUsedAt: string | null;
+  expiresAt: string | null;
 }[]>;
 declare const listApiKeysPaginated: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
@@ -73,11 +73,11 @@ declare const createApiKey: (db: AgentsRunDatabaseClient) => (params: ApiKeyInse
   agentId: string;
   projectId: string;
   tenantId: string;
-  expiresAt: string | null;
   publicId: string;
   keyHash: string;
   keyPrefix: string;
   lastUsedAt: string | null;
+  expiresAt: string | null;
 }>;
 declare const updateApiKey: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;

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

@@ -17,7 +17,6 @@ declare const getMessageById: (db: AgentsRunDatabaseClient) => (params: {
   tenantId: string;
   metadata: MessageMetadata | null;
   content: MessageContent;
-  role: string;
   fromSubAgentId: string | null;
   toSubAgentId: string | null;
   fromExternalAgentId: string | null;
@@ -25,6 +24,7 @@ declare const getMessageById: (db: AgentsRunDatabaseClient) => (params: {
   taskId: string | null;
   a2aTaskId: string | null;
   conversationId: string;
+  role: string;
   fromTeamAgentId: string | null;
   toTeamAgentId: string | null;
   visibility: string;
@@ -151,7 +151,6 @@ declare const createMessage: (db: AgentsRunDatabaseClient) => (params: {
   tenantId: string;
   metadata: MessageMetadata | null;
   content: MessageContent;
-  role: string;
   fromSubAgentId: string | null;
   toSubAgentId: string | null;
   fromExternalAgentId: string | null;
@@ -159,6 +158,7 @@ declare const createMessage: (db: AgentsRunDatabaseClient) => (params: {
   taskId: string | null;
   a2aTaskId: string | null;
   conversationId: string;
+  role: string;
   fromTeamAgentId: string | null;
   toTeamAgentId: string | null;
   visibility: string;
@@ -204,7 +204,6 @@ declare const deleteMessage: (db: AgentsRunDatabaseClient) => (params: {
   tenantId: string;
   metadata: MessageMetadata | null;
   content: MessageContent;
-  role: string;
   fromSubAgentId: string | null;
   toSubAgentId: string | null;
   fromExternalAgentId: string | null;
@@ -212,6 +211,7 @@ declare const deleteMessage: (db: AgentsRunDatabaseClient) => (params: {
   taskId: string | null;
   a2aTaskId: string | null;
   conversationId: string;
+  role: string;
   fromTeamAgentId: string | null;
   toTeamAgentId: string | null;
   visibility: string;

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

@@ -1,4 +1,4 @@
-import { AgentScopeConfig } from "../../db/manage/scope-definitions.js";
+import { AgentScopeConfig, ProjectScopeConfig } from "../../db/manage/scope-definitions.js";
 import { PaginationConfig } from "../../types/utility.js";
 import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
 import { ScheduledTriggerInvocationStatus } from "../../validation/schemas.js";
@@ -254,5 +254,50 @@ declare const listProjectScheduledTriggerInvocationsPaginated: (db: AgentsRunDat
     pages: number;
   };
 }>;
+/**
+ * Get status summary for all invocations with a given scheduledTriggerId.
+ * Used for dataset run status aggregation where scheduledTriggerId = datasetRunId.
+ */
+declare const getScheduledTriggerInvocationStatusSummary: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+  scheduledTriggerId: string;
+}) => Promise<{
+  pending: number;
+  running: number;
+  completed: number;
+  failed: number;
+  cancelled: number;
+}>;
+/**
+ * List invocations by scheduledTriggerId without agentId filter (project-scoped).
+ * Used for dataset run items listing where scheduledTriggerId = datasetRunId.
+ */
+declare const listScheduledTriggerInvocationsByTriggerId: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+  scheduledTriggerId: string;
+  filters?: {
+    status?: string;
+  };
+}) => Promise<{
+  scheduledTriggerId: string;
+  ref: {
+    type: "commit" | "tag" | "branch";
+    name: string;
+    hash: string;
+  } | null;
+  status: "pending" | "failed" | "running" | "completed" | "cancelled";
+  scheduledFor: string;
+  startedAt: string | null;
+  completedAt: string | null;
+  resolvedPayload: Record<string, unknown> | null;
+  conversationIds: string[] | null;
+  attemptNumber: number;
+  idempotencyKey: string;
+  createdAt: string;
+  agentId: string;
+  projectId: string;
+  tenantId: string;
+  id: string;
+}[]>;
 //#endregion
-export { addConversationIdToInvocation, cancelPastPendingInvocationsForTrigger, cancelPendingInvocationsForTrigger, createScheduledTriggerInvocation, deletePendingInvocationsForTrigger, getScheduledTriggerInvocationById, getScheduledTriggerInvocationByIdempotencyKey, getScheduledTriggerRunInfoBatch, listPendingScheduledTriggerInvocations, listProjectScheduledTriggerInvocationsPaginated, listScheduledTriggerInvocationsPaginated, listUpcomingInvocationsForAgentPaginated, markScheduledTriggerInvocationCancelled, markScheduledTriggerInvocationCompleted, markScheduledTriggerInvocationFailed, markScheduledTriggerInvocationRunning, resetCancelledInvocationToPending, updateScheduledTriggerInvocationStatus };
+export { addConversationIdToInvocation, cancelPastPendingInvocationsForTrigger, cancelPendingInvocationsForTrigger, createScheduledTriggerInvocation, deletePendingInvocationsForTrigger, getScheduledTriggerInvocationById, getScheduledTriggerInvocationByIdempotencyKey, getScheduledTriggerInvocationStatusSummary, getScheduledTriggerRunInfoBatch, listPendingScheduledTriggerInvocations, listProjectScheduledTriggerInvocationsPaginated, listScheduledTriggerInvocationsByTriggerId, listScheduledTriggerInvocationsPaginated, listUpcomingInvocationsForAgentPaginated, markScheduledTriggerInvocationCancelled, markScheduledTriggerInvocationCompleted, markScheduledTriggerInvocationFailed, markScheduledTriggerInvocationRunning, resetCancelledInvocationToPending, updateScheduledTriggerInvocationStatus };

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

@@ -223,6 +223,39 @@ const listProjectScheduledTriggerInvocationsPaginated = (db) => async (params) =
 		}
 	};
 };
+/**
+* Get status summary for all invocations with a given scheduledTriggerId.
+* Used for dataset run status aggregation where scheduledTriggerId = datasetRunId.
+*/
+const getScheduledTriggerInvocationStatusSummary = (db) => async (params) => {
+	return (await db.select({
+		status: scheduledTriggerInvocations.status,
+		cnt: count()
+	}).from(scheduledTriggerInvocations).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId))).groupBy(scheduledTriggerInvocations.status)).reduce((acc, { status, cnt }) => {
+		const key = status;
+		if (key in acc) acc[key] += Number(cnt);
+		return acc;
+	}, {
+		pending: 0,
+		running: 0,
+		completed: 0,
+		failed: 0,
+		cancelled: 0
+	});
+};
+/**
+* List invocations by scheduledTriggerId without agentId filter (project-scoped).
+* Used for dataset run items listing where scheduledTriggerId = datasetRunId.
+*/
+const listScheduledTriggerInvocationsByTriggerId = (db) => async (params) => {
+	const conditions = [
+		eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId),
+		eq(scheduledTriggerInvocations.projectId, params.scopes.projectId),
+		eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId)
+	];
+	if (params.filters?.status) conditions.push(eq(scheduledTriggerInvocations.status, params.filters.status));
+	return await db.select().from(scheduledTriggerInvocations).where(and(...conditions)).orderBy(asc(scheduledTriggerInvocations.createdAt));
+};
 
 //#endregion
-export { addConversationIdToInvocation, cancelPastPendingInvocationsForTrigger, cancelPendingInvocationsForTrigger, createScheduledTriggerInvocation, deletePendingInvocationsForTrigger, getScheduledTriggerInvocationById, getScheduledTriggerInvocationByIdempotencyKey, getScheduledTriggerRunInfoBatch, listPendingScheduledTriggerInvocations, listProjectScheduledTriggerInvocationsPaginated, listScheduledTriggerInvocationsPaginated, listUpcomingInvocationsForAgentPaginated, markScheduledTriggerInvocationCancelled, markScheduledTriggerInvocationCompleted, markScheduledTriggerInvocationFailed, markScheduledTriggerInvocationRunning, resetCancelledInvocationToPending, updateScheduledTriggerInvocationStatus };
+export { addConversationIdToInvocation, cancelPastPendingInvocationsForTrigger, cancelPendingInvocationsForTrigger, createScheduledTriggerInvocation, deletePendingInvocationsForTrigger, getScheduledTriggerInvocationById, getScheduledTriggerInvocationByIdempotencyKey, getScheduledTriggerInvocationStatusSummary, getScheduledTriggerRunInfoBatch, listPendingScheduledTriggerInvocations, listProjectScheduledTriggerInvocationsPaginated, listScheduledTriggerInvocationsByTriggerId, listScheduledTriggerInvocationsPaginated, listUpcomingInvocationsForAgentPaginated, markScheduledTriggerInvocationCancelled, markScheduledTriggerInvocationCompleted, markScheduledTriggerInvocationFailed, markScheduledTriggerInvocationRunning, resetCancelledInvocationToPending, updateScheduledTriggerInvocationStatus };

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

@@ -15,12 +15,12 @@ declare const createTask: (db: AgentsRunDatabaseClient) => (params: TaskInsert)
   tenantId: string;
   metadata: TaskMetadataConfig | null;
   subAgentId: string;
+  status: string;
   ref: {
     type: "commit" | "tag" | "branch";
     name: string;
     hash: string;
   } | null;
-  status: string;
   contextId: string;
 }>;
 declare const getTask: (db: AgentsRunDatabaseClient) => (params: {