@rivus/gateway 0.16.2

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 (39) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +6 -0
  3. package/dist/bootstrap/pi-feishu.d.ts +20 -0
  4. package/dist/bootstrap/pi-feishu.js +671 -0
  5. package/dist/chunks/background-session-authority.js +230 -0
  6. package/dist/chunks/background-session-control-input.js +45 -0
  7. package/dist/chunks/background-session-service.d.ts +390 -0
  8. package/dist/chunks/index.d.ts +4703 -0
  9. package/dist/chunks/node-rivus-deployment-manifest.js +1650 -0
  10. package/dist/chunks/rivus-node-entrypoint.js +4464 -0
  11. package/dist/chunks/service.js +12112 -0
  12. package/dist/cli.d.ts +1 -0
  13. package/dist/cli.js +16 -0
  14. package/dist/index.d.ts +3 -0
  15. package/dist/index.js +1215 -0
  16. package/dist/mcp.d.ts +92 -0
  17. package/dist/mcp.js +455 -0
  18. package/package.json +64 -0
  19. package/skills/runtime-management/SKILL.md +65 -0
  20. package/templates/a-share-briefing-analysis.mjs +93 -0
  21. package/templates/a-share-briefing-renderer.mjs +257 -0
  22. package/templates/a-share-index-evidence.mjs +99 -0
  23. package/templates/a-share-market-briefing.mjs +83 -0
  24. package/templates/a-share-market-date.mjs +10 -0
  25. package/templates/a-share-overseas-evidence.mjs +86 -0
  26. package/templates/a-share-policy-evidence.mjs +145 -0
  27. package/templates/a-share-provider-response.mjs +21 -0
  28. package/templates/a-share-sector-evidence.mjs +70 -0
  29. package/templates/acp-stdio-proxy.mjs +58 -0
  30. package/templates/current-weather.mjs +117 -0
  31. package/templates/html-drive-tools.mjs +262 -0
  32. package/templates/https-response-reader.mjs +36 -0
  33. package/templates/langfuse-drive-e2e.mjs +175 -0
  34. package/templates/pi-feishu-deployment.bootstrap.ts +3 -0
  35. package/templates/pi-feishu.bootstrap.ts +242 -0
  36. package/templates/rivus-agents.plugin.mjs +290 -0
  37. package/templates/rivus-langfuse-demo.config.json +37 -0
  38. package/templates/rivus-starter.plugin.mjs +47 -0
  39. package/templates/rivus.config.json +114 -0
@@ -0,0 +1,671 @@
1
+ import { $n as AUTOMATION_SUPPRESSION_PREFIX, B as createBackgroundSessionService, Cn as openPiModelManagementDeployment, G as createFeishuPresentationPreparation, I as openJsonlBackgroundSessionDeliveryStore, J as openJsonlRecoveryControl, K as createFeishuDeploymentEndpoint, M as createBackgroundSessionSupervisor, P as openJsonlBackgroundSessionRepository, Q as createAgentsMdInstructionsProvider, Qn as readAutomationPresentation, Rt as createLazyFeishuWebSocketEventDispatcher, S as resolveFeishuDeliveryChatId, Sn as createPiManagedRunGatewayProvider, Un as createBackgroundSessionHostTools, Y as openJsonlToolOperationLedger$1, Z as createToolBroker$1, _ as openJsonAutomationTickRepository, _n as openJsonlFeishuCardDeliveryLedger, c as createConfiguredFeishuAutomationCardSender, d as createHumanInteractionToolApprovalGateway, dn as openJsonFeishuSessionStore, dt as createConfiguredFeishuMessageReactionSender, et as createFeishuCotPublisher, f as createHumanInteractionEndpointRegistry, ft as createConfiguredFeishuTextReplySender, h as loadRivusDeploymentManifest, it as createJsonFileFeishuCardTargetRegistry, j as resolveBackgroundSessionSupervisorIntervalMs, l as createRoutedHumanInteractionToolApprovalService, lt as createFeishuTopicContextResolver, n as createConfiguredFeishuHumanInteractionPresenter, o as createJsonlHumanInteractionRepository, pn as openJsonlFeishuInboxRepository, pt as createConfiguredFeishuCardRolloverRuntime, q as createAgentHarnessPooledRuntime, sn as createJsonFetchRequest, t as createHumanInteractionService, u as createHumanInteractionModelChangeApproval, un as createSessionScheduler, v as createScheduledAutomation, wt as createConfiguredFeishuOpenApiClient, x as createConfiguredFeishuBackgroundSessionDelivery, y as openJsonlAgentMemoryService, yn as createFeishuCardDeliveryReconciler } from "../chunks/service.js";
2
+ import { f as createBackgroundSessionStepSourceMessageId } from "../chunks/background-session-authority.js";
3
+ import { i as resolveFeishuEndpointCredentials } from "../chunks/node-rivus-deployment-manifest.js";
4
+ import { Effect } from "effect";
5
+ import { createJsonlAgentEventLog, createLangfuseAgentTelemetry, createLegacyAgentHarness, createLegacyProjectMemoryPromptPreparer, createLegacyRivusMemoryToolDescriptor, createSystemClock, createUuidRunIds, createWorkspaceRootHandle, normalizeAgentInvocation, resolveLangfuseTelemetryConfig, toEffectLegacyAgentLoopInput } from "@rivus/runtime";
6
+ import { createHash, randomUUID } from "node:crypto";
7
+ import { mkdir, readFile, stat } from "node:fs/promises";
8
+ import { join, relative } from "node:path";
9
+ import { homedir } from "node:os";
10
+ import { createLegacyPiAgentLoop, createLegacyPiSessionRegistry, createLegacyPiToolNameResolver, createLegacyPiToolProxyDefinitions, createPiAssembly, createPiBashTool, createPiManagedRunBinding, createPiSessionResources, createPiSkillReadTools, createPiSkillRuntime, ensurePiModel, mergePiProviderBaseUrlOverride, resolvePiSessionToolNames, validatePiSkillCommand } from "@rivus/runtime/pi";
11
+ import * as Lark from "@larksuiteoapi/node-sdk";
12
+ //#region src/bootstrap/pi-feishu.ts
13
+ const STATE_DIR = process.env.RIVUS_DEPLOYMENT_STATE_DIR?.trim() || ".rivus/deployment";
14
+ const PI_AGENT_DIR = join(STATE_DIR, "pi-agent");
15
+ const PI_AUTH_FILE = join(STATE_DIR, "pi-auth.json");
16
+ const PI_MODELS_FILE = join(STATE_DIR, "pi-models.json");
17
+ const THINKING_LEVELS = /* @__PURE__ */ new Set([
18
+ "off",
19
+ "minimal",
20
+ "low",
21
+ "medium",
22
+ "high",
23
+ "xhigh"
24
+ ]);
25
+ async function createRivusDeploymentAdapters(context) {
26
+ await mkdir(PI_AGENT_DIR, { recursive: true });
27
+ const hasManagedState = await stat(join(STATE_DIR, "model-management", "model-state.json")).then(() => true, (error) => {
28
+ if (error.code === "ENOENT") return false;
29
+ throw error;
30
+ });
31
+ const piOptions = await createPiSessionOptions(context, context.env.RIVUS_MODEL_MANAGEMENT_ENABLED?.trim() === "true" || hasManagedState);
32
+ let activeModel = piOptions.model;
33
+ const telemetryConfig = resolveLangfuseTelemetryConfig(context.env);
34
+ let telemetry;
35
+ const memory = await openJsonlAgentMemoryService({ filePath: join(STATE_DIR, "memory", "agent-memory.jsonl") });
36
+ const memoryTenantId = context.env.RIVUS_MEMORY_TENANT_ID?.trim() || "local";
37
+ const request = createJsonFetchRequest();
38
+ const createOpenApiClient = (config) => createConfiguredFeishuOpenApiClient({
39
+ config,
40
+ request: (input) => request(input).pipe(Effect.map((response) => response))
41
+ });
42
+ const interactionRegistry = createHumanInteractionEndpointRegistry();
43
+ const manifest = await loadRivusDeploymentManifest(context.manifestPath);
44
+ const defaultEndpoint = manifest.endpoints.find((endpoint) => endpoint.id === manifest.defaultEndpointId);
45
+ const management = await openPiModelManagementDeployment({
46
+ agentId: manifest.defaultAgentId,
47
+ approval: createHumanInteractionModelChangeApproval({
48
+ endpointId: manifest.defaultEndpointId,
49
+ registry: interactionRegistry
50
+ }),
51
+ agentCount: manifest.agents.length,
52
+ endpointId: manifest.defaultEndpointId,
53
+ env: context.env,
54
+ environmentOverrides: context.environmentOverrides ?? process.env,
55
+ ...context.envFilePath ? { envFilePath: context.envFilePath } : {},
56
+ modelAuthPath: PI_AUTH_FILE,
57
+ modelCatalogPath: PI_MODELS_FILE,
58
+ modelRuntime: piOptions.assembly.modelRuntime,
59
+ reply: (messageId, text) => {
60
+ const config = createEndpointConfig(defaultEndpoint, manifest.defaultAgentId, context.env);
61
+ return createConfiguredFeishuTextReplySender({
62
+ config,
63
+ client: createOpenApiClient(config)
64
+ }).reply(messageId, text);
65
+ },
66
+ stateDirectory: STATE_DIR,
67
+ thinkingLevel: piOptions.thinkingLevel ?? "medium"
68
+ });
69
+ try {
70
+ telemetry = telemetryConfig ? createLangfuseAgentTelemetry(telemetryConfig) : void 0;
71
+ if (management && !management.enabled) activeModel = await ensurePiModel({
72
+ modelRuntime: piOptions.assembly.modelRuntime,
73
+ modelsPath: PI_MODELS_FILE,
74
+ target: management.model
75
+ });
76
+ if (management?.enabled && management.recoveryRequired) console.error("Model management requires recovery; business Runs are paused. Query rivus model status --json.");
77
+ if (management?.enabled && management.skillInstallation.status === "conflict") console.error("The runtime-management Skill has local edits; its installed instructions were preserved.");
78
+ const backgroundSessionsConfig = manifest.backgroundSessions;
79
+ const sessionRepository = backgroundSessionsConfig?.enabled ? await openJsonlBackgroundSessionRepository({ filePath: join(STATE_DIR, "background-sessions", "sessions.jsonl") }) : void 0;
80
+ const sessionDeliveries = backgroundSessionsConfig?.enabled ? await openJsonlBackgroundSessionDeliveryStore({ filePath: join(STATE_DIR, "background-sessions", "deliveries.jsonl") }) : void 0;
81
+ const deliveryClients = /* @__PURE__ */ new Map();
82
+ const backgroundService = backgroundSessionsConfig?.enabled ? createBackgroundSessionService({
83
+ clock: { now: () => (/* @__PURE__ */ new Date()).toISOString() },
84
+ deliveries: sessionDeliveries,
85
+ repository: sessionRepository
86
+ }) : void 0;
87
+ const createBackgroundSessionAdapter = (input) => {
88
+ const resolveDeliverySender = async (endpointId) => {
89
+ const existing = deliveryClients.get(endpointId);
90
+ if (existing) return existing;
91
+ const endpoint = manifest.endpoints.find(({ id }) => id === endpointId);
92
+ if (!endpoint) throw new Error(`background session delivery endpoint not found: ${endpointId}`);
93
+ const config = createEndpointConfig(endpoint, endpoint.agentId, context.env);
94
+ const sender = createConfiguredFeishuBackgroundSessionDelivery({
95
+ client: createOpenApiClient(config),
96
+ config
97
+ });
98
+ deliveryClients.set(endpointId, sender);
99
+ return sender;
100
+ };
101
+ const supervisor = createBackgroundSessionSupervisor({
102
+ clock: { now: () => (/* @__PURE__ */ new Date()).toISOString() },
103
+ config: {
104
+ intervalMs: resolveBackgroundSessionSupervisorIntervalMs(input.config.leaseMs),
105
+ leaseMs: input.config.leaseMs,
106
+ leaseRenewalIntervalMs: input.config.leaseRenewalIntervalMs,
107
+ maxConcurrentSessions: input.config.maxConcurrentSessions,
108
+ maxConsecutiveFailures: input.config.maxConsecutiveFailures,
109
+ retryBackoffMs: input.config.retryBackoffMs,
110
+ sessionLifetimeMs: input.config.sessionLifetimeMs
111
+ },
112
+ deliveries: sessionDeliveries,
113
+ deliver: async (delivery) => {
114
+ const session = await sessionRepository.get(delivery.sessionId);
115
+ if (!session) throw new Error(`background session not found for delivery: ${delivery.sessionId}`);
116
+ if (!session.origin.conversationId) throw new Error(`background session has no delivery conversation: ${delivery.sessionId}`);
117
+ return (await resolveDeliverySender(session.origin.endpointId)).deliver({
118
+ chatId: resolveFeishuDeliveryChatId(session.origin.conversationId),
119
+ deliveryId: delivery.deliveryId,
120
+ displayName: session.displayName,
121
+ kind: delivery.kind,
122
+ sessionId: session.sessionId,
123
+ text: delivery.text
124
+ });
125
+ },
126
+ onError: (error) => {
127
+ console.error("Background session supervisor failed", error);
128
+ },
129
+ repository: sessionRepository,
130
+ runStep: async ({ session, signal, wakeText }) => {
131
+ let runId;
132
+ const invocation = {
133
+ allowedActorOpenIds: session.origin.allowedActorOpenIds,
134
+ ...session.origin.conversationId ? { conversationId: session.origin.conversationId } : {},
135
+ endpointId: session.origin.endpointId,
136
+ kind: "background-session",
137
+ ...session.origin.memory ? { memory: session.origin.memory } : {},
138
+ sessionId: session.sessionId,
139
+ sourceMessageId: createBackgroundSessionStepSourceMessageId(session.sessionId, session.stepCount + 1),
140
+ tenantKey: session.origin.tenantKey
141
+ };
142
+ const abortPromise = new Promise((_resolve, reject) => {
143
+ signal.addEventListener("abort", () => {
144
+ if (runId) input.cancel({
145
+ agentId: session.authority.agentId,
146
+ reason: "background session step aborted",
147
+ runId,
148
+ sessionKey: session.authority.sessionKey
149
+ });
150
+ reject(/* @__PURE__ */ new Error("background session step aborted"));
151
+ }, { once: true });
152
+ });
153
+ const runPromise = input.run({
154
+ agentId: session.authority.agentId,
155
+ invocation,
156
+ onUpdate: (update) => {
157
+ if (update.event.type === "agent_run_accepted" && !runId) runId = update.event.runId;
158
+ },
159
+ sessionKey: session.authority.sessionKey,
160
+ text: wakeText
161
+ }).then((result) => readStepRunResult(result));
162
+ return Promise.race([runPromise, abortPromise]);
163
+ },
164
+ sleep
165
+ });
166
+ let running = false;
167
+ return {
168
+ running: () => running,
169
+ status: () => supervisor.status(),
170
+ start: async () => {
171
+ await Effect.runPromise(supervisor.recover());
172
+ await Effect.runPromise(supervisor.start());
173
+ running = true;
174
+ },
175
+ stop: async () => {
176
+ await Effect.runPromise(supervisor.stop());
177
+ running = false;
178
+ }
179
+ };
180
+ };
181
+ return {
182
+ dispose: async () => {
183
+ if (management?.enabled) await management.close();
184
+ await telemetry?.shutdown();
185
+ await piOptions.assembly.dispose();
186
+ },
187
+ createRecoveryControl: () => openJsonlRecoveryControl({
188
+ endpointsDirectory: join(STATE_DIR, "endpoints"),
189
+ instancesDirectory: join(STATE_DIR, "instances")
190
+ }),
191
+ createBackgroundSession: backgroundSessionsConfig?.enabled ? (input) => createBackgroundSessionAdapter(input) : void 0,
192
+ createAutomation: async (input) => {
193
+ const config = createEndpointConfig(input.deliveryEndpoint, input.definition.agentId, context.env);
194
+ const sender = createConfiguredFeishuAutomationCardSender({
195
+ client: createOpenApiClient(config),
196
+ config
197
+ });
198
+ const repository = await openJsonAutomationTickRepository({ filePath: join(STATE_DIR, "automations", createHash("sha256").update(input.automationId).digest("hex").slice(0, 16), "ticks.json") });
199
+ const target = resolveAutomationTarget(input.definition.delivery.targetRef, context.env);
200
+ return createScheduledAutomation({
201
+ automationId: input.automationId,
202
+ binding: {
203
+ agentId: input.definition.agentId,
204
+ bindingId: input.automationId,
205
+ delivery: {
206
+ endpointId: input.definition.delivery.endpointId,
207
+ targetKey: createHash("sha256").update(target).digest("hex"),
208
+ targetType: input.definition.delivery.targetType
209
+ },
210
+ schedule: input.definition.schedule,
211
+ servicePrincipalId: `automation:${input.automationId}`,
212
+ skillGrantRevision: input.definition.runtimeDefinition.skillGrantSet.revision,
213
+ skillIds: input.definition.template.requestedSkillIds,
214
+ templateId: input.definition.templateId,
215
+ timeZone: input.definition.timeZone,
216
+ toolIds: input.definition.template.requestedToolIds
217
+ },
218
+ createInput: input.definition.template.createInput,
219
+ deliver: ({ body, idempotencyKey, presentation }) => Effect.runPromise(sender.send({
220
+ idempotencyKey,
221
+ receiveId: target,
222
+ receiveIdType: input.definition.delivery.targetType,
223
+ markdown: body,
224
+ ...presentation === void 0 ? {} : { presentation }
225
+ })),
226
+ onError: () => {
227
+ console.error(`Scheduled Automation ${input.automationId} failed; it will retry`);
228
+ },
229
+ repository,
230
+ run: async (runInput) => {
231
+ const result = readAutomationRunResult(await input.run(runInput));
232
+ const projected = result.body === void 0 ? void 0 : input.definition.template.createPresentation?.({
233
+ occurrence: runInput.occurrence,
234
+ text: result.body
235
+ });
236
+ const presentation = projected ? readAutomationPresentation(projected) : void 0;
237
+ return {
238
+ ...result,
239
+ ...presentation === void 0 ? {} : { presentation }
240
+ };
241
+ }
242
+ });
243
+ },
244
+ createEndpoint: async (input) => {
245
+ const credentials = resolveFeishuEndpointCredentials(input.definition.credentialRef, context.env);
246
+ const botOpenId = await resolveFeishuBotOpenId(credentials, input.definition.baseUrl);
247
+ const endpointState = join(STATE_DIR, "endpoints", input.endpointId);
248
+ const cardTargets = createJsonFileFeishuCardTargetRegistry({ filePath: join(endpointState, "feishu-card-targets.json") });
249
+ const cardLedger = await openJsonlFeishuCardDeliveryLedger({ filePath: join(endpointState, "feishu-card-delivery.jsonl") });
250
+ const inbox = await openJsonlFeishuInboxRepository({ filePath: join(endpointState, "feishu-inbox.jsonl") });
251
+ const sessionStore = await openJsonFeishuSessionStore({ filePath: join(endpointState, "feishu-session-store.json") });
252
+ const config = createEndpointConfig(input.definition, input.agentId, context.env);
253
+ const openApiClient = createOpenApiClient(config);
254
+ const promptContext = createFeishuTopicContextResolver({
255
+ baseUrl: input.definition.baseUrl,
256
+ client: openApiClient
257
+ });
258
+ const cotPublisher = input.definition.experimental?.cotMessages ? createFeishuCotPublisher({
259
+ baseUrl: resolveExperimentalCotBaseUrl(context.env),
260
+ client: openApiClient,
261
+ minIntervalMs: input.definition.streamMinIntervalMs
262
+ }) : void 0;
263
+ const interactions = createHumanInteractionService({
264
+ clock: { now: () => (/* @__PURE__ */ new Date()).toISOString() },
265
+ presenter: createConfiguredFeishuHumanInteractionPresenter({
266
+ client: openApiClient,
267
+ config
268
+ }),
269
+ repository: createJsonlHumanInteractionRepository({ filePath: join(endpointState, "human-interactions.jsonl") })
270
+ });
271
+ const cardRollover = createConfiguredFeishuCardRolloverRuntime({
272
+ agentName: input.agentId,
273
+ cardTargets,
274
+ client: openApiClient,
275
+ clock: createSystemClock(),
276
+ config,
277
+ ledger: cardLedger,
278
+ onError: (error) => {
279
+ console.error(`Feishu card rollover failed for endpoint ${input.endpointId}`, error);
280
+ },
281
+ ...input.definition.progressDisplay === void 0 ? {} : { progressDisplay: input.definition.progressDisplay },
282
+ sleep,
283
+ title: input.agentId
284
+ });
285
+ const publisher = cardRollover.rollover;
286
+ const endpointEvents = createJsonlAgentEventLog({ filePath: join(STATE_DIR, "instances", input.instanceId, "agent-events.jsonl") });
287
+ const endpointInitialEvents = await Effect.runPromise(endpointEvents.readAll());
288
+ await Effect.runPromise(createFeishuCardDeliveryReconciler({
289
+ events: endpointInitialEvents,
290
+ ledger: cardLedger,
291
+ publish: (action) => publisher.publish(action)
292
+ }).reconcile());
293
+ const replies = createConfiguredFeishuTextReplySender({
294
+ client: openApiClient,
295
+ config
296
+ });
297
+ const reactions = createConfiguredFeishuMessageReactionSender({
298
+ client: openApiClient,
299
+ config
300
+ });
301
+ const prepareCardTarget = (run) => cardRollover.prepareRun(run);
302
+ const endpoint = createFeishuDeploymentEndpoint({
303
+ agentId: input.agentId,
304
+ acknowledgeSteering: (messageId) => reactions.add(messageId),
305
+ botOpenId,
306
+ cardRollover: cardRollover.transport,
307
+ finalizeRun: (runId) => cardRollover.rollover.releaseRun(runId),
308
+ cancel: input.cancel,
309
+ endpointId: input.endpointId,
310
+ eventDispatcher: createLazyFeishuWebSocketEventDispatcher(() => new Lark.EventDispatcher({})),
311
+ groupPolicy: input.definition.groupPolicy,
312
+ handle: input.handle,
313
+ inboxRepository: inbox,
314
+ initialEvents: endpointInitialEvents,
315
+ interactions,
316
+ maxPendingMessages: 100,
317
+ memoryTenantId,
318
+ ...input.projectSpaceId ? { projectSpaceId: input.projectSpaceId } : {},
319
+ onCapacityExceeded: (payload) => replies.reply(payload.event.message.message_id, "Rivus is busy. Please retry in a moment."),
320
+ prepareRun: createFeishuPresentationPreparation({
321
+ ...cotPublisher ? { cotPublisher } : {},
322
+ prepareCardTarget,
323
+ reportCotError: (operation, error) => reportCotError(input.endpointId, operation, error)
324
+ }),
325
+ promptContext,
326
+ publish: (action) => publisher.publish(action),
327
+ reply: (messageId, text) => replies.reply(messageId, text),
328
+ sessionStore,
329
+ ...input.steer ? { steer: input.steer } : {},
330
+ ...cotPublisher ? { publishRunUpdate: (update) => cotPublisher.publish(update).pipe(Effect.catchAll((error) => reportCotError(input.endpointId, "update", error))) } : {},
331
+ sessionNamespace: input.definition.sessionNamespace,
332
+ sleep,
333
+ websocketClient: createLazyFeishuWebSocketClient(credentials, input.definition.baseUrl),
334
+ workerConcurrency: 4
335
+ });
336
+ interactionRegistry.register(input.endpointId, interactions);
337
+ return endpoint;
338
+ },
339
+ createRuntime: async (input) => {
340
+ const instanceState = join(STATE_DIR, "instances", input.instanceId);
341
+ await mkdir(instanceState, { recursive: true });
342
+ const eventLog = createJsonlAgentEventLog({ filePath: join(instanceState, "agent-events.jsonl") });
343
+ const initialEvents = await Effect.runPromise(eventLog.readAll());
344
+ const broker = createToolBroker$1({
345
+ approvals: createRoutedHumanInteractionToolApprovalService(interactionRegistry),
346
+ catalog: input.catalog,
347
+ hostTools: [...input.definition.memory.tool ? [createLegacyRivusMemoryToolDescriptor({ memory })] : [], ...backgroundService ? [...createBackgroundSessionHostTools({
348
+ createSessionId: () => `bg-${randomUUID()}`,
349
+ definition: input.definition,
350
+ service: backgroundService
351
+ })] : []],
352
+ operations: await openJsonlToolOperationLedger$1({ filePath: join(instanceState, "tool-operations.jsonl") }),
353
+ policy: { current: async () => ({
354
+ epoch: 1,
355
+ revokedToolIds: []
356
+ }) }
357
+ });
358
+ const resolveToolName = createLegacyPiToolNameResolver(input.definition.tools);
359
+ const skillRuntime = createPiSkillRuntime(input.definition.skills);
360
+ const workingDirectory = input.projectSpace?.workingDirectory ?? process.cwd();
361
+ const runContexts = management?.enabled ? management.createRunContexts({
362
+ agentId: input.agentId,
363
+ instanceId: input.instanceId,
364
+ toolGrantSet: input.definition.toolGrantSet
365
+ }) : void 0;
366
+ const workspaceRoot = input.projectSpace?.root ?? process.cwd();
367
+ const workspaceInstructions = await createAgentsMdInstructionsProvider().resolve({
368
+ maxBytes: 64 * 1024,
369
+ workingDirectory: relative(workspaceRoot, workingDirectory) || ".",
370
+ workspaceRoot: await createWorkspaceRootHandle(workspaceRoot)
371
+ });
372
+ const prepareProjectMemory = input.projectSpace && input.definition.memory.scopes.includes("project") ? createLegacyProjectMemoryPromptPreparer({
373
+ agentId: input.agentId,
374
+ memory,
375
+ projectId: input.projectSpace.id
376
+ }) : void 0;
377
+ const sessionRegistry = createLegacyPiSessionRegistry({ createSession: async (firstInput) => {
378
+ let activeInput = firstInput;
379
+ const resources = await createPiSessionResources({
380
+ agentDir: PI_AGENT_DIR,
381
+ appendSystemPromptOverride: () => [workspaceInstructions.content, skillRuntime.prompt].filter((content) => content.length > 0),
382
+ cwd: workingDirectory,
383
+ homeDirectory: homedir(),
384
+ ...input.projectSpace ? { projectSkillPaths: input.projectSpace.skillPaths } : {},
385
+ systemPromptOverride: () => input.definition.systemPrompt
386
+ });
387
+ const managedBinding = management?.enabled && runContexts && input.definition.runtimeToolGrantSet.toolIds.includes("bash") ? createPiManagedRunBinding({
388
+ ...resources.bashToolOptions ? { bashToolOptions: resources.bashToolOptions } : {},
389
+ cwd: workingDirectory,
390
+ provider: createPiManagedRunGatewayProvider({
391
+ binDirectory: management.binDirectory,
392
+ contexts: runContexts.registry,
393
+ nodeExecutable: process.execPath,
394
+ socketPath: management.socketPath,
395
+ source: () => toEffectLegacyAgentLoopInput(activeInput)
396
+ })
397
+ }) : void 0;
398
+ const bashTool = input.definition.runtimeToolGrantSet.toolIds.includes("bash") ? managedBinding?.tool ?? createPiBashTool(workingDirectory, resources.bashToolOptions) : void 0;
399
+ const customTools = [
400
+ ...bashTool ? [bashTool] : [],
401
+ ...createPiSkillReadTools({
402
+ cwd: workingDirectory,
403
+ runtimeToolIds: input.definition.runtimeToolGrantSet.toolIds,
404
+ skillPaths: resources.skillNames.size > 0 ? resources.skillPaths : []
405
+ }),
406
+ ...createLegacyPiToolProxyDefinitions({
407
+ agentId: input.agentId,
408
+ approvals: createHumanInteractionToolApprovalGateway({ registry: interactionRegistry }),
409
+ broker,
410
+ getActiveInput: () => activeInput,
411
+ instanceId: input.instanceId,
412
+ memoryScopes: input.definition.memory.scopes,
413
+ toolGrantSet: input.definition.toolGrantSet,
414
+ tools: input.definition.tools
415
+ }),
416
+ ...skillRuntime.tool ? [skillRuntime.tool] : []
417
+ ];
418
+ const activeToolNames = resolvePiSessionToolNames(bashTool ? input.definition.runtimeToolGrantSet.toolIds.filter((toolId) => toolId !== "bash") : input.definition.runtimeToolGrantSet.toolIds, customTools);
419
+ const result = await piOptions.assembly.createSession({
420
+ ...activeModel ? { model: activeModel } : {},
421
+ ...piOptions.thinkingLevel ? { thinkingLevel: piOptions.thinkingLevel } : {},
422
+ ...management?.enabled ? management.runtime.getSessionOptions() : {},
423
+ customTools,
424
+ excludeTools: [],
425
+ resources,
426
+ sessionCwd: workingDirectory,
427
+ sessionDirectory: join(instanceState, "sessions"),
428
+ tools: [...activeToolNames]
429
+ });
430
+ return {
431
+ activate: (loopInput) => {
432
+ activeInput = loopInput;
433
+ managedBinding?.activate({
434
+ runId: loopInput.runId,
435
+ sessionKey: loopInput.sessionKey,
436
+ abortSignal: loopInput.abortSignal,
437
+ ...loopInput.invocation ? { invocation: normalizeAgentInvocation(loopInput.invocation) } : {}
438
+ });
439
+ },
440
+ deactivate: () => managedBinding?.deactivate(),
441
+ dispose: () => result.session.dispose(),
442
+ preparePrompt: async (loopInput) => {
443
+ validatePiSkillCommand(loopInput.text, resources.skillNames);
444
+ return prepareProjectMemory ? prepareProjectMemory(loopInput) : loopInput.text;
445
+ },
446
+ resolveToolName,
447
+ refreshResources: () => resources.refresh(),
448
+ session: result.session
449
+ };
450
+ } });
451
+ const unregisterModels = management?.enabled ? management.runtime.registerSessionRegistry(sessionRegistry) : void 0;
452
+ const loop = createLegacyPiAgentLoop({
453
+ supportsSteering: true,
454
+ disposeSessionAfterRun: false,
455
+ ...management?.enabled ? { runBoundary: management.boundary } : {},
456
+ ...telemetry ? { modelContentObserver: telemetry.modelContentObserver } : {},
457
+ resolveSession: (loopInput) => sessionRegistry.resolve(loopInput)
458
+ });
459
+ const scheduler = createSessionScheduler({
460
+ createRuntime: (sessionKey) => createAgentHarnessPooledRuntime(createLegacyAgentHarness({
461
+ clock: createSystemClock(),
462
+ eventSinks: telemetry ? [eventLog, telemetry.sink] : [eventLog],
463
+ initialEvents: eventsForSession(initialEvents, sessionKey),
464
+ loop,
465
+ runIds: createUuidRunIds(),
466
+ ...input.binding.kind === "background-session" && backgroundSessionsConfig ? { runTimeoutMs: backgroundSessionsConfig.stepTimeoutMs } : {}
467
+ })),
468
+ maxConcurrentSessions: input.binding.kind === "background-session" ? backgroundSessionsConfig?.maxConcurrentSessions ?? 4 : 4,
469
+ maxQueuedRuns: 32
470
+ });
471
+ return {
472
+ ...scheduler,
473
+ dispose: async () => {
474
+ await scheduler.dispose?.();
475
+ await sessionRegistry.disposeAll();
476
+ unregisterModels?.();
477
+ runContexts?.unregister();
478
+ }
479
+ };
480
+ }
481
+ };
482
+ } catch (error) {
483
+ await Promise.allSettled([management?.enabled ? management.close() : Promise.resolve(), telemetry?.shutdown()]);
484
+ throw error;
485
+ }
486
+ }
487
+ function eventsForSession(events, sessionKey) {
488
+ const runIds = new Set(events.filter((event) => event.sessionKey === sessionKey).map((event) => event.runId));
489
+ return events.filter((event) => runIds.has(event.runId));
490
+ }
491
+ function resolveAutomationTarget(targetRef, env) {
492
+ if (!targetRef.startsWith("env:")) throw new Error("Automation delivery targetRef must use env:<VARIABLE>");
493
+ const variable = targetRef.slice(4);
494
+ if (!/^[A-Z][A-Z0-9_]*$/.test(variable)) throw new Error(`Invalid Automation delivery targetRef: ${targetRef}`);
495
+ const target = env[variable]?.trim();
496
+ if (!target) throw new Error(`${variable} is required`);
497
+ return target;
498
+ }
499
+ function resolveExperimentalCotBaseUrl(env) {
500
+ const baseUrl = env.RIVUS_FEISHU_COT_BASE_URL?.trim();
501
+ if (!baseUrl) throw new Error("RIVUS_FEISHU_COT_BASE_URL is required when experimental.cotMessages is enabled");
502
+ return baseUrl;
503
+ }
504
+ function reportCotError(endpointId, operation, error) {
505
+ return Effect.sync(() => {
506
+ const detail = error instanceof Error ? `${error.name}: ${error.message}` : String(error);
507
+ console.error(`Experimental Feishu COT ${operation} failed for endpoint ${endpointId}: ${detail}`);
508
+ });
509
+ }
510
+ function readAutomationRunResult(result) {
511
+ const { finalText, runId } = readStepRunResult(result);
512
+ const trimmedText = finalText.trim();
513
+ if (trimmedText.startsWith("RIVUS_AUTOMATION_SUPPRESSED:")) {
514
+ const reason = trimmedText.slice(AUTOMATION_SUPPRESSION_PREFIX.length).trim();
515
+ if (!reason) throw new Error("Scheduled Automation suppression requires a reason");
516
+ return {
517
+ runId,
518
+ suppressedReason: reason
519
+ };
520
+ }
521
+ if (!trimmedText) throw new Error("Scheduled Automation Agent Run did not produce final text");
522
+ return {
523
+ body: finalText,
524
+ runId
525
+ };
526
+ }
527
+ function readStepRunResult(result) {
528
+ if (result !== null && typeof result === "object" && "finalText" in result && typeof result.finalText === "string" && "runId" in result && typeof result.runId === "string" && result.runId.trim() !== "") return {
529
+ finalText: result.finalText,
530
+ runId: result.runId
531
+ };
532
+ throw new Error("Background session Agent Run did not produce a runId and final text");
533
+ }
534
+ function createLazyFeishuWebSocketClient(credentials, domain) {
535
+ let client;
536
+ let connected = false;
537
+ return {
538
+ connected: () => connected,
539
+ close: () => {
540
+ client?.close();
541
+ client = void 0;
542
+ connected = false;
543
+ },
544
+ start: (options) => new Promise((resolve, reject) => {
545
+ let settled = false;
546
+ const settle = (callback) => {
547
+ if (settled) return;
548
+ settled = true;
549
+ clearTimeout(timeout);
550
+ callback();
551
+ };
552
+ const timeout = setTimeout(() => settle(() => reject(/* @__PURE__ */ new Error("Feishu WebSocket handshake timed out after 15000ms"))), 15e3);
553
+ client ??= new Lark.WSClient({
554
+ ...credentials,
555
+ domain,
556
+ handshakeTimeoutMs: 1e4,
557
+ onError: (error) => {
558
+ connected = false;
559
+ settle(() => reject(error));
560
+ },
561
+ onReady: () => {
562
+ connected = true;
563
+ settle(resolve);
564
+ },
565
+ onReconnected: () => {
566
+ connected = true;
567
+ },
568
+ onReconnecting: () => {
569
+ connected = false;
570
+ }
571
+ });
572
+ client.start(options).catch((error) => {
573
+ connected = false;
574
+ settle(() => reject(error));
575
+ });
576
+ })
577
+ };
578
+ }
579
+ async function resolveFeishuBotOpenId(credentials, domain) {
580
+ const response = await new Lark.Client({
581
+ ...credentials,
582
+ domain
583
+ }).request({
584
+ method: "GET",
585
+ url: "/open-apis/bot/v3/info"
586
+ });
587
+ const openId = response.bot?.open_id;
588
+ if (response.code !== 0 || !openId) throw new Error(`Feishu bot identity lookup failed: ${response.msg ?? "response missing bot.open_id"}`);
589
+ return openId;
590
+ }
591
+ async function createPiSessionOptions(context, deferModel = false) {
592
+ const modelReference = optional(context.env.PI_MODEL);
593
+ const configuredModel = modelReference ? parseModelReference(modelReference) : void 0;
594
+ const provider = configuredModel?.provider;
595
+ const apiKey = await resolvePiApiKey(context.env);
596
+ const baseUrl = optional(context.env.PI_BASE_URL);
597
+ if ((apiKey || baseUrl) && !provider) throw new Error("PI_API_KEY and PI_BASE_URL require PI_MODEL in provider/model form");
598
+ if (baseUrl && !deferModel) await writeProviderBaseUrlOverride(provider, baseUrl);
599
+ const assembly = await createPiAssembly({
600
+ allowModelNetwork: false,
601
+ agentDir: PI_AGENT_DIR,
602
+ authPath: PI_AUTH_FILE,
603
+ cwd: process.cwd(),
604
+ homeDirectory: homedir(),
605
+ modelsPath: baseUrl || deferModel ? PI_MODELS_FILE : null
606
+ });
607
+ const modelRuntime = assembly.modelRuntime;
608
+ if (apiKey) await modelRuntime.setRuntimeApiKey(provider, apiKey);
609
+ const model = configuredModel && !deferModel ? modelRuntime.getModel(configuredModel.provider, configuredModel.modelId) : void 0;
610
+ if (modelReference && !model && !deferModel) throw new Error(`PI_MODEL ${modelReference} was not found in the Pi model runtime`);
611
+ const thinkingLevel = readThinkingLevel(context.env.PI_THINKING_LEVEL) ?? (deferModel && configuredModel ? normalizeThinkingLevel(assembly.resolveThinkingLevel(configuredModel.provider, configuredModel.modelId)) : deferModel ? normalizeThinkingLevel(assembly.resolveThinkingLevel()) : void 0);
612
+ return {
613
+ assembly,
614
+ ...model ? { model } : {},
615
+ ...thinkingLevel ? { thinkingLevel } : {}
616
+ };
617
+ }
618
+ function createEndpointConfig(definition, agentId, env) {
619
+ return {
620
+ agentId,
621
+ feishu: {
622
+ ...resolveFeishuEndpointCredentials(definition.credentialRef, env),
623
+ baseUrl: definition.baseUrl,
624
+ cardStreamLeaseMs: definition.cardStreamLeaseMs,
625
+ streamMinIntervalMs: definition.streamMinIntervalMs
626
+ },
627
+ pi: {}
628
+ };
629
+ }
630
+ async function resolvePiApiKey(env) {
631
+ const inline = optional(env.PI_API_KEY);
632
+ const filePath = optional(env.PI_API_KEY_FILE);
633
+ if (inline && filePath) throw new Error("PI_API_KEY and PI_API_KEY_FILE cannot both be set");
634
+ if (inline) return inline;
635
+ if (!filePath) return void 0;
636
+ const contents = optional(await readFile(filePath, "utf8"));
637
+ if (!contents) throw new Error("PI_API_KEY_FILE must not be empty");
638
+ return contents;
639
+ }
640
+ function optional(value) {
641
+ return value?.trim() || void 0;
642
+ }
643
+ function parseModelReference(reference) {
644
+ const separator = reference.indexOf("/");
645
+ if (separator <= 0 || separator === reference.length - 1) throw new Error("PI_MODEL must use provider/model form, for example zai/glm-5.2");
646
+ return {
647
+ modelId: reference.slice(separator + 1),
648
+ provider: reference.slice(0, separator)
649
+ };
650
+ }
651
+ function readThinkingLevel(value) {
652
+ const level = optional(value);
653
+ if (!level) return void 0;
654
+ if (!THINKING_LEVELS.has(level)) throw new Error("PI_THINKING_LEVEL must be one of off, minimal, low, medium, high, xhigh");
655
+ return level;
656
+ }
657
+ function normalizeThinkingLevel(value) {
658
+ return value !== void 0 && THINKING_LEVELS.has(value) ? value : void 0;
659
+ }
660
+ function sleep(ms) {
661
+ return Effect.promise(() => new Promise((resolve) => setTimeout(resolve, ms)));
662
+ }
663
+ async function writeProviderBaseUrlOverride(provider, baseUrl) {
664
+ await mergePiProviderBaseUrlOverride({
665
+ baseUrl,
666
+ filePath: PI_MODELS_FILE,
667
+ provider
668
+ });
669
+ }
670
+ //#endregion
671
+ export { createRivusDeploymentAdapters };