UNPKG

@genkit-ai/anthropic

Version:

Genkit AI framework plugin for Anthropic APIs.

67 lines (63 loc) 3.48 kB
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 };