@inkeep/agents-core 0.70.1 → 0.70.2

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

@@ -1,6 +1,6 @@
 import { conversations, feedback } from "../../db/runtime/runtime-schema.js";
 import { projectScopedWhere } from "../manage/scope-helpers.js";
-import { and, count, desc, eq, gte, lte } from "drizzle-orm";
+import { and, count, desc, eq, gte, inArray, lte } from "drizzle-orm";
 
 //#region src/data-access/runtime/feedback.ts
 const getFeedbackById = (db) => async (params) => {
@@ -57,6 +57,22 @@ const listFeedback = (db) => async (params) => {
 		total: typeof total[0]?.count === "string" ? Number.parseInt(total[0].count, 10) : total[0]?.count ?? 0
 	};
 };
+const getFeedbackByIds = (db) => async (params) => {
+	if (params.feedbackIds.length === 0) return [];
+	const conversationsJoin = [
+		eq(feedback.tenantId, conversations.tenantId),
+		eq(feedback.projectId, conversations.projectId),
+		eq(feedback.conversationId, conversations.id)
+	];
+	return db.select({
+		id: feedback.id,
+		conversationId: feedback.conversationId,
+		type: feedback.type,
+		details: feedback.details,
+		createdAt: feedback.createdAt,
+		agentId: conversations.agentId
+	}).from(feedback).leftJoin(conversations, and(...conversationsJoin)).where(and(projectScopedWhere(feedback, params.scopes), inArray(feedback.id, params.feedbackIds)));
+};
 const createFeedback = (db) => async (params) => {
 	const now = (/* @__PURE__ */ new Date()).toISOString();
 	const [created] = await db.insert(feedback).values({
@@ -80,4 +96,4 @@ const deleteFeedback = (db) => async (params) => {
 };
 
 //#endregion
-export { createFeedback, deleteFeedback, getFeedbackById, listFeedback, listFeedbackByConversation, updateFeedback };
+export { createFeedback, deleteFeedback, getFeedbackById, getFeedbackByIds, listFeedback, listFeedbackByConversation, updateFeedback };

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

@@ -0,0 +1,49 @@
+import { ResolvedRef } from "../../validation/dolt-schemas.js";
+import { coPilotRuns } from "../../db/runtime/runtime-schema.js";
+import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
+
+//#region src/data-access/runtime/improvementRuns.d.ts
+type CoPilotRunInsert = typeof coPilotRuns.$inferInsert;
+type CoPilotRunSelect = typeof coPilotRuns.$inferSelect;
+type CoPilotRunStatus = 'running' | 'suspended' | 'completed' | 'failed';
+declare const createCoPilotRun: (db: AgentsRunDatabaseClient) => (params: CoPilotRunInsert) => Promise<CoPilotRunSelect>;
+declare const getCoPilotRunByRef: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: {
+    tenantId: string;
+    projectId: string;
+  };
+  ref: ResolvedRef;
+}) => Promise<CoPilotRunSelect | null>;
+declare const getCoPilotRunByBranchName: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: {
+    tenantId: string;
+    projectId: string;
+  };
+  branchName: string;
+}) => Promise<CoPilotRunSelect | null>;
+declare const listCoPilotRunsByBranchName: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: {
+    tenantId: string;
+    projectId: string;
+  };
+  branchName: string;
+}) => Promise<CoPilotRunSelect[]>;
+declare const listCoPilotRuns: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: {
+    tenantId: string;
+    projectId: string;
+  };
+}) => Promise<CoPilotRunSelect[]>;
+declare const updateCoPilotRunStatusByConversationId: (db: AgentsRunDatabaseClient) => (params: {
+  conversationId: string;
+  status: CoPilotRunStatus;
+}) => Promise<CoPilotRunSelect | null>;
+declare const deleteCoPilotRunsByBranchName: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: {
+    tenantId: string;
+    projectId: string;
+  };
+  branchName: string;
+}) => Promise<number>;
+//#endregion
+export { CoPilotRunInsert, CoPilotRunSelect, CoPilotRunStatus, createCoPilotRun, deleteCoPilotRunsByBranchName, getCoPilotRunByBranchName, getCoPilotRunByRef, listCoPilotRuns, listCoPilotRunsByBranchName, updateCoPilotRunStatusByConversationId };

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

@@ -0,0 +1,41 @@
+import { coPilotRuns } from "../../db/runtime/runtime-schema.js";
+import { and, asc, desc, eq, sql } from "drizzle-orm";
+
+//#region src/data-access/runtime/improvementRuns.ts
+const createCoPilotRun = (db) => async (params) => {
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	const [created] = await db.insert(coPilotRuns).values({
+		...params,
+		createdAt: now,
+		updatedAt: now
+	}).returning();
+	return created;
+};
+const getCoPilotRunByRef = (db) => async (params) => {
+	const [match] = await db.select().from(coPilotRuns).where(and(eq(coPilotRuns.tenantId, params.scopes.tenantId), eq(coPilotRuns.projectId, params.scopes.projectId), sql`${coPilotRuns.ref}->>'type' = ${params.ref.type}`, sql`${coPilotRuns.ref}->>'name' = ${params.ref.name}`)).orderBy(desc(coPilotRuns.createdAt)).limit(1);
+	return match ?? null;
+};
+const getCoPilotRunByBranchName = (db) => async (params) => {
+	const [match] = await db.select().from(coPilotRuns).where(and(eq(coPilotRuns.tenantId, params.scopes.tenantId), eq(coPilotRuns.projectId, params.scopes.projectId), sql`${coPilotRuns.ref}->>'type' = 'branch'`, sql`${coPilotRuns.ref}->>'name' = ${params.branchName}`)).orderBy(desc(coPilotRuns.createdAt)).limit(1);
+	return match ?? null;
+};
+const listCoPilotRunsByBranchName = (db) => async (params) => {
+	return await db.select().from(coPilotRuns).where(and(eq(coPilotRuns.tenantId, params.scopes.tenantId), eq(coPilotRuns.projectId, params.scopes.projectId), sql`${coPilotRuns.ref}->>'type' = 'branch'`, sql`${coPilotRuns.ref}->>'name' = ${params.branchName}`)).orderBy(asc(coPilotRuns.createdAt));
+};
+const listCoPilotRuns = (db) => async (params) => {
+	return await db.select().from(coPilotRuns).where(and(eq(coPilotRuns.tenantId, params.scopes.tenantId), eq(coPilotRuns.projectId, params.scopes.projectId))).orderBy(desc(coPilotRuns.createdAt));
+};
+const updateCoPilotRunStatusByConversationId = (db) => async (params) => {
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	const [updated] = await db.update(coPilotRuns).set({
+		status: params.status,
+		updatedAt: now
+	}).where(eq(coPilotRuns.conversationId, params.conversationId)).returning();
+	return updated ?? null;
+};
+const deleteCoPilotRunsByBranchName = (db) => async (params) => {
+	return (await db.delete(coPilotRuns).where(and(eq(coPilotRuns.tenantId, params.scopes.tenantId), eq(coPilotRuns.projectId, params.scopes.projectId), sql`${coPilotRuns.ref}->>'type' = 'branch'`, sql`${coPilotRuns.ref}->>'name' = ${params.branchName}`)).returning()).length;
+};
+
+//#endregion
+export { createCoPilotRun, deleteCoPilotRunsByBranchName, getCoPilotRunByBranchName, getCoPilotRunByRef, listCoPilotRuns, listCoPilotRunsByBranchName, updateCoPilotRunStatusByConversationId };

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

@@ -11,25 +11,25 @@ declare const getMessageById: (db: AgentsRunDatabaseClient) => (params: {
   messageId: string;
 }) => Promise<{
   id: string;
-  content: MessageContent;
-  metadata: MessageMetadata | null;
   createdAt: string;
   updatedAt: string;
-  projectId: string;
+  metadata: MessageMetadata | null;
+  role: string;
+  content: MessageContent;
   tenantId: string;
+  projectId: string;
+  conversationId: string;
   fromSubAgentId: string | null;
   toSubAgentId: string | null;
   fromExternalAgentId: string | null;
   toExternalAgentId: string | null;
-  taskId: string | null;
-  a2aTaskId: string | null;
-  conversationId: string;
-  role: string;
   fromTeamAgentId: string | null;
   toTeamAgentId: string | null;
   visibility: string;
   messageType: string;
+  taskId: string | null;
   parentMessageId: string | null;
+  a2aTaskId: string | null;
   a2aSessionId: string | null;
 } | undefined>;
 declare const listMessages: (db: AgentsRunDatabaseClient) => (params: {
@@ -145,25 +145,25 @@ declare const createMessage: (db: AgentsRunDatabaseClient) => (params: {
   data: Omit<MessageInsert, "tenantId" | "projectId">;
 }) => Promise<{
   id: string;
-  content: MessageContent;
-  metadata: MessageMetadata | null;
   createdAt: string;
   updatedAt: string;
-  projectId: string;
+  metadata: MessageMetadata | null;
+  role: string;
+  content: MessageContent;
   tenantId: string;
+  projectId: string;
+  conversationId: string;
   fromSubAgentId: string | null;
   toSubAgentId: string | null;
   fromExternalAgentId: string | null;
   toExternalAgentId: string | null;
-  taskId: string | null;
-  a2aTaskId: string | null;
-  conversationId: string;
-  role: string;
   fromTeamAgentId: string | null;
   toTeamAgentId: string | null;
   visibility: string;
   messageType: string;
+  taskId: string | null;
   parentMessageId: string | null;
+  a2aTaskId: string | null;
   a2aSessionId: string | null;
 }>;
 declare const updateMessage: (db: AgentsRunDatabaseClient) => (params: {
@@ -198,25 +198,25 @@ declare const deleteMessage: (db: AgentsRunDatabaseClient) => (params: {
   messageId: string;
 }) => Promise<{
   id: string;
-  content: MessageContent;
-  metadata: MessageMetadata | null;
   createdAt: string;
   updatedAt: string;
-  projectId: string;
+  metadata: MessageMetadata | null;
+  role: string;
+  content: MessageContent;
   tenantId: string;
+  projectId: string;
+  conversationId: string;
   fromSubAgentId: string | null;
   toSubAgentId: string | null;
   fromExternalAgentId: string | null;
   toExternalAgentId: string | null;
-  taskId: string | null;
-  a2aTaskId: string | null;
-  conversationId: string;
-  role: string;
   fromTeamAgentId: string | null;
   toTeamAgentId: string | null;
   visibility: string;
   messageType: string;
+  taskId: string | null;
   parentMessageId: string | null;
+  a2aTaskId: string | null;
   a2aSessionId: string | null;
 }>;
 declare const countMessagesByConversation: (db: AgentsRunDatabaseClient) => (params: {

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

@@ -16,8 +16,8 @@ declare const createScheduledTriggerUser: (db: AgentsRunDatabaseClient) => (para
   userId: string;
 }) => Promise<{
   createdAt: string;
-  tenantId: string;
   userId: string;
+  tenantId: string;
   scheduledTriggerId: string;
 }>;
 declare const deleteScheduledTriggerUser: (db: AgentsRunDatabaseClient) => (params: {

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