ai-functions 0.2.19 → 0.3.0

package/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2023 Nathan Clevenger
+Copyright (c) 2024 AI Primitives
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

package/README.md

@@ -1,20 +1,48 @@
-# AI Functions
+# ai-functions
 
-Library for Developing and Managing AI Functions (including OpenAI GPT4 / GPT3.5)
+[![npm version](https://badge.fury.io/js/ai-functions.svg)](https://www.npmjs.com/package/ai-functions)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 
-Key Features:
+A powerful TypeScript library for building AI-powered applications with template literals and structured outputs. Uses `ai` SDK with OpenAI `gpt-4o` by default.
 
-- Enables easy development of AI functions
+## Features
 
-```javascript
-import { AI } from 'ai-functions'
+- 🚀 Template literal API for natural AI prompts
+- 🔄 Async iterator support for streaming responses
+- 📝 Structured output generation with Zod schemas
+- 🎯 Type-safe API with full TypeScript support
+- ⚡️ Built on `ai` SDK for streaming & object generation
+- 🔍 Support for various output formats (objects, arrays, enums)
 
-const { ai, gpt, list } = AI({ apiKey: OPENAI_API_KEY })
+## Installation
+
+```bash
+pnpm add ai-functions
 ```
 
-Then you can use magic `ai` functions:
+## Usage
+
+### Basic Template Literals
+
+```typescript
+import { ai, list } from 'ai-functions'
+
+// Simple text generation
+const text = ai`write a blog post in markdown starting with '# ${title}'`
+
+// Complex objects/arrays dumped to YAML
+const summary = ai`Summarize the itinerary: ${itinerary}`
+
+// Basic list generation
+const things = await list`fun things to do in Miami`
+console.log(things)
+
+// Using async iterator
+for await (const thing of list`fun things to do in Miami`) {
+  console.log(thing)
+}
 
-```javascript
+// Structured output
 const categorizeProduct = ai.categorizeProduct({
   productType: 'App | API | Marketplace | Platform | Packaged Service | Professional Service | Website',
   customer: 'ideal customer profile in 3-5 words',
@@ -25,26 +53,37 @@ const categorizeProduct = ai.categorizeProduct({
 const product = await categorizeProduct({ domain: name })
 ```
 
-you can also use `list` tagged template as a convienence function:
+### Configuration
 
-```javascript
-const things = await list`fun things to do in Miami`
-console.log(things)
+#### Specifying the model
+
+By default `ai-functions` uses the `openai` provider from the `ai` SDK. You can specify any openai model name as a string.
+
+```typescript
+const text = ai`write a blog post in markdown starting with "# ${title}"`({ model: 'gpt-4o' })
 ```
 
-or with Async iterators:
+Or you can pass in any other provider compatible with the `ai` SDK.
 
-```javascript
-for await (const thing of list`fun things to do in Miami`) {
-  console.log(thing)
-}
+```typescript
+import { anthropic } from '@ai-sdk/anthropic'
+
+const text = ai`write a function in TypeScript called ${name}`({ model: anthropic('claude-3-5-sonnet-20241022') })
+```
+
+#### Specifying other options
+
+You can pass in any other options supported by the `ai` SDK, like `system`, `temperature`, `maxTokens`, etc.
+
+```typescript
+const things = await list`fun things to do in ${city}`({ system: 'You are an expert tour guide', temperature: 0.2 })
 ```
 
-Or in a more complex example:
+### Composable Functions & Workflows
 
-```javascript
+```typescript
 const listBlogPosts = (count, topic) => list`${count} blog post titles about ${topic}`
-const writeBlogPost = (title) => gpt`write a blog post in markdown starting with "# ${title}"`
+const writeBlogPost = (title) => ai`write a blog post in markdown starting with '# ${title}'`
 
 async function* writeBlog(count, topic) {
   for await (const title of listBlogPosts(count, topic)) {
@@ -57,3 +96,55 @@ for await (const post of writeBlog(25, 'future of car sales')) {
   console.log({ post })
 }
 ```
+
+### Concurrency
+
+you can specify the `concurrency` option to limit the number of concurrent requests.
+
+```typescript
+activities.map(async (activity) => {
+  const result = await ai`write a paragraph overview of ${activity}`({ concurrency: 5 })
+})
+```
+
+## Development
+
+```bash
+# Install dependencies
+pnpm install
+
+# Run tests
+pnpm test
+
+# Run tests in watch mode
+pnpm test:watch
+
+# Build the package
+pnpm build
+
+# Lint the code
+pnpm lint
+
+# Format the code
+pnpm format
+```
+
+## Contributing
+
+Please read our [Contributing Guide](./CONTRIBUTING.md) to learn about our development process and how to propose bugfixes and improvements.
+
+## License
+
+MIT © [Drivly](https://driv.ly)
+
+## Dependencies
+
+This package uses the following key dependencies:
+
+- `ai` SDK for AI model integration
+- Defaults to `@ai-sdk/openai` provider
+- TypeScript for static typing
+- Zod for schema validation
+- Vitest for testing
+- ESLint for linting
+- Prettier for code formatting

package/bin/cli.js

@@ -0,0 +1,5 @@
+#!/usr/bin/env node
+
+import { cli } from '../dist/cli/index.js'
+
+cli()

package/dist/cli/index.d.ts

@@ -0,0 +1,10 @@
+interface CliOptions {
+    version?: boolean;
+    help?: boolean;
+}
+export declare function parseArgs(args: string[]): CliOptions;
+export declare function showHelp(): void;
+export declare function showVersion(): void;
+export declare function cli(args?: string[]): void;
+export {};
+//# sourceMappingURL=index.d.ts.map

package/dist/cli/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/cli/index.ts"],"names":[],"mappings":"AAEA,UAAU,UAAU;IAClB,OAAO,CAAC,EAAE,OAAO,CAAA;IACjB,IAAI,CAAC,EAAE,OAAO,CAAA;CACf;AAED,wBAAgB,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,UAAU,CAYpD;AAED,wBAAgB,QAAQ,IAAI,IAAI,CAQ/B;AAED,wBAAgB,WAAW,IAAI,IAAI,CAElC;AAED,wBAAgB,GAAG,CAAC,IAAI,GAAE,MAAM,EAA0B,GAAG,IAAI,CAUhE"}

package/dist/cli/index.js

@@ -0,0 +1,38 @@
+import { version } from '../../package.json';
+export function parseArgs(args) {
+    const options = {};
+    for (const arg of args) {
+        if (arg === '-v' || arg === '--version') {
+            options.version = true;
+        }
+        else if (arg === '-h' || arg === '--help') {
+            options.help = true;
+        }
+    }
+    return options;
+}
+export function showHelp() {
+    console.log(`
+Usage: ai-functions [options]
+
+Options:
+  -v, --version  Show version number
+  -h, --help     Show help
+`);
+}
+export function showVersion() {
+    console.log(`v${version}`);
+}
+export function cli(args = process.argv.slice(2)) {
+    const options = parseArgs(args);
+    if (options.version) {
+        showVersion();
+    }
+    else if (options.help) {
+        showHelp();
+    }
+    else {
+        showHelp();
+    }
+}
+//# sourceMappingURL=index.js.map

package/dist/cli/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/cli/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAA;AAO5C,MAAM,UAAU,SAAS,CAAC,IAAc;IACtC,MAAM,OAAO,GAAe,EAAE,CAAA;IAE9B,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,GAAG,KAAK,IAAI,IAAI,GAAG,KAAK,WAAW,EAAE,CAAC;YACxC,OAAO,CAAC,OAAO,GAAG,IAAI,CAAA;QACxB,CAAC;aAAM,IAAI,GAAG,KAAK,IAAI,IAAI,GAAG,KAAK,QAAQ,EAAE,CAAC;YAC5C,OAAO,CAAC,IAAI,GAAG,IAAI,CAAA;QACrB,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAA;AAChB,CAAC;AAED,MAAM,UAAU,QAAQ;IACtB,OAAO,CAAC,GAAG,CAAC;;;;;;CAMb,CAAC,CAAA;AACF,CAAC;AAED,MAAM,UAAU,WAAW;IACzB,OAAO,CAAC,GAAG,CAAC,IAAI,OAAO,EAAE,CAAC,CAAA;AAC5B,CAAC;AAED,MAAM,UAAU,GAAG,CAAC,OAAiB,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IACxD,MAAM,OAAO,GAAG,SAAS,CAAC,IAAI,CAAC,CAAA;IAE/B,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,WAAW,EAAE,CAAA;IACf,CAAC;SAAM,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;QACxB,QAAQ,EAAE,CAAA;IACZ,CAAC;SAAM,CAAC;QACN,QAAQ,EAAE,CAAA;IACZ,CAAC;AACH,CAAC"}

package/dist/cli/index.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=index.test.d.ts.map

package/dist/cli/index.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.test.d.ts","sourceRoot":"","sources":["../../src/cli/index.test.ts"],"names":[],"mappings":""}

package/dist/cli/index.test.js

@@ -0,0 +1,35 @@
+import { describe, it, expect, vi } from 'vitest';
+import { parseArgs, showHelp, showVersion, cli } from './index';
+describe('CLI', () => {
+    it('should parse version flag correctly', () => {
+        const options = parseArgs(['-v']);
+        expect(options.version).toBe(true);
+        const longOptions = parseArgs(['--version']);
+        expect(longOptions.version).toBe(true);
+    });
+    it('should parse help flag correctly', () => {
+        const options = parseArgs(['-h']);
+        expect(options.help).toBe(true);
+        const longOptions = parseArgs(['--help']);
+        expect(longOptions.help).toBe(true);
+    });
+    it('should show version number', () => {
+        const consoleSpy = vi.spyOn(console, 'log');
+        showVersion();
+        expect(consoleSpy).toHaveBeenCalled();
+        consoleSpy.mockRestore();
+    });
+    it('should show help message', () => {
+        const consoleSpy = vi.spyOn(console, 'log');
+        showHelp();
+        expect(consoleSpy).toHaveBeenCalled();
+        consoleSpy.mockRestore();
+    });
+    it('should default to help when no arguments provided', () => {
+        const consoleSpy = vi.spyOn(console, 'log');
+        cli([]);
+        expect(consoleSpy).toHaveBeenCalled();
+        consoleSpy.mockRestore();
+    });
+});
+//# sourceMappingURL=index.test.js.map

package/dist/cli/index.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.test.js","sourceRoot":"","sources":["../../src/cli/index.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AACjD,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,EAAE,MAAM,SAAS,CAAA;AAE/D,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE;IACnB,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;QACjC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAElC,MAAM,WAAW,GAAG,SAAS,CAAC,CAAC,WAAW,CAAC,CAAC,CAAA;QAC5C,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACxC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;QACjC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAE/B,MAAM,WAAW,GAAG,SAAS,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAA;QACzC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACrC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;QAC3C,WAAW,EAAE,CAAA;QACb,MAAM,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAA;QACrC,UAAU,CAAC,WAAW,EAAE,CAAA;IAC1B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;QAC3C,QAAQ,EAAE,CAAA;QACV,MAAM,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAA;QACrC,UAAU,CAAC,WAAW,EAAE,CAAA;IAC1B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;QAC3C,GAAG,CAAC,EAAE,CAAC,CAAA;QACP,MAAM,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAA;QACrC,UAAU,CAAC,WAAW,EAAE,CAAA;IAC1B,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/constants/models.d.ts

@@ -0,0 +1,10 @@
+export declare const MODEL_BEST = "gpt-4o";
+export declare const MODEL_FAST = "gpt-4o-mini";
+export declare const DEFAULT_MODEL = "gpt-4o";
+export declare const FALLBACK_MODEL = "gpt-4o-mini";
+export declare const MODELS: {
+    readonly BEST: "gpt-4o";
+    readonly FAST: "gpt-4o-mini";
+};
+export type ModelType = typeof MODELS[keyof typeof MODELS];
+//# sourceMappingURL=models.d.ts.map

package/dist/constants/models.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../../src/constants/models.ts"],"names":[],"mappings":"AACA,eAAO,MAAM,UAAU,WAAW,CAAA;AAClC,eAAO,MAAM,UAAU,gBAAgB,CAAA;AAGvC,eAAO,MAAM,aAAa,WAAa,CAAA;AACvC,eAAO,MAAM,cAAc,gBAAa,CAAA;AAGxC,eAAO,MAAM,MAAM;;;CAGT,CAAA;AAGV,MAAM,MAAM,SAAS,GAAG,OAAO,MAAM,CAAC,MAAM,OAAO,MAAM,CAAC,CAAA"}

package/dist/constants/models.js

@@ -0,0 +1,12 @@
+// Model identifiers
+export const MODEL_BEST = 'gpt-4o';
+export const MODEL_FAST = 'gpt-4o-mini';
+// Default model configuration
+export const DEFAULT_MODEL = MODEL_BEST;
+export const FALLBACK_MODEL = MODEL_FAST;
+// Model constants for external use
+export const MODELS = {
+    BEST: MODEL_BEST,
+    FAST: MODEL_FAST
+};
+//# sourceMappingURL=models.js.map

package/dist/constants/models.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"models.js","sourceRoot":"","sources":["../../src/constants/models.ts"],"names":[],"mappings":"AAAA,oBAAoB;AACpB,MAAM,CAAC,MAAM,UAAU,GAAG,QAAQ,CAAA;AAClC,MAAM,CAAC,MAAM,UAAU,GAAG,aAAa,CAAA;AAEvC,8BAA8B;AAC9B,MAAM,CAAC,MAAM,aAAa,GAAG,UAAU,CAAA;AACvC,MAAM,CAAC,MAAM,cAAc,GAAG,UAAU,CAAA;AAExC,mCAAmC;AACnC,MAAM,CAAC,MAAM,MAAM,GAAG;IACpB,IAAI,EAAE,UAAU;IAChB,IAAI,EAAE,UAAU;CACR,CAAA"}

package/dist/converters/index.d.ts

@@ -0,0 +1,3 @@
+export * from './model';
+export * from './schema';
+//# sourceMappingURL=index.d.ts.map

package/dist/converters/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/converters/index.ts"],"names":[],"mappings":"AAAA,cAAc,SAAS,CAAA;AACvB,cAAc,UAAU,CAAA"}

package/dist/converters/index.js

@@ -0,0 +1,3 @@
+export * from './model';
+export * from './schema';
+//# sourceMappingURL=index.js.map

package/dist/converters/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/converters/index.ts"],"names":[],"mappings":"AAAA,cAAc,SAAS,CAAA;AACvB,cAAc,UAAU,CAAA"}

package/dist/converters/model.d.ts

@@ -0,0 +1,4 @@
+import { Model } from '../types';
+export declare function getOpenAIClient(): import("@ai-sdk/openai").OpenAIProvider;
+export declare function convertToOpenAIModel(model: Model): any;
+//# sourceMappingURL=model.d.ts.map

package/dist/converters/model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.d.ts","sourceRoot":"","sources":["../../src/converters/model.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,EAAE,MAAM,UAAU,CAAA;AAKhC,wBAAgB,eAAe,4CAO9B;AAED,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,KAAK,OAMhD"}

package/dist/converters/model.js

@@ -0,0 +1,19 @@
+import { createOpenAI } from '@ai-sdk/openai';
+// Create a singleton OpenAI client
+let clientInstance;
+export function getOpenAIClient() {
+    if (!clientInstance) {
+        clientInstance = createOpenAI({
+            baseURL: process.env.AI_GATEWAY,
+        });
+    }
+    return clientInstance;
+}
+export function convertToOpenAIModel(model) {
+    if (typeof model === 'string') {
+        const client = getOpenAIClient();
+        return client(model);
+    }
+    return model;
+}
+//# sourceMappingURL=model.js.map

package/dist/converters/model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.js","sourceRoot":"","sources":["../../src/converters/model.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAA;AAG7C,mCAAmC;AACnC,IAAI,cAA+C,CAAA;AAEnD,MAAM,UAAU,eAAe;IAC7B,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,cAAc,GAAG,YAAY,CAAC;YAC5B,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,UAAU;SAChC,CAAC,CAAA;IACJ,CAAC;IACD,OAAO,cAAc,CAAA;AACvB,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,KAAY;IAC/C,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,eAAe,EAAE,CAAA;QAChC,OAAO,MAAM,CAAC,KAAK,CAAC,CAAA;IACtB,CAAC;IACD,OAAO,KAAK,CAAA;AACd,CAAC"}

package/dist/converters/schema.d.ts

@@ -0,0 +1,4 @@
+import { z } from 'zod';
+import { SimpleSchema } from '../types';
+export declare function convertToZodSchema(schema: SimpleSchema): z.ZodObject<any>;
+//# sourceMappingURL=schema.d.ts.map

package/dist/converters/schema.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.d.ts","sourceRoot":"","sources":["../../src/converters/schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EAAE,YAAY,EAAqB,MAAM,UAAU,CAAA;AAiB1D,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,YAAY,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAQzE"}

package/dist/converters/schema.js

@@ -0,0 +1,25 @@
+import { z } from 'zod';
+function convertValue(value) {
+    if (Array.isArray(value)) {
+        return z.array(z.string().describe(value[0] || ''));
+    }
+    else if (typeof value === 'string') {
+        if (value.includes('|')) {
+            const enumValues = value.split('|').map(v => v.trim());
+            return z.enum(enumValues);
+        }
+        return z.string().describe(value);
+    }
+    else if (typeof value === 'object') {
+        return convertToZodSchema(value);
+    }
+    return z.string();
+}
+export function convertToZodSchema(schema) {
+    const zodSchema = {};
+    for (const [key, value] of Object.entries(schema)) {
+        zodSchema[key] = convertValue(value);
+    }
+    return z.object(zodSchema);
+}
+//# sourceMappingURL=schema.js.map

package/dist/converters/schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.js","sourceRoot":"","sources":["../../src/converters/schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAGvB,SAAS,YAAY,CAAC,KAAwB;IAC5C,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAA;IACrD,CAAC;SAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAA;YACtD,OAAO,CAAC,CAAC,IAAI,CAAC,UAAmC,CAAC,CAAA;QACpD,CAAC;QACD,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAA;IACnC,CAAC;SAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrC,OAAO,kBAAkB,CAAC,KAAK,CAAC,CAAA;IAClC,CAAC;IACD,OAAO,CAAC,CAAC,MAAM,EAAE,CAAA;AACnB,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,MAAoB;IACrD,MAAM,SAAS,GAAiC,EAAE,CAAA;IAElD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAClD,SAAS,CAAC,GAAG,CAAC,GAAG,YAAY,CAAC,KAAK,CAAC,CAAA;IACtC,CAAC;IAED,OAAO,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;AAC5B,CAAC"}

package/dist/core/responses.d.ts

@@ -0,0 +1,5 @@
+import type { GenerateTextResult, StreamingResult } from 'ai';
+export declare function createJsonResponse(result: GenerateTextResult): Response;
+export declare function createStreamResponse(result: StreamingResult): Response;
+export declare function createTextResponse(result: GenerateTextResult): Response;
+//# sourceMappingURL=responses.d.ts.map

package/dist/core/responses.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"responses.d.ts","sourceRoot":"","sources":["../../src/core/responses.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAyB,eAAe,EAAE,MAAM,IAAI,CAAA;AAEpF,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,kBAAkB,GAAG,QAAQ,CAIvE;AAED,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,eAAe,GAAG,QAAQ,CAItE;AAED,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,kBAAkB,GAAG,QAAQ,CAIvE"}

package/dist/core/responses.js

@@ -0,0 +1,16 @@
+export function createJsonResponse(result) {
+    return new Response(JSON.stringify(result), {
+        headers: { 'Content-Type': 'application/json' }
+    });
+}
+export function createStreamResponse(result) {
+    return new Response(result.experimental_stream, {
+        headers: { 'Content-Type': 'text/event-stream' }
+    });
+}
+export function createTextResponse(result) {
+    return new Response(result.text, {
+        headers: { 'Content-Type': 'text/plain' },
+    });
+}
+//# sourceMappingURL=responses.js.map

package/dist/core/responses.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"responses.js","sourceRoot":"","sources":["../../src/core/responses.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,kBAAkB,CAAC,MAA0B;IAC3D,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE;QAC1C,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;KAChD,CAAC,CAAA;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAAuB;IAC1D,OAAO,IAAI,QAAQ,CAAC,MAAM,CAAC,mBAAgD,EAAE;QAC3E,OAAO,EAAE,EAAE,cAAc,EAAE,mBAAmB,EAAE;KACjD,CAAC,CAAA;AACJ,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,MAA0B;IAC3D,OAAO,IAAI,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE;QAC/B,OAAO,EAAE,EAAE,cAAc,EAAE,YAAY,EAAE;KAC1C,CAAC,CAAA;AACJ,CAAC"}

package/dist/core/responses.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=responses.test.d.ts.map

package/dist/core/responses.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"responses.test.d.ts","sourceRoot":"","sources":["../../src/core/responses.test.ts"],"names":[],"mappings":""}

package/dist/core/responses.test.js

@@ -0,0 +1,31 @@
+import { describe, expect, it } from 'vitest';
+import { createJsonResponse, createStreamResponse, createTextResponse } from './responses';
+describe('Response Creation', () => {
+    it('should create JSON response', () => {
+        const mockResult = {
+            text: 'test content',
+            response: { id: 'test', model: 'test-model', created: 123 }
+        };
+        const response = createJsonResponse(mockResult);
+        expect(response).toBeInstanceOf(Response);
+        expect(response.headers.get('Content-Type')).toBe('application/json');
+    });
+    it('should create stream response', () => {
+        const mockResult = {
+            experimental_stream: new ReadableStream()
+        };
+        const response = createStreamResponse(mockResult);
+        expect(response).toBeInstanceOf(Response);
+        expect(response.headers.get('Content-Type')).toBe('text/event-stream');
+    });
+    it('should create text response', () => {
+        const mockResult = {
+            text: 'test content',
+            response: { id: 'test', model: 'test-model', created: 123 }
+        };
+        const response = createTextResponse(mockResult);
+        expect(response).toBeInstanceOf(Response);
+        expect(response.headers.get('Content-Type')).toBe('text/plain');
+    });
+});
+//# sourceMappingURL=responses.test.js.map

package/dist/core/responses.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"responses.test.js","sourceRoot":"","sources":["../../src/core/responses.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAC7C,OAAO,EAAE,kBAAkB,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAA;AAG1F,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,UAAU,GAAG;YACjB,IAAI,EAAE,cAAc;YACpB,QAAQ,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,GAAG,EAAE;SACtC,CAAA;QAEvB,MAAM,QAAQ,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAA;QAC/C,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAA;QACzC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAA;IACvE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,UAAU,GAAG;YACjB,mBAAmB,EAAE,IAAI,cAAc,EAAE;SACvB,CAAA;QAEpB,MAAM,QAAQ,GAAG,oBAAoB,CAAC,UAAU,CAAC,CAAA;QACjD,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAA;QACzC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAA;IACxE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,UAAU,GAAG;YACjB,IAAI,EAAE,cAAc;YACpB,QAAQ,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,GAAG,EAAE;SACtC,CAAA;QAEvB,MAAM,QAAQ,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAA;QAC/C,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAA;QACzC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;IACjE,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/errors.d.ts

@@ -0,0 +1,6 @@
+export declare class AIRequestError extends Error {
+    cause?: unknown | undefined;
+    retryable: boolean;
+    constructor(message: string, cause?: unknown | undefined, retryable?: boolean);
+}
+//# sourceMappingURL=errors.d.ts.map

package/dist/errors.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.d.ts","sourceRoot":"","sources":["../src/errors.ts"],"names":[],"mappings":"AAAA,qBAAa,cAAe,SAAQ,KAAK;IAG9B,KAAK,CAAC,EAAE,OAAO;IACf,SAAS;gBAFhB,OAAO,EAAE,MAAM,EACR,KAAK,CAAC,EAAE,OAAO,YAAA,EACf,SAAS,UAAO;CAK1B"}

package/dist/errors.js

@@ -0,0 +1,9 @@
+export class AIRequestError extends Error {
+    constructor(message, cause, retryable = true) {
+        super(message);
+        this.cause = cause;
+        this.retryable = retryable;
+        this.name = 'AIRequestError';
+    }
+}
+//# sourceMappingURL=errors.js.map

package/dist/errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","sourceRoot":"","sources":["../src/errors.ts"],"names":[],"mappings":"AAAA,MAAM,OAAO,cAAe,SAAQ,KAAK;IACvC,YACE,OAAe,EACR,KAAe,EACf,YAAY,IAAI;QAEvB,KAAK,CAAC,OAAO,CAAC,CAAA;QAHP,UAAK,GAAL,KAAK,CAAU;QACf,cAAS,GAAT,SAAS,CAAO;QAGvB,IAAI,CAAC,IAAI,GAAG,gBAAgB,CAAA;IAC9B,CAAC;CACF"}

package/dist/examples/streaming.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=streaming.test.d.ts.map

package/dist/examples/streaming.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming.test.d.ts","sourceRoot":"","sources":["../../src/examples/streaming.test.ts"],"names":[],"mappings":""}