@inkeep/agents-core 0.0.0-dev-20260120230946 → 0.0.0-dev-20260121032144

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

@@ -1,4 +1,5 @@
 import { 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 { and, eq, inArray } from "drizzle-orm";
 
 //#region src/data-access/manage/evalConfig.ts
@@ -177,6 +178,26 @@ const getEvaluationRunConfigById = (db) => async (params) => {
 const listEvaluationRunConfigs = (db) => async (params) => {
 	return await db.select().from(evaluationRunConfig).where(and(eq(evaluationRunConfig.tenantId, params.scopes.tenantId), eq(evaluationRunConfig.projectId, params.scopes.projectId)));
 };
+const listEvaluationRunConfigsWithSuiteConfigs = (db) => async (params) => {
+	const rows = await db.select({
+		runConfig: evaluationRunConfig,
+		suiteConfigId: evaluationRunConfigEvaluationSuiteConfigRelations.evaluationSuiteConfigId
+	}).from(evaluationRunConfig).leftJoin(evaluationRunConfigEvaluationSuiteConfigRelations, and(eq(evaluationRunConfigEvaluationSuiteConfigRelations.tenantId, evaluationRunConfig.tenantId), eq(evaluationRunConfigEvaluationSuiteConfigRelations.projectId, evaluationRunConfig.projectId), eq(evaluationRunConfigEvaluationSuiteConfigRelations.evaluationRunConfigId, evaluationRunConfig.id))).where(and(eq(evaluationRunConfig.tenantId, params.scopes.tenantId), eq(evaluationRunConfig.projectId, params.scopes.projectId)));
+	const runConfigsById = /* @__PURE__ */ new Map();
+	for (const row of rows) {
+		const runConfig = row.runConfig;
+		const runConfigId = runConfig.id;
+		if (!runConfigsById.has(runConfigId)) {
+			const { tenantId: _tenantId, projectId: _projectId, ...apiRunConfig } = runConfig;
+			runConfigsById.set(runConfigId, {
+				...apiRunConfig,
+				suiteConfigIds: []
+			});
+		}
+		if (row.suiteConfigId) runConfigsById.get(runConfigId)?.suiteConfigIds.push(row.suiteConfigId);
+	}
+	return Array.from(runConfigsById.values());
+};
 const createEvaluationRunConfig = (db) => async (data) => {
 	const now = (/* @__PURE__ */ new Date()).toISOString();
 	const [created] = await db.insert(evaluationRunConfig).values({
@@ -246,6 +267,9 @@ const deleteEvaluationJobConfigEvaluatorRelation = (db) => async (params) => {
 const deleteEvaluationJobConfigEvaluatorRelationsByEvaluator = (db) => async (params) => {
 	return (await db.delete(evaluationJobConfigEvaluatorRelations).where(and(eq(evaluationJobConfigEvaluatorRelations.tenantId, params.scopes.tenantId), eq(evaluationJobConfigEvaluatorRelations.projectId, params.scopes.projectId), eq(evaluationJobConfigEvaluatorRelations.evaluatorId, params.scopes.evaluatorId))).returning()).length;
 };
+const linkDatasetRunToEvaluationJobConfig = (db) => async (params) => {
+	await db.update(datasetRun).set({ evaluationJobConfigId: params.evaluationJobConfigId }).where(and(eq(datasetRun.tenantId, params.scopes.tenantId), eq(datasetRun.projectId, params.scopes.projectId), eq(datasetRun.id, params.scopes.datasetRunId)));
+};
 
 //#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, listDatasetItems, listDatasetRunConfigs, listDatasets, listEvaluationJobConfigs, listEvaluationRunConfigs, listEvaluationSuiteConfigs, listEvaluators, updateDataset, updateDatasetItem, updateDatasetRunConfig, updateEvaluationRunConfig, updateEvaluationSuiteConfig, updateEvaluator };
+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 };

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

@@ -24,13 +24,9 @@ declare const updateFullProjectServerSide: (db: AgentsManageDatabaseClient, logg
 }) => Promise<FullProjectSelect>;
 declare const getFullProject: (db: AgentsManageDatabaseClient, logger?: ProjectLogger) => (params: {
   scopes: ProjectScopeConfig;
-  /** Optional branch name to query from. If not provided, uses the project's main branch. */
-  branchName?: string;
 }) => Promise<FullProjectSelect | null>;
 declare const getFullProjectWithRelationIds: (db: AgentsManageDatabaseClient, logger?: ProjectLogger) => (params: {
   scopes: ProjectScopeConfig;
-  /** Optional branch name to query from. If not provided, uses the project's main branch. */
-  branchName?: string;
 }) => Promise<FullProjectSelectWithRelationIds | null>;
 /**
  * Delete a complete project and cascade to all related entities

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

@@ -3,7 +3,6 @@ import { deleteExternalAgent, listExternalAgents, upsertExternalAgent } from "./
 import { deleteFunction, listFunctions, upsertFunction } from "./functions.js";
 import { deleteCredentialReference, listCredentialReferences, upsertCredentialReference } from "./credentialReferences.js";
 import { deleteTool, listTools, upsertTool } from "./tools.js";
-import { getProjectBranchName, withBranch } from "../../dolt/branch.js";
 import { listAgents } from "./agents.js";
 import { deleteArtifactComponent, listArtifactComponents, upsertArtifactComponent } from "./artifactComponents.js";
 import { deleteDataComponent, listDataComponents, upsertDataComponent } from "./dataComponents.js";
@@ -1091,29 +1090,17 @@ const getFullProjectInternal = (db, logger = defaultLogger) => async (params) =>
 	}
 };
 const getFullProject = (db, logger = defaultLogger) => async (params) => {
-	const { scopes, branchName } = params;
-	const targetBranch = branchName ?? getProjectBranchName(scopes.tenantId, scopes.projectId);
-	return withBranch(db)({
-		branchName: targetBranch,
-		callback: async (txDb) => {
-			return getFullProjectInternal(txDb, logger)({
-				scopes,
-				includeRelationIds: false
-			});
-		}
+	const { scopes } = params;
+	return getFullProjectInternal(db, logger)({
+		scopes,
+		includeRelationIds: false
 	});
 };
 const getFullProjectWithRelationIds = (db, logger = defaultLogger) => async (params) => {
-	const { scopes, branchName } = params;
-	const targetBranch = branchName ?? getProjectBranchName(scopes.tenantId, scopes.projectId);
-	return withBranch(db)({
-		branchName: targetBranch,
-		callback: async (txDb) => {
-			return getFullProjectInternal(txDb, logger)({
-				scopes,
-				includeRelationIds: true
-			});
-		}
+	const { scopes } = params;
+	return getFullProjectInternal(db, logger)({
+		scopes,
+		includeRelationIds: true
 	});
 };
 /**

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

@@ -30,6 +30,13 @@ declare const getToolById: (db: AgentsManageDatabaseClient) => (params: {
   capabilities: ToolServerCapabilities | null;
   lastError: string | null;
 } | null>;
+declare const getMcpToolById: (db: AgentsManageDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+  toolId: string;
+  relationshipId?: string;
+  credentialStoreRegistry?: CredentialStoreRegistry;
+  userId?: string;
+}) => Promise<McpTool | null>;
 declare const listTools: (db: AgentsManageDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   pagination?: PaginationConfig;
@@ -202,4 +209,4 @@ declare const upsertTool: (db: AgentsManageDatabaseClient) => (params: {
   lastError: string | null;
 }>;
 //#endregion
-export { addToolToAgent, createTool, dbResultToMcpTool, deleteTool, getToolById, listTools, removeToolFromAgent, updateTool, upsertSubAgentToolRelation, upsertTool };
+export { addToolToAgent, createTool, dbResultToMcpTool, deleteTool, getMcpToolById, getToolById, listTools, removeToolFromAgent, updateTool, upsertSubAgentToolRelation, upsertTool };

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

@@ -240,6 +240,14 @@ const dbResultToMcpTool = async (dbResult, dbClient, credentialStoreRegistry, re
 const getToolById = (db) => async (params) => {
 	return await db.query.tools.findFirst({ where: and(eq(tools.tenantId, params.scopes.tenantId), eq(tools.projectId, params.scopes.projectId), eq(tools.id, params.toolId)) }) ?? null;
 };
+const getMcpToolById = (db) => async (params) => {
+	const tool = await getToolById(db)({
+		scopes: params.scopes,
+		toolId: params.toolId
+	});
+	if (!tool) return null;
+	return await dbResultToMcpTool(tool, db, params.credentialStoreRegistry, params.relationshipId, params.userId);
+};
 const listTools = (db) => async (params) => {
 	const page = params.pagination?.page || 1;
 	const limit = Math.min(params.pagination?.limit || 10, 100);
@@ -343,4 +351,4 @@ const upsertTool = (db) => async (params) => {
 };
 
 //#endregion
-export { addToolToAgent, createTool, dbResultToMcpTool, deleteTool, getToolById, listTools, removeToolFromAgent, updateTool, upsertSubAgentToolRelation, upsertTool };
+export { addToolToAgent, createTool, dbResultToMcpTool, deleteTool, getMcpToolById, getToolById, listTools, removeToolFromAgent, updateTool, upsertSubAgentToolRelation, upsertTool };

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

@@ -6,14 +6,10 @@ import { TriggerInsert, TriggerSelect, TriggerUpdate } from "../../types/entitie
 
 /**
  * Get a trigger by ID (agent-scoped)
- * Uses withBranch to checkout the project's main branch and query using Drizzle ORM,
- * which handles column name mapping automatically.
  */
 declare const getTriggerById: (db: AgentsManageDatabaseClient) => (params: {
   scopes: AgentScopeConfig;
   triggerId: string;
-  /** If true, uses withBranch to read from the project branch. Default: true for non-branch-scoped clients */
-  useBranchScope?: boolean;
 }) => Promise<TriggerSelect | undefined>;
 /**
  * List all triggers for an agent

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

@@ -1,26 +1,12 @@
 import { triggers } from "../../db/manage/manage-schema.js";
-import { getProjectBranchName, withBranch } from "../../dolt/branch.js";
 import { and, count, desc, eq } from "drizzle-orm";
 
 //#region src/data-access/manage/triggers.ts
 /**
 * Get a trigger by ID (agent-scoped)
-* Uses withBranch to checkout the project's main branch and query using Drizzle ORM,
-* which handles column name mapping automatically.
 */
 const getTriggerById = (db) => async (params) => {
-	const { scopes, triggerId, useBranchScope = true } = params;
-	if (useBranchScope) {
-		const branchName = getProjectBranchName(scopes.tenantId, scopes.projectId);
-		try {
-			return await withBranch(db)({
-				branchName,
-				callback: async (txDb) => {
-					return await txDb.query.triggers.findFirst({ where: and(eq(triggers.tenantId, scopes.tenantId), eq(triggers.projectId, scopes.projectId), eq(triggers.agentId, scopes.agentId), eq(triggers.id, triggerId)) });
-				}
-			});
-		} catch {}
-	}
+	const { scopes, triggerId } = params;
 	return await db.query.triggers.findFirst({ where: and(eq(triggers.tenantId, scopes.tenantId), eq(triggers.projectId, scopes.projectId), eq(triggers.agentId, scopes.agentId), eq(triggers.id, triggerId)) });
 };
 /**

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

@@ -25,7 +25,7 @@ declare const createConversation: (db: AgentsRunDatabaseClient) => (params: Conv
   userId: string | null;
   metadata: ConversationMetadata | null;
   ref: {
-    type: "tag" | "commit" | "branch";
+    type: "commit" | "tag" | "branch";
     name: string;
     hash: string;
   } | null;
@@ -43,7 +43,7 @@ declare const updateConversation: (db: AgentsRunDatabaseClient) => (params: {
   agentId: string | null;
   activeSubAgentId: string;
   ref: {
-    type: "tag" | "commit" | "branch";
+    type: "commit" | "tag" | "branch";
     name: string;
     hash: string;
   } | null;
@@ -69,7 +69,7 @@ declare const updateConversationActiveSubAgent: (db: AgentsRunDatabaseClient) =>
   agentId: string | null;
   activeSubAgentId: string;
   ref: {
-    type: "tag" | "commit" | "branch";
+    type: "commit" | "tag" | "branch";
     name: string;
     hash: string;
   } | null;
@@ -94,7 +94,7 @@ declare const getConversation: (db: AgentsRunDatabaseClient) => (params: {
   userId: string | null;
   metadata: ConversationMetadata | null;
   ref: {
-    type: "tag" | "commit" | "branch";
+    type: "commit" | "tag" | "branch";
     name: string;
     hash: string;
   } | null;
@@ -107,7 +107,7 @@ declare const createOrGetConversation: (db: AgentsRunDatabaseClient) => (input:
   tenantId: string;
   id: string;
   ref: {
-    type: "tag" | "commit" | "branch";
+    type: "commit" | "tag" | "branch";
     name: string;
     hash: string;
   };
@@ -130,7 +130,7 @@ declare const createOrGetConversation: (db: AgentsRunDatabaseClient) => (input:
   userId: string | null;
   metadata: ConversationMetadata | null;
   ref: {
-    type: "tag" | "commit" | "branch";
+    type: "commit" | "tag" | "branch";
     name: string;
     hash: string;
   } | null;
@@ -162,7 +162,7 @@ declare const getActiveAgentForConversation: (db: AgentsRunDatabaseClient) => (p
   userId: string | null;
   metadata: ConversationMetadata | null;
   ref: {
-    type: "tag" | "commit" | "branch";
+    type: "commit" | "tag" | "branch";
     name: string;
     hash: string;
   } | null;

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

@@ -18,15 +18,15 @@ declare const getMessageById: (db: AgentsRunDatabaseClient) => (params: {
   content: MessageContent;
   role: string;
   conversationId: string;
+  taskId: string | null;
+  visibility: string;
   fromSubAgentId: string | null;
   toSubAgentId: string | null;
   fromExternalAgentId: string | null;
   toExternalAgentId: string | null;
   fromTeamAgentId: string | null;
   toTeamAgentId: string | null;
-  visibility: string;
   messageType: string;
-  taskId: string | null;
   parentMessageId: string | null;
   a2aTaskId: string | null;
   a2aSessionId: string | null;
@@ -149,15 +149,15 @@ declare const createMessage: (db: AgentsRunDatabaseClient) => (params: MessageIn
   content: MessageContent;
   role: string;
   conversationId: string;
+  taskId: string | null;
+  visibility: string;
   fromSubAgentId: string | null;
   toSubAgentId: string | null;
   fromExternalAgentId: string | null;
   toExternalAgentId: string | null;
   fromTeamAgentId: string | null;
   toTeamAgentId: string | null;
-  visibility: string;
   messageType: string;
-  taskId: string | null;
   parentMessageId: string | null;
   a2aTaskId: string | null;
   a2aSessionId: string | null;
@@ -202,15 +202,15 @@ declare const deleteMessage: (db: AgentsRunDatabaseClient) => (params: {
   content: MessageContent;
   role: string;
   conversationId: string;
+  taskId: string | null;
+  visibility: string;
   fromSubAgentId: string | null;
   toSubAgentId: string | null;
   fromExternalAgentId: string | null;
   toExternalAgentId: string | null;
   fromTeamAgentId: string | null;
   toTeamAgentId: string | null;
-  visibility: string;
   messageType: string;
-  taskId: string | null;
   parentMessageId: string | null;
   a2aTaskId: string | null;
   a2aSessionId: string | null;

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

@@ -39,5 +39,14 @@ declare const addUserToOrganization: (db: AgentsRunDatabaseClient) => (data: {
   organizationId: string;
   role: string;
 }) => Promise<void>;
+declare const upsertOrganization: (db: AgentsRunDatabaseClient) => (data: {
+  organizationId: string;
+  name: string;
+  slug: string;
+  logo?: string | null;
+  metadata?: string | null;
+}) => Promise<{
+  created: boolean;
+}>;
 //#endregion
-export { addUserToOrganization, getPendingInvitationsByEmail, getUserOrganizations };
+export { addUserToOrganization, getPendingInvitationsByEmail, getUserOrganizations, upsertOrganization };

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

@@ -58,6 +58,18 @@ const addUserToOrganization = (db) => async (data) => {
 		createdAt: /* @__PURE__ */ new Date()
 	});
 };
+const upsertOrganization = (db) => async (data) => {
+	if ((await db.select().from(organization).where(eq(organization.id, data.organizationId)).limit(1)).length > 0) return { created: false };
+	await db.insert(organization).values({
+		id: data.organizationId,
+		name: data.name,
+		slug: data.slug,
+		createdAt: /* @__PURE__ */ new Date(),
+		logo: data.logo ?? null,
+		metadata: data.metadata ?? null
+	});
+	return { created: true };
+};
 
 //#endregion
-export { addUserToOrganization, getPendingInvitationsByEmail, getUserOrganizations };
+export { addUserToOrganization, getPendingInvitationsByEmail, getUserOrganizations, upsertOrganization };

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

@@ -14,7 +14,7 @@ declare const createTask: (db: AgentsRunDatabaseClient) => (params: TaskInsert)
   updatedAt: string;
   metadata: TaskMetadataConfig | null;
   ref: {
-    type: "tag" | "commit" | "branch";
+    type: "commit" | "tag" | "branch";
     name: string;
     hash: string;
   } | null;
@@ -36,7 +36,7 @@ declare const updateTask: (db: AgentsRunDatabaseClient) => (params: {
   updatedAt: string;
   contextId: string;
   ref: {
-    type: "tag" | "commit" | "branch";
+    type: "commit" | "tag" | "branch";
     name: string;
     hash: string;
   } | null;

package/dist/db/manage/manage-client.d.ts

@@ -1,5 +1,6 @@
 import { manage_schema_d_exports } from "./manage-schema.js";
 import { NodePgDatabase } from "drizzle-orm/node-postgres";
+import { Pool } from "pg";
 import { PgliteDatabase } from "drizzle-orm/pglite";
 
 //#region src/db/manage/manage-client.d.ts
@@ -12,6 +13,7 @@ interface AgentsManageDatabaseConfig {
     logQuery: (query: string, params: unknown[]) => void;
   };
 }
+declare function createAgentsManageDatabasePool(config: AgentsManageDatabaseConfig): Pool;
 /**
  * Creates a PostgreSQL database client with connection pooling
  */
@@ -21,4 +23,4 @@ declare function createAgentManageDatabaseConnection(config: AgentsManageDatabas
   release: () => Promise<void>;
 }>;
 //#endregion
-export { AgentsManageDatabaseClient, AgentsManageDatabaseConfig, createAgentManageDatabaseConnection, createAgentsManageDatabaseClient };
+export { AgentsManageDatabaseClient, AgentsManageDatabaseConfig, createAgentManageDatabaseConnection, createAgentsManageDatabaseClient, createAgentsManageDatabasePool };

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

@@ -6,6 +6,18 @@ import { Pool } from "pg";
 
 //#region src/db/manage/manage-client.ts
 loadEnvironmentFiles();
+function createAgentsManageDatabasePool(config) {
+	const connectionString = config.connectionString || env.INKEEP_AGENTS_MANAGE_DATABASE_URL;
+	if (!connectionString) throw new Error("INKEEP_AGENTS_MANAGE_DATABASE_URL environment variable is required. Please set it to your PostgreSQL connection string.");
+	const pool = new Pool({
+		connectionString,
+		max: config.poolSize || Number(env.POSTGRES_POOL_SIZE) || 100
+	});
+	pool.on("error", (err) => {
+		console.error("Unexpected PostgreSQL pool error:", err);
+	});
+	return pool;
+}
 /**
 * Creates a PostgreSQL database client with connection pooling
 */
@@ -53,4 +65,4 @@ function createAgentManageDatabaseConnection(config) {
 }
 
 //#endregion
-export { createAgentManageDatabaseConnection, createAgentsManageDatabaseClient };
+export { createAgentManageDatabaseConnection, createAgentsManageDatabaseClient, createAgentsManageDatabasePool };