import { createAtsAgentTools } from "ats-sdk/agent-tools";
import { generateText, step } from "ai";
import { openai } from "@ai-sdk/openai";

const atsTools = createAtsAgentTools();

const result = await generateText({
  model: openai("gpt-4o"),
  tools: {
    ats_get_jobs: {
      description: atsTools.get_jobs.description,
      parameters: atsTools.get_jobs.parameters,
      execute: async ({ provider, companySlug }) => {
        return atsTools.get_jobs.execute({ provider, companySlug });
      },
    },
    ats_get_job: {
      description: atsTools.get_job.description,
      parameters: atsTools.get_job.parameters,
      execute: async ({ provider, companySlug, jobId }) => {
        return atsTools.get_job.execute({ provider, companySlug, jobId });
      },
    },
  },
  prompt:
    "Find engineering openings at Stripe and check if any are remote in the US.",
});

console.log(result.text);