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,
	});

	console.log(result.choices[0]?.text);
}

run();