import { FireworksAI } from "@simplesagar/fireworksai";

const fireworksAI = new FireworksAI({
	apiKey: process.env["FIREWORKS_API_KEY"] ?? "",
});

async function run() {
	const result = await fireworksAI.completions.create({
		model: "accounts/fireworks/models/llama-v3p1-405b-instruct",
		prompt: "The sky is",
		maxTokens: 160,
		temperature: 1,
		topP: 1,
		topK: 50,
		n: 1,
		stream: true,
	});

	for await (const event of result) {
		process.stdout.write(event.data.choices[0]?.text || "");
	}
}

run();