ai 4.1.55 → 4.1.56

package/dist/index.mjs

@@ -6104,6 +6104,89 @@ function extractReasoningMiddleware({
   };
 }
 
+// core/middleware/simulate-streaming-middleware.ts
+function simulateStreamingMiddleware() {
+  return {
+    middlewareVersion: "v1",
+    wrapStream: async ({ doGenerate }) => {
+      const result = await doGenerate();
+      const simulatedStream = new ReadableStream({
+        start(controller) {
+          controller.enqueue({ type: "response-metadata", ...result.response });
+          if (result.reasoning) {
+            if (typeof result.reasoning === "string") {
+              controller.enqueue({
+                type: "reasoning",
+                textDelta: result.reasoning
+              });
+            } else {
+              for (const reasoning of result.reasoning) {
+                switch (reasoning.type) {
+                  case "text": {
+                    controller.enqueue({
+                      type: "reasoning",
+                      textDelta: reasoning.text
+                    });
+                    if (reasoning.signature != null) {
+                      controller.enqueue({
+                        type: "reasoning-signature",
+                        signature: reasoning.signature
+                      });
+                    }
+                    break;
+                  }
+                  case "redacted": {
+                    controller.enqueue({
+                      type: "redacted-reasoning",
+                      data: reasoning.data
+                    });
+                    break;
+                  }
+                }
+              }
+            }
+          }
+          if (result.text) {
+            controller.enqueue({
+              type: "text-delta",
+              textDelta: result.text
+            });
+          }
+          if (result.toolCalls) {
+            for (const toolCall of result.toolCalls) {
+              controller.enqueue({
+                type: "tool-call-delta",
+                toolCallType: "function",
+                toolCallId: toolCall.toolCallId,
+                toolName: toolCall.toolName,
+                argsTextDelta: toolCall.args
+              });
+              controller.enqueue({
+                type: "tool-call",
+                ...toolCall
+              });
+            }
+          }
+          controller.enqueue({
+            type: "finish",
+            finishReason: result.finishReason,
+            usage: result.usage,
+            logprobs: result.logprobs,
+            providerMetadata: result.providerMetadata
+          });
+          controller.close();
+        }
+      });
+      return {
+        stream: simulatedStream,
+        rawCall: result.rawCall,
+        rawResponse: result.rawResponse,
+        warnings: result.warnings
+      };
+    }
+  };
+}
+
 // core/middleware/wrap-language-model.ts
 var wrapLanguageModel = ({
   model,
@@ -6139,12 +6222,19 @@ var doWrap = ({
     async doGenerate(params) {
       const transformedParams = await doTransform({ params, type: "generate" });
       const doGenerate = async () => model.doGenerate(transformedParams);
-      return wrapGenerate ? wrapGenerate({ doGenerate, params: transformedParams, model }) : doGenerate();
+      const doStream = async () => model.doStream(transformedParams);
+      return wrapGenerate ? wrapGenerate({
+        doGenerate,
+        doStream,
+        params: transformedParams,
+        model
+      }) : doGenerate();
     },
     async doStream(params) {
       const transformedParams = await doTransform({ params, type: "stream" });
+      const doGenerate = async () => model.doGenerate(transformedParams);
       const doStream = async () => model.doStream(transformedParams);
-      return wrapStream ? wrapStream({ doStream, params: transformedParams, model }) : doStream();
+      return wrapStream ? wrapStream({ doGenerate, doStream, params: transformedParams, model }) : doStream();
     }
   };
 };
@@ -7652,6 +7742,7 @@ export {
   processDataStream,
   processTextStream,
   simulateReadableStream,
+  simulateStreamingMiddleware,
   smoothStream,
   streamObject,
   streamText,