@inkeep/agents-core 0.41.2 → 0.43.0

package/dist/data-access/runtime/cascade-delete.d.ts

@@ -0,0 +1,77 @@
+import { AgentScopeConfig, ProjectScopeConfig } from "../../types/utility.js";
+import "../../types/index.js";
+import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
+
+//#region src/data-access/runtime/cascade-delete.d.ts
+/**
+ * Result of a cascade delete operation
+ */
+type CascadeDeleteResult = {
+  conversationsDeleted: number;
+  tasksDeleted: number;
+  contextCacheDeleted: number;
+  apiKeysDeleted: number;
+};
+/**
+ * Delete all runtime entities for a specific branch.
+ * PostgreSQL cascades handle: messages, taskRelations, ledgerArtifacts
+ *
+ * @param db - Runtime database client
+ * @returns Function that performs the cascade delete
+ */
+declare const cascadeDeleteByBranch: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+  fullBranchName: string;
+}) => Promise<CascadeDeleteResult>;
+/**
+ * Delete all runtime entities for a project on a specific branch.
+ * Used when deleting a project from a branch.
+ * PostgreSQL cascades handle: messages, taskRelations, ledgerArtifacts
+ *
+ * @param db - Runtime database client
+ * @returns Function that performs the cascade delete
+ */
+declare const cascadeDeleteByProject: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+  fullBranchName: string;
+}) => Promise<CascadeDeleteResult>;
+/**
+ * Delete all runtime entities for a specific agent on a specific branch.
+ * This includes tasks for the agent and conversations where the agent's
+ * subAgents are active.
+ *
+ * @param db - Runtime database client
+ * @returns Function that performs the cascade delete
+ */
+declare const cascadeDeleteByAgent: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: AgentScopeConfig;
+  fullBranchName: string;
+  subAgentIds: string[];
+}) => Promise<CascadeDeleteResult>;
+/**
+ * Delete all runtime entities for a specific subAgent on a specific branch.
+ *
+ * @param db - Runtime database client
+ * @returns Function that performs the cascade delete
+ */
+declare const cascadeDeleteBySubAgent: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+  subAgentId: string;
+  fullBranchName: string;
+}) => Promise<CascadeDeleteResult>;
+/**
+ * Delete contextCache entries for a specific contextConfig.
+ * Used when deleting a contextConfig from the manage DB.
+ *
+ * @param db - Runtime database client
+ * @returns Function that performs the delete
+ */
+declare const cascadeDeleteByContextConfig: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+  contextConfigId: string;
+  fullBranchName: string;
+}) => Promise<{
+  contextCacheDeleted: number;
+}>;
+//#endregion
+export { CascadeDeleteResult, cascadeDeleteByAgent, cascadeDeleteByBranch, cascadeDeleteByContextConfig, cascadeDeleteByProject, cascadeDeleteBySubAgent };

package/dist/data-access/runtime/cascade-delete.js

@@ -0,0 +1,111 @@
+import { apiKeys, contextCache, conversations, tasks } from "../../db/runtime/runtime-schema.js";
+import { and, eq, inArray, sql } from "drizzle-orm";
+
+//#region src/data-access/runtime/cascade-delete.ts
+/**
+* Delete all runtime entities for a specific branch.
+* PostgreSQL cascades handle: messages, taskRelations, ledgerArtifacts
+*
+* @param db - Runtime database client
+* @returns Function that performs the cascade delete
+*/
+const cascadeDeleteByBranch = (db) => async (params) => {
+	const { scopes, fullBranchName } = params;
+	const contextCacheResult = await db.delete(contextCache).where(and(eq(contextCache.tenantId, scopes.tenantId), eq(contextCache.projectId, scopes.projectId), sql`${contextCache.ref}->>'name' = ${fullBranchName}`)).returning();
+	const conversationsResult = await db.delete(conversations).where(and(eq(conversations.tenantId, scopes.tenantId), eq(conversations.projectId, scopes.projectId), sql`${conversations.ref}->>'name' = ${fullBranchName}`)).returning();
+	const tasksResult = await db.delete(tasks).where(and(eq(tasks.tenantId, scopes.tenantId), eq(tasks.projectId, scopes.projectId), sql`${tasks.ref}->>'name' = ${fullBranchName}`)).returning();
+	return {
+		conversationsDeleted: conversationsResult.length,
+		tasksDeleted: tasksResult.length,
+		contextCacheDeleted: contextCacheResult.length,
+		apiKeysDeleted: 0
+	};
+};
+/**
+* Delete all runtime entities for a project on a specific branch.
+* Used when deleting a project from a branch.
+* PostgreSQL cascades handle: messages, taskRelations, ledgerArtifacts
+*
+* @param db - Runtime database client
+* @returns Function that performs the cascade delete
+*/
+const cascadeDeleteByProject = (db) => async (params) => {
+	const { scopes, fullBranchName } = params;
+	const contextCacheResult = await db.delete(contextCache).where(and(eq(contextCache.tenantId, scopes.tenantId), eq(contextCache.projectId, scopes.projectId), sql`${contextCache.ref}->>'name' = ${fullBranchName}`)).returning();
+	const conversationsResult = await db.delete(conversations).where(and(eq(conversations.tenantId, scopes.tenantId), eq(conversations.projectId, scopes.projectId), sql`${conversations.ref}->>'name' = ${fullBranchName}`)).returning();
+	const tasksResult = await db.delete(tasks).where(and(eq(tasks.tenantId, scopes.tenantId), eq(tasks.projectId, scopes.projectId), sql`${tasks.ref}->>'name' = ${fullBranchName}`)).returning();
+	const apiKeysResult = await db.delete(apiKeys).where(and(eq(apiKeys.tenantId, scopes.tenantId), eq(apiKeys.projectId, scopes.projectId))).returning();
+	return {
+		conversationsDeleted: conversationsResult.length,
+		tasksDeleted: tasksResult.length,
+		contextCacheDeleted: contextCacheResult.length,
+		apiKeysDeleted: apiKeysResult.length
+	};
+};
+/**
+* Delete all runtime entities for a specific agent on a specific branch.
+* This includes tasks for the agent and conversations where the agent's
+* subAgents are active.
+*
+* @param db - Runtime database client
+* @returns Function that performs the cascade delete
+*/
+const cascadeDeleteByAgent = (db) => async (params) => {
+	const { scopes, fullBranchName, subAgentIds } = params;
+	let contextCacheDeleted = 0;
+	let conversationsDeleted = 0;
+	let tasksDeleted = 0;
+	let apiKeysDeleted = 0;
+	if (subAgentIds.length > 0) {
+		const conversationIds = (await db.select({ id: conversations.id }).from(conversations).where(and(eq(conversations.tenantId, scopes.tenantId), eq(conversations.projectId, scopes.projectId), inArray(conversations.activeSubAgentId, subAgentIds), sql`${conversations.ref}->>'name' = ${fullBranchName}`))).map((c) => c.id);
+		if (conversationIds.length > 0) {
+			contextCacheDeleted = (await db.delete(contextCache).where(and(eq(contextCache.tenantId, scopes.tenantId), eq(contextCache.projectId, scopes.projectId), inArray(contextCache.conversationId, conversationIds))).returning()).length;
+			conversationsDeleted = (await db.delete(conversations).where(and(eq(conversations.tenantId, scopes.tenantId), eq(conversations.projectId, scopes.projectId), inArray(conversations.id, conversationIds))).returning()).length;
+		}
+	}
+	tasksDeleted = (await db.delete(tasks).where(and(eq(tasks.tenantId, scopes.tenantId), eq(tasks.projectId, scopes.projectId), eq(tasks.agentId, scopes.agentId), sql`${tasks.ref}->>'name' = ${fullBranchName}`)).returning()).length;
+	apiKeysDeleted = (await db.delete(apiKeys).where(and(eq(apiKeys.tenantId, scopes.tenantId), eq(apiKeys.projectId, scopes.projectId), eq(apiKeys.agentId, scopes.agentId))).returning()).length;
+	return {
+		conversationsDeleted,
+		tasksDeleted,
+		contextCacheDeleted,
+		apiKeysDeleted
+	};
+};
+/**
+* Delete all runtime entities for a specific subAgent on a specific branch.
+*
+* @param db - Runtime database client
+* @returns Function that performs the cascade delete
+*/
+const cascadeDeleteBySubAgent = (db) => async (params) => {
+	const { scopes, subAgentId, fullBranchName } = params;
+	const conversationIds = (await db.select({ id: conversations.id }).from(conversations).where(and(eq(conversations.tenantId, scopes.tenantId), eq(conversations.projectId, scopes.projectId), eq(conversations.activeSubAgentId, subAgentId), sql`${conversations.ref}->>'name' = ${fullBranchName}`))).map((c) => c.id);
+	let contextCacheDeleted = 0;
+	let conversationsDeleted = 0;
+	if (conversationIds.length > 0) {
+		contextCacheDeleted = (await db.delete(contextCache).where(and(eq(contextCache.tenantId, scopes.tenantId), eq(contextCache.projectId, scopes.projectId), inArray(contextCache.conversationId, conversationIds))).returning()).length;
+		conversationsDeleted = (await db.delete(conversations).where(and(eq(conversations.tenantId, scopes.tenantId), eq(conversations.projectId, scopes.projectId), inArray(conversations.id, conversationIds))).returning()).length;
+	}
+	const tasksResult = await db.delete(tasks).where(and(eq(tasks.tenantId, scopes.tenantId), eq(tasks.projectId, scopes.projectId), eq(tasks.subAgentId, subAgentId), sql`${tasks.ref}->>'name' = ${fullBranchName}`)).returning();
+	return {
+		conversationsDeleted,
+		tasksDeleted: tasksResult.length,
+		contextCacheDeleted,
+		apiKeysDeleted: 0
+	};
+};
+/**
+* Delete contextCache entries for a specific contextConfig.
+* Used when deleting a contextConfig from the manage DB.
+*
+* @param db - Runtime database client
+* @returns Function that performs the delete
+*/
+const cascadeDeleteByContextConfig = (db) => async (params) => {
+	const { scopes, contextConfigId, fullBranchName } = params;
+	return { contextCacheDeleted: (await db.delete(contextCache).where(and(eq(contextCache.tenantId, scopes.tenantId), eq(contextCache.projectId, scopes.projectId), eq(contextCache.contextConfigId, contextConfigId), sql`${contextCache.ref}->>'name' = ${fullBranchName}`)).returning()).length };
+};
+
+//#endregion
+export { cascadeDeleteByAgent, cascadeDeleteByBranch, cascadeDeleteByContextConfig, cascadeDeleteByProject, cascadeDeleteBySubAgent };

package/dist/data-access/{contextCache.d.ts → runtime/contextCache.d.ts}

@@ -1,13 +1,13 @@
-import { ContextCacheInsert, ContextCacheSelect } from "../types/entities.js";
-import { ProjectScopeConfig } from "../types/utility.js";
-import { DatabaseClient } from "../db/client.js";
+import { ProjectScopeConfig } from "../../types/utility.js";
+import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
+import { ContextCacheInsert, ContextCacheSelect } from "../../types/entities.js";
 
-//#region src/data-access/contextCache.d.ts
+//#region src/data-access/runtime/contextCache.d.ts
 
 /**
  * Get cached context data for a conversation with optional request hash validation
  */
-declare const getCacheEntry: (db: DatabaseClient) => (params: {
+declare const getCacheEntry: (db: AgentsRunDatabaseClient) => (params: {
   conversationId: string;
   contextConfigId: string;
   contextVariableKey: string;
@@ -16,31 +16,31 @@ declare const getCacheEntry: (db: DatabaseClient) => (params: {
 /**
  * Set cached context data for a conversation
  */
-declare const setCacheEntry: (db: DatabaseClient) => (entry: ContextCacheInsert) => Promise<ContextCacheSelect | null>;
+declare const setCacheEntry: (db: AgentsRunDatabaseClient) => (entry: ContextCacheInsert) => Promise<ContextCacheSelect | null>;
 /**
  * Clear cache entries for a specific conversation
  */
-declare const clearConversationCache: (db: DatabaseClient) => (params: {
+declare const clearConversationCache: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   conversationId: string;
 }) => Promise<number>;
 /**
  * Clear all cache entries for a specific context configuration
  */
-declare const clearContextConfigCache: (db: DatabaseClient) => (params: {
+declare const clearContextConfigCache: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   contextConfigId: string;
 }) => Promise<number>;
 /**
  * Clean up all cache entries for a tenant
  */
-declare const cleanupTenantCache: (db: DatabaseClient) => (params: {
+declare const cleanupTenantCache: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
 }) => Promise<number>;
 /**
  * Invalidate the headers cache for a conversation
  */
-declare const invalidateHeadersCache: (db: DatabaseClient) => (params: {
+declare const invalidateHeadersCache: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   conversationId: string;
   contextConfigId: string;
@@ -48,7 +48,7 @@ declare const invalidateHeadersCache: (db: DatabaseClient) => (params: {
 /**
  * Invalidate specific cache entries for invocation-trigger definitions
  */
-declare const invalidateInvocationDefinitionsCache: (db: DatabaseClient) => (params: {
+declare const invalidateInvocationDefinitionsCache: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   conversationId: string;
   contextConfigId: string;
@@ -57,14 +57,14 @@ declare const invalidateInvocationDefinitionsCache: (db: DatabaseClient) => (par
 /**
  * Get all cache entries for a conversation
  */
-declare const getConversationCacheEntries: (db: DatabaseClient) => (params: {
+declare const getConversationCacheEntries: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   conversationId: string;
 }) => Promise<ContextCacheSelect[]>;
 /**
  * Get all cache entries for a context configuration
  */
-declare const getContextConfigCacheEntries: (db: DatabaseClient) => (params: {
+declare const getContextConfigCacheEntries: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   contextConfigId: string;
 }) => Promise<ContextCacheSelect[]>;

package/dist/data-access/{contextCache.js → runtime/contextCache.js}

@@ -1,8 +1,8 @@
-import { contextCache } from "../db/schema.js";
-import { generateId } from "../utils/conversations.js";
+import { contextCache } from "../../db/runtime/runtime-schema.js";
+import { generateId } from "../../utils/conversations.js";
 import { and, eq } from "drizzle-orm";
 
-//#region src/data-access/contextCache.ts
+//#region src/data-access/runtime/contextCache.ts
 /**
 * Get cached context data for a conversation with optional request hash validation
 */
@@ -35,9 +35,9 @@ const setCacheEntry = (db) => async (entry) => {
 			requestHash: entry.requestHash || null,
 			fetchedAt: (/* @__PURE__ */ new Date()).toISOString(),
 			fetchSource: `${entry.contextConfigId}:${entry.contextVariableKey}`,
-			fetchDurationMs: 0,
 			createdAt: (/* @__PURE__ */ new Date()).toISOString(),
-			updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+			updatedAt: (/* @__PURE__ */ new Date()).toISOString(),
+			ref: entry.ref
 		};
 		const [result] = await db.insert(contextCache).values(cacheData).returning();
 		return {

package/dist/data-access/{conversations.d.ts → runtime/conversations.d.ts}

@@ -1,11 +1,12 @@
-import { ConversationInsert, ConversationSelect, ConversationUpdate } from "../types/entities.js";
-import "../types/index.js";
-import "../index.js";
-import { ConversationHistoryConfig, ConversationMetadata, PaginationConfig, ProjectScopeConfig } from "../types/utility.js";
-import { DatabaseClient } from "../db/client.js";
+import { ResolvedRef } from "../../validation/dolt-schemas.js";
+import { ConversationHistoryConfig, ConversationMetadata, PaginationConfig, ProjectScopeConfig } from "../../types/utility.js";
+import "../../types/index.js";
+import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
+import "../../index.js";
+import { ConversationInsert, ConversationSelect, ConversationUpdate } from "../../types/entities.js";
 
-//#region src/data-access/conversations.d.ts
-declare const listConversations: (db: DatabaseClient) => (params: {
+//#region src/data-access/runtime/conversations.d.ts
+declare const listConversations: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   userId?: string;
   pagination?: PaginationConfig;
@@ -13,19 +14,25 @@ declare const listConversations: (db: DatabaseClient) => (params: {
   conversations: ConversationSelect[];
   total: number;
 }>;
-declare const createConversation: (db: DatabaseClient) => (params: ConversationInsert) => Promise<{
-  tenantId: string;
-  projectId: string;
+declare const createConversation: (db: AgentsRunDatabaseClient) => (params: ConversationInsert) => Promise<{
   id: string;
-  title: string | null;
   createdAt: string;
   updatedAt: string;
+  ref: {
+    type: "commit" | "tag" | "branch";
+    name: string;
+    hash: string;
+  } | null;
   userId: string | null;
   metadata: ConversationMetadata | null;
+  tenantId: string;
+  agentId: string | null;
   activeSubAgentId: string;
+  title: string | null;
   lastContextResolution: string | null;
+  projectId: string;
 }>;
-declare const updateConversation: (db: DatabaseClient) => (params: {
+declare const updateConversation: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   conversationId: string;
   data: ConversationUpdate;
@@ -33,7 +40,13 @@ declare const updateConversation: (db: DatabaseClient) => (params: {
   createdAt: string;
   updatedAt: string;
   userId: string | null;
+  agentId: string | null;
   activeSubAgentId: string;
+  ref: {
+    type: "commit" | "tag" | "branch";
+    name: string;
+    hash: string;
+  } | null;
   title: string | null;
   lastContextResolution: string | null;
   metadata: ConversationMetadata | null;
@@ -41,11 +54,11 @@ declare const updateConversation: (db: DatabaseClient) => (params: {
   tenantId: string;
   id: string;
 }>;
-declare const deleteConversation: (db: DatabaseClient) => (params: {
+declare const deleteConversation: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   conversationId: string;
 }) => Promise<boolean>;
-declare const updateConversationActiveAgent: (db: DatabaseClient) => (params: {
+declare const updateConversationActiveSubAgent: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   conversationId: string;
   activeSubAgentId: string;
@@ -53,7 +66,13 @@ declare const updateConversationActiveAgent: (db: DatabaseClient) => (params: {
   createdAt: string;
   updatedAt: string;
   userId: string | null;
+  agentId: string | null;
   activeSubAgentId: string;
+  ref: {
+    type: "commit" | "tag" | "branch";
+    name: string;
+    hash: string;
+  } | null;
   title: string | null;
   lastContextResolution: string | null;
   metadata: ConversationMetadata | null;
@@ -61,49 +80,67 @@ declare const updateConversationActiveAgent: (db: DatabaseClient) => (params: {
   tenantId: string;
   id: string;
 }>;
-declare const getConversation: (db: DatabaseClient) => (params: {
+declare const getConversation: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   conversationId: string;
 }) => Promise<{
-  tenantId: string;
-  projectId: string;
   id: string;
-  title: string | null;
   createdAt: string;
   updatedAt: string;
+  ref: {
+    type: "commit" | "tag" | "branch";
+    name: string;
+    hash: string;
+  } | null;
   userId: string | null;
   metadata: ConversationMetadata | null;
+  tenantId: string;
+  agentId: string | null;
   activeSubAgentId: string;
+  title: string | null;
   lastContextResolution: string | null;
+  projectId: string;
 } | undefined>;
-declare const createOrGetConversation: (db: DatabaseClient) => (input: ConversationInsert) => Promise<{
+declare const createOrGetConversation: (db: AgentsRunDatabaseClient) => (input: ConversationInsert) => Promise<{
   activeSubAgentId: string;
   projectId: string;
   tenantId: string;
   id: string;
+  ref: {
+    type: "commit" | "tag" | "branch";
+    name: string;
+    hash: string;
+  };
   createdAt?: string | undefined;
   updatedAt?: string | undefined;
   userId?: string | null | undefined;
+  agentId?: string | null | undefined;
   title?: string | null | undefined;
   lastContextResolution?: string | null | undefined;
   metadata?: ConversationMetadata | null | undefined;
   contextConfigId?: string | undefined;
 } | {
-  tenantId: string;
-  projectId: string;
   id: string;
-  title: string | null;
   createdAt: string;
   updatedAt: string;
+  ref: {
+    type: "commit" | "tag" | "branch";
+    name: string;
+    hash: string;
+  } | null;
   userId: string | null;
   metadata: ConversationMetadata | null;
+  tenantId: string;
+  agentId: string | null;
   activeSubAgentId: string;
+  title: string | null;
   lastContextResolution: string | null;
+  projectId: string;
 }>;
 /**
  * Get conversation history with filtering and context management
  */
-declare const getConversationHistory: (db: DatabaseClient) => (params: {
+declare const getConversationHistory: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   conversationId: string;
   options?: ConversationHistoryConfig;
@@ -111,37 +148,49 @@ declare const getConversationHistory: (db: DatabaseClient) => (params: {
 /**
  * Get active agent for a conversation
  */
-declare const getActiveAgentForConversation: (db: DatabaseClient) => (params: {
+declare const getActiveAgentForConversation: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   conversationId: string;
 }) => Promise<{
-  tenantId: string;
-  projectId: string;
   id: string;
-  title: string | null;
   createdAt: string;
   updatedAt: string;
+  ref: {
+    type: "commit" | "tag" | "branch";
+    name: string;
+    hash: string;
+  } | null;
   userId: string | null;
   metadata: ConversationMetadata | null;
+  tenantId: string;
+  agentId: string | null;
   activeSubAgentId: string;
+  title: string | null;
   lastContextResolution: string | null;
+  projectId: string;
 } | undefined>;
 /**
  * Set active agent for a conversation (upsert operation)
  */
-declare const setActiveAgentForConversation: (db: DatabaseClient) => (params: {
+declare const setActiveAgentForConversation: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   conversationId: string;
   subAgentId: string;
+  agentId: string;
+  ref: ResolvedRef;
 }) => Promise<void>;
-declare const setActiveAgentForThread: (db: DatabaseClient) => ({
+declare const setActiveAgentForThread: (db: AgentsRunDatabaseClient) => ({
   scopes,
   threadId,
-  subAgentId
+  subAgentId,
+  agentId,
+  ref
 }: {
   scopes: ProjectScopeConfig;
   threadId: string;
   subAgentId: string;
+  agentId: string;
+  ref: ResolvedRef;
 }) => Promise<void>;
 //#endregion
-export { createConversation, createOrGetConversation, deleteConversation, getActiveAgentForConversation, getConversation, getConversationHistory, listConversations, setActiveAgentForConversation, setActiveAgentForThread, updateConversation, updateConversationActiveAgent };
+export { createConversation, createOrGetConversation, deleteConversation, getActiveAgentForConversation, getConversation, getConversationHistory, listConversations, setActiveAgentForConversation, setActiveAgentForThread, updateConversation, updateConversationActiveSubAgent };

package/dist/data-access/{conversations.js → runtime/conversations.js}

@@ -1,8 +1,8 @@
-import { conversations, messages } from "../db/schema.js";
-import { getConversationId } from "../utils/conversations.js";
+import { conversations, messages } from "../../db/runtime/runtime-schema.js";
+import { getConversationId } from "../../utils/conversations.js";
 import { and, count, desc, eq, inArray } from "drizzle-orm";
 
-//#region src/data-access/conversations.ts
+//#region src/data-access/runtime/conversations.ts
 const listConversations = (db) => async (params) => {
 	const { userId, pagination } = params;
 	const page = pagination?.page || 1;
@@ -44,7 +44,7 @@ const deleteConversation = (db) => async (params) => {
 		return false;
 	}
 };
-const updateConversationActiveAgent = (db) => async (params) => {
+const updateConversationActiveSubAgent = (db) => async (params) => {
 	return updateConversation(db)({
 		scopes: params.scopes,
 		conversationId: params.conversationId,
@@ -77,10 +77,12 @@ const createOrGetConversation = (db) => async (input) => {
 		tenantId: input.tenantId,
 		projectId: input.projectId,
 		userId: input.userId,
+		agentId: input.agentId,
 		activeSubAgentId: input.activeSubAgentId,
 		title: input.title,
 		lastContextResolution: input.lastContextResolution,
 		metadata: input.metadata,
+		ref: input.ref,
 		createdAt: (/* @__PURE__ */ new Date()).toISOString(),
 		updatedAt: (/* @__PURE__ */ new Date()).toISOString()
 	};
@@ -158,6 +160,8 @@ const setActiveAgentForConversation = (db) => async (params) => {
 		tenantId: params.scopes.tenantId,
 		projectId: params.scopes.projectId,
 		activeSubAgentId: params.subAgentId,
+		agentId: params.agentId,
+		ref: params.ref,
 		createdAt: (/* @__PURE__ */ new Date()).toISOString(),
 		updatedAt: (/* @__PURE__ */ new Date()).toISOString()
 	}).onConflictDoUpdate({
@@ -172,13 +176,15 @@ const setActiveAgentForConversation = (db) => async (params) => {
 		}
 	});
 };
-const setActiveAgentForThread = (db) => async ({ scopes, threadId, subAgentId }) => {
+const setActiveAgentForThread = (db) => async ({ scopes, threadId, subAgentId, agentId, ref }) => {
 	return setActiveAgentForConversation(db)({
 		scopes,
 		conversationId: threadId,
-		subAgentId
+		subAgentId,
+		agentId,
+		ref
 	});
 };
 
 //#endregion
-export { createConversation, createOrGetConversation, deleteConversation, getActiveAgentForConversation, getConversation, getConversationHistory, listConversations, setActiveAgentForConversation, setActiveAgentForThread, updateConversation, updateConversationActiveAgent };
+export { createConversation, createOrGetConversation, deleteConversation, getActiveAgentForConversation, getConversation, getConversationHistory, listConversations, setActiveAgentForConversation, setActiveAgentForThread, updateConversation, updateConversationActiveSubAgent };