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

const config: AgentRailsConfig<IAgentMessage[], string> = {
  llm: {
    provider: "openai",
    apiKey: "test-key",
  },
  agent: agent,
  rails: [
    {
      suite: "Test",
      rails: [
        {
          name: "Test",
          input: [{ role: "user", content: "Hello" }],
          expectedBehavior: "Should work",
        },
      ],
    },
  ],
};

export default config;