zeitlich 0.2.36 → 0.2.38

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

@@ -81,6 +81,20 @@ function createThreadManager(config) {
     },
     async delete() {
       await redis.del(redisKey, metaKey);
+    },
+    async length() {
+      await assertThreadExists();
+      return redis.llen(redisKey);
+    },
+    async truncate(length) {
+      await assertThreadExists();
+      if (length <= 0) {
+        await redis.del(redisKey);
+        await redis.expire(metaKey, THREAD_TTL_SECONDS);
+      } else {
+        await redis.ltrim(redisKey, 0, length - 1);
+        await redis.expire(redisKey, THREAD_TTL_SECONDS);
+      }
     }
   };
 }
@@ -123,37 +137,47 @@ function createGoogleGenAIThreadManager(config) {
   const base = createThreadManager(baseConfig);
   const helpers = {
     async appendUserMessage(id, content) {
-      await base.append([{
-        id,
-        content: { role: "user", parts: toParts(content) }
-      }]);
+      await base.append([
+        {
+          id,
+          content: { role: "user", parts: toParts(content) }
+        }
+      ]);
     },
     async appendSystemMessage(id, content) {
       const parts = typeof content === "string" ? [{ text: content }] : content;
       await base.initialize();
-      await base.append([{
-        id,
-        content: { role: "system", parts }
-      }]);
+      await base.append([
+        {
+          id,
+          content: { role: "system", parts }
+        }
+      ]);
     },
     async appendModelContent(id, parts) {
-      await base.append([{
-        id,
-        content: { role: "model", parts }
-      }]);
+      await base.append([
+        {
+          id,
+          content: { role: "model", parts }
+        }
+      ]);
     },
     async appendToolResult(id, toolCallId, toolName, content) {
-      const parts = Array.isArray(content) ? content : [{
-        functionResponse: {
-          id: toolCallId,
-          name: toolName,
-          response: toFunctionResponse(content)
+      const parts = Array.isArray(content) ? content : [
+        {
+          functionResponse: {
+            id: toolCallId,
+            name: toolName,
+            response: toFunctionResponse(content)
+          }
         }
-      }];
-      await base.append([{
-        id,
-        content: { role: "user", parts }
-      }]);
+      ];
+      await base.append([
+        {
+          id,
+          content: { role: "user", parts }
+        }
+      ]);
     },
     async prepareForInvocation() {
       const stored = await base.load();
@@ -171,7 +195,8 @@ function createGoogleGenAIThreadManager(config) {
       const contents = mergeConsecutiveContents(conversationContents);
       return {
         contents: onPreparedMessage ? contents.map((msg, i) => onPreparedMessage(msg, i, contents)) : contents,
-        ...systemInstruction && systemInstruction.length > 0 ? { systemInstruction } : {}
+        ...systemInstruction && systemInstruction.length > 0 ? { systemInstruction } : {},
+        storedLength: stored.length
       };
     }
   };
@@ -203,8 +228,13 @@ function createGoogleGenAIModelInvoker({
   return async function invokeGoogleGenAIModel2(config) {
     const { threadId, threadKey, state } = config;
     const { heartbeat, signal } = getActivityContext();
-    const thread = createGoogleGenAIThreadManager({ redis, threadId, key: threadKey, hooks });
-    const { contents, systemInstruction } = await thread.prepareForInvocation();
+    const thread = createGoogleGenAIThreadManager({
+      redis,
+      threadId,
+      key: threadKey,
+      hooks
+    });
+    const { contents, systemInstruction, storedLength } = await thread.prepareForInvocation();
     const functionDeclarations = toFunctionDeclarations(state.tools);
     const tools = functionDeclarations.length > 0 ? [{ functionDeclarations }] : void 0;
     const stream = await client.models.generateContentStream({
@@ -239,7 +269,8 @@ function createGoogleGenAIModelInvoker({
         inputTokens: lastChunk.usageMetadata?.promptTokenCount,
         outputTokens: lastChunk.usageMetadata?.candidatesTokenCount,
         cachedReadTokens: lastChunk.usageMetadata?.cachedContentTokenCount
-      }
+      },
+      threadLengthAtCall: storedLength
     };
   };
 }
@@ -250,7 +281,12 @@ async function invokeGoogleGenAIModel({
   hooks,
   config
 }) {
-  const invoker = createGoogleGenAIModelInvoker({ redis, client, model, hooks });
+  const invoker = createGoogleGenAIModelInvoker({
+    redis,
+    client,
+    model,
+    hooks
+  });
   return invoker(config);
 }
 
@@ -312,6 +348,14 @@ function createGoogleGenAIAdapter(config) {
         key: threadKey
       });
       await thread.fork(targetThreadId);
+    },
+    async truncateThread(threadId, length, threadKey) {
+      const thread = createGoogleGenAIThreadManager({
+        redis,
+        threadId,
+        key: threadKey
+      });
+      await thread.truncate(length);
     }
   };
   function createActivities(scope) {
@@ -321,7 +365,12 @@ function createGoogleGenAIAdapter(config) {
       Object.entries(threadOps).map(([k, v]) => [`${prefix}${cap(k)}`, v])
     );
   }
-  const makeInvoker = (model, client) => createGoogleGenAIModelInvoker({ redis, client, model, hooks: config.hooks });
+  const makeInvoker = (model, client) => createGoogleGenAIModelInvoker({
+    redis,
+    client,
+    model,
+    hooks: config.hooks
+  });
   const invoker = config.model && config.client ? makeInvoker(config.model, config.client) : (() => {
     throw new Error(
       "No default model provided to createGoogleGenAIAdapter. Either pass `model` in the config or use `createModelInvoker(model)` instead."

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;AAAA,GACF;AACF;;;AC9DA,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,mBAAmB,OAAA,EAAoE;AAC9F,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,IAAA,CAAK,OAAO,CAAC;AAAA,QACjB,EAAA;AAAA,QACA,SAAS,EAAE,IAAA,EAAM,QAAQ,KAAA,EAAO,OAAA,CAAQ,OAAO,CAAA;AAAE,OAClD,CAAC,CAAA;AAAA,IACJ,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,IAAA,CAAK,OAAO,CAAC;AAAA,QACjB,EAAA;AAAA,QACA,OAAA,EAAS,EAAE,IAAA,EAAM,QAAA,EAAU,KAAA;AAAM,OAClC,CAAC,CAAA;AAAA,IACJ,CAAA;AAAA,IAEA,MAAM,kBAAA,CAAmB,EAAA,EAAY,KAAA,EAA8B;AACjE,MAAA,MAAM,IAAA,CAAK,OAAO,CAAC;AAAA,QACjB,EAAA;AAAA,QACA,OAAA,EAAS,EAAE,IAAA,EAAM,OAAA,EAAS,KAAA;AAAM,OACjC,CAAC,CAAA;AAAA,IACJ,CAAA;AAAA,IAEA,MAAM,gBAAA,CACJ,EAAA,EACA,UAAA,EACA,UACA,OAAA,EACe;AACf,MAAA,MAAM,QAAgB,KAAA,CAAM,OAAA,CAAQ,OAAO,CAAA,GACvC,UACA,CAAC;AAAA,QACC,gBAAA,EAAkB;AAAA,UAChB,EAAA,EAAI,UAAA;AAAA,UACJ,IAAA,EAAM,QAAA;AAAA,UACN,QAAA,EAAU,mBAAmB,OAAO;AAAA;AACtC,OACD,CAAA;AAEL,MAAA,MAAM,IAAA,CAAK,OAAO,CAAC;AAAA,QACjB,EAAA;AAAA,QACA,OAAA,EAAS,EAAE,IAAA,EAAM,MAAA,EAAQ,KAAA;AAAM,OAChC,CAAC,CAAA;AAAA,IACJ,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,IAAI,EAAE,iBAAA,KAAsB;AAAC,OACnF;AAAA,IACF;AAAA,GACF;AAEA,EAAA,OAAO,MAAA,CAAO,MAAA,CAAO,IAAA,EAAM,OAAO,CAAA;AACpC;ACxKO,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;;;ACVA,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,MAAA,GAAS,+BAA+B,EAAE,KAAA,EAAO,UAAU,GAAA,EAAK,SAAA,EAAW,OAAO,CAAA;AACxF,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,EAAE,OAAO,MAAA,EAAQ,KAAA,EAAO,OAAO,CAAA;AAC7E,EAAA,OAAO,QAAQ,MAAM,CAAA;AACvB;;;AC3GA,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;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,EAAE,KAAA,EAAO,MAAA,EAAQ,KAAA,EAAO,KAAA,EAAO,MAAA,CAAO,KAAA,EAAO,CAAA;AAE7E,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}\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<StoredContent, Content>;\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\n  extends ProviderThreadManager<StoredContent, GoogleGenAIContent, GoogleGenAIToolResponse, GoogleGenAISystemContent> {\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(content: string | Record<string, unknown>): 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        id,\n        content: { role: \"user\", parts: toParts(content) },\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        id,\n        content: { role: \"system\", parts },\n      }]);\n    },\n\n    async appendModelContent(id: string, parts: Part[]): Promise<void> {\n      await base.append([{\n        id,\n        content: { role: \"model\", parts },\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            functionResponse: {\n              id: toolCallId,\n              name: toolName,\n              response: toFunctionResponse(content),\n            },\n          }];\n\n      await base.append([{\n        id,\n        content: { role: \"user\", parts },\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 ? { systemInstruction } : {}),\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<R, S extends BaseAgentState = BaseAgentState>(\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<S extends BaseAgentState = BaseAgentState> 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<TArgs, TResult, S extends BaseAgentState = BaseAgentState, TToolResponse = JsonValue>(\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 { GoogleGenAI, Content, FunctionDeclaration, Part, GenerateContentResponse } from \"@google/genai\";\nimport type { SerializableToolDefinition } from \"../../../lib/types\";\nimport type { AgentResponse, ModelInvokerConfig } from \"../../../lib/model\";\nimport { createGoogleGenAIThreadManager, type GoogleGenAIThreadManagerHooks } 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({ redis, threadId, key: threadKey, hooks });\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({ redis, client, model, hooks });\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\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({ redis, client, model, hooks: config.hooks });\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/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"]}

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

@@ -1,14 +1,14 @@
-import { G as GoogleGenAIThreadManagerHooks } from '../../../activities-hd4aNnZE.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-hd4aNnZE.cjs';
+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';
 import Redis from 'ioredis';
 import { GoogleGenAI, Content } from '@google/genai';
-import { M as ModelInvokerConfig, A as AgentResponse } from '../../../types-DQ1l_gXL.cjs';
-import '../../../types-Mc_4BCfT.cjs';
+import { M as ModelInvokerConfig, A as AgentResponse } from '../../../types-C06FwR96.cjs';
+import '../../../types-BaOw4hKI.cjs';
 import '@temporalio/common';
 import '@temporalio/workflow';
 import '@temporalio/common/lib/interfaces';
 import 'zod';
-import '../../../types-DxCpFNv_.cjs';
+import '../../../types-yx0LzPGn.cjs';
 
 interface GoogleGenAIModelInvokerConfig {
     redis: Redis;

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

@@ -1,14 +1,14 @@
-import { G as GoogleGenAIThreadManagerHooks } from '../../../activities-BVI2lTwr.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-BVI2lTwr.js';
+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';
 import Redis from 'ioredis';
 import { GoogleGenAI, Content } from '@google/genai';
-import { M as ModelInvokerConfig, A as AgentResponse } from '../../../types-CADc5V_P.js';
-import '../../../types-yiXmqedU.js';
+import { M as ModelInvokerConfig, A as AgentResponse } from '../../../types-DNr31FzL.js';
+import '../../../types-BH_IRryz.js';
 import '@temporalio/common';
 import '@temporalio/workflow';
 import '@temporalio/common/lib/interfaces';
 import 'zod';
-import '../../../types-DxCpFNv_.js';
+import '../../../types-yx0LzPGn.js';
 
 interface GoogleGenAIModelInvokerConfig {
     redis: Redis;

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

@@ -79,6 +79,20 @@ function createThreadManager(config) {
     },
     async delete() {
       await redis.del(redisKey, metaKey);
+    },
+    async length() {
+      await assertThreadExists();
+      return redis.llen(redisKey);
+    },
+    async truncate(length) {
+      await assertThreadExists();
+      if (length <= 0) {
+        await redis.del(redisKey);
+        await redis.expire(metaKey, THREAD_TTL_SECONDS);
+      } else {
+        await redis.ltrim(redisKey, 0, length - 1);
+        await redis.expire(redisKey, THREAD_TTL_SECONDS);
+      }
     }
   };
 }
@@ -121,37 +135,47 @@ function createGoogleGenAIThreadManager(config) {
   const base = createThreadManager(baseConfig);
   const helpers = {
     async appendUserMessage(id, content) {
-      await base.append([{
-        id,
-        content: { role: "user", parts: toParts(content) }
-      }]);
+      await base.append([
+        {
+          id,
+          content: { role: "user", parts: toParts(content) }
+        }
+      ]);
     },
     async appendSystemMessage(id, content) {
       const parts = typeof content === "string" ? [{ text: content }] : content;
       await base.initialize();
-      await base.append([{
-        id,
-        content: { role: "system", parts }
-      }]);
+      await base.append([
+        {
+          id,
+          content: { role: "system", parts }
+        }
+      ]);
     },
     async appendModelContent(id, parts) {
-      await base.append([{
-        id,
-        content: { role: "model", parts }
-      }]);
+      await base.append([
+        {
+          id,
+          content: { role: "model", parts }
+        }
+      ]);
     },
     async appendToolResult(id, toolCallId, toolName, content) {
-      const parts = Array.isArray(content) ? content : [{
-        functionResponse: {
-          id: toolCallId,
-          name: toolName,
-          response: toFunctionResponse(content)
+      const parts = Array.isArray(content) ? content : [
+        {
+          functionResponse: {
+            id: toolCallId,
+            name: toolName,
+            response: toFunctionResponse(content)
+          }
         }
-      }];
-      await base.append([{
-        id,
-        content: { role: "user", parts }
-      }]);
+      ];
+      await base.append([
+        {
+          id,
+          content: { role: "user", parts }
+        }
+      ]);
     },
     async prepareForInvocation() {
       const stored = await base.load();
@@ -169,7 +193,8 @@ function createGoogleGenAIThreadManager(config) {
       const contents = mergeConsecutiveContents(conversationContents);
       return {
         contents: onPreparedMessage ? contents.map((msg, i) => onPreparedMessage(msg, i, contents)) : contents,
-        ...systemInstruction && systemInstruction.length > 0 ? { systemInstruction } : {}
+        ...systemInstruction && systemInstruction.length > 0 ? { systemInstruction } : {},
+        storedLength: stored.length
       };
     }
   };
@@ -201,8 +226,13 @@ function createGoogleGenAIModelInvoker({
   return async function invokeGoogleGenAIModel2(config) {
     const { threadId, threadKey, state } = config;
     const { heartbeat, signal } = getActivityContext();
-    const thread = createGoogleGenAIThreadManager({ redis, threadId, key: threadKey, hooks });
-    const { contents, systemInstruction } = await thread.prepareForInvocation();
+    const thread = createGoogleGenAIThreadManager({
+      redis,
+      threadId,
+      key: threadKey,
+      hooks
+    });
+    const { contents, systemInstruction, storedLength } = await thread.prepareForInvocation();
     const functionDeclarations = toFunctionDeclarations(state.tools);
     const tools = functionDeclarations.length > 0 ? [{ functionDeclarations }] : void 0;
     const stream = await client.models.generateContentStream({
@@ -237,7 +267,8 @@ function createGoogleGenAIModelInvoker({
         inputTokens: lastChunk.usageMetadata?.promptTokenCount,
         outputTokens: lastChunk.usageMetadata?.candidatesTokenCount,
         cachedReadTokens: lastChunk.usageMetadata?.cachedContentTokenCount
-      }
+      },
+      threadLengthAtCall: storedLength
     };
   };
 }
@@ -248,7 +279,12 @@ async function invokeGoogleGenAIModel({
   hooks,
   config
 }) {
-  const invoker = createGoogleGenAIModelInvoker({ redis, client, model, hooks });
+  const invoker = createGoogleGenAIModelInvoker({
+    redis,
+    client,
+    model,
+    hooks
+  });
   return invoker(config);
 }
 
@@ -310,6 +346,14 @@ function createGoogleGenAIAdapter(config) {
         key: threadKey
       });
       await thread.fork(targetThreadId);
+    },
+    async truncateThread(threadId, length, threadKey) {
+      const thread = createGoogleGenAIThreadManager({
+        redis,
+        threadId,
+        key: threadKey
+      });
+      await thread.truncate(length);
     }
   };
   function createActivities(scope) {
@@ -319,7 +363,12 @@ function createGoogleGenAIAdapter(config) {
       Object.entries(threadOps).map(([k, v]) => [`${prefix}${cap(k)}`, v])
     );
   }
-  const makeInvoker = (model, client) => createGoogleGenAIModelInvoker({ redis, client, model, hooks: config.hooks });
+  const makeInvoker = (model, client) => createGoogleGenAIModelInvoker({
+    redis,
+    client,
+    model,
+    hooks: config.hooks
+  });
   const invoker = config.model && config.client ? makeInvoker(config.model, config.client) : (() => {
     throw new Error(
       "No default model provided to createGoogleGenAIAdapter. Either pass `model` in the config or use `createModelInvoker(model)` instead."