import { AgentRailsConfig } from "../src/types";
import { agent, IAgentMessage } from "./mock-agent";

// Type-safe configuration with TypeScript imports
const config: AgentRailsConfig<IAgentMessage[], string> = {
  llm: {
    provider: "openai",
    apiKey: process.env.OPENAI_API_KEY || "test-key",
  },

  agent: async (input: IAgentMessage[]): Promise<string> => {
    return await agent(input);
  },

  rails: [
    {
      suite: "Basic Test Suite",
      rails: [
        {
          name: "Simple test",
          input: [{ role: "user", content: "Hello, world!" }],
          expectedBehavior: "Agent should respond politely",
          goodResponses: ["Hello!", "Hi there!"],
        },
      ],
    },
  ],
};

export default config;