@gammatech/aijsx 0.4.0-beta.8 → 0.5.0-beta.1

package/dist/{createElement-VpIrgHrz.d.mts → createElement--C_Hwi_c.d.mts}

@@ -1,35 +1,83 @@
 import { ZodTypeAny, ZodObject, ZodRawShape } from 'zod';
 
-type Literal = string | number | null | undefined | boolean;
-interface RenderableStream extends AsyncGenerator<string, void, unknown> {
-    [Symbol.asyncIterator]: () => AsyncGenerator<string, void, unknown>;
+interface ChatCompletionRequestPayloads {
 }
-interface RenderResult extends RenderableStream {
-    then: (onResolved: (value: string) => void, onRejected?: (reason?: any) => void) => void;
+interface LogChatCompletionRequest<R extends Record<string, any> = ChatCompletionRequestPayloads[keyof ChatCompletionRequestPayloads]> {
+    requestId: string;
+    startTime: string;
+    model: string;
+    providerRegion?: string;
+    provider?: string;
+    inputMessages: RenderedConversationMessage[];
+    request: R;
 }
-interface Context<T> {
-    Provider: AIComponent<{
-        children: AINode;
-        value: T;
-    }>;
-    defaultValue: T;
-    key: symbol;
+interface LogChatCompletionResponse<R extends Record<string, any> = ChatCompletionRequestPayloads[keyof ChatCompletionRequestPayloads]> extends LogChatCompletionRequest<R> {
+    endTime: string;
+    latency: number;
+    outputMessage: RenderedConversationMessage;
+    finishReason: string;
+    tokensUsed: {
+        prompt: number;
+        completion: number;
+        total: number;
+    };
 }
-type AIComponent<P> = (props: P, context: RenderContext) => Renderable;
-declare const attachedContextSymbol: unique symbol;
-interface AIElement<P> {
-    /** The tag associated with this {@link AIElement}. */
-    tag: AIComponent<P>;
-    /** The component properties. */
-    props: P;
-    /** A function that renders this {@link AIElement} to a {@link Renderable}. */
-    render: (ctx: RenderContext) => Renderable;
-    /** The {@link RenderContext} associated with this {@link Element}. */
-    [attachedContextSymbol]?: Record<symbol, any>;
+
+/**
+ * This can be extended using declare module to add additional providers.
+ */
+type LogLevel = 'error' | 'warn' | 'info' | 'debug';
+type Loggable = string | number | boolean | undefined | null | object;
+type Logger = {
+    error: (...msg: Loggable[]) => void;
+    warn: (...msg: Loggable[]) => void;
+    info: (...msg: Loggable[]) => void;
+    debug: (...msg: Loggable[]) => void;
+    logException(exception: string | Error): void;
+    chatCompletionRequest<K extends keyof ChatCompletionRequestPayloads>(provider: K, payload: LogChatCompletionRequest<ChatCompletionRequestPayloads[K]>): void;
+    chatCompletionResponse<K extends keyof ChatCompletionRequestPayloads>(provider: K, payload: LogChatCompletionResponse<ChatCompletionRequestPayloads[K]>): void;
+};
+declare abstract class LogImplementation {
+    private readonly loggedExceptions;
+    /**
+     * @param ctx The current RenderContext
+     * @param level The log level, e.g. 'error', 'warn', 'info', 'debug'
+     * @param message
+     */
+    abstract log(level: LogLevel, message: string): void;
+    chatCompletionRequest<K extends keyof ChatCompletionRequestPayloads>(provider: K, payload: LogChatCompletionRequest<ChatCompletionRequestPayloads[K]>): void;
+    chatCompletionResponse<K extends keyof ChatCompletionRequestPayloads>(provider: K, payload: LogChatCompletionResponse<ChatCompletionRequestPayloads[K]>): void;
+    /**
+     * Logs exceptions thrown during an element's render.
+     */
+    logException(exception: unknown): void;
+}
+declare class BoundLogger implements Logger {
+    private readonly impl;
+    constructor(impl: LogImplementation);
+    private formatMessage;
+    error: (...msgs: Loggable[]) => void;
+    warn: (...msgs: Loggable[]) => void;
+    info: (...msgs: Loggable[]) => void;
+    debug: (...msgs: Loggable[]) => void;
+    logException: (exception: string | Error) => void;
+    chatCompletionRequest: <K extends keyof ChatCompletionRequestPayloads>(provider: K, payload: LogChatCompletionRequest<ChatCompletionRequestPayloads[K]>) => void;
+    chatCompletionResponse: <K extends keyof ChatCompletionRequestPayloads>(provider: K, payload: LogChatCompletionResponse<ChatCompletionRequestPayloads[K]>) => void;
+}
+declare class NoopLogImplementation extends LogImplementation {
+    log(level: LogLevel, message: string): void;
+}
+declare class ConsoleLogger extends LogImplementation {
+    log(level: LogLevel, message: string): void;
+}
+declare class CombinedLogger extends LogImplementation {
+    private readonly loggers;
+    constructor(loggers: LogImplementation[]);
+    log(...args: Parameters<LogImplementation['log']>): void;
+    logException(...args: Parameters<LogImplementation['logException']>): void;
+    chatCompletionRequest<_K extends keyof ChatCompletionRequestPayloads>(...args: Parameters<LogImplementation['chatCompletionRequest']>): void;
+    chatCompletionResponse<_K extends keyof ChatCompletionRequestPayloads>(...args: Parameters<LogImplementation['chatCompletionResponse']>): void;
 }
-type AINode = Literal | AIElement<any> | AINode[];
-type Renderable = AINode | PromiseLike<Renderable> | RenderableStream;
-type PropsOfAIComponent<T extends AIComponent<any>> = T extends AIComponent<infer P> ? P : never;
 
 type EvaluatorPrimitive = string | number | boolean;
 type EvaluatorResult = {
@@ -80,7 +128,6 @@ interface StreamChain<V extends Record<string, any> = Record<string, any>, R ext
 type IsAsyncGenerator<T> = T extends AsyncGenerator<any, any, any> ? true : false;
 type NotAsyncGenerator<T> = IsAsyncGenerator<T> extends true ? never : T;
 
-declare const LoggerContext: Context<LogImplementation>;
 type ContextValues = Record<symbol, any>;
 type RenderOptions = {
     preserveTags?: boolean;
@@ -90,16 +137,7 @@ type RenderOptions = {
         };
     };
 };
-type RenderTraceInfo = {
-    traceId: string;
-    spanId: string;
-    parentSpanId: string | null;
-    renderType: 'element' | 'chain' | 'prompt' | 'root';
-    key: string;
-    parent: RenderTraceInfo | null;
-};
 interface RenderContext {
-    traceInfo: RenderTraceInfo;
     logger: Logger;
     getContext<T>(context: Context<T>): T;
     render(renderable: Renderable, opts?: RenderOptions): RenderResult;
@@ -115,161 +153,36 @@ type RenderPromptOptions = {
 };
 declare function createContext<T>(defaultValue: T): Context<T>;
 
-interface LogChainStart {
-    startTime: string;
-    chainKey: string;
-    variables: Record<string, any>;
-    chainType: 'function' | 'stream';
-}
-interface LogChainComplete {
-    startTime: string;
-    endTime: string;
-    latency: number;
-    chainKey: string;
-    variables: Record<string, any>;
-    chainType: 'function' | 'stream';
-    output: any;
-}
-interface LogChainError {
-    startTime: string;
-    endTime: string;
-    latency: number;
-    chainKey: string;
-    variables: Record<string, any>;
-    chainType: 'function' | 'stream';
-    error: Error;
-}
-interface LogPromptStart {
-    startTime: string;
-    promptKey: string;
-    variables: Record<string, any>;
-}
-interface LogPromptComplete {
-    startTime: string;
-    endTime: string;
-    latency: number;
-    promptKey: string;
-    variables: Record<string, any>;
-    output: any;
-}
-interface LogPromptError {
-    startTime: string;
-    endTime: string;
-    latency: number;
-    promptKey: string;
-    variables: Record<string, any>;
-    error: Error;
-}
-/**
- * This can be extended using declare module to add additional providers.
- */
-interface ChatCompletionRequestPayloads {
-}
-interface LogChatCompletionRequest<R extends Record<string, any> = ChatCompletionRequestPayloads[keyof ChatCompletionRequestPayloads]> {
-    requestId: string;
-    startTime: string;
-    model: string;
-    providerRegion?: string;
-    provider?: string;
-    inputMessages: RenderedConversationMessage[];
-    request: R;
-}
-interface LogChatCompletionResponse<R extends Record<string, any> = ChatCompletionRequestPayloads[keyof ChatCompletionRequestPayloads]> extends LogChatCompletionRequest<R> {
-    requestId: string;
-    startTime: string;
-    endTime: string;
-    latency: number;
-    outputMessage: RenderedConversationMessage;
-    finishReason: string;
-    tokensUsed: {
-        prompt: number;
-        completion: number;
-        total: number;
-    };
-}
-type LogLevel = 'error' | 'warn' | 'info' | 'debug';
-type Loggable = string | number | boolean | undefined | null | object;
-type LogContext = {
-    renderType: 'element' | 'chain' | 'prompt';
-    key: string;
-    spanId: string;
-    traceId: string;
-    parentSpanId: string;
-};
-type Logger = {
-    error: (...msg: Loggable[]) => void;
-    warn: (...msg: Loggable[]) => void;
-    info: (...msg: Loggable[]) => void;
-    debug: (...msg: Loggable[]) => void;
-    logException: (exception: unknown) => void;
-    chainStart(chain: FunctionChain | StreamChain, payload: LogChainStart): void;
-    chainComplete(chain: FunctionChain | StreamChain, payload: LogChainComplete): void;
-    chainError(chain: FunctionChain | StreamChain, payload: LogChainError): void;
-    promptStart(prompt: Prompt, payload: LogPromptStart): void;
-    promptComplete(prompt: Prompt, payload: LogPromptComplete): void;
-    promptError(prompt: Prompt, payload: LogPromptError): void;
-    chatCompletionRequest: <K extends keyof ChatCompletionRequestPayloads>(provider: K, payload: LogChatCompletionRequest<ChatCompletionRequestPayloads[K]>) => void;
-    chatCompletionResponse: <K extends keyof ChatCompletionRequestPayloads>(provider: K, payload: LogChatCompletionResponse<ChatCompletionRequestPayloads[K]>) => void;
-};
-declare abstract class LogImplementation {
-    protected readonly loggedExceptions: WeakMap<object, boolean>;
-    /**
-     * @param ctx The current RenderContext
-     * @param level The log level, e.g. 'error', 'warn', 'info', 'debug'
-     * @param message
-     */
-    abstract log(ctx: RenderTraceInfo, level: LogLevel, message: string): void;
-    /**
-     * Logs exceptions thrown during an element's render.
-     */
-    logException(ctx: RenderTraceInfo, exception: unknown): void;
-    chainStart(_ctx: RenderTraceInfo, _chain: FunctionChain | StreamChain, _payload: LogChainStart): void;
-    chainComplete(_ctx: RenderTraceInfo, _chain: FunctionChain | StreamChain, _payload: LogChainComplete): void;
-    chainError(_ctx: RenderTraceInfo, _chain: FunctionChain | StreamChain, _payload: LogChainError): void;
-    promptStart(_ctx: RenderTraceInfo, _prompt: Prompt, _payload: LogPromptStart): void;
-    promptComplete(_ctx: RenderTraceInfo, _prompt: Prompt, _payload: LogPromptComplete): void;
-    promptError(_ctx: RenderTraceInfo, _prompt: Prompt, _payload: LogPromptError): void;
-    chatCompletionRequest<K extends keyof ChatCompletionRequestPayloads>(_ctx: RenderTraceInfo, _provider: K, _payload: LogChatCompletionRequest<ChatCompletionRequestPayloads[K]>): void;
-    chatCompletionResponse<K extends keyof ChatCompletionRequestPayloads>(_ctx: RenderTraceInfo, _provider: K, _payload: LogChatCompletionResponse<ChatCompletionRequestPayloads[K]>): void;
-}
-declare class BoundLogger implements Logger {
-    private readonly impl;
-    private readonly ctx;
-    constructor(impl: LogImplementation, ctx: RenderTraceInfo);
-    chainStart(chain: FunctionChain | StreamChain, payload: LogChainStart): void;
-    chainComplete(chain: FunctionChain | StreamChain, payload: LogChainComplete): void;
-    chainError(chain: FunctionChain | StreamChain, payload: LogChainError): void;
-    promptStart(prompt: Prompt, payload: LogPromptStart): void;
-    promptComplete(prompt: Prompt, payload: LogPromptComplete): void;
-    promptError(prompt: Prompt, payload: LogPromptError): void;
-    private formatMessage;
-    error: (...msgs: Loggable[]) => void;
-    warn: (...msgs: Loggable[]) => void;
-    info: (...msgs: Loggable[]) => void;
-    debug: (...msgs: Loggable[]) => void;
-    logException: (exception: unknown) => void;
-    chatCompletionRequest: <K extends keyof ChatCompletionRequestPayloads>(provider: K, payload: LogChatCompletionRequest<ChatCompletionRequestPayloads[K]>) => void;
-    chatCompletionResponse: <K extends keyof ChatCompletionRequestPayloads>(provider: K, payload: LogChatCompletionResponse<ChatCompletionRequestPayloads[K]>) => void;
+type Literal = string | number | null | undefined | boolean;
+interface RenderableStream extends AsyncGenerator<string, void, unknown> {
+    [Symbol.asyncIterator]: () => AsyncGenerator<string, void, unknown>;
 }
-declare class NoopLogImplementation extends LogImplementation {
-    log(_ctx: RenderTraceInfo, _level: LogLevel, _message: string): void;
+interface RenderResult extends RenderableStream {
+    then: (onResolved: (value: string) => void, onRejected?: (reason?: any) => void) => void;
 }
-declare class ConsoleLogger extends LogImplementation {
-    log(ctx: RenderTraceInfo, level: LogLevel, message: string): void;
+interface Context<T> {
+    Provider: AIComponent<{
+        children: AINode;
+        value: T;
+    }>;
+    defaultValue: T;
+    key: symbol;
 }
-declare class CombinedLogger extends LogImplementation {
-    private readonly loggers;
-    constructor(loggers: LogImplementation[]);
-    log(...args: Parameters<LogImplementation['log']>): void;
-    chainStart(...args: Parameters<LogImplementation['chainStart']>): void;
-    chainComplete(...args: Parameters<LogImplementation['chainComplete']>): void;
-    chainError(...args: Parameters<LogImplementation['chainError']>): void;
-    promptStart(...args: Parameters<LogImplementation['promptStart']>): void;
-    promptComplete(...args: Parameters<LogImplementation['promptComplete']>): void;
-    promptError(...args: Parameters<LogImplementation['promptError']>): void;
-    chatCompletionRequest<_K extends keyof ChatCompletionRequestPayloads>(...args: Parameters<LogImplementation['chatCompletionRequest']>): void;
-    chatCompletionResponse<_K extends keyof ChatCompletionRequestPayloads>(...args: Parameters<LogImplementation['chatCompletionResponse']>): void;
+type AIComponent<P> = (props: P, context: RenderContext) => Renderable;
+declare const attachedContextSymbol: unique symbol;
+interface AIElement<P> {
+    /** The tag associated with this {@link AIElement}. */
+    tag: AIComponent<P>;
+    /** The component properties. */
+    props: P;
+    /** A function that renders this {@link AIElement} to a {@link Renderable}. */
+    render: (ctx: RenderContext) => Renderable;
+    /** The {@link RenderContext} associated with this {@link Element}. */
+    [attachedContextSymbol]?: Record<symbol, any>;
 }
+type AINode = Literal | AIElement<any> | AINode[];
+type Renderable = AINode | PromiseLike<Renderable> | RenderableStream;
+type PropsOfAIComponent<T extends AIComponent<any>> = T extends AIComponent<infer P> ? P : never;
 
 type ChatCompletionRole = 'user' | 'system' | 'assistant';
 declare const SystemMessage: (props: {
@@ -292,8 +205,8 @@ declare const computeUsage: (messages: RenderedConversationMessage[]) => {
     total: number;
 };
 declare class ChatCompletionError extends Error {
-    readonly chatCompletionRequest: LogChatCompletionRequest;
-    constructor(message: string, chatCompletionRequest: LogChatCompletionRequest);
+    readonly chatCompletionRequest: Record<string, any>;
+    constructor(message: string, chatCompletionRequest: Record<string, any>);
 }
 
 declare function createAIElement<P extends {
@@ -306,4 +219,4 @@ declare function AIFragment({ children }: {
     children: AINode;
 }): Renderable;
 
-export { type AIComponent as A, type Logger as B, type ContextValues as C, BoundLogger as D, type EvaluatorFn as E, type FunctionChain as F, NoopLogImplementation as G, ConsoleLogger as H, CombinedLogger as I, type Literal as J, type RenderableStream as K, LogImplementation as L, type RenderResult as M, type NotAsyncGenerator as N, attachedContextSymbol as O, type PromptParsed as P, type AIElement as Q, type RenderContext as R, type StreamChain as S, type Renderable as T, UserMessage as U, type PropsOfAIComponent as V, type OutputParser as W, type PromptRenderResult as X, type Prompt as a, type EvaluatorResult as b, type Context as c, type AINode as d, createAIElement as e, AIFragment as f, LoggerContext as g, createContext as h, type RenderTraceInfo as i, type ChatCompletionRole as j, SystemMessage as k, AssistantMessage as l, type RenderedConversationMessage as m, computeUsage as n, ChatCompletionError as o, type LogChainStart as p, type LogChainComplete as q, type LogChainError as r, type LogPromptStart as s, type LogPromptComplete as t, type LogPromptError as u, type ChatCompletionRequestPayloads as v, type LogChatCompletionRequest as w, type LogChatCompletionResponse as x, type LogLevel as y, type LogContext as z };
+export { type AIComponent as A, BoundLogger as B, type ContextValues as C, type Renderable as D, type EvaluatorFn as E, type FunctionChain as F, type PropsOfAIComponent as G, type PromptRenderResult as H, LogImplementation as L, type NotAsyncGenerator as N, type OutputParser as O, type PromptParsed as P, type RenderContext as R, type StreamChain as S, UserMessage as U, type Prompt as a, type EvaluatorResult as b, type Context as c, type AINode as d, createAIElement as e, AIFragment as f, createContext as g, type ChatCompletionRole as h, SystemMessage as i, AssistantMessage as j, type RenderedConversationMessage as k, computeUsage as l, ChatCompletionError as m, type LogLevel as n, type Logger as o, NoopLogImplementation as p, ConsoleLogger as q, CombinedLogger as r, type ChatCompletionRequestPayloads as s, type LogChatCompletionRequest as t, type LogChatCompletionResponse as u, type Literal as v, type RenderableStream as w, type RenderResult as x, attachedContextSymbol as y, type AIElement as z };