import { streamText } from "ai";
import { openai } from "@ai-sdk/openai";

const model = openai("gpt-4o-mini");

const result = streamText({
  model,
  prompt: "Explain quantum entanglement in simple terms.",
});

for await (const textDelta of result.textStream) {
  process.stdout.write(textDelta);
}

const result = streamText({
  model,
  system: "You are a helpful assistant.",
  messages: [
    { role: "user", content: "Tell me a short story about a robot." },
  ],
});

for await (const chunk of result.textStream) {
  process.stdout.write(chunk);
}

const result = streamText({ model, prompt: "Summarise the French Revolution." });

// Stream while displaying
for await (const chunk of result.textStream) {
  process.stdout.write(chunk);
}

// Access the full text once done
const fullText = await result.text;
console.log("\n\nFull:", fullText);