@inkeep/agents-core 0.41.2 → 0.43.0

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

@@ -0,0 +1,120 @@
+import { EvaluationJobFilterCriteria, Filter, ProjectScopeConfig } from "../../types/utility.js";
+import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
+import { ConversationSelect, DatasetRunConversationRelationInsert, DatasetRunConversationRelationSelect, DatasetRunInsert, DatasetRunSelect, EvaluationResultInsert, EvaluationResultSelect, EvaluationResultUpdate, EvaluationRunInsert, EvaluationRunSelect, EvaluationRunUpdate } from "../../types/entities.js";
+
+//#region src/data-access/runtime/evalRuns.d.ts
+declare const getDatasetRunById: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    datasetRunId: string;
+  };
+}) => Promise<DatasetRunSelect | null>;
+declare const listDatasetRuns: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+}) => Promise<DatasetRunSelect[]>;
+declare const listDatasetRunsByConfig: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    datasetRunConfigId: string;
+  };
+}) => Promise<DatasetRunSelect[]>;
+declare const createDatasetRun: (db: AgentsRunDatabaseClient) => (data: DatasetRunInsert) => Promise<DatasetRunSelect>;
+declare const deleteDatasetRun: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    datasetRunId: string;
+  };
+}) => Promise<boolean>;
+declare const getDatasetRunConversationRelations: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    datasetRunId: string;
+  };
+}) => Promise<DatasetRunConversationRelationSelect[]>;
+declare const createDatasetRunConversationRelation: (db: AgentsRunDatabaseClient) => (data: DatasetRunConversationRelationInsert) => Promise<DatasetRunConversationRelationSelect>;
+declare const createDatasetRunConversationRelations: (db: AgentsRunDatabaseClient) => (data: DatasetRunConversationRelationInsert[]) => Promise<DatasetRunConversationRelationSelect[]>;
+declare const deleteDatasetRunConversationRelation: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    datasetRunId: string;
+    conversationId: string;
+  };
+}) => Promise<boolean>;
+declare const deleteDatasetRunConversationRelationsByRun: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    datasetRunId: string;
+  };
+}) => Promise<number>;
+declare const getDatasetRunConversationRelationByConversation: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    conversationId: string;
+  };
+}) => Promise<DatasetRunConversationRelationSelect | null>;
+declare const getEvaluationRunById: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    evaluationRunId: string;
+  };
+}) => Promise<EvaluationRunSelect | null>;
+declare const listEvaluationRuns: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+}) => Promise<EvaluationRunSelect[]>;
+declare const listEvaluationRunsByJobConfigId: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+  evaluationJobConfigId: string;
+}) => Promise<EvaluationRunSelect[]>;
+declare const getEvaluationRunByJobConfigId: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+  evaluationJobConfigId: string;
+}) => Promise<EvaluationRunSelect | null>;
+declare const createEvaluationRun: (db: AgentsRunDatabaseClient) => (data: EvaluationRunInsert) => Promise<EvaluationRunSelect>;
+declare const updateEvaluationRun: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    evaluationRunId: string;
+  };
+  data: EvaluationRunUpdate;
+}) => Promise<EvaluationRunSelect | null>;
+declare const deleteEvaluationRun: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    evaluationRunId: string;
+  };
+}) => Promise<boolean>;
+declare const getEvaluationResultById: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    evaluationResultId: string;
+  };
+}) => Promise<EvaluationResultSelect | null>;
+declare const listEvaluationResults: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+}) => Promise<EvaluationResultSelect[]>;
+declare const listEvaluationResultsByRun: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    evaluationRunId: string;
+  };
+}) => Promise<EvaluationResultSelect[]>;
+declare const listEvaluationResultsByConversation: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    conversationId: string;
+  };
+}) => Promise<EvaluationResultSelect[]>;
+declare const createEvaluationResult: (db: AgentsRunDatabaseClient) => (data: EvaluationResultInsert) => Promise<EvaluationResultSelect>;
+declare const createEvaluationResults: (db: AgentsRunDatabaseClient) => (data: EvaluationResultInsert[]) => Promise<EvaluationResultSelect[]>;
+declare const updateEvaluationResult: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    evaluationResultId: string;
+  };
+  data: EvaluationResultUpdate;
+}) => Promise<EvaluationResultSelect | null>;
+declare const deleteEvaluationResult: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    evaluationResultId: string;
+  };
+}) => Promise<boolean>;
+declare const deleteEvaluationResultsByRun: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig & {
+    evaluationRunId: string;
+  };
+}) => Promise<number>;
+/**
+ * Filter conversations based on evaluation job filter criteria
+ */
+declare const filterConversationsForJob: (db: AgentsRunDatabaseClient) => (params: {
+  scopes: ProjectScopeConfig;
+  jobFilters: Filter<EvaluationJobFilterCriteria> | null | undefined;
+}) => Promise<ConversationSelect[]>;
+//#endregion
+export { createDatasetRun, createDatasetRunConversationRelation, createDatasetRunConversationRelations, createEvaluationResult, createEvaluationResults, createEvaluationRun, deleteDatasetRun, deleteDatasetRunConversationRelation, deleteDatasetRunConversationRelationsByRun, deleteEvaluationResult, deleteEvaluationResultsByRun, deleteEvaluationRun, filterConversationsForJob, getDatasetRunById, getDatasetRunConversationRelationByConversation, getDatasetRunConversationRelations, getEvaluationResultById, getEvaluationRunById, getEvaluationRunByJobConfigId, listDatasetRuns, listDatasetRunsByConfig, listEvaluationResults, listEvaluationResultsByConversation, listEvaluationResultsByRun, listEvaluationRuns, listEvaluationRunsByJobConfigId, updateEvaluationResult, updateEvaluationRun };

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

@@ -0,0 +1,168 @@
+import { conversations, datasetRun, datasetRunConversationRelations, evaluationResult, evaluationRun } from "../../db/runtime/runtime-schema.js";
+import { and, desc, eq, gte, inArray, lte } from "drizzle-orm";
+
+//#region src/data-access/runtime/evalRuns.ts
+const getDatasetRunById = (db) => async (params) => {
+	return (await db.select().from(datasetRun).where(and(eq(datasetRun.tenantId, params.scopes.tenantId), eq(datasetRun.projectId, params.scopes.projectId), eq(datasetRun.id, params.scopes.datasetRunId))).limit(1))[0] ?? null;
+};
+const listDatasetRuns = (db) => async (params) => {
+	return await db.select().from(datasetRun).where(and(eq(datasetRun.tenantId, params.scopes.tenantId), eq(datasetRun.projectId, params.scopes.projectId))).orderBy(desc(datasetRun.createdAt));
+};
+const listDatasetRunsByConfig = (db) => async (params) => {
+	return await db.select().from(datasetRun).where(and(eq(datasetRun.tenantId, params.scopes.tenantId), eq(datasetRun.projectId, params.scopes.projectId), eq(datasetRun.datasetRunConfigId, params.scopes.datasetRunConfigId))).orderBy(desc(datasetRun.createdAt));
+};
+const createDatasetRun = (db) => async (data) => {
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	const [created] = await db.insert(datasetRun).values({
+		...data,
+		createdAt: now,
+		updatedAt: now
+	}).returning();
+	return created;
+};
+const deleteDatasetRun = (db) => async (params) => {
+	return (await db.delete(datasetRun).where(and(eq(datasetRun.tenantId, params.scopes.tenantId), eq(datasetRun.projectId, params.scopes.projectId), eq(datasetRun.id, params.scopes.datasetRunId))).returning()).length > 0;
+};
+const getDatasetRunConversationRelations = (db) => async (params) => {
+	return await db.select().from(datasetRunConversationRelations).where(and(eq(datasetRunConversationRelations.tenantId, params.scopes.tenantId), eq(datasetRunConversationRelations.projectId, params.scopes.projectId), eq(datasetRunConversationRelations.datasetRunId, params.scopes.datasetRunId)));
+};
+const createDatasetRunConversationRelation = (db) => async (data) => {
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	const [created] = await db.insert(datasetRunConversationRelations).values({
+		...data,
+		createdAt: now,
+		updatedAt: now
+	}).returning();
+	return created;
+};
+const createDatasetRunConversationRelations = (db) => async (data) => {
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	const values = data.map((item) => ({
+		...item,
+		createdAt: now,
+		updatedAt: now
+	}));
+	return await db.insert(datasetRunConversationRelations).values(values).returning();
+};
+const deleteDatasetRunConversationRelation = (db) => async (params) => {
+	return (await db.delete(datasetRunConversationRelations).where(and(eq(datasetRunConversationRelations.tenantId, params.scopes.tenantId), eq(datasetRunConversationRelations.projectId, params.scopes.projectId), eq(datasetRunConversationRelations.datasetRunId, params.scopes.datasetRunId), eq(datasetRunConversationRelations.conversationId, params.scopes.conversationId))).returning()).length > 0;
+};
+const deleteDatasetRunConversationRelationsByRun = (db) => async (params) => {
+	return (await db.delete(datasetRunConversationRelations).where(and(eq(datasetRunConversationRelations.tenantId, params.scopes.tenantId), eq(datasetRunConversationRelations.projectId, params.scopes.projectId), eq(datasetRunConversationRelations.datasetRunId, params.scopes.datasetRunId))).returning()).length;
+};
+const getDatasetRunConversationRelationByConversation = (db) => async (params) => {
+	return (await db.select().from(datasetRunConversationRelations).where(and(eq(datasetRunConversationRelations.tenantId, params.scopes.tenantId), eq(datasetRunConversationRelations.projectId, params.scopes.projectId), eq(datasetRunConversationRelations.conversationId, params.scopes.conversationId))).limit(1))[0] || null;
+};
+const getEvaluationRunById = (db) => async (params) => {
+	return (await db.select().from(evaluationRun).where(and(eq(evaluationRun.tenantId, params.scopes.tenantId), eq(evaluationRun.projectId, params.scopes.projectId), eq(evaluationRun.id, params.scopes.evaluationRunId))).limit(1))[0] ?? null;
+};
+const listEvaluationRuns = (db) => async (params) => {
+	return await db.select().from(evaluationRun).where(and(eq(evaluationRun.tenantId, params.scopes.tenantId), eq(evaluationRun.projectId, params.scopes.projectId)));
+};
+const listEvaluationRunsByJobConfigId = (db) => async (params) => {
+	return await db.select().from(evaluationRun).where(and(eq(evaluationRun.tenantId, params.scopes.tenantId), eq(evaluationRun.projectId, params.scopes.projectId), eq(evaluationRun.evaluationJobConfigId, params.evaluationJobConfigId)));
+};
+const getEvaluationRunByJobConfigId = (db) => async (params) => {
+	return (await db.select().from(evaluationRun).where(and(eq(evaluationRun.tenantId, params.scopes.tenantId), eq(evaluationRun.projectId, params.scopes.projectId), eq(evaluationRun.evaluationJobConfigId, params.evaluationJobConfigId))).limit(1))[0] ?? null;
+};
+const createEvaluationRun = (db) => async (data) => {
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	const [created] = await db.insert(evaluationRun).values({
+		...data,
+		createdAt: now,
+		updatedAt: now
+	}).returning();
+	return created;
+};
+const updateEvaluationRun = (db) => async (params) => {
+	const updateData = { updatedAt: (/* @__PURE__ */ new Date()).toISOString() };
+	for (const [key, value] of Object.entries(params.data)) if (value !== void 0) updateData[key] = value;
+	const [updated] = await db.update(evaluationRun).set(updateData).where(and(eq(evaluationRun.tenantId, params.scopes.tenantId), eq(evaluationRun.projectId, params.scopes.projectId), eq(evaluationRun.id, params.scopes.evaluationRunId))).returning();
+	return updated ?? null;
+};
+const deleteEvaluationRun = (db) => async (params) => {
+	return (await db.delete(evaluationRun).where(and(eq(evaluationRun.tenantId, params.scopes.tenantId), eq(evaluationRun.projectId, params.scopes.projectId), eq(evaluationRun.id, params.scopes.evaluationRunId))).returning()).length > 0;
+};
+const getEvaluationResultById = (db) => async (params) => {
+	return await db.query.evaluationResult.findFirst({ where: and(eq(evaluationResult.tenantId, params.scopes.tenantId), eq(evaluationResult.projectId, params.scopes.projectId), eq(evaluationResult.id, params.scopes.evaluationResultId)) }) ?? null;
+};
+const listEvaluationResults = (db) => async (params) => {
+	return await db.query.evaluationResult.findMany({ where: and(eq(evaluationResult.tenantId, params.scopes.tenantId), eq(evaluationResult.projectId, params.scopes.projectId)) });
+};
+const listEvaluationResultsByRun = (db) => async (params) => {
+	return await db.query.evaluationResult.findMany({ where: and(eq(evaluationResult.tenantId, params.scopes.tenantId), eq(evaluationResult.projectId, params.scopes.projectId), eq(evaluationResult.evaluationRunId, params.scopes.evaluationRunId)) });
+};
+const listEvaluationResultsByConversation = (db) => async (params) => {
+	return await db.query.evaluationResult.findMany({ where: and(eq(evaluationResult.tenantId, params.scopes.tenantId), eq(evaluationResult.projectId, params.scopes.projectId), eq(evaluationResult.conversationId, params.scopes.conversationId)) });
+};
+const createEvaluationResult = (db) => async (data) => {
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	const [created] = await db.insert(evaluationResult).values({
+		...data,
+		createdAt: now,
+		updatedAt: now
+	}).returning();
+	return created;
+};
+const createEvaluationResults = (db) => async (data) => {
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	const values = data.map((item) => ({
+		...item,
+		createdAt: now,
+		updatedAt: now
+	}));
+	return await db.insert(evaluationResult).values(values).returning();
+};
+const updateEvaluationResult = (db) => async (params) => {
+	const updateData = { updatedAt: (/* @__PURE__ */ new Date()).toISOString() };
+	for (const [key, value] of Object.entries(params.data)) if (value !== void 0) updateData[key] = value;
+	const [updated] = await db.update(evaluationResult).set(updateData).where(and(eq(evaluationResult.tenantId, params.scopes.tenantId), eq(evaluationResult.projectId, params.scopes.projectId), eq(evaluationResult.id, params.scopes.evaluationResultId))).returning();
+	return updated ?? null;
+};
+const deleteEvaluationResult = (db) => async (params) => {
+	return (await db.delete(evaluationResult).where(and(eq(evaluationResult.tenantId, params.scopes.tenantId), eq(evaluationResult.projectId, params.scopes.projectId), eq(evaluationResult.id, params.scopes.evaluationResultId))).returning()).length > 0;
+};
+const deleteEvaluationResultsByRun = (db) => async (params) => {
+	return (await db.delete(evaluationResult).where(and(eq(evaluationResult.tenantId, params.scopes.tenantId), eq(evaluationResult.projectId, params.scopes.projectId), eq(evaluationResult.evaluationRunId, params.scopes.evaluationRunId))).returning()).length;
+};
+/**
+* Helper to extract plain filter criteria from a Filter wrapper.
+* Currently only handles plain objects - and/or combinators are not yet supported.
+*/
+function extractPlainFilterCriteria(filter) {
+	if (!filter) return null;
+	if ("and" in filter || "or" in filter) return null;
+	return filter;
+}
+/**
+* Filter conversations based on evaluation job filter criteria
+*/
+const filterConversationsForJob = (db) => async (params) => {
+	const { scopes, jobFilters: rawJobFilters } = params;
+	const jobFilters = extractPlainFilterCriteria(rawJobFilters);
+	const { tenantId, projectId } = scopes;
+	const whereConditions = [eq(conversations.tenantId, tenantId), eq(conversations.projectId, projectId)];
+	if (jobFilters?.conversationIds && Array.isArray(jobFilters.conversationIds) && jobFilters.conversationIds.length > 0) whereConditions.push(inArray(conversations.id, jobFilters.conversationIds));
+	if (jobFilters?.dateRange) {
+		const { startDate, endDate } = jobFilters.dateRange;
+		if (startDate) whereConditions.push(gte(conversations.createdAt, startDate));
+		if (endDate) whereConditions.push(lte(conversations.createdAt, endDate));
+	}
+	if (jobFilters?.datasetRunIds && Array.isArray(jobFilters.datasetRunIds) && jobFilters.datasetRunIds.length > 0) {
+		const allConversationIds = /* @__PURE__ */ new Set();
+		for (const datasetRunId of jobFilters.datasetRunIds) {
+			const relations$1 = await getDatasetRunConversationRelations(db)({ scopes: {
+				tenantId,
+				projectId,
+				datasetRunId
+			} });
+			for (const relation of relations$1) allConversationIds.add(relation.conversationId);
+		}
+		if (allConversationIds.size > 0) whereConditions.push(inArray(conversations.id, Array.from(allConversationIds)));
+		else return [];
+	}
+	return await db.select().from(conversations).where(and(...whereConditions));
+};
+
+//#endregion
+export { createDatasetRun, createDatasetRunConversationRelation, createDatasetRunConversationRelations, createEvaluationResult, createEvaluationResults, createEvaluationRun, deleteDatasetRun, deleteDatasetRunConversationRelation, deleteDatasetRunConversationRelationsByRun, deleteEvaluationResult, deleteEvaluationResultsByRun, deleteEvaluationRun, filterConversationsForJob, getDatasetRunById, getDatasetRunConversationRelationByConversation, getDatasetRunConversationRelations, getEvaluationResultById, getEvaluationRunById, getEvaluationRunByJobConfigId, listDatasetRuns, listDatasetRunsByConfig, listEvaluationResults, listEvaluationResultsByConversation, listEvaluationResultsByRun, listEvaluationRuns, listEvaluationRunsByJobConfigId, updateEvaluationResult, updateEvaluationRun };

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

@@ -1,14 +1,14 @@
-import { Artifact } from "../types/a2a.js";
-import { LedgerArtifactSelect } from "../types/entities.js";
-import "../types/index.js";
-import { ProjectScopeConfig } from "../types/utility.js";
-import { DatabaseClient } from "../db/client.js";
+import { Artifact } from "../../types/a2a.js";
+import { ProjectScopeConfig } from "../../types/utility.js";
+import "../../types/index.js";
+import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
+import { LedgerArtifactSelect } from "../../types/entities.js";
 
-//#region src/data-access/ledgerArtifacts.d.ts
+//#region src/data-access/runtime/ledgerArtifacts.d.ts
 /**
  * Atomic upsert operation for a single artifact - prevents race conditions
  */
-declare const upsertLedgerArtifact: (db: DatabaseClient) => (params: {
+declare const upsertLedgerArtifact: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   contextId: string;
   taskId: string;
@@ -21,7 +21,7 @@ declare const upsertLedgerArtifact: (db: DatabaseClient) => (params: {
 /**
  * Save one or more artifacts to the ledger
  */
-declare const addLedgerArtifacts: (db: DatabaseClient) => (params: {
+declare const addLedgerArtifacts: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   contextId: string;
   taskId?: string | null;
@@ -33,7 +33,7 @@ declare const addLedgerArtifacts: (db: DatabaseClient) => (params: {
  * At least one of taskId, toolCallId, toolCallIds, or artifactId must be provided.
  * Use toolCallIds for batch queries to avoid N+1 query problems.
  */
-declare const getLedgerArtifacts: (db: DatabaseClient) => (params: {
+declare const getLedgerArtifacts: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   taskId?: string;
   toolCallId?: string;
@@ -43,28 +43,28 @@ declare const getLedgerArtifacts: (db: DatabaseClient) => (params: {
 /**
  * Get ledger artifacts by context ID
  */
-declare const getLedgerArtifactsByContext: (db: DatabaseClient) => (params: {
+declare const getLedgerArtifactsByContext: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   contextId: string;
 }) => Promise<LedgerArtifactSelect[]>;
 /**
  * Delete ledger artifacts by task ID
  */
-declare const deleteLedgerArtifactsByTask: (db: DatabaseClient) => (params: {
+declare const deleteLedgerArtifactsByTask: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   taskId: string;
 }) => Promise<boolean>;
 /**
  * Delete ledger artifacts by context ID
  */
-declare const deleteLedgerArtifactsByContext: (db: DatabaseClient) => (params: {
+declare const deleteLedgerArtifactsByContext: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   contextId: string;
 }) => Promise<boolean>;
 /**
  * Count ledger artifacts by task ID
  */
-declare const countLedgerArtifactsByTask: (db: DatabaseClient) => (params: {
+declare const countLedgerArtifactsByTask: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   taskId: string;
 }) => Promise<number>;

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

@@ -1,8 +1,8 @@
-import { ledgerArtifacts } from "../db/schema.js";
-import { generateId } from "../utils/conversations.js";
+import { ledgerArtifacts } from "../../db/runtime/runtime-schema.js";
+import { generateId } from "../../utils/conversations.js";
 import { and, count, eq, inArray } from "drizzle-orm";
 
-//#region src/data-access/ledgerArtifacts.ts
+//#region src/data-access/runtime/ledgerArtifacts.ts
 /**
 * Validate artifact data before database insertion
 */

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

@@ -1,22 +1,21 @@
-import { MessageInsert, MessageUpdate } from "../types/entities.js";
-import "../types/index.js";
-import "../index.js";
-import { MessageContent, MessageMetadata, MessageVisibility, PaginationConfig, ProjectScopeConfig } from "../types/utility.js";
-import { DatabaseClient } from "../db/client.js";
+import { MessageContent, MessageMetadata, MessageVisibility, PaginationConfig, ProjectScopeConfig } from "../../types/utility.js";
+import "../../types/index.js";
+import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
+import "../../index.js";
+import { MessageInsert, MessageUpdate } from "../../types/entities.js";
 
-//#region src/data-access/messages.d.ts
-declare const getMessageById: (db: DatabaseClient) => (params: {
+//#region src/data-access/runtime/messages.d.ts
+declare const getMessageById: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   messageId: string;
 }) => Promise<{
-  tenantId: string;
-  projectId: string;
   id: string;
   createdAt: string;
   updatedAt: string;
   metadata: MessageMetadata | null;
-  content: MessageContent;
   role: string;
+  tenantId: string;
+  projectId: string;
   conversationId: string;
   fromSubAgentId: string | null;
   toSubAgentId: string | null;
@@ -24,6 +23,7 @@ declare const getMessageById: (db: DatabaseClient) => (params: {
   toExternalAgentId: string | null;
   fromTeamAgentId: string | null;
   toTeamAgentId: string | null;
+  content: MessageContent;
   visibility: string;
   messageType: string;
   taskId: string | null;
@@ -31,7 +31,7 @@ declare const getMessageById: (db: DatabaseClient) => (params: {
   a2aTaskId: string | null;
   a2aSessionId: string | null;
 } | undefined>;
-declare const listMessages: (db: DatabaseClient) => (params: {
+declare const listMessages: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   pagination: PaginationConfig;
 }) => Promise<{
@@ -57,7 +57,7 @@ declare const listMessages: (db: DatabaseClient) => (params: {
   tenantId: string;
   id: string;
 }[]>;
-declare const getMessagesByConversation: (db: DatabaseClient) => (params: {
+declare const getMessagesByConversation: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   conversationId: string;
   pagination: PaginationConfig;
@@ -84,7 +84,7 @@ declare const getMessagesByConversation: (db: DatabaseClient) => (params: {
   tenantId: string;
   id: string;
 }[]>;
-declare const getMessagesByTask: (db: DatabaseClient) => (params: {
+declare const getMessagesByTask: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   taskId: string;
   pagination: PaginationConfig;
@@ -111,7 +111,7 @@ declare const getMessagesByTask: (db: DatabaseClient) => (params: {
   tenantId: string;
   id: string;
 }[]>;
-declare const getVisibleMessages: (db: DatabaseClient) => (params: {
+declare const getVisibleMessages: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   conversationId: string;
   visibility?: MessageVisibility[];
@@ -139,15 +139,14 @@ declare const getVisibleMessages: (db: DatabaseClient) => (params: {
   tenantId: string;
   id: string;
 }[]>;
-declare const createMessage: (db: DatabaseClient) => (params: MessageInsert) => Promise<{
-  tenantId: string;
-  projectId: string;
+declare const createMessage: (db: AgentsRunDatabaseClient) => (params: MessageInsert) => Promise<{
   id: string;
   createdAt: string;
   updatedAt: string;
   metadata: MessageMetadata | null;
-  content: MessageContent;
   role: string;
+  tenantId: string;
+  projectId: string;
   conversationId: string;
   fromSubAgentId: string | null;
   toSubAgentId: string | null;
@@ -155,6 +154,7 @@ declare const createMessage: (db: DatabaseClient) => (params: MessageInsert) =>
   toExternalAgentId: string | null;
   fromTeamAgentId: string | null;
   toTeamAgentId: string | null;
+  content: MessageContent;
   visibility: string;
   messageType: string;
   taskId: string | null;
@@ -162,7 +162,7 @@ declare const createMessage: (db: DatabaseClient) => (params: MessageInsert) =>
   a2aTaskId: string | null;
   a2aSessionId: string | null;
 }>;
-declare const updateMessage: (db: DatabaseClient) => (params: {
+declare const updateMessage: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   messageId: string;
   data: MessageUpdate;
@@ -189,18 +189,17 @@ declare const updateMessage: (db: DatabaseClient) => (params: {
   tenantId: string;
   id: string;
 }>;
-declare const deleteMessage: (db: DatabaseClient) => (params: {
+declare const deleteMessage: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   messageId: string;
 }) => Promise<{
-  tenantId: string;
-  projectId: string;
   id: string;
   createdAt: string;
   updatedAt: string;
   metadata: MessageMetadata | null;
-  content: MessageContent;
   role: string;
+  tenantId: string;
+  projectId: string;
   conversationId: string;
   fromSubAgentId: string | null;
   toSubAgentId: string | null;
@@ -208,6 +207,7 @@ declare const deleteMessage: (db: DatabaseClient) => (params: {
   toExternalAgentId: string | null;
   fromTeamAgentId: string | null;
   toTeamAgentId: string | null;
+  content: MessageContent;
   visibility: string;
   messageType: string;
   taskId: string | null;
@@ -215,7 +215,7 @@ declare const deleteMessage: (db: DatabaseClient) => (params: {
   a2aTaskId: string | null;
   a2aSessionId: string | null;
 }>;
-declare const countMessagesByConversation: (db: DatabaseClient) => (params: {
+declare const countMessagesByConversation: (db: AgentsRunDatabaseClient) => (params: {
   scopes: ProjectScopeConfig;
   conversationId: string;
 }) => Promise<number>;

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

@@ -1,7 +1,7 @@
-import { messages } from "../db/schema.js";
+import { messages } from "../../db/runtime/runtime-schema.js";
 import { and, asc, count, desc, eq, inArray } from "drizzle-orm";
 
-//#region src/data-access/messages.ts
+//#region src/data-access/runtime/messages.ts
 const getMessageById = (db) => async (params) => {
 	return db.query.messages.findFirst({ where: and(eq(messages.tenantId, params.scopes.tenantId), eq(messages.projectId, params.scopes.projectId), eq(messages.id, params.messageId)) });
 };

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

@@ -1,7 +1,7 @@
-import { UserOrganization } from "../auth/auth-validation-schemas.js";
-import { DatabaseClient } from "../db/client.js";
+import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
+import { UserOrganization } from "../../auth/auth-validation-schemas.js";
 
-//#region src/data-access/organizations.d.ts
+//#region src/data-access/runtime/organizations.d.ts
 
 /**
  * Organization and Member data access layer
@@ -12,14 +12,14 @@ import { DatabaseClient } from "../db/client.js";
  * Queries Better Auth's member and organization tables
  * Returns Date for createdAt (converted to string at API boundary)
  */
-declare const getUserOrganizations: (db: DatabaseClient) => (userId: string) => Promise<Array<Omit<UserOrganization, "createdAt"> & {
+declare const getUserOrganizationsFromDb: (db: AgentsRunDatabaseClient) => (userId: string) => Promise<Array<Omit<UserOrganization, "createdAt"> & {
   createdAt: Date;
 }>>;
 /**
  * Get pending invitations for a user by email
  * Returns invitations with status 'pending' that haven't expired
  */
-declare const getPendingInvitationsByEmail: (db: DatabaseClient) => (email: string) => Promise<{
+declare const getPendingInvitationsByEmail: (db: AgentsRunDatabaseClient) => (email: string) => Promise<{
   id: string;
   email: string;
   organizationId: string;
@@ -34,10 +34,19 @@ declare const getPendingInvitationsByEmail: (db: DatabaseClient) => (email: stri
  * Add user to organization
  * Directly inserts into Better Auth's member table
  */
-declare const addUserToOrganization: (db: DatabaseClient) => (data: {
+declare const addUserToOrganization: (db: AgentsRunDatabaseClient) => (data: {
   userId: string;
   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, getUserOrganizationsFromDb, upsertOrganization };

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

@@ -1,7 +1,7 @@
-import { invitation, member, organization } from "../auth/auth-schema.js";
+import { invitation, member, organization } from "../../auth/auth-schema.js";
 import { and, desc, eq } from "drizzle-orm";
 
-//#region src/data-access/organizations.ts
+//#region src/data-access/runtime/organizations.ts
 /**
 * Organization and Member data access layer
 * All database queries for Better Auth's organization/member tables
@@ -11,7 +11,7 @@ import { and, desc, eq } from "drizzle-orm";
 * Queries Better Auth's member and organization tables
 * Returns Date for createdAt (converted to string at API boundary)
 */
-const getUserOrganizations = (db) => async (userId) => {
+const getUserOrganizationsFromDb = (db) => async (userId) => {
 	return (await db.select({
 		id: member.id,
 		userId: member.userId,
@@ -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, getUserOrganizationsFromDb, upsertOrganization };