ai
Version:
AI SDK by Vercel - The AI Toolkit for TypeScript and JavaScript
247 lines (199 loc) • 5.13 kB
text/mdx
---
title: UIMessage
description: API Reference for UIMessage
---
# `UIMessage`
`UIMessage` serves as the source of truth for your application's state, representing the complete message history including metadata, data parts, and all contextual information. In contrast to `ModelMessage`, which represents the state or context passed to the model, `UIMessage` contains the full application state needed for UI rendering and client-side functionality.
## Type Safety
`UIMessage` is designed to be type-safe and accepts three generic parameters to ensure proper typing throughout your application:
1. **`METADATA`** - Custom metadata type for additional message information
2. **`DATA_PARTS`** - Custom data part types for structured data components
3. **`TOOLS`** - Tool definitions for type-safe tool interactions
## Creating Your Own UIMessage Type
Here's an example of how to create a custom typed UIMessage for your application:
```typescript
import { InferUITools, ToolSet, UIMessage, tool } from 'ai';
import z from 'zod';
const metadataSchema = z.object({
someMetadata: z.string().datetime(),
});
type MyMetadata = z.infer<typeof metadataSchema>;
const dataPartSchema = z.object({
someDataPart: z.object({}),
anotherDataPart: z.object({}),
});
type MyDataPart = z.infer<typeof dataPartSchema>;
const tools = {
someTool: tool({}),
} satisfies ToolSet;
type MyTools = InferUITools<typeof tools>;
export type MyUIMessage = UIMessage<MyMetadata, MyDataPart, MyTools>;
```
## `UIMessage` Interface
```typescript
interface UIMessage<
METADATA = unknown,
DATA_PARTS extends UIDataTypes = UIDataTypes,
TOOLS extends UITools = UITools,
> {
/**
* A unique identifier for the message.
*/
id: string;
/**
* The role of the message.
*/
role: 'system' | 'user' | 'assistant';
/**
* The metadata of the message.
*/
metadata?: METADATA;
/**
* The parts of the message. Use this for rendering the message in the UI.
*/
parts: Array<UIMessagePart<DATA_PARTS, TOOLS>>;
}
```
## `UIMessagePart` Types
### `TextUIPart`
A text part of a message.
```typescript
type TextUIPart = {
type: 'text';
/**
* The text content.
*/
text: string;
/**
* The state of the text part.
*/
state?: 'streaming' | 'done';
};
```
### `ReasoningUIPart`
A reasoning part of a message.
```typescript
type ReasoningUIPart = {
type: 'reasoning';
/**
* The reasoning text.
*/
text: string;
/**
* The state of the reasoning part.
*/
state?: 'streaming' | 'done';
/**
* The provider metadata.
*/
providerMetadata?: Record<string, any>;
};
```
### `ToolUIPart`
A tool part of a message that represents tool invocations and their results.
<Note>
The type is based on the name of the tool (e.g., `tool-someTool` for a tool
named `someTool`).
</Note>
```typescript
type ToolUIPart<TOOLS extends UITools = UITools> = ValueOf<{
[NAME in keyof TOOLS & string]: {
type: `tool-${NAME}`;
toolCallId: string;
} & (
| {
state: 'input-streaming';
input: DeepPartial<TOOLS[NAME]['input']> | undefined;
providerExecuted?: boolean;
output?: never;
errorText?: never;
}
| {
state: 'input-available';
input: TOOLS[NAME]['input'];
providerExecuted?: boolean;
output?: never;
errorText?: never;
}
| {
state: 'output-available';
input: TOOLS[NAME]['input'];
output: TOOLS[NAME]['output'];
errorText?: never;
providerExecuted?: boolean;
}
| {
state: 'output-error';
input: TOOLS[NAME]['input'];
output?: never;
errorText: string;
providerExecuted?: boolean;
}
);
}>;
```
### `SourceUrlUIPart`
A source URL part of a message.
```typescript
type SourceUrlUIPart = {
type: 'source-url';
sourceId: string;
url: string;
title?: string;
providerMetadata?: Record<string, any>;
};
```
### `SourceDocumentUIPart`
A document source part of a message.
```typescript
type SourceDocumentUIPart = {
type: 'source-document';
sourceId: string;
mediaType: string;
title: string;
filename?: string;
providerMetadata?: Record<string, any>;
};
```
### `FileUIPart`
A file part of a message.
```typescript
type FileUIPart = {
type: 'file';
/**
* IANA media type of the file.
*/
mediaType: string;
/**
* Optional filename of the file.
*/
filename?: string;
/**
* The URL of the file.
* It can either be a URL to a hosted file or a Data URL.
*/
url: string;
};
```
### `DataUIPart`
A data part of a message for custom data types.
<Note>
The type is based on the name of the data part (e.g., `data-someDataPart` for
a data part named `someDataPart`).
</Note>
```typescript
type DataUIPart<DATA_TYPES extends UIDataTypes> = ValueOf<{
[NAME in keyof DATA_TYPES & string]: {
type: `data-${NAME}`;
id?: string;
data: DATA_TYPES[NAME];
};
}>;
```
### `StepStartUIPart`
A step boundary part of a message.
```typescript
type StepStartUIPart = {
type: 'step-start';
};
```