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

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: "BaseURL Test",
      rails: [
        {
          name: "Test baseUrl imports",
          input: [{ role: "user", content: "Hello" }],
          expectedBehavior: "Should work with baseUrl imports",
        },
      ],
    },
  ],
};

export default config;