/**
 * Core types for AgentRails framework
 */
/**
 * Tool call definition for validation
 */
export interface ToolCall {
    id: string;
    type: string;
    function?: {
        name: string;
        arguments?: Record<string, any>;
    };
}
/**
 * Rail case definition (formerly TestCase)
 * Can be defined inline in config or loaded from YAML
 */
export interface RailCase<TInput = any> {
    name: string;
    description?: string;
    input: TInput;
    expectedBehavior?: string;
    goodResponses?: string[];
    badResponses?: string[];
    maxTimeAllowed?: number;
    expectedToolCalls?: string[];
    timeout?: number;
    metadata?: Record<string, any>;
}
/**
 * Rail suite definition (formerly TestSuite)
 * Can be defined inline in config or loaded from YAML
 */
export interface RailSuite<TInput = any> {
    suite: string;
    description?: string;
    rails: RailCase<TInput>[];
}
/**
 * Agent implementation that users must provide
 */
export interface AgentFunction<TInput = any, TOutput = any> {
    (input: TInput): Promise<TOutput>;
}
/**
 * Rail result for a single rail case (formerly TestResult)
 */
export interface RailResult<TInput = any, TOutput = any> {
    railName: string;
    passed: boolean;
    actualResponse: TOutput;
    evaluationReasoning?: string;
    error?: string;
    duration: number;
    toolCalls?: ToolCall[];
    toolCallValidation?: {
        expected: string[];
        actual: string[];
        passed: boolean;
    };
}
/**
 * Complete rail run results (formerly TestRunResult)
 */
export interface RailRunResult<TInput = any, TOutput = any> {
    suiteName: string;
    totalRails: number;
    passed: number;
    failed: number;
    duration: number;
    results: RailResult<TInput, TOutput>[];
}
/**
 * Supported LLM providers for evaluation
 */
export type LLMProvider = "openai" | "anthropic" | "google" | "grok";
/**
 * LLM configuration options
 */
export interface LLMConfig {
    provider: LLMProvider;
    apiKey: string;
    model?: string;
    temperature?: number;
    baseURL?: string;
}
/**
 * AgentRails configuration
 */
export interface AgentRailsConfig<TInput = any, TOutput = any> {
    llm: LLMConfig;
    rails?: RailSuite<TInput>[];
    testMatch?: string[];
    timeout?: number;
    agent: AgentFunction<TInput, TOutput>;
    features?: {
        e2eTesting?: boolean;
        toolCallValidation?: boolean;
        [key: string]: any;
    };
}
/**
 * LLM Evaluator interface for extensibility
 */
export interface LLMEvaluator<TInput = any, TOutput = any> {
    evaluate(input: TInput, actualResponse: TOutput, expectedBehavior?: string, goodResponses?: string[], badResponses?: string[]): Promise<{
        passed: boolean;
        reasoning: string;
    }>;
}
export type TestCase<TInput = any> = RailCase<TInput>;
export type TestSuite<TInput = any> = RailSuite<TInput>;
export type TestResult<TInput = any, TOutput = any> = RailResult<TInput, TOutput>;
export type TestRunResult<TInput = any, TOutput = any> = RailRunResult<TInput, TOutput>;
//# sourceMappingURL=types.d.ts.map