updates

vercel · Dec 2, 2024 · 9789507 · 9789507
1 parent 103da13
commit 9789507
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 14 deletions.
diff --git a/content/cookbook/01-next/122-caching-middleware.mdx b/content/cookbook/01-next/122-caching-middleware.mdx
@@ -104,10 +104,9 @@ export const cacheMiddleware: LanguageModelV1Middleware = {
     }
 
     // If not cached, proceed with streaming
-    const { stream, rawCall, rawResponse, request, warnings } =
-      await doStream();
+    const { stream, ...rest } = await doStream();
 
-    const fullResponse: unknown[] = [];
+    const fullResponse: LanguageModelV1StreamPart[] = [];
 
     const transformStream = new TransformStream<
       LanguageModelV1StreamPart,
@@ -125,10 +124,7 @@ export const cacheMiddleware: LanguageModelV1Middleware = {
 
     return {
       stream: stream.pipeThrough(transformStream),
-      rawCall,
-      rawResponse,
-      request,
-      warnings,
+      ...rest,
     };
   },
 };

diff --git a/content/docs/06-advanced/04-caching.mdx b/content/docs/06-advanced/04-caching.mdx
@@ -58,10 +58,9 @@ export const cacheMiddleware: LanguageModelV1Middleware = {
     }
 
     // If not cached, proceed with streaming
-    const { stream, rawCall, rawResponse, request, warnings } =
-      await doStream();
+    const { stream, ...rest } = await doStream();
 
-    const fullResponse: unknown[] = [];
+    const fullResponse: LanguageModelV1StreamPart[] = [];
 
     const transformStream = new TransformStream<
       LanguageModelV1StreamPart,
@@ -79,10 +78,7 @@ export const cacheMiddleware: LanguageModelV1Middleware = {
 
     return {
       stream: stream.pipeThrough(transformStream),
-      rawCall,
-      rawResponse,
-      request,
-      warnings,
+      ...rest,
     };
   },
 };