import { Logger } from '../../../logger/logger.cjs';
import { GetRunContext, MiddlewareType } from '../../../context.cjs';
import { a as ChatModel, c as ChatModelEmitter, l as ChatModelToolChoiceSupport, C as ChatConfig, f as ChatModelInput, i as ChatModelOutput, j as ChatModelParameters, b as ChatModelCache, d as ChatModelEvents } from '../../../chat-C20nYj46.cjs';
import { Message, MessageContentPart } from '../../../backend/message.cjs';
import { WatsonxClient, WatsonxClientSettings } from './client.cjs';
import { TextChatResultChoice, TextChatUsage, TextChatParams } from '@ibm-cloud/watsonx-ai/dist/watsonx-ai-ml/vml_v1.js';
import { E as Emitter } from '../../../emitter-D4OcelZ9.cjs';
import 'pino';
import '../../../errors.cjs';
import '../../../internals/types.cjs';
import '../../../internals/helpers/guards.cjs';
import '../../../internals/serializable.cjs';
import '../../../internals/helpers/promise.cjs';
import 'promise-based-task';
import '../../../cache/base.cjs';
import '../../../backend/constants.cjs';
import '../../../tools/base.cjs';
import 'ajv';
import '../../../internals/helpers/schema.cjs';
import 'zod';
import 'zod-to-json-schema';
import '../../../template.cjs';
import 'ai';
import '@ibm-cloud/watsonx-ai';
import 'ibm-cloud-sdk-core';
import '../../../backend/client.cjs';

declare class WatsonxChatModel extends ChatModel {
    readonly modelId: string;
    protected readonly client: WatsonxClient;
    readonly emitter: ChatModelEmitter;
    readonly toolChoiceSupport: ChatModelToolChoiceSupport[];
    get providerId(): string;
    constructor(modelId?: string, client?: WatsonxClient | WatsonxClientSettings);
    config(config: ChatConfig): this;
    protected _create(input: ChatModelInput, run: GetRunContext<any>): Promise<ChatModelOutput>;
    _createStream(input: ChatModelInput, run: GetRunContext<this>): AsyncGenerator<ChatModelOutput, void, unknown>;
    protected extractOutput(choices: TextChatResultChoice[], usage?: TextChatUsage, id?: string, toolCallIdsByName?: Map<string, string>): {
        finishReason: ChatModelOutput["finishReason"];
        usage: {
            completionTokens: number;
            promptTokens: number;
            totalTokens: number;
        } | undefined;
        messages: Message<MessageContentPart, string>[];
    };
    protected prepareInput(overrides: ChatModelInput): Promise<TextChatParams>;
    createSnapshot(): {
        modelId: string;
        parameters: ChatModelParameters;
        client: WatsonxClient;
        cache: ChatModelCache;
        emitter: Emitter<ChatModelEvents>;
        middlewares: MiddlewareType<any>[];
        logger: Logger;
        toolChoiceSupport: ChatModelToolChoiceSupport[];
        toolCallFallbackViaResponseFormat: boolean;
        modelSupportsToolCalling: boolean;
        retryOnEmptyResponse: boolean;
        fixInvalidToolCalls: boolean;
    };
    loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
}

export { WatsonxChatModel };