import { load } from "@std/dotenv";

const env = await load({ envPath: ".env" });

const mistralApiKey = env["MISTRAL_APIKEY"];
const mistralApiUrl = "https://api.mistral.ai/v1/chat/completions";

export const callLLM = async (prompt: string) => {
  const response = await fetch(mistralApiUrl, {
    method: "POST",
    headers: {
      "Content-Type": "application/json",
      Authorization: `Bearer ${mistralApiKey}`,
    },
    body: JSON.stringify({
      model: "mistral-small-latest",
      response_format: { type: "json_object" },
      messages: [{ role: "user", content: prompt }],
    }),
  });

  const data = await response.json();
  return data.choices[0].message.content;
};