@gammatech/aijsx 0.3.0-beta.1 → 0.3.0-beta.11

package/dist/{chunk-2JPDWMO2.mjs → chunk-QIA7MVVR.mjs}

@@ -1,6 +1,5 @@
 // src/createElement.ts
-var parseAsString = (result) => result;
-function createAIElement(tag, props, children) {
+function createAIElement(tag, props, ...children) {
   const propsToPass = {
     ...props ?? {},
     ...children.length === 0 ? {} : { children: children.length === 1 ? children[0] : children }
@@ -8,16 +7,12 @@ function createAIElement(tag, props, children) {
   const result = {
     tag,
     props: propsToPass,
-    parse: "parse" in tag ? tag.parse : parseAsString,
     render: (ctx) => {
       return tag(propsToPass, ctx);
     }
   };
   return result;
 }
-function isParsedAIElement(value) {
-  return value !== null && typeof value === "object" && "tag" in value && "parse" in value && "render" in value;
-}
 function isAIElement(value) {
   return value !== null && typeof value === "object" && "tag" in value && "render" in value;
 }
@@ -29,10 +24,23 @@ function AIFragment({ children }) {
   return children;
 }
 
+// src/jsx-runtime.ts
+function jsx(type, config, maybeKey) {
+  const configWithKey = maybeKey !== void 0 ? { ...config, key: maybeKey } : config;
+  const children = config && Array.isArray(config.children) ? config.children : [];
+  return createAIElement(type, configWithKey, ...children);
+}
+var jsxDEV = jsx;
+var jsxs = jsx;
+var Fragment = AIFragment;
+
 export {
   createAIElement,
-  isParsedAIElement,
   isAIElement,
   isLiteral,
-  AIFragment
+  AIFragment,
+  jsx,
+  jsxDEV,
+  jsxs,
+  Fragment
 };

package/dist/{createElement-jIxJ-zr7.d.mts → createElement-SURWK7y8.d.mts}

@@ -1,7 +1,6 @@
+import { ZodTypeAny, ZodObject, ZodRawShape } from 'zod';
+
 type Literal = string | number | null | undefined | boolean;
-type JSONSerializable = string | number | boolean | null | JSONSerializable[] | {
-    [key: string]: JSONSerializable;
-};
 interface RenderableStream {
     [Symbol.asyncIterator]: () => AsyncGenerator<string, void, unknown>;
 }
@@ -12,37 +11,77 @@ interface Context<T> {
     Provider: AIComponent<{
         children: AINode;
         value: T;
-    }, string>;
+    }>;
     defaultValue: T;
     key: symbol;
 }
-interface AIComponentString<P, _Result extends string> {
-    (props: P, context: RenderContext): Renderable;
-}
-interface AIComponentParsed<P, ParsedResult extends JSONSerializable = string> {
-    (props: P, context: RenderContext): Renderable;
-    parse: (result: string) => ParsedResult;
-}
-type AINode = Literal | AIElement<AIComponent<any, any>> | AINode[];
-type Renderable = AINode | PromiseLike<Renderable> | RenderableStream;
-type AIComponent<P, R extends string | JSONSerializable> = R extends string ? AIComponentString<P, R> : R extends JSONSerializable ? AIComponentParsed<P, R> : never;
+type AIComponent<P> = (props: P, context: RenderContext) => Renderable;
 declare const attachedContextSymbol: unique symbol;
-interface AIElement<C extends AIComponent<any, any>> {
+interface AIElement<P> {
     /** The tag associated with this {@link AIElement}. */
-    tag: C;
+    tag: AIComponent<P>;
     /** The component properties. */
-    props: PropsOfAIComponent<C>;
+    props: P;
     /** A function that renders this {@link AIElement} to a {@link Renderable}. */
     render: (ctx: RenderContext) => Renderable;
-    parse: (result: string) => ParsedTypeOfAIComponent<C>;
     /** The {@link RenderContext} associated with this {@link Element}. */
     [attachedContextSymbol]?: Record<symbol, any>;
 }
-type PropsOfAIComponent<T extends AIComponent<any, any>> = T extends AIComponent<infer P, any> ? P : never;
-type ParsedTypeOfAIComponent<T extends AIComponent<any, any>> = T extends AIComponent<any, infer P> ? P : never;
-type ParsedTypeOfRenderable<R extends Renderable> = R extends AIElement<infer C> ? C extends AIComponentParsed<any, infer U> ? U : string : string;
+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 = {
+    key: string;
+    result: EvaluatorPrimitive | EvaluatorPrimitive[];
+    passes: boolean;
+    description?: string;
+    debug?: string;
+};
+type EvaluatorFn<V extends Record<string, any> = Record<string, any>, O = any> = (variables: V, result: O) => Promise<EvaluatorResult | EvaluatorResult[]>;
+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>;
+    schema: ZodObject<ZodRawShape>;
+    evaluators: EvaluatorFn<V>[];
+    messageSchema: ZodTypeAny | null;
+    outputSchema: ZodTypeAny;
+    metadata: Record<string, any>;
+}
+interface PromptParsed<V extends Record<string, any> = Record<string, any>, O = any> extends Prompt<V> {
+    evaluators: EvaluatorFn<V, O>[];
+    messageSchema: null;
+    parse: (result: string) => O;
+}
+interface FunctionChain<V extends Record<string, any> = Record<string, any>, R extends NotAsyncGenerator<any> = any> {
+    key: string;
+    type: 'function';
+    run: (vars: V, context: RenderContext) => R;
+    schema: ZodObject<ZodRawShape>;
+    messageSchema: null;
+    outputSchema: ZodTypeAny;
+}
+interface StreamChain<V extends Record<string, any> = Record<string, any>, R extends AsyncGenerator<any, any, any> = AsyncGenerator<any, any, any>> {
+    key: string;
+    type: 'stream';
+    run: (vars: V, context: RenderContext) => R;
+    schema: ZodObject<ZodRawShape>;
+    messageSchema: ZodTypeAny;
+    outputSchema: null;
+}
+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;
     renderedProps?: {
@@ -53,13 +92,21 @@ type RenderOptions = {
 };
 interface RenderContext {
     parentContext: RenderContext | null;
-    element: AIElement<AIComponent<any, any>>;
+    element: AIElement<any>;
     renderId: string;
     logger: Logger;
     getContext<T>(context: Context<T>): T;
     render(renderable: Renderable, opts?: RenderOptions): RenderResult;
-    renderJson<R extends Renderable>(renderable: R): Promise<ParsedTypeOfRenderable<R>>;
-}
+    render<V extends Record<string, any>, O>(prompt: PromptParsed<V, O>, variables: V, options?: RenderPromptOptions): Promise<O>;
+    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>;
 
 /**
@@ -163,11 +210,14 @@ declare class ChatCompletionError extends Error {
     constructor(message: string, chatCompletionRequest: LogChatCompletionRequest);
 }
 
+declare function createAIElement<P extends {
+    children: C;
+}, C>(tag: AIComponent<P>, props: Omit<P, 'children'> | null, ...children: [C]): AIElement<P>;
 declare function createAIElement<P extends {
     children: C[];
-}, C, T extends AIComponent<P, any>>(tag: T, props: Omit<P, 'children'> | null, children: C[]): AIElement<T>;
+}, C>(tag: AIComponent<P>, props: Omit<P, 'children'> | null, ...children: C[]): AIElement<P>;
 declare function AIFragment({ children }: {
     children: AINode;
 }): Renderable;
 
-export { type AINode as A, BoundLogger as B, type Context as C, type ParsedTypeOfRenderable as D, type JSONSerializable as J, LogImplementation as L, NoopLogImplementation as N, type PropsOfAIComponent as P, type RenderContext as R, SystemMessage as S, UserMessage as U, AIFragment as a, LoggerContext as b, createAIElement as c, createContext as d, type ChatCompletionRole as e, AssistantMessage as f, type RenderedConversationMessage as g, computeUsage as h, ChatCompletionError as i, type ChatCompletionRequestPayloads as j, type LogChatCompletionRequest as k, type LogChatCompletionResponse as l, type LogLevel as m, type Logger as n, ConsoleLogger as o, CombinedLogger as p, type Literal as q, type RenderableStream as r, type RenderResult as s, type AIComponentString as t, type AIComponentParsed as u, type Renderable as v, type AIComponent as w, attachedContextSymbol as x, type AIElement as y, type ParsedTypeOfAIComponent 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 EvaluatorResult as H, type PromptRenderResult as I, 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 Context as b, type AINode as c, createAIElement as d, AIFragment as e, LoggerContext 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 ChatCompletionRequestPayloads as n, type LogChatCompletionRequest as o, type LogChatCompletionResponse as p, type LogLevel as q, type Logger as r, NoopLogImplementation as s, ConsoleLogger as t, CombinedLogger as u, type Literal as v, type RenderableStream as w, type RenderResult as x, attachedContextSymbol as y, type AIElement as z };

package/dist/{createElement-jIxJ-zr7.d.ts → createElement-SURWK7y8.d.ts}

@@ -1,7 +1,6 @@
+import { ZodTypeAny, ZodObject, ZodRawShape } from 'zod';
+
 type Literal = string | number | null | undefined | boolean;
-type JSONSerializable = string | number | boolean | null | JSONSerializable[] | {
-    [key: string]: JSONSerializable;
-};
 interface RenderableStream {
     [Symbol.asyncIterator]: () => AsyncGenerator<string, void, unknown>;
 }
@@ -12,37 +11,77 @@ interface Context<T> {
     Provider: AIComponent<{
         children: AINode;
         value: T;
-    }, string>;
+    }>;
     defaultValue: T;
     key: symbol;
 }
-interface AIComponentString<P, _Result extends string> {
-    (props: P, context: RenderContext): Renderable;
-}
-interface AIComponentParsed<P, ParsedResult extends JSONSerializable = string> {
-    (props: P, context: RenderContext): Renderable;
-    parse: (result: string) => ParsedResult;
-}
-type AINode = Literal | AIElement<AIComponent<any, any>> | AINode[];
-type Renderable = AINode | PromiseLike<Renderable> | RenderableStream;
-type AIComponent<P, R extends string | JSONSerializable> = R extends string ? AIComponentString<P, R> : R extends JSONSerializable ? AIComponentParsed<P, R> : never;
+type AIComponent<P> = (props: P, context: RenderContext) => Renderable;
 declare const attachedContextSymbol: unique symbol;
-interface AIElement<C extends AIComponent<any, any>> {
+interface AIElement<P> {
     /** The tag associated with this {@link AIElement}. */
-    tag: C;
+    tag: AIComponent<P>;
     /** The component properties. */
-    props: PropsOfAIComponent<C>;
+    props: P;
     /** A function that renders this {@link AIElement} to a {@link Renderable}. */
     render: (ctx: RenderContext) => Renderable;
-    parse: (result: string) => ParsedTypeOfAIComponent<C>;
     /** The {@link RenderContext} associated with this {@link Element}. */
     [attachedContextSymbol]?: Record<symbol, any>;
 }
-type PropsOfAIComponent<T extends AIComponent<any, any>> = T extends AIComponent<infer P, any> ? P : never;
-type ParsedTypeOfAIComponent<T extends AIComponent<any, any>> = T extends AIComponent<any, infer P> ? P : never;
-type ParsedTypeOfRenderable<R extends Renderable> = R extends AIElement<infer C> ? C extends AIComponentParsed<any, infer U> ? U : string : string;
+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 = {
+    key: string;
+    result: EvaluatorPrimitive | EvaluatorPrimitive[];
+    passes: boolean;
+    description?: string;
+    debug?: string;
+};
+type EvaluatorFn<V extends Record<string, any> = Record<string, any>, O = any> = (variables: V, result: O) => Promise<EvaluatorResult | EvaluatorResult[]>;
+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>;
+    schema: ZodObject<ZodRawShape>;
+    evaluators: EvaluatorFn<V>[];
+    messageSchema: ZodTypeAny | null;
+    outputSchema: ZodTypeAny;
+    metadata: Record<string, any>;
+}
+interface PromptParsed<V extends Record<string, any> = Record<string, any>, O = any> extends Prompt<V> {
+    evaluators: EvaluatorFn<V, O>[];
+    messageSchema: null;
+    parse: (result: string) => O;
+}
+interface FunctionChain<V extends Record<string, any> = Record<string, any>, R extends NotAsyncGenerator<any> = any> {
+    key: string;
+    type: 'function';
+    run: (vars: V, context: RenderContext) => R;
+    schema: ZodObject<ZodRawShape>;
+    messageSchema: null;
+    outputSchema: ZodTypeAny;
+}
+interface StreamChain<V extends Record<string, any> = Record<string, any>, R extends AsyncGenerator<any, any, any> = AsyncGenerator<any, any, any>> {
+    key: string;
+    type: 'stream';
+    run: (vars: V, context: RenderContext) => R;
+    schema: ZodObject<ZodRawShape>;
+    messageSchema: ZodTypeAny;
+    outputSchema: null;
+}
+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;
     renderedProps?: {
@@ -53,13 +92,21 @@ type RenderOptions = {
 };
 interface RenderContext {
     parentContext: RenderContext | null;
-    element: AIElement<AIComponent<any, any>>;
+    element: AIElement<any>;
     renderId: string;
     logger: Logger;
     getContext<T>(context: Context<T>): T;
     render(renderable: Renderable, opts?: RenderOptions): RenderResult;
-    renderJson<R extends Renderable>(renderable: R): Promise<ParsedTypeOfRenderable<R>>;
-}
+    render<V extends Record<string, any>, O>(prompt: PromptParsed<V, O>, variables: V, options?: RenderPromptOptions): Promise<O>;
+    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>;
 
 /**
@@ -163,11 +210,14 @@ declare class ChatCompletionError extends Error {
     constructor(message: string, chatCompletionRequest: LogChatCompletionRequest);
 }
 
+declare function createAIElement<P extends {
+    children: C;
+}, C>(tag: AIComponent<P>, props: Omit<P, 'children'> | null, ...children: [C]): AIElement<P>;
 declare function createAIElement<P extends {
     children: C[];
-}, C, T extends AIComponent<P, any>>(tag: T, props: Omit<P, 'children'> | null, children: C[]): AIElement<T>;
+}, C>(tag: AIComponent<P>, props: Omit<P, 'children'> | null, ...children: C[]): AIElement<P>;
 declare function AIFragment({ children }: {
     children: AINode;
 }): Renderable;
 
-export { type AINode as A, BoundLogger as B, type Context as C, type ParsedTypeOfRenderable as D, type JSONSerializable as J, LogImplementation as L, NoopLogImplementation as N, type PropsOfAIComponent as P, type RenderContext as R, SystemMessage as S, UserMessage as U, AIFragment as a, LoggerContext as b, createAIElement as c, createContext as d, type ChatCompletionRole as e, AssistantMessage as f, type RenderedConversationMessage as g, computeUsage as h, ChatCompletionError as i, type ChatCompletionRequestPayloads as j, type LogChatCompletionRequest as k, type LogChatCompletionResponse as l, type LogLevel as m, type Logger as n, ConsoleLogger as o, CombinedLogger as p, type Literal as q, type RenderableStream as r, type RenderResult as s, type AIComponentString as t, type AIComponentParsed as u, type Renderable as v, type AIComponent as w, attachedContextSymbol as x, type AIElement as y, type ParsedTypeOfAIComponent 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 EvaluatorResult as H, type PromptRenderResult as I, 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 Context as b, type AINode as c, createAIElement as d, AIFragment as e, LoggerContext 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 ChatCompletionRequestPayloads as n, type LogChatCompletionRequest as o, type LogChatCompletionResponse as p, type LogLevel as q, type Logger as r, NoopLogImplementation as s, ConsoleLogger as t, CombinedLogger as u, type Literal as v, type RenderableStream as w, type RenderResult as x, attachedContextSymbol as y, type AIElement as z };

package/dist/index.d.mts

@@ -1,5 +1,6 @@
-import { L as LogImplementation, R as RenderContext, C as Context, A as AINode } from './createElement-jIxJ-zr7.mjs';
-export { w as AIComponent, u as AIComponentParsed, t as AIComponentString, y as AIElement, a as AIFragment, f as AssistantMessage, B as BoundLogger, i as ChatCompletionError, j as ChatCompletionRequestPayloads, e as ChatCompletionRole, p as CombinedLogger, o as ConsoleLogger, J as JSONSerializable, q as Literal, k as LogChatCompletionRequest, l as LogChatCompletionResponse, m as LogLevel, n as Logger, b as LoggerContext, N as NoopLogImplementation, z as ParsedTypeOfAIComponent, D as ParsedTypeOfRenderable, P as PropsOfAIComponent, s as RenderResult, v as Renderable, r as RenderableStream, g as RenderedConversationMessage, S as SystemMessage, U as UserMessage, x as attachedContextSymbol, h as computeUsage, c as createAIElement, d as createContext } from './createElement-jIxJ-zr7.mjs';
+import { R as RenderContext, L as LogImplementation, C as ContextValues, A as AIComponent, E as EvaluatorFn, P as PromptParsed, a as Prompt, N as NotAsyncGenerator, F as FunctionChain, S as StreamChain, b as Context, c as AINode } from './createElement-SURWK7y8.mjs';
+export { z as AIElement, e as AIFragment, j as AssistantMessage, B as BoundLogger, m as ChatCompletionError, n as ChatCompletionRequestPayloads, h as ChatCompletionRole, u as CombinedLogger, t as ConsoleLogger, H as EvaluatorResult, v as Literal, o as LogChatCompletionRequest, p as LogChatCompletionResponse, q as LogLevel, r as Logger, f as LoggerContext, s as NoopLogImplementation, O as OutputParser, I as PromptRenderResult, G as PropsOfAIComponent, x as RenderResult, D as Renderable, w as RenderableStream, k as RenderedConversationMessage, i as SystemMessage, U as UserMessage, y as attachedContextSymbol, l as computeUsage, d as createAIElement, g as createContext } from './createElement-SURWK7y8.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';
@@ -7,10 +8,54 @@ import AnthropicClient from '@anthropic-ai/sdk';
 export { default as AnthropicClient } from '@anthropic-ai/sdk';
 export { countTokens as countAnthropicTokens } from '@anthropic-ai/tokenizer';
 
-declare function createRenderContext({ logger, rootRenderId, }?: {
+type CreateRenderContextOptions = {
     logger?: LogImplementation;
     rootRenderId?: string;
-}): RenderContext;
+    contextValues?: ContextValues;
+};
+declare function createRenderContext({ logger, rootRenderId, contextValues, }?: CreateRenderContextOptions): RenderContext;
+
+/**
+ * This is a helper function to create a JsxPrompt type
+ */
+declare function createPrompt<K extends string, I extends ZodObject<ZodRawShape>, O extends ZodTypeAny = ZodString>(config: {
+    key: K;
+    schema: I;
+    component: AIComponent<z.infer<I>>;
+    evaluators?: EvaluatorFn<z.infer<I>, z.infer<O>>[];
+    metadata?: Record<string, any>;
+    outputParser: {
+        schema: O;
+        parse: (result: string) => z.infer<O>;
+    };
+}): PromptParsed<z.infer<I>, z.infer<O>>;
+declare function createPrompt<K extends string, I extends ZodObject<ZodRawShape>>(config: {
+    key: K;
+    schema: I;
+    component: AIComponent<z.infer<I>>;
+    evaluators?: EvaluatorFn<z.infer<I>, string>[];
+    metadata?: Record<string, any>;
+}): Prompt<z.infer<I>>;
+
+declare function createFunctionChain<I extends ZodObject<ZodRawShape>, R extends NotAsyncGenerator<any>>(chain: {
+    key: string;
+    run: (variables: z.infer<I>, context: RenderContext) => R;
+    outputSchema?: ZodTypeAny;
+    schema: I;
+}): FunctionChain<z.infer<I>, R>;
+declare function createStreamChain<I extends ZodObject<ZodRawShape>, M extends ZodTypeAny, R extends AsyncGenerator<any, any, any>>(chain: {
+    key: string;
+    run: (variables: z.infer<I>, context: RenderContext) => R;
+    schema: I;
+    messageSchema?: M;
+}): StreamChain<z.infer<I>, R>;
+
+declare class PromptParseVariablesError extends Error {
+}
+declare class ChainParseVariablesError extends Error {
+}
+declare class PromptInvalidOutputError extends Error {
+}
 
 type OpenAIChatCompletionRequest = OpenAI.Chat.Completions.ChatCompletionCreateParamsStreaming;
 type OpenAIChatMessage = ChatCompletionSystemMessageParam | ChatCompletionUserMessageParam | ChatCompletionAssistantMessageParam;
@@ -92,4 +137,4 @@ type AnthropicChatCompletionProps = {
  */
 declare function AnthropicChatCompletion(props: AnthropicChatCompletionProps, { render, logger, getContext }: RenderContext): AsyncGenerator<string, void, unknown>;
 
-export { AINode, AnthropicChatCompletion, type AnthropicChatCompletionRequest, AnthropicClientContext, ContentTypeImage, Context, LogImplementation, OpenAIChatCompletion, type OpenAIChatCompletionRequest, type OpenAIChatMessage, OpenAIClientContext, OpenAIVisionChatCompletion, RenderContext, type ValidAnthropicChatModel, type ValidOpenAIChatModel, type ValidOpenAIVisionModel, createRenderContext, defaultMaxTokens, tokenCountForOpenAIMessage, tokenCountForOpenAIVisionMessage, tokenLimitForChatModel, tokenizer };
+export { AIComponent, AINode, AnthropicChatCompletion, type AnthropicChatCompletionRequest, AnthropicClientContext, ChainParseVariablesError, ContentTypeImage, Context, EvaluatorFn, FunctionChain, LogImplementation, NotAsyncGenerator, OpenAIChatCompletion, type OpenAIChatCompletionRequest, type OpenAIChatMessage, OpenAIClientContext, OpenAIVisionChatCompletion, Prompt, PromptInvalidOutputError, PromptParseVariablesError, PromptParsed, RenderContext, StreamChain, type ValidAnthropicChatModel, type ValidOpenAIChatModel, type ValidOpenAIVisionModel, createFunctionChain, createPrompt, createRenderContext, createStreamChain, defaultMaxTokens, tokenCountForOpenAIMessage, tokenCountForOpenAIVisionMessage, tokenLimitForChatModel, tokenizer };

package/dist/index.d.ts

@@ -1,5 +1,6 @@
-import { L as LogImplementation, R as RenderContext, C as Context, A as AINode } from './createElement-jIxJ-zr7.js';
-export { w as AIComponent, u as AIComponentParsed, t as AIComponentString, y as AIElement, a as AIFragment, f as AssistantMessage, B as BoundLogger, i as ChatCompletionError, j as ChatCompletionRequestPayloads, e as ChatCompletionRole, p as CombinedLogger, o as ConsoleLogger, J as JSONSerializable, q as Literal, k as LogChatCompletionRequest, l as LogChatCompletionResponse, m as LogLevel, n as Logger, b as LoggerContext, N as NoopLogImplementation, z as ParsedTypeOfAIComponent, D as ParsedTypeOfRenderable, P as PropsOfAIComponent, s as RenderResult, v as Renderable, r as RenderableStream, g as RenderedConversationMessage, S as SystemMessage, U as UserMessage, x as attachedContextSymbol, h as computeUsage, c as createAIElement, d as createContext } from './createElement-jIxJ-zr7.js';
+import { R as RenderContext, L as LogImplementation, C as ContextValues, A as AIComponent, E as EvaluatorFn, P as PromptParsed, a as Prompt, N as NotAsyncGenerator, F as FunctionChain, S as StreamChain, b as Context, c as AINode } from './createElement-SURWK7y8.js';
+export { z as AIElement, e as AIFragment, j as AssistantMessage, B as BoundLogger, m as ChatCompletionError, n as ChatCompletionRequestPayloads, h as ChatCompletionRole, u as CombinedLogger, t as ConsoleLogger, H as EvaluatorResult, v as Literal, o as LogChatCompletionRequest, p as LogChatCompletionResponse, q as LogLevel, r as Logger, f as LoggerContext, s as NoopLogImplementation, O as OutputParser, I as PromptRenderResult, G as PropsOfAIComponent, x as RenderResult, D as Renderable, w as RenderableStream, k as RenderedConversationMessage, i as SystemMessage, U as UserMessage, y as attachedContextSymbol, l as computeUsage, d as createAIElement, g as createContext } from './createElement-SURWK7y8.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';
@@ -7,10 +8,54 @@ import AnthropicClient from '@anthropic-ai/sdk';
 export { default as AnthropicClient } from '@anthropic-ai/sdk';
 export { countTokens as countAnthropicTokens } from '@anthropic-ai/tokenizer';
 
-declare function createRenderContext({ logger, rootRenderId, }?: {
+type CreateRenderContextOptions = {
     logger?: LogImplementation;
     rootRenderId?: string;
-}): RenderContext;
+    contextValues?: ContextValues;
+};
+declare function createRenderContext({ logger, rootRenderId, contextValues, }?: CreateRenderContextOptions): RenderContext;
+
+/**
+ * This is a helper function to create a JsxPrompt type
+ */
+declare function createPrompt<K extends string, I extends ZodObject<ZodRawShape>, O extends ZodTypeAny = ZodString>(config: {
+    key: K;
+    schema: I;
+    component: AIComponent<z.infer<I>>;
+    evaluators?: EvaluatorFn<z.infer<I>, z.infer<O>>[];
+    metadata?: Record<string, any>;
+    outputParser: {
+        schema: O;
+        parse: (result: string) => z.infer<O>;
+    };
+}): PromptParsed<z.infer<I>, z.infer<O>>;
+declare function createPrompt<K extends string, I extends ZodObject<ZodRawShape>>(config: {
+    key: K;
+    schema: I;
+    component: AIComponent<z.infer<I>>;
+    evaluators?: EvaluatorFn<z.infer<I>, string>[];
+    metadata?: Record<string, any>;
+}): Prompt<z.infer<I>>;
+
+declare function createFunctionChain<I extends ZodObject<ZodRawShape>, R extends NotAsyncGenerator<any>>(chain: {
+    key: string;
+    run: (variables: z.infer<I>, context: RenderContext) => R;
+    outputSchema?: ZodTypeAny;
+    schema: I;
+}): FunctionChain<z.infer<I>, R>;
+declare function createStreamChain<I extends ZodObject<ZodRawShape>, M extends ZodTypeAny, R extends AsyncGenerator<any, any, any>>(chain: {
+    key: string;
+    run: (variables: z.infer<I>, context: RenderContext) => R;
+    schema: I;
+    messageSchema?: M;
+}): StreamChain<z.infer<I>, R>;
+
+declare class PromptParseVariablesError extends Error {
+}
+declare class ChainParseVariablesError extends Error {
+}
+declare class PromptInvalidOutputError extends Error {
+}
 
 type OpenAIChatCompletionRequest = OpenAI.Chat.Completions.ChatCompletionCreateParamsStreaming;
 type OpenAIChatMessage = ChatCompletionSystemMessageParam | ChatCompletionUserMessageParam | ChatCompletionAssistantMessageParam;
@@ -92,4 +137,4 @@ type AnthropicChatCompletionProps = {
  */
 declare function AnthropicChatCompletion(props: AnthropicChatCompletionProps, { render, logger, getContext }: RenderContext): AsyncGenerator<string, void, unknown>;
 
-export { AINode, AnthropicChatCompletion, type AnthropicChatCompletionRequest, AnthropicClientContext, ContentTypeImage, Context, LogImplementation, OpenAIChatCompletion, type OpenAIChatCompletionRequest, type OpenAIChatMessage, OpenAIClientContext, OpenAIVisionChatCompletion, RenderContext, type ValidAnthropicChatModel, type ValidOpenAIChatModel, type ValidOpenAIVisionModel, createRenderContext, defaultMaxTokens, tokenCountForOpenAIMessage, tokenCountForOpenAIVisionMessage, tokenLimitForChatModel, tokenizer };
+export { AIComponent, AINode, AnthropicChatCompletion, type AnthropicChatCompletionRequest, AnthropicClientContext, ChainParseVariablesError, ContentTypeImage, Context, EvaluatorFn, FunctionChain, LogImplementation, NotAsyncGenerator, OpenAIChatCompletion, type OpenAIChatCompletionRequest, type OpenAIChatMessage, OpenAIClientContext, OpenAIVisionChatCompletion, Prompt, PromptInvalidOutputError, PromptParseVariablesError, PromptParsed, RenderContext, StreamChain, type ValidAnthropicChatModel, type ValidOpenAIChatModel, type ValidOpenAIVisionModel, createFunctionChain, createPrompt, createRenderContext, createStreamChain, defaultMaxTokens, tokenCountForOpenAIMessage, tokenCountForOpenAIVisionMessage, tokenLimitForChatModel, tokenizer };