@inkeep/agents-core 0.0.0-dev-20260331193203 → 0.0.0-dev-20260331194528

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

@@ -4,6 +4,10 @@ import { boolean, doublePrecision, foreignKey, index, jsonb, numeric, pgTable, p
 
 //#region src/db/manage/manage-schema.ts
 var manage_schema_exports = /* @__PURE__ */ __exportAll({
+	agentDatasetRelations: () => agentDatasetRelations,
+	agentDatasetRelationsRelations: () => agentDatasetRelationsRelations,
+	agentEvaluatorRelations: () => agentEvaluatorRelations,
+	agentEvaluatorRelationsRelations: () => agentEvaluatorRelationsRelations,
 	agentRelations: () => agentRelations,
 	agentToolRelationsRelations: () => agentToolRelationsRelations,
 	agents: () => agents,
@@ -761,22 +765,18 @@ const dataset = pgTable("dataset", {
 	name: "dataset_project_fk"
 }).onDelete("cascade")]);
 /**
-*
 * Individual test case within a dataset. Contains the input messages to send
-* to an agent and optionally expected output or simulation configuration.
+* to an agent and optionally expected output.
 * When a dataset run executes, it creates conversations from these items.
 *
-*
 * Includes: input (messages array with optional headers), expected output (array of messages),
-* simulation agent (stopWhen conditions, prompt/modelConfig), and timestamps
-* simulationAgent is for when a user wants to create a multi-turn simulation aka a simulating agent is creating input messages based on a persona
+* and timestamps.
 */
 const datasetItem = pgTable("dataset_item", {
 	...projectScoped,
 	datasetId: varchar("dataset_id", { length: 256 }).notNull(),
 	input: jsonb("input").$type().notNull(),
 	expectedOutput: jsonb("expected_output").$type(),
-	simulationAgent: jsonb("simulation_agent").$type(),
 	...timestamps
 }, (table) => [primaryKey({ columns: [
 	table.tenantId,
@@ -1367,6 +1367,100 @@ const subAgentTeamAgentRelationsRelations = relations(subAgentTeamAgentRelations
 	})
 }));
 /**
+* Links agents to datasets. Many-to-many relationship that scopes a dataset
+* to specific agents. When a dataset has agent relations, it is only associated
+* with those agents. When it has NO agent relations, it is treated as
+* project-wide and available to all agents.
+*
+* Includes: agentId, datasetId, and timestamps
+*/
+const agentDatasetRelations = pgTable("agent_dataset_relations", {
+	...projectScoped,
+	agentId: varchar("agent_id", { length: 256 }).notNull(),
+	datasetId: varchar("dataset_id", { length: 256 }).notNull(),
+	...timestamps
+}, (table) => [
+	primaryKey({ columns: [
+		table.tenantId,
+		table.projectId,
+		table.id
+	] }),
+	foreignKey({
+		columns: [
+			table.tenantId,
+			table.projectId,
+			table.agentId
+		],
+		foreignColumns: [
+			agents.tenantId,
+			agents.projectId,
+			agents.id
+		],
+		name: "agent_dataset_relations_agent_fk"
+	}).onDelete("cascade"),
+	foreignKey({
+		columns: [
+			table.tenantId,
+			table.projectId,
+			table.datasetId
+		],
+		foreignColumns: [
+			dataset.tenantId,
+			dataset.projectId,
+			dataset.id
+		],
+		name: "agent_dataset_relations_dataset_fk"
+	}).onDelete("cascade"),
+	unique("agent_dataset_relations_unique").on(table.tenantId, table.projectId, table.agentId, table.datasetId)
+]);
+/**
+* Links agents to evaluators. Many-to-many relationship that scopes an evaluator
+* to specific agents. When an evaluator has agent relations, it is only associated
+* with those agents. When it has NO agent relations, it is treated as
+* project-wide and available to all agents.
+*
+* Includes: agentId, evaluatorId, and timestamps
+*/
+const agentEvaluatorRelations = pgTable("agent_evaluator_relations", {
+	...projectScoped,
+	agentId: varchar("agent_id", { length: 256 }).notNull(),
+	evaluatorId: varchar("evaluator_id", { length: 256 }).notNull(),
+	...timestamps
+}, (table) => [
+	primaryKey({ columns: [
+		table.tenantId,
+		table.projectId,
+		table.id
+	] }),
+	foreignKey({
+		columns: [
+			table.tenantId,
+			table.projectId,
+			table.agentId
+		],
+		foreignColumns: [
+			agents.tenantId,
+			agents.projectId,
+			agents.id
+		],
+		name: "agent_evaluator_relations_agent_fk"
+	}).onDelete("cascade"),
+	foreignKey({
+		columns: [
+			table.tenantId,
+			table.projectId,
+			table.evaluatorId
+		],
+		foreignColumns: [
+			evaluator.tenantId,
+			evaluator.projectId,
+			evaluator.id
+		],
+		name: "agent_evaluator_relations_evaluator_fk"
+	}).onDelete("cascade"),
+	unique("agent_evaluator_relations_unique").on(table.tenantId, table.projectId, table.agentId, table.evaluatorId)
+]);
+/**
 * Links agents to dataset run configs. Many-to-many relationship that
 * allows one dataset run config to use multiple agents, and one agent to be used
 * by multiple dataset run configs.
@@ -1411,6 +1505,58 @@ const datasetRunConfigAgentRelations = pgTable("dataset_run_config_agent_relatio
 		name: "dataset_run_config_agent_relations_agent_fk"
 	}).onDelete("cascade")
 ]);
+const agentDatasetRelationsRelations = relations(agentDatasetRelations, ({ one }) => ({
+	agent: one(agents, {
+		fields: [
+			agentDatasetRelations.tenantId,
+			agentDatasetRelations.projectId,
+			agentDatasetRelations.agentId
+		],
+		references: [
+			agents.tenantId,
+			agents.projectId,
+			agents.id
+		]
+	}),
+	dataset: one(dataset, {
+		fields: [
+			agentDatasetRelations.tenantId,
+			agentDatasetRelations.projectId,
+			agentDatasetRelations.datasetId
+		],
+		references: [
+			dataset.tenantId,
+			dataset.projectId,
+			dataset.id
+		]
+	})
+}));
+const agentEvaluatorRelationsRelations = relations(agentEvaluatorRelations, ({ one }) => ({
+	agent: one(agents, {
+		fields: [
+			agentEvaluatorRelations.tenantId,
+			agentEvaluatorRelations.projectId,
+			agentEvaluatorRelations.agentId
+		],
+		references: [
+			agents.tenantId,
+			agents.projectId,
+			agents.id
+		]
+	}),
+	evaluator: one(evaluator, {
+		fields: [
+			agentEvaluatorRelations.tenantId,
+			agentEvaluatorRelations.projectId,
+			agentEvaluatorRelations.evaluatorId
+		],
+		references: [
+			evaluator.tenantId,
+			evaluator.projectId,
+			evaluator.id
+		]
+	})
+}));
 
 //#endregion
-export { agentRelations, agentToolRelationsRelations, agents, artifactComponents, artifactComponentsRelations, contextConfigs, contextConfigsRelations, credentialReferences, credentialReferencesRelations, dataComponents, dataComponentsRelations, dataset, datasetItem, datasetRunConfig, datasetRunConfigAgentRelations, evaluationJobConfig, evaluationJobConfigEvaluatorRelations, evaluationRunConfig, evaluationRunConfigEvaluationSuiteConfigRelations, evaluationSuiteConfig, evaluationSuiteConfigEvaluatorRelations, evaluator, externalAgents, externalAgentsRelations, functionTools, functionToolsRelations, functions, functionsRelations, manage_schema_exports, projects, projectsRelations, skillFiles, skillFilesRelations, skills, skillsRelations, subAgentArtifactComponents, subAgentArtifactComponentsRelations, subAgentDataComponents, subAgentDataComponentsRelations, subAgentExternalAgentRelations, subAgentExternalAgentRelationsRelations, subAgentFunctionToolRelations, subAgentFunctionToolRelationsRelations, subAgentRelations, subAgentRelationsRelations, subAgentSkills, subAgentSkillsRelations, subAgentTeamAgentRelations, subAgentTeamAgentRelationsRelations, subAgentToolRelations, subAgents, subAgentsRelations, tools, toolsRelations, triggers };
+export { agentDatasetRelations, agentDatasetRelationsRelations, agentEvaluatorRelations, agentEvaluatorRelationsRelations, agentRelations, agentToolRelationsRelations, agents, artifactComponents, artifactComponentsRelations, contextConfigs, contextConfigsRelations, credentialReferences, credentialReferencesRelations, dataComponents, dataComponentsRelations, dataset, datasetItem, datasetRunConfig, datasetRunConfigAgentRelations, evaluationJobConfig, evaluationJobConfigEvaluatorRelations, evaluationRunConfig, evaluationRunConfigEvaluationSuiteConfigRelations, evaluationSuiteConfig, evaluationSuiteConfigEvaluatorRelations, evaluator, externalAgents, externalAgentsRelations, functionTools, functionToolsRelations, functions, functionsRelations, manage_schema_exports, projects, projectsRelations, skillFiles, skillFilesRelations, skills, skillsRelations, subAgentArtifactComponents, subAgentArtifactComponentsRelations, subAgentDataComponents, subAgentDataComponentsRelations, subAgentExternalAgentRelations, subAgentExternalAgentRelationsRelations, subAgentFunctionToolRelations, subAgentFunctionToolRelationsRelations, subAgentRelations, subAgentRelationsRelations, subAgentSkills, subAgentSkillsRelations, subAgentTeamAgentRelations, subAgentTeamAgentRelationsRelations, subAgentToolRelations, subAgents, subAgentsRelations, tools, toolsRelations, triggers };