zeitlich 0.2.38 → 0.2.40

package/dist/adapters/thread/google-genai/index.cjs

@@ -2,8 +2,22 @@
 
 var activity = require('@temporalio/activity');
 
-// src/lib/thread/manager.ts
+// src/adapters/thread/google-genai/adapter-id.ts
+var ADAPTER_ID = "googleGenAI";
+
+// src/lib/thread/keys.ts
 var THREAD_TTL_SECONDS = 60 * 60 * 24 * 90;
+function getThreadListKey(threadKey, threadId) {
+  return `${threadKey}:thread:${threadId}`;
+}
+function getThreadMetaKey(threadKey, threadId) {
+  return `${threadKey}:meta:thread:${threadId}`;
+}
+function getThreadStateKey(threadKey, threadId) {
+  return `${threadKey}:state:thread:${threadId}`;
+}
+
+// src/lib/thread/manager.ts
 var APPEND_IDEMPOTENT_SCRIPT = `
 if redis.call('EXISTS', KEYS[1]) == 1 then
   return 0
@@ -15,8 +29,8 @@ redis.call('EXPIRE', KEYS[2], tonumber(ARGV[1]))
 redis.call('SET', KEYS[1], '1', 'EX', tonumber(ARGV[1]))
 return 1
 `;
-function getThreadKey(threadId, key) {
-  return `${key}:thread:${threadId}`;
+function getDedupKey(threadId, id) {
+  return `dedup:${id}:thread:${threadId}`;
 }
 function createThreadManager(config) {
   const {
@@ -27,8 +41,9 @@ function createThreadManager(config) {
     deserialize = (raw) => JSON.parse(raw),
     idOf
   } = config;
-  const redisKey = getThreadKey(threadId, key);
-  const metaKey = getThreadKey(threadId, `${key}:meta`);
+  const redisKey = getThreadListKey(key, threadId);
+  const metaKey = getThreadMetaKey(key, threadId);
+  const stateKey = getThreadStateKey(key, threadId);
   async function assertThreadExists() {
     const exists = await redis.exists(metaKey);
     if (!exists) {
@@ -50,7 +65,7 @@ function createThreadManager(config) {
       await assertThreadExists();
       if (idOf) {
         const dedupId = messages.map(idOf).join(":");
-        const dedupKey = getThreadKey(threadId, `dedup:${dedupId}`);
+        const dedupKey = getDedupKey(threadId, dedupId);
         await redis.eval(
           APPEND_IDEMPOTENT_SCRIPT,
           2,
@@ -67,34 +82,98 @@ function createThreadManager(config) {
     async fork(newThreadId) {
       await assertThreadExists();
       const data = await redis.lrange(redisKey, 0, -1);
+      const stateRaw = await redis.get(stateKey);
       const forked = createThreadManager({
         ...config,
         threadId: newThreadId
       });
       await forked.initialize();
       if (data.length > 0) {
-        const newKey = getThreadKey(newThreadId, key);
+        const newKey = getThreadListKey(key, newThreadId);
         await redis.rpush(newKey, ...data);
         await redis.expire(newKey, THREAD_TTL_SECONDS);
       }
+      if (stateRaw != null) {
+        const newStateKey = getThreadStateKey(key, newThreadId);
+        await redis.set(newStateKey, stateRaw, "EX", THREAD_TTL_SECONDS);
+      }
       return forked;
     },
+    async replaceAll(messages) {
+      await assertThreadExists();
+      if (!idOf) {
+        throw new Error(
+          "replaceAll requires the thread manager to be configured with `idOf`"
+        );
+      }
+      const existing = await redis.lrange(redisKey, 0, -1);
+      const existingIds = existing.map((raw) => idOf(deserialize(raw))).filter((id) => typeof id === "string");
+      await redis.del(redisKey);
+      if (existingIds.length > 0) {
+        await redis.del(
+          ...existingIds.map((id) => getDedupKey(threadId, id))
+        );
+      }
+      if (messages.length > 0) {
+        await redis.rpush(redisKey, ...messages.map(serialize));
+        await redis.expire(redisKey, THREAD_TTL_SECONDS);
+      }
+      await redis.expire(metaKey, THREAD_TTL_SECONDS);
+    },
     async delete() {
-      await redis.del(redisKey, metaKey);
+      await redis.del(redisKey, metaKey, stateKey);
+    },
+    async loadState() {
+      const raw = await redis.get(stateKey);
+      if (raw == null) return null;
+      return JSON.parse(raw);
+    },
+    async saveState(state) {
+      await assertThreadExists();
+      await redis.set(
+        stateKey,
+        JSON.stringify(state),
+        "EX",
+        THREAD_TTL_SECONDS
+      );
+    },
+    async deleteState() {
+      await redis.del(stateKey);
     },
     async length() {
       await assertThreadExists();
       return redis.llen(redisKey);
     },
-    async truncate(length) {
+    async truncateFromId(messageId) {
       await assertThreadExists();
-      if (length <= 0) {
+      if (!idOf) {
+        throw new Error(
+          "truncateFromId requires the thread manager to be configured with `idOf`"
+        );
+      }
+      const data = await redis.lrange(redisKey, 0, -1);
+      let idx = -1;
+      const removedIds = [];
+      for (let i = 0; i < data.length; i++) {
+        const raw = data[i];
+        if (raw === void 0) continue;
+        const id = idOf(deserialize(raw));
+        if (idx === -1 && id === messageId) idx = i;
+        if (idx !== -1) removedIds.push(id);
+      }
+      if (idx === -1) return;
+      if (idx === 0) {
         await redis.del(redisKey);
         await redis.expire(metaKey, THREAD_TTL_SECONDS);
       } else {
-        await redis.ltrim(redisKey, 0, length - 1);
+        await redis.ltrim(redisKey, 0, idx - 1);
         await redis.expire(redisKey, THREAD_TTL_SECONDS);
       }
+      if (removedIds.length > 0) {
+        await redis.del(
+          ...removedIds.map((id) => getDedupKey(threadId, id))
+        );
+      }
     }
   };
 }
@@ -195,12 +274,33 @@ function createGoogleGenAIThreadManager(config) {
       const contents = mergeConsecutiveContents(conversationContents);
       return {
         contents: onPreparedMessage ? contents.map((msg, i) => onPreparedMessage(msg, i, contents)) : contents,
-        ...systemInstruction && systemInstruction.length > 0 ? { systemInstruction } : {},
-        storedLength: stored.length
+        ...systemInstruction && systemInstruction.length > 0 ? { systemInstruction } : {}
       };
     }
   };
-  return Object.assign(base, helpers);
+  const manager = Object.assign(base, helpers);
+  const originalFork = manager.fork.bind(manager);
+  manager.fork = async (newThreadId) => {
+    await originalFork(newThreadId);
+    const forked = createGoogleGenAIThreadManager({
+      ...config,
+      threadId: newThreadId
+    });
+    const { onForkPrepareThread, onForkTransform } = config.hooks ?? {};
+    if (!onForkPrepareThread && !onForkTransform) {
+      return forked;
+    }
+    let next = await forked.load();
+    if (onForkPrepareThread) {
+      next = await onForkPrepareThread(next);
+    }
+    if (onForkTransform) {
+      next = next.map((msg, i) => onForkTransform(msg, i, next));
+    }
+    await forked.replaceAll(next);
+    return forked;
+  };
+  return manager;
 }
 function getActivityContext() {
   try {
@@ -226,7 +326,7 @@ function createGoogleGenAIModelInvoker({
   hooks
 }) {
   return async function invokeGoogleGenAIModel2(config) {
-    const { threadId, threadKey, state } = config;
+    const { threadId, threadKey, state, assistantMessageId } = config;
     const { heartbeat, signal } = getActivityContext();
     const thread = createGoogleGenAIThreadManager({
       redis,
@@ -234,7 +334,8 @@ function createGoogleGenAIModelInvoker({
       key: threadKey,
       hooks
     });
-    const { contents, systemInstruction, storedLength } = await thread.prepareForInvocation();
+    await thread.truncateFromId(assistantMessageId);
+    const { contents, systemInstruction } = await thread.prepareForInvocation();
     const functionDeclarations = toFunctionDeclarations(state.tools);
     const tools = functionDeclarations.length > 0 ? [{ functionDeclarations }] : void 0;
     const stream = await client.models.generateContentStream({
@@ -269,8 +370,7 @@ function createGoogleGenAIModelInvoker({
         inputTokens: lastChunk.usageMetadata?.promptTokenCount,
         outputTokens: lastChunk.usageMetadata?.candidatesTokenCount,
         cachedReadTokens: lastChunk.usageMetadata?.cachedContentTokenCount
-      },
-      threadLengthAtCall: storedLength
+      }
     };
   };
 }
@@ -291,7 +391,6 @@ async function invokeGoogleGenAIModel({
 }
 
 // src/adapters/thread/google-genai/activities.ts
-var ADAPTER_PREFIX = "googleGenAI";
 function createGoogleGenAIAdapter(config) {
   const { redis } = config;
   const threadOps = {
@@ -345,21 +444,38 @@ function createGoogleGenAIAdapter(config) {
       const thread = createGoogleGenAIThreadManager({
         redis,
         threadId: sourceThreadId,
-        key: threadKey
+        key: threadKey,
+        hooks: config.hooks
       });
       await thread.fork(targetThreadId);
     },
-    async truncateThread(threadId, length, threadKey) {
+    async truncateThread(threadId, messageId, threadKey) {
+      const thread = createGoogleGenAIThreadManager({
+        redis,
+        threadId,
+        key: threadKey
+      });
+      await thread.truncateFromId(messageId);
+    },
+    async loadThreadState(threadId, threadKey) {
+      const thread = createGoogleGenAIThreadManager({
+        redis,
+        threadId,
+        key: threadKey
+      });
+      return thread.loadState();
+    },
+    async saveThreadState(threadId, state, threadKey) {
       const thread = createGoogleGenAIThreadManager({
         redis,
         threadId,
         key: threadKey
       });
-      await thread.truncate(length);
+      await thread.saveState(state);
     }
   };
   function createActivities(scope) {
-    const prefix = scope ? `${ADAPTER_PREFIX}${scope.charAt(0).toUpperCase()}${scope.slice(1)}` : ADAPTER_PREFIX;
+    const prefix = scope ? `${ADAPTER_ID}${scope.charAt(0).toUpperCase()}${scope.slice(1)}` : ADAPTER_ID;
     const cap = (s) => s.charAt(0).toUpperCase() + s.slice(1);
     return Object.fromEntries(
       Object.entries(threadOps).map(([k, v]) => [`${prefix}${cap(k)}`, v])
@@ -384,6 +500,7 @@ function createGoogleGenAIAdapter(config) {
   };
 }
 
+exports.ADAPTER_ID = ADAPTER_ID;
 exports.createGoogleGenAIAdapter = createGoogleGenAIAdapter;
 exports.createGoogleGenAIModelInvoker = createGoogleGenAIModelInvoker;
 exports.createGoogleGenAIThreadManager = createGoogleGenAIThreadManager;

package/dist/adapters/thread/google-genai/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/lib/thread/manager.ts","../../../../src/adapters/thread/google-genai/thread-manager.ts","../../../../src/lib/activity.ts","../../../../src/adapters/thread/google-genai/model-invoker.ts","../../../../src/adapters/thread/google-genai/activities.ts"],"names":["Context","invokeGoogleGenAIModel"],"mappings":";;;;;AAEA,IAAM,kBAAA,GAAqB,EAAA,GAAK,EAAA,GAAK,EAAA,GAAK,EAAA;AAW1C,IAAM,wBAAA,GAA2B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,CAAA;AAYjC,SAAS,YAAA,CAAa,UAAkB,GAAA,EAAqB;AAC3D,EAAA,OAAO,CAAA,EAAG,GAAG,CAAA,QAAA,EAAW,QAAQ,CAAA,CAAA;AAClC;AAMO,SAAS,oBACd,MAAA,EACsB;AACtB,EAAA,MAAM;AAAA,IACJ,KAAA;AAAA,IACA,QAAA;AAAA,IACA,GAAA,GAAM,UAAA;AAAA,IACN,SAAA,GAAY,CAAC,CAAA,KAAiB,IAAA,CAAK,UAAU,CAAC,CAAA;AAAA,IAC9C,WAAA,GAAc,CAAC,GAAA,KAAmB,IAAA,CAAK,MAAM,GAAG,CAAA;AAAA,IAChD;AAAA,GACF,GAAI,MAAA;AACJ,EAAA,MAAM,QAAA,GAAW,YAAA,CAAa,QAAA,EAAU,GAAG,CAAA;AAC3C,EAAA,MAAM,OAAA,GAAU,YAAA,CAAa,QAAA,EAAU,CAAA,EAAG,GAAG,CAAA,KAAA,CAAO,CAAA;AAEpD,EAAA,eAAe,kBAAA,GAAoC;AACjD,IAAA,MAAM,MAAA,GAAS,MAAM,KAAA,CAAM,MAAA,CAAO,OAAO,CAAA;AACzC,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,QAAA,EAAW,QAAQ,CAAA,QAAA,EAAW,GAAG,CAAA,gBAAA,CAAkB,CAAA;AAAA,IACrE;AAAA,EACF;AAEA,EAAA,OAAO;AAAA,IACL,MAAM,UAAA,GAA4B;AAChC,MAAA,MAAM,KAAA,CAAM,IAAI,QAAQ,CAAA;AACxB,MAAA,MAAM,KAAA,CAAM,GAAA,CAAI,OAAA,EAAS,GAAA,EAAK,MAAM,kBAAkB,CAAA;AAAA,IACxD,CAAA;AAAA,IAEA,MAAM,IAAA,GAAqB;AACzB,MAAA,MAAM,kBAAA,EAAmB;AACzB,MAAA,MAAM,OAAO,MAAM,KAAA,CAAM,MAAA,CAAO,QAAA,EAAU,GAAG,EAAE,CAAA;AAC/C,MAAA,OAAO,IAAA,CAAK,IAAI,WAAW,CAAA;AAAA,IAC7B,CAAA;AAAA,IAEA,MAAM,OAAO,QAAA,EAA8B;AACzC,MAAA,IAAI,QAAA,CAAS,WAAW,CAAA,EAAG;AAC3B,MAAA,MAAM,kBAAA,EAAmB;AAEzB,MAAA,IAAI,IAAA,EAAM;AACR,QAAA,MAAM,UAAU,QAAA,CAAS,GAAA,CAAI,IAAI,CAAA,CAAE,KAAK,GAAG,CAAA;AAC3C,QAAA,MAAM,QAAA,GAAW,YAAA,CAAa,QAAA,EAAU,CAAA,MAAA,EAAS,OAAO,CAAA,CAAE,CAAA;AAC1D,QAAA,MAAM,KAAA,CAAM,IAAA;AAAA,UACV,wBAAA;AAAA,UACA,CAAA;AAAA,UACA,QAAA;AAAA,UACA,QAAA;AAAA,UACA,OAAO,kBAAkB,CAAA;AAAA,UACzB,GAAG,QAAA,CAAS,GAAA,CAAI,SAAS;AAAA,SAC3B;AAAA,MACF,CAAA,MAAO;AACL,QAAA,MAAM,MAAM,KAAA,CAAM,QAAA,EAAU,GAAG,QAAA,CAAS,GAAA,CAAI,SAAS,CAAC,CAAA;AACtD,QAAA,MAAM,KAAA,CAAM,MAAA,CAAO,QAAA,EAAU,kBAAkB,CAAA;AAAA,MACjD;AAAA,IACF,CAAA;AAAA,IAEA,MAAM,KAAK,WAAA,EAAoD;AAC7D,MAAA,MAAM,kBAAA,EAAmB;AACzB,MAAA,MAAM,OAAO,MAAM,KAAA,CAAM,MAAA,CAAO,QAAA,EAAU,GAAG,EAAE,CAAA;AAC/C,MAAA,MAAM,SAAS,mBAAA,CAAoB;AAAA,QACjC,GAAG,MAAA;AAAA,QACH,QAAA,EAAU;AAAA,OACX,CAAA;AACD,MAAA,MAAM,OAAO,UAAA,EAAW;AACxB,MAAA,IAAI,IAAA,CAAK,SAAS,CAAA,EAAG;AACnB,QAAA,MAAM,MAAA,GAAS,YAAA,CAAa,WAAA,EAAa,GAAG,CAAA;AAC5C,QAAA,MAAM,KAAA,CAAM,KAAA,CAAM,MAAA,EAAQ,GAAG,IAAI,CAAA;AACjC,QAAA,MAAM,KAAA,CAAM,MAAA,CAAO,MAAA,EAAQ,kBAAkB,CAAA;AAAA,MAC/C;AACA,MAAA,OAAO,MAAA;AAAA,IACT,CAAA;AAAA,IAEA,MAAM,MAAA,GAAwB;AAC5B,MAAA,MAAM,KAAA,CAAM,GAAA,CAAI,QAAA,EAAU,OAAO,CAAA;AAAA,IACnC,CAAA;AAAA,IAEA,MAAM,MAAA,GAA0B;AAC9B,MAAA,MAAM,kBAAA,EAAmB;AACzB,MAAA,OAAO,KAAA,CAAM,KAAK,QAAQ,CAAA;AAAA,IAC5B,CAAA;AAAA,IAEA,MAAM,SAAS,MAAA,EAA+B;AAC5C,MAAA,MAAM,kBAAA,EAAmB;AACzB,MAAA,IAAI,UAAU,CAAA,EAAG;AACf,QAAA,MAAM,KAAA,CAAM,IAAI,QAAQ,CAAA;AACxB,QAAA,MAAM,KAAA,CAAM,MAAA,CAAO,OAAA,EAAS,kBAAkB,CAAA;AAAA,MAChD,CAAA,MAAO;AACL,QAAA,MAAM,KAAA,CAAM,KAAA,CAAM,QAAA,EAAU,CAAA,EAAG,SAAS,CAAC,CAAA;AACzC,QAAA,MAAM,KAAA,CAAM,MAAA,CAAO,QAAA,EAAU,kBAAkB,CAAA;AAAA,MACjD;AAAA,IAGF;AAAA,GACF;AACF;;;ACvEA,SAAS,gBAAgB,GAAA,EAA4B;AACnD,EAAA,OAAO,GAAA,CAAI,EAAA;AACb;AAGA,SAAS,QAAQ,OAAA,EAAqC;AACpD,EAAA,IAAI,OAAO,YAAY,QAAA,EAAU;AAC/B,IAAA,OAAO,CAAC,EAAE,IAAA,EAAM,OAAA,EAAS,CAAA;AAAA,EAC3B;AACA,EAAA,OAAO,OAAA;AACT;AAGA,SAAS,mBACP,OAAA,EACyB;AACzB,EAAA,IAAI,OAAO,YAAY,QAAA,EAAU;AAC/B,IAAA,OAAO,OAAA;AAAA,EACT;AACA,EAAA,OAAO,EAAE,QAAQ,OAAA,EAAQ;AAC3B;AAOA,SAAS,yBAAyB,QAAA,EAAgC;AAChE,EAAA,MAAM,SAAoB,EAAC;AAC3B,EAAA,KAAA,MAAW,WAAW,QAAA,EAAU;AAC9B,IAAA,MAAM,IAAA,GAAO,MAAA,CAAO,MAAA,CAAO,MAAA,GAAS,CAAC,CAAA;AACrC,IAAA,IAAI,IAAA,IAAQ,IAAA,CAAK,IAAA,KAAS,OAAA,CAAQ,IAAA,EAAM;AACtC,MAAA,IAAA,CAAK,KAAA,GAAQ,CAAC,GAAI,IAAA,CAAK,KAAA,IAAS,EAAC,EAAI,GAAI,OAAA,CAAQ,KAAA,IAAS,EAAG,CAAA;AAAA,IAC/D,CAAA,MAAO;AACL,MAAA,MAAA,CAAO,IAAA,CAAK,EAAE,GAAG,OAAA,EAAS,KAAA,EAAO,CAAC,GAAI,OAAA,CAAQ,KAAA,IAAS,EAAG,CAAA,EAAG,CAAA;AAAA,IAC/D;AAAA,EACF;AACA,EAAA,OAAO,MAAA;AACT;AAOO,SAAS,+BACd,MAAA,EAC0B;AAC1B,EAAA,MAAM,UAAA,GAAiD;AAAA,IACrD,OAAO,MAAA,CAAO,KAAA;AAAA,IACd,UAAU,MAAA,CAAO,QAAA;AAAA,IACjB,KAAK,MAAA,CAAO,GAAA;AAAA,IACZ,IAAA,EAAM;AAAA,GACR;AAEA,EAAA,MAAM,IAAA,GAAO,oBAAoB,UAAU,CAAA;AAE3C,EAAA,MAAM,OAAA,GAA6D;AAAA,IACjE,MAAM,iBAAA,CACJ,EAAA,EACA,OAAA,EACe;AACf,MAAA,MAAM,KAAK,MAAA,CAAO;AAAA,QAChB;AAAA,UACE,EAAA;AAAA,UACA,SAAS,EAAE,IAAA,EAAM,QAAQ,KAAA,EAAO,OAAA,CAAQ,OAAO,CAAA;AAAE;AACnD,OACD,CAAA;AAAA,IACH,CAAA;AAAA,IAEA,MAAM,mBAAA,CACJ,EAAA,EACA,OAAA,EACe;AACf,MAAA,MAAM,KAAA,GACJ,OAAO,OAAA,KAAY,QAAA,GAAW,CAAC,EAAE,IAAA,EAAM,OAAA,EAAS,CAAA,GAAI,OAAA;AACtD,MAAA,MAAM,KAAK,UAAA,EAAW;AACtB,MAAA,MAAM,KAAK,MAAA,CAAO;AAAA,QAChB;AAAA,UACE,EAAA;AAAA,UACA,OAAA,EAAS,EAAE,IAAA,EAAM,QAAA,EAAU,KAAA;AAAM;AACnC,OACD,CAAA;AAAA,IACH,CAAA;AAAA,IAEA,MAAM,kBAAA,CAAmB,EAAA,EAAY,KAAA,EAA8B;AACjE,MAAA,MAAM,KAAK,MAAA,CAAO;AAAA,QAChB;AAAA,UACE,EAAA;AAAA,UACA,OAAA,EAAS,EAAE,IAAA,EAAM,OAAA,EAAS,KAAA;AAAM;AAClC,OACD,CAAA;AAAA,IACH,CAAA;AAAA,IAEA,MAAM,gBAAA,CACJ,EAAA,EACA,UAAA,EACA,UACA,OAAA,EACe;AACf,MAAA,MAAM,KAAA,GAAgB,KAAA,CAAM,OAAA,CAAQ,OAAO,IACtC,OAAA,GACD;AAAA,QACE;AAAA,UACE,gBAAA,EAAkB;AAAA,YAChB,EAAA,EAAI,UAAA;AAAA,YACJ,IAAA,EAAM,QAAA;AAAA,YACN,QAAA,EAAU,mBAAmB,OAAO;AAAA;AACtC;AACF,OACF;AAEJ,MAAA,MAAM,KAAK,MAAA,CAAO;AAAA,QAChB;AAAA,UACE,EAAA;AAAA,UACA,OAAA,EAAS,EAAE,IAAA,EAAM,MAAA,EAAQ,KAAA;AAAM;AACjC,OACD,CAAA;AAAA,IACH,CAAA;AAAA,IAEA,MAAM,oBAAA,GAA8D;AAClE,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,IAAA,EAAK;AAC/B,MAAA,MAAM,EAAE,gBAAA,EAAkB,iBAAA,EAAkB,GAAI,MAAA,CAAO,SAAS,EAAC;AACjE,MAAA,MAAM,MAAA,GAAS,gBAAA,GACX,MAAA,CAAO,GAAA,CAAI,CAAC,GAAA,EAAK,CAAA,KAAM,gBAAA,CAAiB,GAAA,EAAK,CAAA,EAAG,MAAM,CAAC,CAAA,GACvD,MAAA;AAEJ,MAAA,IAAI,iBAAA;AACJ,MAAA,MAAM,uBAAkC,EAAC;AAEzC,MAAA,KAAA,MAAW,QAAQ,MAAA,EAAQ;AACzB,QAAA,IAAI,IAAA,CAAK,OAAA,CAAQ,IAAA,KAAS,QAAA,EAAU;AAClC,UAAA,iBAAA,GAAoB,IAAA,CAAK,OAAA,CAAQ,KAAA,IAAS,EAAC;AAAA,QAC7C,CAAA,MAAO;AACL,UAAA,oBAAA,CAAqB,IAAA,CAAK,KAAK,OAAO,CAAA;AAAA,QACxC;AAAA,MACF;AAEA,MAAA,MAAM,QAAA,GAAW,yBAAyB,oBAAoB,CAAA;AAC9D,MAAA,OAAO;AAAA,QACL,QAAA,EAAU,iBAAA,GACN,QAAA,CAAS,GAAA,CAAI,CAAC,GAAA,EAAK,CAAA,KAAM,iBAAA,CAAkB,GAAA,EAAK,CAAA,EAAG,QAAQ,CAAC,CAAA,GAC5D,QAAA;AAAA,QACJ,GAAI,qBAAqB,iBAAA,CAAkB,MAAA,GAAS,IAChD,EAAE,iBAAA,KACF,EAAC;AAAA,QACL,cAAc,MAAA,CAAO;AAAA,OACvB;AAAA,IACF;AAAA,GACF;AAEA,EAAA,OAAO,MAAA,CAAO,MAAA,CAAO,IAAA,EAAM,OAAO,CAAA;AACpC;AChMO,SAAS,kBAAA,GAGd;AACA,EAAA,IAAI;AACF,IAAA,MAAM,GAAA,GAAMA,iBAAQ,OAAA,EAAQ;AAC5B,IAAA,OAAO,EAAE,WAAW,MAAM,GAAA,CAAI,WAAU,EAAG,MAAA,EAAQ,IAAI,kBAAA,EAAmB;AAAA,EAC5E,CAAA,CAAA,MAAQ;AACN,IAAA,OAAO,EAAC;AAAA,EACV;AACF;;;ACDA,SAAS,uBACP,KAAA,EACuB;AACvB,EAAA,OAAO,KAAA,CAAM,GAAA,CAAI,CAAC,CAAA,MAAO;AAAA,IACvB,MAAM,CAAA,CAAE,IAAA;AAAA,IACR,aAAa,CAAA,CAAE,WAAA;AAAA,IACf,sBAAsB,CAAA,CAAE;AAAA,GAC1B,CAAE,CAAA;AACJ;AA0BO,SAAS,6BAAA,CAA8B;AAAA,EAC5C,KAAA;AAAA,EACA,MAAA;AAAA,EACA,KAAA;AAAA,EACA;AACF,CAAA,EAAkC;AAChC,EAAA,OAAO,eAAeC,wBACpB,MAAA,EACiC;AACjC,IAAA,MAAM,EAAE,QAAA,EAAU,SAAA,EAAW,KAAA,EAAM,GAAI,MAAA;AACvC,IAAA,MAAM,EAAE,SAAA,EAAW,MAAA,EAAO,GAAI,kBAAA,EAAmB;AAEjD,IAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,MAC5C,KAAA;AAAA,MACA,QAAA;AAAA,MACA,GAAA,EAAK,SAAA;AAAA,MACL;AAAA,KACD,CAAA;AACD,IAAA,MAAM,EAAE,QAAA,EAAU,iBAAA,EAAmB,cAAa,GAChD,MAAM,OAAO,oBAAA,EAAqB;AAEpC,IAAA,MAAM,oBAAA,GAAuB,sBAAA,CAAuB,KAAA,CAAM,KAAK,CAAA;AAC/D,IAAA,MAAM,KAAA,GACJ,qBAAqB,MAAA,GAAS,CAAA,GAAI,CAAC,EAAE,oBAAA,EAAsB,CAAA,GAAI,MAAA;AAEjE,IAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,MAAA,CAAO,qBAAA,CAAsB;AAAA,MACvD,KAAA;AAAA,MACA,QAAA;AAAA,MACA,MAAA,EAAQ;AAAA,QACN,GAAI,iBAAA,GAAoB,EAAE,iBAAA,KAAsB,EAAC;AAAA,QACjD,GAAI,KAAA,GAAQ,EAAE,KAAA,KAAU,EAAC;AAAA,QACzB,WAAA,EAAa;AAAA;AACf,KACD,CAAA;AAED,IAAA,MAAM,WAAmB,EAAC;AAC1B,IAAA,IAAI,SAAA;AACJ,IAAA,WAAA,MAAiB,SAAS,MAAA,EAAQ;AAChC,MAAA,SAAA,GAAY,KAAA;AACZ,MAAA,QAAA,CAAS,IAAA,CAAK,GAAI,KAAA,CAAM,UAAA,GAAa,CAAC,CAAA,EAAG,OAAA,EAAS,KAAA,IAAS,EAAG,CAAA;AAC9D,MAAA,SAAA,IAAY;AAAA,IACd;AAEA,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,MAAM,IAAI,MAAM,wDAAwD,CAAA;AAAA,IAC1E;AAEA,IAAA,MAAM,YAAA,GAAwB,EAAE,IAAA,EAAM,OAAA,EAAS,OAAO,QAAA,EAAS;AAC/D,IAAA,MAAM,aAAA,GAAgB,SAAA,CAAU,aAAA,IAAiB,EAAC;AAElD,IAAA,OAAO;AAAA,MACL,OAAA,EAAS,YAAA;AAAA,MACT,YAAA,EAAc,aAAA,CAAc,GAAA,CAAI,CAAC,EAAA,MAAQ;AAAA,QACvC,IAAI,EAAA,CAAG,EAAA;AAAA,QACP,IAAA,EAAM,GAAG,IAAA,IAAQ,EAAA;AAAA,QACjB,IAAA,EAAM,EAAA,CAAG,IAAA,IAAQ;AAAC,OACpB,CAAE,CAAA;AAAA,MACF,KAAA,EAAO;AAAA,QACL,WAAA,EAAa,UAAU,aAAA,EAAe,gBAAA;AAAA,QACtC,YAAA,EAAc,UAAU,aAAA,EAAe,oBAAA;AAAA,QACvC,gBAAA,EAAkB,UAAU,aAAA,EAAe;AAAA,OAC7C;AAAA,MACA,kBAAA,EAAoB;AAAA,KACtB;AAAA,EACF,CAAA;AACF;AAOA,eAAsB,sBAAA,CAAuB;AAAA,EAC3C,KAAA;AAAA,EACA,MAAA;AAAA,EACA,KAAA;AAAA,EACA,KAAA;AAAA,EACA;AACF,CAAA,EAMoC;AAClC,EAAA,MAAM,UAAU,6BAAA,CAA8B;AAAA,IAC5C,KAAA;AAAA,IACA,MAAA;AAAA,IACA,KAAA;AAAA,IACA;AAAA,GACD,CAAA;AACD,EAAA,OAAO,QAAQ,MAAM,CAAA;AACvB;;;AC/HA,IAAM,cAAA,GAAiB,aAAA;AAmHhB,SAAS,yBACd,MAAA,EACoB;AACpB,EAAA,MAAM,EAAE,OAAM,GAAI,MAAA;AAElB,EAAA,MAAM,SAAA,GAA2C;AAAA,IAC/C,MAAM,gBAAA,CACJ,QAAA,EACA,SAAA,EACe;AACf,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA;AAAA,QACA,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,MAAM,OAAO,UAAA,EAAW;AAAA,IAC1B,CAAA;AAAA,IAEA,MAAM,kBAAA,CACJ,QAAA,EACA,EAAA,EACA,SACA,SAAA,EACe;AACf,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA;AAAA,QACA,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,MAAM,MAAA,CAAO,iBAAA,CAAkB,EAAA,EAAI,OAAO,CAAA;AAAA,IAC5C,CAAA;AAAA,IAEA,MAAM,mBAAA,CACJ,QAAA,EACA,EAAA,EACA,SACA,SAAA,EACe;AACf,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA;AAAA,QACA,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,MAAM,MAAA,CAAO,mBAAA,CAAoB,EAAA,EAAI,OAAO,CAAA;AAAA,IAC9C,CAAA;AAAA,IAEA,MAAM,gBAAA,CAAiB,EAAA,EAAY,GAAA,EAAsC;AACvE,MAAA,MAAM,EAAE,QAAA,EAAU,SAAA,EAAW,UAAA,EAAY,QAAA,EAAU,SAAQ,GAAI,GAAA;AAC/D,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA;AAAA,QACA,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,MAAM,MAAA,CAAO,gBAAA;AAAA,QACX,EAAA;AAAA,QACA,UAAA;AAAA,QACA,QAAA;AAAA,QACA;AAAA,OACF;AAAA,IACF,CAAA;AAAA,IAEA,MAAM,kBAAA,CACJ,QAAA,EACA,EAAA,EACA,SACA,SAAA,EACe;AACf,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA;AAAA,QACA,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,MAAM,OAAO,kBAAA,CAAmB,EAAA,EAAI,OAAA,CAAQ,KAAA,IAAS,EAAE,CAAA;AAAA,IACzD,CAAA;AAAA,IAEA,MAAM,UAAA,CACJ,cAAA,EACA,cAAA,EACA,SAAA,EACe;AACf,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA,EAAU,cAAA;AAAA,QACV,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,MAAM,MAAA,CAAO,KAAK,cAAc,CAAA;AAAA,IAClC,CAAA;AAAA,IAEA,MAAM,cAAA,CACJ,QAAA,EACA,MAAA,EACA,SAAA,EACe;AACf,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA;AAAA,QACA,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,MAAM,MAAA,CAAO,SAAS,MAAM,CAAA;AAAA,IAC9B;AAAA,GACF;AAEA,EAAA,SAAS,iBACP,KAAA,EACyB;AACzB,IAAA,MAAM,SAAS,KAAA,GACX,CAAA,EAAG,cAAc,CAAA,EAAG,MAAM,MAAA,CAAO,CAAC,CAAA,CAAE,WAAA,EAAa,CAAA,EAAG,KAAA,CAAM,KAAA,CAAM,CAAC,CAAC,CAAA,CAAA,GAClE,cAAA;AACJ,IAAA,MAAM,GAAA,GAAM,CAAC,CAAA,KAAsB,CAAA,CAAE,MAAA,CAAO,CAAC,CAAA,CAAE,WAAA,EAAY,GAAI,CAAA,CAAE,KAAA,CAAM,CAAC,CAAA;AACxE,IAAA,OAAO,MAAA,CAAO,WAAA;AAAA,MACZ,OAAO,OAAA,CAAQ,SAAS,EAAE,GAAA,CAAI,CAAC,CAAC,CAAA,EAAG,CAAC,MAAM,CAAC,CAAA,EAAG,MAAM,CAAA,EAAG,GAAA,CAAI,CAAC,CAAC,CAAA,CAAA,EAAI,CAAC,CAAC;AAAA,KACrE;AAAA,EACF;AAEA,EAAA,MAAM,WAAA,GAAc,CAClB,KAAA,EACA,MAAA,KAEA,6BAAA,CAA8B;AAAA,IAC5B,KAAA;AAAA,IACA,MAAA;AAAA,IACA,KAAA;AAAA,IACA,OAAO,MAAA,CAAO;AAAA,GACf,CAAA;AAEH,EAAA,MAAM,OAAA,GACJ,MAAA,CAAO,KAAA,IAAS,MAAA,CAAO,MAAA,GACnB,WAAA,CAAY,MAAA,CAAO,KAAA,EAAO,MAAA,CAAO,MAAM,CAAA,IACrC,MAAM;AACN,IAAA,MAAM,IAAI,KAAA;AAAA,MACR;AAAA,KAEF;AAAA,EACF,CAAA,CAAA;AAEN,EAAA,OAAO;AAAA,IACL,gBAAA;AAAA,IACA,OAAA;AAAA,IACA,kBAAA,EAAoB,WAAA;AAAA,IACpB,WAAA,EAAa,CAAC,OAAA,KAAY;AAAA,GAC5B;AACF","file":"index.cjs","sourcesContent":["import type { ThreadManagerConfig, BaseThreadManager } from \"./types\";\n\nconst THREAD_TTL_SECONDS = 60 * 60 * 24 * 90; // 90 days\n\n/**\n * Lua script for atomic idempotent append.\n * Checks a dedup key; if it exists the message was already appended and we\n * return 0. Otherwise appends all messages to the list, sets TTL on both\n * the list and the dedup key, and returns 1.\n *\n * KEYS[1] = dedup key, KEYS[2] = list key\n * ARGV[1] = TTL seconds, ARGV[2..N] = serialised messages\n */\nconst APPEND_IDEMPOTENT_SCRIPT = `\nif redis.call('EXISTS', KEYS[1]) == 1 then\n  return 0\nend\nfor i = 2, #ARGV do\n  redis.call('RPUSH', KEYS[2], ARGV[i])\nend\nredis.call('EXPIRE', KEYS[2], tonumber(ARGV[1]))\nredis.call('SET', KEYS[1], '1', 'EX', tonumber(ARGV[1]))\nreturn 1\n`;\n\nfunction getThreadKey(threadId: string, key: string): string {\n  return `${key}:thread:${threadId}`;\n}\n\n/**\n * Creates a generic thread manager for handling conversation state in Redis.\n * Framework-agnostic — works with any serializable message type.\n */\nexport function createThreadManager<T>(\n  config: ThreadManagerConfig<T>\n): BaseThreadManager<T> {\n  const {\n    redis,\n    threadId,\n    key = \"messages\",\n    serialize = (m: T): string => JSON.stringify(m),\n    deserialize = (raw: string): T => JSON.parse(raw) as T,\n    idOf,\n  } = config;\n  const redisKey = getThreadKey(threadId, key);\n  const metaKey = getThreadKey(threadId, `${key}:meta`);\n\n  async function assertThreadExists(): Promise<void> {\n    const exists = await redis.exists(metaKey);\n    if (!exists) {\n      throw new Error(`Thread \"${threadId}\" (key: ${key}) does not exist`);\n    }\n  }\n\n  return {\n    async initialize(): Promise<void> {\n      await redis.del(redisKey);\n      await redis.set(metaKey, \"1\", \"EX\", THREAD_TTL_SECONDS);\n    },\n\n    async load(): Promise<T[]> {\n      await assertThreadExists();\n      const data = await redis.lrange(redisKey, 0, -1);\n      return data.map(deserialize);\n    },\n\n    async append(messages: T[]): Promise<void> {\n      if (messages.length === 0) return;\n      await assertThreadExists();\n\n      if (idOf) {\n        const dedupId = messages.map(idOf).join(\":\");\n        const dedupKey = getThreadKey(threadId, `dedup:${dedupId}`);\n        await redis.eval(\n          APPEND_IDEMPOTENT_SCRIPT,\n          2,\n          dedupKey,\n          redisKey,\n          String(THREAD_TTL_SECONDS),\n          ...messages.map(serialize)\n        );\n      } else {\n        await redis.rpush(redisKey, ...messages.map(serialize));\n        await redis.expire(redisKey, THREAD_TTL_SECONDS);\n      }\n    },\n\n    async fork(newThreadId: string): Promise<BaseThreadManager<T>> {\n      await assertThreadExists();\n      const data = await redis.lrange(redisKey, 0, -1);\n      const forked = createThreadManager({\n        ...config,\n        threadId: newThreadId,\n      });\n      await forked.initialize();\n      if (data.length > 0) {\n        const newKey = getThreadKey(newThreadId, key);\n        await redis.rpush(newKey, ...data);\n        await redis.expire(newKey, THREAD_TTL_SECONDS);\n      }\n      return forked;\n    },\n\n    async delete(): Promise<void> {\n      await redis.del(redisKey, metaKey);\n    },\n\n    async length(): Promise<number> {\n      await assertThreadExists();\n      return redis.llen(redisKey);\n    },\n\n    async truncate(length: number): Promise<void> {\n      await assertThreadExists();\n      if (length <= 0) {\n        await redis.del(redisKey);\n        await redis.expire(metaKey, THREAD_TTL_SECONDS);\n      } else {\n        await redis.ltrim(redisKey, 0, length - 1);\n        await redis.expire(redisKey, THREAD_TTL_SECONDS);\n      }\n      // Dedup keys for removed messages are left to expire via their TTL.\n      // Post-truncate appends use fresh ids so collisions do not occur in practice.\n    },\n  };\n}\n","import type Redis from \"ioredis\";\nimport type { Content, Part } from \"@google/genai\";\nimport { createThreadManager } from \"../../../lib/thread/manager\";\nimport type {\n  ProviderThreadManager,\n  ThreadManagerConfig,\n  ThreadManagerHooks,\n} from \"../../../lib/thread/types\";\nimport type { GoogleGenAIToolResponse } from \"./activities\";\n\n/** SDK-native content type for Google GenAI human messages */\nexport type GoogleGenAIContent = string | Part[];\n\n/** SDK-native content type for Google GenAI system instructions */\nexport type GoogleGenAISystemContent = string | Part[];\n\n/** A Content with a unique ID for idempotent Redis storage */\nexport interface StoredContent {\n  id: string;\n  content: Content;\n}\n\nexport type GoogleGenAIThreadManagerHooks = ThreadManagerHooks<\n  StoredContent,\n  Content\n>;\n\nexport interface GoogleGenAIThreadManagerConfig {\n  redis: Redis;\n  threadId: string;\n  /** Thread key, defaults to 'messages' */\n  key?: string;\n  hooks?: GoogleGenAIThreadManagerHooks;\n}\n\n/** Prepared payload ready to send to the Google GenAI API */\nexport interface GoogleGenAIInvocationPayload {\n  contents: Content[];\n  systemInstruction?: Part[];\n  /** Number of stored messages loaded from Redis before preparation. */\n  storedLength: number;\n}\n\n/** Thread manager with Google GenAI Content convenience helpers */\nexport interface GoogleGenAIThreadManager extends ProviderThreadManager<\n  StoredContent,\n  GoogleGenAIContent,\n  GoogleGenAIToolResponse,\n  GoogleGenAISystemContent\n> {\n  appendModelContent(id: string, parts: Part[]): Promise<void>;\n  prepareForInvocation(): Promise<GoogleGenAIInvocationPayload>;\n}\n\nfunction storedContentId(msg: StoredContent): string {\n  return msg.id;\n}\n\n/** Normalise content into Part[] */\nfunction toParts(content: GoogleGenAIContent): Part[] {\n  if (typeof content === \"string\") {\n    return [{ text: content }];\n  }\n  return content;\n}\n\n/** Convert a string or object into a Record suitable for functionResponse.response */\nfunction toFunctionResponse(\n  content: string | Record<string, unknown>\n): Record<string, unknown> {\n  if (typeof content === \"object\") {\n    return content;\n  }\n  return { result: content };\n}\n\n/**\n * Merge consecutive Content objects sharing the same role.\n * The Gemini API requires alternating user/model turns; without\n * merging, multiple sequential tool-result messages would violate this.\n */\nfunction mergeConsecutiveContents(contents: Content[]): Content[] {\n  const merged: Content[] = [];\n  for (const content of contents) {\n    const last = merged[merged.length - 1];\n    if (last && last.role === content.role) {\n      last.parts = [...(last.parts ?? []), ...(content.parts ?? [])];\n    } else {\n      merged.push({ ...content, parts: [...(content.parts ?? [])] });\n    }\n  }\n  return merged;\n}\n\n/**\n * Creates a Google GenAI-specific thread manager that stores StoredContent\n * instances in Redis and provides convenience helpers for creating and\n * appending typed Content messages.\n */\nexport function createGoogleGenAIThreadManager(\n  config: GoogleGenAIThreadManagerConfig\n): GoogleGenAIThreadManager {\n  const baseConfig: ThreadManagerConfig<StoredContent> = {\n    redis: config.redis,\n    threadId: config.threadId,\n    key: config.key,\n    idOf: storedContentId,\n  };\n\n  const base = createThreadManager(baseConfig);\n\n  const helpers: Omit<GoogleGenAIThreadManager, keyof typeof base> = {\n    async appendUserMessage(\n      id: string,\n      content: GoogleGenAIContent\n    ): Promise<void> {\n      await base.append([\n        {\n          id,\n          content: { role: \"user\", parts: toParts(content) },\n        },\n      ]);\n    },\n\n    async appendSystemMessage(\n      id: string,\n      content: GoogleGenAISystemContent\n    ): Promise<void> {\n      const parts: Part[] =\n        typeof content === \"string\" ? [{ text: content }] : content;\n      await base.initialize();\n      await base.append([\n        {\n          id,\n          content: { role: \"system\", parts },\n        },\n      ]);\n    },\n\n    async appendModelContent(id: string, parts: Part[]): Promise<void> {\n      await base.append([\n        {\n          id,\n          content: { role: \"model\", parts },\n        },\n      ]);\n    },\n\n    async appendToolResult(\n      id: string,\n      toolCallId: string,\n      toolName: string,\n      content: GoogleGenAIToolResponse\n    ): Promise<void> {\n      const parts: Part[] = Array.isArray(content)\n        ? (content as Part[])\n        : [\n            {\n              functionResponse: {\n                id: toolCallId,\n                name: toolName,\n                response: toFunctionResponse(content),\n              },\n            },\n          ];\n\n      await base.append([\n        {\n          id,\n          content: { role: \"user\", parts },\n        },\n      ]);\n    },\n\n    async prepareForInvocation(): Promise<GoogleGenAIInvocationPayload> {\n      const stored = await base.load();\n      const { onPrepareMessage, onPreparedMessage } = config.hooks ?? {};\n      const mapped = onPrepareMessage\n        ? stored.map((msg, i) => onPrepareMessage(msg, i, stored))\n        : stored;\n\n      let systemInstruction: Part[] | undefined;\n      const conversationContents: Content[] = [];\n\n      for (const item of mapped) {\n        if (item.content.role === \"system\") {\n          systemInstruction = item.content.parts ?? [];\n        } else {\n          conversationContents.push(item.content);\n        }\n      }\n\n      const contents = mergeConsecutiveContents(conversationContents);\n      return {\n        contents: onPreparedMessage\n          ? contents.map((msg, i) => onPreparedMessage(msg, i, contents))\n          : contents,\n        ...(systemInstruction && systemInstruction.length > 0\n          ? { systemInstruction }\n          : {}),\n        storedLength: stored.length,\n      };\n    },\n  };\n\n  return Object.assign(base, helpers);\n}\n","import { Context } from \"@temporalio/activity\";\nimport type { WorkflowClient } from \"@temporalio/client\";\nimport type { BaseAgentState, RunAgentConfig } from \"./types\";\nimport type { JsonValue } from \"./state/types\";\nimport type {\n  ActivityToolHandler,\n  RouterContext,\n  ToolHandlerResponse,\n} from \"./tool-router/types\";\n\n/**\n * Safely retrieve Temporal activity heartbeat and cancellation signal.\n * Returns empty object when called outside a Temporal activity (e.g. tests).\n */\nexport function getActivityContext(): {\n  heartbeat?: () => void;\n  signal?: AbortSignal;\n} {\n  try {\n    const ctx = Context.current();\n    return { heartbeat: () => ctx.heartbeat(), signal: ctx.cancellationSignal };\n  } catch {\n    return {};\n  }\n}\n\n/**\n * Query the parent workflow's state from within an activity.\n * Resolves the workflow handle from the current activity context.\n */\nexport async function queryParentWorkflowState<T>(\n  client: WorkflowClient\n): Promise<T> {\n  const { workflowExecution } = Context.current().info;\n  const handle = client.getHandle(\n    workflowExecution.workflowId,\n    workflowExecution.runId\n  );\n  return handle.query<T>(\"getAgentState\");\n}\n\n/**\n * Wraps a handler into a scope-prefixed `RunAgentActivity` by auto-fetching\n * the parent workflow's agent state before each invocation.\n *\n * Returns a `Record` with a single key `run<Scope>` so it can be spread\n * into the activities object alongside adapter activities.\n *\n * @param scope - Workflow scope used to derive the activity name.\n *   `\"myAgentWorkflow\"` produces `{ runMyAgentWorkflow: fn }`.\n *\n * @example\n * ```typescript\n * import { createRunAgentActivity } from 'zeitlich';\n *\n * return {\n *   ...adapter.createActivities(\"myAgentWorkflow\"),\n *   ...createRunAgentActivity(client, adapter.invoker, \"myAgentWorkflow\"),\n * };\n * ```\n */\nexport function createRunAgentActivity<\n  R,\n  S extends BaseAgentState = BaseAgentState,\n>(\n  client: WorkflowClient,\n  handler: (config: RunAgentConfig & { state: S }) => Promise<R>,\n  scope: string\n): Record<string, (config: RunAgentConfig) => Promise<R>> {\n  const name = `run${scope.charAt(0).toUpperCase()}${scope.slice(1)}`;\n  return {\n    [name]: async (config: RunAgentConfig) => {\n      const state = await queryParentWorkflowState<S>(client);\n      return handler({ ...config, state });\n    },\n  };\n}\n\n/**\n * Context injected into tool handlers created via {@link withParentWorkflowState}.\n */\nexport interface AgentStateContext<\n  S extends BaseAgentState = BaseAgentState,\n> extends RouterContext {\n  state: S;\n}\n\n/**\n * Wraps a tool handler into an `ActivityToolHandler` by auto-fetching the\n * parent workflow's agent state before each invocation.\n *\n * @typeParam S - Custom agent state type (defaults to `BaseAgentState`)\n *\n * @example\n * ```typescript\n * import { withParentWorkflowState, type AgentStateContext } from 'zeitlich';\n *\n * // With custom state:\n * interface MyState extends BaseAgentState { customField: string }\n * const myHandler = withParentWorkflowState<MyArgs, MyResult, MyState>(\n *   client,\n *   async (args, ctx) => {\n *     console.log(ctx.state.customField);\n *     return { toolResponse: 'done', data: null };\n *   },\n * );\n * ```\n */\nexport function withParentWorkflowState<\n  TArgs,\n  TResult,\n  S extends BaseAgentState = BaseAgentState,\n  TToolResponse = JsonValue,\n>(\n  client: WorkflowClient,\n  handler: (\n    args: TArgs,\n    context: AgentStateContext<S>\n  ) => Promise<ToolHandlerResponse<TResult, TToolResponse>>\n): ActivityToolHandler<TArgs, TResult, RouterContext, TToolResponse> {\n  return async (args, context) => {\n    const state = await queryParentWorkflowState<S>(client);\n    return handler(args, { ...context, state });\n  };\n}\n","import type Redis from \"ioredis\";\nimport type {\n  GoogleGenAI,\n  Content,\n  FunctionDeclaration,\n  Part,\n  GenerateContentResponse,\n} from \"@google/genai\";\nimport type { SerializableToolDefinition } from \"../../../lib/types\";\nimport type { AgentResponse, ModelInvokerConfig } from \"../../../lib/model\";\nimport {\n  createGoogleGenAIThreadManager,\n  type GoogleGenAIThreadManagerHooks,\n} from \"./thread-manager\";\nimport { getActivityContext } from \"../../../lib/activity\";\n\nexport interface GoogleGenAIModelInvokerConfig {\n  redis: Redis;\n  client: GoogleGenAI;\n  model: string;\n  hooks?: GoogleGenAIThreadManagerHooks;\n}\n\nfunction toFunctionDeclarations(\n  tools: SerializableToolDefinition[]\n): FunctionDeclaration[] {\n  return tools.map((t) => ({\n    name: t.name,\n    description: t.description,\n    parametersJsonSchema: t.schema,\n  }));\n}\n\n/**\n * Creates a Google GenAI model invoker that satisfies the generic\n * `ModelInvoker<Content>` contract.\n *\n * Internally streams the response and emits Temporal heartbeats on each\n * chunk so that long-running LLM calls remain visible to the scheduler.\n * The caller is responsible for appending the response to the thread.\n *\n * @example\n * ```typescript\n * import { createGoogleGenAIModelInvoker } from 'zeitlich/adapters/thread/google-genai';\n * import { createRunAgentActivity } from 'zeitlich';\n * import { GoogleGenAI } from '@google/genai';\n *\n * const client = new GoogleGenAI({ apiKey: process.env.GEMINI_API_KEY });\n * const invoker = createGoogleGenAIModelInvoker({\n *   redis,\n *   client,\n *   model: 'gemini-2.5-flash',\n * });\n *\n * return { ...createRunAgentActivity(client, invoker, \"myAgent\") };\n * ```\n */\nexport function createGoogleGenAIModelInvoker({\n  redis,\n  client,\n  model,\n  hooks,\n}: GoogleGenAIModelInvokerConfig) {\n  return async function invokeGoogleGenAIModel(\n    config: ModelInvokerConfig\n  ): Promise<AgentResponse<Content>> {\n    const { threadId, threadKey, state } = config;\n    const { heartbeat, signal } = getActivityContext();\n\n    const thread = createGoogleGenAIThreadManager({\n      redis,\n      threadId,\n      key: threadKey,\n      hooks,\n    });\n    const { contents, systemInstruction, storedLength } =\n      await thread.prepareForInvocation();\n\n    const functionDeclarations = toFunctionDeclarations(state.tools);\n    const tools =\n      functionDeclarations.length > 0 ? [{ functionDeclarations }] : undefined;\n\n    const stream = await client.models.generateContentStream({\n      model,\n      contents,\n      config: {\n        ...(systemInstruction ? { systemInstruction } : {}),\n        ...(tools ? { tools } : {}),\n        abortSignal: signal,\n      },\n    });\n\n    const allParts: Part[] = [];\n    let lastChunk: GenerateContentResponse | undefined;\n    for await (const chunk of stream) {\n      lastChunk = chunk;\n      allParts.push(...(chunk.candidates?.[0]?.content?.parts ?? []));\n      heartbeat?.();\n    }\n\n    if (!lastChunk) {\n      throw new Error(\"Google GenAI stream ended without producing any chunks\");\n    }\n\n    const modelContent: Content = { role: \"model\", parts: allParts };\n    const functionCalls = lastChunk.functionCalls ?? [];\n\n    return {\n      message: modelContent,\n      rawToolCalls: functionCalls.map((fc) => ({\n        id: fc.id,\n        name: fc.name ?? \"\",\n        args: fc.args ?? {},\n      })),\n      usage: {\n        inputTokens: lastChunk.usageMetadata?.promptTokenCount,\n        outputTokens: lastChunk.usageMetadata?.candidatesTokenCount,\n        cachedReadTokens: lastChunk.usageMetadata?.cachedContentTokenCount,\n      },\n      threadLengthAtCall: storedLength,\n    };\n  };\n}\n\n/**\n * Standalone function for one-shot Google GenAI model invocation.\n * Convenience wrapper around createGoogleGenAIModelInvoker for cases\n * where you don't need to reuse the invoker.\n */\nexport async function invokeGoogleGenAIModel({\n  redis,\n  client,\n  model,\n  hooks,\n  config,\n}: {\n  redis: Redis;\n  client: GoogleGenAI;\n  model: string;\n  hooks?: GoogleGenAIThreadManagerHooks;\n  config: ModelInvokerConfig;\n}): Promise<AgentResponse<Content>> {\n  const invoker = createGoogleGenAIModelInvoker({\n    redis,\n    client,\n    model,\n    hooks,\n  });\n  return invoker(config);\n}\n","import type Redis from \"ioredis\";\nimport type { GoogleGenAI, Content, Part } from \"@google/genai\";\nimport type { ToolResultConfig } from \"../../../lib/types\";\nimport type {\n  ActivityToolHandler,\n  RouterContext,\n  ToolHandlerResponse,\n} from \"../../../lib/tool-router/types\";\nimport type {\n  ThreadOps,\n  PrefixedThreadOps,\n  ScopedPrefix,\n} from \"../../../lib/session/types\";\nimport type { ModelInvoker } from \"../../../lib/model\";\nimport {\n  createGoogleGenAIThreadManager,\n  type GoogleGenAIContent,\n  type GoogleGenAISystemContent,\n  type GoogleGenAIThreadManagerHooks,\n} from \"./thread-manager\";\nimport { createGoogleGenAIModelInvoker } from \"./model-invoker\";\n\nconst ADAPTER_PREFIX = \"googleGenAI\" as const;\n\nexport type GoogleGenAIThreadOps<TScope extends string = \"\"> =\n  PrefixedThreadOps<\n    ScopedPrefix<TScope, typeof ADAPTER_PREFIX>,\n    GoogleGenAIContent\n  >;\n\nexport interface GoogleGenAIAdapterConfig {\n  redis: Redis;\n  client?: GoogleGenAI;\n  /** Default model name (e.g. 'gemini-2.5-flash'). If omitted, use `createModelInvoker()` */\n  model?: string;\n  hooks?: GoogleGenAIThreadManagerHooks;\n}\n\n/**\n * Tool response type accepted by the Google GenAI adapter.\n *\n * Handlers can return:\n * - **`string`** — plain text, wrapped in a `functionResponse` part.\n * - **`Record<string, unknown>`** — structured object used as `functionResponse.response`.\n * - **`Part[]`** — pre-built parts used directly as `Content.parts`.\n *   The handler is responsible for building correct Part objects (e.g. `functionResponse`,\n *   `inlineData`, `text`). Use `context.toolCallId` and `context.toolName` to construct\n *   `functionResponse` parts.\n *\n * @example\n * ```typescript\n * adapter.wrapHandler(async (args, ctx) => ({\n *   toolResponse: [\n *     { functionResponse: { id: ctx.toolCallId, name: ctx.toolName, response: { result: \"done\" } } },\n *     { inlineData: { data: base64, mimeType: \"image/png\" } },\n *   ],\n *   data: null,\n * }));\n * ```\n */\nexport type GoogleGenAIToolResponse = string | Record<string, unknown> | Part[];\n\nexport interface GoogleGenAIAdapter {\n  /** Model invoker using the default model (only available when `model` was provided) */\n  invoker: ModelInvoker<Content>;\n  /** Create an invoker for a specific model name (for multi-model setups) */\n  createModelInvoker(model: string, client: GoogleGenAI): ModelInvoker<Content>;\n  /**\n   * Create prefixed thread activities for registration on the worker.\n   *\n   * @param scope - Workflow name appended to the adapter prefix.\n   *   Use different scopes for the main agent vs subagents to avoid collisions.\n   *\n   * @example\n   * ```typescript\n   * adapter.createActivities(\"codingAgent\")\n   * // → { googleGenAICodingAgentInitializeThread, googleGenAICodingAgentAppendHumanMessage, … }\n   *\n   * adapter.createActivities(\"researchAgent\")\n   * // → { googleGenAIResearchAgentInitializeThread, … }\n   * ```\n   */\n  createActivities<S extends string = \"\">(scope?: S): GoogleGenAIThreadOps<S>;\n\n  /**\n   * Identity wrapper that types a tool handler for this adapter.\n   * Constrains `toolResponse` to {@link GoogleGenAIToolResponse}.\n   */\n  wrapHandler<TArgs, TResult, TContext extends RouterContext = RouterContext>(\n    handler: (\n      args: TArgs,\n      context: TContext\n    ) => Promise<ToolHandlerResponse<TResult, GoogleGenAIToolResponse>>\n  ): ActivityToolHandler<TArgs, TResult, TContext, GoogleGenAIToolResponse>;\n}\n\n/**\n * Creates a Google GenAI adapter that bundles thread operations and model\n * invocation using the `@google/genai` SDK.\n *\n * Use `createActivities(scope)` to register scoped thread operations as\n * Temporal activities on the worker. The `invoker` (or invokers created via\n * `createModelInvoker`) should be wrapped with `createRunAgentActivity`.\n *\n * @example\n * ```typescript\n * import { createGoogleGenAIAdapter } from 'zeitlich/adapters/thread/google-genai';\n * import { createRunAgentActivity } from 'zeitlich';\n * import { GoogleGenAI } from '@google/genai';\n *\n * const client = new GoogleGenAI({ apiKey: process.env.GEMINI_API_KEY });\n * const adapter = createGoogleGenAIAdapter({ redis, client, model: 'gemini-2.5-flash' });\n *\n * export function createActivities(temporalClient: WorkflowClient) {\n *   return {\n *     ...adapter.createActivities(\"codingAgent\"),\n *     ...createRunAgentActivity(temporalClient, adapter.invoker, \"codingAgent\"),\n *   };\n * }\n * ```\n *\n * @example Multi-agent worker (main + subagent share the adapter)\n * ```typescript\n * export function createActivities(temporalClient: WorkflowClient) {\n *   return {\n *     ...adapter.createActivities(\"codingAgent\"),\n *     ...adapter.createActivities(\"researchAgent\"),\n *     ...createRunAgentActivity(temporalClient, adapter.invoker, \"codingAgent\"),\n *     ...createRunAgentActivity(\n *       temporalClient,\n *       adapter.createModelInvoker('gemini-2.5-pro'),\n *       \"researchAgent\",\n *     ),\n *   };\n * }\n * ```\n */\nexport function createGoogleGenAIAdapter(\n  config: GoogleGenAIAdapterConfig\n): GoogleGenAIAdapter {\n  const { redis } = config;\n\n  const threadOps: ThreadOps<GoogleGenAIContent> = {\n    async initializeThread(\n      threadId: string,\n      threadKey?: string\n    ): Promise<void> {\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId,\n        key: threadKey,\n      });\n      await thread.initialize();\n    },\n\n    async appendHumanMessage(\n      threadId: string,\n      id: string,\n      content: GoogleGenAIContent,\n      threadKey?: string\n    ): Promise<void> {\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId,\n        key: threadKey,\n      });\n      await thread.appendUserMessage(id, content);\n    },\n\n    async appendSystemMessage(\n      threadId: string,\n      id: string,\n      content: GoogleGenAISystemContent,\n      threadKey?: string\n    ): Promise<void> {\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId,\n        key: threadKey,\n      });\n      await thread.appendSystemMessage(id, content);\n    },\n\n    async appendToolResult(id: string, cfg: ToolResultConfig): Promise<void> {\n      const { threadId, threadKey, toolCallId, toolName, content } = cfg;\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId,\n        key: threadKey,\n      });\n      await thread.appendToolResult(\n        id,\n        toolCallId,\n        toolName,\n        content as GoogleGenAIToolResponse\n      );\n    },\n\n    async appendAgentMessage(\n      threadId: string,\n      id: string,\n      message: Content,\n      threadKey?: string\n    ): Promise<void> {\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId,\n        key: threadKey,\n      });\n      await thread.appendModelContent(id, message.parts ?? []);\n    },\n\n    async forkThread(\n      sourceThreadId: string,\n      targetThreadId: string,\n      threadKey?: string\n    ): Promise<void> {\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId: sourceThreadId,\n        key: threadKey,\n      });\n      await thread.fork(targetThreadId);\n    },\n\n    async truncateThread(\n      threadId: string,\n      length: number,\n      threadKey?: string,\n    ): Promise<void> {\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId,\n        key: threadKey,\n      });\n      await thread.truncate(length);\n    },\n  };\n\n  function createActivities<S extends string = \"\">(\n    scope?: S\n  ): GoogleGenAIThreadOps<S> {\n    const prefix = scope\n      ? `${ADAPTER_PREFIX}${scope.charAt(0).toUpperCase()}${scope.slice(1)}`\n      : ADAPTER_PREFIX;\n    const cap = (s: string): string => s.charAt(0).toUpperCase() + s.slice(1);\n    return Object.fromEntries(\n      Object.entries(threadOps).map(([k, v]) => [`${prefix}${cap(k)}`, v])\n    ) as GoogleGenAIThreadOps<S>;\n  }\n\n  const makeInvoker = (\n    model: string,\n    client: GoogleGenAI\n  ): ModelInvoker<Content> =>\n    createGoogleGenAIModelInvoker({\n      redis,\n      client,\n      model,\n      hooks: config.hooks,\n    });\n\n  const invoker: ModelInvoker<Content> =\n    config.model && config.client\n      ? makeInvoker(config.model, config.client)\n      : ((() => {\n          throw new Error(\n            \"No default model provided to createGoogleGenAIAdapter. \" +\n              \"Either pass `model` in the config or use `createModelInvoker(model)` instead.\"\n          );\n        }) as unknown as ModelInvoker<Content>);\n\n  return {\n    createActivities,\n    invoker,\n    createModelInvoker: makeInvoker,\n    wrapHandler: (handler) => handler,\n  };\n}\n"]}
+{"version":3,"sources":["../../../../src/adapters/thread/google-genai/adapter-id.ts","../../../../src/lib/thread/keys.ts","../../../../src/lib/thread/manager.ts","../../../../src/adapters/thread/google-genai/thread-manager.ts","../../../../src/lib/activity.ts","../../../../src/adapters/thread/google-genai/model-invoker.ts","../../../../src/adapters/thread/google-genai/activities.ts"],"names":["Context","invokeGoogleGenAIModel"],"mappings":";;;;;AAYO,IAAM,UAAA,GAAa;;;ACsBnB,IAAM,kBAAA,GAAqB,EAAA,GAAK,EAAA,GAAK,EAAA,GAAK,EAAA;AAc1C,SAAS,gBAAA,CACd,WACA,QAAA,EACQ;AACR,EAAA,OAAO,CAAA,EAAG,SAAS,CAAA,QAAA,EAAW,QAAQ,CAAA,CAAA;AACxC;AAeO,SAAS,gBAAA,CACd,WACA,QAAA,EACQ;AACR,EAAA,OAAO,CAAA,EAAG,SAAS,CAAA,aAAA,EAAgB,QAAQ,CAAA,CAAA;AAC7C;AAeO,SAAS,iBAAA,CACd,WACA,QAAA,EACQ;AACR,EAAA,OAAO,CAAA,EAAG,SAAS,CAAA,cAAA,EAAiB,QAAQ,CAAA,CAAA;AAC9C;;;AC3EA,IAAM,wBAAA,GAA2B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,CAAA;AAYjC,SAAS,WAAA,CAAY,UAAkB,EAAA,EAAoB;AACzD,EAAA,OAAO,CAAA,MAAA,EAAS,EAAE,CAAA,QAAA,EAAW,QAAQ,CAAA,CAAA;AACvC;AAMO,SAAS,oBACd,MAAA,EACsB;AACtB,EAAA,MAAM;AAAA,IACJ,KAAA;AAAA,IACA,QAAA;AAAA,IACA,GAAA,GAAM,UAAA;AAAA,IACN,SAAA,GAAY,CAAC,CAAA,KAAiB,IAAA,CAAK,UAAU,CAAC,CAAA;AAAA,IAC9C,WAAA,GAAc,CAAC,GAAA,KAAmB,IAAA,CAAK,MAAM,GAAG,CAAA;AAAA,IAChD;AAAA,GACF,GAAI,MAAA;AACJ,EAAA,MAAM,QAAA,GAAW,gBAAA,CAAiB,GAAA,EAAK,QAAQ,CAAA;AAC/C,EAAA,MAAM,OAAA,GAAU,gBAAA,CAAiB,GAAA,EAAK,QAAQ,CAAA;AAC9C,EAAA,MAAM,QAAA,GAAW,iBAAA,CAAkB,GAAA,EAAK,QAAQ,CAAA;AAEhD,EAAA,eAAe,kBAAA,GAAoC;AACjD,IAAA,MAAM,MAAA,GAAS,MAAM,KAAA,CAAM,MAAA,CAAO,OAAO,CAAA;AACzC,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,QAAA,EAAW,QAAQ,CAAA,QAAA,EAAW,GAAG,CAAA,gBAAA,CAAkB,CAAA;AAAA,IACrE;AAAA,EACF;AAEA,EAAA,OAAO;AAAA,IACL,MAAM,UAAA,GAA4B;AAChC,MAAA,MAAM,KAAA,CAAM,IAAI,QAAQ,CAAA;AACxB,MAAA,MAAM,KAAA,CAAM,GAAA,CAAI,OAAA,EAAS,GAAA,EAAK,MAAM,kBAAkB,CAAA;AAAA,IACxD,CAAA;AAAA,IAEA,MAAM,IAAA,GAAqB;AACzB,MAAA,MAAM,kBAAA,EAAmB;AACzB,MAAA,MAAM,OAAO,MAAM,KAAA,CAAM,MAAA,CAAO,QAAA,EAAU,GAAG,EAAE,CAAA;AAC/C,MAAA,OAAO,IAAA,CAAK,IAAI,WAAW,CAAA;AAAA,IAC7B,CAAA;AAAA,IAEA,MAAM,OAAO,QAAA,EAA8B;AACzC,MAAA,IAAI,QAAA,CAAS,WAAW,CAAA,EAAG;AAC3B,MAAA,MAAM,kBAAA,EAAmB;AAEzB,MAAA,IAAI,IAAA,EAAM;AACR,QAAA,MAAM,UAAU,QAAA,CAAS,GAAA,CAAI,IAAI,CAAA,CAAE,KAAK,GAAG,CAAA;AAC3C,QAAA,MAAM,QAAA,GAAW,WAAA,CAAY,QAAA,EAAU,OAAO,CAAA;AAC9C,QAAA,MAAM,KAAA,CAAM,IAAA;AAAA,UACV,wBAAA;AAAA,UACA,CAAA;AAAA,UACA,QAAA;AAAA,UACA,QAAA;AAAA,UACA,OAAO,kBAAkB,CAAA;AAAA,UACzB,GAAG,QAAA,CAAS,GAAA,CAAI,SAAS;AAAA,SAC3B;AAAA,MACF,CAAA,MAAO;AACL,QAAA,MAAM,MAAM,KAAA,CAAM,QAAA,EAAU,GAAG,QAAA,CAAS,GAAA,CAAI,SAAS,CAAC,CAAA;AACtD,QAAA,MAAM,KAAA,CAAM,MAAA,CAAO,QAAA,EAAU,kBAAkB,CAAA;AAAA,MACjD;AAAA,IACF,CAAA;AAAA,IAEA,MAAM,KAAK,WAAA,EAAoD;AAC7D,MAAA,MAAM,kBAAA,EAAmB;AACzB,MAAA,MAAM,OAAO,MAAM,KAAA,CAAM,MAAA,CAAO,QAAA,EAAU,GAAG,EAAE,CAAA;AAC/C,MAAA,MAAM,QAAA,GAAW,MAAM,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA;AACzC,MAAA,MAAM,SAAS,mBAAA,CAAoB;AAAA,QACjC,GAAG,MAAA;AAAA,QACH,QAAA,EAAU;AAAA,OACX,CAAA;AACD,MAAA,MAAM,OAAO,UAAA,EAAW;AACxB,MAAA,IAAI,IAAA,CAAK,SAAS,CAAA,EAAG;AACnB,QAAA,MAAM,MAAA,GAAS,gBAAA,CAAiB,GAAA,EAAK,WAAW,CAAA;AAChD,QAAA,MAAM,KAAA,CAAM,KAAA,CAAM,MAAA,EAAQ,GAAG,IAAI,CAAA;AACjC,QAAA,MAAM,KAAA,CAAM,MAAA,CAAO,MAAA,EAAQ,kBAAkB,CAAA;AAAA,MAC/C;AACA,MAAA,IAAI,YAAY,IAAA,EAAM;AACpB,QAAA,MAAM,WAAA,GAAc,iBAAA,CAAkB,GAAA,EAAK,WAAW,CAAA;AACtD,QAAA,MAAM,KAAA,CAAM,GAAA,CAAI,WAAA,EAAa,QAAA,EAAU,MAAM,kBAAkB,CAAA;AAAA,MACjE;AACA,MAAA,OAAO,MAAA;AAAA,IACT,CAAA;AAAA,IAEA,MAAM,WAAW,QAAA,EAA8B;AAC7C,MAAA,MAAM,kBAAA,EAAmB;AACzB,MAAA,IAAI,CAAC,IAAA,EAAM;AACT,QAAA,MAAM,IAAI,KAAA;AAAA,UACR;AAAA,SACF;AAAA,MACF;AACA,MAAA,MAAM,WAAW,MAAM,KAAA,CAAM,MAAA,CAAO,QAAA,EAAU,GAAG,EAAE,CAAA;AACnD,MAAA,MAAM,cAAc,QAAA,CACjB,GAAA,CAAI,CAAC,GAAA,KAAQ,KAAK,WAAA,CAAY,GAAG,CAAC,CAAC,EACnC,MAAA,CAAO,CAAC,EAAA,KAAqB,OAAO,OAAO,QAAQ,CAAA;AACtD,MAAA,MAAM,KAAA,CAAM,IAAI,QAAQ,CAAA;AACxB,MAAA,IAAI,WAAA,CAAY,SAAS,CAAA,EAAG;AAC1B,QAAA,MAAM,KAAA,CAAM,GAAA;AAAA,UACV,GAAG,YAAY,GAAA,CAAI,CAAC,OAAO,WAAA,CAAY,QAAA,EAAU,EAAE,CAAC;AAAA,SACtD;AAAA,MACF;AACA,MAAA,IAAI,QAAA,CAAS,SAAS,CAAA,EAAG;AACvB,QAAA,MAAM,MAAM,KAAA,CAAM,QAAA,EAAU,GAAG,QAAA,CAAS,GAAA,CAAI,SAAS,CAAC,CAAA;AACtD,QAAA,MAAM,KAAA,CAAM,MAAA,CAAO,QAAA,EAAU,kBAAkB,CAAA;AAAA,MACjD;AACA,MAAA,MAAM,KAAA,CAAM,MAAA,CAAO,OAAA,EAAS,kBAAkB,CAAA;AAAA,IAChD,CAAA;AAAA,IAEA,MAAM,MAAA,GAAwB;AAC5B,MAAA,MAAM,KAAA,CAAM,GAAA,CAAI,QAAA,EAAU,OAAA,EAAS,QAAQ,CAAA;AAAA,IAC7C,CAAA;AAAA,IAEA,MAAM,SAAA,GAAkD;AACtD,MAAA,MAAM,GAAA,GAAM,MAAM,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA;AACpC,MAAA,IAAI,GAAA,IAAO,MAAM,OAAO,IAAA;AACxB,MAAA,OAAO,IAAA,CAAK,MAAM,GAAG,CAAA;AAAA,IACvB,CAAA;AAAA,IAEA,MAAM,UAAU,KAAA,EAA4C;AAC1D,MAAA,MAAM,kBAAA,EAAmB;AACzB,MAAA,MAAM,KAAA,CAAM,GAAA;AAAA,QACV,QAAA;AAAA,QACA,IAAA,CAAK,UAAU,KAAK,CAAA;AAAA,QACpB,IAAA;AAAA,QACA;AAAA,OACF;AAAA,IACF,CAAA;AAAA,IAEA,MAAM,WAAA,GAA6B;AACjC,MAAA,MAAM,KAAA,CAAM,IAAI,QAAQ,CAAA;AAAA,IAC1B,CAAA;AAAA,IAEA,MAAM,MAAA,GAA0B;AAC9B,MAAA,MAAM,kBAAA,EAAmB;AACzB,MAAA,OAAO,KAAA,CAAM,KAAK,QAAQ,CAAA;AAAA,IAC5B,CAAA;AAAA,IAEA,MAAM,eAAe,SAAA,EAAkC;AACrD,MAAA,MAAM,kBAAA,EAAmB;AACzB,MAAA,IAAI,CAAC,IAAA,EAAM;AACT,QAAA,MAAM,IAAI,KAAA;AAAA,UACR;AAAA,SACF;AAAA,MACF;AACA,MAAA,MAAM,OAAO,MAAM,KAAA,CAAM,MAAA,CAAO,QAAA,EAAU,GAAG,EAAE,CAAA;AAC/C,MAAA,IAAI,GAAA,GAAM,EAAA;AACV,MAAA,MAAM,aAAuB,EAAC;AAC9B,MAAA,KAAA,IAAS,CAAA,GAAI,CAAA,EAAG,CAAA,GAAI,IAAA,CAAK,QAAQ,CAAA,EAAA,EAAK;AACpC,QAAA,MAAM,GAAA,GAAM,KAAK,CAAC,CAAA;AAClB,QAAA,IAAI,QAAQ,MAAA,EAAW;AACvB,QAAA,MAAM,EAAA,GAAK,IAAA,CAAK,WAAA,CAAY,GAAG,CAAC,CAAA;AAChC,QAAA,IAAI,GAAA,KAAQ,EAAA,IAAM,EAAA,KAAO,SAAA,EAAW,GAAA,GAAM,CAAA;AAC1C,QAAA,IAAI,GAAA,KAAQ,EAAA,EAAI,UAAA,CAAW,IAAA,CAAK,EAAE,CAAA;AAAA,MACpC;AACA,MAAA,IAAI,QAAQ,EAAA,EAAI;AAChB,MAAA,IAAI,QAAQ,CAAA,EAAG;AACb,QAAA,MAAM,KAAA,CAAM,IAAI,QAAQ,CAAA;AACxB,QAAA,MAAM,KAAA,CAAM,MAAA,CAAO,OAAA,EAAS,kBAAkB,CAAA;AAAA,MAChD,CAAA,MAAO;AACL,QAAA,MAAM,KAAA,CAAM,KAAA,CAAM,QAAA,EAAU,CAAA,EAAG,MAAM,CAAC,CAAA;AACtC,QAAA,MAAM,KAAA,CAAM,MAAA,CAAO,QAAA,EAAU,kBAAkB,CAAA;AAAA,MACjD;AAKA,MAAA,IAAI,UAAA,CAAW,SAAS,CAAA,EAAG;AACzB,QAAA,MAAM,KAAA,CAAM,GAAA;AAAA,UACV,GAAG,WAAW,GAAA,CAAI,CAAC,OAAO,WAAA,CAAY,QAAA,EAAU,EAAE,CAAC;AAAA,SACrD;AAAA,MACF;AAAA,IACF;AAAA,GACF;AACF;;;ACvJA,SAAS,gBAAgB,GAAA,EAA4B;AACnD,EAAA,OAAO,GAAA,CAAI,EAAA;AACb;AAGA,SAAS,QAAQ,OAAA,EAAqC;AACpD,EAAA,IAAI,OAAO,YAAY,QAAA,EAAU;AAC/B,IAAA,OAAO,CAAC,EAAE,IAAA,EAAM,OAAA,EAAS,CAAA;AAAA,EAC3B;AACA,EAAA,OAAO,OAAA;AACT;AAGA,SAAS,mBACP,OAAA,EACyB;AACzB,EAAA,IAAI,OAAO,YAAY,QAAA,EAAU;AAC/B,IAAA,OAAO,OAAA;AAAA,EACT;AACA,EAAA,OAAO,EAAE,QAAQ,OAAA,EAAQ;AAC3B;AAOA,SAAS,yBAAyB,QAAA,EAAgC;AAChE,EAAA,MAAM,SAAoB,EAAC;AAC3B,EAAA,KAAA,MAAW,WAAW,QAAA,EAAU;AAC9B,IAAA,MAAM,IAAA,GAAO,MAAA,CAAO,MAAA,CAAO,MAAA,GAAS,CAAC,CAAA;AACrC,IAAA,IAAI,IAAA,IAAQ,IAAA,CAAK,IAAA,KAAS,OAAA,CAAQ,IAAA,EAAM;AACtC,MAAA,IAAA,CAAK,KAAA,GAAQ,CAAC,GAAI,IAAA,CAAK,KAAA,IAAS,EAAC,EAAI,GAAI,OAAA,CAAQ,KAAA,IAAS,EAAG,CAAA;AAAA,IAC/D,CAAA,MAAO;AACL,MAAA,MAAA,CAAO,IAAA,CAAK,EAAE,GAAG,OAAA,EAAS,KAAA,EAAO,CAAC,GAAI,OAAA,CAAQ,KAAA,IAAS,EAAG,CAAA,EAAG,CAAA;AAAA,IAC/D;AAAA,EACF;AACA,EAAA,OAAO,MAAA;AACT;AAOO,SAAS,+BACd,MAAA,EAC0B;AAC1B,EAAA,MAAM,UAAA,GAAiD;AAAA,IACrD,OAAO,MAAA,CAAO,KAAA;AAAA,IACd,UAAU,MAAA,CAAO,QAAA;AAAA,IACjB,KAAK,MAAA,CAAO,GAAA;AAAA,IACZ,IAAA,EAAM;AAAA,GACR;AAEA,EAAA,MAAM,IAAA,GAAO,oBAAoB,UAAU,CAAA;AAE3C,EAAA,MAAM,OAAA,GAA6D;AAAA,IACjE,MAAM,iBAAA,CACJ,EAAA,EACA,OAAA,EACe;AACf,MAAA,MAAM,KAAK,MAAA,CAAO;AAAA,QAChB;AAAA,UACE,EAAA;AAAA,UACA,SAAS,EAAE,IAAA,EAAM,QAAQ,KAAA,EAAO,OAAA,CAAQ,OAAO,CAAA;AAAE;AACnD,OACD,CAAA;AAAA,IACH,CAAA;AAAA,IAEA,MAAM,mBAAA,CACJ,EAAA,EACA,OAAA,EACe;AACf,MAAA,MAAM,KAAA,GACJ,OAAO,OAAA,KAAY,QAAA,GAAW,CAAC,EAAE,IAAA,EAAM,OAAA,EAAS,CAAA,GAAI,OAAA;AACtD,MAAA,MAAM,KAAK,UAAA,EAAW;AACtB,MAAA,MAAM,KAAK,MAAA,CAAO;AAAA,QAChB;AAAA,UACE,EAAA;AAAA,UACA,OAAA,EAAS,EAAE,IAAA,EAAM,QAAA,EAAU,KAAA;AAAM;AACnC,OACD,CAAA;AAAA,IACH,CAAA;AAAA,IAEA,MAAM,kBAAA,CAAmB,EAAA,EAAY,KAAA,EAA8B;AACjE,MAAA,MAAM,KAAK,MAAA,CAAO;AAAA,QAChB;AAAA,UACE,EAAA;AAAA,UACA,OAAA,EAAS,EAAE,IAAA,EAAM,OAAA,EAAS,KAAA;AAAM;AAClC,OACD,CAAA;AAAA,IACH,CAAA;AAAA,IAEA,MAAM,gBAAA,CACJ,EAAA,EACA,UAAA,EACA,UACA,OAAA,EACe;AACf,MAAA,MAAM,KAAA,GAAgB,KAAA,CAAM,OAAA,CAAQ,OAAO,IACtC,OAAA,GACD;AAAA,QACE;AAAA,UACE,gBAAA,EAAkB;AAAA,YAChB,EAAA,EAAI,UAAA;AAAA,YACJ,IAAA,EAAM,QAAA;AAAA,YACN,QAAA,EAAU,mBAAmB,OAAO;AAAA;AACtC;AACF,OACF;AAEJ,MAAA,MAAM,KAAK,MAAA,CAAO;AAAA,QAChB;AAAA,UACE,EAAA;AAAA,UACA,OAAA,EAAS,EAAE,IAAA,EAAM,MAAA,EAAQ,KAAA;AAAM;AACjC,OACD,CAAA;AAAA,IACH,CAAA;AAAA,IAEA,MAAM,oBAAA,GAA8D;AAClE,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,IAAA,EAAK;AAC/B,MAAA,MAAM,EAAE,gBAAA,EAAkB,iBAAA,EAAkB,GAAI,MAAA,CAAO,SAAS,EAAC;AACjE,MAAA,MAAM,MAAA,GAAS,gBAAA,GACX,MAAA,CAAO,GAAA,CAAI,CAAC,GAAA,EAAK,CAAA,KAAM,gBAAA,CAAiB,GAAA,EAAK,CAAA,EAAG,MAAM,CAAC,CAAA,GACvD,MAAA;AAEJ,MAAA,IAAI,iBAAA;AACJ,MAAA,MAAM,uBAAkC,EAAC;AAEzC,MAAA,KAAA,MAAW,QAAQ,MAAA,EAAQ;AACzB,QAAA,IAAI,IAAA,CAAK,OAAA,CAAQ,IAAA,KAAS,QAAA,EAAU;AAClC,UAAA,iBAAA,GAAoB,IAAA,CAAK,OAAA,CAAQ,KAAA,IAAS,EAAC;AAAA,QAC7C,CAAA,MAAO;AACL,UAAA,oBAAA,CAAqB,IAAA,CAAK,KAAK,OAAO,CAAA;AAAA,QACxC;AAAA,MACF;AAEA,MAAA,MAAM,QAAA,GAAW,yBAAyB,oBAAoB,CAAA;AAC9D,MAAA,OAAO;AAAA,QACL,QAAA,EAAU,iBAAA,GACN,QAAA,CAAS,GAAA,CAAI,CAAC,GAAA,EAAK,CAAA,KAAM,iBAAA,CAAkB,GAAA,EAAK,CAAA,EAAG,QAAQ,CAAC,CAAA,GAC5D,QAAA;AAAA,QACJ,GAAI,qBAAqB,iBAAA,CAAkB,MAAA,GAAS,IAChD,EAAE,iBAAA,KACF;AAAC,OACP;AAAA,IACF;AAAA,GACF;AAEA,EAAA,MAAM,OAAA,GAAU,MAAA,CAAO,MAAA,CAAO,IAAA,EAAM,OAAO,CAAA;AAE3C,EAAA,MAAM,YAAA,GAAe,OAAA,CAAQ,IAAA,CAAK,IAAA,CAAK,OAAO,CAAA;AAC9C,EAAA,OAAA,CAAQ,IAAA,GAAO,OACb,WAAA,KACsC;AACtC,IAAA,MAAM,aAAa,WAAW,CAAA;AAC9B,IAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,MAC5C,GAAG,MAAA;AAAA,MACH,QAAA,EAAU;AAAA,KACX,CAAA;AACD,IAAA,MAAM,EAAE,mBAAA,EAAqB,eAAA,EAAgB,GAAI,MAAA,CAAO,SAAS,EAAC;AAClE,IAAA,IAAI,CAAC,mBAAA,IAAuB,CAAC,eAAA,EAAiB;AAC5C,MAAA,OAAO,MAAA;AAAA,IACT;AACA,IAAA,IAAI,IAAA,GAAO,MAAM,MAAA,CAAO,IAAA,EAAK;AAC7B,IAAA,IAAI,mBAAA,EAAqB;AACvB,MAAA,IAAA,GAAO,MAAM,oBAAoB,IAAI,CAAA;AAAA,IACvC;AACA,IAAA,IAAI,eAAA,EAAiB;AACnB,MAAA,IAAA,GAAO,IAAA,CAAK,IAAI,CAAC,GAAA,EAAK,MAAM,eAAA,CAAgB,GAAA,EAAK,CAAA,EAAG,IAAI,CAAC,CAAA;AAAA,IAC3D;AACA,IAAA,MAAM,MAAA,CAAO,WAAW,IAAI,CAAA;AAC5B,IAAA,OAAO,MAAA;AAAA,EACT,CAAA;AAEA,EAAA,OAAO,OAAA;AACT;ACvNO,SAAS,kBAAA,GAGd;AACA,EAAA,IAAI;AACF,IAAA,MAAM,GAAA,GAAMA,iBAAQ,OAAA,EAAQ;AAC5B,IAAA,OAAO,EAAE,WAAW,MAAM,GAAA,CAAI,WAAU,EAAG,MAAA,EAAQ,IAAI,kBAAA,EAAmB;AAAA,EAC5E,CAAA,CAAA,MAAQ;AACN,IAAA,OAAO,EAAC;AAAA,EACV;AACF;;;ACDA,SAAS,uBACP,KAAA,EACuB;AACvB,EAAA,OAAO,KAAA,CAAM,GAAA,CAAI,CAAC,CAAA,MAAO;AAAA,IACvB,MAAM,CAAA,CAAE,IAAA;AAAA,IACR,aAAa,CAAA,CAAE,WAAA;AAAA,IACf,sBAAsB,CAAA,CAAE;AAAA,GAC1B,CAAE,CAAA;AACJ;AA0BO,SAAS,6BAAA,CAA8B;AAAA,EAC5C,KAAA;AAAA,EACA,MAAA;AAAA,EACA,KAAA;AAAA,EACA;AACF,CAAA,EAAkC;AAChC,EAAA,OAAO,eAAeC,wBACpB,MAAA,EACiC;AACjC,IAAA,MAAM,EAAE,QAAA,EAAU,SAAA,EAAW,KAAA,EAAO,oBAAmB,GAAI,MAAA;AAC3D,IAAA,MAAM,EAAE,SAAA,EAAW,MAAA,EAAO,GAAI,kBAAA,EAAmB;AAEjD,IAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,MAC5C,KAAA;AAAA,MACA,QAAA;AAAA,MACA,GAAA,EAAK,SAAA;AAAA,MACL;AAAA,KACD,CAAA;AAKD,IAAA,MAAM,MAAA,CAAO,eAAe,kBAAkB,CAAA;AAC9C,IAAA,MAAM,EAAE,QAAA,EAAU,iBAAA,EAAkB,GAClC,MAAM,OAAO,oBAAA,EAAqB;AAEpC,IAAA,MAAM,oBAAA,GAAuB,sBAAA,CAAuB,KAAA,CAAM,KAAK,CAAA;AAC/D,IAAA,MAAM,KAAA,GACJ,qBAAqB,MAAA,GAAS,CAAA,GAAI,CAAC,EAAE,oBAAA,EAAsB,CAAA,GAAI,MAAA;AAEjE,IAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,MAAA,CAAO,qBAAA,CAAsB;AAAA,MACvD,KAAA;AAAA,MACA,QAAA;AAAA,MACA,MAAA,EAAQ;AAAA,QACN,GAAI,iBAAA,GAAoB,EAAE,iBAAA,KAAsB,EAAC;AAAA,QACjD,GAAI,KAAA,GAAQ,EAAE,KAAA,KAAU,EAAC;AAAA,QACzB,WAAA,EAAa;AAAA;AACf,KACD,CAAA;AAED,IAAA,MAAM,WAAmB,EAAC;AAC1B,IAAA,IAAI,SAAA;AACJ,IAAA,WAAA,MAAiB,SAAS,MAAA,EAAQ;AAChC,MAAA,SAAA,GAAY,KAAA;AACZ,MAAA,QAAA,CAAS,IAAA,CAAK,GAAI,KAAA,CAAM,UAAA,GAAa,CAAC,CAAA,EAAG,OAAA,EAAS,KAAA,IAAS,EAAG,CAAA;AAC9D,MAAA,SAAA,IAAY;AAAA,IACd;AAEA,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,MAAM,IAAI,MAAM,wDAAwD,CAAA;AAAA,IAC1E;AAEA,IAAA,MAAM,YAAA,GAAwB,EAAE,IAAA,EAAM,OAAA,EAAS,OAAO,QAAA,EAAS;AAC/D,IAAA,MAAM,aAAA,GAAgB,SAAA,CAAU,aAAA,IAAiB,EAAC;AAElD,IAAA,OAAO;AAAA,MACL,OAAA,EAAS,YAAA;AAAA,MACT,YAAA,EAAc,aAAA,CAAc,GAAA,CAAI,CAAC,EAAA,MAAQ;AAAA,QACvC,IAAI,EAAA,CAAG,EAAA;AAAA,QACP,IAAA,EAAM,GAAG,IAAA,IAAQ,EAAA;AAAA,QACjB,IAAA,EAAM,EAAA,CAAG,IAAA,IAAQ;AAAC,OACpB,CAAE,CAAA;AAAA,MACF,KAAA,EAAO;AAAA,QACL,WAAA,EAAa,UAAU,aAAA,EAAe,gBAAA;AAAA,QACtC,YAAA,EAAc,UAAU,aAAA,EAAe,oBAAA;AAAA,QACvC,gBAAA,EAAkB,UAAU,aAAA,EAAe;AAAA;AAC7C,KACF;AAAA,EACF,CAAA;AACF;AAOA,eAAsB,sBAAA,CAAuB;AAAA,EAC3C,KAAA;AAAA,EACA,MAAA;AAAA,EACA,KAAA;AAAA,EACA,KAAA;AAAA,EACA;AACF,CAAA,EAMoC;AAClC,EAAA,MAAM,UAAU,6BAAA,CAA8B;AAAA,IAC5C,KAAA;AAAA,IACA,MAAA;AAAA,IACA,KAAA;AAAA,IACA;AAAA,GACD,CAAA;AACD,EAAA,OAAO,QAAQ,MAAM,CAAA;AACvB;;;AChBO,SAAS,yBACd,MAAA,EACoB;AACpB,EAAA,MAAM,EAAE,OAAM,GAAI,MAAA;AAElB,EAAA,MAAM,SAAA,GAA2C;AAAA,IAC/C,MAAM,gBAAA,CACJ,QAAA,EACA,SAAA,EACe;AACf,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA;AAAA,QACA,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,MAAM,OAAO,UAAA,EAAW;AAAA,IAC1B,CAAA;AAAA,IAEA,MAAM,kBAAA,CACJ,QAAA,EACA,EAAA,EACA,SACA,SAAA,EACe;AACf,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA;AAAA,QACA,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,MAAM,MAAA,CAAO,iBAAA,CAAkB,EAAA,EAAI,OAAO,CAAA;AAAA,IAC5C,CAAA;AAAA,IAEA,MAAM,mBAAA,CACJ,QAAA,EACA,EAAA,EACA,SACA,SAAA,EACe;AACf,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA;AAAA,QACA,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,MAAM,MAAA,CAAO,mBAAA,CAAoB,EAAA,EAAI,OAAO,CAAA;AAAA,IAC9C,CAAA;AAAA,IAEA,MAAM,gBAAA,CAAiB,EAAA,EAAY,GAAA,EAAsC;AACvE,MAAA,MAAM,EAAE,QAAA,EAAU,SAAA,EAAW,UAAA,EAAY,QAAA,EAAU,SAAQ,GAAI,GAAA;AAC/D,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA;AAAA,QACA,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,MAAM,MAAA,CAAO,gBAAA;AAAA,QACX,EAAA;AAAA,QACA,UAAA;AAAA,QACA,QAAA;AAAA,QACA;AAAA,OACF;AAAA,IACF,CAAA;AAAA,IAEA,MAAM,kBAAA,CACJ,QAAA,EACA,EAAA,EACA,SACA,SAAA,EACe;AACf,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA;AAAA,QACA,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,MAAM,OAAO,kBAAA,CAAmB,EAAA,EAAI,OAAA,CAAQ,KAAA,IAAS,EAAE,CAAA;AAAA,IACzD,CAAA;AAAA,IAEA,MAAM,UAAA,CACJ,cAAA,EACA,cAAA,EACA,SAAA,EACe;AACf,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA,EAAU,cAAA;AAAA,QACV,GAAA,EAAK,SAAA;AAAA,QACL,OAAO,MAAA,CAAO;AAAA,OACf,CAAA;AACD,MAAA,MAAM,MAAA,CAAO,KAAK,cAAc,CAAA;AAAA,IAClC,CAAA;AAAA,IAEA,MAAM,cAAA,CACJ,QAAA,EACA,SAAA,EACA,SAAA,EACe;AACf,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA;AAAA,QACA,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,MAAM,MAAA,CAAO,eAAe,SAAS,CAAA;AAAA,IACvC,CAAA;AAAA,IAEA,MAAM,eAAA,CACJ,QAAA,EACA,SAAA,EACsC;AACtC,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA;AAAA,QACA,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,OAAO,OAAO,SAAA,EAAU;AAAA,IAC1B,CAAA;AAAA,IAEA,MAAM,eAAA,CACJ,QAAA,EACA,KAAA,EACA,SAAA,EACe;AACf,MAAA,MAAM,SAAS,8BAAA,CAA+B;AAAA,QAC5C,KAAA;AAAA,QACA,QAAA;AAAA,QACA,GAAA,EAAK;AAAA,OACN,CAAA;AACD,MAAA,MAAM,MAAA,CAAO,UAAU,KAAK,CAAA;AAAA,IAC9B;AAAA,GACF;AAEA,EAAA,SAAS,iBACP,KAAA,EACyB;AACzB,IAAA,MAAM,SAAS,KAAA,GACX,CAAA,EAAG,UAAU,CAAA,EAAG,MAAM,MAAA,CAAO,CAAC,CAAA,CAAE,WAAA,EAAa,CAAA,EAAG,KAAA,CAAM,KAAA,CAAM,CAAC,CAAC,CAAA,CAAA,GAC9D,UAAA;AACJ,IAAA,MAAM,GAAA,GAAM,CAAC,CAAA,KAAsB,CAAA,CAAE,MAAA,CAAO,CAAC,CAAA,CAAE,WAAA,EAAY,GAAI,CAAA,CAAE,KAAA,CAAM,CAAC,CAAA;AACxE,IAAA,OAAO,MAAA,CAAO,WAAA;AAAA,MACZ,OAAO,OAAA,CAAQ,SAAS,EAAE,GAAA,CAAI,CAAC,CAAC,CAAA,EAAG,CAAC,MAAM,CAAC,CAAA,EAAG,MAAM,CAAA,EAAG,GAAA,CAAI,CAAC,CAAC,CAAA,CAAA,EAAI,CAAC,CAAC;AAAA,KACrE;AAAA,EACF;AAEA,EAAA,MAAM,WAAA,GAAc,CAClB,KAAA,EACA,MAAA,KAEA,6BAAA,CAA8B;AAAA,IAC5B,KAAA;AAAA,IACA,MAAA;AAAA,IACA,KAAA;AAAA,IACA,OAAO,MAAA,CAAO;AAAA,GACf,CAAA;AAEH,EAAA,MAAM,OAAA,GACJ,MAAA,CAAO,KAAA,IAAS,MAAA,CAAO,MAAA,GACnB,WAAA,CAAY,MAAA,CAAO,KAAA,EAAO,MAAA,CAAO,MAAM,CAAA,IACrC,MAAM;AACN,IAAA,MAAM,IAAI,KAAA;AAAA,MACR;AAAA,KAEF;AAAA,EACF,CAAA,CAAA;AAEN,EAAA,OAAO;AAAA,IACL,gBAAA;AAAA,IACA,OAAA;AAAA,IACA,kBAAA,EAAoB,WAAA;AAAA,IACpB,WAAA,EAAa,CAAC,OAAA,KAAY;AAAA,GAC5B;AACF","file":"index.cjs","sourcesContent":["/**\n * Public adapter identity for the Google GenAI thread adapter.\n *\n * This value is wire format — it appears as the prefix for Temporal\n * activity names (e.g. `googleGenAICodingAgentInitializeThread`) and\n * must never change, since renaming it would orphan existing persisted\n * threads and break in-flight workflows.\n *\n * Re-exported from `zeitlich/adapters/thread/google-genai` so downstream\n * consumers can use the exact same literal the adapter uses internally,\n * typed as the narrow string literal `\"googleGenAI\"`.\n */\nexport const ADAPTER_ID = \"googleGenAI\" as const;\n\n/** Narrow string-literal type for {@link ADAPTER_ID}. */\nexport type AdapterId = typeof ADAPTER_ID;\n","/**\n * Public helpers for zeitlich's Redis thread storage layout.\n *\n * These are the exact string-building primitives zeitlich's internal thread\n * manager uses for every adapter. Downstream consumers that need to read a\n * persisted thread (for evaluation, observability, admin tooling, etc.)\n * should use these helpers rather than reconstructing the key layout by\n * hand — the layout is versioned with this module, so upgrading zeitlich\n * keeps the consumer in sync.\n *\n * The layout is adapter-agnostic: every thread adapter stores messages the\n * same way.\n *\n * @example\n * ```typescript\n * import {\n *   getThreadListKey,\n *   getThreadMetaKey,\n *   THREAD_TTL_SECONDS,\n * } from 'zeitlich';\n *\n * const listKey = getThreadListKey('messages', threadId);\n * const metaKey = getThreadMetaKey('messages', threadId);\n * const ttl = await redis.ttl(listKey); // <= THREAD_TTL_SECONDS\n * ```\n */\n\n/**\n * TTL (in seconds) applied to every thread list and thread meta key that\n * zeitlich's {@link createThreadManager} writes. Exposed so downstream\n * consumers can size their Redis retention / query windows to match.\n *\n * Current value: 90 days.\n */\nexport const THREAD_TTL_SECONDS = 60 * 60 * 24 * 90;\n\n/**\n * Build the Redis list key that holds a thread's serialized messages.\n *\n * Mirrors the exact key used internally by zeitlich's thread manager,\n * so a consumer calling `redis.lrange(getThreadListKey(key, id), 0, -1)`\n * sees the same data the writer wrote.\n *\n * @param threadKey - Thread key (defaults to `\"messages\"` inside the\n *                    thread manager, but downstream adapters may pass\n *                    their own value).\n * @param threadId  - Thread id as provided to the thread manager.\n */\nexport function getThreadListKey(\n  threadKey: string,\n  threadId: string\n): string {\n  return `${threadKey}:thread:${threadId}`;\n}\n\n/**\n * Build the Redis key that stores a thread's existence marker / metadata.\n *\n * Zeitlich treats the presence of this key as \"thread has been\n * initialized\"; append/load/fork/truncate operations fail when it is\n * missing. Consumers can use it as a cheap existence probe without\n * scanning the message list.\n *\n * @param threadKey - Thread key (defaults to `\"messages\"` inside the\n *                    thread manager, but downstream adapters may pass\n *                    their own value).\n * @param threadId  - Thread id as provided to the thread manager.\n */\nexport function getThreadMetaKey(\n  threadKey: string,\n  threadId: string\n): string {\n  return `${threadKey}:meta:thread:${threadId}`;\n}\n\n/**\n * Build the Redis key that stores a thread's persisted state slice\n * (tasks + custom state) written by zeitlich's session loop on every\n * exit path.\n *\n * Consumers can read this key with `redis.get(getThreadStateKey(key, id))`\n * and `JSON.parse` the result into a {@link PersistedThreadState}.\n *\n * @param threadKey - Thread key (defaults to `\"messages\"` inside the\n *                    thread manager, but downstream adapters may pass\n *                    their own value).\n * @param threadId  - Thread id as provided to the thread manager.\n */\nexport function getThreadStateKey(\n  threadKey: string,\n  threadId: string\n): string {\n  return `${threadKey}:state:thread:${threadId}`;\n}\n","import type { PersistedThreadState } from \"../state/types\";\nimport type { ThreadManagerConfig, BaseThreadManager } from \"./types\";\nimport {\n  THREAD_TTL_SECONDS,\n  getThreadListKey,\n  getThreadMetaKey,\n  getThreadStateKey,\n} from \"./keys\";\n\n/**\n * Lua script for atomic idempotent append.\n * Checks a dedup key; if it exists the message was already appended and we\n * return 0. Otherwise appends all messages to the list, sets TTL on both\n * the list and the dedup key, and returns 1.\n *\n * KEYS[1] = dedup key, KEYS[2] = list key\n * ARGV[1] = TTL seconds, ARGV[2..N] = serialised messages\n */\nconst APPEND_IDEMPOTENT_SCRIPT = `\nif redis.call('EXISTS', KEYS[1]) == 1 then\n  return 0\nend\nfor i = 2, #ARGV do\n  redis.call('RPUSH', KEYS[2], ARGV[i])\nend\nredis.call('EXPIRE', KEYS[2], tonumber(ARGV[1]))\nredis.call('SET', KEYS[1], '1', 'EX', tonumber(ARGV[1]))\nreturn 1\n`;\n\nfunction getDedupKey(threadId: string, id: string): string {\n  return `dedup:${id}:thread:${threadId}`;\n}\n\n/**\n * Creates a generic thread manager for handling conversation state in Redis.\n * Framework-agnostic — works with any serializable message type.\n */\nexport function createThreadManager<T>(\n  config: ThreadManagerConfig<T>\n): BaseThreadManager<T> {\n  const {\n    redis,\n    threadId,\n    key = \"messages\",\n    serialize = (m: T): string => JSON.stringify(m),\n    deserialize = (raw: string): T => JSON.parse(raw) as T,\n    idOf,\n  } = config;\n  const redisKey = getThreadListKey(key, threadId);\n  const metaKey = getThreadMetaKey(key, threadId);\n  const stateKey = getThreadStateKey(key, threadId);\n\n  async function assertThreadExists(): Promise<void> {\n    const exists = await redis.exists(metaKey);\n    if (!exists) {\n      throw new Error(`Thread \"${threadId}\" (key: ${key}) does not exist`);\n    }\n  }\n\n  return {\n    async initialize(): Promise<void> {\n      await redis.del(redisKey);\n      await redis.set(metaKey, \"1\", \"EX\", THREAD_TTL_SECONDS);\n    },\n\n    async load(): Promise<T[]> {\n      await assertThreadExists();\n      const data = await redis.lrange(redisKey, 0, -1);\n      return data.map(deserialize);\n    },\n\n    async append(messages: T[]): Promise<void> {\n      if (messages.length === 0) return;\n      await assertThreadExists();\n\n      if (idOf) {\n        const dedupId = messages.map(idOf).join(\":\");\n        const dedupKey = getDedupKey(threadId, dedupId);\n        await redis.eval(\n          APPEND_IDEMPOTENT_SCRIPT,\n          2,\n          dedupKey,\n          redisKey,\n          String(THREAD_TTL_SECONDS),\n          ...messages.map(serialize)\n        );\n      } else {\n        await redis.rpush(redisKey, ...messages.map(serialize));\n        await redis.expire(redisKey, THREAD_TTL_SECONDS);\n      }\n    },\n\n    async fork(newThreadId: string): Promise<BaseThreadManager<T>> {\n      await assertThreadExists();\n      const data = await redis.lrange(redisKey, 0, -1);\n      const stateRaw = await redis.get(stateKey);\n      const forked = createThreadManager({\n        ...config,\n        threadId: newThreadId,\n      });\n      await forked.initialize();\n      if (data.length > 0) {\n        const newKey = getThreadListKey(key, newThreadId);\n        await redis.rpush(newKey, ...data);\n        await redis.expire(newKey, THREAD_TTL_SECONDS);\n      }\n      if (stateRaw != null) {\n        const newStateKey = getThreadStateKey(key, newThreadId);\n        await redis.set(newStateKey, stateRaw, \"EX\", THREAD_TTL_SECONDS);\n      }\n      return forked;\n    },\n\n    async replaceAll(messages: T[]): Promise<void> {\n      await assertThreadExists();\n      if (!idOf) {\n        throw new Error(\n          \"replaceAll requires the thread manager to be configured with `idOf`\"\n        );\n      }\n      const existing = await redis.lrange(redisKey, 0, -1);\n      const existingIds = existing\n        .map((raw) => idOf(deserialize(raw)))\n        .filter((id): id is string => typeof id === \"string\");\n      await redis.del(redisKey);\n      if (existingIds.length > 0) {\n        await redis.del(\n          ...existingIds.map((id) => getDedupKey(threadId, id))\n        );\n      }\n      if (messages.length > 0) {\n        await redis.rpush(redisKey, ...messages.map(serialize));\n        await redis.expire(redisKey, THREAD_TTL_SECONDS);\n      }\n      await redis.expire(metaKey, THREAD_TTL_SECONDS);\n    },\n\n    async delete(): Promise<void> {\n      await redis.del(redisKey, metaKey, stateKey);\n    },\n\n    async loadState(): Promise<PersistedThreadState | null> {\n      const raw = await redis.get(stateKey);\n      if (raw == null) return null;\n      return JSON.parse(raw) as PersistedThreadState;\n    },\n\n    async saveState(state: PersistedThreadState): Promise<void> {\n      await assertThreadExists();\n      await redis.set(\n        stateKey,\n        JSON.stringify(state),\n        \"EX\",\n        THREAD_TTL_SECONDS\n      );\n    },\n\n    async deleteState(): Promise<void> {\n      await redis.del(stateKey);\n    },\n\n    async length(): Promise<number> {\n      await assertThreadExists();\n      return redis.llen(redisKey);\n    },\n\n    async truncateFromId(messageId: string): Promise<void> {\n      await assertThreadExists();\n      if (!idOf) {\n        throw new Error(\n          \"truncateFromId requires the thread manager to be configured with `idOf`\"\n        );\n      }\n      const data = await redis.lrange(redisKey, 0, -1);\n      let idx = -1;\n      const removedIds: string[] = [];\n      for (let i = 0; i < data.length; i++) {\n        const raw = data[i];\n        if (raw === undefined) continue;\n        const id = idOf(deserialize(raw));\n        if (idx === -1 && id === messageId) idx = i;\n        if (idx !== -1) removedIds.push(id);\n      }\n      if (idx === -1) return;\n      if (idx === 0) {\n        await redis.del(redisKey);\n        await redis.expire(metaKey, THREAD_TTL_SECONDS);\n      } else {\n        await redis.ltrim(redisKey, 0, idx - 1);\n        await redis.expire(redisKey, THREAD_TTL_SECONDS);\n      }\n      // Clear dedup markers for the removed messages so that a rewind\n      // retry which reuses the same ids (e.g. the same assistantId) can\n      // re-append without the idempotent-append Lua script treating it\n      // as a duplicate.\n      if (removedIds.length > 0) {\n        await redis.del(\n          ...removedIds.map((id) => getDedupKey(threadId, id))\n        );\n      }\n    },\n  };\n}\n","import type Redis from \"ioredis\";\nimport type { Content, Part } from \"@google/genai\";\nimport { createThreadManager } from \"../../../lib/thread/manager\";\nimport type {\n  ProviderThreadManager,\n  ThreadManagerConfig,\n  ThreadManagerHooks,\n} from \"../../../lib/thread/types\";\nimport type { GoogleGenAIToolResponse } from \"./activities\";\n\n/** SDK-native content type for Google GenAI human messages */\nexport type GoogleGenAIContent = string | Part[];\n\n/** SDK-native content type for Google GenAI system instructions */\nexport type GoogleGenAISystemContent = string | Part[];\n\n/** A Content with a unique ID for idempotent Redis storage */\nexport interface StoredContent {\n  id: string;\n  content: Content;\n}\n\nexport type GoogleGenAIThreadManagerHooks = ThreadManagerHooks<\n  StoredContent,\n  Content\n>;\n\nexport interface GoogleGenAIThreadManagerConfig {\n  redis: Redis;\n  threadId: string;\n  /** Thread key, defaults to 'messages' */\n  key?: string;\n  hooks?: GoogleGenAIThreadManagerHooks;\n}\n\n/** Prepared payload ready to send to the Google GenAI API */\nexport interface GoogleGenAIInvocationPayload {\n  contents: Content[];\n  systemInstruction?: Part[];\n}\n\n/** Thread manager with Google GenAI Content convenience helpers */\nexport interface GoogleGenAIThreadManager extends ProviderThreadManager<\n  StoredContent,\n  GoogleGenAIContent,\n  GoogleGenAIToolResponse,\n  GoogleGenAISystemContent\n> {\n  appendModelContent(id: string, parts: Part[]): Promise<void>;\n  prepareForInvocation(): Promise<GoogleGenAIInvocationPayload>;\n}\n\nfunction storedContentId(msg: StoredContent): string {\n  return msg.id;\n}\n\n/** Normalise content into Part[] */\nfunction toParts(content: GoogleGenAIContent): Part[] {\n  if (typeof content === \"string\") {\n    return [{ text: content }];\n  }\n  return content;\n}\n\n/** Convert a string or object into a Record suitable for functionResponse.response */\nfunction toFunctionResponse(\n  content: string | Record<string, unknown>\n): Record<string, unknown> {\n  if (typeof content === \"object\") {\n    return content;\n  }\n  return { result: content };\n}\n\n/**\n * Merge consecutive Content objects sharing the same role.\n * The Gemini API requires alternating user/model turns; without\n * merging, multiple sequential tool-result messages would violate this.\n */\nfunction mergeConsecutiveContents(contents: Content[]): Content[] {\n  const merged: Content[] = [];\n  for (const content of contents) {\n    const last = merged[merged.length - 1];\n    if (last && last.role === content.role) {\n      last.parts = [...(last.parts ?? []), ...(content.parts ?? [])];\n    } else {\n      merged.push({ ...content, parts: [...(content.parts ?? [])] });\n    }\n  }\n  return merged;\n}\n\n/**\n * Creates a Google GenAI-specific thread manager that stores StoredContent\n * instances in Redis and provides convenience helpers for creating and\n * appending typed Content messages.\n */\nexport function createGoogleGenAIThreadManager(\n  config: GoogleGenAIThreadManagerConfig\n): GoogleGenAIThreadManager {\n  const baseConfig: ThreadManagerConfig<StoredContent> = {\n    redis: config.redis,\n    threadId: config.threadId,\n    key: config.key,\n    idOf: storedContentId,\n  };\n\n  const base = createThreadManager(baseConfig);\n\n  const helpers: Omit<GoogleGenAIThreadManager, keyof typeof base> = {\n    async appendUserMessage(\n      id: string,\n      content: GoogleGenAIContent\n    ): Promise<void> {\n      await base.append([\n        {\n          id,\n          content: { role: \"user\", parts: toParts(content) },\n        },\n      ]);\n    },\n\n    async appendSystemMessage(\n      id: string,\n      content: GoogleGenAISystemContent\n    ): Promise<void> {\n      const parts: Part[] =\n        typeof content === \"string\" ? [{ text: content }] : content;\n      await base.initialize();\n      await base.append([\n        {\n          id,\n          content: { role: \"system\", parts },\n        },\n      ]);\n    },\n\n    async appendModelContent(id: string, parts: Part[]): Promise<void> {\n      await base.append([\n        {\n          id,\n          content: { role: \"model\", parts },\n        },\n      ]);\n    },\n\n    async appendToolResult(\n      id: string,\n      toolCallId: string,\n      toolName: string,\n      content: GoogleGenAIToolResponse\n    ): Promise<void> {\n      const parts: Part[] = Array.isArray(content)\n        ? (content as Part[])\n        : [\n            {\n              functionResponse: {\n                id: toolCallId,\n                name: toolName,\n                response: toFunctionResponse(content),\n              },\n            },\n          ];\n\n      await base.append([\n        {\n          id,\n          content: { role: \"user\", parts },\n        },\n      ]);\n    },\n\n    async prepareForInvocation(): Promise<GoogleGenAIInvocationPayload> {\n      const stored = await base.load();\n      const { onPrepareMessage, onPreparedMessage } = config.hooks ?? {};\n      const mapped = onPrepareMessage\n        ? stored.map((msg, i) => onPrepareMessage(msg, i, stored))\n        : stored;\n\n      let systemInstruction: Part[] | undefined;\n      const conversationContents: Content[] = [];\n\n      for (const item of mapped) {\n        if (item.content.role === \"system\") {\n          systemInstruction = item.content.parts ?? [];\n        } else {\n          conversationContents.push(item.content);\n        }\n      }\n\n      const contents = mergeConsecutiveContents(conversationContents);\n      return {\n        contents: onPreparedMessage\n          ? contents.map((msg, i) => onPreparedMessage(msg, i, contents))\n          : contents,\n        ...(systemInstruction && systemInstruction.length > 0\n          ? { systemInstruction }\n          : {}),\n      };\n    },\n  };\n\n  const manager = Object.assign(base, helpers);\n\n  const originalFork = manager.fork.bind(manager);\n  manager.fork = async (\n    newThreadId: string\n  ): Promise<GoogleGenAIThreadManager> => {\n    await originalFork(newThreadId);\n    const forked = createGoogleGenAIThreadManager({\n      ...config,\n      threadId: newThreadId,\n    });\n    const { onForkPrepareThread, onForkTransform } = config.hooks ?? {};\n    if (!onForkPrepareThread && !onForkTransform) {\n      return forked;\n    }\n    let next = await forked.load();\n    if (onForkPrepareThread) {\n      next = await onForkPrepareThread(next);\n    }\n    if (onForkTransform) {\n      next = next.map((msg, i) => onForkTransform(msg, i, next));\n    }\n    await forked.replaceAll(next);\n    return forked;\n  };\n\n  return manager;\n}\n","import { Context } from \"@temporalio/activity\";\nimport type { WorkflowClient } from \"@temporalio/client\";\nimport type { BaseAgentState, RunAgentConfig } from \"./types\";\nimport type { JsonValue } from \"./state/types\";\nimport type {\n  ActivityToolHandler,\n  RouterContext,\n  ToolHandlerResponse,\n} from \"./tool-router/types\";\n\n/**\n * Safely retrieve Temporal activity heartbeat and cancellation signal.\n * Returns empty object when called outside a Temporal activity (e.g. tests).\n */\nexport function getActivityContext(): {\n  heartbeat?: () => void;\n  signal?: AbortSignal;\n} {\n  try {\n    const ctx = Context.current();\n    return { heartbeat: () => ctx.heartbeat(), signal: ctx.cancellationSignal };\n  } catch {\n    return {};\n  }\n}\n\n/**\n * Query the parent workflow's state from within an activity.\n * Resolves the workflow handle from the current activity context.\n */\nexport async function queryParentWorkflowState<T>(\n  client: WorkflowClient\n): Promise<T> {\n  const { workflowExecution } = Context.current().info;\n  const handle = client.getHandle(\n    workflowExecution.workflowId,\n    workflowExecution.runId\n  );\n  return handle.query<T>(\"getAgentState\");\n}\n\n/**\n * Wraps a handler into a scope-prefixed `RunAgentActivity` by auto-fetching\n * the parent workflow's agent state before each invocation.\n *\n * Returns a `Record` with a single key `run<Scope>` so it can be spread\n * into the activities object alongside adapter activities.\n *\n * @param scope - Workflow scope used to derive the activity name.\n *   `\"myAgentWorkflow\"` produces `{ runMyAgentWorkflow: fn }`.\n *\n * @example\n * ```typescript\n * import { createRunAgentActivity } from 'zeitlich';\n *\n * return {\n *   ...adapter.createActivities(\"myAgentWorkflow\"),\n *   ...createRunAgentActivity(client, adapter.invoker, \"myAgentWorkflow\"),\n * };\n * ```\n */\nexport function createRunAgentActivity<\n  R,\n  S extends BaseAgentState = BaseAgentState,\n>(\n  client: WorkflowClient,\n  handler: (config: RunAgentConfig & { state: S }) => Promise<R>,\n  scope: string\n): Record<string, (config: RunAgentConfig) => Promise<R>> {\n  const name = `run${scope.charAt(0).toUpperCase()}${scope.slice(1)}`;\n  return {\n    [name]: async (config: RunAgentConfig) => {\n      const state = await queryParentWorkflowState<S>(client);\n      return handler({ ...config, state });\n    },\n  };\n}\n\n/**\n * Context injected into tool handlers created via {@link withParentWorkflowState}.\n */\nexport interface AgentStateContext<\n  S extends BaseAgentState = BaseAgentState,\n> extends RouterContext {\n  state: S;\n}\n\n/**\n * Wraps a tool handler into an `ActivityToolHandler` by auto-fetching the\n * parent workflow's agent state before each invocation.\n *\n * @typeParam S - Custom agent state type (defaults to `BaseAgentState`)\n *\n * @example\n * ```typescript\n * import { withParentWorkflowState, type AgentStateContext } from 'zeitlich';\n *\n * // With custom state:\n * interface MyState extends BaseAgentState { customField: string }\n * const myHandler = withParentWorkflowState<MyArgs, MyResult, MyState>(\n *   client,\n *   async (args, ctx) => {\n *     console.log(ctx.state.customField);\n *     return { toolResponse: 'done', data: null };\n *   },\n * );\n * ```\n */\nexport function withParentWorkflowState<\n  TArgs,\n  TResult,\n  S extends BaseAgentState = BaseAgentState,\n  TToolResponse = JsonValue,\n>(\n  client: WorkflowClient,\n  handler: (\n    args: TArgs,\n    context: AgentStateContext<S>\n  ) => Promise<ToolHandlerResponse<TResult, TToolResponse>>\n): ActivityToolHandler<TArgs, TResult, RouterContext, TToolResponse> {\n  return async (args, context) => {\n    const state = await queryParentWorkflowState<S>(client);\n    return handler(args, { ...context, state });\n  };\n}\n","import type Redis from \"ioredis\";\nimport type {\n  GoogleGenAI,\n  Content,\n  FunctionDeclaration,\n  Part,\n  GenerateContentResponse,\n} from \"@google/genai\";\nimport type { SerializableToolDefinition } from \"../../../lib/types\";\nimport type { AgentResponse, ModelInvokerConfig } from \"../../../lib/model\";\nimport {\n  createGoogleGenAIThreadManager,\n  type GoogleGenAIThreadManagerHooks,\n} from \"./thread-manager\";\nimport { getActivityContext } from \"../../../lib/activity\";\n\nexport interface GoogleGenAIModelInvokerConfig {\n  redis: Redis;\n  client: GoogleGenAI;\n  model: string;\n  hooks?: GoogleGenAIThreadManagerHooks;\n}\n\nfunction toFunctionDeclarations(\n  tools: SerializableToolDefinition[]\n): FunctionDeclaration[] {\n  return tools.map((t) => ({\n    name: t.name,\n    description: t.description,\n    parametersJsonSchema: t.schema,\n  }));\n}\n\n/**\n * Creates a Google GenAI model invoker that satisfies the generic\n * `ModelInvoker<Content>` contract.\n *\n * Internally streams the response and emits Temporal heartbeats on each\n * chunk so that long-running LLM calls remain visible to the scheduler.\n * The caller is responsible for appending the response to the thread.\n *\n * @example\n * ```typescript\n * import { createGoogleGenAIModelInvoker } from 'zeitlich/adapters/thread/google-genai';\n * import { createRunAgentActivity } from 'zeitlich';\n * import { GoogleGenAI } from '@google/genai';\n *\n * const client = new GoogleGenAI({ apiKey: process.env.GEMINI_API_KEY });\n * const invoker = createGoogleGenAIModelInvoker({\n *   redis,\n *   client,\n *   model: 'gemini-2.5-flash',\n * });\n *\n * return { ...createRunAgentActivity(client, invoker, \"myAgent\") };\n * ```\n */\nexport function createGoogleGenAIModelInvoker({\n  redis,\n  client,\n  model,\n  hooks,\n}: GoogleGenAIModelInvokerConfig) {\n  return async function invokeGoogleGenAIModel(\n    config: ModelInvokerConfig\n  ): Promise<AgentResponse<Content>> {\n    const { threadId, threadKey, state, assistantMessageId } = config;\n    const { heartbeat, signal } = getActivityContext();\n\n    const thread = createGoogleGenAIThreadManager({\n      redis,\n      threadId,\n      key: threadKey,\n      hooks,\n    });\n    // Truncate the thread starting at the id the assistant message\n    // will be stored under. No-op on the first attempt; on rewind\n    // retry / Temporal reset it wipes the prior attempt's assistant\n    // + tool results so the LLM sees the original pre-call state.\n    await thread.truncateFromId(assistantMessageId);\n    const { contents, systemInstruction } =\n      await thread.prepareForInvocation();\n\n    const functionDeclarations = toFunctionDeclarations(state.tools);\n    const tools =\n      functionDeclarations.length > 0 ? [{ functionDeclarations }] : undefined;\n\n    const stream = await client.models.generateContentStream({\n      model,\n      contents,\n      config: {\n        ...(systemInstruction ? { systemInstruction } : {}),\n        ...(tools ? { tools } : {}),\n        abortSignal: signal,\n      },\n    });\n\n    const allParts: Part[] = [];\n    let lastChunk: GenerateContentResponse | undefined;\n    for await (const chunk of stream) {\n      lastChunk = chunk;\n      allParts.push(...(chunk.candidates?.[0]?.content?.parts ?? []));\n      heartbeat?.();\n    }\n\n    if (!lastChunk) {\n      throw new Error(\"Google GenAI stream ended without producing any chunks\");\n    }\n\n    const modelContent: Content = { role: \"model\", parts: allParts };\n    const functionCalls = lastChunk.functionCalls ?? [];\n\n    return {\n      message: modelContent,\n      rawToolCalls: functionCalls.map((fc) => ({\n        id: fc.id,\n        name: fc.name ?? \"\",\n        args: fc.args ?? {},\n      })),\n      usage: {\n        inputTokens: lastChunk.usageMetadata?.promptTokenCount,\n        outputTokens: lastChunk.usageMetadata?.candidatesTokenCount,\n        cachedReadTokens: lastChunk.usageMetadata?.cachedContentTokenCount,\n      },\n    };\n  };\n}\n\n/**\n * Standalone function for one-shot Google GenAI model invocation.\n * Convenience wrapper around createGoogleGenAIModelInvoker for cases\n * where you don't need to reuse the invoker.\n */\nexport async function invokeGoogleGenAIModel({\n  redis,\n  client,\n  model,\n  hooks,\n  config,\n}: {\n  redis: Redis;\n  client: GoogleGenAI;\n  model: string;\n  hooks?: GoogleGenAIThreadManagerHooks;\n  config: ModelInvokerConfig;\n}): Promise<AgentResponse<Content>> {\n  const invoker = createGoogleGenAIModelInvoker({\n    redis,\n    client,\n    model,\n    hooks,\n  });\n  return invoker(config);\n}\n","import type Redis from \"ioredis\";\nimport type { GoogleGenAI, Content, Part } from \"@google/genai\";\nimport type { ToolResultConfig } from \"../../../lib/types\";\nimport type { PersistedThreadState } from \"../../../lib/state/types\";\nimport type {\n  ActivityToolHandler,\n  RouterContext,\n  ToolHandlerResponse,\n} from \"../../../lib/tool-router/types\";\nimport type {\n  ThreadOps,\n  PrefixedThreadOps,\n  ScopedPrefix,\n} from \"../../../lib/session/types\";\nimport type { ModelInvoker } from \"../../../lib/model\";\nimport {\n  createGoogleGenAIThreadManager,\n  type GoogleGenAIContent,\n  type GoogleGenAISystemContent,\n  type GoogleGenAIThreadManagerHooks,\n} from \"./thread-manager\";\nimport { createGoogleGenAIModelInvoker } from \"./model-invoker\";\nimport { ADAPTER_ID } from \"./adapter-id\";\n\nexport type GoogleGenAIThreadOps<TScope extends string = \"\"> =\n  PrefixedThreadOps<\n    ScopedPrefix<TScope, typeof ADAPTER_ID>,\n    GoogleGenAIContent\n  >;\n\nexport interface GoogleGenAIAdapterConfig {\n  redis: Redis;\n  client?: GoogleGenAI;\n  /** Default model name (e.g. 'gemini-2.5-flash'). If omitted, use `createModelInvoker()` */\n  model?: string;\n  hooks?: GoogleGenAIThreadManagerHooks;\n}\n\n/**\n * Tool response type accepted by the Google GenAI adapter.\n *\n * Handlers can return:\n * - **`string`** — plain text, wrapped in a `functionResponse` part.\n * - **`Record<string, unknown>`** — structured object used as `functionResponse.response`.\n * - **`Part[]`** — pre-built parts used directly as `Content.parts`.\n *   The handler is responsible for building correct Part objects (e.g. `functionResponse`,\n *   `inlineData`, `text`). Use `context.toolCallId` and `context.toolName` to construct\n *   `functionResponse` parts.\n *\n * @example\n * ```typescript\n * adapter.wrapHandler(async (args, ctx) => ({\n *   toolResponse: [\n *     { functionResponse: { id: ctx.toolCallId, name: ctx.toolName, response: { result: \"done\" } } },\n *     { inlineData: { data: base64, mimeType: \"image/png\" } },\n *   ],\n *   data: null,\n * }));\n * ```\n */\nexport type GoogleGenAIToolResponse = string | Record<string, unknown> | Part[];\n\nexport interface GoogleGenAIAdapter {\n  /** Model invoker using the default model (only available when `model` was provided) */\n  invoker: ModelInvoker<Content>;\n  /** Create an invoker for a specific model name (for multi-model setups) */\n  createModelInvoker(model: string, client: GoogleGenAI): ModelInvoker<Content>;\n  /**\n   * Create prefixed thread activities for registration on the worker.\n   *\n   * @param scope - Workflow name appended to the adapter prefix.\n   *   Use different scopes for the main agent vs subagents to avoid collisions.\n   *\n   * @example\n   * ```typescript\n   * adapter.createActivities(\"codingAgent\")\n   * // → { googleGenAICodingAgentInitializeThread, googleGenAICodingAgentAppendHumanMessage, … }\n   *\n   * adapter.createActivities(\"researchAgent\")\n   * // → { googleGenAIResearchAgentInitializeThread, … }\n   * ```\n   */\n  createActivities<S extends string = \"\">(scope?: S): GoogleGenAIThreadOps<S>;\n\n  /**\n   * Identity wrapper that types a tool handler for this adapter.\n   * Constrains `toolResponse` to {@link GoogleGenAIToolResponse}.\n   */\n  wrapHandler<TArgs, TResult, TContext extends RouterContext = RouterContext>(\n    handler: (\n      args: TArgs,\n      context: TContext\n    ) => Promise<ToolHandlerResponse<TResult, GoogleGenAIToolResponse>>\n  ): ActivityToolHandler<TArgs, TResult, TContext, GoogleGenAIToolResponse>;\n}\n\n/**\n * Creates a Google GenAI adapter that bundles thread operations and model\n * invocation using the `@google/genai` SDK.\n *\n * Use `createActivities(scope)` to register scoped thread operations as\n * Temporal activities on the worker. The `invoker` (or invokers created via\n * `createModelInvoker`) should be wrapped with `createRunAgentActivity`.\n *\n * @example\n * ```typescript\n * import { createGoogleGenAIAdapter } from 'zeitlich/adapters/thread/google-genai';\n * import { createRunAgentActivity } from 'zeitlich';\n * import { GoogleGenAI } from '@google/genai';\n *\n * const client = new GoogleGenAI({ apiKey: process.env.GEMINI_API_KEY });\n * const adapter = createGoogleGenAIAdapter({ redis, client, model: 'gemini-2.5-flash' });\n *\n * export function createActivities(temporalClient: WorkflowClient) {\n *   return {\n *     ...adapter.createActivities(\"codingAgent\"),\n *     ...createRunAgentActivity(temporalClient, adapter.invoker, \"codingAgent\"),\n *   };\n * }\n * ```\n *\n * @example Multi-agent worker (main + subagent share the adapter)\n * ```typescript\n * export function createActivities(temporalClient: WorkflowClient) {\n *   return {\n *     ...adapter.createActivities(\"codingAgent\"),\n *     ...adapter.createActivities(\"researchAgent\"),\n *     ...createRunAgentActivity(temporalClient, adapter.invoker, \"codingAgent\"),\n *     ...createRunAgentActivity(\n *       temporalClient,\n *       adapter.createModelInvoker('gemini-2.5-pro'),\n *       \"researchAgent\",\n *     ),\n *   };\n * }\n * ```\n */\nexport function createGoogleGenAIAdapter(\n  config: GoogleGenAIAdapterConfig\n): GoogleGenAIAdapter {\n  const { redis } = config;\n\n  const threadOps: ThreadOps<GoogleGenAIContent> = {\n    async initializeThread(\n      threadId: string,\n      threadKey?: string\n    ): Promise<void> {\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId,\n        key: threadKey,\n      });\n      await thread.initialize();\n    },\n\n    async appendHumanMessage(\n      threadId: string,\n      id: string,\n      content: GoogleGenAIContent,\n      threadKey?: string\n    ): Promise<void> {\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId,\n        key: threadKey,\n      });\n      await thread.appendUserMessage(id, content);\n    },\n\n    async appendSystemMessage(\n      threadId: string,\n      id: string,\n      content: GoogleGenAISystemContent,\n      threadKey?: string\n    ): Promise<void> {\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId,\n        key: threadKey,\n      });\n      await thread.appendSystemMessage(id, content);\n    },\n\n    async appendToolResult(id: string, cfg: ToolResultConfig): Promise<void> {\n      const { threadId, threadKey, toolCallId, toolName, content } = cfg;\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId,\n        key: threadKey,\n      });\n      await thread.appendToolResult(\n        id,\n        toolCallId,\n        toolName,\n        content as GoogleGenAIToolResponse\n      );\n    },\n\n    async appendAgentMessage(\n      threadId: string,\n      id: string,\n      message: Content,\n      threadKey?: string\n    ): Promise<void> {\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId,\n        key: threadKey,\n      });\n      await thread.appendModelContent(id, message.parts ?? []);\n    },\n\n    async forkThread(\n      sourceThreadId: string,\n      targetThreadId: string,\n      threadKey?: string\n    ): Promise<void> {\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId: sourceThreadId,\n        key: threadKey,\n        hooks: config.hooks,\n      });\n      await thread.fork(targetThreadId);\n    },\n\n    async truncateThread(\n      threadId: string,\n      messageId: string,\n      threadKey?: string,\n    ): Promise<void> {\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId,\n        key: threadKey,\n      });\n      await thread.truncateFromId(messageId);\n    },\n\n    async loadThreadState(\n      threadId: string,\n      threadKey?: string\n    ): Promise<PersistedThreadState | null> {\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId,\n        key: threadKey,\n      });\n      return thread.loadState();\n    },\n\n    async saveThreadState(\n      threadId: string,\n      state: PersistedThreadState,\n      threadKey?: string\n    ): Promise<void> {\n      const thread = createGoogleGenAIThreadManager({\n        redis,\n        threadId,\n        key: threadKey,\n      });\n      await thread.saveState(state);\n    },\n  };\n\n  function createActivities<S extends string = \"\">(\n    scope?: S\n  ): GoogleGenAIThreadOps<S> {\n    const prefix = scope\n      ? `${ADAPTER_ID}${scope.charAt(0).toUpperCase()}${scope.slice(1)}`\n      : ADAPTER_ID;\n    const cap = (s: string): string => s.charAt(0).toUpperCase() + s.slice(1);\n    return Object.fromEntries(\n      Object.entries(threadOps).map(([k, v]) => [`${prefix}${cap(k)}`, v])\n    ) as GoogleGenAIThreadOps<S>;\n  }\n\n  const makeInvoker = (\n    model: string,\n    client: GoogleGenAI\n  ): ModelInvoker<Content> =>\n    createGoogleGenAIModelInvoker({\n      redis,\n      client,\n      model,\n      hooks: config.hooks,\n    });\n\n  const invoker: ModelInvoker<Content> =\n    config.model && config.client\n      ? makeInvoker(config.model, config.client)\n      : ((() => {\n          throw new Error(\n            \"No default model provided to createGoogleGenAIAdapter. \" +\n              \"Either pass `model` in the config or use `createModelInvoker(model)` instead.\"\n          );\n        }) as unknown as ModelInvoker<Content>);\n\n  return {\n    createActivities,\n    invoker,\n    createModelInvoker: makeInvoker,\n    wrapHandler: (handler) => handler,\n  };\n}\n"]}

package/dist/adapters/thread/google-genai/index.d.cts

@@ -1,9 +1,10 @@
-import { G as GoogleGenAIThreadManagerHooks } from '../../../activities-CDcwkRZs.cjs';
-export { a as GoogleGenAIAdapter, b as GoogleGenAIAdapterConfig, c as GoogleGenAIContent, d as GoogleGenAIInvocationPayload, e as GoogleGenAIThreadManager, f as GoogleGenAIThreadManagerConfig, g as GoogleGenAIThreadOps, h as GoogleGenAIToolResponse, S as StoredContent, i as createGoogleGenAIAdapter, j as createGoogleGenAIThreadManager } from '../../../activities-CDcwkRZs.cjs';
+export { A as ADAPTER_ID, a as AdapterId } from '../../../adapter-id-BB-mmrts.cjs';
+import { G as GoogleGenAIThreadManagerHooks } from '../../../activities-CvUrG3YG.cjs';
+export { a as GoogleGenAIAdapter, b as GoogleGenAIAdapterConfig, c as GoogleGenAIContent, d as GoogleGenAIInvocationPayload, e as GoogleGenAIThreadManager, f as GoogleGenAIThreadManagerConfig, g as GoogleGenAIThreadOps, h as GoogleGenAIToolResponse, S as StoredContent, i as createGoogleGenAIAdapter, j as createGoogleGenAIThreadManager } from '../../../activities-CvUrG3YG.cjs';
 import Redis from 'ioredis';
 import { GoogleGenAI, Content } from '@google/genai';
-import { M as ModelInvokerConfig, A as AgentResponse } from '../../../types-C06FwR96.cjs';
-import '../../../types-BaOw4hKI.cjs';
+import { M as ModelInvokerConfig, A as AgentResponse } from '../../../types-Cn2r3ol3.cjs';
+import '../../../types-CuISs0Ub.cjs';
 import '@temporalio/common';
 import '@temporalio/workflow';
 import '@temporalio/common/lib/interfaces';

package/dist/adapters/thread/google-genai/index.d.ts

@@ -1,9 +1,10 @@
-import { G as GoogleGenAIThreadManagerHooks } from '../../../activities-BKhMtKDd.js';
-export { a as GoogleGenAIAdapter, b as GoogleGenAIAdapterConfig, c as GoogleGenAIContent, d as GoogleGenAIInvocationPayload, e as GoogleGenAIThreadManager, f as GoogleGenAIThreadManagerConfig, g as GoogleGenAIThreadOps, h as GoogleGenAIToolResponse, S as StoredContent, i as createGoogleGenAIAdapter, j as createGoogleGenAIThreadManager } from '../../../activities-BKhMtKDd.js';
+export { A as ADAPTER_ID, a as AdapterId } from '../../../adapter-id-BB-mmrts.js';
+import { G as GoogleGenAIThreadManagerHooks } from '../../../activities-CULxRzJ1.js';
+export { a as GoogleGenAIAdapter, b as GoogleGenAIAdapterConfig, c as GoogleGenAIContent, d as GoogleGenAIInvocationPayload, e as GoogleGenAIThreadManager, f as GoogleGenAIThreadManagerConfig, g as GoogleGenAIThreadOps, h as GoogleGenAIToolResponse, S as StoredContent, i as createGoogleGenAIAdapter, j as createGoogleGenAIThreadManager } from '../../../activities-CULxRzJ1.js';
 import Redis from 'ioredis';
 import { GoogleGenAI, Content } from '@google/genai';
-import { M as ModelInvokerConfig, A as AgentResponse } from '../../../types-DNr31FzL.js';
-import '../../../types-BH_IRryz.js';
+import { M as ModelInvokerConfig, A as AgentResponse } from '../../../types-DeQH84C_.js';
+import '../../../types-C7OoY7h8.js';
 import '@temporalio/common';
 import '@temporalio/workflow';
 import '@temporalio/common/lib/interfaces';