@kenkaiiii/gg-ai 5.87.1 → 5.87.2

package/dist/index.cjs

@@ -1806,7 +1806,7 @@ function parseToolArguments(argsJson) {
 
 // src/utils/provider-fetch.ts
 var import_undici = require("undici");
-var PROVIDER_TRANSPORT_TIMEOUT_MS = 30 * 6e4;
+var PROVIDER_TRANSPORT_TIMEOUT_MS = 45 * 6e4;
 function nodeUsesEnvProxy({
   env = process.env,
   execArgv = process.execArgv,
@@ -1834,6 +1834,25 @@ function createProviderFetch(timeoutMs, { useEnvProxy = nodeUsesEnvProxy() } = {
   };
   return Object.assign(providerFetch2, { Response: globalThis.Response });
 }
+function withBodyActivity(fetchImpl, onActivity) {
+  return async (input, init2) => {
+    const response = await fetchImpl(input, init2);
+    if (!response.body || !response.headers.get("content-type")?.includes("text/event-stream")) {
+      return response;
+    }
+    const tap = new TransformStream({
+      transform(chunk, controller) {
+        onActivity();
+        controller.enqueue(chunk);
+      }
+    });
+    return new Response(response.body.pipeThrough(tap), {
+      status: response.status,
+      statusText: response.statusText,
+      headers: response.headers
+    });
+  };
+}
 var providerFetch = createProviderFetch(
   PROVIDER_TRANSPORT_TIMEOUT_MS
 );
@@ -2134,8 +2153,10 @@ async function* runStream(options) {
   const keepalive = { type: "keepalive" };
   let receivedAnyEvent = false;
   try {
+    const onActivity = options.onTransportActivity;
+    const streamClient = onActivity ? client.withOptions({ fetch: withBodyActivity(options.fetch ?? providerFetch, onActivity) }) : client;
     const stream2 = await send(
-      (p, o) => client.messages.create(p, o)
+      (p, o) => streamClient.messages.create(p, o)
     );
     for await (const event of stream2) {
       receivedAnyEvent = true;