import { openai } from '@ai-sdk/openai'; import { generateId, type ModelMessage } from 'ai'; import { createStreamableValue, getMutableAIState as $getMutableAIState, streamUI, type createAI, } from '@ai-sdk/rsc'; import { Message, BotMessage } from './message'; import { z } from 'zod'; type AIProviderNoActions = ReturnType>; // typed wrapper *without* actions defined to avoid circular dependencies const getMutableAIState = $getMutableAIState; // mock function to fetch weather data const fetchWeatherData = async (location: string) => { await new Promise(resolve => setTimeout(resolve, 1000)); return { temperature: '72°F' }; }; export async function submitUserMessage(content: string) { 'use server'; const aiState = getMutableAIState(); aiState.update({ ...aiState.get(), messages: [ ...aiState.get().messages, { id: generateId(), role: 'user', content }, ], }); let textStream: undefined | ReturnType>; let textNode: React.ReactNode; const result = await streamUI({ model: openai('gpt-5-mini'), initial: Working on that..., instructions: 'You are a weather assistant.', messages: aiState .get() .messages.map(({ role, content }) => ({ role, content }) as ModelMessage), text: ({ content, done, delta }) => { if (!textStream) { textStream = createStreamableValue(''); textNode = ; } if (done) { textStream.done(); aiState.update({ ...aiState.get(), messages: [ ...aiState.get().messages, { id: generateId(), role: 'assistant', content }, ], }); } else { textStream.append(delta); } return textNode; }, tools: { get_current_weather: { description: 'Get the current weather', inputSchema: z.object({ location: z.string(), }), generate: async function* ({ location }) { yield ( Loading weather for {location} ); const { temperature } = await fetchWeatherData(location); return ( The temperature in {location} is{' '} {temperature} ); }, }, }, onFinish: event => { // your own logic, e.g. for saving the chat history or recording usage console.log(`[onFinish]: ${JSON.stringify(event, null, 2)}`); }, }); return { id: generateId(), display: result.value, }; } export type ClientMessage = ModelMessage & { id: string; }; export type AIState = { chatId: string; messages: ClientMessage[]; }; export type UIState = { id: string; display: React.ReactNode; }[];