import { GoogleGenerativeAI } from "../dist/index.js";

const genAI = new GoogleGenerativeAI("YOUR_API");

const model = genAI.getGenerativeModel({ model: "gemini-2.0-flash" });

const callbacks = {
    onData: (chunk) => console.log("Data: ", chunk),
    //onEnd: (fullText) => console.log("Final Text: ", fullText),
    //onError: (error) => console.error("Error: ", error),
  };

  try {
    const stream = await model.generateContentStream("Explain how AI works", {}, callbacks);
  } catch (error) {
    console.error("Stream generation failed:", error);
  }