ai 7.0.0-beta.113 → 7.0.0-beta.115

package/src/transcribe/transcribe-result.ts

@@ -1,6 +1,6 @@
-import { JSONObject } from '@ai-sdk/provider';
-import { TranscriptionModelResponseMetadata } from '../types/transcription-model-response-metadata';
-import { Warning } from '../types/warning';
+import type { JSONObject } from '@ai-sdk/provider';
+import type { TranscriptionModelResponseMetadata } from '../types/transcription-model-response-metadata';
+import type { Warning } from '../types/warning';
 
 /**
  * The result of a `transcribe` call.

package/src/transcribe/transcribe.ts

@@ -1,21 +1,21 @@
-import { JSONObject } from '@ai-sdk/provider';
-import { ProviderOptions, withUserAgentSuffix } from '@ai-sdk/provider-utils';
+import type { JSONObject } from '@ai-sdk/provider';
+import {
+  detectMediaType,
+  withUserAgentSuffix,
+  type ProviderOptions,
+} from '@ai-sdk/provider-utils';
 import { NoTranscriptGeneratedError } from '../error/no-transcript-generated-error';
 import { logWarnings } from '../logger/log-warnings';
-import { DataContent } from '../prompt';
+import type { DataContent } from '../prompt';
 import { convertDataContentToUint8Array } from '../prompt/data-content';
-import { TranscriptionModel } from '../types/transcription-model';
-import { TranscriptionModelResponseMetadata } from '../types/transcription-model-response-metadata';
-import {
-  audioMediaTypeSignatures,
-  detectMediaType,
-} from '../util/detect-media-type';
+import type { TranscriptionModel } from '../types/transcription-model';
+import type { TranscriptionModelResponseMetadata } from '../types/transcription-model-response-metadata';
 import { createDownload } from '../util/download/create-download';
 import { prepareRetries } from '../util/prepare-retries';
-import { TranscriptionResult } from './transcribe-result';
+import type { TranscriptionResult } from './transcribe-result';
 import { VERSION } from '../version';
 import { resolveTranscriptionModel } from '../model/resolve-model';
-import { Warning } from '../types';
+import type { Warning } from '../types';
 /**
  * Generates transcripts using a transcription model.
  *
@@ -124,7 +124,7 @@ export async function transcribe({
       mediaType:
         detectMediaType({
           data: audioData,
-          signatures: audioMediaTypeSignatures,
+          topLevelType: 'audio',
         }) ?? 'audio/wav',
     }),
   );

package/src/types/embedding-model-middleware.ts

@@ -1,4 +1,4 @@
-import { EmbeddingModelV4Middleware } from '@ai-sdk/provider';
+import type { EmbeddingModelV4Middleware } from '@ai-sdk/provider';
 
 /**
  * Middleware for embedding models.

package/src/types/embedding-model.ts

@@ -1,4 +1,4 @@
-import {
+import type {
   EmbeddingModelV2,
   EmbeddingModelV3,
   EmbeddingModelV4,

package/src/types/image-model-middleware.ts

@@ -1,4 +1,4 @@
-import { ImageModelV4Middleware } from '@ai-sdk/provider';
+import type { ImageModelV4Middleware } from '@ai-sdk/provider';
 
 /**
  * Middleware for image models.

package/src/types/image-model.ts

@@ -1,4 +1,4 @@
-import {
+import type {
   ImageModelV2,
   ImageModelV3,
   ImageModelV4,

package/src/types/json-value.ts

@@ -1,4 +1,4 @@
-import { JSONValue as OriginalJSONValue } from '@ai-sdk/provider';
+import type { JSONValue as OriginalJSONValue } from '@ai-sdk/provider';
 import { z } from 'zod/v4';
 
 export const jsonValueSchema: z.ZodType<JSONValue> = z.lazy(() =>

package/src/types/language-model-middleware.ts

@@ -1,4 +1,4 @@
-import { LanguageModelV4Middleware } from '@ai-sdk/provider';
+import type { LanguageModelV4Middleware } from '@ai-sdk/provider';
 
 /**
  * Middleware for language models.

package/src/types/language-model.ts

@@ -1,5 +1,5 @@
-import { GatewayModelId } from '@ai-sdk/gateway';
-import {
+import type { GatewayModelId } from '@ai-sdk/gateway';
+import type {
   LanguageModelV2,
   LanguageModelV3,
   LanguageModelV4,

package/src/types/provider-metadata.ts

@@ -1,4 +1,4 @@
-import { SharedV4ProviderMetadata } from '@ai-sdk/provider';
+import type { SharedV4ProviderMetadata } from '@ai-sdk/provider';
 import { z } from 'zod/v4';
 import { jsonValueSchema } from './json-value';
 

package/src/types/provider-reference.ts

@@ -1,3 +1,3 @@
-import { SharedV4ProviderReference } from '@ai-sdk/provider';
+import type { SharedV4ProviderReference } from '@ai-sdk/provider';
 
 export type ProviderReference = SharedV4ProviderReference;

package/src/types/provider.ts

@@ -1,7 +1,7 @@
-import { EmbeddingModel } from './embedding-model';
-import { LanguageModel } from './language-model';
-import { ImageModel } from './image-model';
-import { RerankingModel } from './reranking-model';
+import type { EmbeddingModel } from './embedding-model';
+import type { LanguageModel } from './language-model';
+import type { ImageModel } from './image-model';
+import type { RerankingModel } from './reranking-model';
 
 /**
  * Provider for language, text embedding, and image models.

package/src/types/reranking-model.ts

@@ -1,4 +1,4 @@
-import { RerankingModelV3, RerankingModelV4 } from '@ai-sdk/provider';
+import type { RerankingModelV3, RerankingModelV4 } from '@ai-sdk/provider';
 
 /**
  * Reranking model that is used by the AI SDK.

package/src/types/speech-model.ts

@@ -1,4 +1,8 @@
-import { SpeechModelV2, SpeechModelV3, SpeechModelV4 } from '@ai-sdk/provider';
+import type {
+  SpeechModelV2,
+  SpeechModelV3,
+  SpeechModelV4,
+} from '@ai-sdk/provider';
 
 /**
  * Speech model that is used by the AI SDK.

package/src/types/transcription-model.ts

@@ -1,4 +1,4 @@
-import {
+import type {
   TranscriptionModelV2,
   TranscriptionModelV3,
   TranscriptionModelV4,

package/src/types/usage.ts

@@ -1,4 +1,4 @@
-import {
+import type {
   ImageModelV4Usage,
   JSONObject,
   LanguageModelV4Usage,

package/src/types/video-model.ts

@@ -1,4 +1,4 @@
-import {
+import type {
   Experimental_VideoModelV3,
   Experimental_VideoModelV4,
   SharedV4ProviderMetadata,

package/src/types/warning.ts

@@ -1,4 +1,4 @@
-import { SharedV4Warning } from '@ai-sdk/provider';
+import type { SharedV4Warning } from '@ai-sdk/provider';
 
 /**
  * Warning from the model provider for this call. The call will proceed, but e.g.

package/src/ui/call-completion-api.ts

@@ -1,12 +1,12 @@
 import {
   parseJsonEventStream,
-  ParseResult,
   withUserAgentSuffix,
   getRuntimeEnvironmentUserAgent,
+  type ParseResult,
 } from '@ai-sdk/provider-utils';
 import {
-  UIMessageChunk,
   uiMessageChunkSchema,
+  type UIMessageChunk,
 } from '../ui-message-stream/ui-message-chunks';
 import { consumeStream } from '../util/consume-stream';
 import { processTextStream } from './process-text-stream';

package/src/ui/chat-transport.ts

@@ -1,6 +1,6 @@
-import { UIMessageChunk } from '../ui-message-stream';
-import { ChatRequestOptions } from './chat';
-import { UIMessage } from './ui-messages';
+import type { UIMessageChunk } from '../ui-message-stream';
+import type { ChatRequestOptions } from './chat';
+import type { UIMessage } from './ui-messages';
 
 /**
  * Transport interface for handling chat message communication and streaming.

package/src/ui/chat.ts

@@ -1,26 +1,23 @@
 import {
-  FlexibleSchema,
   generateId as generateIdFunc,
-  IdGenerator,
-  InferSchema,
+  type FlexibleSchema,
+  type IdGenerator,
+  type InferSchema,
 } from '@ai-sdk/provider-utils';
-import { FinishReason } from '../types/language-model';
-import { UIMessageChunk } from '../ui-message-stream/ui-message-chunks';
+import type { FinishReason } from '../types/language-model';
+import type { UIMessageChunk } from '../ui-message-stream/ui-message-chunks';
 import { consumeStream } from '../util/consume-stream';
 import { SerialJobExecutor } from '../util/serial-job-executor';
-import { ChatTransport } from './chat-transport';
+import type { ChatTransport } from './chat-transport';
 import { convertFileListToFileUIParts } from './convert-file-list-to-file-ui-parts';
 import { DefaultChatTransport } from './default-chat-transport';
 import {
   createStreamingUIMessageState,
   processUIMessageStream,
-  StreamingUIMessageState,
+  type StreamingUIMessageState,
 } from './process-ui-message-stream';
 import {
-  InferUIMessageToolCall,
   isToolUIPart,
-  UIMessagePart,
-  UITools,
   type DataUIPart,
   type FileUIPart,
   type InferUIMessageData,
@@ -28,8 +25,10 @@ import {
   type InferUIMessageTools,
   type UIDataTypes,
   type UIMessage,
+  type InferUIMessageToolCall,
+  type UIMessagePart,
+  type UITools,
 } from './ui-messages';
-
 export type CreateUIMessage<UI_MESSAGE extends UIMessage> = Omit<
   UI_MESSAGE,
   'id' | 'role'

package/src/ui/convert-file-list-to-file-ui-parts.ts

@@ -1,4 +1,4 @@
-import { FileUIPart } from './ui-messages';
+import type { FileUIPart } from './ui-messages';
 
 export async function convertFileListToFileUIParts(
   files: FileList | undefined,

package/src/ui/convert-to-model-messages.ts

@@ -1,38 +1,37 @@
 import {
-  AssistantContent,
-  CustomPart,
-  FilePart,
   isNonNullable,
-  ModelMessage,
-  TextPart,
-  ToolApprovalResponse,
-  ToolResultPart,
+  type ToolSet,
+  type AssistantContent,
+  type CustomPart,
+  type FilePart,
+  type ModelMessage,
+  type TextPart,
+  type ToolApprovalResponse,
+  type ToolResultPart,
 } from '@ai-sdk/provider-utils';
-import type { ToolSet } from '@ai-sdk/provider-utils';
 import { createToolModelOutput } from '../prompt/create-tool-model-output';
 import { MessageConversionError } from '../prompt/message-conversion-error';
 import {
-  CustomContentUIPart,
-  DataUIPart,
-  DynamicToolUIPart,
-  FileUIPart,
   getToolName,
-  InferUIMessageData,
-  InferUIMessageTools,
   isCustomContentUIPart,
   isDataUIPart,
   isFileUIPart,
   isReasoningFileUIPart,
   isReasoningUIPart,
-  ReasoningFileUIPart,
   isTextUIPart,
   isToolUIPart,
-  ReasoningUIPart,
-  TextUIPart,
-  ToolUIPart,
-  UIMessage,
+  type CustomContentUIPart,
+  type DataUIPart,
+  type DynamicToolUIPart,
+  type FileUIPart,
+  type InferUIMessageData,
+  type InferUIMessageTools,
+  type ReasoningFileUIPart,
+  type ReasoningUIPart,
+  type TextUIPart,
+  type ToolUIPart,
+  type UIMessage,
 } from './ui-messages';
-
 /**
  * Converts an array of UI messages from useChat into an array of ModelMessages that can be used
  * with the AI functions (e.g. `streamText`, `generateText`).
@@ -114,7 +113,13 @@ export async function convertToModelMessages<UI_MESSAGE extends UIMessage>(
                   type: 'file' as const,
                   mediaType: part.mediaType,
                   filename: part.filename,
-                  data: part.providerReference ?? part.url,
+                  data:
+                    part.providerReference != null
+                      ? {
+                          type: 'reference' as const,
+                          reference: part.providerReference,
+                        }
+                      : { type: 'url' as const, url: new URL(part.url) },
                   ...(part.providerMetadata != null
                     ? { providerOptions: part.providerMetadata }
                     : {}),
@@ -176,7 +181,13 @@ export async function convertToModelMessages<UI_MESSAGE extends UIMessage>(
                   type: 'file' as const,
                   mediaType: part.mediaType,
                   filename: part.filename,
-                  data: part.providerReference ?? part.url,
+                  data:
+                    part.providerReference != null
+                      ? {
+                          type: 'reference' as const,
+                          reference: part.providerReference,
+                        }
+                      : { type: 'url' as const, url: new URL(part.url) },
                   ...(part.providerMetadata != null
                     ? { providerOptions: part.providerMetadata }
                     : {}),
@@ -184,7 +195,7 @@ export async function convertToModelMessages<UI_MESSAGE extends UIMessage>(
               } else if (isReasoningFileUIPart(part)) {
                 content.push({
                   type: 'reasoning-file' as const,
-                  data: part.url,
+                  data: { type: 'url' as const, url: new URL(part.url) },
                   mediaType: part.mediaType,
                   providerOptions: part.providerMetadata,
                 });

package/src/ui/default-chat-transport.ts

@@ -1,13 +1,13 @@
-import { parseJsonEventStream, ParseResult } from '@ai-sdk/provider-utils';
+import { parseJsonEventStream, type ParseResult } from '@ai-sdk/provider-utils';
 import {
-  UIMessageChunk,
   uiMessageChunkSchema,
+  type UIMessageChunk,
 } from '../ui-message-stream/ui-message-chunks';
 import {
   HttpChatTransport,
-  HttpChatTransportInitOptions,
+  type HttpChatTransportInitOptions,
 } from './http-chat-transport';
-import { UIMessage } from './ui-messages';
+import type { UIMessage } from './ui-messages';
 
 export class DefaultChatTransport<
   UI_MESSAGE extends UIMessage,

package/src/ui/direct-chat-transport.ts

@@ -1,11 +1,11 @@
-import { Agent } from '../agent/agent';
-import { Output } from '../generate-text/output';
-import { UIMessageStreamOptions } from '../generate-text/stream-text-result';
+import type { Agent } from '../agent/agent';
+import type { Output } from '../generate-text/output';
+import type { UIMessageStreamOptions } from '../generate-text/stream-text-result';
 import type { Context, ToolSet } from '@ai-sdk/provider-utils';
-import { UIMessageChunk } from '../ui-message-stream/ui-message-chunks';
-import { ChatTransport } from './chat-transport';
+import type { UIMessageChunk } from '../ui-message-stream/ui-message-chunks';
+import type { ChatTransport } from './chat-transport';
 import { convertToModelMessages } from './convert-to-model-messages';
-import { InferUITools, UIMessage } from './ui-messages';
+import type { InferUITools, UIMessage } from './ui-messages';
 import { validateUIMessages } from './validate-ui-messages';
 
 /**

package/src/ui/http-chat-transport.ts

@@ -1,12 +1,12 @@
 import {
-  FetchFunction,
-  Resolvable,
   normalizeHeaders,
   resolve,
+  type FetchFunction,
+  type Resolvable,
 } from '@ai-sdk/provider-utils';
-import { UIMessageChunk } from '../ui-message-stream/ui-message-chunks';
-import { ChatTransport } from './chat-transport';
-import { UIMessage } from './ui-messages';
+import type { UIMessageChunk } from '../ui-message-stream/ui-message-chunks';
+import type { ChatTransport } from './chat-transport';
+import type { UIMessage } from './ui-messages';
 
 export type PrepareSendMessagesRequest<UI_MESSAGE extends UIMessage> = (
   options: {

package/src/ui/last-assistant-message-is-complete-with-approval-responses.ts

@@ -1,5 +1,4 @@
 import { isToolUIPart, type UIMessage } from './ui-messages';
-
 /**
  * Check if the last message is an assistant message with completed tool call approvals.
  * The last step of the message must have at least one tool approval response and

package/src/ui/last-assistant-message-is-complete-with-tool-calls.ts

@@ -1,5 +1,4 @@
 import { isToolUIPart, type UIMessage } from './ui-messages';
-
 /**
  * Check if the last message is an assistant message with completed tool calls.
  * The last step of the message must have at least one tool invocation and

package/src/ui/process-ui-message-stream.ts

@@ -1,35 +1,34 @@
-import { FlexibleSchema, validateTypes } from '@ai-sdk/provider-utils';
+import { validateTypes, type FlexibleSchema } from '@ai-sdk/provider-utils';
 import { UIMessageStreamError } from '../error/ui-message-stream-error';
-import { ProviderMetadata } from '../types';
-import { FinishReason } from '../types/language-model';
+import type { ProviderMetadata } from '../types';
+import type { FinishReason } from '../types/language-model';
 import {
-  DataUIMessageChunk,
-  InferUIMessageChunk,
   isDataUIMessageChunk,
-  UIMessageChunk,
+  type DataUIMessageChunk,
+  type InferUIMessageChunk,
+  type UIMessageChunk,
 } from '../ui-message-stream/ui-message-chunks';
-import { ErrorHandler } from '../util/error-handler';
+import type { ErrorHandler } from '../util/error-handler';
 import { mergeObjects } from '../util/merge-objects';
 import { parsePartialJson } from '../util/parse-partial-json';
-import { UIDataTypesToSchemas } from './chat';
+import type { UIDataTypesToSchemas } from './chat';
 import {
-  CustomContentUIPart,
-  DataUIPart,
-  DynamicToolUIPart,
   getStaticToolName,
-  InferUIMessageData,
-  InferUIMessageMetadata,
-  InferUIMessageToolCall,
-  InferUIMessageTools,
   isStaticToolUIPart,
   isToolUIPart,
-  ReasoningUIPart,
-  TextUIPart,
-  ToolUIPart,
-  UIMessage,
-  UIMessagePart,
+  type CustomContentUIPart,
+  type DataUIPart,
+  type DynamicToolUIPart,
+  type InferUIMessageData,
+  type InferUIMessageMetadata,
+  type InferUIMessageToolCall,
+  type InferUIMessageTools,
+  type ReasoningUIPart,
+  type TextUIPart,
+  type ToolUIPart,
+  type UIMessage,
+  type UIMessagePart,
 } from './ui-messages';
-
 export type StreamingUIMessageState<UI_MESSAGE extends UIMessage> = {
   message: UI_MESSAGE;
   activeTextParts: Record<string, TextUIPart>;

package/src/ui/text-stream-chat-transport.ts

@@ -1,10 +1,10 @@
-import { UIMessageChunk } from '../ui-message-stream/ui-message-chunks';
+import type { UIMessageChunk } from '../ui-message-stream/ui-message-chunks';
 import {
   HttpChatTransport,
-  HttpChatTransportInitOptions,
+  type HttpChatTransportInitOptions,
 } from './http-chat-transport';
 import { transformTextToUiMessageStream } from './transform-text-to-ui-message-stream';
-import { UIMessage } from './ui-messages';
+import type { UIMessage } from './ui-messages';
 
 export class TextStreamChatTransport<
   UI_MESSAGE extends UIMessage,

package/src/ui/transform-text-to-ui-message-stream.ts

@@ -1,4 +1,4 @@
-import { UIMessageChunk } from '../ui-message-stream/ui-message-chunks';
+import type { UIMessageChunk } from '../ui-message-stream/ui-message-chunks';
 
 export function transformTextToUiMessageStream({
   stream,

package/src/ui/ui-messages.ts

@@ -1,14 +1,14 @@
-import {
+import type {
   InferToolInput,
   InferToolOutput,
   Tool,
   ToolCall,
   ToolSet,
 } from '@ai-sdk/provider-utils';
-import { ProviderMetadata } from '../types/provider-metadata';
-import { ProviderReference } from '../types/provider-reference';
-import { DeepPartial } from '../util/deep-partial';
-import { ValueOf } from '../util/value-of';
+import type { ProviderMetadata } from '../types/provider-metadata';
+import type { ProviderReference } from '../types/provider-reference';
+import type { DeepPartial } from '../util/deep-partial';
+import type { ValueOf } from '../util/value-of';
 
 /**
  * The data types that can be used in the UI message for the UI message data parts.
@@ -180,7 +180,14 @@ export type FileUIPart = {
   type: 'file';
 
   /**
-   * IANA media type of the file.
+   * Either a full IANA media type (`type/subtype`, e.g. `image/png`) or just
+   * the top-level IANA segment (e.g. `image`, `audio`, `video`, `text`).
+   *
+   * `*`-subtype wildcards (e.g. `image/*`) are normalized as equivalent to the
+   * top-level segment alone (e.g. `image`). Providers can use the helpers in
+   * `@ai-sdk/provider-utils` (`isFullMediaType`, `getTopLevelMediaType`,
+   * `detectMediaType`) to resolve the field according to their API
+   * requirements.
    *
    * @see https://www.iana.org/assignments/media-types/media-types.xhtml
    */

package/src/ui/use-completion.ts

@@ -1,4 +1,4 @@
-import { FetchFunction } from '@ai-sdk/provider-utils';
+import type { FetchFunction } from '@ai-sdk/provider-utils';
 
 export type CompletionRequestOptions = {
   /**

package/src/ui/validate-ui-messages.ts

@@ -1,15 +1,15 @@
 import { TypeValidationError } from '@ai-sdk/provider';
 import {
-  FlexibleSchema,
   lazySchema,
-  Tool,
   validateTypes,
   zodSchema,
+  type FlexibleSchema,
+  type Tool,
 } from '@ai-sdk/provider-utils';
 import { z } from 'zod/v4';
 import { InvalidArgumentError } from '../error';
 import { providerMetadataSchema } from '../types/provider-metadata';
-import {
+import type {
   DataUIPart,
   InferUIMessageData,
   InferUIMessageTools,

package/src/ui-message-stream/create-ui-message-stream-response.ts

@@ -1,8 +1,8 @@
 import { prepareHeaders } from '../util/prepare-headers';
 import { JsonToSseTransformStream } from './json-to-sse-transform-stream';
 import { UI_MESSAGE_STREAM_HEADERS } from './ui-message-stream-headers';
-import { UIMessageChunk } from './ui-message-chunks';
-import { UIMessageStreamResponseInit } from './ui-message-stream-response-init';
+import type { UIMessageChunk } from './ui-message-chunks';
+import type { UIMessageStreamResponseInit } from './ui-message-stream-response-init';
 
 /**
  * Creates a Response object from a UI message stream.

package/src/ui-message-stream/create-ui-message-stream.ts

@@ -1,14 +1,14 @@
 import {
   generateId as generateIdFunc,
   getErrorMessage,
-  IdGenerator,
+  type IdGenerator,
 } from '@ai-sdk/provider-utils';
-import { UIMessage } from '../ui/ui-messages';
+import type { UIMessage } from '../ui/ui-messages';
 import { handleUIMessageStreamFinish } from './handle-ui-message-stream-finish';
-import { InferUIMessageChunk } from './ui-message-chunks';
-import { UIMessageStreamOnFinishCallback } from './ui-message-stream-on-finish-callback';
-import { UIMessageStreamOnStepFinishCallback } from './ui-message-stream-on-step-finish-callback';
-import { UIMessageStreamWriter } from './ui-message-stream-writer';
+import type { InferUIMessageChunk } from './ui-message-chunks';
+import type { UIMessageStreamOnFinishCallback } from './ui-message-stream-on-finish-callback';
+import type { UIMessageStreamOnStepFinishCallback } from './ui-message-stream-on-step-finish-callback';
+import type { UIMessageStreamWriter } from './ui-message-stream-writer';
 
 /**
  * Creates a UI message stream that can be used to send messages to the client.

package/src/ui-message-stream/get-response-ui-message-id.ts

@@ -1,5 +1,5 @@
-import { IdGenerator } from '@ai-sdk/provider-utils';
-import { UIMessage } from '../ui/ui-messages';
+import type { IdGenerator } from '@ai-sdk/provider-utils';
+import type { UIMessage } from '../ui/ui-messages';
 
 /**
  * Determines the message ID to use for a response message.

package/src/ui-message-stream/handle-ui-message-stream-finish.ts

@@ -1,13 +1,13 @@
 import {
   createStreamingUIMessageState,
   processUIMessageStream,
-  StreamingUIMessageState,
+  type StreamingUIMessageState,
 } from '../ui/process-ui-message-stream';
-import { UIMessage } from '../ui/ui-messages';
-import { ErrorHandler } from '../util/error-handler';
-import { InferUIMessageChunk, UIMessageChunk } from './ui-message-chunks';
-import { UIMessageStreamOnFinishCallback } from './ui-message-stream-on-finish-callback';
-import { UIMessageStreamOnStepFinishCallback } from './ui-message-stream-on-step-finish-callback';
+import type { UIMessage } from '../ui/ui-messages';
+import type { ErrorHandler } from '../util/error-handler';
+import type { InferUIMessageChunk, UIMessageChunk } from './ui-message-chunks';
+import type { UIMessageStreamOnFinishCallback } from './ui-message-stream-on-finish-callback';
+import type { UIMessageStreamOnStepFinishCallback } from './ui-message-stream-on-step-finish-callback';
 
 export function handleUIMessageStreamFinish<UI_MESSAGE extends UIMessage>({
   messageId,