@genkit-ai/anthropic
Version:
Genkit AI framework plugin for Anthropic APIs.
67 lines (63 loc) • 3.48 kB
TypeScript
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 };