@gammatech/aijsx 0.5.0-beta.1 → 0.5.0-dev.2024-03-11

package/dist/{createElement--C_Hwi_c.d.ts → jsx-dev-runtime-zWb34twz.d.ts}

@@ -1,83 +1,32 @@
 import { ZodTypeAny, ZodObject, ZodRawShape } from 'zod';
 
-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> {
-    endTime: string;
-    latency: number;
-    outputMessage: RenderedConversationMessage;
-    finishReason: string;
-    tokensUsed: {
-        prompt: number;
-        completion: number;
-        total: number;
-    };
-}
-
-/**
- * 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;
+type Literal = string | number | null | undefined | boolean;
+interface RenderResult extends AsyncGenerator<string, void> {
+    then(onFulfilled: (value: string) => string | PromiseLike<string>, onRejected?: (reason: any) => string | PromiseLike<string>): PromiseLike<string>;
 }
-declare class ConsoleLogger extends LogImplementation {
-    log(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;
-    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 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> | AsyncGenerator<string, void>;
+type PropsOfAIComponent<T extends AIComponent<any>> = T extends AIComponent<infer P> ? P : never;
 
 type EvaluatorPrimitive = string | number | boolean;
 type EvaluatorResult = {
@@ -92,9 +41,6 @@ type OutputParser<O> = {
     schema: ZodTypeAny;
     parse: (result: string) => O;
 };
-interface PromptRenderResult<O> extends RenderableStream {
-    then: (onResolved: (value: O) => void, onRejected?: (reason?: any) => void) => void;
-}
 interface Prompt<V extends Record<string, any> = Record<string, any>> {
     key: string;
     component: AIComponent<V>;
@@ -128,6 +74,115 @@ 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;
 
+type SpanContext = {
+    traceId: string;
+    spanId: string;
+};
+type SpanAttributes = Record<string, any>;
+type SpanStatus = 'unset' | 'ok' | 'error';
+interface ReadableSpan<Attrs extends SpanAttributes = SpanAttributes> {
+    name: string;
+    status: SpanStatus;
+    spanContext: SpanContext;
+    parentSpanId: string | null;
+    attributes: Attrs;
+    events: SpanEvent[];
+    startTime: string;
+    endTime: string | null;
+    ended: boolean;
+    duration: number;
+}
+interface Span<Attrs extends SpanAttributes = SpanAttributes> extends ReadableSpan<Attrs> {
+    setAttributes(attributes: Partial<Attrs>): this;
+    end(endTime?: [number, number]): void;
+    addEvent(name: string, attributes: SpanAttributes, startTime?: [number, number]): this;
+    recordException(exception: Error | string, time?: [number, number]): this;
+}
+interface SpanEvent {
+    name: string;
+    timestamp: string;
+    time: [number, number];
+    attributes: SpanAttributes;
+}
+interface TracingContextKey<_T> {
+    key: symbol;
+}
+interface TracingContext {
+    /**
+     * Get a value from the context.
+     *
+     * @param key key which identifies a context value
+     */
+    getValue<T>(key: TracingContextKey<T>): T | undefined;
+    /**
+     * Create a new context which inherits from this context and has
+     * the given key set to the given value.
+     *
+     * @param key context key for which to set the value
+     * @param value value to set for the given key
+     */
+    setValue<T>(key: TracingContextKey<T>, value: T): TracingContext;
+    /**
+     * Return a new context which inherits from this context but does
+     * not contain a value for the given key.
+     *
+     * @param key context key for which to clear a value
+     */
+    deleteValue<T>(key: TracingContextKey<T>): TracingContext;
+}
+interface TracingContextManager {
+    /**
+     * Get the current active context
+     */
+    active(): TracingContext;
+    /**
+     * Run the fn callback with object set as the current active context
+     * @param context Any object to set as the current active context
+     * @param fn A callback to be immediately run within a specific context
+     * @param thisArg optional receiver to be used for calling fn
+     * @param args optional arguments forwarded to fn
+     */
+    with<A extends unknown[], F extends (...args: A) => ReturnType<F>>(context: TracingContext, fn: F, ...args: A): ReturnType<F>;
+    /**
+     * Bind an object as the current context (or a specific one)
+     * @param [context] Optionally specify the context which you want to assign
+     * @param target Any object to which a context need to be set
+     */
+    bind<T>(context: TracingContext, target: T): T;
+    /**
+     * Enable context management
+     */
+    enable(): this;
+    /**
+     * Disable context management
+     */
+    disable(): this;
+}
+interface SpanProcessor {
+    /**
+     * Called when a {@link Span} is started, if the `span.isRecording()`
+     * returns true.
+     * @param span the Span that just started.
+     */
+    onStart?(span: ReadableSpan): void;
+    onSpanEvent?(event: SpanEvent, span: ReadableSpan): void;
+    onSpanException?(exceptionEvent: SpanEvent, span: ReadableSpan): void;
+    /**
+     * Called when a {@link ReadableSpan} is ended, if the `span.isRecording()`
+     * returns true.
+     * @param span the Span that just ended.
+     */
+    onEnd(span: ReadableSpan): void;
+}
+interface Tracer {
+    getActiveSpan(): Span | null;
+    trace<F extends (span: Span) => ReturnType<F>>(name: string, attributes: SpanAttributes, fn: F): ReturnType<F>;
+}
+interface SpanExporter {
+    export(spans: ReadableSpan[]): Promise<void>;
+    shutdown?(): Promise<void>;
+}
+
 type ContextValues = Record<symbol, any>;
 type RenderOptions = {
     preserveTags?: boolean;
@@ -138,6 +193,7 @@ type RenderOptions = {
     };
 };
 interface RenderContext {
+    tracer: Tracer;
     logger: Logger;
     getContext<T>(context: Context<T>): T;
     render(renderable: Renderable, opts?: RenderOptions): RenderResult;
@@ -145,44 +201,87 @@ interface RenderContext {
     render<V extends Record<string, any>>(prompt: Prompt<V>, variables: V): RenderResult;
     render<V extends Record<string, any>, O extends AsyncGenerator<any>>(chain: StreamChain<V, O>, variables: V): AsyncGenerator<O, void>;
     render<V extends Record<string, any>, O>(chain: FunctionChain<V, O>, variables: V): O;
-    runChain<V extends Record<string, any>, R>(chain: FunctionChain<V, R>, variables: V): R;
-    runChain<V extends Record<string, any>, R extends AsyncGenerator<any>>(chain: StreamChain<V, R>, variables: V): R;
 }
 type RenderPromptOptions = {
     validateOutput?: boolean;
 };
 declare function createContext<T>(defaultValue: T): Context<T>;
 
-type Literal = string | number | null | undefined | boolean;
-interface RenderableStream extends AsyncGenerator<string, void, unknown> {
-    [Symbol.asyncIterator]: () => AsyncGenerator<string, void, unknown>;
+/**
+ * This can be extended using declare module to add additional providers.
+ */
+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]> {
+    startTime: number;
+    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> {
+    latency: number;
+    outputMessage: RenderedConversationMessage;
+    finishReason: string | null;
+    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>;
+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: 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;
+};
+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: RenderContext, level: LogLevel, message: string): void;
+    /**
+     * Logs exceptions thrown during an element's render.
+     */
+    logException(ctx: RenderContext, exception: unknown): void;
+    chatCompletionRequest<K extends keyof ChatCompletionRequestPayloads>(_ctx: RenderContext, _provider: K, _payload: LogChatCompletionRequest<ChatCompletionRequestPayloads[K]>): void;
+    chatCompletionResponse<K extends keyof ChatCompletionRequestPayloads>(_ctx: RenderContext, _provider: K, _payload: LogChatCompletionResponse<ChatCompletionRequestPayloads[K]>): void;
+}
+declare class BoundLogger implements Logger {
+    private readonly impl;
+    private readonly ctx;
+    constructor(impl: LogImplementation, ctx: RenderContext);
+    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;
+}
+declare class NoopLogImplementation extends LogImplementation {
+    log(_ctx: RenderContext, _level: LogLevel, _message: string): void;
+}
+declare class ConsoleLogger extends LogImplementation {
+    log(ctx: RenderContext, level: LogLevel, message: string): void;
+}
+declare class CombinedLogger extends LogImplementation {
+    private readonly loggers;
+    constructor(loggers: LogImplementation[]);
+    log(...args: Parameters<LogImplementation['log']>): 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 ChatCompletionRole = 'user' | 'system' | 'assistant';
 declare const SystemMessage: (props: {
@@ -205,8 +304,8 @@ declare const computeUsage: (messages: RenderedConversationMessage[]) => {
     total: number;
 };
 declare class ChatCompletionError extends Error {
-    readonly chatCompletionRequest: Record<string, any>;
-    constructor(message: string, chatCompletionRequest: Record<string, any>);
+    readonly chatCompletionRequest: LogChatCompletionRequest;
+    constructor(message: string, chatCompletionRequest: LogChatCompletionRequest);
 }
 
 declare function createAIElement<P extends {
@@ -219,4 +318,30 @@ declare function AIFragment({ children }: {
     children: AINode;
 }): Renderable;
 
-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 };
+/**
+ * The is used as an import source for ts/js files as the JSX transpile functinos
+ */
+
+/** @hidden */
+declare namespace JSX {
+    type ElementType = AIComponent<any>;
+    interface Element extends AIElement<any> {
+    }
+    interface IntrinsicElements {
+    }
+    interface ElementChildrenAttribute {
+        children: {};
+    }
+}
+/** @hidden */
+declare function jsx(type: any, config: any, maybeKey?: any): AIElement<{
+    children: any[];
+}>;
+/** @hidden */
+declare const jsxDEV: typeof jsx;
+/** @hidden */
+declare const jsxs: typeof jsx;
+/** @hidden */
+declare const Fragment: typeof AIFragment;
+
+export { jsxDEV as $, type AIComponent as A, BoundLogger as B, type ContextValues as C, type RenderResult as D, type EvaluatorFn as E, type FunctionChain as F, attachedContextSymbol as G, type AIElement as H, type Renderable as I, JSX as J, type PropsOfAIComponent as K, LogImplementation as L, type SpanContext as M, type NotAsyncGenerator as N, type SpanStatus as O, type PromptParsed as P, type Span as Q, type RenderContext as R, type SpanProcessor as S, type Tracer as T, UserMessage as U, type SpanEvent as V, type TracingContextKey as W, type TracingContext as X, type TracingContextManager as Y, type OutputParser as Z, jsx as _, type ReadableSpan as a, jsxs as a0, Fragment as a1, type SpanExporter as b, type AINode as c, type SpanAttributes as d, type Prompt as e, type StreamChain as f, type EvaluatorResult as g, type Context as h, createAIElement as i, AIFragment as j, createContext as k, type ChatCompletionRole as l, SystemMessage as m, AssistantMessage as n, type RenderedConversationMessage as o, computeUsage as p, ChatCompletionError as q, type ChatCompletionRequestPayloads as r, type LogChatCompletionRequest as s, type LogChatCompletionResponse as t, type LogLevel as u, type Logger as v, NoopLogImplementation as w, ConsoleLogger as x, CombinedLogger as y, type Literal as z };

package/dist/jsx-dev-runtime.d.mts

@@ -1,3 +1,2 @@
-export { Fragment, JSX, jsx, jsxDEV, jsxs } from './jsx-runtime.mjs';
-import './createElement--C_Hwi_c.mjs';
+export { a1 as Fragment, J as JSX, _ as jsx, $ as jsxDEV, a0 as jsxs } from './jsx-dev-runtime-zWb34twz.mjs';
 import 'zod';

package/dist/jsx-dev-runtime.d.ts

@@ -1,3 +1,2 @@
-export { Fragment, JSX, jsx, jsxDEV, jsxs } from './jsx-runtime.js';
-import './createElement--C_Hwi_c.js';
+export { a1 as Fragment, J as JSX, _ as jsx, $ as jsxDEV, a0 as jsxs } from './jsx-dev-runtime-zWb34twz.js';
 import 'zod';

package/dist/jsx-runtime.d.mts

@@ -1,30 +1,2 @@
-import { A as AIComponent, z as AIElement, f as AIFragment } from './createElement--C_Hwi_c.mjs';
+export { a1 as Fragment, J as JSX, _ as jsx, $ as jsxDEV, a0 as jsxs } from './jsx-dev-runtime-zWb34twz.mjs';
 import 'zod';
-
-/**
- * The is used as an import source for ts/js files as the JSX transpile functinos
- */
-
-/** @hidden */
-declare namespace JSX {
-    type ElementType = AIComponent<any>;
-    interface Element extends AIElement<any> {
-    }
-    interface IntrinsicElements {
-    }
-    interface ElementChildrenAttribute {
-        children: {};
-    }
-}
-/** @hidden */
-declare function jsx(type: any, config: any, maybeKey?: any): AIElement<{
-    children: any[];
-}>;
-/** @hidden */
-declare const jsxDEV: typeof jsx;
-/** @hidden */
-declare const jsxs: typeof jsx;
-/** @hidden */
-declare const Fragment: typeof AIFragment;
-
-export { Fragment, JSX, jsx, jsxDEV, jsxs };

package/dist/jsx-runtime.d.ts

@@ -1,30 +1,2 @@
-import { A as AIComponent, z as AIElement, f as AIFragment } from './createElement--C_Hwi_c.js';
+export { a1 as Fragment, J as JSX, _ as jsx, $ as jsxDEV, a0 as jsxs } from './jsx-dev-runtime-zWb34twz.js';
 import 'zod';
-
-/**
- * The is used as an import source for ts/js files as the JSX transpile functinos
- */
-
-/** @hidden */
-declare namespace JSX {
-    type ElementType = AIComponent<any>;
-    interface Element extends AIElement<any> {
-    }
-    interface IntrinsicElements {
-    }
-    interface ElementChildrenAttribute {
-        children: {};
-    }
-}
-/** @hidden */
-declare function jsx(type: any, config: any, maybeKey?: any): AIElement<{
-    children: any[];
-}>;
-/** @hidden */
-declare const jsxDEV: typeof jsx;
-/** @hidden */
-declare const jsxs: typeof jsx;
-/** @hidden */
-declare const Fragment: typeof AIFragment;
-
-export { Fragment, JSX, jsx, jsxDEV, jsxs };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gammatech/aijsx",
-  "version": "0.5.0-beta.1",
+  "version": "0.5.0-dev.2024-03-11",
   "description": "Rewrite of aijsx",
   "author": "Jordan Garcia",
   "license": "MIT",
@@ -9,6 +9,7 @@
   "types": "./dist/index.d.ts",
   "scripts": {
     "dev": "ts-node ./src/test.tsx",
+    "bench": "npx tsx ./benchmark/bench.tsx",
     "test": "jest --verbose",
     "test:watch": "jest --watch --verbose --silent false",
     "build": "yarn check-types && tsup",
@@ -17,7 +18,7 @@
     "check-types": "tsc --skipLibCheck --noEmit"
   },
   "dependencies": {
-    "@anthropic-ai/sdk": "^0.12.0",
+    "@anthropic-ai/sdk": "0.17.1",
     "@anthropic-ai/tokenizer": "^0.0.4",
     "fast-xml-parser": "^4.3.4",
     "js-tiktoken": "^1.0.8",