import { ChatOpenAI } from '@langchain/openai';
import { z } from 'zod';

// Define your tools with Zod schemas
const GetWeather = {
  name: "GetWeather",
  description: "Get the current weather in a given location",
  schema: z.object({
    location: z.string().describe("The city and state, e.g. San Francisco, CA")
  }),
};

const GetPopulation = {
  name: "GetPopulation",
  description: "Get the current population in a given location",
  schema: z.object({
    location: z.string().describe("The city and state, e.g. San Francisco, CA")
  }),
};

// Instantiate the ChatOpenAI model
const llm = new ChatOpenAI({
  model: "gpt-4o",
  temperature: 0,
  maxTokens: undefined,
  timeout: undefined,
  maxRetries: 2,
});

// Bind tools with strict schema enforcement
const llmWithTools = llm.bindTools(
  [GetWeather, GetPopulation],
  {
    strict: true  // enforce tool args schema is respected
  }
);

// Invoke the model with a query
const aiMsg = await llmWithTools.invoke(
  "Which city is hotter today and which is bigger: LA or NY?"
);
console.log(aiMsg.tool_calls);