@inkeep/agents-core 0.1.0

package/dist/data-access/tools.d.ts

@@ -0,0 +1,277 @@
+import type { DatabaseClient } from '../db/client.js';
+import type { McpTool, McpToolStatus, PaginationConfig, ScopeConfig, ToolInsert, ToolSelect, ToolUpdate } from '../types/index.js';
+export declare const dbResultToMcpTool: (dbResult: ToolSelect) => McpTool;
+export declare const getToolById: (db: DatabaseClient) => (params: {
+    scopes: ScopeConfig;
+    toolId: string;
+}) => Promise<{
+    tenantId: string;
+    projectId: string;
+    id: string;
+    name: string;
+    config: {
+        type: "mcp";
+        mcp: import("../index.js").ToolMcpConfig;
+    };
+    credentialReferenceId: string | null;
+    createdAt: string;
+    updatedAt: string;
+    status: string;
+    headers: Record<string, string> | null;
+    imageUrl: string | null;
+    capabilities: import("../index.js").ToolServerCapabilities | null;
+    lastHealthCheck: string | null;
+    lastError: string | null;
+    availableTools: import("../index.js").McpToolDefinition[] | null;
+    lastToolsSync: string | null;
+} | null>;
+export declare const listTools: (db: DatabaseClient) => (params: {
+    scopes: ScopeConfig;
+    pagination?: PaginationConfig;
+}) => Promise<{
+    data: {
+        tenantId: string;
+        projectId: string;
+        id: string;
+        name: string;
+        config: {
+            type: "mcp";
+            mcp: import("../index.js").ToolMcpConfig;
+        };
+        credentialReferenceId: string | null;
+        headers: Record<string, string> | null;
+        imageUrl: string | null;
+        capabilities: import("../index.js").ToolServerCapabilities | null;
+        status: string;
+        lastHealthCheck: string | null;
+        lastError: string | null;
+        availableTools: import("../index.js").McpToolDefinition[] | null;
+        lastToolsSync: string | null;
+        createdAt: string;
+        updatedAt: string;
+    }[];
+    pagination: {
+        page: number;
+        limit: number;
+        total: number;
+        pages: number;
+    };
+}>;
+export declare const getToolsByStatus: (db: DatabaseClient) => (params: {
+    scopes: ScopeConfig;
+    status: string;
+}) => Promise<{
+    tenantId: string;
+    projectId: string;
+    id: string;
+    name: string;
+    config: {
+        type: "mcp";
+        mcp: import("../index.js").ToolMcpConfig;
+    };
+    credentialReferenceId: string | null;
+    headers: Record<string, string> | null;
+    imageUrl: string | null;
+    capabilities: import("../index.js").ToolServerCapabilities | null;
+    status: string;
+    lastHealthCheck: string | null;
+    lastError: string | null;
+    availableTools: import("../index.js").McpToolDefinition[] | null;
+    lastToolsSync: string | null;
+    createdAt: string;
+    updatedAt: string;
+}[]>;
+export declare const listToolsByStatus: (db: DatabaseClient) => (params: {
+    scopes: ScopeConfig;
+    status: McpToolStatus;
+}) => Promise<{
+    tenantId: string;
+    projectId: string;
+    id: string;
+    name: string;
+    config: {
+        type: "mcp";
+        mcp: import("../index.js").ToolMcpConfig;
+    };
+    credentialReferenceId: string | null;
+    createdAt: string;
+    updatedAt: string;
+    status: string;
+    headers: Record<string, string> | null;
+    imageUrl: string | null;
+    capabilities: import("../index.js").ToolServerCapabilities | null;
+    lastHealthCheck: string | null;
+    lastError: string | null;
+    availableTools: import("../index.js").McpToolDefinition[] | null;
+    lastToolsSync: string | null;
+}[]>;
+export declare const createTool: (db: DatabaseClient) => (params: ToolInsert) => Promise<{
+    tenantId: string;
+    projectId: string;
+    id: string;
+    name: string;
+    config: {
+        type: "mcp";
+        mcp: import("../index.js").ToolMcpConfig;
+    };
+    credentialReferenceId: string | null;
+    createdAt: string;
+    updatedAt: string;
+    status: string;
+    headers: Record<string, string> | null;
+    imageUrl: string | null;
+    capabilities: import("../index.js").ToolServerCapabilities | null;
+    lastHealthCheck: string | null;
+    lastError: string | null;
+    availableTools: import("../index.js").McpToolDefinition[] | null;
+    lastToolsSync: string | null;
+}>;
+export declare const updateTool: (db: DatabaseClient) => (params: {
+    scopes: ScopeConfig;
+    toolId: string;
+    data: ToolUpdate;
+}) => Promise<{
+    tenantId: string;
+    projectId: string;
+    id: string;
+    name: string;
+    config: {
+        type: "mcp";
+        mcp: import("../index.js").ToolMcpConfig;
+    };
+    credentialReferenceId: string | null;
+    createdAt: string;
+    updatedAt: string;
+    status: string;
+    headers: Record<string, string> | null;
+    imageUrl: string | null;
+    capabilities: import("../index.js").ToolServerCapabilities | null;
+    lastHealthCheck: string | null;
+    lastError: string | null;
+    availableTools: import("../index.js").McpToolDefinition[] | null;
+    lastToolsSync: string | null;
+}>;
+export declare const deleteTool: (db: DatabaseClient) => (params: {
+    scopes: ScopeConfig;
+    toolId: string;
+}) => Promise<boolean>;
+export declare const addToolToAgent: (db: DatabaseClient) => (params: {
+    scopes: ScopeConfig;
+    agentId: string;
+    toolId: string;
+}) => Promise<{
+    tenantId: string;
+    projectId: string;
+    id: string;
+    createdAt: string;
+    updatedAt: string;
+    agentId: string;
+    toolId: string;
+}>;
+export declare const removeToolFromAgent: (db: DatabaseClient) => (params: {
+    scopes: ScopeConfig;
+    agentId: string;
+    toolId: string;
+}) => Promise<{
+    tenantId: string;
+    projectId: string;
+    id: string;
+    createdAt: string;
+    updatedAt: string;
+    agentId: string;
+    toolId: string;
+}>;
+/**
+ * Upsert agent-tool relation (create if it doesn't exist, no-op if it does)
+ */
+export declare const upsertAgentToolRelation: (db: DatabaseClient) => (params: {
+    scopes: ScopeConfig;
+    agentId: string;
+    toolId: string;
+}) => Promise<{
+    tenantId: string;
+    projectId: string;
+    id: string;
+    createdAt: string;
+    updatedAt: string;
+    agentId: string;
+    toolId: string;
+}>;
+export declare const updateToolStatus: (db: DatabaseClient) => (params: {
+    scopes: ScopeConfig;
+    toolId: string;
+    status: string;
+    lastHealthCheck?: string;
+    lastError?: string;
+}) => Promise<{
+    tenantId: string;
+    projectId: string;
+    id: string;
+    name: string;
+    config: {
+        type: "mcp";
+        mcp: import("../index.js").ToolMcpConfig;
+    };
+    credentialReferenceId: string | null;
+    createdAt: string;
+    updatedAt: string;
+    status: string;
+    headers: Record<string, string> | null;
+    imageUrl: string | null;
+    capabilities: import("../index.js").ToolServerCapabilities | null;
+    lastHealthCheck: string | null;
+    lastError: string | null;
+    availableTools: import("../index.js").McpToolDefinition[] | null;
+    lastToolsSync: string | null;
+}>;
+/**
+ * Upsert a tool (create if it doesn't exist, update if it does)
+ */
+export declare const upsertTool: (db: DatabaseClient) => (params: {
+    data: ToolInsert;
+}) => Promise<{
+    tenantId: string;
+    projectId: string;
+    id: string;
+    name: string;
+    config: {
+        type: "mcp";
+        mcp: import("../index.js").ToolMcpConfig;
+    };
+    credentialReferenceId: string | null;
+    createdAt: string;
+    updatedAt: string;
+    status: string;
+    headers: Record<string, string> | null;
+    imageUrl: string | null;
+    capabilities: import("../index.js").ToolServerCapabilities | null;
+    lastHealthCheck: string | null;
+    lastError: string | null;
+    availableTools: import("../index.js").McpToolDefinition[] | null;
+    lastToolsSync: string | null;
+}>;
+export declare const getHealthyToolsForAgent: (db: DatabaseClient) => (params: {
+    scopes: ScopeConfig;
+    agentId: string;
+}) => Promise<{
+    tenantId: string;
+    projectId: string;
+    id: string;
+    name: string;
+    config: {
+        type: "mcp";
+        mcp: import("../index.js").ToolMcpConfig;
+    };
+    credentialReferenceId: string | null;
+    headers: Record<string, string> | null;
+    imageUrl: string | null;
+    capabilities: import("../index.js").ToolServerCapabilities | null;
+    status: string;
+    lastHealthCheck: string | null;
+    lastError: string | null;
+    availableTools: import("../index.js").McpToolDefinition[] | null;
+    lastToolsSync: string | null;
+    createdAt: string;
+    updatedAt: string;
+}[]>;
+//# sourceMappingURL=tools.d.ts.map

package/dist/data-access/tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../../src/data-access/tools.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAEtD,OAAO,KAAK,EACV,OAAO,EACP,aAAa,EACb,gBAAgB,EAChB,WAAW,EACX,UAAU,EACV,UAAU,EACV,UAAU,EACX,MAAM,mBAAmB,CAAC;AAG3B,eAAO,MAAM,iBAAiB,GAAI,UAAU,UAAU,KAAG,OA4BxD,CAAC;AAEF,eAAO,MAAM,WAAW,GACrB,IAAI,cAAc,MAAY,QAAQ;IAAE,MAAM,EAAE,WAAW,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE;;;;;;;;;;;;;;;;;;;;SAS7E,CAAC;AAEJ,eAAO,MAAM,SAAS,GACnB,IAAI,cAAc,MACZ,QAAQ;IAAE,MAAM,EAAE,WAAW,CAAC;IAAC,UAAU,CAAC,EAAE,gBAAgB,CAAA;CAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA4BpE,CAAC;AAEJ,eAAO,MAAM,gBAAgB,GAC1B,IAAI,cAAc,MAAY,QAAQ;IAAE,MAAM,EAAE,WAAW,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE;;;;;;;;;;;;;;;;;;;;IAW7E,CAAC;AAEJ,eAAO,MAAM,iBAAiB,GAC3B,IAAI,cAAc,MAAY,QAAQ;IAAE,MAAM,EAAE,WAAW,CAAC;IAAC,MAAM,EAAE,aAAa,CAAA;CAAE;;;;;;;;;;;;;;;;;;;;IAUpF,CAAC;AAEJ,eAAO,MAAM,UAAU,GAAI,IAAI,cAAc,MAAY,QAAQ,UAAU;;;;;;;;;;;;;;;;;;;;EAa1E,CAAC;AAEF,eAAO,MAAM,UAAU,GACpB,IAAI,cAAc,MACZ,QAAQ;IAAE,MAAM,EAAE,WAAW,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,UAAU,CAAA;CAAE;;;;;;;;;;;;;;;;;;;;EAmBvE,CAAC;AAEJ,eAAO,MAAM,UAAU,GACpB,IAAI,cAAc,MAAY,QAAQ;IAAE,MAAM,EAAE,WAAW,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,qBAa7E,CAAC;AAEJ,eAAO,MAAM,cAAc,GACxB,IAAI,cAAc,MACZ,QAAQ;IAAE,MAAM,EAAE,WAAW,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE;;;;;;;;EAkBtE,CAAC;AAEJ,eAAO,MAAM,mBAAmB,GAC7B,IAAI,cAAc,MACZ,QAAQ;IAAE,MAAM,EAAE,WAAW,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE;;;;;;;;EActE,CAAC;AAEJ;;GAEG;AACH,eAAO,MAAM,uBAAuB,GACjC,IAAI,cAAc,MACZ,QAAQ;IAAE,MAAM,EAAE,WAAW,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE;;;;;;;;EAkBtE,CAAC;AAEJ,eAAO,MAAM,gBAAgB,GAC1B,IAAI,cAAc,MACZ,QAAQ;IACb,MAAM,EAAE,WAAW,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;;;;;;;;;;;;;;;;;;;;EAUA,CAAC;AAEJ;;GAEG;AACH,eAAO,MAAM,UAAU,GAAI,IAAI,cAAc,MAAY,QAAQ;IAAE,IAAI,EAAE,UAAU,CAAA;CAAE;;;;;;;;;;;;;;;;;;;;EAyBpF,CAAC;AAGF,eAAO,MAAM,uBAAuB,GACjC,IAAI,cAAc,MAAY,QAAQ;IAAE,MAAM,EAAE,WAAW,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE;;;;;;;;;;;;;;;;;;;;IAiB9E,CAAC"}

package/dist/data-access/tools.js

@@ -0,0 +1,183 @@
+import { and, count, desc, eq } from 'drizzle-orm';
+import { nanoid } from 'nanoid';
+import { agentToolRelations, tools } from '../db/schema.js';
+// Helper function to convert database result to McpTool
+export const dbResultToMcpTool = (dbResult) => {
+    const { headers, capabilities, credentialReferenceId, lastError, availableTools, imageUrl, lastHealthCheck, lastToolsSync, createdAt, updatedAt, ...rest } = dbResult;
+    return {
+        ...rest,
+        status: dbResult.status,
+        capabilities: capabilities || undefined,
+        credentialReferenceId: credentialReferenceId || undefined,
+        lastHealthCheck: lastHealthCheck ? new Date(lastHealthCheck) : undefined,
+        lastToolsSync: lastToolsSync ? new Date(lastToolsSync) : undefined,
+        createdAt: new Date(createdAt),
+        updatedAt: new Date(updatedAt),
+        lastError: lastError || undefined,
+        availableTools: availableTools || undefined,
+        headers: headers || undefined,
+        imageUrl: imageUrl || undefined,
+    };
+};
+export const getToolById = (db) => async (params) => {
+    const result = await db.query.tools.findFirst({
+        where: and(eq(tools.tenantId, params.scopes.tenantId), eq(tools.projectId, params.scopes.projectId), eq(tools.id, params.toolId)),
+    });
+    return result ?? null;
+};
+export const listTools = (db) => async (params) => {
+    const page = params.pagination?.page || 1;
+    const limit = Math.min(params.pagination?.limit || 10, 100);
+    const offset = (page - 1) * limit;
+    const whereClause = and(eq(tools.tenantId, params.scopes.tenantId), eq(tools.projectId, params.scopes.projectId));
+    const [query, totalResult] = await Promise.all([
+        db
+            .select()
+            .from(tools)
+            .where(whereClause)
+            .limit(limit)
+            .offset(offset)
+            .orderBy(desc(tools.createdAt)),
+        db.select({ count: count() }).from(tools).where(whereClause),
+    ]);
+    const total = totalResult[0]?.count || 0;
+    const pages = Math.ceil(total / limit);
+    return {
+        data: query,
+        pagination: { page, limit, total, pages },
+    };
+};
+export const getToolsByStatus = (db) => async (params) => {
+    return db
+        .select()
+        .from(tools)
+        .where(and(eq(tools.tenantId, params.scopes.tenantId), eq(tools.projectId, params.scopes.projectId), eq(tools.status, params.status)));
+};
+export const listToolsByStatus = (db) => async (params) => {
+    const toolsList = await db.query.tools.findMany({
+        where: and(eq(tools.tenantId, params.scopes.tenantId), eq(tools.status, params.status), eq(tools.projectId, params.scopes.projectId)),
+        orderBy: desc(tools.createdAt),
+    });
+    return toolsList;
+};
+export const createTool = (db) => async (params) => {
+    const now = new Date().toISOString();
+    const [created] = await db
+        .insert(tools)
+        .values({
+        ...params,
+        createdAt: now,
+        updatedAt: now,
+    })
+        .returning();
+    return created;
+};
+export const updateTool = (db) => async (params) => {
+    const now = new Date().toISOString();
+    const [updated] = await db
+        .update(tools)
+        .set({
+        ...params.data,
+        updatedAt: now,
+    })
+        .where(and(eq(tools.tenantId, params.scopes.tenantId), eq(tools.projectId, params.scopes.projectId), eq(tools.id, params.toolId)))
+        .returning();
+    return updated ?? null;
+};
+export const deleteTool = (db) => async (params) => {
+    const [deleted] = await db
+        .delete(tools)
+        .where(and(eq(tools.tenantId, params.scopes.tenantId), eq(tools.projectId, params.scopes.projectId), eq(tools.id, params.toolId)))
+        .returning();
+    return deleted ? true : false;
+};
+export const addToolToAgent = (db) => async (params) => {
+    const id = nanoid();
+    const now = new Date().toISOString();
+    const [created] = await db
+        .insert(agentToolRelations)
+        .values({
+        id,
+        tenantId: params.scopes.tenantId,
+        projectId: params.scopes.projectId,
+        agentId: params.agentId,
+        toolId: params.toolId,
+        createdAt: now,
+        updatedAt: now,
+    })
+        .returning();
+    return created;
+};
+export const removeToolFromAgent = (db) => async (params) => {
+    const [deleted] = await db
+        .delete(agentToolRelations)
+        .where(and(eq(agentToolRelations.tenantId, params.scopes.tenantId), eq(agentToolRelations.projectId, params.scopes.projectId), eq(agentToolRelations.agentId, params.agentId), eq(agentToolRelations.toolId, params.toolId)))
+        .returning();
+    return deleted;
+};
+/**
+ * Upsert agent-tool relation (create if it doesn't exist, no-op if it does)
+ */
+export const upsertAgentToolRelation = (db) => async (params) => {
+    // Check if relation already exists
+    const existing = await db.query.agentToolRelations.findFirst({
+        where: and(eq(agentToolRelations.tenantId, params.scopes.tenantId), eq(agentToolRelations.projectId, params.scopes.projectId), eq(agentToolRelations.agentId, params.agentId), eq(agentToolRelations.toolId, params.toolId)),
+    });
+    if (!existing) {
+        // Create the relation if it doesn't exist
+        return await addToolToAgent(db)(params);
+    }
+    // Return existing relation if it already exists
+    return existing;
+};
+export const updateToolStatus = (db) => async (params) => {
+    return updateTool(db)({
+        scopes: params.scopes,
+        toolId: params.toolId,
+        data: {
+            status: params.status,
+            lastHealthCheck: params.lastHealthCheck || new Date().toISOString(),
+            lastError: params.lastError,
+        },
+    });
+};
+/**
+ * Upsert a tool (create if it doesn't exist, update if it does)
+ */
+export const upsertTool = (db) => async (params) => {
+    const scopes = { tenantId: params.data.tenantId, projectId: params.data.projectId };
+    const existing = await getToolById(db)({
+        scopes,
+        toolId: params.data.id,
+    });
+    if (existing) {
+        // Update existing tool
+        return await updateTool(db)({
+            scopes,
+            toolId: params.data.id,
+            data: {
+                name: params.data.name,
+                config: params.data.config,
+                credentialReferenceId: params.data.credentialReferenceId,
+                imageUrl: params.data.imageUrl,
+                headers: params.data.headers,
+            },
+        });
+    }
+    else {
+        // Create new tool
+        return await createTool(db)(params.data);
+    }
+};
+// Get healthy tools for agent execution
+export const getHealthyToolsForAgent = (db) => async (params) => {
+    const healthyTools = await db
+        .select({
+        tool: tools,
+    })
+        .from(tools)
+        .innerJoin(agentToolRelations, and(eq(tools.id, agentToolRelations.toolId), eq(agentToolRelations.agentId, params.agentId), eq(agentToolRelations.tenantId, params.scopes.tenantId)))
+        .where(and(eq(tools.tenantId, params.scopes.tenantId), eq(tools.status, 'healthy')));
+    return healthyTools.map((row) => row.tool);
+};
+//# sourceMappingURL=tools.js.map

package/dist/data-access/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../src/data-access/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,aAAa,CAAC;AACnD,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEhC,OAAO,EAAE,kBAAkB,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AAW5D,wDAAwD;AACxD,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,QAAoB,EAAW,EAAE;IACjE,MAAM,EACJ,OAAO,EACP,YAAY,EACZ,qBAAqB,EACrB,SAAS,EACT,cAAc,EACd,QAAQ,EACR,eAAe,EACf,aAAa,EACb,SAAS,EACT,SAAS,EACT,GAAG,IAAI,EACR,GAAG,QAAQ,CAAC;IACb,OAAO;QACL,GAAG,IAAI;QACP,MAAM,EAAE,QAAQ,CAAC,MAAuB;QACxC,YAAY,EAAE,YAAY,IAAI,SAAS;QACvC,qBAAqB,EAAE,qBAAqB,IAAI,SAAS;QACzD,eAAe,EAAE,eAAe,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,SAAS;QACxE,aAAa,EAAE,aAAa,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS;QAClE,SAAS,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC;QAC9B,SAAS,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC;QAC9B,SAAS,EAAE,SAAS,IAAI,SAAS;QACjC,cAAc,EAAE,cAAc,IAAI,SAAS;QAC3C,OAAO,EAAE,OAAO,IAAI,SAAS;QAC7B,QAAQ,EAAE,QAAQ,IAAI,SAAS;KAChC,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,WAAW,GACtB,CAAC,EAAkB,EAAE,EAAE,CAAC,KAAK,EAAE,MAA+C,EAAE,EAAE;IAChF,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC;QAC5C,KAAK,EAAE,GAAG,CACR,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EAC1C,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,EAC5C,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,MAAM,CAAC,MAAM,CAAC,CAC5B;KACF,CAAC,CAAC;IACH,OAAO,MAAM,IAAI,IAAI,CAAC;AACxB,CAAC,CAAC;AAEJ,MAAM,CAAC,MAAM,SAAS,GACpB,CAAC,EAAkB,EAAE,EAAE,CACvB,KAAK,EAAE,MAA8D,EAAE,EAAE;IACvE,MAAM,IAAI,GAAG,MAAM,CAAC,UAAU,EAAE,IAAI,IAAI,CAAC,CAAC;IAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,KAAK,IAAI,EAAE,EAAE,GAAG,CAAC,CAAC;IAC5D,MAAM,MAAM,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;IAElC,MAAM,WAAW,GAAG,GAAG,CACrB,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EAC1C,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAC7C,CAAC;IAEF,MAAM,CAAC,KAAK,EAAE,WAAW,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;QAC7C,EAAE;aACC,MAAM,EAAE;aACR,IAAI,CAAC,KAAK,CAAC;aACX,KAAK,CAAC,WAAW,CAAC;aAClB,KAAK,CAAC,KAAK,CAAC;aACZ,MAAM,CAAC,MAAM,CAAC;aACd,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACjC,EAAE,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC;KAC7D,CAAC,CAAC;IAEH,MAAM,KAAK,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC;IACzC,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC;IAEvC,OAAO;QACL,IAAI,EAAE,KAAK;QACX,UAAU,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;KAC1C,CAAC;AACJ,CAAC,CAAC;AAEJ,MAAM,CAAC,MAAM,gBAAgB,GAC3B,CAAC,EAAkB,EAAE,EAAE,CAAC,KAAK,EAAE,MAA+C,EAAE,EAAE;IAChF,OAAO,EAAE;SACN,MAAM,EAAE;SACR,IAAI,CAAC,KAAK,CAAC;SACX,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EAC1C,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,EAC5C,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAChC,CACF,CAAC;AACN,CAAC,CAAC;AAEJ,MAAM,CAAC,MAAM,iBAAiB,GAC5B,CAAC,EAAkB,EAAE,EAAE,CAAC,KAAK,EAAE,MAAsD,EAAE,EAAE;IACvF,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC9C,KAAK,EAAE,GAAG,CACR,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EAC1C,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,EAC/B,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAC7C;QACD,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC;KAC/B,CAAC,CAAC;IACH,OAAO,SAAS,CAAC;AACnB,CAAC,CAAC;AAEJ,MAAM,CAAC,MAAM,UAAU,GAAG,CAAC,EAAkB,EAAE,EAAE,CAAC,KAAK,EAAE,MAAkB,EAAE,EAAE;IAC7E,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAErC,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,EAAE;SACvB,MAAM,CAAC,KAAK,CAAC;SACb,MAAM,CAAC;QACN,GAAG,MAAM;QACT,SAAS,EAAE,GAAG;QACd,SAAS,EAAE,GAAG;KACf,CAAC;SACD,SAAS,EAAE,CAAC;IAEf,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,UAAU,GACrB,CAAC,EAAkB,EAAE,EAAE,CACvB,KAAK,EAAE,MAAiE,EAAE,EAAE;IAC1E,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAErC,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,EAAE;SACvB,MAAM,CAAC,KAAK,CAAC;SACb,GAAG,CAAC;QACH,GAAG,MAAM,CAAC,IAAI;QACd,SAAS,EAAE,GAAG;KACf,CAAC;SACD,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EAC1C,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,EAC5C,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,MAAM,CAAC,MAAM,CAAC,CAC5B,CACF;SACA,SAAS,EAAE,CAAC;IAEf,OAAO,OAAO,IAAI,IAAI,CAAC;AACzB,CAAC,CAAC;AAEJ,MAAM,CAAC,MAAM,UAAU,GACrB,CAAC,EAAkB,EAAE,EAAE,CAAC,KAAK,EAAE,MAA+C,EAAE,EAAE;IAChF,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,EAAE;SACvB,MAAM,CAAC,KAAK,CAAC;SACb,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EAC1C,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,EAC5C,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,MAAM,CAAC,MAAM,CAAC,CAC5B,CACF;SACA,SAAS,EAAE,CAAC;IAEf,OAAO,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;AAChC,CAAC,CAAC;AAEJ,MAAM,CAAC,MAAM,cAAc,GACzB,CAAC,EAAkB,EAAE,EAAE,CACvB,KAAK,EAAE,MAAgE,EAAE,EAAE;IACzE,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;IACpB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAErC,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,EAAE;SACvB,MAAM,CAAC,kBAAkB,CAAC;SAC1B,MAAM,CAAC;QACN,EAAE;QACF,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ;QAChC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS;QAClC,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,SAAS,EAAE,GAAG;QACd,SAAS,EAAE,GAAG;KACf,CAAC;SACD,SAAS,EAAE,CAAC;IAEf,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEJ,MAAM,CAAC,MAAM,mBAAmB,GAC9B,CAAC,EAAkB,EAAE,EAAE,CACvB,KAAK,EAAE,MAAgE,EAAE,EAAE;IACzE,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,EAAE;SACvB,MAAM,CAAC,kBAAkB,CAAC;SAC1B,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EACvD,EAAE,CAAC,kBAAkB,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,EACzD,EAAE,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,EAC9C,EAAE,CAAC,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAC7C,CACF;SACA,SAAS,EAAE,CAAC;IAEf,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEJ;;GAEG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAClC,CAAC,EAAkB,EAAE,EAAE,CACvB,KAAK,EAAE,MAAgE,EAAE,EAAE;IACzE,mCAAmC;IACnC,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,KAAK,CAAC,kBAAkB,CAAC,SAAS,CAAC;QAC3D,KAAK,EAAE,GAAG,CACR,EAAE,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EACvD,EAAE,CAAC,kBAAkB,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,EACzD,EAAE,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,EAC9C,EAAE,CAAC,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAC7C;KACF,CAAC,CAAC;IAEH,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,0CAA0C;QAC1C,OAAO,MAAM,cAAc,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;IAC1C,CAAC;IAED,gDAAgD;IAChD,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC;AAEJ,MAAM,CAAC,MAAM,gBAAgB,GAC3B,CAAC,EAAkB,EAAE,EAAE,CACvB,KAAK,EAAE,MAMN,EAAE,EAAE;IACH,OAAO,UAAU,CAAC,EAAE,CAAC,CAAC;QACpB,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,IAAI,EAAE;YACJ,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,eAAe,EAAE,MAAM,CAAC,eAAe,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnE,SAAS,EAAE,MAAM,CAAC,SAAS;SAC5B;KACF,CAAC,CAAC;AACL,CAAC,CAAC;AAEJ;;GAEG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,CAAC,EAAkB,EAAE,EAAE,CAAC,KAAK,EAAE,MAA4B,EAAE,EAAE;IACvF,MAAM,MAAM,GAAG,EAAE,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;IAEpF,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,EAAE,CAAC,CAAC;QACrC,MAAM;QACN,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE;KACvB,CAAC,CAAC;IAEH,IAAI,QAAQ,EAAE,CAAC;QACb,uBAAuB;QACvB,OAAO,MAAM,UAAU,CAAC,EAAE,CAAC,CAAC;YAC1B,MAAM;YACN,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE;YACtB,IAAI,EAAE;gBACJ,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI;gBACtB,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM;gBAC1B,qBAAqB,EAAE,MAAM,CAAC,IAAI,CAAC,qBAAqB;gBACxD,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ;gBAC9B,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO;aAC7B;SACF,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,kBAAkB;QAClB,OAAO,MAAM,UAAU,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC;AACH,CAAC,CAAC;AAEF,wCAAwC;AACxC,MAAM,CAAC,MAAM,uBAAuB,GAClC,CAAC,EAAkB,EAAE,EAAE,CAAC,KAAK,EAAE,MAAgD,EAAE,EAAE;IACjF,MAAM,YAAY,GAAG,MAAM,EAAE;SAC1B,MAAM,CAAC;QACN,IAAI,EAAE,KAAK;KACZ,CAAC;SACD,IAAI,CAAC,KAAK,CAAC;SACX,SAAS,CACR,kBAAkB,EAClB,GAAG,CACD,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,kBAAkB,CAAC,MAAM,CAAC,EACvC,EAAE,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,EAC9C,EAAE,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CACxD,CACF;SACA,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC;IAEvF,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;AAC7C,CAAC,CAAC"}

package/dist/data-access/validation.d.ts

@@ -0,0 +1,17 @@
+import type { DatabaseClient } from '../db/client.js';
+/**
+ * Validates that a project exists before performing database operations
+ * This provides runtime validation even when foreign key constraints are not enforced
+ */
+export declare const validateProjectExists: (db: DatabaseClient, tenantId: string, projectId: string) => Promise<void>;
+/**
+ * Wraps a database operation with project validation
+ * Ensures the project exists before executing the operation
+ */
+export declare const withProjectValidation: <T extends (...args: any[]) => Promise<any>>(db: DatabaseClient, operation: T) => (params: Parameters<T>[0]) => Promise<ReturnType<T>>;
+/**
+ * Creates a validated version of data access functions
+ * Automatically adds project validation to insert/update operations
+ */
+export declare const createValidatedDataAccess: <T extends Record<string, any>>(db: DatabaseClient, dataAccessFunctions: T) => T;
+//# sourceMappingURL=validation.d.ts.map

package/dist/data-access/validation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"validation.d.ts","sourceRoot":"","sources":["../../src/data-access/validation.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAGtD;;;GAGG;AACH,eAAO,MAAM,qBAAqB,GAChC,IAAI,cAAc,EAClB,UAAU,MAAM,EAChB,WAAW,MAAM,KAChB,OAAO,CAAC,IAAI,CAWd,CAAC;AAEF;;;GAGG;AACH,eAAO,MAAM,qBAAqB,GAAI,CAAC,SAAS,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,OAAO,CAAC,GAAG,CAAC,EAC9E,IAAI,cAAc,EAClB,WAAW,CAAC,MAEE,QAAQ,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAG,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAW/D,CAAC;AAEF;;;GAGG;AACH,eAAO,MAAM,yBAAyB,GAAI,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EACrE,IAAI,cAAc,EAClB,qBAAqB,CAAC,KACrB,CAiBF,CAAC"}

package/dist/data-access/validation.js

@@ -0,0 +1,52 @@
+import { projectExistsInTable } from './projects.js';
+/**
+ * Validates that a project exists before performing database operations
+ * This provides runtime validation even when foreign key constraints are not enforced
+ */
+export const validateProjectExists = async (db, tenantId, projectId) => {
+    const exists = await projectExistsInTable(db)({
+        scopes: { tenantId, projectId }
+    });
+    if (!exists) {
+        throw new Error(`Project with ID "${projectId}" does not exist for tenant "${tenantId}". ` +
+            `Please create the project first before adding resources to it.`);
+    }
+};
+/**
+ * Wraps a database operation with project validation
+ * Ensures the project exists before executing the operation
+ */
+export const withProjectValidation = (db, operation) => {
+    return async (params) => {
+        // Extract tenantId and projectId from params
+        const tenantId = params.tenantId || params.scopes?.tenantId;
+        const projectId = params.projectId || params.scopes?.projectId;
+        if (tenantId && projectId) {
+            await validateProjectExists(db, tenantId, projectId);
+        }
+        return operation(params);
+    };
+};
+/**
+ * Creates a validated version of data access functions
+ * Automatically adds project validation to insert/update operations
+ */
+export const createValidatedDataAccess = (db, dataAccessFunctions) => {
+    const validated = {};
+    for (const [key, fn] of Object.entries(dataAccessFunctions)) {
+        if (typeof fn === 'function') {
+            // Add validation to create/insert/update operations
+            if (key.startsWith('create') || key.startsWith('insert') || key.startsWith('update')) {
+                validated[key] = withProjectValidation(db, fn);
+            }
+            else {
+                validated[key] = fn;
+            }
+        }
+        else {
+            validated[key] = fn;
+        }
+    }
+    return validated;
+};
+//# sourceMappingURL=validation.js.map

package/dist/data-access/validation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"validation.js","sourceRoot":"","sources":["../../src/data-access/validation.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAErD;;;GAGG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,KAAK,EACxC,EAAkB,EAClB,QAAgB,EAChB,SAAiB,EACF,EAAE;IACjB,MAAM,MAAM,GAAG,MAAM,oBAAoB,CAAC,EAAE,CAAC,CAAC;QAC5C,MAAM,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE;KAChC,CAAC,CAAC;IAEH,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CACb,oBAAoB,SAAS,gCAAgC,QAAQ,KAAK;YAC1E,gEAAgE,CACjE,CAAC;IACJ,CAAC;AACH,CAAC,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,CACnC,EAAkB,EAClB,SAAY,EACZ,EAAE;IACF,OAAO,KAAK,EAAE,MAAwB,EAA0B,EAAE;QAChE,6CAA6C;QAC7C,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC;QAC5D,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC;QAE/D,IAAI,QAAQ,IAAI,SAAS,EAAE,CAAC;YAC1B,MAAM,qBAAqB,CAAC,EAAE,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;QACvD,CAAC;QAED,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC;IAC3B,CAAC,CAAC;AACJ,CAAC,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,CACvC,EAAkB,EAClB,mBAAsB,EACnB,EAAE;IACL,MAAM,SAAS,GAAG,EAAO,CAAC;IAE1B,KAAK,MAAM,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,EAAE,CAAC;QAC5D,IAAI,OAAO,EAAE,KAAK,UAAU,EAAE,CAAC;YAC7B,oDAAoD;YACpD,IAAI,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACrF,SAAS,CAAC,GAAc,CAAC,GAAG,qBAAqB,CAAC,EAAE,EAAE,EAAE,CAAe,CAAC;YAC1E,CAAC;iBAAM,CAAC;gBACN,SAAS,CAAC,GAAc,CAAC,GAAG,EAAE,CAAC;YACjC,CAAC;QACH,CAAC;aAAM,CAAC;YACN,SAAS,CAAC,GAAc,CAAC,GAAG,EAAE,CAAC;QACjC,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC,CAAC"}

package/dist/db/clean.d.ts

@@ -0,0 +1,6 @@
+/**
+ * Truncates all tables in the database, respecting foreign key constraints
+ * Tables are cleared in dependency order (child tables first, then parent tables)
+ */
+export declare function cleanDatabase(): Promise<void>;
+//# sourceMappingURL=clean.d.ts.map

package/dist/db/clean.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"clean.d.ts","sourceRoot":"","sources":["../../src/db/clean.ts"],"names":[],"mappings":"AA4BA;;;GAGG;AACH,wBAAsB,aAAa,kBA6DlC"}

package/dist/db/clean.js

@@ -0,0 +1,81 @@
+import { sql } from 'drizzle-orm';
+import { env } from '../env.js';
+import { agentArtifactComponents, agentDataComponents, agentGraph, agentRelations, agents, agentToolRelations, artifactComponents, contextCache, contextConfigs, conversations, credentialReferences, dataComponents, externalAgents, ledgerArtifacts, messages, taskRelations, tasks, tools, } from '@inkeep/agents-core';
+import { createDatabaseClient } from './client.js';
+const dbClient = createDatabaseClient({
+    url: env.DB_FILE_NAME,
+});
+/**
+ * Truncates all tables in the database, respecting foreign key constraints
+ * Tables are cleared in dependency order (child tables first, then parent tables)
+ */
+export async function cleanDatabase() {
+    console.log(`🗑️  Cleaning database for environment: ${env.ENVIRONMENT}`);
+    console.log(`📁 Using database: ${env.DB_FILE_NAME}`);
+    console.log('---');
+    try {
+        // Disable foreign key constraints temporarily to make cleanup easier
+        await dbClient.run(sql `PRAGMA foreign_keys = OFF`);
+        // Order matters: clear dependent tables first
+        const tablesToClear = [
+            { table: messages, name: 'messages' },
+            { table: conversations, name: 'conversations' },
+            { table: taskRelations, name: 'task_relations' },
+            { table: tasks, name: 'tasks' },
+            { table: agentArtifactComponents, name: 'agent_artifact_components' },
+            { table: agentDataComponents, name: 'agent_data_components' },
+            { table: agentToolRelations, name: 'agent_tool_relations' },
+            { table: agentRelations, name: 'agent_relations' },
+            { table: agentGraph, name: 'agent_graph' },
+            { table: artifactComponents, name: 'artifact_components' },
+            { table: dataComponents, name: 'data_components' },
+            { table: tools, name: 'tools' },
+            { table: agents, name: 'agents' },
+            { table: externalAgents, name: 'external_agents' },
+            { table: ledgerArtifacts, name: 'ledger_artifacts' },
+            { table: credentialReferences, name: 'credential_references' },
+            { table: contextConfigs, name: 'context_configs' },
+            { table: contextCache, name: 'context_cache' },
+        ];
+        for (const { table, name } of tablesToClear) {
+            try {
+                await dbClient.delete(table).run();
+                console.log(`✅ Cleared table: ${name}`);
+            }
+            catch (error) {
+                // Handle case where table doesn't exist
+                const errorMessage = error?.message || '';
+                const causeMessage = error?.cause?.message || '';
+                if (errorMessage.includes('no such table') ||
+                    causeMessage.includes('no such table') ||
+                    (error?.cause?.code === 'SQLITE_ERROR' && causeMessage.includes('no such table'))) {
+                    console.log(`⚠️  Table ${name} doesn't exist, skipping`);
+                }
+                else {
+                    throw error;
+                }
+            }
+        }
+        // Re-enable foreign key constraints
+        await dbClient.run(sql `PRAGMA foreign_keys = ON`);
+        console.log('---');
+        console.log('🎉 Database cleaned successfully');
+    }
+    catch (error) {
+        console.error('❌ Failed to clean database:', error);
+        throw error;
+    }
+}
+// Run the clean function if executed directly
+if (import.meta.url === new URL(import.meta.url).href) {
+    cleanDatabase()
+        .then(() => {
+        console.log('Database cleanup completed');
+        process.exit(0);
+    })
+        .catch((error) => {
+        console.error('Database cleanup failed:', error);
+        process.exit(1);
+    });
+}
+//# sourceMappingURL=clean.js.map

package/dist/db/clean.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"clean.js","sourceRoot":"","sources":["../../src/db/clean.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,EAAE,GAAG,EAAE,MAAM,WAAW,CAAC;AAChC,OAAO,EACL,uBAAuB,EACvB,mBAAmB,EACnB,UAAU,EACV,cAAc,EACd,MAAM,EACN,kBAAkB,EAClB,kBAAkB,EAClB,YAAY,EACZ,cAAc,EACd,aAAa,EACb,oBAAoB,EACpB,cAAc,EACd,cAAc,EACd,eAAe,EACf,QAAQ,EACR,aAAa,EACb,KAAK,EACL,KAAK,GACN,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAEnD,MAAM,QAAQ,GAAG,oBAAoB,CAAC;IACpC,GAAG,EAAE,GAAG,CAAC,YAAY;CACtB,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa;IACjC,OAAO,CAAC,GAAG,CAAC,2CAA2C,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC;IAC1E,OAAO,CAAC,GAAG,CAAC,sBAAsB,GAAG,CAAC,YAAY,EAAE,CAAC,CAAC;IACtD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAEnB,IAAI,CAAC;QACH,qEAAqE;QACrE,MAAM,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAA,2BAA2B,CAAC,CAAC;QAEnD,8CAA8C;QAC9C,MAAM,aAAa,GAAG;YACpB,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE;YACrC,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,eAAe,EAAE;YAC/C,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,gBAAgB,EAAE;YAChD,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE;YAC/B,EAAE,KAAK,EAAE,uBAAuB,EAAE,IAAI,EAAE,2BAA2B,EAAE;YACrE,EAAE,KAAK,EAAE,mBAAmB,EAAE,IAAI,EAAE,uBAAuB,EAAE;YAC7D,EAAE,KAAK,EAAE,kBAAkB,EAAE,IAAI,EAAE,sBAAsB,EAAE;YAC3D,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,iBAAiB,EAAE;YAClD,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,aAAa,EAAE;YAC1C,EAAE,KAAK,EAAE,kBAAkB,EAAE,IAAI,EAAE,qBAAqB,EAAE;YAC1D,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,iBAAiB,EAAE;YAClD,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE;YAC/B,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE;YACjC,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,iBAAiB,EAAE;YAClD,EAAE,KAAK,EAAE,eAAe,EAAE,IAAI,EAAE,kBAAkB,EAAE;YACpD,EAAE,KAAK,EAAE,oBAAoB,EAAE,IAAI,EAAE,uBAAuB,EAAE;YAC9D,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,iBAAiB,EAAE;YAClD,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,eAAe,EAAE;SAC/C,CAAC;QAEF,KAAK,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,aAAa,EAAE,CAAC;YAC5C,IAAI,CAAC;gBACH,MAAM,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,CAAC;gBACnC,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,EAAE,CAAC,CAAC;YAC1C,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,wCAAwC;gBACxC,MAAM,YAAY,GAAG,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;gBAC1C,MAAM,YAAY,GAAG,KAAK,EAAE,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;gBAEjD,IACE,YAAY,CAAC,QAAQ,CAAC,eAAe,CAAC;oBACtC,YAAY,CAAC,QAAQ,CAAC,eAAe,CAAC;oBACtC,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,KAAK,cAAc,IAAI,YAAY,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,EACjF,CAAC;oBACD,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,0BAA0B,CAAC,CAAC;gBAC3D,CAAC;qBAAM,CAAC;oBACN,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;QACH,CAAC;QAED,oCAAoC;QACpC,MAAM,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAA,0BAA0B,CAAC,CAAC;QAElD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACnB,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAClD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;QACpD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,8CAA8C;AAC9C,IAAI,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;IACtD,aAAa,EAAE;SACZ,IAAI,CAAC,GAAG,EAAE;QACT,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;QAC1C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC;SACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;QACf,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;QACjD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;AACP,CAAC"}