Tool calling

import OpenAI from "openai";

const client = new OpenAI({
  apiKey: process.env.LIMITLESS_API_KEY,
  baseURL: process.env.LIMITLESS_BASE_URL,
  maxRetries: 0,
});

const messages = [{ role: "user", content: "Is the blue notebook in stock?" }];
const tools = [{
  type: "function",
  function: {
    name: "check_stock",
    description: "Check a product's available quantity.",
    parameters: {
      type: "object",
      properties: { product: { type: "string" } },
      required: ["product"],
      additionalProperties: false,
    },
  },
}];

const first = await client.chat.completions.create({
  model: process.env.LIMITLESS_MODEL,
  messages,
  tools,
  tool_choice: "auto",
  max_tokens: 512,
});

const assistant = first.choices[0]?.message;
if (!assistant) throw new Error("No assistant message returned");
messages.push(assistant);

for (const call of assistant.tool_calls ?? []) {
  if (call.type !== "function" || call.function.name !== "check_stock") {
    throw new Error("Unexpected tool request");
  }
  const args = JSON.parse(call.function.arguments);
  if (typeof args.product !== "string" || args.product.length > 100) {
    throw new Error("Invalid tool arguments");
  }
  // Example fixture. Replace with your authorized inventory lookup.
  const stock = { product: args.product, quantity: 12 };
  messages.push({ role: "tool", tool_call_id: call.id, content: JSON.stringify(stock) });
}

if (assistant.tool_calls?.length) {
  const final = await client.chat.completions.create({
    model: process.env.LIMITLESS_MODEL,
    messages,
    max_tokens: 512,
  });
  console.log(final.choices[0]?.message.content);
} else {
  console.log(assistant.content);
}