import type { ONIModel, ChatParams } from "../models/index.js";
import type { StateGraph } from "../graph.js";
import type { ONIStreamEvent, CustomStreamEvent, MessageStreamEvent, StreamMode } from "../types.js";
interface MockResponse {
    role: "assistant";
    content: string;
    toolCalls?: Array<{
        id: string;
        name: string;
        args: Record<string, unknown>;
    }>;
}
interface MockModelInstance extends ONIModel {
    callHistory: ChatParams[];
    reset(): void;
}
export declare function mockModel(responses: MockResponse[]): MockModelInstance;
interface GraphAssertions {
    hasNode?: string[];
    nodeCount?: number;
    hasEdge?: [string, string][];
}
export declare function assertGraph<S extends Record<string, unknown>>(graph: StateGraph<S>, assertions: GraphAssertions): void;
interface TestHarness<S extends Record<string, unknown>> {
    invoke(input: Partial<S>, config?: Record<string, unknown>): Promise<S>;
    collectStream(input: Partial<S>, streamMode?: StreamMode): Promise<(ONIStreamEvent<S> | CustomStreamEvent | MessageStreamEvent)[]>;
}
export declare function createTestHarness<S extends Record<string, unknown>>(graph: StateGraph<S>, options?: {
    checkpointer?: import("../types.js").ONICheckpointer<S>;
    store?: import("../store/index.js").BaseStore;
}): TestHarness<S>;
export type { MockResponse, MockModelInstance, GraphAssertions, TestHarness };
//# sourceMappingURL=index.d.ts.map