ai-functions 2.1.1 → 2.3.0

package/dist/generate.d.ts

@@ -8,7 +8,7 @@
  *
  * @packageDocumentation
  */
-import { generateText as sdkGenerateText, streamText as sdkStreamText, type GenerateObjectResult, type StreamObjectResult, type LanguageModel } from 'ai';
+import { generateText as sdkGenerateText, streamText as sdkStreamText, type LanguageModel } from 'ai';
 type ModelArg = string | LanguageModel;
 interface GenerateObjectOptions<T> {
     model: ModelArg;
@@ -102,7 +102,11 @@ interface GenerateTextOptions {
  * })
  * ```
  */
-export declare function generateObject<T>(options: GenerateObjectOptions<T>): Promise<GenerateObjectResult<T>>;
+export declare function generateObject<T>(options: GenerateObjectOptions<T>): Promise<{
+    object: T;
+    usage?: unknown;
+    warnings?: unknown[] | undefined;
+}>;
 /**
  * Generate text from a prompt using AI
  *
@@ -145,7 +149,9 @@ export declare function generateText(options: GenerateTextOptions): Promise<Awai
  * }
  * ```
  */
-export declare function streamObject<T>(options: GenerateObjectOptions<T>): Promise<StreamObjectResult<T, T, never>>;
+export declare function streamObject<T>(options: GenerateObjectOptions<T>): Promise<{
+    partialObjectStream: AsyncIterable<T>;
+}>;
 /**
  * Stream text from a prompt using AI
  *

package/dist/generate.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"generate.d.ts","sourceRoot":"","sources":["../src/generate.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EAEL,YAAY,IAAI,eAAe,EAE/B,UAAU,IAAI,aAAa,EAC3B,KAAK,oBAAoB,EAEzB,KAAK,kBAAkB,EAEvB,KAAK,aAAa,EACnB,MAAM,IAAI,CAAA;AAIX,KAAK,QAAQ,GAAG,MAAM,GAAG,aAAa,CAAA;AAGtC,UAAU,qBAAqB,CAAC,CAAC;IAC/B,KAAK,EAAE,QAAQ,CAAA;IACf,MAAM,EAAE,CAAC,CAAA;IACT,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,QAAQ,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;IAC5E,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAAA;IAC/B,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,gBAAgB,CAAC,EAAE,MAAM,CAAA;IACzB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,WAAW,CAAC,EAAE,WAAW,CAAA;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAChC,sBAAsB,CAAC,EAAE;QAAE,SAAS,CAAC,EAAE,OAAO,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KAAE,CAAA;CACzG;AAED,UAAU,mBAAmB;IAC3B,KAAK,EAAE,QAAQ,CAAA;IACf,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,QAAQ,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;IAC5E,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,gBAAgB,CAAC,EAAE,MAAM,CAAA;IACzB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,WAAW,CAAC,EAAE,WAAW,CAAA;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAChC,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IAC/B,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,UAAU,GAAG;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAA;IAC9E,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB,sBAAsB,CAAC,EAAE;QAAE,SAAS,CAAC,EAAE,OAAO,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KAAE,CAAA;CACzG;AAqCD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmCG;AACH,wBAAsB,cAAc,CAAC,CAAC,EACpC,OAAO,EAAE,qBAAqB,CAAC,CAAC,CAAC,GAChC,OAAO,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,CAUlC;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,wBAAsB,YAAY,CAChC,OAAO,EAAE,mBAAmB,GAC3B,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,OAAO,eAAe,CAAC,CAAC,CAAC,CAMtD;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAsB,YAAY,CAAC,CAAC,EAClC,OAAO,EAAE,qBAAqB,CAAC,CAAC,CAAC,GAChC,OAAO,CAAC,kBAAkB,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC,CAU1C;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAsB,UAAU,CAC9B,OAAO,EAAE,mBAAmB,GAC3B,OAAO,CAAC,UAAU,CAAC,OAAO,aAAa,CAAC,CAAC,CAM3C"}
+{"version":3,"file":"generate.d.ts","sourceRoot":"","sources":["../src/generate.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EACL,YAAY,IAAI,eAAe,EAC/B,UAAU,IAAI,aAAa,EAI3B,KAAK,aAAa,EACnB,MAAM,IAAI,CAAA;AAKX,KAAK,QAAQ,GAAG,MAAM,GAAG,aAAa,CAAA;AAGtC,UAAU,qBAAqB,CAAC,CAAC;IAC/B,KAAK,EAAE,QAAQ,CAAA;IACf,MAAM,EAAE,CAAC,CAAA;IACT,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,QAAQ,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;IAC5E,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAAA;IAC/B,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,gBAAgB,CAAC,EAAE,MAAM,CAAA;IACzB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,WAAW,CAAC,EAAE,WAAW,CAAA;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAChC,sBAAsB,CAAC,EAAE;QACvB,SAAS,CAAC,EAAE,OAAO,CAAA;QACnB,UAAU,CAAC,EAAE,MAAM,CAAA;QACnB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KAClC,CAAA;CACF;AAED,UAAU,mBAAmB;IAC3B,KAAK,EAAE,QAAQ,CAAA;IACf,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,QAAQ,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;IAC5E,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,gBAAgB,CAAC,EAAE,MAAM,CAAA;IACzB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,WAAW,CAAC,EAAE,WAAW,CAAA;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAChC,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IAC/B,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,UAAU,GAAG;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAA;IAC9E,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB,sBAAsB,CAAC,EAAE;QACvB,SAAS,CAAC,EAAE,OAAO,CAAA;QACnB,UAAU,CAAC,EAAE,MAAM,CAAA;QACnB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KAClC,CAAA;CACF;AA2BD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmCG;AACH,wBAAsB,cAAc,CAAC,CAAC,EACpC,OAAO,EAAE,qBAAqB,CAAC,CAAC,CAAC,GAChC,OAAO,CAAC;IAAE,MAAM,EAAE,CAAC,CAAC;IAAC,KAAK,CAAC,EAAE,OAAO,CAAC;IAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,GAAG,SAAS,CAAA;CAAE,CAAC,CAW3E;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,wBAAsB,YAAY,CAChC,OAAO,EAAE,mBAAmB,GAC3B,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,OAAO,eAAe,CAAC,CAAC,CAAC,CAMtD;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAsB,YAAY,CAAC,CAAC,EAClC,OAAO,EAAE,qBAAqB,CAAC,CAAC,CAAC,GAChC,OAAO,CAAC;IAAE,mBAAmB,EAAE,aAAa,CAAC,CAAC,CAAC,CAAA;CAAE,CAAC,CAapD;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAsB,UAAU,CAC9B,OAAO,EAAE,mBAAmB,GAC3B,OAAO,CAAC,UAAU,CAAC,OAAO,aAAa,CAAC,CAAC,CAM3C"}

package/dist/generate.js

@@ -8,8 +8,9 @@
  *
  * @packageDocumentation
  */
-import { generateObject as sdkGenerateObject, generateText as sdkGenerateText, streamObject as sdkStreamObject, streamText as sdkStreamText } from 'ai';
+import { generateText as sdkGenerateText, streamText as sdkStreamText, Output, } from 'ai';
 import { schema as convertSchema } from './schema.js';
+import { isZodSchema } from './type-guards.js';
 /**
  * Resolve model string to LanguageModel instance
  * Uses ai-providers for model routing with Cloudflare AI Gateway support
@@ -23,15 +24,6 @@ async function resolveModel(modelArg) {
     const { model } = await import('ai-providers');
     return model(modelArg);
 }
-/**
- * Check if value is a Zod schema
- */
-function isZodSchema(value) {
-    return value !== null &&
-        typeof value === 'object' &&
-        '_def' in value &&
-        'parse' in value;
-}
 /**
  * Convert schema to Zod if needed
  */
@@ -80,13 +72,14 @@ function resolveSchema(schemaArg) {
 export async function generateObject(options) {
     const model = await resolveModel(options.model);
     const schema = resolveSchema(options.schema);
-    // Use 'as any' to handle AI SDK v4 API variance
-    return sdkGenerateObject({
-        ...options,
+    const { schema: _schema, mode: _mode, ...rest } = options;
+    // Using unknown cast for SDK compatibility - the AI SDK has complex type unions
+    const result = await sdkGenerateText({
+        ...rest,
         model,
-        schema,
-        output: 'object'
+        output: Output.object({ schema }),
     });
+    return { object: result.output, usage: result.usage, warnings: result.warnings };
 }
 /**
  * Generate text from a prompt using AI
@@ -115,7 +108,7 @@ export async function generateText(options) {
     const model = await resolveModel(options.model);
     return sdkGenerateText({
         ...options,
-        model
+        model,
     });
 }
 /**
@@ -139,13 +132,16 @@ export async function generateText(options) {
 export async function streamObject(options) {
     const model = await resolveModel(options.model);
     const schema = resolveSchema(options.schema);
-    // Use 'as any' to handle AI SDK API variance
-    return sdkStreamObject({
-        ...options,
+    const { schema: _schema, mode: _mode, ...rest } = options;
+    // Using unknown cast for SDK compatibility - the AI SDK has complex type unions
+    const result = await sdkStreamText({
+        ...rest,
         model,
-        schema,
-        output: 'object'
+        output: Output.object({ schema }),
     });
+    return {
+        partialObjectStream: result.partialOutputStream,
+    };
 }
 /**
  * Stream text from a prompt using AI
@@ -168,7 +164,7 @@ export async function streamText(options) {
     const model = await resolveModel(options.model);
     return sdkStreamText({
         ...options,
-        model
+        model,
     });
 }
 //# sourceMappingURL=generate.js.map

package/dist/generate.js.map

@@ -1 +1 @@
-{"version":3,"file":"generate.js","sourceRoot":"","sources":["../src/generate.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EACL,cAAc,IAAI,iBAAiB,EACnC,YAAY,IAAI,eAAe,EAC/B,YAAY,IAAI,eAAe,EAC/B,UAAU,IAAI,aAAa,EAM5B,MAAM,IAAI,CAAA;AACX,OAAO,EAAE,MAAM,IAAI,aAAa,EAAqB,MAAM,aAAa,CAAA;AA+CxE;;;GAGG;AACH,KAAK,UAAU,YAAY,CAAC,QAAkB;IAC5C,mCAAmC;IACnC,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACjC,OAAO,QAAQ,CAAA;IACjB,CAAC;IAED,wCAAwC;IACxC,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,CAAA;IAC9C,OAAO,KAAK,CAAC,QAAQ,CAAC,CAAA;AACxB,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,KAAc;IACjC,OAAO,KAAK,KAAK,IAAI;QACnB,OAAO,KAAK,KAAK,QAAQ;QACzB,MAAM,IAAI,KAAK;QACf,OAAO,IAAI,KAAK,CAAA;AACpB,CAAC;AAED;;GAEG;AACH,SAAS,aAAa,CAAC,SAAoB;IACzC,IAAI,WAAW,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3B,OAAO,SAAS,CAAA;IAClB,CAAC;IACD,OAAO,aAAa,CAAC,SAAyB,CAAC,CAAA;AACjD,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmCG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,OAAiC;IAEjC,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;IAC/C,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,MAAmB,CAAC,CAAA;IACzD,gDAAgD;IAChD,OAAO,iBAAiB,CAAC;QACvB,GAAG,OAAO;QACV,KAAK;QACL,MAAM;QACN,MAAM,EAAE,QAAQ;KACV,CAAqC,CAAA;AAC/C,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAA4B;IAE5B,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;IAC/C,OAAO,eAAe,CAAC;QACrB,GAAG,OAAO;QACV,KAAK;KACmC,CAAC,CAAA;AAC7C,CAAC;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAAiC;IAEjC,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;IAC/C,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,MAAmB,CAAC,CAAA;IACzD,6CAA6C;IAC7C,OAAO,eAAe,CAAC;QACrB,GAAG,OAAO;QACV,KAAK;QACL,MAAM;QACN,MAAM,EAAE,QAAQ;KACV,CAA+C,CAAA;AACzD,CAAC;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,OAA4B;IAE5B,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;IAC/C,OAAO,aAAa,CAAC;QACnB,GAAG,OAAO;QACV,KAAK;KACiC,CAAC,CAAA;AAC3C,CAAC"}
+{"version":3,"file":"generate.js","sourceRoot":"","sources":["../src/generate.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EACL,YAAY,IAAI,eAAe,EAC/B,UAAU,IAAI,aAAa,EAC3B,MAAM,GAIP,MAAM,IAAI,CAAA;AACX,OAAO,EAAE,MAAM,IAAI,aAAa,EAAqB,MAAM,aAAa,CAAA;AACxE,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAA;AAuD9C;;;GAGG;AACH,KAAK,UAAU,YAAY,CAAC,QAAkB;IAC5C,mCAAmC;IACnC,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACjC,OAAO,QAAQ,CAAA;IACjB,CAAC;IAED,wCAAwC;IACxC,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,CAAA;IAC9C,OAAO,KAAK,CAAC,QAAQ,CAAC,CAAA;AACxB,CAAC;AAED;;GAEG;AACH,SAAS,aAAa,CAAC,SAAoB;IACzC,IAAI,WAAW,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3B,OAAO,SAAS,CAAA;IAClB,CAAC;IACD,OAAO,aAAa,CAAC,SAAyB,CAAC,CAAA;AACjD,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmCG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,OAAiC;IAEjC,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;IAC/C,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,MAAmB,CAAC,CAAA;IACzD,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAA;IACzD,gFAAgF;IAChF,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC;QACnC,GAAG,IAAI;QACP,KAAK;QACL,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;KACkB,CAAC,CAAA;IACtD,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,MAAW,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAA;AACvF,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAA4B;IAE5B,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;IAC/C,OAAO,eAAe,CAAC;QACrB,GAAG,OAAO;QACV,KAAK;KACmC,CAAC,CAAA;AAC7C,CAAC;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAAiC;IAEjC,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;IAC/C,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,MAAmB,CAAC,CAAA;IACzD,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAA;IACzD,gFAAgF;IAChF,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC;QACjC,GAAG,IAAI;QACP,KAAK;QACL,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;KACgB,CAAC,CAAA;IACpD,OAAO;QACL,mBAAmB,EAAE,MAAM,CAAC,mBAAuC;KACpE,CAAA;AACH,CAAC;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,OAA4B;IAE5B,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;IAC/C,OAAO,aAAa,CAAC;QACnB,GAAG,OAAO;QACV,KAAK;KACiC,CAAC,CAAA;AAC3C,CAAC"}

package/dist/index.d.ts

@@ -1,29 +1,44 @@
 /**
- * ai-functions - Core AI primitives for building intelligent applications
+ * ai-functions - Full-featured AI primitives for building intelligent applications
  *
- * This is the foundational package that all other primitives depend on.
- * It provides:
- * - RPC primitives via rpc.do
- * - AI function types and interfaces
- * - Core AI() and ai() constructors
- * - Embedding utilities from AI SDK
- * - Cloudflare Workers AI provider (default for embeddings)
- * - Tagged template literal support for all functions
- * - Async iterator support for list/extract
+ * This package provides the complete feature set including:
+ * - Core primitives from ai-core (generate, AIPromise, templates, context)
+ * - Batch processing (BatchQueue, BatchMapPromise)
+ * - Retry/resilience patterns (RetryPolicy, CircuitBreaker, FallbackChain)
+ * - Budget tracking (BudgetTracker, TokenCounter)
+ * - Caching (MemoryCache, EmbeddingCache, GenerationCache)
+ * - Tool orchestration (AgenticLoop, ToolRouter)
+ * - Embeddings
+ * - Provider integrations
+ *
+ * For lightweight usage with just core primitives, use ai-core directly.
  *
  * @packageDocumentation
  */
-export * from './types.js';
-export * from './ai.js';
-export * from './embeddings.js';
-export { generateObject, generateText, streamObject, streamText } from './generate.js';
+export type { AIFunctionDefinition, JSONSchema, AIGenerateOptions, AIGenerateResult, AIFunctionCall, AIClient, ImageOptions, ImageResult, VideoOptions, VideoResult, WriteOptions, TemplateFunction as CoreTemplateFunction, ListItem, ListResult, NamedList, ListsResult, CodeLanguage, GenerativeOutputType, HumanChannel, LegacyHumanChannel, SchemaLimitations, BaseFunctionDefinition, CodeFunctionDefinition, CodeFunctionResult, CodeGenerationDefinition, GenerativeFunctionDefinition, GenerativeFunctionResult, AgenticFunctionDefinition, AgenticExecutionState, HumanFunctionDefinition, HumanFunctionResult, FunctionDefinition, DefinedFunction, FunctionRegistry, AutoDefineResult, } from './types.js';
 export { schema, type SimpleSchema } from './schema.js';
-export * from './providers/index.js';
 export { parseTemplate, createTemplateFunction, createChainablePromise, createStreamableList, withBatch, type FunctionOptions, type TemplateFunction, type BatchableFunction, type StreamableList, type ChainablePromise, } from './template.js';
-export { AIPromise, isAIPromise, getRawPromise, createTextPromise, createObjectPromise, createListPromise, createListsPromise, createBooleanPromise, createExtractPromise, createAITemplateFunction, parseTemplateWithDependencies, AI_PROMISE_SYMBOL, RAW_PROMISE_SYMBOL, type AIPromiseOptions, } from './ai-promise.js';
-export { generate, type GenerateType, type GenerateOptions, ai as aiPrompt, // Renamed to avoid conflict with AIProxy from ai.ts
-write, code, list, lists, extract, summarize, is, diagram, slides, image, video, do, research, read, browse, decide, ask, approve, review, type HumanOptions, type HumanResult, } from './primitives.js';
-export { BatchQueue, createBatch, withBatch as withBatchQueue, registerBatchAdapter, getBatchAdapter, isBatchMode, deferToBatch, BATCH_MODE_SYMBOL, type BatchMode as BatchQueueMode, type BatchProvider, type BatchStatus, type BatchItem, type BatchJob, type BatchResult, type BatchSubmitResult, type BatchAdapter, type BatchQueueOptions, type DeferredOptions, } from './batch-queue.js';
+export { AIPromise, isAIPromise, getRawPromise, createTextPromise, createObjectPromise, createListPromise, createListsPromise, createBooleanPromise, createExtractPromise, createAITemplateFunction, parseTemplateWithDependencies, AI_PROMISE_SYMBOL, RAW_PROMISE_SYMBOL, type AIPromiseOptions, type StreamingAIPromise, type StreamOptions, } from './ai-promise.js';
+export { generateObject, generateText, streamObject, streamText } from './generate.js';
+export { generate, type GenerateType, type GenerateOptions, ai, write, code, list, lists, extract, summarize, is, diagram, slides, image, video, do, research, read, browse, decide, ask, approve, review, type HumanOptions, type HumanResult, } from './primitives.js';
+export { configure, getContext, withContext, getGlobalContext, resetContext, getModel, getProvider, type ExecutionContext, } from './context.js';
+export { isZodSchema } from './type-guards.js';
+export { AI, ai as aiProxy, define, defineFunction, functions, createFunctionRegistry, resetGlobalRegistry, withTemplate, type AIProxy, createSchemaFunctions, type AISchemaOptions, type SchemaFunctions, type InferSimpleSchemaResult, createDefinedFunction, convertArgsToJSONSchema, fillTemplate, generateCode, } from './ai.js';
+export { ai as aiPrompt } from './primitives.js';
+export * from './embeddings.js';
+export * from './providers/index.js';
+export { BatchQueue, createBatch, withBatch as withBatchQueue, registerBatchAdapter, getBatchAdapter, registerFlexAdapter, getFlexAdapter, hasFlexAdapter, tiersForModel, modelSupportsTier, isBatchMode, deferToBatch, BATCH_MODE_SYMBOL, type BatchMode as BatchQueueMode, type BatchProvider, type BatchStatus, type BatchItem, type BatchJob, type BatchResult, type BatchSubmitResult, type BatchAdapter, type FlexAdapter, type BatchQueueOptions, type DeferredOptions, } from './batch-queue.js';
+export { policyFor as modelPolicyFor, type ModelPolicy, type BatchTier, type RetryPolicyData, type CircuitBreakerPolicyData, type ErrorCategoryName, } from 'language-models';
 export { BatchMapPromise, createBatchMap, isBatchMapPromise, BATCH_MAP_SYMBOL, type CapturedOperation, type BatchMapOptions, } from './batch-map.js';
-export { configure, getContext, withContext, getGlobalContext, resetContext, getModel, getProvider, getBatchMode, getBatchThreshold, shouldUseBatchAPI, type ExecutionContext, type BatchMode, } from './context.js';
+export { getBatchMode, getBatchThreshold, shouldUseBatchAPI, getFlexThreshold, getExecutionTier, isFlexAvailable, type BatchMode, type ContextBudgetConfig, type ExecutionTier, } from './context.js';
+export { BudgetTracker, TokenCounter, RequestContext, BudgetExceededError, createRequestContext, withBudget, type BudgetConfig, type BudgetAlert, type BudgetSnapshot, type TokenUsage, type RequestInfo, type RequestContextOptions, type ModelPricing, type WithBudgetOptions, type RemainingBudget, type CheckBudgetOptions, } from './budget.js';
+export { AgenticLoop, ToolRouter, ToolValidator, createTool, createToolset, wrapTool, cachedTool, rateLimitedTool, timeoutTool, createAgenticLoop, type Tool, type ToolCall, type ToolResult, type FormattedToolResult, type ValidationResult, type ModelResponse, type Message, type StepInfo, type LoopOptions, type RunOptions, type ToolCallResult, type SDKToolResult, type LoopResult, type LoopStreamEvent, } from './tool-orchestration.js';
+export { MemoryCache, EmbeddingCache, GenerationCache, withCache, hashKey, createCacheKey, type CacheStorage, type CacheEntry, type CacheOptions, type CacheStats, type MemoryCacheOptions, type CacheKeyType, type EmbeddingCacheOptions, type BatchEmbeddingResult, type GenerationParams, type GenerationCacheGetOptions, type WithCacheOptions, type CachedFunction, } from './cache.js';
+export { RetryableError, NonRetryableError, NetworkError, RateLimitError, CircuitOpenError, ErrorCategory, classifyError, calculateBackoff, RetryPolicy, CircuitBreaker, FallbackChain, withRetry, type JitterStrategy, type BackoffOptions, type RetryOptions, type RetryInfo, type BatchItemResult, type CircuitState, type CircuitBreakerOptions, type CircuitBreakerMetrics, type FallbackModel, type FallbackOptions, type FallbackMetrics, } from './retry.js';
+export { DigitalObjectsFunctionRegistry, createDigitalObjectsRegistry, FUNCTION_NOUNS, FUNCTION_VERBS, type StoredFunctionDefinition, type FunctionCallData, type DigitalObjectsRegistryOptions, } from './digital-objects-registry.js';
+export { configureLogger, getLogger, resetLogger, getLogLevel, setLogLevel, type Logger, type LoggerConfig, } from './logger.js';
+export { cacheMiddleware, embeddingCacheMiddleware, budgetMiddleware, traceMiddleware, type CacheMiddlewareOptions, type CacheMiddlewareStore, type EmbedCacheMiddlewareOptions, type EmbedCacheMiddlewareStore, type BudgetMiddlewareOptions, type PricingOverlay, type TraceEvent, type TraceEventKind, type TraceMiddlewareOptions, } from './middleware/index.js';
+export { wrapForV3, type WrapForV3Options } from './wrap-for-v3.js';
+export type { LanguageModelV3 } from '@ai-sdk/provider';
+export { InMemoryEvalLogStore, getEvalLogStore, configureEvalLogStore, type EvalLogEntry, type EvalLogListOptions, type EvalLogStore, } from './eval-log/index.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAMH,cAAc,YAAY,CAAA;AAC1B,cAAc,SAAS,CAAA;AAGvB,cAAc,iBAAiB,CAAA;AAG/B,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,eAAe,CAAA;AAGtF,OAAO,EAAE,MAAM,EAAE,KAAK,YAAY,EAAE,MAAM,aAAa,CAAA;AAGvD,cAAc,sBAAsB,CAAA;AAGpC,OAAO,EACL,aAAa,EACb,sBAAsB,EACtB,sBAAsB,EACtB,oBAAoB,EACpB,SAAS,EACT,KAAK,eAAe,EACpB,KAAK,gBAAgB,EACrB,KAAK,iBAAiB,EACtB,KAAK,cAAc,EACnB,KAAK,gBAAgB,GACtB,MAAM,eAAe,CAAA;AAGtB,OAAO,EACL,SAAS,EACT,WAAW,EACX,aAAa,EACb,iBAAiB,EACjB,mBAAmB,EACnB,iBAAiB,EACjB,kBAAkB,EAClB,oBAAoB,EACpB,oBAAoB,EACpB,wBAAwB,EACxB,6BAA6B,EAC7B,iBAAiB,EACjB,kBAAkB,EAClB,KAAK,gBAAgB,GACtB,MAAM,iBAAiB,CAAA;AAGxB,OAAO,EAEL,QAAQ,EACR,KAAK,YAAY,EACjB,KAAK,eAAe,EAGpB,EAAE,IAAI,QAAQ,EAAE,oDAAoD;AACpE,KAAK,EACL,IAAI,EACJ,IAAI,EACJ,KAAK,EACL,OAAO,EACP,SAAS,EACT,EAAE,EACF,OAAO,EACP,MAAM,EACN,KAAK,EACL,KAAK,EAGL,EAAE,EACF,QAAQ,EAGR,IAAI,EACJ,MAAM,EAGN,MAAM,EAGN,GAAG,EACH,OAAO,EACP,MAAM,EACN,KAAK,YAAY,EACjB,KAAK,WAAW,GACjB,MAAM,iBAAiB,CAAA;AAGxB,OAAO,EACL,UAAU,EACV,WAAW,EACX,SAAS,IAAI,cAAc,EAC3B,oBAAoB,EACpB,eAAe,EACf,WAAW,EACX,YAAY,EACZ,iBAAiB,EACjB,KAAK,SAAS,IAAI,cAAc,EAChC,KAAK,aAAa,EAClB,KAAK,WAAW,EAChB,KAAK,SAAS,EACd,KAAK,QAAQ,EACb,KAAK,WAAW,EAChB,KAAK,iBAAiB,EACtB,KAAK,YAAY,EACjB,KAAK,iBAAiB,EACtB,KAAK,eAAe,GACrB,MAAM,kBAAkB,CAAA;AAGzB,OAAO,EACL,eAAe,EACf,cAAc,EACd,iBAAiB,EACjB,gBAAgB,EAChB,KAAK,iBAAiB,EACtB,KAAK,eAAe,GACrB,MAAM,gBAAgB,CAAA;AAGvB,OAAO,EACL,SAAS,EACT,UAAU,EACV,WAAW,EACX,gBAAgB,EAChB,YAAY,EACZ,QAAQ,EACR,WAAW,EACX,YAAY,EACZ,iBAAiB,EACjB,iBAAiB,EACjB,KAAK,gBAAgB,EACrB,KAAK,SAAS,GACf,MAAM,cAAc,CAAA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAOH,YAAY,EACV,oBAAoB,EACpB,UAAU,EACV,iBAAiB,EACjB,gBAAgB,EAChB,cAAc,EACd,QAAQ,EACR,YAAY,EACZ,WAAW,EACX,YAAY,EACZ,WAAW,EACX,YAAY,EACZ,gBAAgB,IAAI,oBAAoB,EACxC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,WAAW,EACX,YAAY,EACZ,oBAAoB,EACpB,YAAY,EACZ,kBAAkB,EAClB,iBAAiB,EACjB,sBAAsB,EACtB,sBAAsB,EACtB,kBAAkB,EAClB,wBAAwB,EACxB,4BAA4B,EAC5B,wBAAwB,EACxB,yBAAyB,EACzB,qBAAqB,EACrB,uBAAuB,EACvB,mBAAmB,EACnB,kBAAkB,EAClB,eAAe,EACf,gBAAgB,EAChB,gBAAgB,GACjB,MAAM,YAAY,CAAA;AAGnB,OAAO,EAAE,MAAM,EAAE,KAAK,YAAY,EAAE,MAAM,aAAa,CAAA;AAGvD,OAAO,EACL,aAAa,EACb,sBAAsB,EACtB,sBAAsB,EACtB,oBAAoB,EACpB,SAAS,EACT,KAAK,eAAe,EACpB,KAAK,gBAAgB,EACrB,KAAK,iBAAiB,EACtB,KAAK,cAAc,EACnB,KAAK,gBAAgB,GACtB,MAAM,eAAe,CAAA;AAGtB,OAAO,EACL,SAAS,EACT,WAAW,EACX,aAAa,EACb,iBAAiB,EACjB,mBAAmB,EACnB,iBAAiB,EACjB,kBAAkB,EAClB,oBAAoB,EACpB,oBAAoB,EACpB,wBAAwB,EACxB,6BAA6B,EAC7B,iBAAiB,EACjB,kBAAkB,EAClB,KAAK,gBAAgB,EACrB,KAAK,kBAAkB,EACvB,KAAK,aAAa,GACnB,MAAM,iBAAiB,CAAA;AAGxB,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,eAAe,CAAA;AAGtF,OAAO,EACL,QAAQ,EACR,KAAK,YAAY,EACjB,KAAK,eAAe,EACpB,EAAE,EACF,KAAK,EACL,IAAI,EACJ,IAAI,EACJ,KAAK,EACL,OAAO,EACP,SAAS,EACT,EAAE,EACF,OAAO,EACP,MAAM,EACN,KAAK,EACL,KAAK,EACL,EAAE,EACF,QAAQ,EACR,IAAI,EACJ,MAAM,EACN,MAAM,EACN,GAAG,EACH,OAAO,EACP,MAAM,EACN,KAAK,YAAY,EACjB,KAAK,WAAW,GACjB,MAAM,iBAAiB,CAAA;AAGxB,OAAO,EACL,SAAS,EACT,UAAU,EACV,WAAW,EACX,gBAAgB,EAChB,YAAY,EACZ,QAAQ,EACR,WAAW,EACX,KAAK,gBAAgB,GACtB,MAAM,cAAc,CAAA;AAGrB,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAA;AAO9C,OAAO,EACL,EAAE,EACF,EAAE,IAAI,OAAO,EACb,MAAM,EACN,cAAc,EACd,SAAS,EACT,sBAAsB,EACtB,mBAAmB,EACnB,YAAY,EACZ,KAAK,OAAO,EAEZ,qBAAqB,EACrB,KAAK,eAAe,EACpB,KAAK,eAAe,EACpB,KAAK,uBAAuB,EAC5B,qBAAqB,EACrB,uBAAuB,EACvB,YAAY,EACZ,YAAY,GACb,MAAM,SAAS,CAAA;AAGhB,OAAO,EAAE,EAAE,IAAI,QAAQ,EAAE,MAAM,iBAAiB,CAAA;AAGhD,cAAc,iBAAiB,CAAA;AAG/B,cAAc,sBAAsB,CAAA;AAOpC,OAAO,EACL,UAAU,EACV,WAAW,EACX,SAAS,IAAI,cAAc,EAC3B,oBAAoB,EACpB,eAAe,EACf,mBAAmB,EACnB,cAAc,EACd,cAAc,EACd,aAAa,EACb,iBAAiB,EACjB,WAAW,EACX,YAAY,EACZ,iBAAiB,EACjB,KAAK,SAAS,IAAI,cAAc,EAChC,KAAK,aAAa,EAClB,KAAK,WAAW,EAChB,KAAK,SAAS,EACd,KAAK,QAAQ,EACb,KAAK,WAAW,EAChB,KAAK,iBAAiB,EACtB,KAAK,YAAY,EACjB,KAAK,WAAW,EAChB,KAAK,iBAAiB,EACtB,KAAK,eAAe,GACrB,MAAM,kBAAkB,CAAA;AAGzB,OAAO,EACL,SAAS,IAAI,cAAc,EAC3B,KAAK,WAAW,EAChB,KAAK,SAAS,EACd,KAAK,eAAe,EACpB,KAAK,wBAAwB,EAC7B,KAAK,iBAAiB,GACvB,MAAM,iBAAiB,CAAA;AAGxB,OAAO,EACL,eAAe,EACf,cAAc,EACd,iBAAiB,EACjB,gBAAgB,EAChB,KAAK,iBAAiB,EACtB,KAAK,eAAe,GACrB,MAAM,gBAAgB,CAAA;AAGvB,OAAO,EACL,YAAY,EACZ,iBAAiB,EACjB,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,EAChB,eAAe,EACf,KAAK,SAAS,EACd,KAAK,mBAAmB,EACxB,KAAK,aAAa,GACnB,MAAM,cAAc,CAAA;AAGrB,OAAO,EACL,aAAa,EACb,YAAY,EACZ,cAAc,EACd,mBAAmB,EACnB,oBAAoB,EACpB,UAAU,EACV,KAAK,YAAY,EACjB,KAAK,WAAW,EAChB,KAAK,cAAc,EACnB,KAAK,UAAU,EACf,KAAK,WAAW,EAChB,KAAK,qBAAqB,EAC1B,KAAK,YAAY,EACjB,KAAK,iBAAiB,EACtB,KAAK,eAAe,EACpB,KAAK,kBAAkB,GACxB,MAAM,aAAa,CAAA;AAGpB,OAAO,EAEL,WAAW,EACX,UAAU,EACV,aAAa,EAGb,UAAU,EACV,aAAa,EACb,QAAQ,EACR,UAAU,EACV,eAAe,EACf,WAAW,EACX,iBAAiB,EAGjB,KAAK,IAAI,EACT,KAAK,QAAQ,EACb,KAAK,UAAU,EACf,KAAK,mBAAmB,EACxB,KAAK,gBAAgB,EACrB,KAAK,aAAa,EAClB,KAAK,OAAO,EACZ,KAAK,QAAQ,EACb,KAAK,WAAW,EAChB,KAAK,UAAU,EACf,KAAK,cAAc,EACnB,KAAK,aAAa,EAClB,KAAK,UAAU,EACf,KAAK,eAAe,GACrB,MAAM,yBAAyB,CAAA;AAGhC,OAAO,EAEL,WAAW,EAGX,cAAc,EACd,eAAe,EAGf,SAAS,EAGT,OAAO,EACP,cAAc,EAGd,KAAK,YAAY,EACjB,KAAK,UAAU,EACf,KAAK,YAAY,EACjB,KAAK,UAAU,EACf,KAAK,kBAAkB,EACvB,KAAK,YAAY,EACjB,KAAK,qBAAqB,EAC1B,KAAK,oBAAoB,EACzB,KAAK,gBAAgB,EACrB,KAAK,yBAAyB,EAC9B,KAAK,gBAAgB,EACrB,KAAK,cAAc,GACpB,MAAM,YAAY,CAAA;AAGnB,OAAO,EAEL,cAAc,EACd,iBAAiB,EACjB,YAAY,EACZ,cAAc,EACd,gBAAgB,EAChB,aAAa,EACb,aAAa,EAGb,gBAAgB,EAGhB,WAAW,EAGX,cAAc,EAGd,aAAa,EAGb,SAAS,EAGT,KAAK,cAAc,EACnB,KAAK,cAAc,EACnB,KAAK,YAAY,EACjB,KAAK,SAAS,EACd,KAAK,eAAe,EACpB,KAAK,YAAY,EACjB,KAAK,qBAAqB,EAC1B,KAAK,qBAAqB,EAC1B,KAAK,aAAa,EAClB,KAAK,eAAe,EACpB,KAAK,eAAe,GACrB,MAAM,YAAY,CAAA;AAGnB,OAAO,EACL,8BAA8B,EAC9B,4BAA4B,EAC5B,cAAc,EACd,cAAc,EACd,KAAK,wBAAwB,EAC7B,KAAK,gBAAgB,EACrB,KAAK,6BAA6B,GACnC,MAAM,+BAA+B,CAAA;AAGtC,OAAO,EACL,eAAe,EACf,SAAS,EACT,WAAW,EACX,WAAW,EACX,WAAW,EACX,KAAK,MAAM,EACX,KAAK,YAAY,GAClB,MAAM,aAAa,CAAA;AAOpB,OAAO,EACL,eAAe,EACf,wBAAwB,EACxB,gBAAgB,EAChB,eAAe,EACf,KAAK,sBAAsB,EAC3B,KAAK,oBAAoB,EACzB,KAAK,2BAA2B,EAChC,KAAK,yBAAyB,EAC9B,KAAK,uBAAuB,EAC5B,KAAK,cAAc,EACnB,KAAK,UAAU,EACf,KAAK,cAAc,EACnB,KAAK,sBAAsB,GAC5B,MAAM,uBAAuB,CAAA;AAG9B,OAAO,EAAE,SAAS,EAAE,KAAK,gBAAgB,EAAE,MAAM,kBAAkB,CAAA;AAMnE,YAAY,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAA;AAGvD,OAAO,EACL,oBAAoB,EACpB,eAAe,EACf,qBAAqB,EACrB,KAAK,YAAY,EACjB,KAAK,kBAAkB,EACvB,KAAK,YAAY,GAClB,MAAM,qBAAqB,CAAA"}

package/dist/index.js

@@ -1,54 +1,101 @@
 /**
- * ai-functions - Core AI primitives for building intelligent applications
+ * ai-functions - Full-featured AI primitives for building intelligent applications
  *
- * This is the foundational package that all other primitives depend on.
- * It provides:
- * - RPC primitives via rpc.do
- * - AI function types and interfaces
- * - Core AI() and ai() constructors
- * - Embedding utilities from AI SDK
- * - Cloudflare Workers AI provider (default for embeddings)
- * - Tagged template literal support for all functions
- * - Async iterator support for list/extract
+ * This package provides the complete feature set including:
+ * - Core primitives from ai-core (generate, AIPromise, templates, context)
+ * - Batch processing (BatchQueue, BatchMapPromise)
+ * - Retry/resilience patterns (RetryPolicy, CircuitBreaker, FallbackChain)
+ * - Budget tracking (BudgetTracker, TokenCounter)
+ * - Caching (MemoryCache, EmbeddingCache, GenerationCache)
+ * - Tool orchestration (AgenticLoop, ToolRouter)
+ * - Embeddings
+ * - Provider integrations
+ *
+ * For lightweight usage with just core primitives, use ai-core directly.
  *
  * @packageDocumentation
  */
-// TODO: Re-enable RPC exports when rpc.do is published
-// export * from 'rpc.do'
-// Export AI function types and interfaces
-export * from './types.js';
-export * from './ai.js';
-// Export embedding utilities
-export * from './embeddings.js';
-// Export generation functions with smart model routing
-export { generateObject, generateText, streamObject, streamText } from './generate.js';
-// Export simplified schema helper
+// Schema exports
 export { schema } from './schema.js';
-// Export providers
-export * from './providers/index.js';
-// Export template utilities
+// Template exports
 export { parseTemplate, createTemplateFunction, createChainablePromise, createStreamableList, withBatch, } from './template.js';
-// Export AIPromise utilities for promise pipelining
+// AIPromise exports
 export { AIPromise, isAIPromise, getRawPromise, createTextPromise, createObjectPromise, createListPromise, createListsPromise, createBooleanPromise, createExtractPromise, createAITemplateFunction, parseTemplateWithDependencies, AI_PROMISE_SYMBOL, RAW_PROMISE_SYMBOL, } from './ai-promise.js';
-// Export AI primitives
-export { 
-// Core generate primitive
-generate, 
-// Generative functions
-ai as aiPrompt, // Renamed to avoid conflict with AIProxy from ai.ts
-write, code, list, lists, extract, summarize, is, diagram, slides, image, video, 
-// Agentic functions
-do, research, 
-// Web functions
-read, browse, 
-// Decision functions
-decide, 
-// Human-in-the-loop functions
-ask, approve, review, } from './primitives.js';
+// Generation exports
+export { generateObject, generateText, streamObject, streamText } from './generate.js';
+// Primitives
+export { generate, ai, write, code, list, lists, extract, summarize, is, diagram, slides, image, video, do, research, read, browse, decide, ask, approve, review, } from './primitives.js';
+// Context exports
+export { configure, getContext, withContext, getGlobalContext, resetContext, getModel, getProvider, } from './context.js';
+// Type guards
+export { isZodSchema } from './type-guards.js';
+// ============================================================================
+// Export AI Proxy (the AI class/proxy from ai.ts)
+// This is separate from the core 'ai' template function
+// ============================================================================
+export { AI, ai as aiProxy, define, defineFunction, functions, createFunctionRegistry, resetGlobalRegistry, withTemplate, 
+// Also export from new split modules
+createSchemaFunctions, createDefinedFunction, convertArgsToJSONSchema, fillTemplate, generateCode, } from './ai.js';
+// Also export 'ai' primitive as 'aiPrompt' for convenience
+export { ai as aiPrompt } from './primitives.js';
+// Export embedding utilities (not in ai-core)
+export * from './embeddings.js';
+// Export providers (not in ai-core)
+export * from './providers/index.js';
+// ============================================================================
+// Extended exports (NOT in ai-core)
+// ============================================================================
 // Export batch processing
-export { BatchQueue, createBatch, withBatch as withBatchQueue, registerBatchAdapter, getBatchAdapter, isBatchMode, deferToBatch, BATCH_MODE_SYMBOL, } from './batch-queue.js';
+export { BatchQueue, createBatch, withBatch as withBatchQueue, registerBatchAdapter, getBatchAdapter, registerFlexAdapter, getFlexAdapter, hasFlexAdapter, tiersForModel, modelSupportsTier, isBatchMode, deferToBatch, BATCH_MODE_SYMBOL, } from './batch-queue.js';
+// Export per-model policy data (sourced from language-models)
+export { policyFor as modelPolicyFor, } from 'language-models';
 // Export batch map for automatic batching
 export { BatchMapPromise, createBatchMap, isBatchMapPromise, BATCH_MAP_SYMBOL, } from './batch-map.js';
-// Export execution context
-export { configure, getContext, withContext, getGlobalContext, resetContext, getModel, getProvider, getBatchMode, getBatchThreshold, shouldUseBatchAPI, } from './context.js';
+// Export additional execution context features (full version with batch/budget)
+export { getBatchMode, getBatchThreshold, shouldUseBatchAPI, getFlexThreshold, getExecutionTier, isFlexAvailable, } from './context.js';
+// Export budget tracking and request tracing
+export { BudgetTracker, TokenCounter, RequestContext, BudgetExceededError, createRequestContext, withBudget, } from './budget.js';
+// Export agentic tool orchestration
+export { 
+// Core classes
+AgenticLoop, ToolRouter, ToolValidator, 
+// Tool composition utilities
+createTool, createToolset, wrapTool, cachedTool, rateLimitedTool, timeoutTool, createAgenticLoop, } from './tool-orchestration.js';
+// Export caching layer for embeddings and generations
+export { 
+// Core cache storage
+MemoryCache, 
+// Specialized caches
+EmbeddingCache, GenerationCache, 
+// Cache wrapper
+withCache, 
+// Utilities
+hashKey, createCacheKey, } from './cache.js';
+// Export retry/fallback patterns with exponential backoff
+export { 
+// Error types and classification
+RetryableError, NonRetryableError, NetworkError, RateLimitError, CircuitOpenError, ErrorCategory, classifyError, 
+// Backoff calculation
+calculateBackoff, 
+// Retry policy
+RetryPolicy, 
+// Circuit breaker
+CircuitBreaker, 
+// Fallback chain
+FallbackChain, 
+// Convenience helper
+withRetry, } from './retry.js';
+// Export digital-objects function registry
+export { DigitalObjectsFunctionRegistry, createDigitalObjectsRegistry, FUNCTION_NOUNS, FUNCTION_VERBS, } from './digital-objects-registry.js';
+// Export configurable logger
+export { configureLogger, getLogger, resetLogger, getLogLevel, setLogLevel, } from './logger.js';
+// ============================================================================
+// V3 middleware stack — cache / budget / trace via wrapLanguageModel
+// ============================================================================
+// Composable middleware primitives
+export { cacheMiddleware, embeddingCacheMiddleware, budgetMiddleware, traceMiddleware, } from './middleware/index.js';
+// Convenience composer
+export { wrapForV3 } from './wrap-for-v3.js';
+// Pluggable persistence primitive for trace/eval entries
+export { InMemoryEvalLogStore, getEvalLogStore, configureEvalLogStore, } from './eval-log/index.js';
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,uDAAuD;AACvD,yBAAyB;AAEzB,0CAA0C;AAC1C,cAAc,YAAY,CAAA;AAC1B,cAAc,SAAS,CAAA;AAEvB,6BAA6B;AAC7B,cAAc,iBAAiB,CAAA;AAE/B,uDAAuD;AACvD,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,eAAe,CAAA;AAEtF,kCAAkC;AAClC,OAAO,EAAE,MAAM,EAAqB,MAAM,aAAa,CAAA;AAEvD,mBAAmB;AACnB,cAAc,sBAAsB,CAAA;AAEpC,4BAA4B;AAC5B,OAAO,EACL,aAAa,EACb,sBAAsB,EACtB,sBAAsB,EACtB,oBAAoB,EACpB,SAAS,GAMV,MAAM,eAAe,CAAA;AAEtB,oDAAoD;AACpD,OAAO,EACL,SAAS,EACT,WAAW,EACX,aAAa,EACb,iBAAiB,EACjB,mBAAmB,EACnB,iBAAiB,EACjB,kBAAkB,EAClB,oBAAoB,EACpB,oBAAoB,EACpB,wBAAwB,EACxB,6BAA6B,EAC7B,iBAAiB,EACjB,kBAAkB,GAEnB,MAAM,iBAAiB,CAAA;AAExB,uBAAuB;AACvB,OAAO;AACL,0BAA0B;AAC1B,QAAQ;AAIR,uBAAuB;AACvB,EAAE,IAAI,QAAQ,EAAE,oDAAoD;AACpE,KAAK,EACL,IAAI,EACJ,IAAI,EACJ,KAAK,EACL,OAAO,EACP,SAAS,EACT,EAAE,EACF,OAAO,EACP,MAAM,EACN,KAAK,EACL,KAAK;AAEL,oBAAoB;AACpB,EAAE,EACF,QAAQ;AAER,gBAAgB;AAChB,IAAI,EACJ,MAAM;AAEN,qBAAqB;AACrB,MAAM;AAEN,8BAA8B;AAC9B,GAAG,EACH,OAAO,EACP,MAAM,GAGP,MAAM,iBAAiB,CAAA;AAExB,0BAA0B;AAC1B,OAAO,EACL,UAAU,EACV,WAAW,EACX,SAAS,IAAI,cAAc,EAC3B,oBAAoB,EACpB,eAAe,EACf,WAAW,EACX,YAAY,EACZ,iBAAiB,GAWlB,MAAM,kBAAkB,CAAA;AAEzB,0CAA0C;AAC1C,OAAO,EACL,eAAe,EACf,cAAc,EACd,iBAAiB,EACjB,gBAAgB,GAGjB,MAAM,gBAAgB,CAAA;AAEvB,2BAA2B;AAC3B,OAAO,EACL,SAAS,EACT,UAAU,EACV,WAAW,EACX,gBAAgB,EAChB,YAAY,EACZ,QAAQ,EACR,WAAW,EACX,YAAY,EACZ,iBAAiB,EACjB,iBAAiB,GAGlB,MAAM,cAAc,CAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AA6CH,iBAAiB;AACjB,OAAO,EAAE,MAAM,EAAqB,MAAM,aAAa,CAAA;AAEvD,mBAAmB;AACnB,OAAO,EACL,aAAa,EACb,sBAAsB,EACtB,sBAAsB,EACtB,oBAAoB,EACpB,SAAS,GAMV,MAAM,eAAe,CAAA;AAEtB,oBAAoB;AACpB,OAAO,EACL,SAAS,EACT,WAAW,EACX,aAAa,EACb,iBAAiB,EACjB,mBAAmB,EACnB,iBAAiB,EACjB,kBAAkB,EAClB,oBAAoB,EACpB,oBAAoB,EACpB,wBAAwB,EACxB,6BAA6B,EAC7B,iBAAiB,EACjB,kBAAkB,GAInB,MAAM,iBAAiB,CAAA;AAExB,qBAAqB;AACrB,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,eAAe,CAAA;AAEtF,aAAa;AACb,OAAO,EACL,QAAQ,EAGR,EAAE,EACF,KAAK,EACL,IAAI,EACJ,IAAI,EACJ,KAAK,EACL,OAAO,EACP,SAAS,EACT,EAAE,EACF,OAAO,EACP,MAAM,EACN,KAAK,EACL,KAAK,EACL,EAAE,EACF,QAAQ,EACR,IAAI,EACJ,MAAM,EACN,MAAM,EACN,GAAG,EACH,OAAO,EACP,MAAM,GAGP,MAAM,iBAAiB,CAAA;AAExB,kBAAkB;AAClB,OAAO,EACL,SAAS,EACT,UAAU,EACV,WAAW,EACX,gBAAgB,EAChB,YAAY,EACZ,QAAQ,EACR,WAAW,GAEZ,MAAM,cAAc,CAAA;AAErB,cAAc;AACd,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAA;AAE9C,+EAA+E;AAC/E,kDAAkD;AAClD,wDAAwD;AACxD,+EAA+E;AAE/E,OAAO,EACL,EAAE,EACF,EAAE,IAAI,OAAO,EACb,MAAM,EACN,cAAc,EACd,SAAS,EACT,sBAAsB,EACtB,mBAAmB,EACnB,YAAY;AAEZ,qCAAqC;AACrC,qBAAqB,EAIrB,qBAAqB,EACrB,uBAAuB,EACvB,YAAY,EACZ,YAAY,GACb,MAAM,SAAS,CAAA;AAEhB,2DAA2D;AAC3D,OAAO,EAAE,EAAE,IAAI,QAAQ,EAAE,MAAM,iBAAiB,CAAA;AAEhD,8CAA8C;AAC9C,cAAc,iBAAiB,CAAA;AAE/B,oCAAoC;AACpC,cAAc,sBAAsB,CAAA;AAEpC,+EAA+E;AAC/E,oCAAoC;AACpC,+EAA+E;AAE/E,0BAA0B;AAC1B,OAAO,EACL,UAAU,EACV,WAAW,EACX,SAAS,IAAI,cAAc,EAC3B,oBAAoB,EACpB,eAAe,EACf,mBAAmB,EACnB,cAAc,EACd,cAAc,EACd,aAAa,EACb,iBAAiB,EACjB,WAAW,EACX,YAAY,EACZ,iBAAiB,GAYlB,MAAM,kBAAkB,CAAA;AAEzB,8DAA8D;AAC9D,OAAO,EACL,SAAS,IAAI,cAAc,GAM5B,MAAM,iBAAiB,CAAA;AAExB,0CAA0C;AAC1C,OAAO,EACL,eAAe,EACf,cAAc,EACd,iBAAiB,EACjB,gBAAgB,GAGjB,MAAM,gBAAgB,CAAA;AAEvB,gFAAgF;AAChF,OAAO,EACL,YAAY,EACZ,iBAAiB,EACjB,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,EAChB,eAAe,GAIhB,MAAM,cAAc,CAAA;AAErB,6CAA6C;AAC7C,OAAO,EACL,aAAa,EACb,YAAY,EACZ,cAAc,EACd,mBAAmB,EACnB,oBAAoB,EACpB,UAAU,GAWX,MAAM,aAAa,CAAA;AAEpB,oCAAoC;AACpC,OAAO;AACL,eAAe;AACf,WAAW,EACX,UAAU,EACV,aAAa;AAEb,6BAA6B;AAC7B,UAAU,EACV,aAAa,EACb,QAAQ,EACR,UAAU,EACV,eAAe,EACf,WAAW,EACX,iBAAiB,GAiBlB,MAAM,yBAAyB,CAAA;AAEhC,sDAAsD;AACtD,OAAO;AACL,qBAAqB;AACrB,WAAW;AAEX,qBAAqB;AACrB,cAAc,EACd,eAAe;AAEf,gBAAgB;AAChB,SAAS;AAET,YAAY;AACZ,OAAO,EACP,cAAc,GAef,MAAM,YAAY,CAAA;AAEnB,0DAA0D;AAC1D,OAAO;AACL,iCAAiC;AACjC,cAAc,EACd,iBAAiB,EACjB,YAAY,EACZ,cAAc,EACd,gBAAgB,EAChB,aAAa,EACb,aAAa;AAEb,sBAAsB;AACtB,gBAAgB;AAEhB,eAAe;AACf,WAAW;AAEX,kBAAkB;AAClB,cAAc;AAEd,iBAAiB;AACjB,aAAa;AAEb,qBAAqB;AACrB,SAAS,GAcV,MAAM,YAAY,CAAA;AAEnB,2CAA2C;AAC3C,OAAO,EACL,8BAA8B,EAC9B,4BAA4B,EAC5B,cAAc,EACd,cAAc,GAIf,MAAM,+BAA+B,CAAA;AAEtC,6BAA6B;AAC7B,OAAO,EACL,eAAe,EACf,SAAS,EACT,WAAW,EACX,WAAW,EACX,WAAW,GAGZ,MAAM,aAAa,CAAA;AAEpB,+EAA+E;AAC/E,qEAAqE;AACrE,+EAA+E;AAE/E,mCAAmC;AACnC,OAAO,EACL,eAAe,EACf,wBAAwB,EACxB,gBAAgB,EAChB,eAAe,GAUhB,MAAM,uBAAuB,CAAA;AAE9B,uBAAuB;AACvB,OAAO,EAAE,SAAS,EAAyB,MAAM,kBAAkB,CAAA;AAQnE,yDAAyD;AACzD,OAAO,EACL,oBAAoB,EACpB,eAAe,EACf,qBAAqB,GAItB,MAAM,qBAAqB,CAAA"}

package/dist/logger.d.ts

@@ -0,0 +1,118 @@
+/**
+ * Configurable Logger for AI Functions
+ *
+ * Provides a pluggable logging interface that defaults to console but can be
+ * configured to use any logging implementation. This allows library consumers
+ * to integrate with their own logging infrastructure.
+ *
+ * @example
+ * ```ts
+ * import { configureLogger, getLogger } from 'ai-functions'
+ *
+ * // Use with default console logger
+ * const logger = getLogger()
+ * logger.warn('Something happened')
+ *
+ * // Configure a custom logger
+ * configureLogger({
+ *   debug: (msg, ...args) => myLogger.debug(msg, ...args),
+ *   info: (msg, ...args) => myLogger.info(msg, ...args),
+ *   warn: (msg, ...args) => myLogger.warn(msg, ...args),
+ *   error: (msg, ...args) => myLogger.error(msg, ...args),
+ * })
+ *
+ * // Disable logging entirely
+ * configureLogger(null)
+ * ```
+ *
+ * @packageDocumentation
+ */
+/**
+ * Logger interface that matches the standard console methods.
+ *
+ * All methods are optional to allow partial implementations.
+ * Missing methods will be no-ops when called.
+ */
+export interface Logger {
+    /** Debug level logging (verbose) */
+    debug?: (message: string, ...args: unknown[]) => void;
+    /** Info level logging (general information) */
+    info?: (message: string, ...args: unknown[]) => void;
+    /** Warning level logging (potential issues) */
+    warn?: (message: string, ...args: unknown[]) => void;
+    /** Error level logging (errors and exceptions) */
+    error?: (message: string, ...args: unknown[]) => void;
+}
+/**
+ * Logger configuration options
+ */
+export interface LoggerConfig {
+    /** The logger implementation to use */
+    logger?: Logger | null;
+    /** Minimum log level to output */
+    level?: 'debug' | 'info' | 'warn' | 'error' | 'silent';
+}
+/**
+ * Configure the global logger for ai-functions.
+ *
+ * @param config - Logger configuration or just a logger instance
+ *
+ * @example
+ * ```ts
+ * // Use a custom logger
+ * configureLogger({
+ *   logger: {
+ *     debug: (msg) => myLogger.debug(msg),
+ *     info: (msg) => myLogger.info(msg),
+ *     warn: (msg) => myLogger.warn(msg),
+ *     error: (msg) => myLogger.error(msg),
+ *   },
+ *   level: 'info',
+ * })
+ *
+ * // Just set the log level
+ * configureLogger({ level: 'debug' })
+ *
+ * // Disable all logging
+ * configureLogger({ logger: null })
+ * // or
+ * configureLogger({ level: 'silent' })
+ *
+ * // Reset to defaults
+ * configureLogger({})
+ * ```
+ */
+export declare function configureLogger(config: LoggerConfig | Logger | null): void;
+/**
+ * Get the current logger instance.
+ *
+ * Returns a logger that respects the configured log level.
+ * Methods for levels below the configured minimum will be no-ops.
+ *
+ * @returns Logger instance with level filtering applied
+ *
+ * @example
+ * ```ts
+ * const logger = getLogger()
+ * logger.warn('This is a warning')
+ * logger.error('This is an error', { details: 'foo' })
+ * ```
+ */
+export declare function getLogger(): Required<Logger>;
+/**
+ * Reset the logger to default configuration.
+ *
+ * Restores the console-based logger with 'warn' minimum level.
+ */
+export declare function resetLogger(): void;
+/**
+ * Get the current log level
+ */
+export declare function getLogLevel(): string;
+/**
+ * Set the minimum log level
+ *
+ * @param level - Minimum level to log ('debug' | 'info' | 'warn' | 'error' | 'silent')
+ */
+export declare function setLogLevel(level: 'debug' | 'info' | 'warn' | 'error' | 'silent'): void;
+//# sourceMappingURL=logger.d.ts.map

package/dist/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../src/logger.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AAMH;;;;;GAKG;AACH,MAAM,WAAW,MAAM;IACrB,oCAAoC;IACpC,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,IAAI,CAAA;IACrD,+CAA+C;IAC/C,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,IAAI,CAAA;IACpD,+CAA+C;IAC/C,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,IAAI,CAAA;IACpD,kDAAkD;IAClD,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,IAAI,CAAA;CACtD;AAED;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,uCAAuC;IACvC,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;IACtB,kCAAkC;IAClC,KAAK,CAAC,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,QAAQ,CAAA;CACvD;AAyCD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,YAAY,GAAG,MAAM,GAAG,IAAI,GAAG,IAAI,CAiC1E;AAED;;;;;;;;;;;;;;GAcG;AACH,wBAAgB,SAAS,IAAI,QAAQ,CAAC,MAAM,CAAC,CAyB5C;AAED;;;;GAIG;AACH,wBAAgB,WAAW,IAAI,IAAI,CAGlC;AAED;;GAEG;AACH,wBAAgB,WAAW,IAAI,MAAM,CAEpC;AAED;;;;GAIG;AACH,wBAAgB,WAAW,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,QAAQ,GAAG,IAAI,CAEvF"}