@gammatech/aijsx 0.8.1-dev.2024-05-24.2 → 0.9.0-dev.2024-05-28

package/dist/index.d.mts

@@ -1,13 +1,46 @@
-import { R as RenderContext, L as LogImplementation, S as SpanProcessor, C as ContextValues, a as Context, A as AINode, T as Tracer, b as ReadableSpan, c as SpanExporter, d as AIComponent, e as SpanAttributes, P as Prompt, f as ChatCompletionRequestPayloads, g as RenderedConversationMessage, E as EvaluatorResult, h as EvaluatorFn, i as PromptParsed, N as NotAsyncGenerator, F as FunctionChain, j as StreamChain, k as ChatCompletionClientAndProvider, J as JSX } from './jsx-dev-runtime-Kh2H5_Cf.mjs';
-export { I as AIElement, m as AIFragment, q as AssistantMessage, B as BoundLogger, s as ChatCompletionError, o as ChatCompletionRole, z as CombinedLogger, y as ConsoleLogger, D as Literal, t as LogChatCompletionRequest, u as LogChatCompletionResponse, v as LogLevel, w as Logger, x as NoopLogImplementation, _ as OutputParser, M as PropsOfAIComponent, G as RenderResult, K as Renderable, V as Span, O as SpanContext, W as SpanEvent, Q as SpanStatus, p as SystemMessage, Y as TracingContext, X as TracingContextKey, Z as TracingContextManager, U as UserMessage, H as attachedContextSymbol, r as computeUsage, l as createAIElement, n as createContext } from './jsx-dev-runtime-Kh2H5_Cf.mjs';
+import { L as LogChatCompletionRequest, A as AINode, C as ChatMessage, R as RenderContext, a as LogImplementation, S as SpanProcessor, b as ContextValues, c as Context, T as Tracer, d as ReadableSpan, e as SpanExporter, f as AIComponent, g as SpanAttributes, P as Prompt, h as ChatCompletionRequestPayloads, D as DebugMessage, E as EvaluatorResult, i as EvaluatorFn, j as PromptParsed, N as NotAsyncGenerator, F as FunctionChain, k as StreamChain, J as JSX } from './jsx-dev-runtime-uCZBuaDe.mjs';
+export { G as AIElement, m as AIFragment, B as BoundLogger, o as ChatRole, w as CombinedLogger, v as ConsoleLogger, I as ImagePart, p as ImagePartProps, x as Literal, q as LogChatCompletionResponse, r as LogLevel, s as Logger, u as NoopLogImplementation, Y as OutputParser, K as PropsOfAIComponent, y as RenderResult, H as Renderable, Q as Span, M as SpanContext, U as SpanEvent, O as SpanStatus, W as TracingContext, V as TracingContextKey, X as TracingContextManager, z as attachedContextSymbol, l as createAIElement, n as createContext, t as toDebugMessage } from './jsx-dev-runtime-uCZBuaDe.mjs';
 import { ZodObject, ZodRawShape, ZodTypeAny, ZodString, z } from 'zod';
 import { OpenAI } from 'openai';
 export { OpenAI as OpenAIClient } from 'openai';
 import { ChatCompletionSystemMessageParam, ChatCompletionUserMessageParam, ChatCompletionAssistantMessageParam, ChatCompletionCreateParams } from 'openai/resources';
 import AnthropicClient from '@anthropic-ai/sdk';
 export { default as AnthropicClient } from '@anthropic-ai/sdk';
-import { ImageBlockParam } from '@anthropic-ai/sdk/resources';
-export { countTokens as countAnthropicTokens } from '@anthropic-ai/tokenizer';
+
+declare class ChatCompletionError extends Error {
+    readonly chatCompletionRequest: LogChatCompletionRequest;
+    readonly status: number | undefined;
+    readonly shouldRetry: boolean;
+    readonly name = "ChatCompletionError";
+    constructor(message: string, chatCompletionRequest: LogChatCompletionRequest, status: number | undefined, shouldRetry?: boolean);
+}
+
+declare const SystemMessage: (props: {
+    children: AINode;
+}) => AINode;
+declare const UserMessage: (props: {
+    children: AINode;
+}) => AINode;
+declare const AssistantMessage: (props: {
+    children: AINode;
+}) => AINode;
+
+type TokenizerFn = (message: ChatMessage) => number;
+declare const computeUsage: (messages: ChatMessage[], tokenizer: TokenizerFn) => {
+    prompt: number;
+    completion: number;
+    total: number;
+};
+
+type ChatCompletionClientAndProvider<K> = {
+    client: K;
+    provider?: string;
+    providerRegion?: string;
+    costFn?: (model: string, usage: {
+        prompt: number;
+        completion: number;
+    }) => number;
+};
 
 type CreateRenderContextOptions = {
     logger?: LogImplementation;
@@ -129,7 +162,7 @@ declare namespace AISpanAttributes {
         retryCount?: number;
         requestType: K;
         chatCompletionRequest: ChatCompletionRequestPayloads[K];
-        inputMessages: RenderedConversationMessage[];
+        inputMessages: DebugMessage[];
         finishReason: string | null;
         output: string;
         tokensUsed: Usage;
@@ -200,7 +233,8 @@ declare module '@gammatech/aijsx' {
         openai: OpenAIChatCompletionRequest;
     }
 }
-type ValidOpenAIChatModel = 'gpt-4o' | 'gpt-4o-2024-05-13' | 'gpt-4-turbo-2024-04-09' | 'gpt-4-turbo' | 'gpt-4' | 'gpt-4-0314' | 'gpt-4-0613' | 'gpt-4-32k' | 'gpt-4-32k-0314' | 'gpt-4-32k-0613' | 'gpt-4-1106-preview' | 'gpt-4-0125-preview' | 'gpt-3.5-turbo' | 'gpt-3.5-turbo-0301' | 'gpt-3.5-turbo-0613' | 'gpt-3.5-turbo-16k' | 'gpt-3.5-turbo-16k-0613' | 'gpt-3.5-turbo-1106' | 'gpt-3.5-turbo-0125';
+type ValidOpenAIVisionModel = 'gpt-4o' | 'gpt-4o-2024-05-13' | 'gpt-4-turbo-2024-04-09' | 'gpt-4-turbo' | 'gpt-4-vision-preview';
+type ValidOpenAIChatModel = ValidOpenAIVisionModel | 'gpt-4' | 'gpt-4-0314' | 'gpt-4-0613' | 'gpt-4-32k' | 'gpt-4-32k-0314' | 'gpt-4-32k-0613' | 'gpt-4-1106-preview' | 'gpt-4-0125-preview' | 'gpt-3.5-turbo' | 'gpt-3.5-turbo-0301' | 'gpt-3.5-turbo-0613' | 'gpt-3.5-turbo-16k' | 'gpt-3.5-turbo-16k-0613' | 'gpt-3.5-turbo-1106' | 'gpt-3.5-turbo-0125';
 declare const OpenAIClientContext: Context<() => ChatCompletionClientAndProvider<OpenAI>>;
 type OpenAIChatCompletionProps = {
     model: ValidOpenAIChatModel;
@@ -213,33 +247,7 @@ type OpenAIChatCompletionProps = {
 };
 declare function OpenAIChatCompletion(props: OpenAIChatCompletionProps, ctx: RenderContext): AINode;
 
-type ValidOpenAIVisionModel = 'gpt-4o' | 'gpt-4o-2024-05-13' | 'gpt-4-turbo-2024-04-09' | 'gpt-4-turbo' | 'gpt-4-vision-preview';
-declare const ContentTypeImage: (_props: {
-    url: string;
-    dimensions?: {
-        width: number;
-        height: number;
-    };
-    detail?: 'auto' | 'high' | 'low';
-}) => null;
-type OpenAIVisionChatCompletionProps = {
-    model?: ValidOpenAIVisionModel;
-    maxTokens?: number;
-    stop?: string | string[];
-    temperature?: number;
-    maxRetries?: number;
-    children: AINode;
-};
-declare function OpenAIVisionChatCompletion(props: OpenAIVisionChatCompletionProps, ctx: RenderContext): AINode;
-declare function OpenAIVisionChatCompletionInner(props: OpenAIVisionChatCompletionProps, { logger, render, tracer, getContext }: RenderContext): AsyncGenerator<string, void, unknown>;
-
-declare const tokenizer: {
-    encode: (text: string) => number[];
-    decode: (tokens: number[]) => string;
-};
-declare function tokenLimitForChatModel(model: ValidOpenAIChatModel): number | undefined;
-declare function tokenCountForOpenAIMessage(message: OpenAIChatMessage): number;
-declare function tokenCountForOpenAIVisionMessage(message: OpenAIChatMessage): number;
+declare const openaiTokenizer: TokenizerFn;
 
 type AnthropicChatCompletionRequest = AnthropicClient.Messages.MessageStreamParams;
 declare module '@gammatech/aijsx' {
@@ -262,23 +270,7 @@ type AnthropicChatCompletionProps = {
     children: AINode;
 };
 declare function AnthropicChatCompletion(props: AnthropicChatCompletionProps, ctx: RenderContext): JSX.Element;
-/**
- * An AI.JSX component that invokes an Anthropic Large Language Model.
- * @param children The children to render.
- * @param chatModel The chat model to use.
- * @param completionModel The completion model to use.
- * @param client The Anthropic client.
- */
-declare function AnthropicChatCompletionInner(props: AnthropicChatCompletionProps, { render, logger, tracer, getContext }: RenderContext): AsyncGenerator<string, void, unknown>;
 
-type ClaudeImageBlockBase64 = {
-    data: string;
-    mediaType: ImageBlockParam.Source['media_type'];
-};
-type ClaudeImageBlockUrl = {
-    url: string;
-};
-type ClaudeImageBlockProps = ClaudeImageBlockBase64 | ClaudeImageBlockUrl;
-declare const ClaudeImageBlock: (props: ClaudeImageBlockProps) => Promise<JSX.Element>;
+declare const anthropicTokenizer: TokenizerFn;
 
-export { AIComponent, AINode, AISpanAttributes, AISpanProcessor, AnthropicChatCompletion, AnthropicChatCompletionInner, type AnthropicChatCompletionRequest, AnthropicClientContext, ChatCompletionClientAndProvider, type ChatCompletionRequestPayloads, ClaudeImageBlock, ContentTypeImage, Context, type CostFn, DefaultMaxRetriesContext, EnrichingSpanProcessor, EvaluatorFn, EvaluatorResult, Fallback, FunctionChain, LogImplementation, NotAsyncGenerator, OpenAIChatCompletion, type OpenAIChatCompletionRequest, type OpenAIChatMessage, OpenAIClientContext, OpenAIVisionChatCompletion, OpenAIVisionChatCompletionInner, ParseVariablesError, ProcessedAISpanAttributes, type Prompt, PromptInvalidOutputError, PromptParsed, ReadableSpan, RenderContext, RenderedConversationMessage, Retry, RetryCountContext, SpanAttributes, SpanExporter, SpanProcessor, StreamChain, Trace, Tracer, type ValidAnthropicChatModel, type ValidOpenAIChatModel, type ValidOpenAIVisionModel, createFunctionChain, createPrompt, createRenderContext, createStreamChain, evaluatePrompt, tokenCountForOpenAIMessage, tokenCountForOpenAIVisionMessage, tokenLimitForChatModel, tokenizer, tracing };
+export { AIComponent, AINode, AISpanAttributes, AISpanProcessor, AnthropicChatCompletion, type AnthropicChatCompletionRequest, AnthropicClientContext, AssistantMessage, type ChatCompletionClientAndProvider, ChatCompletionError, type ChatCompletionRequestPayloads, ChatMessage, Context, type CostFn, DebugMessage, DefaultMaxRetriesContext, EnrichingSpanProcessor, EvaluatorFn, EvaluatorResult, Fallback, FunctionChain, LogChatCompletionRequest, LogImplementation, NotAsyncGenerator, OpenAIChatCompletion, type OpenAIChatCompletionRequest, type OpenAIChatMessage, OpenAIClientContext, ParseVariablesError, ProcessedAISpanAttributes, type Prompt, PromptInvalidOutputError, PromptParsed, ReadableSpan, RenderContext, Retry, RetryCountContext, SpanAttributes, SpanExporter, SpanProcessor, StreamChain, SystemMessage, type TokenizerFn, Trace, Tracer, UserMessage, type ValidAnthropicChatModel, type ValidOpenAIChatModel, type ValidOpenAIVisionModel, anthropicTokenizer, computeUsage, createFunctionChain, createPrompt, createRenderContext, createStreamChain, evaluatePrompt, openaiTokenizer, tracing };

package/dist/index.d.ts

@@ -1,13 +1,46 @@
-import { R as RenderContext, L as LogImplementation, S as SpanProcessor, C as ContextValues, a as Context, A as AINode, T as Tracer, b as ReadableSpan, c as SpanExporter, d as AIComponent, e as SpanAttributes, P as Prompt, f as ChatCompletionRequestPayloads, g as RenderedConversationMessage, E as EvaluatorResult, h as EvaluatorFn, i as PromptParsed, N as NotAsyncGenerator, F as FunctionChain, j as StreamChain, k as ChatCompletionClientAndProvider, J as JSX } from './jsx-dev-runtime-Kh2H5_Cf.js';
-export { I as AIElement, m as AIFragment, q as AssistantMessage, B as BoundLogger, s as ChatCompletionError, o as ChatCompletionRole, z as CombinedLogger, y as ConsoleLogger, D as Literal, t as LogChatCompletionRequest, u as LogChatCompletionResponse, v as LogLevel, w as Logger, x as NoopLogImplementation, _ as OutputParser, M as PropsOfAIComponent, G as RenderResult, K as Renderable, V as Span, O as SpanContext, W as SpanEvent, Q as SpanStatus, p as SystemMessage, Y as TracingContext, X as TracingContextKey, Z as TracingContextManager, U as UserMessage, H as attachedContextSymbol, r as computeUsage, l as createAIElement, n as createContext } from './jsx-dev-runtime-Kh2H5_Cf.js';
+import { L as LogChatCompletionRequest, A as AINode, C as ChatMessage, R as RenderContext, a as LogImplementation, S as SpanProcessor, b as ContextValues, c as Context, T as Tracer, d as ReadableSpan, e as SpanExporter, f as AIComponent, g as SpanAttributes, P as Prompt, h as ChatCompletionRequestPayloads, D as DebugMessage, E as EvaluatorResult, i as EvaluatorFn, j as PromptParsed, N as NotAsyncGenerator, F as FunctionChain, k as StreamChain, J as JSX } from './jsx-dev-runtime-uCZBuaDe.js';
+export { G as AIElement, m as AIFragment, B as BoundLogger, o as ChatRole, w as CombinedLogger, v as ConsoleLogger, I as ImagePart, p as ImagePartProps, x as Literal, q as LogChatCompletionResponse, r as LogLevel, s as Logger, u as NoopLogImplementation, Y as OutputParser, K as PropsOfAIComponent, y as RenderResult, H as Renderable, Q as Span, M as SpanContext, U as SpanEvent, O as SpanStatus, W as TracingContext, V as TracingContextKey, X as TracingContextManager, z as attachedContextSymbol, l as createAIElement, n as createContext, t as toDebugMessage } from './jsx-dev-runtime-uCZBuaDe.js';
 import { ZodObject, ZodRawShape, ZodTypeAny, ZodString, z } from 'zod';
 import { OpenAI } from 'openai';
 export { OpenAI as OpenAIClient } from 'openai';
 import { ChatCompletionSystemMessageParam, ChatCompletionUserMessageParam, ChatCompletionAssistantMessageParam, ChatCompletionCreateParams } from 'openai/resources';
 import AnthropicClient from '@anthropic-ai/sdk';
 export { default as AnthropicClient } from '@anthropic-ai/sdk';
-import { ImageBlockParam } from '@anthropic-ai/sdk/resources';
-export { countTokens as countAnthropicTokens } from '@anthropic-ai/tokenizer';
+
+declare class ChatCompletionError extends Error {
+    readonly chatCompletionRequest: LogChatCompletionRequest;
+    readonly status: number | undefined;
+    readonly shouldRetry: boolean;
+    readonly name = "ChatCompletionError";
+    constructor(message: string, chatCompletionRequest: LogChatCompletionRequest, status: number | undefined, shouldRetry?: boolean);
+}
+
+declare const SystemMessage: (props: {
+    children: AINode;
+}) => AINode;
+declare const UserMessage: (props: {
+    children: AINode;
+}) => AINode;
+declare const AssistantMessage: (props: {
+    children: AINode;
+}) => AINode;
+
+type TokenizerFn = (message: ChatMessage) => number;
+declare const computeUsage: (messages: ChatMessage[], tokenizer: TokenizerFn) => {
+    prompt: number;
+    completion: number;
+    total: number;
+};
+
+type ChatCompletionClientAndProvider<K> = {
+    client: K;
+    provider?: string;
+    providerRegion?: string;
+    costFn?: (model: string, usage: {
+        prompt: number;
+        completion: number;
+    }) => number;
+};
 
 type CreateRenderContextOptions = {
     logger?: LogImplementation;
@@ -129,7 +162,7 @@ declare namespace AISpanAttributes {
         retryCount?: number;
         requestType: K;
         chatCompletionRequest: ChatCompletionRequestPayloads[K];
-        inputMessages: RenderedConversationMessage[];
+        inputMessages: DebugMessage[];
         finishReason: string | null;
         output: string;
         tokensUsed: Usage;
@@ -200,7 +233,8 @@ declare module '@gammatech/aijsx' {
         openai: OpenAIChatCompletionRequest;
     }
 }
-type ValidOpenAIChatModel = 'gpt-4o' | 'gpt-4o-2024-05-13' | 'gpt-4-turbo-2024-04-09' | 'gpt-4-turbo' | 'gpt-4' | 'gpt-4-0314' | 'gpt-4-0613' | 'gpt-4-32k' | 'gpt-4-32k-0314' | 'gpt-4-32k-0613' | 'gpt-4-1106-preview' | 'gpt-4-0125-preview' | 'gpt-3.5-turbo' | 'gpt-3.5-turbo-0301' | 'gpt-3.5-turbo-0613' | 'gpt-3.5-turbo-16k' | 'gpt-3.5-turbo-16k-0613' | 'gpt-3.5-turbo-1106' | 'gpt-3.5-turbo-0125';
+type ValidOpenAIVisionModel = 'gpt-4o' | 'gpt-4o-2024-05-13' | 'gpt-4-turbo-2024-04-09' | 'gpt-4-turbo' | 'gpt-4-vision-preview';
+type ValidOpenAIChatModel = ValidOpenAIVisionModel | 'gpt-4' | 'gpt-4-0314' | 'gpt-4-0613' | 'gpt-4-32k' | 'gpt-4-32k-0314' | 'gpt-4-32k-0613' | 'gpt-4-1106-preview' | 'gpt-4-0125-preview' | 'gpt-3.5-turbo' | 'gpt-3.5-turbo-0301' | 'gpt-3.5-turbo-0613' | 'gpt-3.5-turbo-16k' | 'gpt-3.5-turbo-16k-0613' | 'gpt-3.5-turbo-1106' | 'gpt-3.5-turbo-0125';
 declare const OpenAIClientContext: Context<() => ChatCompletionClientAndProvider<OpenAI>>;
 type OpenAIChatCompletionProps = {
     model: ValidOpenAIChatModel;
@@ -213,33 +247,7 @@ type OpenAIChatCompletionProps = {
 };
 declare function OpenAIChatCompletion(props: OpenAIChatCompletionProps, ctx: RenderContext): AINode;
 
-type ValidOpenAIVisionModel = 'gpt-4o' | 'gpt-4o-2024-05-13' | 'gpt-4-turbo-2024-04-09' | 'gpt-4-turbo' | 'gpt-4-vision-preview';
-declare const ContentTypeImage: (_props: {
-    url: string;
-    dimensions?: {
-        width: number;
-        height: number;
-    };
-    detail?: 'auto' | 'high' | 'low';
-}) => null;
-type OpenAIVisionChatCompletionProps = {
-    model?: ValidOpenAIVisionModel;
-    maxTokens?: number;
-    stop?: string | string[];
-    temperature?: number;
-    maxRetries?: number;
-    children: AINode;
-};
-declare function OpenAIVisionChatCompletion(props: OpenAIVisionChatCompletionProps, ctx: RenderContext): AINode;
-declare function OpenAIVisionChatCompletionInner(props: OpenAIVisionChatCompletionProps, { logger, render, tracer, getContext }: RenderContext): AsyncGenerator<string, void, unknown>;
-
-declare const tokenizer: {
-    encode: (text: string) => number[];
-    decode: (tokens: number[]) => string;
-};
-declare function tokenLimitForChatModel(model: ValidOpenAIChatModel): number | undefined;
-declare function tokenCountForOpenAIMessage(message: OpenAIChatMessage): number;
-declare function tokenCountForOpenAIVisionMessage(message: OpenAIChatMessage): number;
+declare const openaiTokenizer: TokenizerFn;
 
 type AnthropicChatCompletionRequest = AnthropicClient.Messages.MessageStreamParams;
 declare module '@gammatech/aijsx' {
@@ -262,23 +270,7 @@ type AnthropicChatCompletionProps = {
     children: AINode;
 };
 declare function AnthropicChatCompletion(props: AnthropicChatCompletionProps, ctx: RenderContext): JSX.Element;
-/**
- * An AI.JSX component that invokes an Anthropic Large Language Model.
- * @param children The children to render.
- * @param chatModel The chat model to use.
- * @param completionModel The completion model to use.
- * @param client The Anthropic client.
- */
-declare function AnthropicChatCompletionInner(props: AnthropicChatCompletionProps, { render, logger, tracer, getContext }: RenderContext): AsyncGenerator<string, void, unknown>;
 
-type ClaudeImageBlockBase64 = {
-    data: string;
-    mediaType: ImageBlockParam.Source['media_type'];
-};
-type ClaudeImageBlockUrl = {
-    url: string;
-};
-type ClaudeImageBlockProps = ClaudeImageBlockBase64 | ClaudeImageBlockUrl;
-declare const ClaudeImageBlock: (props: ClaudeImageBlockProps) => Promise<JSX.Element>;
+declare const anthropicTokenizer: TokenizerFn;
 
-export { AIComponent, AINode, AISpanAttributes, AISpanProcessor, AnthropicChatCompletion, AnthropicChatCompletionInner, type AnthropicChatCompletionRequest, AnthropicClientContext, ChatCompletionClientAndProvider, type ChatCompletionRequestPayloads, ClaudeImageBlock, ContentTypeImage, Context, type CostFn, DefaultMaxRetriesContext, EnrichingSpanProcessor, EvaluatorFn, EvaluatorResult, Fallback, FunctionChain, LogImplementation, NotAsyncGenerator, OpenAIChatCompletion, type OpenAIChatCompletionRequest, type OpenAIChatMessage, OpenAIClientContext, OpenAIVisionChatCompletion, OpenAIVisionChatCompletionInner, ParseVariablesError, ProcessedAISpanAttributes, type Prompt, PromptInvalidOutputError, PromptParsed, ReadableSpan, RenderContext, RenderedConversationMessage, Retry, RetryCountContext, SpanAttributes, SpanExporter, SpanProcessor, StreamChain, Trace, Tracer, type ValidAnthropicChatModel, type ValidOpenAIChatModel, type ValidOpenAIVisionModel, createFunctionChain, createPrompt, createRenderContext, createStreamChain, evaluatePrompt, tokenCountForOpenAIMessage, tokenCountForOpenAIVisionMessage, tokenLimitForChatModel, tokenizer, tracing };
+export { AIComponent, AINode, AISpanAttributes, AISpanProcessor, AnthropicChatCompletion, type AnthropicChatCompletionRequest, AnthropicClientContext, AssistantMessage, type ChatCompletionClientAndProvider, ChatCompletionError, type ChatCompletionRequestPayloads, ChatMessage, Context, type CostFn, DebugMessage, DefaultMaxRetriesContext, EnrichingSpanProcessor, EvaluatorFn, EvaluatorResult, Fallback, FunctionChain, LogChatCompletionRequest, LogImplementation, NotAsyncGenerator, OpenAIChatCompletion, type OpenAIChatCompletionRequest, type OpenAIChatMessage, OpenAIClientContext, ParseVariablesError, ProcessedAISpanAttributes, type Prompt, PromptInvalidOutputError, PromptParsed, ReadableSpan, RenderContext, Retry, RetryCountContext, SpanAttributes, SpanExporter, SpanProcessor, StreamChain, SystemMessage, type TokenizerFn, Trace, Tracer, UserMessage, type ValidAnthropicChatModel, type ValidOpenAIChatModel, type ValidOpenAIVisionModel, anthropicTokenizer, computeUsage, createFunctionChain, createPrompt, createRenderContext, createStreamChain, evaluatePrompt, openaiTokenizer, tracing };