import { AgentRailsConfig } from "agentrails";

const config: AgentRailsConfig<string, string> = {
  llm: {
    provider: "openai",
    apiKey: process.env.OPENAI_API_KEY || "test-key",
  },
  agent: async (input: string) => {
    return `Mock response for: ${input}`;
  },
  rails: [
    {
      suite: "Test Suite",
      rails: [
        {
          name: "Basic test",
          input: "Hello",
          expectedBehavior: "Should respond politely",
          goodResponses: ["Hello! How can I help you?"],
          badResponses: ["I don't care"],
        },
      ],
    },
  ],
};

export default config;