fix (ai/core): correct types in streamObject (#2208)

vercel · Jul 8, 2024 · 915e93f · 915e93f
1 parent 6a11cfa
commit 915e93f
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/packages/core/core/generate-object/stream-object.ts b/packages/core/core/generate-object/stream-object.ts
@@ -147,7 +147,10 @@ Warnings from the model provider (e.g. unsupported settings).
   }
 
   let callOptions: LanguageModelV1CallOptions;
-  let transformer: Transformer<LanguageModelV1StreamPart>;
+  let transformer: Transformer<
+    LanguageModelV1StreamPart,
+    string | Omit<LanguageModelV1StreamPart, 'text-delta'>
+  >;
 
   switch (mode) {
     case 'json': {
@@ -343,7 +346,9 @@ Response headers.
     schema,
     onFinish,
   }: {
-    stream: ReadableStream<string | ObjectStreamInputPart>;
+    stream: ReadableStream<
+      string | Omit<LanguageModelV1StreamPart, 'text-delta'>
+    >;
     warnings: CallWarning[] | undefined;
     rawResponse?: {
       headers?: Record<string, string>;