@inkeep/agents-core 0.47.5 → 0.48.1
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.
- package/README.md +1 -1
- package/dist/auth/auth-schema.d.ts +83 -83
- package/dist/auth/auth-validation-schemas.d.ts +148 -148
- package/dist/auth/auth.d.ts +5 -5
- package/dist/client-exports.d.ts +13 -2
- package/dist/client-exports.js +5 -4
- package/dist/data-access/index.d.ts +6 -1
- package/dist/data-access/index.js +6 -1
- package/dist/data-access/manage/agentFull.js +154 -1
- package/dist/data-access/manage/agents.d.ts +4 -4
- package/dist/data-access/manage/agents.js +56 -4
- package/dist/data-access/manage/artifactComponents.d.ts +41 -5
- package/dist/data-access/manage/functionTools.d.ts +3 -3
- package/dist/data-access/manage/projectFull.js +97 -0
- package/dist/data-access/manage/scheduledTriggers.d.ts +80 -0
- package/dist/data-access/manage/scheduledTriggers.js +76 -0
- package/dist/data-access/manage/scheduledWorkflows.d.ts +29 -0
- package/dist/data-access/manage/scheduledWorkflows.js +32 -0
- package/dist/data-access/manage/skills.d.ts +109 -0
- package/dist/data-access/manage/skills.js +122 -0
- package/dist/data-access/manage/tools.d.ts +4 -4
- package/dist/data-access/manage/triggers.d.ts +2 -2
- package/dist/data-access/runtime/scheduledTriggerInvocations.d.ts +233 -0
- package/dist/data-access/runtime/scheduledTriggerInvocations.js +226 -0
- package/dist/data-access/runtime/workAppSlack.d.ts +55 -0
- package/dist/data-access/runtime/workAppSlack.js +146 -0
- package/dist/db/manage/manage-schema.d.ts +1519 -589
- package/dist/db/manage/manage-schema.js +240 -15
- package/dist/db/runtime/runtime-schema.d.ts +1236 -234
- package/dist/db/runtime/runtime-schema.js +108 -3
- package/dist/index.d.ts +15 -6
- package/dist/index.js +13 -5
- package/dist/types/entities.d.ts +15 -2
- package/dist/types/index.d.ts +2 -2
- package/dist/utils/index.d.ts +5 -2
- package/dist/utils/index.js +5 -2
- package/dist/utils/slack-link-token.d.ts +57 -0
- package/dist/utils/slack-link-token.js +112 -0
- package/dist/utils/slack-user-token.d.ts +65 -0
- package/dist/utils/slack-user-token.js +129 -0
- package/dist/utils/sse-parser.d.ts +35 -0
- package/dist/utils/sse-parser.js +71 -0
- package/dist/utils/tracer-factory.d.ts +11 -2
- package/dist/utils/tracer-factory.js +24 -5
- package/dist/utils/tracer.d.ts +2 -2
- package/dist/utils/tracer.js +2 -2
- package/dist/utils/trigger-auth.d.ts +1 -1
- package/dist/validation/drizzle-schema-helpers.d.ts +3 -3
- package/dist/validation/extend-schemas.d.ts +34 -0
- package/dist/validation/extend-schemas.js +33 -0
- package/dist/validation/index.d.ts +2 -2
- package/dist/validation/index.js +2 -2
- package/dist/validation/json-schemas.d.ts +28 -0
- package/dist/validation/json-schemas.js +56 -0
- package/dist/validation/schemas.d.ts +4930 -1640
- package/dist/validation/schemas.js +159 -38
- package/dist/validation/stream-event-schemas.d.ts +0 -2
- package/dist/validation/stream-event-schemas.js +1 -2
- package/drizzle/manage/0007_nice_lilandra.sql +3 -0
- package/drizzle/manage/0008_friendly_mentallo.sql +32 -0
- package/drizzle/manage/0009_chilly_old_lace.sql +39 -0
- package/drizzle/manage/meta/0007_snapshot.json +3148 -0
- package/drizzle/manage/meta/0008_snapshot.json +3391 -0
- package/drizzle/manage/meta/0009_snapshot.json +3670 -0
- package/drizzle/manage/meta/_journal.json +21 -0
- package/drizzle/runtime/0012_greedy_hulk.sql +84 -0
- package/drizzle/runtime/0013_huge_white_queen.sql +19 -0
- package/drizzle/runtime/meta/0007_snapshot.json +1 -1
- package/drizzle/runtime/meta/0012_snapshot.json +3622 -0
- package/drizzle/runtime/meta/0013_snapshot.json +3746 -0
- package/drizzle/runtime/meta/_journal.json +14 -0
- package/package.json +1 -1
|
@@ -0,0 +1,226 @@
|
|
|
1
|
+
import { scheduledTriggerInvocations } from "../../db/runtime/runtime-schema.js";
|
|
2
|
+
import { and, asc, count, desc, eq, gte, inArray, lte, ne, sql } from "drizzle-orm";
|
|
3
|
+
|
|
4
|
+
//#region src/data-access/runtime/scheduledTriggerInvocations.ts
|
|
5
|
+
/**
|
|
6
|
+
* Get a scheduled trigger invocation by ID (agent-scoped)
|
|
7
|
+
*/
|
|
8
|
+
const getScheduledTriggerInvocationById = (db) => async (params) => {
|
|
9
|
+
return await db.query.scheduledTriggerInvocations.findFirst({ where: and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.id, params.invocationId)) });
|
|
10
|
+
};
|
|
11
|
+
/**
|
|
12
|
+
* Get a scheduled trigger invocation by idempotency key
|
|
13
|
+
* Used to check if an invocation already exists for a given schedule
|
|
14
|
+
*/
|
|
15
|
+
const getScheduledTriggerInvocationByIdempotencyKey = (db) => async (params) => {
|
|
16
|
+
return await db.query.scheduledTriggerInvocations.findFirst({ where: eq(scheduledTriggerInvocations.idempotencyKey, params.idempotencyKey) });
|
|
17
|
+
};
|
|
18
|
+
/**
|
|
19
|
+
* List scheduled trigger invocations with optional filtering (agent-scoped)
|
|
20
|
+
*/
|
|
21
|
+
const listScheduledTriggerInvocationsPaginated = (db) => async (params) => {
|
|
22
|
+
const page = params.pagination?.page || 1;
|
|
23
|
+
const limit = Math.min(params.pagination?.limit || 10, 100);
|
|
24
|
+
const offset = (page - 1) * limit;
|
|
25
|
+
const conditions = [
|
|
26
|
+
eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId),
|
|
27
|
+
eq(scheduledTriggerInvocations.projectId, params.scopes.projectId),
|
|
28
|
+
eq(scheduledTriggerInvocations.agentId, params.scopes.agentId),
|
|
29
|
+
eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId)
|
|
30
|
+
];
|
|
31
|
+
if (params.filters?.status) conditions.push(eq(scheduledTriggerInvocations.status, params.filters.status));
|
|
32
|
+
if (params.filters?.from) conditions.push(gte(scheduledTriggerInvocations.scheduledFor, params.filters.from));
|
|
33
|
+
if (params.filters?.to) conditions.push(lte(scheduledTriggerInvocations.scheduledFor, params.filters.to));
|
|
34
|
+
const whereClause = and(...conditions);
|
|
35
|
+
const [data, totalResult] = await Promise.all([db.select().from(scheduledTriggerInvocations).where(whereClause).limit(limit).offset(offset).orderBy(desc(scheduledTriggerInvocations.scheduledFor)), db.select({ count: count() }).from(scheduledTriggerInvocations).where(whereClause)]);
|
|
36
|
+
const total = totalResult[0]?.count || 0;
|
|
37
|
+
return {
|
|
38
|
+
data,
|
|
39
|
+
pagination: {
|
|
40
|
+
page,
|
|
41
|
+
limit,
|
|
42
|
+
total,
|
|
43
|
+
pages: Math.ceil(total / limit)
|
|
44
|
+
}
|
|
45
|
+
};
|
|
46
|
+
};
|
|
47
|
+
/**
|
|
48
|
+
* List pending invocations for a trigger, ordered by scheduledFor (earliest first)
|
|
49
|
+
* Used by workflow to get the next invocation to execute
|
|
50
|
+
*/
|
|
51
|
+
const listPendingScheduledTriggerInvocations = (db) => async (params) => {
|
|
52
|
+
const maxLimit = Math.min(params.limit || 10, 100);
|
|
53
|
+
return await db.select().from(scheduledTriggerInvocations).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.status, "pending"))).orderBy(asc(scheduledTriggerInvocations.scheduledFor)).limit(maxLimit);
|
|
54
|
+
};
|
|
55
|
+
/**
|
|
56
|
+
* Delete all pending invocations for a trigger
|
|
57
|
+
* Used when cron expression changes or trigger is disabled
|
|
58
|
+
*/
|
|
59
|
+
const deletePendingInvocationsForTrigger = (db) => async (params) => {
|
|
60
|
+
return (await db.delete(scheduledTriggerInvocations).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.status, "pending"))).returning()).length;
|
|
61
|
+
};
|
|
62
|
+
/**
|
|
63
|
+
* Create a new scheduled trigger invocation (agent-scoped)
|
|
64
|
+
*/
|
|
65
|
+
const createScheduledTriggerInvocation = (db) => async (params) => {
|
|
66
|
+
return (await db.insert(scheduledTriggerInvocations).values(params).returning())[0];
|
|
67
|
+
};
|
|
68
|
+
/**
|
|
69
|
+
* Update scheduled trigger invocation status (agent-scoped)
|
|
70
|
+
*/
|
|
71
|
+
const updateScheduledTriggerInvocationStatus = (db) => async (params) => {
|
|
72
|
+
return (await db.update(scheduledTriggerInvocations).set(params.data).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.id, params.invocationId))).returning())[0];
|
|
73
|
+
};
|
|
74
|
+
/**
|
|
75
|
+
* Mark invocation as running
|
|
76
|
+
*/
|
|
77
|
+
const markScheduledTriggerInvocationRunning = (db) => async (params) => {
|
|
78
|
+
const now = (/* @__PURE__ */ new Date()).toISOString();
|
|
79
|
+
return (await db.update(scheduledTriggerInvocations).set({
|
|
80
|
+
status: "running",
|
|
81
|
+
startedAt: sql`COALESCE(${scheduledTriggerInvocations.startedAt}, ${now})`
|
|
82
|
+
}).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.id, params.invocationId))).returning())[0];
|
|
83
|
+
};
|
|
84
|
+
/**
|
|
85
|
+
* Mark invocation as completed
|
|
86
|
+
* Note: Will not update if status is already 'cancelled' to respect user cancellation
|
|
87
|
+
*/
|
|
88
|
+
const markScheduledTriggerInvocationCompleted = (db) => async (params) => {
|
|
89
|
+
return (await db.update(scheduledTriggerInvocations).set({
|
|
90
|
+
status: "completed",
|
|
91
|
+
completedAt: (/* @__PURE__ */ new Date()).toISOString()
|
|
92
|
+
}).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.id, params.invocationId), ne(scheduledTriggerInvocations.status, "cancelled"))).returning())[0];
|
|
93
|
+
};
|
|
94
|
+
/**
|
|
95
|
+
* Mark invocation as failed
|
|
96
|
+
* Note: Will not update if status is already 'cancelled' to respect user cancellation
|
|
97
|
+
*/
|
|
98
|
+
const markScheduledTriggerInvocationFailed = (db) => async (params) => {
|
|
99
|
+
return (await db.update(scheduledTriggerInvocations).set({
|
|
100
|
+
status: "failed",
|
|
101
|
+
completedAt: (/* @__PURE__ */ new Date()).toISOString()
|
|
102
|
+
}).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.id, params.invocationId), ne(scheduledTriggerInvocations.status, "cancelled"))).returning())[0];
|
|
103
|
+
};
|
|
104
|
+
/**
|
|
105
|
+
* Add a conversation ID to the invocation's conversationIds array
|
|
106
|
+
* Used to track all conversations created during retries
|
|
107
|
+
*/
|
|
108
|
+
const addConversationIdToInvocation = (db) => async (params) => {
|
|
109
|
+
return (await db.update(scheduledTriggerInvocations).set({ conversationIds: sql`${scheduledTriggerInvocations.conversationIds} || ${JSON.stringify([params.conversationId])}::jsonb` }).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.id, params.invocationId))).returning())[0];
|
|
110
|
+
};
|
|
111
|
+
/**
|
|
112
|
+
* Mark invocation as cancelled
|
|
113
|
+
*/
|
|
114
|
+
const markScheduledTriggerInvocationCancelled = (db) => async (params) => {
|
|
115
|
+
return (await db.update(scheduledTriggerInvocations).set({
|
|
116
|
+
status: "cancelled",
|
|
117
|
+
completedAt: (/* @__PURE__ */ new Date()).toISOString()
|
|
118
|
+
}).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerInvocations.id, params.invocationId))).returning())[0];
|
|
119
|
+
};
|
|
120
|
+
/**
|
|
121
|
+
* Cancel all pending invocations for a trigger
|
|
122
|
+
* Used when a trigger is deleted
|
|
123
|
+
*/
|
|
124
|
+
const cancelPendingInvocationsForTrigger = (db) => async (params) => {
|
|
125
|
+
return (await db.update(scheduledTriggerInvocations).set({
|
|
126
|
+
status: "cancelled",
|
|
127
|
+
completedAt: (/* @__PURE__ */ new Date()).toISOString()
|
|
128
|
+
}).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), inArray(scheduledTriggerInvocations.status, ["pending", "running"]))).returning()).length;
|
|
129
|
+
};
|
|
130
|
+
/**
|
|
131
|
+
* Cancel only PAST pending invocations for a trigger (scheduledFor <= now)
|
|
132
|
+
* Used when a trigger is disabled - keeps future invocations pending
|
|
133
|
+
*/
|
|
134
|
+
const cancelPastPendingInvocationsForTrigger = (db) => async (params) => {
|
|
135
|
+
const now = (/* @__PURE__ */ new Date()).toISOString();
|
|
136
|
+
return (await db.update(scheduledTriggerInvocations).set({
|
|
137
|
+
status: "cancelled",
|
|
138
|
+
completedAt: now
|
|
139
|
+
}).where(and(eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId), eq(scheduledTriggerInvocations.agentId, params.scopes.agentId), eq(scheduledTriggerInvocations.scheduledTriggerId, params.scheduledTriggerId), inArray(scheduledTriggerInvocations.status, ["pending", "running"]), lte(scheduledTriggerInvocations.scheduledFor, now))).returning()).length;
|
|
140
|
+
};
|
|
141
|
+
/**
|
|
142
|
+
* Get run info for multiple scheduled triggers in a single query
|
|
143
|
+
* Returns last run (completed/failed) and next pending run for each trigger
|
|
144
|
+
*/
|
|
145
|
+
const getScheduledTriggerRunInfoBatch = (db) => async (params) => {
|
|
146
|
+
if (params.triggerIds.length === 0) return /* @__PURE__ */ new Map();
|
|
147
|
+
const { tenantId, projectId } = params.scopes;
|
|
148
|
+
const allInvocations = await db.select({
|
|
149
|
+
scheduledTriggerId: scheduledTriggerInvocations.scheduledTriggerId,
|
|
150
|
+
status: scheduledTriggerInvocations.status,
|
|
151
|
+
scheduledFor: scheduledTriggerInvocations.scheduledFor,
|
|
152
|
+
completedAt: scheduledTriggerInvocations.completedAt,
|
|
153
|
+
conversationIds: scheduledTriggerInvocations.conversationIds
|
|
154
|
+
}).from(scheduledTriggerInvocations).where(and(eq(scheduledTriggerInvocations.tenantId, tenantId), eq(scheduledTriggerInvocations.projectId, projectId), inArray(scheduledTriggerInvocations.scheduledTriggerId, params.triggerIds.map((t) => t.triggerId)))).orderBy(desc(scheduledTriggerInvocations.completedAt));
|
|
155
|
+
const result = /* @__PURE__ */ new Map();
|
|
156
|
+
for (const trigger of params.triggerIds) result.set(trigger.triggerId, {
|
|
157
|
+
lastRunAt: null,
|
|
158
|
+
lastRunStatus: null,
|
|
159
|
+
lastRunConversationIds: [],
|
|
160
|
+
nextRunAt: null
|
|
161
|
+
});
|
|
162
|
+
for (const inv of allInvocations) {
|
|
163
|
+
const triggerInfo = result.get(inv.scheduledTriggerId);
|
|
164
|
+
if (!triggerInfo) continue;
|
|
165
|
+
if (inv.status === "pending" && !triggerInfo.nextRunAt) triggerInfo.nextRunAt = inv.scheduledFor;
|
|
166
|
+
if ((inv.status === "completed" || inv.status === "failed") && !triggerInfo.lastRunAt) {
|
|
167
|
+
triggerInfo.lastRunAt = inv.completedAt;
|
|
168
|
+
triggerInfo.lastRunStatus = inv.status;
|
|
169
|
+
triggerInfo.lastRunConversationIds = inv.conversationIds || [];
|
|
170
|
+
}
|
|
171
|
+
}
|
|
172
|
+
return result;
|
|
173
|
+
};
|
|
174
|
+
/**
|
|
175
|
+
* List upcoming invocations across ALL triggers for an agent with pagination
|
|
176
|
+
* Used for the upcoming runs dashboard with full pagination support
|
|
177
|
+
*/
|
|
178
|
+
const listUpcomingInvocationsForAgentPaginated = (db) => async (params) => {
|
|
179
|
+
const page = params.pagination?.page || 1;
|
|
180
|
+
const limit = Math.min(params.pagination?.limit || 20, 100);
|
|
181
|
+
const offset = (page - 1) * limit;
|
|
182
|
+
const statusCondition = params.includeRunning ? inArray(scheduledTriggerInvocations.status, ["pending", "running"]) : eq(scheduledTriggerInvocations.status, "pending");
|
|
183
|
+
const whereClause = and(...[
|
|
184
|
+
eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId),
|
|
185
|
+
eq(scheduledTriggerInvocations.projectId, params.scopes.projectId),
|
|
186
|
+
eq(scheduledTriggerInvocations.agentId, params.scopes.agentId),
|
|
187
|
+
statusCondition
|
|
188
|
+
]);
|
|
189
|
+
const [data, totalResult] = await Promise.all([db.select().from(scheduledTriggerInvocations).where(whereClause).limit(limit).offset(offset).orderBy(asc(scheduledTriggerInvocations.scheduledFor)), db.select({ count: count() }).from(scheduledTriggerInvocations).where(whereClause)]);
|
|
190
|
+
const total = totalResult[0]?.count || 0;
|
|
191
|
+
return {
|
|
192
|
+
data,
|
|
193
|
+
pagination: {
|
|
194
|
+
page,
|
|
195
|
+
limit,
|
|
196
|
+
total,
|
|
197
|
+
pages: Math.ceil(total / limit)
|
|
198
|
+
}
|
|
199
|
+
};
|
|
200
|
+
};
|
|
201
|
+
/**
|
|
202
|
+
* List all invocations across ALL triggers for a PROJECT with pagination
|
|
203
|
+
* Used for the project-level invocations dashboard
|
|
204
|
+
*/
|
|
205
|
+
const listProjectScheduledTriggerInvocationsPaginated = (db) => async (params) => {
|
|
206
|
+
const page = params.pagination?.page || 1;
|
|
207
|
+
const limit = Math.min(params.pagination?.limit || 20, 100);
|
|
208
|
+
const offset = (page - 1) * limit;
|
|
209
|
+
const conditions = [eq(scheduledTriggerInvocations.tenantId, params.scopes.tenantId), eq(scheduledTriggerInvocations.projectId, params.scopes.projectId)];
|
|
210
|
+
if (params.filters?.status) conditions.push(eq(scheduledTriggerInvocations.status, params.filters.status));
|
|
211
|
+
const whereClause = and(...conditions);
|
|
212
|
+
const [data, totalResult] = await Promise.all([db.select().from(scheduledTriggerInvocations).where(whereClause).limit(limit).offset(offset).orderBy(desc(scheduledTriggerInvocations.createdAt)), db.select({ count: count() }).from(scheduledTriggerInvocations).where(whereClause)]);
|
|
213
|
+
const total = totalResult[0]?.count || 0;
|
|
214
|
+
return {
|
|
215
|
+
data,
|
|
216
|
+
pagination: {
|
|
217
|
+
page,
|
|
218
|
+
limit,
|
|
219
|
+
total,
|
|
220
|
+
pages: Math.ceil(total / limit)
|
|
221
|
+
}
|
|
222
|
+
};
|
|
223
|
+
};
|
|
224
|
+
|
|
225
|
+
//#endregion
|
|
226
|
+
export { addConversationIdToInvocation, cancelPastPendingInvocationsForTrigger, cancelPendingInvocationsForTrigger, createScheduledTriggerInvocation, deletePendingInvocationsForTrigger, getScheduledTriggerInvocationById, getScheduledTriggerInvocationByIdempotencyKey, getScheduledTriggerRunInfoBatch, listPendingScheduledTriggerInvocations, listProjectScheduledTriggerInvocationsPaginated, listScheduledTriggerInvocationsPaginated, listUpcomingInvocationsForAgentPaginated, markScheduledTriggerInvocationCancelled, markScheduledTriggerInvocationCompleted, markScheduledTriggerInvocationFailed, markScheduledTriggerInvocationRunning, updateScheduledTriggerInvocationStatus };
|
|
@@ -0,0 +1,55 @@
|
|
|
1
|
+
import { workAppSlackChannelAgentConfigs, workAppSlackUserMappings, workAppSlackWorkspaces } from "../../db/runtime/runtime-schema.js";
|
|
2
|
+
import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
|
|
3
|
+
|
|
4
|
+
//#region src/data-access/runtime/workAppSlack.d.ts
|
|
5
|
+
type WorkAppSlackWorkspaceInsert = typeof workAppSlackWorkspaces.$inferInsert;
|
|
6
|
+
type WorkAppSlackWorkspaceSelect = typeof workAppSlackWorkspaces.$inferSelect;
|
|
7
|
+
type WorkAppSlackUserMappingInsert = typeof workAppSlackUserMappings.$inferInsert;
|
|
8
|
+
type WorkAppSlackUserMappingSelect = typeof workAppSlackUserMappings.$inferSelect;
|
|
9
|
+
type WorkAppSlackChannelAgentConfigInsert = typeof workAppSlackChannelAgentConfigs.$inferInsert;
|
|
10
|
+
type WorkAppSlackChannelAgentConfigSelect = typeof workAppSlackChannelAgentConfigs.$inferSelect;
|
|
11
|
+
declare const createWorkAppSlackWorkspace: (db: AgentsRunDatabaseClient) => (data: Omit<WorkAppSlackWorkspaceInsert, "id" | "createdAt" | "updatedAt">) => Promise<WorkAppSlackWorkspaceSelect>;
|
|
12
|
+
declare const findWorkAppSlackWorkspaceByTeamId: (db: AgentsRunDatabaseClient) => (tenantId: string, slackTeamId: string) => Promise<WorkAppSlackWorkspaceSelect | null>;
|
|
13
|
+
/**
|
|
14
|
+
* Find a workspace by Slack team ID only (without tenant filter).
|
|
15
|
+
* Slack team IDs are globally unique, so this is safe for resolving
|
|
16
|
+
* the tenant from an incoming Slack event where tenant is unknown.
|
|
17
|
+
*/
|
|
18
|
+
declare const findWorkAppSlackWorkspaceBySlackTeamId: (db: AgentsRunDatabaseClient) => (slackTeamId: string) => Promise<WorkAppSlackWorkspaceSelect | null>;
|
|
19
|
+
/**
|
|
20
|
+
* Find a workspace by its Nango connection ID.
|
|
21
|
+
*
|
|
22
|
+
* One Nango connection = one OAuth token = one Slack workspace.
|
|
23
|
+
* The nangoConnectionId should be globally unique (not per-tenant).
|
|
24
|
+
* The schema has a unique constraint on nangoConnectionId ensuring this.
|
|
25
|
+
*/
|
|
26
|
+
declare const findWorkAppSlackWorkspaceByNangoConnectionId: (db: AgentsRunDatabaseClient) => (nangoConnectionId: string) => Promise<WorkAppSlackWorkspaceSelect | null>;
|
|
27
|
+
declare const listWorkAppSlackWorkspacesByTenant: (db: AgentsRunDatabaseClient) => (tenantId: string) => Promise<WorkAppSlackWorkspaceSelect[]>;
|
|
28
|
+
declare const updateWorkAppSlackWorkspace: (db: AgentsRunDatabaseClient) => (id: string, data: Partial<Pick<WorkAppSlackWorkspaceInsert, "status" | "slackTeamName">>) => Promise<WorkAppSlackWorkspaceSelect | null>;
|
|
29
|
+
declare const deleteWorkAppSlackWorkspace: (db: AgentsRunDatabaseClient) => (id: string) => Promise<boolean>;
|
|
30
|
+
declare const deleteWorkAppSlackWorkspaceByNangoConnectionId: (db: AgentsRunDatabaseClient) => (nangoConnectionId: string) => Promise<boolean>;
|
|
31
|
+
declare const findWorkAppSlackUserMapping: (db: AgentsRunDatabaseClient) => (tenantId: string, slackUserId: string, slackTeamId: string, clientId?: string) => Promise<WorkAppSlackUserMappingSelect | null>;
|
|
32
|
+
declare const findWorkAppSlackUserMappingByInkeepUserId: (db: AgentsRunDatabaseClient) => (inkeepUserId: string) => Promise<WorkAppSlackUserMappingSelect[]>;
|
|
33
|
+
/**
|
|
34
|
+
* Find a user mapping by Slack user ID and team ID only (ignores tenant).
|
|
35
|
+
* Use this when you need to find the user's tenant from their mapping.
|
|
36
|
+
*/
|
|
37
|
+
declare const findWorkAppSlackUserMappingBySlackUser: (db: AgentsRunDatabaseClient) => (slackUserId: string, slackTeamId: string, clientId?: string) => Promise<WorkAppSlackUserMappingSelect | null>;
|
|
38
|
+
declare const listWorkAppSlackUserMappingsByTeam: (db: AgentsRunDatabaseClient) => (tenantId: string, slackTeamId: string) => Promise<WorkAppSlackUserMappingSelect[]>;
|
|
39
|
+
declare const createWorkAppSlackUserMapping: (db: AgentsRunDatabaseClient) => (data: Omit<WorkAppSlackUserMappingInsert, "id" | "clientId" | "createdAt" | "updatedAt"> & {
|
|
40
|
+
clientId?: string;
|
|
41
|
+
}) => Promise<WorkAppSlackUserMappingSelect>;
|
|
42
|
+
declare const deleteWorkAppSlackUserMapping: (db: AgentsRunDatabaseClient) => (tenantId: string, slackUserId: string, slackTeamId: string, clientId?: string) => Promise<boolean>;
|
|
43
|
+
declare const deleteAllWorkAppSlackUserMappingsByTeam: (db: AgentsRunDatabaseClient) => (tenantId: string, slackTeamId: string, clientId?: string) => Promise<number>;
|
|
44
|
+
declare const createWorkAppSlackChannelAgentConfig: (db: AgentsRunDatabaseClient) => (data: Omit<WorkAppSlackChannelAgentConfigInsert, "id" | "createdAt" | "updatedAt">) => Promise<WorkAppSlackChannelAgentConfigSelect>;
|
|
45
|
+
declare const findWorkAppSlackChannelAgentConfig: (db: AgentsRunDatabaseClient) => (tenantId: string, slackTeamId: string, slackChannelId: string) => Promise<WorkAppSlackChannelAgentConfigSelect | null>;
|
|
46
|
+
declare const listWorkAppSlackChannelAgentConfigsByTeam: (db: AgentsRunDatabaseClient) => (tenantId: string, slackTeamId: string) => Promise<WorkAppSlackChannelAgentConfigSelect[]>;
|
|
47
|
+
/**
|
|
48
|
+
* Atomic upsert using onConflictDoUpdate to avoid TOCTOU race conditions.
|
|
49
|
+
* Uses the unique constraint on (tenantId, slackTeamId, slackChannelId).
|
|
50
|
+
*/
|
|
51
|
+
declare const upsertWorkAppSlackChannelAgentConfig: (db: AgentsRunDatabaseClient) => (data: Omit<WorkAppSlackChannelAgentConfigInsert, "id" | "createdAt" | "updatedAt">) => Promise<WorkAppSlackChannelAgentConfigSelect>;
|
|
52
|
+
declare const deleteWorkAppSlackChannelAgentConfig: (db: AgentsRunDatabaseClient) => (tenantId: string, slackTeamId: string, slackChannelId: string) => Promise<boolean>;
|
|
53
|
+
declare const deleteAllWorkAppSlackChannelAgentConfigsByTeam: (db: AgentsRunDatabaseClient) => (tenantId: string, slackTeamId: string) => Promise<number>;
|
|
54
|
+
//#endregion
|
|
55
|
+
export { WorkAppSlackChannelAgentConfigInsert, WorkAppSlackChannelAgentConfigSelect, WorkAppSlackUserMappingInsert, WorkAppSlackUserMappingSelect, WorkAppSlackWorkspaceInsert, WorkAppSlackWorkspaceSelect, createWorkAppSlackChannelAgentConfig, createWorkAppSlackUserMapping, createWorkAppSlackWorkspace, deleteAllWorkAppSlackChannelAgentConfigsByTeam, deleteAllWorkAppSlackUserMappingsByTeam, deleteWorkAppSlackChannelAgentConfig, deleteWorkAppSlackUserMapping, deleteWorkAppSlackWorkspace, deleteWorkAppSlackWorkspaceByNangoConnectionId, findWorkAppSlackChannelAgentConfig, findWorkAppSlackUserMapping, findWorkAppSlackUserMappingByInkeepUserId, findWorkAppSlackUserMappingBySlackUser, findWorkAppSlackWorkspaceByNangoConnectionId, findWorkAppSlackWorkspaceBySlackTeamId, findWorkAppSlackWorkspaceByTeamId, listWorkAppSlackChannelAgentConfigsByTeam, listWorkAppSlackUserMappingsByTeam, listWorkAppSlackWorkspacesByTenant, updateWorkAppSlackWorkspace, upsertWorkAppSlackChannelAgentConfig };
|
|
@@ -0,0 +1,146 @@
|
|
|
1
|
+
import { workAppSlackChannelAgentConfigs, workAppSlackUserMappings, workAppSlackWorkspaces } from "../../db/runtime/runtime-schema.js";
|
|
2
|
+
import { and, eq } from "drizzle-orm";
|
|
3
|
+
import { nanoid } from "nanoid";
|
|
4
|
+
|
|
5
|
+
//#region src/data-access/runtime/workAppSlack.ts
|
|
6
|
+
const DEFAULT_CLIENT_ID = "work-apps-slack";
|
|
7
|
+
const createWorkAppSlackWorkspace = (db) => async (data) => {
|
|
8
|
+
const id = `wsw_${nanoid(21)}`;
|
|
9
|
+
const now = (/* @__PURE__ */ new Date()).toISOString();
|
|
10
|
+
const [result] = await db.insert(workAppSlackWorkspaces).values({
|
|
11
|
+
id,
|
|
12
|
+
...data,
|
|
13
|
+
createdAt: now,
|
|
14
|
+
updatedAt: now
|
|
15
|
+
}).returning();
|
|
16
|
+
return result;
|
|
17
|
+
};
|
|
18
|
+
const findWorkAppSlackWorkspaceByTeamId = (db) => async (tenantId, slackTeamId) => {
|
|
19
|
+
return (await db.select().from(workAppSlackWorkspaces).where(and(eq(workAppSlackWorkspaces.tenantId, tenantId), eq(workAppSlackWorkspaces.slackTeamId, slackTeamId))).limit(1))[0] || null;
|
|
20
|
+
};
|
|
21
|
+
/**
|
|
22
|
+
* Find a workspace by Slack team ID only (without tenant filter).
|
|
23
|
+
* Slack team IDs are globally unique, so this is safe for resolving
|
|
24
|
+
* the tenant from an incoming Slack event where tenant is unknown.
|
|
25
|
+
*/
|
|
26
|
+
const findWorkAppSlackWorkspaceBySlackTeamId = (db) => async (slackTeamId) => {
|
|
27
|
+
return (await db.select().from(workAppSlackWorkspaces).where(eq(workAppSlackWorkspaces.slackTeamId, slackTeamId)).limit(1))[0] || null;
|
|
28
|
+
};
|
|
29
|
+
/**
|
|
30
|
+
* Find a workspace by its Nango connection ID.
|
|
31
|
+
*
|
|
32
|
+
* One Nango connection = one OAuth token = one Slack workspace.
|
|
33
|
+
* The nangoConnectionId should be globally unique (not per-tenant).
|
|
34
|
+
* The schema has a unique constraint on nangoConnectionId ensuring this.
|
|
35
|
+
*/
|
|
36
|
+
const findWorkAppSlackWorkspaceByNangoConnectionId = (db) => async (nangoConnectionId) => {
|
|
37
|
+
return (await db.select().from(workAppSlackWorkspaces).where(eq(workAppSlackWorkspaces.nangoConnectionId, nangoConnectionId)))[0] || null;
|
|
38
|
+
};
|
|
39
|
+
const listWorkAppSlackWorkspacesByTenant = (db) => async (tenantId) => {
|
|
40
|
+
return db.select().from(workAppSlackWorkspaces).where(eq(workAppSlackWorkspaces.tenantId, tenantId));
|
|
41
|
+
};
|
|
42
|
+
const updateWorkAppSlackWorkspace = (db) => async (id, data) => {
|
|
43
|
+
const [result] = await db.update(workAppSlackWorkspaces).set({
|
|
44
|
+
...data,
|
|
45
|
+
updatedAt: (/* @__PURE__ */ new Date()).toISOString()
|
|
46
|
+
}).where(eq(workAppSlackWorkspaces.id, id)).returning();
|
|
47
|
+
return result || null;
|
|
48
|
+
};
|
|
49
|
+
const deleteWorkAppSlackWorkspace = (db) => async (id) => {
|
|
50
|
+
return (await db.delete(workAppSlackWorkspaces).where(eq(workAppSlackWorkspaces.id, id)).returning()).length > 0;
|
|
51
|
+
};
|
|
52
|
+
const deleteWorkAppSlackWorkspaceByNangoConnectionId = (db) => async (nangoConnectionId) => {
|
|
53
|
+
return (await db.delete(workAppSlackWorkspaces).where(eq(workAppSlackWorkspaces.nangoConnectionId, nangoConnectionId)).returning()).length > 0;
|
|
54
|
+
};
|
|
55
|
+
const findWorkAppSlackUserMapping = (db) => async (tenantId, slackUserId, slackTeamId, clientId = DEFAULT_CLIENT_ID) => {
|
|
56
|
+
return (await db.select().from(workAppSlackUserMappings).where(and(eq(workAppSlackUserMappings.tenantId, tenantId), eq(workAppSlackUserMappings.clientId, clientId), eq(workAppSlackUserMappings.slackUserId, slackUserId), eq(workAppSlackUserMappings.slackTeamId, slackTeamId))).limit(1))[0] || null;
|
|
57
|
+
};
|
|
58
|
+
const findWorkAppSlackUserMappingByInkeepUserId = (db) => async (inkeepUserId) => {
|
|
59
|
+
return db.select().from(workAppSlackUserMappings).where(eq(workAppSlackUserMappings.inkeepUserId, inkeepUserId));
|
|
60
|
+
};
|
|
61
|
+
/**
|
|
62
|
+
* Find a user mapping by Slack user ID and team ID only (ignores tenant).
|
|
63
|
+
* Use this when you need to find the user's tenant from their mapping.
|
|
64
|
+
*/
|
|
65
|
+
const findWorkAppSlackUserMappingBySlackUser = (db) => async (slackUserId, slackTeamId, clientId = DEFAULT_CLIENT_ID) => {
|
|
66
|
+
return (await db.select().from(workAppSlackUserMappings).where(and(eq(workAppSlackUserMappings.clientId, clientId), eq(workAppSlackUserMappings.slackUserId, slackUserId), eq(workAppSlackUserMappings.slackTeamId, slackTeamId))).limit(1))[0] || null;
|
|
67
|
+
};
|
|
68
|
+
const listWorkAppSlackUserMappingsByTeam = (db) => async (tenantId, slackTeamId) => {
|
|
69
|
+
return db.select().from(workAppSlackUserMappings).where(and(eq(workAppSlackUserMappings.tenantId, tenantId), eq(workAppSlackUserMappings.slackTeamId, slackTeamId)));
|
|
70
|
+
};
|
|
71
|
+
const createWorkAppSlackUserMapping = (db) => async (data) => {
|
|
72
|
+
const id = `wsum_${nanoid(21)}`;
|
|
73
|
+
const now = (/* @__PURE__ */ new Date()).toISOString();
|
|
74
|
+
const [result] = await db.insert(workAppSlackUserMappings).values({
|
|
75
|
+
id,
|
|
76
|
+
clientId: data.clientId || DEFAULT_CLIENT_ID,
|
|
77
|
+
...data,
|
|
78
|
+
linkedAt: now,
|
|
79
|
+
createdAt: now,
|
|
80
|
+
updatedAt: now
|
|
81
|
+
}).returning();
|
|
82
|
+
return result;
|
|
83
|
+
};
|
|
84
|
+
const deleteWorkAppSlackUserMapping = (db) => async (tenantId, slackUserId, slackTeamId, clientId = DEFAULT_CLIENT_ID) => {
|
|
85
|
+
return (await db.delete(workAppSlackUserMappings).where(and(eq(workAppSlackUserMappings.tenantId, tenantId), eq(workAppSlackUserMappings.clientId, clientId), eq(workAppSlackUserMappings.slackUserId, slackUserId), eq(workAppSlackUserMappings.slackTeamId, slackTeamId))).returning()).length > 0;
|
|
86
|
+
};
|
|
87
|
+
const deleteAllWorkAppSlackUserMappingsByTeam = (db) => async (tenantId, slackTeamId, clientId = DEFAULT_CLIENT_ID) => {
|
|
88
|
+
return (await db.delete(workAppSlackUserMappings).where(and(eq(workAppSlackUserMappings.tenantId, tenantId), eq(workAppSlackUserMappings.clientId, clientId), eq(workAppSlackUserMappings.slackTeamId, slackTeamId))).returning()).length;
|
|
89
|
+
};
|
|
90
|
+
const createWorkAppSlackChannelAgentConfig = (db) => async (data) => {
|
|
91
|
+
const id = `wscac_${nanoid(21)}`;
|
|
92
|
+
const now = (/* @__PURE__ */ new Date()).toISOString();
|
|
93
|
+
const [result] = await db.insert(workAppSlackChannelAgentConfigs).values({
|
|
94
|
+
id,
|
|
95
|
+
...data,
|
|
96
|
+
createdAt: now,
|
|
97
|
+
updatedAt: now
|
|
98
|
+
}).returning();
|
|
99
|
+
return result;
|
|
100
|
+
};
|
|
101
|
+
const findWorkAppSlackChannelAgentConfig = (db) => async (tenantId, slackTeamId, slackChannelId) => {
|
|
102
|
+
return (await db.select().from(workAppSlackChannelAgentConfigs).where(and(eq(workAppSlackChannelAgentConfigs.tenantId, tenantId), eq(workAppSlackChannelAgentConfigs.slackTeamId, slackTeamId), eq(workAppSlackChannelAgentConfigs.slackChannelId, slackChannelId))).limit(1))[0] || null;
|
|
103
|
+
};
|
|
104
|
+
const listWorkAppSlackChannelAgentConfigsByTeam = (db) => async (tenantId, slackTeamId) => {
|
|
105
|
+
return db.select().from(workAppSlackChannelAgentConfigs).where(and(eq(workAppSlackChannelAgentConfigs.tenantId, tenantId), eq(workAppSlackChannelAgentConfigs.slackTeamId, slackTeamId)));
|
|
106
|
+
};
|
|
107
|
+
/**
|
|
108
|
+
* Atomic upsert using onConflictDoUpdate to avoid TOCTOU race conditions.
|
|
109
|
+
* Uses the unique constraint on (tenantId, slackTeamId, slackChannelId).
|
|
110
|
+
*/
|
|
111
|
+
const upsertWorkAppSlackChannelAgentConfig = (db) => async (data) => {
|
|
112
|
+
const id = `wscac_${nanoid(21)}`;
|
|
113
|
+
const now = (/* @__PURE__ */ new Date()).toISOString();
|
|
114
|
+
const [result] = await db.insert(workAppSlackChannelAgentConfigs).values({
|
|
115
|
+
id,
|
|
116
|
+
...data,
|
|
117
|
+
createdAt: now,
|
|
118
|
+
updatedAt: now
|
|
119
|
+
}).onConflictDoUpdate({
|
|
120
|
+
target: [
|
|
121
|
+
workAppSlackChannelAgentConfigs.tenantId,
|
|
122
|
+
workAppSlackChannelAgentConfigs.slackTeamId,
|
|
123
|
+
workAppSlackChannelAgentConfigs.slackChannelId
|
|
124
|
+
],
|
|
125
|
+
set: {
|
|
126
|
+
projectId: data.projectId,
|
|
127
|
+
agentId: data.agentId,
|
|
128
|
+
agentName: data.agentName,
|
|
129
|
+
slackChannelName: data.slackChannelName,
|
|
130
|
+
slackChannelType: data.slackChannelType,
|
|
131
|
+
enabled: data.enabled,
|
|
132
|
+
configuredByUserId: data.configuredByUserId,
|
|
133
|
+
updatedAt: now
|
|
134
|
+
}
|
|
135
|
+
}).returning();
|
|
136
|
+
return result;
|
|
137
|
+
};
|
|
138
|
+
const deleteWorkAppSlackChannelAgentConfig = (db) => async (tenantId, slackTeamId, slackChannelId) => {
|
|
139
|
+
return (await db.delete(workAppSlackChannelAgentConfigs).where(and(eq(workAppSlackChannelAgentConfigs.tenantId, tenantId), eq(workAppSlackChannelAgentConfigs.slackTeamId, slackTeamId), eq(workAppSlackChannelAgentConfigs.slackChannelId, slackChannelId))).returning()).length > 0;
|
|
140
|
+
};
|
|
141
|
+
const deleteAllWorkAppSlackChannelAgentConfigsByTeam = (db) => async (tenantId, slackTeamId) => {
|
|
142
|
+
return (await db.delete(workAppSlackChannelAgentConfigs).where(and(eq(workAppSlackChannelAgentConfigs.tenantId, tenantId), eq(workAppSlackChannelAgentConfigs.slackTeamId, slackTeamId))).returning()).length;
|
|
143
|
+
};
|
|
144
|
+
|
|
145
|
+
//#endregion
|
|
146
|
+
export { createWorkAppSlackChannelAgentConfig, createWorkAppSlackUserMapping, createWorkAppSlackWorkspace, deleteAllWorkAppSlackChannelAgentConfigsByTeam, deleteAllWorkAppSlackUserMappingsByTeam, deleteWorkAppSlackChannelAgentConfig, deleteWorkAppSlackUserMapping, deleteWorkAppSlackWorkspace, deleteWorkAppSlackWorkspaceByNangoConnectionId, findWorkAppSlackChannelAgentConfig, findWorkAppSlackUserMapping, findWorkAppSlackUserMappingByInkeepUserId, findWorkAppSlackUserMappingBySlackUser, findWorkAppSlackWorkspaceByNangoConnectionId, findWorkAppSlackWorkspaceBySlackTeamId, findWorkAppSlackWorkspaceByTeamId, listWorkAppSlackChannelAgentConfigsByTeam, listWorkAppSlackUserMappingsByTeam, listWorkAppSlackWorkspacesByTenant, updateWorkAppSlackWorkspace, upsertWorkAppSlackChannelAgentConfig };
|