@gammatech/aijsx 0.3.0-beta.2 → 0.3.0-beta.4

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-_dLhPtA-.d.mts → createElement-Mo9PV8GD.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,35 +11,59 @@ 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> = (input: {
+    promptKey: string;
+    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[];
+    metadata?: Record<string, any>;
+}
+interface PromptParsed<V extends Record<string, any> = Record<string, any>, O = any> extends Prompt<V> {
+    outputParser: OutputParser<O>;
+}
+type Chain<V extends Record<string, any> = Record<string, any>, R = any> = {
+    run: (vars: V, context: RenderContext) => R;
+    schema: ZodObject<ZodRawShape>;
+};
 
 declare const LoggerContext: Context<LogImplementation>;
 type RenderOptions = {
@@ -53,13 +76,18 @@ 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>(chain: Chain<V, O>, variables: V): O;
 }
+type RenderPromptOptions = {
+    validateOutput?: boolean;
+};
 declare function createContext<T>(defaultValue: T): Context<T>;
 
 /**
@@ -163,11 +191,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 AIComponentParsed 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, type Renderable as a, type AINode as b, createAIElement as c, AIFragment as d, LoggerContext as e, createContext as f, type ChatCompletionRole as g, AssistantMessage as h, type RenderedConversationMessage as i, computeUsage as j, ChatCompletionError as k, type ChatCompletionRequestPayloads as l, type LogChatCompletionRequest as m, type LogChatCompletionResponse as n, type LogLevel as o, type Logger as p, ConsoleLogger as q, CombinedLogger as r, type Literal as s, type RenderableStream as t, type RenderResult as u, type AIComponentString as v, type AIComponent as w, attachedContextSymbol as x, type AIElement as y, type ParsedTypeOfAIComponent as z };
+export { type AINode as A, BoundLogger as B, type Context as C, type EvaluatorResult as D, type EvaluatorFn as E, type PromptRenderResult as F, LogImplementation as L, NoopLogImplementation as N, type OutputParser as O, type PromptParsed as P, type RenderContext as R, SystemMessage as S, UserMessage as U, type AIComponent as a, type Prompt as b, type Chain as c, createAIElement as d, AIFragment as e, LoggerContext as f, createContext as g, type ChatCompletionRole as h, AssistantMessage as i, type RenderedConversationMessage as j, computeUsage as k, ChatCompletionError as l, type ChatCompletionRequestPayloads as m, type LogChatCompletionRequest as n, type LogChatCompletionResponse as o, type LogLevel as p, type Logger as q, ConsoleLogger as r, CombinedLogger as s, type Literal as t, type RenderableStream as u, type RenderResult as v, attachedContextSymbol as w, type AIElement as x, type Renderable as y, type PropsOfAIComponent as z };

package/dist/{createElement-_dLhPtA-.d.ts → createElement-Mo9PV8GD.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,35 +11,59 @@ 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> = (input: {
+    promptKey: string;
+    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[];
+    metadata?: Record<string, any>;
+}
+interface PromptParsed<V extends Record<string, any> = Record<string, any>, O = any> extends Prompt<V> {
+    outputParser: OutputParser<O>;
+}
+type Chain<V extends Record<string, any> = Record<string, any>, R = any> = {
+    run: (vars: V, context: RenderContext) => R;
+    schema: ZodObject<ZodRawShape>;
+};
 
 declare const LoggerContext: Context<LogImplementation>;
 type RenderOptions = {
@@ -53,13 +76,18 @@ 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>(chain: Chain<V, O>, variables: V): O;
 }
+type RenderPromptOptions = {
+    validateOutput?: boolean;
+};
 declare function createContext<T>(defaultValue: T): Context<T>;
 
 /**
@@ -163,11 +191,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 AIComponentParsed 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, type Renderable as a, type AINode as b, createAIElement as c, AIFragment as d, LoggerContext as e, createContext as f, type ChatCompletionRole as g, AssistantMessage as h, type RenderedConversationMessage as i, computeUsage as j, ChatCompletionError as k, type ChatCompletionRequestPayloads as l, type LogChatCompletionRequest as m, type LogChatCompletionResponse as n, type LogLevel as o, type Logger as p, ConsoleLogger as q, CombinedLogger as r, type Literal as s, type RenderableStream as t, type RenderResult as u, type AIComponentString as v, type AIComponent as w, attachedContextSymbol as x, type AIElement as y, type ParsedTypeOfAIComponent as z };
+export { type AINode as A, BoundLogger as B, type Context as C, type EvaluatorResult as D, type EvaluatorFn as E, type PromptRenderResult as F, LogImplementation as L, NoopLogImplementation as N, type OutputParser as O, type PromptParsed as P, type RenderContext as R, SystemMessage as S, UserMessage as U, type AIComponent as a, type Prompt as b, type Chain as c, createAIElement as d, AIFragment as e, LoggerContext as f, createContext as g, type ChatCompletionRole as h, AssistantMessage as i, type RenderedConversationMessage as j, computeUsage as k, ChatCompletionError as l, type ChatCompletionRequestPayloads as m, type LogChatCompletionRequest as n, type LogChatCompletionResponse as o, type LogLevel as p, type Logger as q, ConsoleLogger as r, CombinedLogger as s, type Literal as t, type RenderableStream as u, type RenderResult as v, attachedContextSymbol as w, type AIElement as x, type Renderable as y, type PropsOfAIComponent as z };

package/dist/index.d.mts

@@ -1,21 +1,18 @@
-import { L as LogImplementation, R as RenderContext, J as JSONSerializable, a as Renderable, A as AIComponentParsed, C as Context, b as AINode } from './createElement-_dLhPtA-.mjs';
-export { w as AIComponent, v as AIComponentString, y as AIElement, d as AIFragment, h as AssistantMessage, B as BoundLogger, k as ChatCompletionError, l as ChatCompletionRequestPayloads, g as ChatCompletionRole, r as CombinedLogger, q as ConsoleLogger, s as Literal, m as LogChatCompletionRequest, n as LogChatCompletionResponse, o as LogLevel, p as Logger, e as LoggerContext, N as NoopLogImplementation, z as ParsedTypeOfAIComponent, D as ParsedTypeOfRenderable, P as PropsOfAIComponent, u as RenderResult, t as RenderableStream, i as RenderedConversationMessage, S as SystemMessage, U as UserMessage, x as attachedContextSymbol, j as computeUsage, c as createAIElement, f as createContext } from './createElement-_dLhPtA-.mjs';
+import { R as RenderContext, L as LogImplementation, C as Context, A as AINode, a as AIComponent, E as EvaluatorFn, P as PromptParsed, b as Prompt, c as Chain } from './createElement-Mo9PV8GD.mjs';
+export { x as AIElement, e as AIFragment, i as AssistantMessage, B as BoundLogger, l as ChatCompletionError, m as ChatCompletionRequestPayloads, h as ChatCompletionRole, s as CombinedLogger, r as ConsoleLogger, D as EvaluatorResult, t as Literal, n as LogChatCompletionRequest, o as LogChatCompletionResponse, p as LogLevel, q as Logger, f as LoggerContext, N as NoopLogImplementation, O as OutputParser, F as PromptRenderResult, z as PropsOfAIComponent, v as RenderResult, y as Renderable, u as RenderableStream, j as RenderedConversationMessage, S as SystemMessage, U as UserMessage, w as attachedContextSymbol, k as computeUsage, d as createAIElement, g as createContext } from './createElement-Mo9PV8GD.mjs';
 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';
 export { countTokens as countAnthropicTokens } from '@anthropic-ai/tokenizer';
+import { ZodObject, ZodRawShape, ZodTypeAny, ZodString, z } from 'zod';
 
-declare function createRenderContext({ logger, rootRenderId, }?: {
+type CreateRenderContextOptions = {
     logger?: LogImplementation;
     rootRenderId?: string;
-}): RenderContext;
-
-declare function createAIComponentParsed<P, R extends JSONSerializable>(config: {
-    render: (props: P, context: RenderContext) => Renderable;
-    parse: (result: string) => R;
-}): AIComponentParsed<P, R>;
+};
+declare function createRenderContext({ logger, rootRenderId, }?: CreateRenderContextOptions): RenderContext;
 
 type OpenAIChatCompletionRequest = OpenAI.Chat.Completions.ChatCompletionCreateParamsStreaming;
 type OpenAIChatMessage = ChatCompletionSystemMessageParam | ChatCompletionUserMessageParam | ChatCompletionAssistantMessageParam;
@@ -97,4 +94,37 @@ type AnthropicChatCompletionProps = {
  */
 declare function AnthropicChatCompletion(props: AnthropicChatCompletionProps, { render, logger, getContext }: RenderContext): AsyncGenerator<string, void, unknown>;
 
-export { AIComponentParsed, AINode, AnthropicChatCompletion, type AnthropicChatCompletionRequest, AnthropicClientContext, ContentTypeImage, Context, JSONSerializable, LogImplementation, OpenAIChatCompletion, type OpenAIChatCompletionRequest, type OpenAIChatMessage, OpenAIClientContext, OpenAIVisionChatCompletion, RenderContext, Renderable, type ValidAnthropicChatModel, type ValidOpenAIChatModel, type ValidOpenAIVisionModel, createAIComponentParsed, createRenderContext, defaultMaxTokens, tokenCountForOpenAIMessage, tokenCountForOpenAIVisionMessage, tokenLimitForChatModel, tokenizer };
+/**
+ * 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 createChain<I extends ZodObject<ZodRawShape>, R>(config: {
+    run: (vars: z.infer<I>, context: RenderContext) => R;
+    schema: I;
+}): Chain<z.infer<I>, R>;
+
+declare class PromptParseVariablesError extends Error {
+}
+declare class ChainParseVariablesError extends Error {
+}
+declare class PromptInvalidOutputError extends Error {
+}
+
+export { AIComponent, AINode, AnthropicChatCompletion, type AnthropicChatCompletionRequest, AnthropicClientContext, Chain, ChainParseVariablesError, ContentTypeImage, Context, EvaluatorFn, LogImplementation, OpenAIChatCompletion, type OpenAIChatCompletionRequest, type OpenAIChatMessage, OpenAIClientContext, OpenAIVisionChatCompletion, Prompt, PromptInvalidOutputError, PromptParseVariablesError, PromptParsed, RenderContext, type ValidAnthropicChatModel, type ValidOpenAIChatModel, type ValidOpenAIVisionModel, createChain, createPrompt, createRenderContext, defaultMaxTokens, tokenCountForOpenAIMessage, tokenCountForOpenAIVisionMessage, tokenLimitForChatModel, tokenizer };

package/dist/index.d.ts

@@ -1,21 +1,18 @@
-import { L as LogImplementation, R as RenderContext, J as JSONSerializable, a as Renderable, A as AIComponentParsed, C as Context, b as AINode } from './createElement-_dLhPtA-.js';
-export { w as AIComponent, v as AIComponentString, y as AIElement, d as AIFragment, h as AssistantMessage, B as BoundLogger, k as ChatCompletionError, l as ChatCompletionRequestPayloads, g as ChatCompletionRole, r as CombinedLogger, q as ConsoleLogger, s as Literal, m as LogChatCompletionRequest, n as LogChatCompletionResponse, o as LogLevel, p as Logger, e as LoggerContext, N as NoopLogImplementation, z as ParsedTypeOfAIComponent, D as ParsedTypeOfRenderable, P as PropsOfAIComponent, u as RenderResult, t as RenderableStream, i as RenderedConversationMessage, S as SystemMessage, U as UserMessage, x as attachedContextSymbol, j as computeUsage, c as createAIElement, f as createContext } from './createElement-_dLhPtA-.js';
+import { R as RenderContext, L as LogImplementation, C as Context, A as AINode, a as AIComponent, E as EvaluatorFn, P as PromptParsed, b as Prompt, c as Chain } from './createElement-Mo9PV8GD.js';
+export { x as AIElement, e as AIFragment, i as AssistantMessage, B as BoundLogger, l as ChatCompletionError, m as ChatCompletionRequestPayloads, h as ChatCompletionRole, s as CombinedLogger, r as ConsoleLogger, D as EvaluatorResult, t as Literal, n as LogChatCompletionRequest, o as LogChatCompletionResponse, p as LogLevel, q as Logger, f as LoggerContext, N as NoopLogImplementation, O as OutputParser, F as PromptRenderResult, z as PropsOfAIComponent, v as RenderResult, y as Renderable, u as RenderableStream, j as RenderedConversationMessage, S as SystemMessage, U as UserMessage, w as attachedContextSymbol, k as computeUsage, d as createAIElement, g as createContext } from './createElement-Mo9PV8GD.js';
 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';
 export { countTokens as countAnthropicTokens } from '@anthropic-ai/tokenizer';
+import { ZodObject, ZodRawShape, ZodTypeAny, ZodString, z } from 'zod';
 
-declare function createRenderContext({ logger, rootRenderId, }?: {
+type CreateRenderContextOptions = {
     logger?: LogImplementation;
     rootRenderId?: string;
-}): RenderContext;
-
-declare function createAIComponentParsed<P, R extends JSONSerializable>(config: {
-    render: (props: P, context: RenderContext) => Renderable;
-    parse: (result: string) => R;
-}): AIComponentParsed<P, R>;
+};
+declare function createRenderContext({ logger, rootRenderId, }?: CreateRenderContextOptions): RenderContext;
 
 type OpenAIChatCompletionRequest = OpenAI.Chat.Completions.ChatCompletionCreateParamsStreaming;
 type OpenAIChatMessage = ChatCompletionSystemMessageParam | ChatCompletionUserMessageParam | ChatCompletionAssistantMessageParam;
@@ -97,4 +94,37 @@ type AnthropicChatCompletionProps = {
  */
 declare function AnthropicChatCompletion(props: AnthropicChatCompletionProps, { render, logger, getContext }: RenderContext): AsyncGenerator<string, void, unknown>;
 
-export { AIComponentParsed, AINode, AnthropicChatCompletion, type AnthropicChatCompletionRequest, AnthropicClientContext, ContentTypeImage, Context, JSONSerializable, LogImplementation, OpenAIChatCompletion, type OpenAIChatCompletionRequest, type OpenAIChatMessage, OpenAIClientContext, OpenAIVisionChatCompletion, RenderContext, Renderable, type ValidAnthropicChatModel, type ValidOpenAIChatModel, type ValidOpenAIVisionModel, createAIComponentParsed, createRenderContext, defaultMaxTokens, tokenCountForOpenAIMessage, tokenCountForOpenAIVisionMessage, tokenLimitForChatModel, tokenizer };
+/**
+ * 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 createChain<I extends ZodObject<ZodRawShape>, R>(config: {
+    run: (vars: z.infer<I>, context: RenderContext) => R;
+    schema: I;
+}): Chain<z.infer<I>, R>;
+
+declare class PromptParseVariablesError extends Error {
+}
+declare class ChainParseVariablesError extends Error {
+}
+declare class PromptInvalidOutputError extends Error {
+}
+
+export { AIComponent, AINode, AnthropicChatCompletion, type AnthropicChatCompletionRequest, AnthropicClientContext, Chain, ChainParseVariablesError, ContentTypeImage, Context, EvaluatorFn, LogImplementation, OpenAIChatCompletion, type OpenAIChatCompletionRequest, type OpenAIChatMessage, OpenAIClientContext, OpenAIVisionChatCompletion, Prompt, PromptInvalidOutputError, PromptParseVariablesError, PromptParsed, RenderContext, type ValidAnthropicChatModel, type ValidOpenAIChatModel, type ValidOpenAIVisionModel, createChain, createPrompt, createRenderContext, defaultMaxTokens, tokenCountForOpenAIMessage, tokenCountForOpenAIVisionMessage, tokenLimitForChatModel, tokenizer };