@ykurilov/agent-chat-flows 0.63.2 → 0.64.0

package/dist/index.js

@@ -14,7 +14,7 @@ import {
   sideTaskFieldKnown,
   sideTaskRecord,
   sideTaskStanding
-} from "./chunk-H3AN5YUN.js";
+} from "./chunk-3K72EVAT.js";
 import {
   FACT_PREFIX,
   MAX_FACTS,
@@ -26,7 +26,7 @@ import {
   fieldValueText,
   isFactEntry,
   sameFact
-} from "./chunk-KGWXTWBZ.js";
+} from "./chunk-22FWN7NU.js";
 import {
   AGENT_FLOW_FORMAT,
   AGENT_FLOW_FORMAT_VERSION,
@@ -92,6 +92,7 @@ import {
   noteSideQuestion,
   overdueOnEntry,
   parseBooleanAnswer,
+  parseDuration,
   parseFlow,
   placeOfPath,
   returnTarget,
@@ -105,7 +106,7 @@ import {
   withDayAnchor,
   withNudgeStep,
   withoutUndeliveredTouch
-} from "./chunk-LT4QELAZ.js";
+} from "./chunk-5RDY7QBA.js";
 import {
   appliedInTime,
   byVersionAndState,
@@ -1020,6 +1021,15 @@ function testChatOffset(data) {
   return typeof offset === "number" && Number.isFinite(offset) ? offset : 0;
 }
 var chatTime = (run, serverNow) => serverNow + (run?.timeOffset ?? 0);
+var FlowTestClockError = class extends Error {
+  constructor(code, message) {
+    super(message);
+    this.code = code;
+    this.name = "FlowTestClockError";
+  }
+  code;
+};
+var NEXT_FIRST_MESSAGE_STEPS = 50;
 var TURN_WAIT_LIMIT = 5;
 async function placeRun(runtime, port, chat, flow, now) {
   const { values } = await port.getChatState(chat.ownerId, chat.chatId);
@@ -1049,7 +1059,8 @@ async function advanceTestClock(input) {
   const { runtime, port, timers, ownerId, chatId, clock } = input;
   const offset = testChatOffset(await port.getChatData(ownerId, chatId));
   if (offset === null) throw new Error(`\u0427\u0430\u0442 ${chatId} \u043D\u0435 \u0442\u0435\u0441\u0442\u043E\u0432\u044B\u0439: \u0447\u0430\u0441\u044B \u043F\u0435\u0440\u0435\u043C\u0430\u0442\u044B\u0432\u0430\u044E\u0442\u0441\u044F \u0442\u043E\u043B\u044C\u043A\u043E \u0443 \u0442\u0435\u0441\u0442\u043E\u0432\u044B\u0445 \u0447\u0430\u0442\u043E\u0432`);
-  const ms = clock.to === "next_day_start" ? await msToNextDay(runtime, port, ownerId, chatId, offset) : clock.ms;
+  if (clock.to === "next_first_message") return advanceToNextFirstMessage({ runtime, port, timers, ownerId, chatId, offset });
+  const ms = clock.to === "next_day_start" ? await msToNextDay(runtime, port, ownerId, chatId, offset) : clock.ms ?? Number.NaN;
   if (!Number.isFinite(ms) || ms <= 0) throw new Error("\u041F\u0435\u0440\u0435\u043C\u043E\u0442\u043A\u0430 \u0447\u0430\u0441\u043E\u0432: \u043D\u0443\u0436\u0435\u043D \u0441\u0434\u0432\u0438\u0433 \u0432\u043F\u0435\u0440\u0451\u0434, \u043C\u0441 \u0431\u043E\u043B\u044C\u0448\u0435 \u043D\u0443\u043B\u044F");
   const timeOffset = offset + ms;
   await port.updateChatData(ownerId, chatId, { [FLOW_TEST_CHAT_KEY]: { timeOffset } });
@@ -1062,6 +1073,37 @@ async function advanceTestClock(input) {
   await turnFinished(port, ownerId, chatId);
   return { timeOffset };
 }
+var nothingScheduled = (chatId) => new FlowTestClockError("nothing_scheduled", `\u0427\u0430\u0442 ${chatId}: \u0432\u043F\u0435\u0440\u0435\u0434\u0438 \u0441\u0446\u0435\u043D\u0430\u0440\u0438\u0439 \u0441\u0430\u043C \u043D\u0438\u0447\u0435\u0433\u043E \u043D\u0435 \u043D\u0430\u043F\u0438\u0448\u0435\u0442 \u2014 \u043F\u0435\u0440\u0435\u043C\u0430\u0442\u044B\u0432\u0430\u0442\u044C \u043D\u0435 \u043A \u0447\u0435\u043C\u0443`);
+async function advanceToNextFirstMessage(input) {
+  const { runtime, port, timers, ownerId, chatId } = input;
+  const pinned = timers && await runtime.chatFlow(ownerId, chatId);
+  if (!timers?.watch || !pinned) throw nothingScheduled(chatId);
+  let timeOffset = input.offset;
+  for (let step = 0; step < NEXT_FIRST_MESSAGE_STEPS; step += 1) {
+    const run = await placeRun(runtime, port, { ownerId, chatId }, pinned.flow, runtime.now() + timeOffset);
+    if (!run) throw nothingScheduled(chatId);
+    if (step === 0) await timers.reschedule({ ownerId, chatId, run: { ...run, timeOffset }, branchHead: run.answerId ?? null });
+    const record = await runtime.store.getChat(chatId);
+    if (!record || record.endedAt !== null || record.dueAt === null) throw nothingScheduled(chatId);
+    const serverNow = runtime.now();
+    const shift = Math.max(0, Date.parse(record.dueAt) - serverNow);
+    if (shift > 0) {
+      timeOffset += shift;
+      await port.updateChatData(ownerId, chatId, { [FLOW_TEST_CHAT_KEY]: { timeOffset } });
+      await runtime.store.setDue(chatId, { at: new Date(serverNow).toISOString(), exitId: record.dueExitId ?? FLOW_FIRST_EXIT, branchHead: record.dueBranchHead });
+    }
+    let reason = null;
+    const unwatch = timers.watch(chatId, (fired) => reason ??= fired);
+    try {
+      await timers.tick();
+      await turnFinished(port, ownerId, chatId);
+    } finally {
+      unwatch();
+    }
+    if (reason) return { timeOffset, reason };
+  }
+  throw new FlowTestClockError("not_reached", `\u0427\u0430\u0442 ${chatId}: \u0441\u0440\u043E\u043A \u0441\u043E\u043E\u0431\u0449\u0435\u043D\u0438\u044F \u043E\u0442\u043A\u043B\u0430\u0434\u044B\u0432\u0430\u043B\u0441\u044F ${NEXT_FIRST_MESSAGE_STEPS} \u0440\u0430\u0437 \u043F\u043E\u0434\u0440\u044F\u0434 \u2014 \u043F\u0435\u0440\u0435\u043C\u043E\u0442\u043A\u0430 \u043E\u0441\u0442\u0430\u043D\u043E\u0432\u043B\u0435\u043D\u0430`);
+}
 
 // src/runtime/flow-finish.ts
 var FLOW_DEADLINE_NOTICE = "flow_deadline";
@@ -1501,6 +1543,10 @@ function createFlowTimers(input) {
   const owner = `flow-timers:${randomUUID()}`;
   let interval;
   let running = null;
+  const watchers = /* @__PURE__ */ new Map();
+  const fired = (chatId, reason) => {
+    for (const onFired of watchers.get(chatId) ?? []) onFired(reason);
+  };
   const dueInput = (due, branchHead, offset) => ({ at: isoOf2(due.at - offset), exitId: due.exitId, branchHead });
   async function placeOf2(record, chatPort, run, failedAt) {
     const pausedUntil = record.pausedUntil === null ? null : Date.parse(record.pausedUntil);
@@ -1625,11 +1671,14 @@ function createFlowTimers(input) {
       await store.setDue(chatId, null);
       const endAt = flow.schedule ? deadlineAt(run, timeZone, flow.schedule) ?? wake.at : wake.at;
       await input.onDeadline?.({ record, flow, run, at: endAt, timeZone });
+      const deadlineState = flow.schedule?.deadlineState;
+      fired(chatId, { kind: "deadline", stateId: deadlineState && stateById(flow, deadlineState) ? deadlineState : run.state });
       return;
     }
     if (due.giveUp) {
       await store.setDue(chatId, null);
       await input.onGiveUp?.({ record, flow, run, at });
+      fired(chatId, { kind: "give_up", stateId: run.state });
       return;
     }
     const transitions = [];
@@ -1710,6 +1759,7 @@ function createFlowTimers(input) {
     const answered = nudgeStep === void 0 ? noteAgentAnswer(next, at) : withNudgeStep(noteAgentAnswer(next, at), nudgeStep, nudgeStepOwn === true);
     const following = nextWake(flow, answered, at, timeZone, { pausedUntil: context.pausedUntil ?? null });
     await store.setDue(chatId, following ? dueInput(following, result.messageId ?? null, offset) : null, { unlessDelivery: true });
+    fired(chatId, { kind: reasons[0].kind, stateId: next.state, ...moved && { exitId: due.exitId }, reasons });
     if (!delivery) {
       await firstMessages?.record({ ownerId, at, timeZone });
       return;
@@ -1764,7 +1814,15 @@ function createFlowTimers(input) {
       });
     },
     tick,
-    reschedule
+    reschedule,
+    watch(chatId, onFired) {
+      const listeners = watchers.get(chatId) ?? /* @__PURE__ */ new Set();
+      watchers.set(chatId, listeners.add(onFired));
+      return () => {
+        listeners.delete(onFired);
+        if (listeners.size === 0) watchers.delete(chatId);
+      };
+    }
   };
 }
 
@@ -3289,6 +3347,7 @@ export {
   FLOW_TIMER_SPANS,
   FLOW_TIMER_UNITS,
   FLOW_TIME_ZONE_TOOL,
+  FlowTestClockError,
   RETURN_TARGET,
   activeExits,
   agentFlowSchema,
@@ -3333,6 +3392,7 @@ export {
   noteRejectedClaim,
   noteSideQuestion,
   openSideTasks,
+  parseDuration,
   parseFlow,
   placeOfPath,
   resolveFirstMessages,