@inkeep/agents-core 0.41.2 → 0.43.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (258) hide show
  1. package/dist/api-client/base-client.d.ts +87 -8
  2. package/dist/api-client/base-client.js +174 -1
  3. package/dist/api-client/eval-api-client.d.ts +47 -0
  4. package/dist/api-client/eval-api-client.js +65 -0
  5. package/dist/api-client/index.d.ts +4 -0
  6. package/dist/api-client/index.js +5 -0
  7. package/dist/api-client/manage-api-client.d.ts +34 -0
  8. package/dist/api-client/manage-api-client.js +104 -0
  9. package/dist/auth/auth.d.ts +86 -20
  10. package/dist/auth/auth.js +60 -2
  11. package/dist/auth/authz/client.d.ts +87 -0
  12. package/dist/auth/authz/client.js +196 -0
  13. package/dist/auth/authz/config.d.ts +103 -0
  14. package/dist/auth/authz/config.js +93 -0
  15. package/dist/auth/authz/index.d.ts +5 -0
  16. package/dist/auth/authz/index.js +6 -0
  17. package/dist/auth/authz/permissions.d.ts +53 -0
  18. package/dist/auth/authz/permissions.js +83 -0
  19. package/dist/auth/authz/sync.d.ts +106 -0
  20. package/dist/auth/authz/sync.js +321 -0
  21. package/dist/auth/permissions.d.ts +13 -13
  22. package/dist/auth/permissions.js +2 -181
  23. package/dist/client-exports.d.ts +9 -3
  24. package/dist/client-exports.js +4 -2
  25. package/dist/constants/context-breakdown.d.ts +61 -0
  26. package/dist/constants/context-breakdown.js +124 -0
  27. package/dist/constants/execution-limits-shared/defaults.d.ts +1 -1
  28. package/dist/constants/execution-limits-shared/defaults.js +1 -1
  29. package/dist/constants/execution-limits-shared/index.d.ts +1 -1
  30. package/dist/constants/otel-attributes.d.ts +4 -0
  31. package/dist/constants/otel-attributes.js +4 -0
  32. package/dist/context/ContextConfig.d.ts +2 -2
  33. package/dist/context/ContextConfig.js +3 -3
  34. package/dist/context/TemplateEngine.d.ts +0 -6
  35. package/dist/context/TemplateEngine.js +4 -19
  36. package/dist/context/index.d.ts +1 -5
  37. package/dist/context/index.js +1 -5
  38. package/dist/credential-stores/keychain-store.d.ts +20 -8
  39. package/dist/credential-stores/keychain-store.js +107 -43
  40. package/dist/credential-stuffer/CredentialStuffer.d.ts +1 -1
  41. package/dist/data-access/index.d.ts +34 -26
  42. package/dist/data-access/index.js +34 -26
  43. package/dist/data-access/manage/agentFull.d.ts +36 -0
  44. package/dist/data-access/{agentFull.js → manage/agentFull.js} +209 -7
  45. package/dist/data-access/{agents.d.ts → manage/agents.d.ts} +64 -63
  46. package/dist/data-access/{agents.js → manage/agents.js} +80 -27
  47. package/dist/data-access/{artifactComponents.d.ts → manage/artifactComponents.d.ts} +33 -33
  48. package/dist/data-access/{artifactComponents.js → manage/artifactComponents.js} +5 -5
  49. package/dist/data-access/{contextConfigs.d.ts → manage/contextConfigs.d.ts} +26 -26
  50. package/dist/data-access/{contextConfigs.js → manage/contextConfigs.js} +3 -3
  51. package/dist/data-access/{credentialReferences.d.ts → manage/credentialReferences.d.ts} +17 -17
  52. package/dist/data-access/{credentialReferences.js → manage/credentialReferences.js} +2 -2
  53. package/dist/data-access/{dataComponents.d.ts → manage/dataComponents.d.ts} +26 -26
  54. package/dist/data-access/{dataComponents.js → manage/dataComponents.js} +7 -7
  55. package/dist/data-access/manage/evalConfig.d.ts +221 -0
  56. package/dist/data-access/manage/evalConfig.js +275 -0
  57. package/dist/data-access/{externalAgents.d.ts → manage/externalAgents.d.ts} +16 -16
  58. package/dist/data-access/{externalAgents.js → manage/externalAgents.js} +2 -2
  59. package/dist/data-access/manage/functionTools.d.ts +242 -0
  60. package/dist/data-access/{functionTools.js → manage/functionTools.js} +124 -30
  61. package/dist/data-access/{functions.d.ts → manage/functions.d.ts} +9 -9
  62. package/dist/data-access/{functions.js → manage/functions.js} +3 -3
  63. package/dist/data-access/manage/projectFull.d.ts +38 -0
  64. package/dist/data-access/{projectFull.js → manage/projectFull.js} +64 -65
  65. package/dist/data-access/manage/projectLifecycle.d.ts +119 -0
  66. package/dist/data-access/manage/projectLifecycle.js +234 -0
  67. package/dist/data-access/manage/projects.d.ts +75 -0
  68. package/dist/data-access/{projects.js → manage/projects.js} +15 -16
  69. package/dist/data-access/{subAgentExternalAgentRelations.d.ts → manage/subAgentExternalAgentRelations.d.ts} +43 -43
  70. package/dist/data-access/{subAgentExternalAgentRelations.js → manage/subAgentExternalAgentRelations.js} +2 -2
  71. package/dist/data-access/{subAgentRelations.d.ts → manage/subAgentRelations.d.ts} +65 -65
  72. package/dist/data-access/{subAgentRelations.js → manage/subAgentRelations.js} +3 -3
  73. package/dist/data-access/{subAgentTeamAgentRelations.d.ts → manage/subAgentTeamAgentRelations.d.ts} +43 -43
  74. package/dist/data-access/{subAgentTeamAgentRelations.js → manage/subAgentTeamAgentRelations.js} +2 -2
  75. package/dist/data-access/{subAgents.d.ts → manage/subAgents.d.ts} +28 -28
  76. package/dist/data-access/{subAgents.js → manage/subAgents.js} +4 -4
  77. package/dist/data-access/{tools.d.ts → manage/tools.d.ts} +65 -52
  78. package/dist/data-access/{tools.js → manage/tools.js} +109 -64
  79. package/dist/data-access/manage/triggers.d.ts +106 -0
  80. package/dist/data-access/manage/triggers.js +81 -0
  81. package/dist/data-access/{apiKeys.d.ts → runtime/apiKeys.d.ts} +37 -37
  82. package/dist/data-access/{apiKeys.js → runtime/apiKeys.js} +3 -3
  83. package/dist/data-access/runtime/cascade-delete.d.ts +77 -0
  84. package/dist/data-access/runtime/cascade-delete.js +111 -0
  85. package/dist/data-access/{contextCache.d.ts → runtime/contextCache.d.ts} +13 -13
  86. package/dist/data-access/{contextCache.js → runtime/contextCache.js} +5 -5
  87. package/dist/data-access/{conversations.d.ts → runtime/conversations.d.ts} +80 -31
  88. package/dist/data-access/{conversations.js → runtime/conversations.js} +13 -7
  89. package/dist/data-access/runtime/evalRuns.d.ts +120 -0
  90. package/dist/data-access/runtime/evalRuns.js +168 -0
  91. package/dist/data-access/{ledgerArtifacts.d.ts → runtime/ledgerArtifacts.d.ts} +13 -13
  92. package/dist/data-access/{ledgerArtifacts.js → runtime/ledgerArtifacts.js} +3 -3
  93. package/dist/data-access/{messages.d.ts → runtime/messages.d.ts} +24 -24
  94. package/dist/data-access/{messages.js → runtime/messages.js} +2 -2
  95. package/dist/data-access/{organizations.d.ts → runtime/organizations.d.ts} +16 -7
  96. package/dist/data-access/{organizations.js → runtime/organizations.js} +16 -4
  97. package/dist/data-access/runtime/projects.d.ts +62 -0
  98. package/dist/data-access/runtime/projects.js +90 -0
  99. package/dist/data-access/runtime/tasks.d.ts +55 -0
  100. package/dist/data-access/{tasks.js → runtime/tasks.js} +2 -2
  101. package/dist/data-access/runtime/triggerInvocations.d.ts +62 -0
  102. package/dist/data-access/runtime/triggerInvocations.js +54 -0
  103. package/dist/data-access/runtime/users.d.ts +19 -0
  104. package/dist/data-access/{users.js → runtime/users.js} +2 -2
  105. package/dist/data-access/validation.d.ts +4 -4
  106. package/dist/data-access/validation.js +1 -1
  107. package/dist/db/clean.d.ts +8 -4
  108. package/dist/db/clean.js +14 -105
  109. package/dist/db/delete.d.ts +1 -1
  110. package/dist/db/delete.js +7 -10
  111. package/dist/db/manage/dolt-cleanup.d.ts +51 -0
  112. package/dist/db/manage/dolt-cleanup.js +132 -0
  113. package/dist/db/manage/manage-client.d.ts +26 -0
  114. package/dist/db/manage/manage-client.js +68 -0
  115. package/dist/db/{schema.d.ts → manage/manage-schema.d.ts} +1257 -969
  116. package/dist/db/{schema.js → manage/manage-schema.js} +436 -334
  117. package/dist/db/manage/test-manage-client.d.ts +27 -0
  118. package/dist/db/manage/test-manage-client.js +68 -0
  119. package/dist/db/runtime/runtime-client.d.ts +20 -0
  120. package/dist/db/runtime/runtime-client.js +30 -0
  121. package/dist/db/runtime/runtime-schema.d.ts +2834 -0
  122. package/dist/db/runtime/runtime-schema.js +483 -0
  123. package/dist/db/runtime/test-runtime-client.d.ts +27 -0
  124. package/dist/db/{test-client.js → runtime/test-runtime-client.js} +11 -25
  125. package/dist/db/utils.d.ts +6 -0
  126. package/dist/db/utils.js +42 -0
  127. package/dist/dolt/branch.d.ts +62 -0
  128. package/dist/dolt/branch.js +82 -0
  129. package/dist/dolt/branches-api.d.ts +108 -0
  130. package/dist/dolt/branches-api.js +162 -0
  131. package/dist/dolt/commit.d.ts +94 -0
  132. package/dist/dolt/commit.js +103 -0
  133. package/dist/dolt/diff.d.ts +27 -0
  134. package/dist/dolt/diff.js +21 -0
  135. package/dist/dolt/index.d.ts +10 -0
  136. package/dist/dolt/index.js +11 -0
  137. package/dist/dolt/merge.d.ts +63 -0
  138. package/dist/dolt/merge.js +81 -0
  139. package/dist/dolt/migrate-all-branches.d.ts +4 -0
  140. package/dist/dolt/migrate-all-branches.js +83 -0
  141. package/dist/dolt/migrate-dolt.d.ts +1 -0
  142. package/dist/dolt/migrate-dolt.js +25 -0
  143. package/dist/dolt/ref-helpers.d.ts +19 -0
  144. package/dist/dolt/ref-helpers.js +65 -0
  145. package/dist/dolt/ref-middleware.d.ts +82 -0
  146. package/dist/dolt/ref-middleware.js +217 -0
  147. package/dist/dolt/ref-scope.d.ts +101 -0
  148. package/dist/dolt/ref-scope.js +231 -0
  149. package/dist/dolt/schema-sync.d.ts +135 -0
  150. package/dist/dolt/schema-sync.js +255 -0
  151. package/dist/env.d.ts +6 -4
  152. package/dist/env.js +3 -2
  153. package/dist/index.d.ts +73 -46
  154. package/dist/index.js +76 -49
  155. package/dist/types/@napi-rs__keyring/index.d.ts +14 -0
  156. package/dist/types/entities.d.ts +81 -2
  157. package/dist/types/index.d.ts +3 -3
  158. package/dist/types/utility.d.ts +46 -5
  159. package/dist/types/utility.js +2 -1
  160. package/dist/utils/JsonTransformer.d.ts +42 -0
  161. package/dist/utils/JsonTransformer.js +103 -0
  162. package/dist/utils/apiKeys.d.ts +5 -1
  163. package/dist/utils/apiKeys.js +11 -1
  164. package/dist/utils/colors.d.ts +34 -0
  165. package/dist/utils/colors.js +49 -0
  166. package/dist/utils/credential-store-utils.d.ts +1 -1
  167. package/dist/utils/format-messages.d.ts +1 -1
  168. package/dist/utils/index.d.ts +8 -4
  169. package/dist/utils/index.js +8 -4
  170. package/dist/utils/internal-service-auth.d.ts +79 -0
  171. package/dist/utils/internal-service-auth.js +140 -0
  172. package/dist/utils/jmespath-utils.d.ts +152 -0
  173. package/dist/utils/jmespath-utils.js +213 -0
  174. package/dist/utils/jwt-helpers.d.ts +56 -0
  175. package/dist/utils/jwt-helpers.js +90 -0
  176. package/dist/utils/mcp-client.d.ts +1 -1
  177. package/dist/utils/mcp-client.js +1 -1
  178. package/dist/utils/service-token-auth.d.ts +9 -27
  179. package/dist/utils/service-token-auth.js +48 -96
  180. package/dist/utils/signature-validation.d.ts +2 -0
  181. package/dist/utils/signature-validation.js +3 -0
  182. package/dist/utils/template-interpolation.d.ts +22 -0
  183. package/dist/utils/template-interpolation.js +62 -0
  184. package/dist/utils/third-party-mcp-servers/composio-client.d.ts +13 -1
  185. package/dist/utils/third-party-mcp-servers/composio-client.js +47 -29
  186. package/dist/utils/third-party-mcp-servers/index.d.ts +2 -2
  187. package/dist/utils/third-party-mcp-servers/index.js +2 -2
  188. package/dist/utils/trigger-auth.d.ts +85 -0
  189. package/dist/utils/trigger-auth.js +233 -0
  190. package/dist/validation/agentFull.js +2 -4
  191. package/dist/validation/dolt-schemas.d.ts +49 -0
  192. package/dist/validation/dolt-schemas.js +44 -0
  193. package/dist/validation/drizzle-schema-helpers.d.ts +4 -26
  194. package/dist/validation/drizzle-schema-helpers.js +5 -151
  195. package/dist/validation/index.d.ts +5 -4
  196. package/dist/validation/index.js +4 -3
  197. package/dist/validation/render-validation.js +19 -0
  198. package/dist/validation/schemas.d.ts +18223 -5148
  199. package/dist/validation/schemas.js +559 -12
  200. package/dist/validation/stream-event-schemas.d.ts +96 -1
  201. package/dist/validation/stream-event-schemas.js +67 -2
  202. package/drizzle/manage/0000_tearful_rhodey.sql +414 -0
  203. package/drizzle/manage/0001_broken_wendell_vaughn.sql +19 -0
  204. package/drizzle/manage/0002_bent_sunfire.sql +1 -0
  205. package/drizzle/manage/0003_tiny_captain_universe.sql +8 -0
  206. package/drizzle/manage/0004_curious_phil_sheldon.sql +2 -0
  207. package/drizzle/manage/0005_silent_shatterstar.sql +53 -0
  208. package/drizzle/manage/meta/0000_snapshot.json +2987 -0
  209. package/drizzle/manage/meta/0001_snapshot.json +3115 -0
  210. package/drizzle/manage/meta/0002_snapshot.json +3115 -0
  211. package/drizzle/manage/meta/0003_snapshot.json +3134 -0
  212. package/drizzle/manage/meta/0004_snapshot.json +3141 -0
  213. package/drizzle/manage/meta/0005_snapshot.json +3141 -0
  214. package/drizzle/manage/meta/_journal.json +48 -0
  215. package/drizzle/runtime/0008_silly_preak.sql +127 -0
  216. package/drizzle/runtime/0009_freezing_leo.sql +17 -0
  217. package/drizzle/runtime/meta/0008_snapshot.json +2263 -0
  218. package/drizzle/runtime/meta/0009_snapshot.json +2397 -0
  219. package/drizzle/{meta → runtime/meta}/_journal.json +14 -0
  220. package/package.json +56 -18
  221. package/spicedb/schema.zed +114 -0
  222. package/dist/context/ContextFetcher.d.ts +0 -73
  223. package/dist/context/ContextFetcher.js +0 -291
  224. package/dist/context/ContextResolver.d.ts +0 -60
  225. package/dist/context/ContextResolver.js +0 -278
  226. package/dist/context/context.d.ts +0 -27
  227. package/dist/context/context.js +0 -128
  228. package/dist/context/contextCache.d.ts +0 -58
  229. package/dist/context/contextCache.js +0 -177
  230. package/dist/data-access/agentFull.d.ts +0 -33
  231. package/dist/data-access/functionTools.d.ts +0 -169
  232. package/dist/data-access/projectFull.d.ts +0 -32
  233. package/dist/data-access/projects.d.ts +0 -71
  234. package/dist/data-access/tasks.d.ts +0 -45
  235. package/dist/data-access/users.d.ts +0 -19
  236. package/dist/db/client.d.ts +0 -20
  237. package/dist/db/client.js +0 -28
  238. package/dist/db/test-client.d.ts +0 -31
  239. package/dist/middleware/contextValidation.d.ts +0 -46
  240. package/dist/middleware/contextValidation.js +0 -280
  241. package/dist/middleware/index.d.ts +0 -2
  242. package/dist/middleware/index.js +0 -3
  243. package/dist/utils/execution.d.ts +0 -22
  244. package/dist/utils/execution.js +0 -25
  245. /package/drizzle/{0000_exotic_mysterio.sql → runtime/0000_exotic_mysterio.sql} +0 -0
  246. /package/drizzle/{0001_calm_sheva_callister.sql → runtime/0001_calm_sheva_callister.sql} +0 -0
  247. /package/drizzle/{0002_puzzling_goblin_queen.sql → runtime/0002_puzzling_goblin_queen.sql} +0 -0
  248. /package/drizzle/{0003_sweet_human_robot.sql → runtime/0003_sweet_human_robot.sql} +0 -0
  249. /package/drizzle/{0004_cuddly_shooting_star.sql → runtime/0004_cuddly_shooting_star.sql} +0 -0
  250. /package/drizzle/{0005_reflective_starfox.sql → runtime/0005_reflective_starfox.sql} +0 -0
  251. /package/drizzle/{0006_stale_thaddeus_ross.sql → runtime/0006_stale_thaddeus_ross.sql} +0 -0
  252. /package/drizzle/{0007_slim_karma.sql → runtime/0007_slim_karma.sql} +0 -0
  253. /package/drizzle/{meta → runtime/meta}/0000_snapshot.json +0 -0
  254. /package/drizzle/{meta → runtime/meta}/0001_snapshot.json +0 -0
  255. /package/drizzle/{meta → runtime/meta}/0003_snapshot.json +0 -0
  256. /package/drizzle/{meta → runtime/meta}/0005_snapshot.json +0 -0
  257. /package/drizzle/{meta → runtime/meta}/0006_snapshot.json +0 -0
  258. /package/drizzle/{meta → runtime/meta}/0007_snapshot.json +0 -0
@@ -0,0 +1,62 @@
1
+ import { PaginationConfig } from "../../types/utility.js";
2
+ import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
3
+ import { ProjectMetadataInsert, ProjectMetadataSelect } from "../../types/entities.js";
4
+
5
+ //#region src/data-access/runtime/projects.d.ts
6
+ interface ProjectMetadataPaginatedResult {
7
+ data: ProjectMetadataSelect[];
8
+ pagination: {
9
+ page: number;
10
+ limit: number;
11
+ total: number;
12
+ pages: number;
13
+ };
14
+ }
15
+ /**
16
+ * Get a project from the runtime DB by ID
17
+ */
18
+ declare const getProjectMetadata: (db: AgentsRunDatabaseClient) => (params: {
19
+ tenantId: string;
20
+ projectId: string;
21
+ }) => Promise<ProjectMetadataSelect | null>;
22
+ /**
23
+ * List all runtimeProjects for a tenant from the runtime DB
24
+ */
25
+ declare const listProjectsMetadata: (db: AgentsRunDatabaseClient) => (params: {
26
+ tenantId: string;
27
+ }) => Promise<ProjectMetadataSelect[]>;
28
+ /**
29
+ * List runtimeProjects with pagination from the runtime DB
30
+ * @param projectIds - Optional array of project IDs to filter by. If undefined, returns all projects.
31
+ */
32
+ declare const listProjectsMetadataPaginated: (db: AgentsRunDatabaseClient) => (params: {
33
+ tenantId: string;
34
+ pagination?: PaginationConfig;
35
+ projectIds?: string[];
36
+ }) => Promise<ProjectMetadataPaginatedResult>;
37
+ /**
38
+ * Create a project in the runtime DB
39
+ */
40
+ declare const createProjectMetadata: (db: AgentsRunDatabaseClient) => (params: ProjectMetadataInsert) => Promise<ProjectMetadataSelect>;
41
+ /**
42
+ * Delete a project from the runtime DB
43
+ */
44
+ declare const deleteProjectMetadata: (db: AgentsRunDatabaseClient) => (params: {
45
+ tenantId: string;
46
+ projectId: string;
47
+ }) => Promise<boolean>;
48
+ /**
49
+ * Check if a project exists in the runtime DB
50
+ */
51
+ declare const projectsMetadataExists: (db: AgentsRunDatabaseClient) => (params: {
52
+ tenantId: string;
53
+ projectId: string;
54
+ }) => Promise<boolean>;
55
+ /**
56
+ * Count runtimeProjects for a tenant
57
+ */
58
+ declare const countProjectsInRuntime: (db: AgentsRunDatabaseClient) => (params: {
59
+ tenantId: string;
60
+ }) => Promise<number>;
61
+ //#endregion
62
+ export { ProjectMetadataPaginatedResult, countProjectsInRuntime, createProjectMetadata, deleteProjectMetadata, getProjectMetadata, listProjectsMetadata, listProjectsMetadataPaginated, projectsMetadataExists };
@@ -0,0 +1,90 @@
1
+ import { projectMetadata } from "../../db/runtime/runtime-schema.js";
2
+ import { and, count, desc, eq, inArray } from "drizzle-orm";
3
+
4
+ //#region src/data-access/runtime/projects.ts
5
+ /**
6
+ * Get a project from the runtime DB by ID
7
+ */
8
+ const getProjectMetadata = (db) => async (params) => {
9
+ return await db.query.projectMetadata.findFirst({ where: and(eq(projectMetadata.tenantId, params.tenantId), eq(projectMetadata.id, params.projectId)) }) ?? null;
10
+ };
11
+ /**
12
+ * List all runtimeProjects for a tenant from the runtime DB
13
+ */
14
+ const listProjectsMetadata = (db) => async (params) => {
15
+ return await db.query.projectMetadata.findMany({
16
+ where: eq(projectMetadata.tenantId, params.tenantId),
17
+ orderBy: [desc(projectMetadata.createdAt)]
18
+ });
19
+ };
20
+ /**
21
+ * List runtimeProjects with pagination from the runtime DB
22
+ * @param projectIds - Optional array of project IDs to filter by. If undefined, returns all projects.
23
+ */
24
+ const listProjectsMetadataPaginated = (db) => async (params) => {
25
+ const page = params.pagination?.page || 1;
26
+ const limit = Math.min(params.pagination?.limit || 10, 100);
27
+ const offset = (page - 1) * limit;
28
+ const conditions = [eq(projectMetadata.tenantId, params.tenantId)];
29
+ if (params.projectIds !== void 0) {
30
+ if (params.projectIds.length === 0) return {
31
+ data: [],
32
+ pagination: {
33
+ page,
34
+ limit,
35
+ total: 0,
36
+ pages: 0
37
+ }
38
+ };
39
+ conditions.push(inArray(projectMetadata.id, params.projectIds));
40
+ }
41
+ const whereClause = and(...conditions);
42
+ const [data, totalResult] = await Promise.all([db.select().from(projectMetadata).where(whereClause).limit(limit).offset(offset).orderBy(desc(projectMetadata.createdAt)), db.select({ count: count() }).from(projectMetadata).where(whereClause)]);
43
+ const total = totalResult[0]?.count || 0;
44
+ const totalNumber = typeof total === "string" ? Number.parseInt(total, 10) : total;
45
+ return {
46
+ data,
47
+ pagination: {
48
+ page,
49
+ limit,
50
+ total: totalNumber,
51
+ pages: Math.ceil(totalNumber / limit)
52
+ }
53
+ };
54
+ };
55
+ /**
56
+ * Create a project in the runtime DB
57
+ */
58
+ const createProjectMetadata = (db) => async (params) => {
59
+ const now = (/* @__PURE__ */ new Date()).toISOString();
60
+ const [project] = await db.insert(projectMetadata).values({
61
+ id: params.id,
62
+ tenantId: params.tenantId,
63
+ createdBy: params.createdBy ?? null,
64
+ mainBranchName: params.mainBranchName,
65
+ createdAt: now
66
+ }).returning();
67
+ return project;
68
+ };
69
+ /**
70
+ * Delete a project from the runtime DB
71
+ */
72
+ const deleteProjectMetadata = (db) => async (params) => {
73
+ return (await db.delete(projectMetadata).where(and(eq(projectMetadata.tenantId, params.tenantId), eq(projectMetadata.id, params.projectId))).returning()).length > 0;
74
+ };
75
+ /**
76
+ * Check if a project exists in the runtime DB
77
+ */
78
+ const projectsMetadataExists = (db) => async (params) => {
79
+ return await getProjectMetadata(db)(params) !== null;
80
+ };
81
+ /**
82
+ * Count runtimeProjects for a tenant
83
+ */
84
+ const countProjectsInRuntime = (db) => async (params) => {
85
+ const total = (await db.select({ count: count() }).from(projectMetadata).where(eq(projectMetadata.tenantId, params.tenantId)))[0]?.count || 0;
86
+ return typeof total === "string" ? Number.parseInt(total, 10) : total;
87
+ };
88
+
89
+ //#endregion
90
+ export { countProjectsInRuntime, createProjectMetadata, deleteProjectMetadata, getProjectMetadata, listProjectsMetadata, listProjectsMetadataPaginated, projectsMetadataExists };
@@ -0,0 +1,55 @@
1
+ import { TaskMetadataConfig } from "../../types/utility.js";
2
+ import "../../types/index.js";
3
+ import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
4
+ import "../../index.js";
5
+ import { TaskInsert, TaskSelect } from "../../types/entities.js";
6
+
7
+ //#region src/data-access/runtime/tasks.d.ts
8
+ declare const createTask: (db: AgentsRunDatabaseClient) => (params: TaskInsert) => Promise<{
9
+ id: string;
10
+ createdAt: string;
11
+ updatedAt: string;
12
+ ref: {
13
+ type: "commit" | "tag" | "branch";
14
+ name: string;
15
+ hash: string;
16
+ } | null;
17
+ metadata: TaskMetadataConfig | null;
18
+ status: string;
19
+ tenantId: string;
20
+ agentId: string;
21
+ projectId: string;
22
+ contextId: string;
23
+ subAgentId: string;
24
+ }>;
25
+ declare const getTask: (db: AgentsRunDatabaseClient) => (params: {
26
+ id: string;
27
+ }) => Promise<TaskSelect | null>;
28
+ declare const updateTask: (db: AgentsRunDatabaseClient) => (params: {
29
+ taskId: string;
30
+ data: {
31
+ status?: string;
32
+ metadata?: any;
33
+ };
34
+ }) => Promise<{
35
+ createdAt: string;
36
+ updatedAt: string;
37
+ contextId: string;
38
+ ref: {
39
+ type: "commit" | "tag" | "branch";
40
+ name: string;
41
+ hash: string;
42
+ } | null;
43
+ status: string;
44
+ metadata: TaskMetadataConfig | null;
45
+ subAgentId: string;
46
+ agentId: string;
47
+ projectId: string;
48
+ tenantId: string;
49
+ id: string;
50
+ }>;
51
+ declare const listTaskIdsByContextId: (db: AgentsRunDatabaseClient) => (params: {
52
+ contextId: string;
53
+ }) => Promise<string[]>;
54
+ //#endregion
55
+ export { createTask, getTask, listTaskIdsByContextId, updateTask };
@@ -1,7 +1,7 @@
1
- import { tasks } from "../db/schema.js";
1
+ import { tasks } from "../../db/runtime/runtime-schema.js";
2
2
  import { and, eq } from "drizzle-orm";
3
3
 
4
- //#region src/data-access/tasks.ts
4
+ //#region src/data-access/runtime/tasks.ts
5
5
  const createTask = (db) => async (params) => {
6
6
  const now = (/* @__PURE__ */ new Date()).toISOString();
7
7
  const [created] = await db.insert(tasks).values({
@@ -0,0 +1,62 @@
1
+ import { AgentScopeConfig, PaginationConfig } from "../../types/utility.js";
2
+ import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
3
+ import { TriggerInvocationInsert, TriggerInvocationSelect, TriggerInvocationUpdate } from "../../types/entities.js";
4
+
5
+ //#region src/data-access/runtime/triggerInvocations.d.ts
6
+
7
+ /**
8
+ * Get a trigger invocation by ID (agent-scoped)
9
+ */
10
+ declare const getTriggerInvocationById: (db: AgentsRunDatabaseClient) => (params: {
11
+ scopes: AgentScopeConfig;
12
+ triggerId: string;
13
+ invocationId: string;
14
+ }) => Promise<TriggerInvocationSelect | undefined>;
15
+ /**
16
+ * List trigger invocations with optional filtering (agent-scoped)
17
+ */
18
+ declare const listTriggerInvocationsPaginated: (db: AgentsRunDatabaseClient) => (params: {
19
+ scopes: AgentScopeConfig;
20
+ triggerId: string;
21
+ pagination?: PaginationConfig;
22
+ filters?: {
23
+ status?: "pending" | "success" | "failed";
24
+ from?: string;
25
+ to?: string;
26
+ };
27
+ }) => Promise<{
28
+ data: {
29
+ triggerId: string;
30
+ conversationId: string | null;
31
+ status: string;
32
+ requestPayload: unknown;
33
+ transformedPayload: unknown;
34
+ errorMessage: string | null;
35
+ createdAt: string;
36
+ agentId: string;
37
+ projectId: string;
38
+ tenantId: string;
39
+ id: string;
40
+ }[];
41
+ pagination: {
42
+ page: number;
43
+ limit: number;
44
+ total: number;
45
+ pages: number;
46
+ };
47
+ }>;
48
+ /**
49
+ * Create a new trigger invocation (agent-scoped)
50
+ */
51
+ declare const createTriggerInvocation: (db: AgentsRunDatabaseClient) => (params: TriggerInvocationInsert) => Promise<TriggerInvocationSelect>;
52
+ /**
53
+ * Update trigger invocation status (agent-scoped)
54
+ */
55
+ declare const updateTriggerInvocationStatus: (db: AgentsRunDatabaseClient) => (params: {
56
+ scopes: AgentScopeConfig;
57
+ triggerId: string;
58
+ invocationId: string;
59
+ data: TriggerInvocationUpdate;
60
+ }) => Promise<TriggerInvocationSelect>;
61
+ //#endregion
62
+ export { createTriggerInvocation, getTriggerInvocationById, listTriggerInvocationsPaginated, updateTriggerInvocationStatus };
@@ -0,0 +1,54 @@
1
+ import { triggerInvocations } from "../../db/runtime/runtime-schema.js";
2
+ import { and, count, desc, eq, gte, lte } from "drizzle-orm";
3
+
4
+ //#region src/data-access/runtime/triggerInvocations.ts
5
+ /**
6
+ * Get a trigger invocation by ID (agent-scoped)
7
+ */
8
+ const getTriggerInvocationById = (db) => async (params) => {
9
+ return await db.query.triggerInvocations.findFirst({ where: and(eq(triggerInvocations.tenantId, params.scopes.tenantId), eq(triggerInvocations.projectId, params.scopes.projectId), eq(triggerInvocations.agentId, params.scopes.agentId), eq(triggerInvocations.triggerId, params.triggerId), eq(triggerInvocations.id, params.invocationId)) });
10
+ };
11
+ /**
12
+ * List trigger invocations with optional filtering (agent-scoped)
13
+ */
14
+ const listTriggerInvocationsPaginated = (db) => async (params) => {
15
+ const page = params.pagination?.page || 1;
16
+ const limit = Math.min(params.pagination?.limit || 10, 100);
17
+ const offset = (page - 1) * limit;
18
+ const conditions = [
19
+ eq(triggerInvocations.tenantId, params.scopes.tenantId),
20
+ eq(triggerInvocations.projectId, params.scopes.projectId),
21
+ eq(triggerInvocations.agentId, params.scopes.agentId),
22
+ eq(triggerInvocations.triggerId, params.triggerId)
23
+ ];
24
+ if (params.filters?.status) conditions.push(eq(triggerInvocations.status, params.filters.status));
25
+ if (params.filters?.from) conditions.push(gte(triggerInvocations.createdAt, params.filters.from));
26
+ if (params.filters?.to) conditions.push(lte(triggerInvocations.createdAt, params.filters.to));
27
+ const whereClause = and(...conditions);
28
+ const [data, totalResult] = await Promise.all([db.select().from(triggerInvocations).where(whereClause).limit(limit).offset(offset).orderBy(desc(triggerInvocations.createdAt)), db.select({ count: count() }).from(triggerInvocations).where(whereClause)]);
29
+ const total = totalResult[0]?.count || 0;
30
+ return {
31
+ data,
32
+ pagination: {
33
+ page,
34
+ limit,
35
+ total,
36
+ pages: Math.ceil(total / limit)
37
+ }
38
+ };
39
+ };
40
+ /**
41
+ * Create a new trigger invocation (agent-scoped)
42
+ */
43
+ const createTriggerInvocation = (db) => async (params) => {
44
+ return (await db.insert(triggerInvocations).values(params).returning())[0];
45
+ };
46
+ /**
47
+ * Update trigger invocation status (agent-scoped)
48
+ */
49
+ const updateTriggerInvocationStatus = (db) => async (params) => {
50
+ return (await db.update(triggerInvocations).set(params.data).where(and(eq(triggerInvocations.tenantId, params.scopes.tenantId), eq(triggerInvocations.projectId, params.scopes.projectId), eq(triggerInvocations.agentId, params.scopes.agentId), eq(triggerInvocations.triggerId, params.triggerId), eq(triggerInvocations.id, params.invocationId))).returning())[0];
51
+ };
52
+
53
+ //#endregion
54
+ export { createTriggerInvocation, getTriggerInvocationById, listTriggerInvocationsPaginated, updateTriggerInvocationStatus };
@@ -0,0 +1,19 @@
1
+ import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
2
+ import { User } from "../../auth/auth-validation-schemas.js";
3
+
4
+ //#region src/data-access/runtime/users.d.ts
5
+
6
+ /**
7
+ * User data access layer
8
+ * All database queries for Better Auth's user table
9
+ */
10
+ /**
11
+ * Get user by ID
12
+ */
13
+ declare const getUserById: (db: AgentsRunDatabaseClient) => (userId: string) => Promise<User | null>;
14
+ /**
15
+ * Get user by email
16
+ */
17
+ declare const getUserByEmail: (db: AgentsRunDatabaseClient) => (email: string) => Promise<User | null>;
18
+ //#endregion
19
+ export { getUserByEmail, getUserById };
@@ -1,7 +1,7 @@
1
- import { user } from "../auth/auth-schema.js";
1
+ import { user } from "../../auth/auth-schema.js";
2
2
  import { eq } from "drizzle-orm";
3
3
 
4
- //#region src/data-access/users.ts
4
+ //#region src/data-access/runtime/users.ts
5
5
  /**
6
6
  * User data access layer
7
7
  * All database queries for Better Auth's user table
@@ -1,4 +1,4 @@
1
- import { DatabaseClient } from "../db/client.js";
1
+ import { AgentsManageDatabaseClient } from "../db/manage/manage-client.js";
2
2
 
3
3
  //#region src/data-access/validation.d.ts
4
4
 
@@ -6,16 +6,16 @@ import { DatabaseClient } from "../db/client.js";
6
6
  * Validates that a project exists before performing database operations
7
7
  * This provides runtime validation even when foreign key constraints are not enforced
8
8
  */
9
- declare const validateProjectExists: (db: DatabaseClient, tenantId: string, projectId: string) => Promise<void>;
9
+ declare const validateProjectExists: (db: AgentsManageDatabaseClient, tenantId: string, projectId: string) => Promise<void>;
10
10
  /**
11
11
  * Wraps a database operation with project validation
12
12
  * Ensures the project exists before executing the operation
13
13
  */
14
- declare const withProjectValidation: <T extends (...args: any[]) => Promise<any>>(db: DatabaseClient, operation: T) => (params: Parameters<T>[0]) => Promise<ReturnType<T>>;
14
+ declare const withProjectValidation: <T extends (...args: any[]) => Promise<any>>(db: AgentsManageDatabaseClient, operation: T) => (params: Parameters<T>[0]) => Promise<ReturnType<T>>;
15
15
  /**
16
16
  * Creates a validated version of data access functions
17
17
  * Automatically adds project validation to insert/update operations
18
18
  */
19
- declare const createValidatedDataAccess: <T extends Record<string, any>>(db: DatabaseClient, dataAccessFunctions: T) => T;
19
+ declare const createValidatedDataAccess: <T extends Record<string, any>>(db: AgentsManageDatabaseClient, dataAccessFunctions: T) => T;
20
20
  //#endregion
21
21
  export { createValidatedDataAccess, validateProjectExists, withProjectValidation };
@@ -1,4 +1,4 @@
1
- import { projectExistsInTable } from "./projects.js";
1
+ import { projectExistsInTable } from "./manage/projects.js";
2
2
 
3
3
  //#region src/data-access/validation.ts
4
4
  /**
@@ -1,8 +1,12 @@
1
+ import { AgentsManageDatabaseClient } from "./manage/manage-client.js";
2
+ import { AgentsRunDatabaseClient } from "./runtime/runtime-client.js";
3
+
1
4
  //#region src/db/clean.d.ts
5
+
2
6
  /**
3
- * Truncates all tables in the database, respecting foreign key constraints
4
- * Tables are cleared in dependency order (child tables first, then parent tables)
7
+ * Cleans up test database by removing all data but keeping schema
8
+ * Dynamically gets all tables from the public schema and truncates them
5
9
  */
6
- declare function cleanDatabase(): Promise<void>;
10
+ declare function cleanupDatabase(db: AgentsManageDatabaseClient | AgentsRunDatabaseClient): Promise<void>;
7
11
  //#endregion
8
- export { cleanDatabase };
12
+ export { cleanupDatabase };
package/dist/db/clean.js CHANGED
@@ -1,115 +1,24 @@
1
- import { agents, artifactComponents, contextCache, contextConfigs, conversations, credentialReferences, dataComponents, externalAgents, ledgerArtifacts, messages, subAgentArtifactComponents, subAgentDataComponents, subAgentRelations, subAgentToolRelations, subAgents, taskRelations, tasks, tools } from "./schema.js";
2
- import { env } from "../env.js";
3
- import { createDatabaseClient } from "./client.js";
4
- import "../index.js";
1
+ import { sql } from "drizzle-orm";
5
2
 
6
3
  //#region src/db/clean.ts
7
- const dbClient = createDatabaseClient();
8
4
  /**
9
- * Truncates all tables in the database, respecting foreign key constraints
10
- * Tables are cleared in dependency order (child tables first, then parent tables)
5
+ * Cleans up test database by removing all data but keeping schema
6
+ * Dynamically gets all tables from the public schema and truncates them
11
7
  */
12
- async function cleanDatabase() {
13
- console.log(`🗑️ Cleaning database for environment: ${env.ENVIRONMENT}`);
8
+ async function cleanupDatabase(db) {
14
9
  try {
15
- const tablesToClear = [
16
- {
17
- table: messages,
18
- name: "messages"
19
- },
20
- {
21
- table: conversations,
22
- name: "conversations"
23
- },
24
- {
25
- table: taskRelations,
26
- name: "task_relations"
27
- },
28
- {
29
- table: tasks,
30
- name: "tasks"
31
- },
32
- {
33
- table: subAgentArtifactComponents,
34
- name: "agent_artifact_components"
35
- },
36
- {
37
- table: subAgentDataComponents,
38
- name: "agent_data_components"
39
- },
40
- {
41
- table: subAgentToolRelations,
42
- name: "agent_tool_relations"
43
- },
44
- {
45
- table: subAgentRelations,
46
- name: "agent_relations"
47
- },
48
- {
49
- table: agents,
50
- name: "agent"
51
- },
52
- {
53
- table: artifactComponents,
54
- name: "artifact_components"
55
- },
56
- {
57
- table: dataComponents,
58
- name: "data_components"
59
- },
60
- {
61
- table: tools,
62
- name: "tools"
63
- },
64
- {
65
- table: subAgents,
66
- name: "agents"
67
- },
68
- {
69
- table: externalAgents,
70
- name: "external_agents"
71
- },
72
- {
73
- table: ledgerArtifacts,
74
- name: "ledger_artifacts"
75
- },
76
- {
77
- table: credentialReferences,
78
- name: "credential_references"
79
- },
80
- {
81
- table: contextConfigs,
82
- name: "context_configs"
83
- },
84
- {
85
- table: contextCache,
86
- name: "context_cache"
87
- }
88
- ];
89
- for (const { table, name } of tablesToClear) try {
90
- await dbClient.delete(table);
91
- console.log(`✅ Cleared table: ${name}`);
92
- } catch (error) {
93
- const errorMessage = error?.message || "";
94
- const causeMessage = error?.cause?.message || "";
95
- const errorCode = error?.code || error?.cause?.code || "";
96
- if (errorMessage.includes("no such table") || causeMessage.includes("no such table") || errorMessage.includes("does not exist") || causeMessage.includes("does not exist") || errorCode === "42P01") console.log(`⚠️ Table ${name} doesn't exist, skipping`);
97
- else throw error;
98
- }
99
- console.log("---");
100
- console.log("🎉 Database cleaned successfully");
10
+ const tables = (await db.execute(sql.raw(`
11
+ SELECT tablename
12
+ FROM pg_tables
13
+ WHERE schemaname = 'public'
14
+ `))).rows.map((row) => row.tablename);
15
+ if (tables.length === 0) return;
16
+ const tableList = tables.map((t) => `"${t}"`).join(", ");
17
+ await db.execute(sql.raw(`TRUNCATE TABLE ${tableList} RESTART IDENTITY CASCADE`));
101
18
  } catch (error) {
102
- console.error(" Failed to clean database:", error);
103
- throw error;
19
+ console.debug("Could not clean database:", error);
104
20
  }
105
21
  }
106
- if (import.meta.url === new URL(import.meta.url).href) cleanDatabase().then(() => {
107
- console.log("Database cleanup completed");
108
- process.exit(0);
109
- }).catch((error) => {
110
- console.error("Database cleanup failed:", error);
111
- process.exit(1);
112
- });
113
22
 
114
23
  //#endregion
115
- export { cleanDatabase };
24
+ export { cleanupDatabase };
@@ -3,6 +3,6 @@
3
3
  * Drops all tables, sequences, types, and functions from the public schema
4
4
  * WARNING: This is destructive and cannot be undone!
5
5
  */
6
- declare function deleteDatabase(): Promise<void>;
6
+ declare function deleteDatabase(type: 'manage' | 'run'): Promise<void>;
7
7
  //#endregion
8
8
  export { deleteDatabase };
package/dist/db/delete.js CHANGED
@@ -1,5 +1,6 @@
1
1
  import { env } from "../env.js";
2
- import { createDatabaseClient } from "./client.js";
2
+ import { createAgentsManageDatabaseClient } from "./manage/manage-client.js";
3
+ import { createAgentsRunDatabaseClient } from "./runtime/runtime-client.js";
3
4
  import { sql } from "drizzle-orm";
4
5
 
5
6
  //#region src/db/delete.ts
@@ -7,11 +8,14 @@ import { sql } from "drizzle-orm";
7
8
  * Drops all tables, sequences, types, and functions from the public schema
8
9
  * WARNING: This is destructive and cannot be undone!
9
10
  */
10
- async function deleteDatabase() {
11
+ async function deleteDatabase(type) {
11
12
  console.log(`🗑️ Deleting all database objects for environment: ${env.ENVIRONMENT}`);
12
13
  console.log("---");
13
- const dbClient = createDatabaseClient();
14
+ const dbClient = type === "manage" ? createAgentsManageDatabaseClient({}) : createAgentsRunDatabaseClient({});
14
15
  try {
16
+ console.log("Dropping workflow schema...");
17
+ await dbClient.execute(sql`DROP SCHEMA IF EXISTS workflow CASCADE`);
18
+ console.log("✅ Workflow schema dropped");
15
19
  console.log("Dropping public schema and all objects...");
16
20
  await dbClient.execute(sql`DROP SCHEMA IF EXISTS public CASCADE`);
17
21
  console.log("✅ Public schema dropped");
@@ -25,13 +29,6 @@ async function deleteDatabase() {
25
29
  throw error;
26
30
  }
27
31
  }
28
- if (import.meta.url === new URL(import.meta.url).href) deleteDatabase().then(() => {
29
- console.log("Database deletion completed");
30
- process.exit(0);
31
- }).catch((error) => {
32
- console.error("Database deletion failed:", error);
33
- process.exit(1);
34
- });
35
32
 
36
33
  //#endregion
37
34
  export { deleteDatabase };
@@ -0,0 +1,51 @@
1
+ import { AgentsManageDatabaseClient } from "./manage-client.js";
2
+
3
+ //#region src/db/manage/dolt-cleanup.d.ts
4
+
5
+ /**
6
+ * Get a database client for cleanup operations
7
+ * Allows passing a custom client or uses a default one
8
+ */
9
+ declare const getIntegrationTestClient: (db?: AgentsManageDatabaseClient) => AgentsManageDatabaseClient;
10
+ /**
11
+ * Delete all branches matching a prefix pattern
12
+ * Used for cleaning up integration test data
13
+ */
14
+ declare const cleanupBranchesByPrefix: (prefix: string, db?: AgentsManageDatabaseClient) => Promise<void>;
15
+ /**
16
+ * Delete all tags matching a prefix pattern
17
+ * Used for cleaning up integration test data
18
+ */
19
+ declare const cleanupTagsByPrefix: (prefix: string, db?: AgentsManageDatabaseClient) => Promise<void>;
20
+ /**
21
+ * Delete specific branches by name
22
+ * Used for cleaning up integration test data
23
+ */
24
+ declare const cleanupBranches: (branchNames: Set<string>, db?: AgentsManageDatabaseClient) => Promise<void>;
25
+ /**
26
+ * Delete specific tags by name
27
+ * Used for cleaning up integration test data
28
+ */
29
+ declare const cleanupTags: (tagNames: Set<string>, db?: AgentsManageDatabaseClient) => Promise<void>;
30
+ /**
31
+ * Comprehensive cleanup for test data by prefix
32
+ * Used for cleaning up integration test data
33
+ */
34
+ declare const cleanupTestData: (prefix: string, branches?: Set<string>, tags?: Set<string>, db?: AgentsManageDatabaseClient) => Promise<void>;
35
+ /**
36
+ * Delete all branches matching a tenant prefix
37
+ * Used for cleaning up integration test data
38
+ */
39
+ declare const cleanupTenantBranches: (tenantId: string, db?: AgentsManageDatabaseClient) => Promise<void>;
40
+ /**
41
+ * Comprehensive cleanup for a tenant: deletes tags and branches
42
+ * Used for cleaning up integration test data
43
+ */
44
+ declare const cleanupTenant: (tenantId: string, tagNames?: Set<string>, db?: AgentsManageDatabaseClient) => Promise<void>;
45
+ /**
46
+ * Cleanup multiple tenants
47
+ * Used for cleaning up integration test data
48
+ */
49
+ declare const cleanupTenants: (tenantIds: Set<string>, tagNames?: Set<string>, db?: AgentsManageDatabaseClient) => Promise<void>;
50
+ //#endregion
51
+ export { cleanupBranches, cleanupBranchesByPrefix, cleanupTags, cleanupTagsByPrefix, cleanupTenant, cleanupTenantBranches, cleanupTenants, cleanupTestData, getIntegrationTestClient };