import { expect, test } from "bun:test";
import { FireworksAI } from "./src";

const fireworksAI = new FireworksAI({
    bearerAuth: process.env.FIREWORKS_API_KEY,
});

test("Generate an image from a prompt", async () => {
    const result = await fireworksAI.images.generateFromPrompt({
        accountId: "fireworks",
        modelId: "stable-diffusion-xl-1024-v1-0",
        bodyText2imageGen: {
            prompt: "A futuristic cityscape",
            negativePrompt: "cloudy day",
            height: 1024,
            width: 1024,
            cfgScale: 7,
            sampler: "K_EULER",
            samples: 1,
            seed: 0,
            steps: 30,
            safetyCheck: false,
        },
    });

    expect(result).toBeDefined();
});

test("Create a chat completion", async () => {

    const result = await fireworksAI.completions.createChatCompletion({
        model: "accounts/fireworks/models/llama-v3p1-8b-instruct",
        messages: [
            {
                role: "user",
                content: "Hello, how are you?",
            },
        ],
        
    });
    expect(result.choices).toBeDefined();
    expect(result.choices[0].message).toBeDefined();
    
});