import { LanguageModelV3, LanguageModelV3CallOptions } from '@ai-sdk/provider';
import { AxAIService, AxGenIn, AxGenOut, AxAgentic } from '@ax-llm/ax/index.js';
import { ModelMessage } from 'ai';
import { ReactNode } from 'react';
import { z } from 'zod';

type AxConfig = {
    fetch?: typeof fetch;
};
type Streamable = ReactNode | Promise<ReactNode>;
type Renderer<T> = (args: T) => Streamable | Generator<Streamable, Streamable, void> | AsyncGenerator<Streamable, Streamable, void>;
declare class AxAgentProvider<IN extends AxGenIn, OUT extends AxGenOut> {
    private readonly config?;
    private readonly funcInfo;
    private generateFunction;
    private updateState;
    constructor({ agent, updateState, generate, config, }: Readonly<{
        agent: AxAgentic<IN, OUT>;
        updateState: (msgs: readonly ModelMessage[]) => void;
        generate: Renderer<OUT>;
        config?: Readonly<AxConfig>;
    }>);
    get description(): string;
    get parameters(): z.ZodTypeAny;
    get generate(): Renderer<IN>;
}
declare class AxAIProvider implements LanguageModelV3 {
    readonly specificationVersion: "v3";
    readonly supportedUrls: Record<string, RegExp[]>;
    private readonly ai;
    private readonly config?;
    modelId: string;
    constructor(ai: AxAIService, config?: Readonly<AxConfig>);
    get provider(): string;
    doGenerate(options: LanguageModelV3CallOptions): Promise<Awaited<ReturnType<LanguageModelV3['doGenerate']>>>;
    doStream(options: LanguageModelV3CallOptions): Promise<Awaited<ReturnType<LanguageModelV3['doStream']>>>;
}

export { AxAIProvider, AxAgentProvider };