const PayloadSchema = z.object({
 model: z.string().default("gpt-3.5-turbo"),
 prompt: z.string().default("Hello, how are you?"),
});
export const agentTask = schemaTask({
 id: "agent-task",
 schema: PayloadSchema,
 run: async (payload) => {
 const model = openai(payload.model);
 let messages: CoreMessage[] = [
 {
 role: "user",
 content: payload.prompt,
 }
 ];
 let iteration = 0;
 while (true) {
 iteration++;
 const result = streamText({
 model,
 messages,
 });
 await metadata.stream(`openai.${iteration}`, result.fullStream);
 const systemResponse = await result.text;
 messages.push({
 role: "system",
 content: systemResponse,
 });
 // This token will expire in 1 day
 const token = await wait.createToken({ timeout: "1d" });
 metadata.set("resumeToken", token); // So we can access the token through Realtime
 // Now wait for the next message
 const nextMessage = await wait.forToken<CoreUserMessage>(token);
 if (nextMessage.ok) {
 messages.push(nextMessage.data);
 } else {
 break;
 }
 }
 },
});

// Define our context type for this specific task
interface ChatContext {
 messages: CoreMessage[];
}
export const agentTask = schemaTask({
 id: "agent-task",
 schema: PayloadSchema,
 run: async (payload) => {
 const model = openai(payload.model);
 // Initial context with starting messages
 const initialContext: ChatContext = {
 messages: [
 {
 role: "user",
 content: payload.prompt,
 },
 ],
 };
 // Use the reusable conversation loop with explicit generic types and anonymous functions
 const finalContext = await agentLoop<ChatContext, CoreUserMessage>(
 initialContext,
 async (context, iteration) => {
 const result = streamText({
 model,
 messages: context.messages,
 });
 await metadata.stream(`openai.${iteration}`, result.fullStream);
 const systemResponse = await result.text;
 return {
 ...context,
 messages: [
 ...context.messages,
 {
 role: "system",
 content: systemResponse,
 },
 ],
 };
 },
 (context, userMessage) => ({
 ...context,
 messages: [...context.messages, userMessage],
 })
 );
 // Optionally do something with the final context
 return finalContext;
 },
});