run_model.js

import { AzureOpenAI } from "openai";
import dotenv from "dotenv";

dotenv.config();

export async function main() {
  // Load environment variables
  const endpoint = process.env["AZURE_OPENAI_ENDPOINT"];
  const apiKey = process.env["AZURE_OPENAI_API_KEY"];
  const apiVersion = "2025-01-01-preview";
  const deployment = "gpt-35-turbo"; // This must match your deployment name
  const model = process.env["MODEL_NAME"];

  // Check for missing environment variables
  if (!endpoint || !apiKey || apiKey === "<REPLACE_WITH_YOUR_KEY_VALUE_HERE>") {
    console.error("Error: AZURE_OPENAI_ENDPOINT and AZURE_OPENAI_API_KEY must be set in your .env file.");
    return;
  }

  // Get user prompt from command line arguments
  const userPrompt = process.argv[2];

  if (!userPrompt) {
    console.error("Error: Please provide a prompt as a command-line argument.");
    console.error('Usage: node run_model.js "Your prompt here"');
    return;
  }

  const client = new AzureOpenAI({ endpoint, apiKey, apiVersion, deployment });

  // In-memory conversation history
  const messages = [
    { role: "system", content: "You are a helpful assistant." },
    { role: "user", content: userPrompt },
  ];

  console.log(`Sending prompt to Azure OpenAI: "${userPrompt}"`);

  const result = await client.chat.completions.create({
    model : model,
    messages: messages,
    max_tokens: 800,
    temperature: 0.7,
  });

  console.log("Assistant:", result.choices[0].message.content);
}

main().catch((err) => {
  console.error("The sample encountered an error:", err);
});