# Save Messages To Database
Sometimes conversations with language models can get interesting and you might want to save the state of so you can revisit it or continue the conversation later.
`createAI` has an experimental callback function called `onSetAIState` that gets called whenever the AI state changes. You can use this to save the AI state to a file or a database.
## Client
```tsx filename='app/layout.tsx'
import { ServerMessage } from './actions';
import { AI } from './ai';
export default function RootLayout({
children,
}: Readonly<{
children: React.ReactNode;
}>) {
// get chat history from database
const history: ServerMessage[] = getChat();
return (
{children}
);
}
```
```tsx filename='app/page.tsx'
'use client';
import { useState } from 'react';
import { ClientMessage } from './actions';
import { useActions, useUIState } from '@ai-sdk/rsc';
import { generateId } from 'ai';
// Allow streaming responses up to 30 seconds
export const maxDuration = 30;
export default function Home() {
const [input, setInput] = useState('');
const [conversation, setConversation] = useUIState();
const { continueConversation } = useActions();
return (
{conversation.map((message: ClientMessage) => (
{message.role}: {message.display}
))}
{
setInput(event.target.value);
}}
/>
);
}
```
## Server
We will use the callback function to listen to state changes and save the conversation once we receive a `done` event.
```tsx filename='app/actions.tsx'
'use server';
import { getAIState, getMutableAIState, streamUI } from '@ai-sdk/rsc';
import { openai } from '@ai-sdk/openai';
import { ReactNode } from 'react';
import { z } from 'zod';
import { generateId } from 'ai';
import { Stock } from '@ai-studio/components/stock';
export interface ServerMessage {
role: 'user' | 'assistant' | 'function';
content: string;
}
export interface ClientMessage {
id: string;
role: 'user' | 'assistant' | 'function';
display: ReactNode;
}
export async function continueConversation(
input: string,
): Promise {
'use server';
const history = getMutableAIState();
const result = await streamUI({
model: openai('gpt-3.5-turbo'),
messages: [...history.get(), { role: 'user', content: input }],
text: ({ content, done }) => {
if (done) {
history.done([
...history.get(),
{ role: 'user', content: input },
{ role: 'assistant', content },
]);
}
return
{content}
;
},
tools: {
showStockInformation: {
description:
'Get stock information for symbol for the last numOfMonths months',
inputSchema: z.object({
symbol: z
.string()
.describe('The stock symbol to get information for'),
numOfMonths: z
.number()
.describe('The number of months to get historical information for'),
}),
generate: async ({ symbol, numOfMonths }) => {
history.done([
...history.get(),
{
role: 'function',
name: 'showStockInformation',
content: JSON.stringify({ symbol, numOfMonths }),
},
]);
return ;
},
},
},
});
return {
id: generateId(),
role: 'assistant',
display: result.value,
};
}
```
```ts filename='app/ai.ts'
import { createAI } from '@ai-sdk/rsc';
import { ServerMessage, ClientMessage, continueConversation } from './actions';
export const AI = createAI({
actions: {
continueConversation,
},
onSetAIState: async ({ state, done }) => {
'use server';
if (done) {
saveChat(state);
}
},
onGetUIState: async () => {
'use server';
const history: ServerMessage[] = getAIState();
return history.map(({ role, content }) => ({
id: generateId(),
role,
display:
role === 'function' ? : content,
}));
},
});
```
## Navigation
- [Generate Text](/cookbook/rsc/generate-text)
- [Generate Text with Chat Prompt](/cookbook/rsc/generate-text-with-chat-prompt)
- [Stream Text](/cookbook/rsc/stream-text)
- [Stream Text with Chat Prompt](/cookbook/rsc/stream-text-with-chat-prompt)
- [Generate Object](/cookbook/rsc/generate-object)
- [Stream Object](/cookbook/rsc/stream-object)
- [Call Tools](/cookbook/rsc/call-tools)
- [Call Tools in Parallel](/cookbook/rsc/call-tools-in-parallel)
- [Save Messages To Database](/cookbook/rsc/save-messages-to-database)
- [Restore Messages From Database](/cookbook/rsc/restore-messages-from-database)
- [Render Visual Interface in Chat](/cookbook/rsc/render-visual-interface-in-chat)
- [Stream Updates to Visual Interfaces](/cookbook/rsc/stream-updates-to-visual-interfaces)
- [Record Token Usage after Streaming User Interfaces](/cookbook/rsc/stream-ui-record-token-usage)
[Full Sitemap](/sitemap.md)