import { BetaMessageStream } from '@anthropic-ai/sdk/lib/BetaMessageStream.js';
import { BetaMessage, BetaRawMessageStreamEvent, MessageCreateParamsNonStreaming, MessageCreateParamsStreaming, BetaTool, BetaMessageParam, BetaTextBlockParam, BetaImageBlockParam, BetaRequestDocumentBlock, BetaToolUseBlockParam, BetaToolResultBlockParam, BetaThinkingBlockParam, BetaRedactedThinkingBlockParam } from '@anthropic-ai/sdk/resources/beta/messages';
import { Part, GenerateRequest, GenerateResponseData } from 'genkit';
import { ClaudeRunnerParams, AnthropicConfigSchema } from '../types.js';
import { BaseRunner } from './base.js';
import { RunnerTypes } from './types.js';
import '@anthropic-ai/sdk';
import '@anthropic-ai/sdk/resources/messages';
import 'genkit/model';

/**
 * Copyright 2025 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

interface BetaRunnerTypes extends RunnerTypes {
    Message: BetaMessage;
    Stream: BetaMessageStream;
    StreamEvent: BetaRawMessageStreamEvent;
    RequestBody: MessageCreateParamsNonStreaming;
    StreamingRequestBody: MessageCreateParamsStreaming;
    Tool: BetaTool;
    MessageParam: BetaMessageParam;
    ToolResponseContent: BetaTextBlockParam | BetaImageBlockParam;
    ContentBlockParam: BetaTextBlockParam | BetaImageBlockParam | BetaRequestDocumentBlock | BetaToolUseBlockParam | BetaToolResultBlockParam | BetaThinkingBlockParam | BetaRedactedThinkingBlockParam;
}
/**
 * Runner for the Anthropic Beta API.
 */
declare class BetaRunner extends BaseRunner<BetaRunnerTypes> {
    constructor(params: ClaudeRunnerParams);
    /**
     * Map a Genkit Part -> Anthropic beta content block param.
     * Supports: text, images (base64 data URLs), PDFs (document source),
     * tool_use (client tool request), tool_result (client tool response).
     */
    protected toAnthropicMessageContent(part: Part): BetaTextBlockParam | BetaImageBlockParam | BetaRequestDocumentBlock | BetaToolUseBlockParam | BetaToolResultBlockParam | BetaThinkingBlockParam | BetaRedactedThinkingBlockParam;
    protected createMessage(body: MessageCreateParamsNonStreaming, abortSignal: AbortSignal): Promise<BetaMessage>;
    protected streamMessages(body: MessageCreateParamsStreaming, abortSignal: AbortSignal): BetaMessageStream;
    /**
     * Build non-streaming request body.
     */
    protected toAnthropicRequestBody(modelName: string, request: GenerateRequest<typeof AnthropicConfigSchema>): MessageCreateParamsNonStreaming;
    /**
     * Build streaming request body.
     */
    protected toAnthropicStreamingRequestBody(modelName: string, request: GenerateRequest<typeof AnthropicConfigSchema>): MessageCreateParamsStreaming;
    protected toGenkitResponse(message: BetaMessage): GenerateResponseData;
    protected toGenkitPart(event: BetaRawMessageStreamEvent): Part | undefined;
    private fromBetaContentBlock;
    private fromBetaStopReason;
    private isStructuredOutputEnabled;
}

export { BetaRunner };