import { Factory } from "fishery";
import { ObjectId } from "mongodb";
import { faker } from "@faker-js/faker";
import type { Agent } from "../../../talkpilot";

export const agentFactory = Factory.define<Agent>(() => ({
  name: faker.person.fullName(),
  language: "en",
  voice: "alloy",
  instructions: faker.lorem.paragraph(),
  labels: [],
  clientId: new ObjectId().toString(),
  role: "assistant",
  gender: faker.helpers.arrayElement(["male", "female"]),
}));

export function createAgent(overrides?: Partial<Agent>): Agent {
  return agentFactory.build(overrides);
}