@inkeep/agents-core 0.47.4 → 0.48.0

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

@@ -1,6 +1,6 @@
 import { __exportAll } from "../../_virtual/rolldown_runtime.js";
 import { relations } from "drizzle-orm";
-import { boolean, doublePrecision, foreignKey, jsonb, pgTable, primaryKey, text, timestamp, unique, varchar } from "drizzle-orm/pg-core";
+import { boolean, doublePrecision, foreignKey, index, jsonb, numeric, pgTable, primaryKey, text, timestamp, unique, varchar } from "drizzle-orm/pg-core";
 
 //#region src/db/manage/manage-schema.ts
 var manage_schema_exports = /* @__PURE__ */ __exportAll({
@@ -34,6 +34,12 @@ var manage_schema_exports = /* @__PURE__ */ __exportAll({
 	functionsRelations: () => functionsRelations,
 	projects: () => projects,
 	projectsRelations: () => projectsRelations,
+	scheduledTriggers: () => scheduledTriggers,
+	scheduledTriggersRelations: () => scheduledTriggersRelations,
+	scheduledWorkflows: () => scheduledWorkflows,
+	scheduledWorkflowsRelations: () => scheduledWorkflowsRelations,
+	skills: () => skills,
+	skillsRelations: () => skillsRelations,
 	subAgentArtifactComponents: () => subAgentArtifactComponents,
 	subAgentArtifactComponentsRelations: () => subAgentArtifactComponentsRelations,
 	subAgentDataComponents: () => subAgentDataComponents,
@@ -44,6 +50,8 @@ var manage_schema_exports = /* @__PURE__ */ __exportAll({
 	subAgentFunctionToolRelationsRelations: () => subAgentFunctionToolRelationsRelations,
 	subAgentRelations: () => subAgentRelations,
 	subAgentRelationsRelations: () => subAgentRelationsRelations,
+	subAgentSkills: () => subAgentSkills,
+	subAgentSkillsRelations: () => subAgentSkillsRelations,
 	subAgentTeamAgentRelations: () => subAgentTeamAgentRelations,
 	subAgentTeamAgentRelationsRelations: () => subAgentTeamAgentRelationsRelations,
 	subAgentToolRelations: () => subAgentToolRelations,
@@ -86,8 +94,7 @@ const projects = pgTable("projects", {
 }, (table) => [primaryKey({ columns: [table.tenantId, table.id] })]);
 const agents = pgTable("agent", {
 	...projectScoped,
-	name: varchar("name", { length: 256 }).notNull(),
-	description: text("description"),
+	...uiProperties,
 	defaultSubAgentId: varchar("default_sub_agent_id", { length: 256 }),
 	contextConfigId: varchar("context_config_id", { length: 256 }),
 	models: jsonb("models").$type(),
@@ -164,6 +171,83 @@ const triggers = pgTable("triggers", {
 		name: "triggers_credential_reference_fk"
 	}).onDelete("set null")
 ]);
+const scheduledTriggers = pgTable("scheduled_triggers", {
+	...agentScoped,
+	...uiProperties,
+	enabled: boolean("enabled").notNull().default(true),
+	cronExpression: varchar("cron_expression", { length: 256 }),
+	cronTimezone: varchar("cron_timezone", { length: 64 }).default("UTC"),
+	runAt: timestamp("run_at", {
+		withTimezone: true,
+		mode: "string"
+	}),
+	payload: jsonb("payload").$type(),
+	messageTemplate: text("message_template"),
+	maxRetries: numeric("max_retries", { mode: "number" }).notNull().default(1),
+	retryDelaySeconds: numeric("retry_delay_seconds", { mode: "number" }).notNull().default(60),
+	timeoutSeconds: numeric("timeout_seconds", { mode: "number" }).notNull().default(780),
+	...timestamps
+}, (table) => [primaryKey({ columns: [
+	table.tenantId,
+	table.projectId,
+	table.agentId,
+	table.id
+] }), foreignKey({
+	columns: [
+		table.tenantId,
+		table.projectId,
+		table.agentId
+	],
+	foreignColumns: [
+		agents.tenantId,
+		agents.projectId,
+		agents.id
+	],
+	name: "scheduled_triggers_agent_fk"
+}).onDelete("cascade")]);
+const scheduledWorkflows = pgTable("scheduled_workflows", {
+	...agentScoped,
+	...uiProperties,
+	workflowRunId: varchar("workflow_run_id", { length: 256 }),
+	status: varchar("status", { length: 50 }).notNull().default("pending"),
+	scheduledTriggerId: varchar("scheduled_trigger_id", { length: 256 }).notNull(),
+	...timestamps
+}, (table) => [
+	primaryKey({ columns: [
+		table.tenantId,
+		table.projectId,
+		table.agentId,
+		table.id
+	] }),
+	foreignKey({
+		columns: [
+			table.tenantId,
+			table.projectId,
+			table.agentId
+		],
+		foreignColumns: [
+			agents.tenantId,
+			agents.projectId,
+			agents.id
+		],
+		name: "scheduled_workflows_agent_fk"
+	}).onDelete("cascade"),
+	foreignKey({
+		columns: [
+			table.tenantId,
+			table.projectId,
+			table.agentId,
+			table.scheduledTriggerId
+		],
+		foreignColumns: [
+			scheduledTriggers.tenantId,
+			scheduledTriggers.projectId,
+			scheduledTriggers.agentId,
+			scheduledTriggers.id
+		],
+		name: "scheduled_workflows_trigger_fk"
+	}).onDelete("cascade")
+]);
 const subAgents = pgTable("sub_agents", {
 	...agentScoped,
 	...uiProperties,
@@ -196,6 +280,67 @@ const subAgents = pgTable("sub_agents", {
 	],
 	name: "sub_agents_agents_fk"
 }).onDelete("cascade")]);
+const skills = pgTable("skills", {
+	...projectScoped,
+	id: varchar("id", { length: 64 }).notNull(),
+	name: varchar("name", { length: 64 }).notNull(),
+	description: text("description").notNull(),
+	content: text("content").notNull(),
+	metadata: jsonb("metadata").$type(),
+	...timestamps
+}, (table) => [primaryKey({ columns: [
+	table.tenantId,
+	table.projectId,
+	table.id
+] }), foreignKey({
+	columns: [table.tenantId, table.projectId],
+	foreignColumns: [projects.tenantId, projects.id],
+	name: "skills_project_fk"
+}).onDelete("cascade")]);
+const subAgentSkills = pgTable("sub_agent_skills", {
+	...subAgentScoped,
+	skillId: varchar("skill_id", { length: 64 }).notNull(),
+	index: numeric({ mode: "number" }).notNull().default(0),
+	alwaysLoaded: boolean("always_loaded").notNull().default(false),
+	...timestamps
+}, (table) => [
+	primaryKey({ columns: [
+		table.tenantId,
+		table.projectId,
+		table.agentId,
+		table.id
+	] }),
+	foreignKey({
+		columns: [
+			table.tenantId,
+			table.projectId,
+			table.agentId,
+			table.subAgentId
+		],
+		foreignColumns: [
+			subAgents.tenantId,
+			subAgents.projectId,
+			subAgents.agentId,
+			subAgents.id
+		],
+		name: "sub_agent_skills_sub_agent_fk"
+	}).onDelete("cascade"),
+	foreignKey({
+		columns: [
+			table.tenantId,
+			table.projectId,
+			table.skillId
+		],
+		foreignColumns: [
+			skills.tenantId,
+			skills.projectId,
+			skills.id
+		],
+		name: "sub_agent_skills_skill_fk"
+	}).onDelete("cascade"),
+	unique("sub_agent_skills_sub_agent_skill_unique").on(table.subAgentId, table.skillId),
+	index("sub_agent_skills_skill_idx").on(table.skillId)
+]);
 const subAgentRelations = pgTable("sub_agent_relations", {
 	...agentScoped,
 	sourceSubAgentId: varchar("source_sub_agent_id", { length: 256 }).notNull(),
@@ -246,7 +391,7 @@ const externalAgents = pgTable("external_agents", {
 const dataComponents = pgTable("data_components", {
 	...projectScoped,
 	...uiProperties,
-	props: jsonb("props").$type(),
+	props: jsonb("props").$type().notNull(),
 	render: jsonb("render").$type(),
 	...timestamps
 }, (table) => [primaryKey({ columns: [
@@ -358,8 +503,7 @@ const subAgentArtifactComponents = pgTable("sub_agent_artifact_components", {
 ]);
 const tools = pgTable("tools", {
 	...projectScoped,
-	name: varchar("name", { length: 256 }).notNull(),
-	description: text("description"),
+	...uiProperties,
 	config: jsonb("config").$type().notNull(),
 	credentialReferenceId: varchar("credential_reference_id", { length: 256 }),
 	credentialScope: varchar("credential_scope", { length: 50 }).notNull().default("project"),
@@ -388,8 +532,7 @@ const tools = pgTable("tools", {
 ]);
 const functionTools = pgTable("function_tools", {
 	...agentScoped,
-	name: varchar("name", { length: 256 }).notNull(),
-	description: text("description"),
+	...uiProperties,
 	functionId: varchar("function_id", { length: 256 }).notNull(),
 	...timestamps
 }, (table) => [
@@ -620,7 +763,7 @@ const subAgentFunctionToolRelations = pgTable("sub_agent_function_tool_relations
 ]);
 const credentialReferences = pgTable("credential_references", {
 	...projectScoped,
-	name: varchar("name", { length: 256 }).notNull(),
+	name: uiProperties.name,
 	type: varchar("type", { length: 256 }).notNull(),
 	credentialStoreId: varchar("credential_store_id", { length: 256 }).notNull(),
 	retrievalParams: jsonb("retrieval_params").$type(),
@@ -648,13 +791,13 @@ const credentialReferences = pgTable("credential_references", {
 * runs where conversations are created from dataset items. Each datasetRun
 * specifies which agent to use when executing the dataset.
 *
-* one to many relationship with datasetItem
+* one-to-many relationship with datasetItem
 *
 * Includes: name and timestamps
 */
 const dataset = pgTable("dataset", {
 	...projectScoped,
-	name: varchar("name", { length: 256 }).notNull(),
+	name: uiProperties.name,
 	...timestamps
 }, (table) => [primaryKey({ columns: [
 	table.tenantId,
@@ -983,7 +1126,8 @@ const projectsRelations = relations(projects, ({ many }) => ({
 	externalAgents: many(externalAgents),
 	dataComponents: many(dataComponents),
 	artifactComponents: many(artifactComponents),
-	credentialReferences: many(credentialReferences)
+	credentialReferences: many(credentialReferences),
+	skills: many(skills)
 }));
 const contextConfigsRelations = relations(contextConfigs, ({ many, one }) => ({
 	project: one(projects, {
@@ -1003,7 +1147,8 @@ const subAgentsRelations = relations(subAgents, ({ many, one }) => ({
 	toolRelations: many(subAgentToolRelations),
 	functionToolRelations: many(subAgentFunctionToolRelations),
 	dataComponentRelations: many(subAgentDataComponents),
-	artifactComponentRelations: many(subAgentArtifactComponents)
+	artifactComponentRelations: many(subAgentArtifactComponents),
+	skillRelations: many(subAgentSkills)
 }));
 const agentRelations = relations(agents, ({ one, many }) => ({
 	project: one(projects, {
@@ -1018,7 +1163,52 @@ const agentRelations = relations(agents, ({ one, many }) => ({
 		fields: [agents.contextConfigId],
 		references: [contextConfigs.id]
 	}),
-	functionTools: many(functionTools)
+	functionTools: many(functionTools),
+	scheduledWorkflows: many(scheduledWorkflows),
+	scheduledTriggers: many(scheduledTriggers)
+}));
+const scheduledTriggersRelations = relations(scheduledTriggers, ({ one }) => ({
+	agent: one(agents, {
+		fields: [
+			scheduledTriggers.tenantId,
+			scheduledTriggers.projectId,
+			scheduledTriggers.agentId
+		],
+		references: [
+			agents.tenantId,
+			agents.projectId,
+			agents.id
+		]
+	}),
+	scheduledWorkflow: one(scheduledWorkflows)
+}));
+const scheduledWorkflowsRelations = relations(scheduledWorkflows, ({ one }) => ({
+	agent: one(agents, {
+		fields: [
+			scheduledWorkflows.tenantId,
+			scheduledWorkflows.projectId,
+			scheduledWorkflows.agentId
+		],
+		references: [
+			agents.tenantId,
+			agents.projectId,
+			agents.id
+		]
+	}),
+	scheduledTrigger: one(scheduledTriggers, {
+		fields: [
+			scheduledWorkflows.tenantId,
+			scheduledWorkflows.projectId,
+			scheduledWorkflows.agentId,
+			scheduledWorkflows.scheduledTriggerId
+		],
+		references: [
+			scheduledTriggers.tenantId,
+			scheduledTriggers.projectId,
+			scheduledTriggers.agentId,
+			scheduledTriggers.id
+		]
+	})
 }));
 const externalAgentsRelations = relations(externalAgents, ({ one, many }) => ({
 	project: one(projects, {
@@ -1094,6 +1284,41 @@ const subAgentDataComponentsRelations = relations(subAgentDataComponents, ({ one
 		references: [dataComponents.id]
 	})
 }));
+const skillsRelations = relations(skills, ({ one, many }) => ({
+	project: one(projects, {
+		fields: [skills.tenantId, skills.projectId],
+		references: [projects.tenantId, projects.id]
+	}),
+	subAgentRelations: many(subAgentSkills)
+}));
+const subAgentSkillsRelations = relations(subAgentSkills, ({ one }) => ({
+	subAgent: one(subAgents, {
+		fields: [
+			subAgentSkills.tenantId,
+			subAgentSkills.projectId,
+			subAgentSkills.agentId,
+			subAgentSkills.subAgentId
+		],
+		references: [
+			subAgents.tenantId,
+			subAgents.projectId,
+			subAgents.agentId,
+			subAgents.id
+		]
+	}),
+	skill: one(skills, {
+		fields: [
+			subAgentSkills.tenantId,
+			subAgentSkills.projectId,
+			subAgentSkills.skillId
+		],
+		references: [
+			skills.tenantId,
+			skills.projectId,
+			skills.id
+		]
+	})
+}));
 const functionsRelations = relations(functions, ({ many, one }) => ({
 	functionTools: many(functionTools),
 	project: one(projects, {
@@ -1261,4 +1486,4 @@ const datasetRunConfigAgentRelations = pgTable("dataset_run_config_agent_relatio
 ]);
 
 //#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, subAgentArtifactComponents, subAgentArtifactComponentsRelations, subAgentDataComponents, subAgentDataComponentsRelations, subAgentExternalAgentRelations, subAgentExternalAgentRelationsRelations, subAgentFunctionToolRelations, subAgentFunctionToolRelationsRelations, subAgentRelations, subAgentRelationsRelations, subAgentTeamAgentRelations, subAgentTeamAgentRelationsRelations, subAgentToolRelations, subAgents, subAgentsRelations, tools, toolsRelations, triggers };
+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, scheduledTriggers, scheduledTriggersRelations, scheduledWorkflows, scheduledWorkflowsRelations, skills, skillsRelations, subAgentArtifactComponents, subAgentArtifactComponentsRelations, subAgentDataComponents, subAgentDataComponentsRelations, subAgentExternalAgentRelations, subAgentExternalAgentRelationsRelations, subAgentFunctionToolRelations, subAgentFunctionToolRelationsRelations, subAgentRelations, subAgentRelationsRelations, subAgentSkills, subAgentSkillsRelations, subAgentTeamAgentRelations, subAgentTeamAgentRelationsRelations, subAgentToolRelations, subAgents, subAgentsRelations, tools, toolsRelations, triggers };