ai-functions 0.3.0 → 0.4.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/.turbo/turbo-build.log +5 -0
- package/.turbo/turbo-test.log +105 -0
- package/README.md +190 -86
- package/TODO.md +138 -0
- package/dist/ai-promise.d.ts +219 -0
- package/dist/ai-promise.d.ts.map +1 -0
- package/dist/ai-promise.js +610 -0
- package/dist/ai-promise.js.map +1 -0
- package/dist/ai.d.ts +285 -0
- package/dist/ai.d.ts.map +1 -0
- package/dist/ai.js +842 -0
- package/dist/ai.js.map +1 -0
- package/dist/batch/anthropic.d.ts +23 -0
- package/dist/batch/anthropic.d.ts.map +1 -0
- package/dist/batch/anthropic.js +257 -0
- package/dist/batch/anthropic.js.map +1 -0
- package/dist/batch/bedrock.d.ts +64 -0
- package/dist/batch/bedrock.d.ts.map +1 -0
- package/dist/batch/bedrock.js +586 -0
- package/dist/batch/bedrock.js.map +1 -0
- package/dist/batch/cloudflare.d.ts +37 -0
- package/dist/batch/cloudflare.d.ts.map +1 -0
- package/dist/batch/cloudflare.js +289 -0
- package/dist/batch/cloudflare.js.map +1 -0
- package/dist/batch/google.d.ts +41 -0
- package/dist/batch/google.d.ts.map +1 -0
- package/dist/batch/google.js +360 -0
- package/dist/batch/google.js.map +1 -0
- package/dist/batch/index.d.ts +31 -0
- package/dist/batch/index.d.ts.map +1 -0
- package/dist/batch/index.js +31 -0
- package/dist/batch/index.js.map +1 -0
- package/dist/batch/memory.d.ts +44 -0
- package/dist/batch/memory.d.ts.map +1 -0
- package/dist/batch/memory.js +188 -0
- package/dist/batch/memory.js.map +1 -0
- package/dist/batch/openai.d.ts +37 -0
- package/dist/batch/openai.d.ts.map +1 -0
- package/dist/batch/openai.js +403 -0
- package/dist/batch/openai.js.map +1 -0
- package/dist/batch-map.d.ts +125 -0
- package/dist/batch-map.d.ts.map +1 -0
- package/dist/batch-map.js +406 -0
- package/dist/batch-map.js.map +1 -0
- package/dist/batch-queue.d.ts +273 -0
- package/dist/batch-queue.d.ts.map +1 -0
- package/dist/batch-queue.js +271 -0
- package/dist/batch-queue.js.map +1 -0
- package/dist/context.d.ts +133 -0
- package/dist/context.d.ts.map +1 -0
- package/dist/context.js +267 -0
- package/dist/context.js.map +1 -0
- package/dist/embeddings.d.ts +123 -0
- package/dist/embeddings.d.ts.map +1 -0
- package/dist/embeddings.js +170 -0
- package/dist/embeddings.js.map +1 -0
- package/dist/eval/index.d.ts +8 -0
- package/dist/eval/index.d.ts.map +1 -0
- package/dist/eval/index.js +8 -0
- package/dist/eval/index.js.map +1 -0
- package/dist/eval/models.d.ts +66 -0
- package/dist/eval/models.d.ts.map +1 -0
- package/dist/eval/models.js +120 -0
- package/dist/eval/models.js.map +1 -0
- package/dist/eval/runner.d.ts +64 -0
- package/dist/eval/runner.d.ts.map +1 -0
- package/dist/eval/runner.js +148 -0
- package/dist/eval/runner.js.map +1 -0
- package/dist/generate.d.ts +168 -0
- package/dist/generate.d.ts.map +1 -0
- package/dist/generate.js +174 -0
- package/dist/generate.js.map +1 -0
- package/dist/index.d.ts +29 -4
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +53 -52
- package/dist/index.js.map +1 -1
- package/dist/primitives.d.ts +292 -0
- package/dist/primitives.d.ts.map +1 -0
- package/dist/primitives.js +471 -0
- package/dist/primitives.js.map +1 -0
- package/dist/providers/cloudflare.d.ts +9 -0
- package/dist/providers/cloudflare.d.ts.map +1 -0
- package/dist/providers/cloudflare.js +9 -0
- package/dist/providers/cloudflare.js.map +1 -0
- package/dist/providers/index.d.ts +9 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +9 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/schema.d.ts +54 -0
- package/dist/schema.d.ts.map +1 -0
- package/dist/schema.js +109 -0
- package/dist/schema.js.map +1 -0
- package/dist/template.d.ts +73 -0
- package/dist/template.d.ts.map +1 -0
- package/dist/template.js +129 -0
- package/dist/template.js.map +1 -0
- package/dist/types.d.ts +474 -106
- package/dist/types.d.ts.map +1 -1
- package/dist/types.js +4 -8
- package/dist/types.js.map +1 -1
- package/evalite.config.ts +19 -0
- package/evals/README.md +212 -0
- package/evals/classification.eval.ts +108 -0
- package/evals/marketing.eval.ts +370 -0
- package/evals/math.eval.ts +94 -0
- package/evals/run-evals.ts +166 -0
- package/evals/structured-output.eval.ts +143 -0
- package/evals/writing.eval.ts +117 -0
- package/examples/batch-blog-posts.ts +160 -0
- package/package.json +57 -57
- package/src/ai-promise.ts +784 -0
- package/src/ai.ts +1183 -0
- package/src/batch/anthropic.ts +375 -0
- package/src/batch/bedrock.ts +801 -0
- package/src/batch/cloudflare.ts +421 -0
- package/src/batch/google.ts +491 -0
- package/src/batch/index.ts +31 -0
- package/src/batch/memory.ts +253 -0
- package/src/batch/openai.ts +557 -0
- package/src/batch-map.ts +534 -0
- package/src/batch-queue.ts +493 -0
- package/src/context.ts +332 -0
- package/src/embeddings.ts +244 -0
- package/src/eval/index.ts +8 -0
- package/src/eval/models.ts +158 -0
- package/src/eval/runner.ts +217 -0
- package/src/generate.ts +245 -0
- package/src/index.ts +154 -0
- package/src/primitives.ts +612 -0
- package/src/providers/cloudflare.ts +15 -0
- package/src/providers/index.ts +14 -0
- package/src/schema.ts +147 -0
- package/src/template.ts +209 -0
- package/src/types.ts +540 -0
- package/test/README.md +105 -0
- package/test/ai-proxy.test.ts +192 -0
- package/test/async-iterators.test.ts +327 -0
- package/test/batch-background.test.ts +482 -0
- package/test/batch-blog-posts.test.ts +387 -0
- package/test/blog-generation.test.ts +510 -0
- package/test/browse-read.test.ts +611 -0
- package/test/core-functions.test.ts +694 -0
- package/test/decide.test.ts +393 -0
- package/test/define.test.ts +274 -0
- package/test/e2e-bedrock-manual.ts +163 -0
- package/test/e2e-bedrock.test.ts +191 -0
- package/test/e2e-flex-gateway.ts +157 -0
- package/test/e2e-flex-manual.ts +183 -0
- package/test/e2e-flex.test.ts +209 -0
- package/test/e2e-google-manual.ts +178 -0
- package/test/e2e-google.test.ts +216 -0
- package/test/embeddings.test.ts +284 -0
- package/test/evals/define-function.eval.test.ts +379 -0
- package/test/evals/primitives.eval.test.ts +384 -0
- package/test/function-types.test.ts +492 -0
- package/test/generate-core.test.ts +319 -0
- package/test/generate.test.ts +163 -0
- package/test/implicit-batch.test.ts +422 -0
- package/test/schema.test.ts +109 -0
- package/test/tagged-templates.test.ts +302 -0
- package/tsconfig.json +10 -0
- package/vitest.config.ts +42 -0
- package/LICENSE +0 -21
- package/bin/cli.js +0 -5
- package/dist/cli/index.d.ts +0 -10
- package/dist/cli/index.d.ts.map +0 -1
- package/dist/cli/index.js +0 -38
- package/dist/cli/index.js.map +0 -1
- package/dist/cli/index.test.d.ts +0 -2
- package/dist/cli/index.test.d.ts.map +0 -1
- package/dist/cli/index.test.js +0 -35
- package/dist/cli/index.test.js.map +0 -1
- package/dist/constants/models.d.ts +0 -10
- package/dist/constants/models.d.ts.map +0 -1
- package/dist/constants/models.js +0 -12
- package/dist/constants/models.js.map +0 -1
- package/dist/converters/index.d.ts +0 -3
- package/dist/converters/index.d.ts.map +0 -1
- package/dist/converters/index.js +0 -3
- package/dist/converters/index.js.map +0 -1
- package/dist/converters/model.d.ts +0 -4
- package/dist/converters/model.d.ts.map +0 -1
- package/dist/converters/model.js +0 -19
- package/dist/converters/model.js.map +0 -1
- package/dist/converters/schema.d.ts +0 -4
- package/dist/converters/schema.d.ts.map +0 -1
- package/dist/converters/schema.js +0 -25
- package/dist/converters/schema.js.map +0 -1
- package/dist/core/responses.d.ts +0 -5
- package/dist/core/responses.d.ts.map +0 -1
- package/dist/core/responses.js +0 -16
- package/dist/core/responses.js.map +0 -1
- package/dist/core/responses.test.d.ts +0 -2
- package/dist/core/responses.test.d.ts.map +0 -1
- package/dist/core/responses.test.js +0 -31
- package/dist/core/responses.test.js.map +0 -1
- package/dist/errors.d.ts +0 -6
- package/dist/errors.d.ts.map +0 -1
- package/dist/errors.js +0 -9
- package/dist/errors.js.map +0 -1
- package/dist/examples/streaming.test.d.ts +0 -2
- package/dist/examples/streaming.test.d.ts.map +0 -1
- package/dist/examples/streaming.test.js +0 -176
- package/dist/examples/streaming.test.js.map +0 -1
- package/dist/factory/__tests__/index.test.d.ts +0 -2
- package/dist/factory/__tests__/index.test.d.ts.map +0 -1
- package/dist/factory/__tests__/index.test.js +0 -430
- package/dist/factory/__tests__/index.test.js.map +0 -1
- package/dist/factory/__tests__/list.test.d.ts +0 -2
- package/dist/factory/__tests__/list.test.d.ts.map +0 -1
- package/dist/factory/__tests__/list.test.js +0 -92
- package/dist/factory/__tests__/list.test.js.map +0 -1
- package/dist/factory/index.d.ts +0 -20
- package/dist/factory/index.d.ts.map +0 -1
- package/dist/factory/index.js +0 -287
- package/dist/factory/index.js.map +0 -1
- package/dist/factory/index.test.d.ts +0 -2
- package/dist/factory/index.test.d.ts.map +0 -1
- package/dist/factory/index.test.js +0 -287
- package/dist/factory/index.test.js.map +0 -1
- package/dist/factory/list.d.ts +0 -3
- package/dist/factory/list.d.ts.map +0 -1
- package/dist/factory/list.js +0 -221
- package/dist/factory/list.js.map +0 -1
- package/dist/factory/list.test.d.ts +0 -2
- package/dist/factory/list.test.d.ts.map +0 -1
- package/dist/factory/list.test.js +0 -84
- package/dist/factory/list.test.js.map +0 -1
- package/dist/generate/index.d.ts +0 -5
- package/dist/generate/index.d.ts.map +0 -1
- package/dist/generate/index.js +0 -17
- package/dist/generate/index.js.map +0 -1
- package/dist/index.test.d.ts +0 -2
- package/dist/index.test.d.ts.map +0 -1
- package/dist/index.test.js +0 -59
- package/dist/index.test.js.map +0 -1
- package/dist/list/await.d.ts +0 -3
- package/dist/list/await.d.ts.map +0 -1
- package/dist/list/await.js +0 -28
- package/dist/list/await.js.map +0 -1
- package/dist/list/constants.d.ts +0 -4
- package/dist/list/constants.d.ts.map +0 -1
- package/dist/list/constants.js +0 -5
- package/dist/list/constants.js.map +0 -1
- package/dist/list/create-function.d.ts +0 -3
- package/dist/list/create-function.d.ts.map +0 -1
- package/dist/list/create-function.js +0 -11
- package/dist/list/create-function.js.map +0 -1
- package/dist/list/index.d.ts +0 -4
- package/dist/list/index.d.ts.map +0 -1
- package/dist/list/index.js +0 -5
- package/dist/list/index.js.map +0 -1
- package/dist/list/prompt.d.ts +0 -3
- package/dist/list/prompt.d.ts.map +0 -1
- package/dist/list/prompt.js +0 -6
- package/dist/list/prompt.js.map +0 -1
- package/dist/list/schemas.d.ts +0 -4
- package/dist/list/schemas.d.ts.map +0 -1
- package/dist/list/schemas.js +0 -8
- package/dist/list/schemas.js.map +0 -1
- package/dist/list/stream.d.ts +0 -3
- package/dist/list/stream.d.ts.map +0 -1
- package/dist/list/stream.js +0 -33
- package/dist/list/stream.js.map +0 -1
- package/dist/list/types.d.ts +0 -11
- package/dist/list/types.d.ts.map +0 -1
- package/dist/list/types.js +0 -2
- package/dist/list/types.js.map +0 -1
- package/dist/list/validation.d.ts +0 -3
- package/dist/list/validation.d.ts.map +0 -1
- package/dist/list/validation.js +0 -12
- package/dist/list/validation.js.map +0 -1
- package/dist/providers/config.d.ts +0 -4
- package/dist/providers/config.d.ts.map +0 -1
- package/dist/providers/config.js +0 -21
- package/dist/providers/config.js.map +0 -1
- package/dist/providers/config.test.d.ts +0 -2
- package/dist/providers/config.test.d.ts.map +0 -1
- package/dist/providers/config.test.js +0 -37
- package/dist/providers/config.test.js.map +0 -1
- package/dist/proxy/constants.d.ts +0 -4
- package/dist/proxy/constants.d.ts.map +0 -1
- package/dist/proxy/constants.js +0 -5
- package/dist/proxy/constants.js.map +0 -1
- package/dist/proxy/create-function.d.ts +0 -4
- package/dist/proxy/create-function.d.ts.map +0 -1
- package/dist/proxy/create-function.js +0 -24
- package/dist/proxy/create-function.js.map +0 -1
- package/dist/proxy/create-proxy.d.ts +0 -2
- package/dist/proxy/create-proxy.d.ts.map +0 -1
- package/dist/proxy/create-proxy.js +0 -11
- package/dist/proxy/create-proxy.js.map +0 -1
- package/dist/proxy/function-generator.d.ts +0 -9
- package/dist/proxy/function-generator.d.ts.map +0 -1
- package/dist/proxy/function-generator.js +0 -29
- package/dist/proxy/function-generator.js.map +0 -1
- package/dist/proxy/index.d.ts +0 -4
- package/dist/proxy/index.d.ts.map +0 -1
- package/dist/proxy/index.js +0 -4
- package/dist/proxy/index.js.map +0 -1
- package/dist/proxy/prompt.d.ts +0 -2
- package/dist/proxy/prompt.d.ts.map +0 -1
- package/dist/proxy/prompt.js +0 -6
- package/dist/proxy/prompt.js.map +0 -1
- package/dist/proxy/types.d.ts +0 -7
- package/dist/proxy/types.d.ts.map +0 -1
- package/dist/proxy/types.js +0 -2
- package/dist/proxy/types.js.map +0 -1
- package/dist/queue/manager.d.ts +0 -5
- package/dist/queue/manager.d.ts.map +0 -1
- package/dist/queue/manager.js +0 -37
- package/dist/queue/manager.js.map +0 -1
- package/dist/queue/manager.test.d.ts +0 -2
- package/dist/queue/manager.test.d.ts.map +0 -1
- package/dist/queue/manager.test.js +0 -52
- package/dist/queue/manager.test.js.map +0 -1
- package/dist/schema-converter.d.ts +0 -4
- package/dist/schema-converter.d.ts.map +0 -1
- package/dist/schema-converter.js +0 -30
- package/dist/schema-converter.js.map +0 -1
- package/dist/stream/index.d.ts +0 -7
- package/dist/stream/index.d.ts.map +0 -1
- package/dist/stream/index.js +0 -23
- package/dist/stream/index.js.map +0 -1
- package/dist/streaming/utils.d.ts +0 -4
- package/dist/streaming/utils.d.ts.map +0 -1
- package/dist/streaming/utils.js +0 -131
- package/dist/streaming/utils.js.map +0 -1
- package/dist/streaming/utils.test.d.ts +0 -2
- package/dist/streaming/utils.test.d.ts.map +0 -1
- package/dist/streaming/utils.test.js +0 -84
- package/dist/streaming/utils.test.js.map +0 -1
- package/dist/templates/result.d.ts +0 -7
- package/dist/templates/result.d.ts.map +0 -1
- package/dist/templates/result.js +0 -40
- package/dist/templates/result.js.map +0 -1
- package/dist/templates/result.test.d.ts +0 -2
- package/dist/templates/result.test.d.ts.map +0 -1
- package/dist/templates/result.test.js +0 -75
- package/dist/templates/result.test.js.map +0 -1
- package/dist/test/setup.d.ts +0 -2
- package/dist/test/setup.d.ts.map +0 -1
- package/dist/test/setup.js +0 -21
- package/dist/test/setup.js.map +0 -1
- package/dist/test-types.d.ts +0 -13
- package/dist/test-types.d.ts.map +0 -1
- package/dist/test-types.js +0 -55
- package/dist/test-types.js.map +0 -1
- package/dist/types/index.d.ts +0 -4
- package/dist/types/index.d.ts.map +0 -1
- package/dist/types/index.js +0 -4
- package/dist/types/index.js.map +0 -1
- package/dist/types/list.d.ts +0 -10
- package/dist/types/list.d.ts.map +0 -1
- package/dist/types/list.js +0 -2
- package/dist/types/list.js.map +0 -1
- package/dist/types/model.d.ts +0 -7
- package/dist/types/model.d.ts.map +0 -1
- package/dist/types/model.js +0 -2
- package/dist/types/model.js.map +0 -1
- package/dist/types/options.d.ts +0 -25
- package/dist/types/options.d.ts.map +0 -1
- package/dist/types/options.js +0 -2
- package/dist/types/options.js.map +0 -1
- package/dist/types/schema.d.ts +0 -5
- package/dist/types/schema.d.ts.map +0 -1
- package/dist/types/schema.js +0 -2
- package/dist/types/schema.js.map +0 -1
- package/dist/utils/__tests__/request-handler.test.d.ts +0 -2
- package/dist/utils/__tests__/request-handler.test.d.ts.map +0 -1
- package/dist/utils/__tests__/request-handler.test.js +0 -134
- package/dist/utils/__tests__/request-handler.test.js.map +0 -1
- package/dist/utils/__tests__/schema.test.d.ts +0 -2
- package/dist/utils/__tests__/schema.test.d.ts.map +0 -1
- package/dist/utils/__tests__/schema.test.js +0 -49
- package/dist/utils/__tests__/schema.test.js.map +0 -1
- package/dist/utils/__tests__/stream-progress.test.d.ts +0 -2
- package/dist/utils/__tests__/stream-progress.test.d.ts.map +0 -1
- package/dist/utils/__tests__/stream-progress.test.js +0 -85
- package/dist/utils/__tests__/stream-progress.test.js.map +0 -1
- package/dist/utils/index.d.ts +0 -2
- package/dist/utils/index.d.ts.map +0 -1
- package/dist/utils/index.js +0 -2
- package/dist/utils/index.js.map +0 -1
- package/dist/utils/request-handler.d.ts +0 -17
- package/dist/utils/request-handler.d.ts.map +0 -1
- package/dist/utils/request-handler.js +0 -105
- package/dist/utils/request-handler.js.map +0 -1
- package/dist/utils/schema.d.ts +0 -11
- package/dist/utils/schema.d.ts.map +0 -1
- package/dist/utils/schema.js +0 -51
- package/dist/utils/schema.js.map +0 -1
- package/dist/utils/stream-progress.d.ts +0 -17
- package/dist/utils/stream-progress.d.ts.map +0 -1
- package/dist/utils/stream-progress.js +0 -86
- package/dist/utils/stream-progress.js.map +0 -1
- package/dist/utils/validation.d.ts +0 -3
- package/dist/utils/validation.d.ts.map +0 -1
- package/dist/utils/validation.js +0 -30
- package/dist/utils/validation.js.map +0 -1
|
@@ -0,0 +1,302 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Tests for tagged template literal support
|
|
3
|
+
*
|
|
4
|
+
* Every function should support:
|
|
5
|
+
* 1. Tagged template syntax: fn`prompt ${variable}`
|
|
6
|
+
* 2. Objects/arrays auto-convert to YAML in templates
|
|
7
|
+
* 3. Options chaining: fn`prompt`({ model: '...' })
|
|
8
|
+
*/
|
|
9
|
+
|
|
10
|
+
import { describe, it, expect } from 'vitest'
|
|
11
|
+
import { stringify } from 'yaml'
|
|
12
|
+
|
|
13
|
+
// ============================================================================
|
|
14
|
+
// Template Parsing Tests (Unit Tests - No AI calls)
|
|
15
|
+
// ============================================================================
|
|
16
|
+
|
|
17
|
+
/**
|
|
18
|
+
* Parse a tagged template literal into a prompt string
|
|
19
|
+
* Objects and arrays are converted to YAML for readability
|
|
20
|
+
*/
|
|
21
|
+
function parseTemplate(strings: TemplateStringsArray, ...values: unknown[]): string {
|
|
22
|
+
return strings.reduce((result, str, i) => {
|
|
23
|
+
const value = values[i]
|
|
24
|
+
if (value === undefined) {
|
|
25
|
+
return result + str
|
|
26
|
+
}
|
|
27
|
+
|
|
28
|
+
// Convert objects/arrays to YAML
|
|
29
|
+
if (typeof value === 'object' && value !== null) {
|
|
30
|
+
const yaml = stringify(value).trim()
|
|
31
|
+
return result + str + '\n' + yaml
|
|
32
|
+
}
|
|
33
|
+
|
|
34
|
+
// Primitives use toString()
|
|
35
|
+
return result + str + String(value)
|
|
36
|
+
}, '')
|
|
37
|
+
}
|
|
38
|
+
|
|
39
|
+
/**
|
|
40
|
+
* Create a template function that supports both call styles
|
|
41
|
+
*/
|
|
42
|
+
function createTemplateFunction<T>(
|
|
43
|
+
handler: (prompt: string, options?: Record<string, unknown>) => Promise<T>
|
|
44
|
+
): {
|
|
45
|
+
(strings: TemplateStringsArray, ...values: unknown[]): Promise<T> & { (options?: Record<string, unknown>): Promise<T> }
|
|
46
|
+
(prompt: string, options?: Record<string, unknown>): Promise<T>
|
|
47
|
+
} {
|
|
48
|
+
function templateFn(promptOrStrings: string | TemplateStringsArray, ...args: unknown[]) {
|
|
49
|
+
// Tagged template literal
|
|
50
|
+
if (Array.isArray(promptOrStrings) && 'raw' in promptOrStrings) {
|
|
51
|
+
const prompt = parseTemplate(promptOrStrings as TemplateStringsArray, ...args)
|
|
52
|
+
|
|
53
|
+
// Return a thenable that also accepts options
|
|
54
|
+
const promise = handler(prompt)
|
|
55
|
+
|
|
56
|
+
// Allow chaining with options: fn`prompt`({ model: '...' })
|
|
57
|
+
// Create a function that also implements then/catch for Promise interface
|
|
58
|
+
const chainable = (options?: Record<string, unknown>) => handler(prompt, options)
|
|
59
|
+
|
|
60
|
+
// Make it thenable by adding then/catch from the original promise
|
|
61
|
+
;(chainable as unknown as Promise<T>).then = promise.then.bind(promise)
|
|
62
|
+
;(chainable as unknown as Promise<T>).catch = promise.catch.bind(promise)
|
|
63
|
+
|
|
64
|
+
return chainable
|
|
65
|
+
}
|
|
66
|
+
|
|
67
|
+
// Regular function call
|
|
68
|
+
return handler(promptOrStrings as string, args[0] as Record<string, unknown> | undefined)
|
|
69
|
+
}
|
|
70
|
+
|
|
71
|
+
return templateFn as ReturnType<typeof createTemplateFunction<T>>
|
|
72
|
+
}
|
|
73
|
+
|
|
74
|
+
describe('template parsing', () => {
|
|
75
|
+
it('parses simple string interpolation', () => {
|
|
76
|
+
const topic = 'TypeScript'
|
|
77
|
+
const result = parseTemplate`Write about ${topic}`
|
|
78
|
+
expect(result).toBe('Write about TypeScript')
|
|
79
|
+
})
|
|
80
|
+
|
|
81
|
+
it('parses multiple interpolations', () => {
|
|
82
|
+
const topic = 'TypeScript'
|
|
83
|
+
const audience = 'beginners'
|
|
84
|
+
const result = parseTemplate`Write about ${topic} for ${audience}`
|
|
85
|
+
expect(result).toBe('Write about TypeScript for beginners')
|
|
86
|
+
})
|
|
87
|
+
|
|
88
|
+
it('converts objects to YAML', () => {
|
|
89
|
+
const context = { topic: 'TypeScript', level: 'beginner' }
|
|
90
|
+
const result = parseTemplate`Write a post about ${{ context }}`
|
|
91
|
+
|
|
92
|
+
expect(result).toContain('context:')
|
|
93
|
+
expect(result).toContain('topic: TypeScript')
|
|
94
|
+
expect(result).toContain('level: beginner')
|
|
95
|
+
})
|
|
96
|
+
|
|
97
|
+
it('converts arrays to YAML lists', () => {
|
|
98
|
+
const topics = ['React', 'Vue', 'Angular']
|
|
99
|
+
const result = parseTemplate`Compare ${topics}`
|
|
100
|
+
|
|
101
|
+
expect(result).toContain('- React')
|
|
102
|
+
expect(result).toContain('- Vue')
|
|
103
|
+
expect(result).toContain('- Angular')
|
|
104
|
+
})
|
|
105
|
+
|
|
106
|
+
it('handles nested objects', () => {
|
|
107
|
+
const brand = {
|
|
108
|
+
hero: 'developers',
|
|
109
|
+
problem: {
|
|
110
|
+
internal: 'complexity',
|
|
111
|
+
external: 'time constraints',
|
|
112
|
+
},
|
|
113
|
+
}
|
|
114
|
+
const result = parseTemplate`Create a story for ${{ brand }}`
|
|
115
|
+
|
|
116
|
+
expect(result).toContain('brand:')
|
|
117
|
+
expect(result).toContain('hero: developers')
|
|
118
|
+
expect(result).toContain('problem:')
|
|
119
|
+
expect(result).toContain('internal: complexity')
|
|
120
|
+
expect(result).toContain('external: time constraints')
|
|
121
|
+
})
|
|
122
|
+
|
|
123
|
+
it('handles numbers and booleans', () => {
|
|
124
|
+
const count = 10
|
|
125
|
+
const active = true
|
|
126
|
+
const result = parseTemplate`Generate ${count} items, active: ${active}`
|
|
127
|
+
|
|
128
|
+
expect(result).toBe('Generate 10 items, active: true')
|
|
129
|
+
})
|
|
130
|
+
|
|
131
|
+
it('handles null and undefined gracefully', () => {
|
|
132
|
+
const value = null
|
|
133
|
+
const result = parseTemplate`Value is ${value}`
|
|
134
|
+
expect(result).toContain('Value is')
|
|
135
|
+
})
|
|
136
|
+
|
|
137
|
+
it('preserves template structure with objects inline', () => {
|
|
138
|
+
const requirements = {
|
|
139
|
+
pages: ['home', 'about', 'contact'],
|
|
140
|
+
features: ['dark mode', 'responsive'],
|
|
141
|
+
}
|
|
142
|
+
const result = parseTemplate`marketing site${{ requirements }}`
|
|
143
|
+
|
|
144
|
+
expect(result).toContain('marketing site')
|
|
145
|
+
expect(result).toContain('requirements:')
|
|
146
|
+
expect(result).toContain('pages:')
|
|
147
|
+
expect(result).toContain('- home')
|
|
148
|
+
expect(result).toContain('features:')
|
|
149
|
+
expect(result).toContain('- dark mode')
|
|
150
|
+
})
|
|
151
|
+
})
|
|
152
|
+
|
|
153
|
+
describe('template function creation', () => {
|
|
154
|
+
it('creates a function that handles tagged templates', async () => {
|
|
155
|
+
const mockHandler = async (prompt: string) => `Processed: ${prompt}`
|
|
156
|
+
const fn = createTemplateFunction(mockHandler)
|
|
157
|
+
|
|
158
|
+
const result = await fn`Hello ${'world'}`
|
|
159
|
+
expect(result).toBe('Processed: Hello world')
|
|
160
|
+
})
|
|
161
|
+
|
|
162
|
+
it('creates a function that handles regular calls', async () => {
|
|
163
|
+
const mockHandler = async (prompt: string) => `Processed: ${prompt}`
|
|
164
|
+
const fn = createTemplateFunction(mockHandler)
|
|
165
|
+
|
|
166
|
+
const result = await fn('Hello world')
|
|
167
|
+
expect(result).toBe('Processed: Hello world')
|
|
168
|
+
})
|
|
169
|
+
|
|
170
|
+
it('supports options chaining on tagged templates', async () => {
|
|
171
|
+
let capturedOptions: Record<string, unknown> | undefined
|
|
172
|
+
const mockHandler = async (prompt: string, options?: Record<string, unknown>) => {
|
|
173
|
+
capturedOptions = options
|
|
174
|
+
return `Processed: ${prompt}`
|
|
175
|
+
}
|
|
176
|
+
const fn = createTemplateFunction(mockHandler)
|
|
177
|
+
|
|
178
|
+
const result = await fn`Hello world`({ model: 'claude-opus-4-5' })
|
|
179
|
+
|
|
180
|
+
expect(result).toBe('Processed: Hello world')
|
|
181
|
+
expect(capturedOptions).toEqual({ model: 'claude-opus-4-5' })
|
|
182
|
+
})
|
|
183
|
+
|
|
184
|
+
it('passes options in regular calls', async () => {
|
|
185
|
+
let capturedOptions: Record<string, unknown> | undefined
|
|
186
|
+
const mockHandler = async (prompt: string, options?: Record<string, unknown>) => {
|
|
187
|
+
capturedOptions = options
|
|
188
|
+
return `Processed: ${prompt}`
|
|
189
|
+
}
|
|
190
|
+
const fn = createTemplateFunction(mockHandler)
|
|
191
|
+
|
|
192
|
+
await fn('Hello world', { model: 'gpt-5-1', temperature: 0.7 })
|
|
193
|
+
|
|
194
|
+
expect(capturedOptions).toEqual({ model: 'gpt-5-1', temperature: 0.7 })
|
|
195
|
+
})
|
|
196
|
+
|
|
197
|
+
it('converts complex objects to YAML in templates', async () => {
|
|
198
|
+
let capturedPrompt = ''
|
|
199
|
+
const mockHandler = async (prompt: string) => {
|
|
200
|
+
capturedPrompt = prompt
|
|
201
|
+
return 'ok'
|
|
202
|
+
}
|
|
203
|
+
const fn = createTemplateFunction(mockHandler)
|
|
204
|
+
|
|
205
|
+
const brand = { hero: 'developers', guide: 'ai-functions' }
|
|
206
|
+
await fn`Create story for ${{ brand }}`
|
|
207
|
+
|
|
208
|
+
expect(capturedPrompt).toContain('brand:')
|
|
209
|
+
expect(capturedPrompt).toContain('hero: developers')
|
|
210
|
+
expect(capturedPrompt).toContain('guide: ai-functions')
|
|
211
|
+
})
|
|
212
|
+
})
|
|
213
|
+
|
|
214
|
+
describe('options parameter', () => {
|
|
215
|
+
it('supports model option', async () => {
|
|
216
|
+
let capturedOptions: Record<string, unknown> | undefined
|
|
217
|
+
const mockHandler = async (_prompt: string, options?: Record<string, unknown>) => {
|
|
218
|
+
capturedOptions = options
|
|
219
|
+
return 'ok'
|
|
220
|
+
}
|
|
221
|
+
const fn = createTemplateFunction(mockHandler)
|
|
222
|
+
|
|
223
|
+
await fn`test`({ model: 'claude-opus-4-5' })
|
|
224
|
+
|
|
225
|
+
expect(capturedOptions?.model).toBe('claude-opus-4-5')
|
|
226
|
+
})
|
|
227
|
+
|
|
228
|
+
it('supports thinking option', async () => {
|
|
229
|
+
let capturedOptions: Record<string, unknown> | undefined
|
|
230
|
+
const mockHandler = async (_prompt: string, options?: Record<string, unknown>) => {
|
|
231
|
+
capturedOptions = options
|
|
232
|
+
return 'ok'
|
|
233
|
+
}
|
|
234
|
+
const fn = createTemplateFunction(mockHandler)
|
|
235
|
+
|
|
236
|
+
await fn`complex analysis`({ thinking: 'high' })
|
|
237
|
+
|
|
238
|
+
expect(capturedOptions?.thinking).toBe('high')
|
|
239
|
+
})
|
|
240
|
+
|
|
241
|
+
it('supports thinking as token budget number', async () => {
|
|
242
|
+
let capturedOptions: Record<string, unknown> | undefined
|
|
243
|
+
const mockHandler = async (_prompt: string, options?: Record<string, unknown>) => {
|
|
244
|
+
capturedOptions = options
|
|
245
|
+
return 'ok'
|
|
246
|
+
}
|
|
247
|
+
const fn = createTemplateFunction(mockHandler)
|
|
248
|
+
|
|
249
|
+
await fn`complex analysis`({ thinking: 10000 })
|
|
250
|
+
|
|
251
|
+
expect(capturedOptions?.thinking).toBe(10000)
|
|
252
|
+
})
|
|
253
|
+
|
|
254
|
+
it('supports temperature option', async () => {
|
|
255
|
+
let capturedOptions: Record<string, unknown> | undefined
|
|
256
|
+
const mockHandler = async (_prompt: string, options?: Record<string, unknown>) => {
|
|
257
|
+
capturedOptions = options
|
|
258
|
+
return 'ok'
|
|
259
|
+
}
|
|
260
|
+
const fn = createTemplateFunction(mockHandler)
|
|
261
|
+
|
|
262
|
+
await fn`creative writing`({ temperature: 0.9 })
|
|
263
|
+
|
|
264
|
+
expect(capturedOptions?.temperature).toBe(0.9)
|
|
265
|
+
})
|
|
266
|
+
|
|
267
|
+
it('supports maxTokens option', async () => {
|
|
268
|
+
let capturedOptions: Record<string, unknown> | undefined
|
|
269
|
+
const mockHandler = async (_prompt: string, options?: Record<string, unknown>) => {
|
|
270
|
+
capturedOptions = options
|
|
271
|
+
return 'ok'
|
|
272
|
+
}
|
|
273
|
+
const fn = createTemplateFunction(mockHandler)
|
|
274
|
+
|
|
275
|
+
await fn`long article`({ maxTokens: 4000 })
|
|
276
|
+
|
|
277
|
+
expect(capturedOptions?.maxTokens).toBe(4000)
|
|
278
|
+
})
|
|
279
|
+
|
|
280
|
+
it('supports multiple options together', async () => {
|
|
281
|
+
let capturedOptions: Record<string, unknown> | undefined
|
|
282
|
+
const mockHandler = async (_prompt: string, options?: Record<string, unknown>) => {
|
|
283
|
+
capturedOptions = options
|
|
284
|
+
return 'ok'
|
|
285
|
+
}
|
|
286
|
+
const fn = createTemplateFunction(mockHandler)
|
|
287
|
+
|
|
288
|
+
await fn`complex task`({
|
|
289
|
+
model: 'claude-opus-4-5',
|
|
290
|
+
thinking: 'high',
|
|
291
|
+
temperature: 0.7,
|
|
292
|
+
maxTokens: 8000,
|
|
293
|
+
})
|
|
294
|
+
|
|
295
|
+
expect(capturedOptions).toEqual({
|
|
296
|
+
model: 'claude-opus-4-5',
|
|
297
|
+
thinking: 'high',
|
|
298
|
+
temperature: 0.7,
|
|
299
|
+
maxTokens: 8000,
|
|
300
|
+
})
|
|
301
|
+
})
|
|
302
|
+
})
|
package/tsconfig.json
ADDED
package/vitest.config.ts
ADDED
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
import { defineConfig } from 'vitest/config'
|
|
2
|
+
import { config } from 'dotenv'
|
|
3
|
+
import { resolve } from 'path'
|
|
4
|
+
import { existsSync } from 'fs'
|
|
5
|
+
|
|
6
|
+
// Load .env from current directory, parent directories, and root
|
|
7
|
+
// This supports primitives being used as a submodule
|
|
8
|
+
const envPaths = [
|
|
9
|
+
resolve(process.cwd(), '.env'),
|
|
10
|
+
resolve(process.cwd(), '..', '.env'),
|
|
11
|
+
resolve(process.cwd(), '..', '..', '.env'),
|
|
12
|
+
resolve(process.cwd(), '..', '..', '..', '.env'),
|
|
13
|
+
]
|
|
14
|
+
|
|
15
|
+
for (const envPath of envPaths) {
|
|
16
|
+
if (existsSync(envPath)) {
|
|
17
|
+
config({ path: envPath })
|
|
18
|
+
}
|
|
19
|
+
}
|
|
20
|
+
|
|
21
|
+
export default defineConfig({
|
|
22
|
+
test: {
|
|
23
|
+
globals: false,
|
|
24
|
+
environment: 'node',
|
|
25
|
+
include: ['test/**/*.test.ts'],
|
|
26
|
+
// Exclude runtime-specific tests (use separate configs)
|
|
27
|
+
exclude: [
|
|
28
|
+
'test/rpc/workers/**',
|
|
29
|
+
'test/rpc/bun/**',
|
|
30
|
+
'node_modules/**',
|
|
31
|
+
],
|
|
32
|
+
testTimeout: 60000, // AI calls can take time
|
|
33
|
+
hookTimeout: 30000,
|
|
34
|
+
// Run tests sequentially to avoid rate limiting
|
|
35
|
+
pool: 'forks',
|
|
36
|
+
poolOptions: {
|
|
37
|
+
forks: {
|
|
38
|
+
singleFork: true,
|
|
39
|
+
},
|
|
40
|
+
},
|
|
41
|
+
},
|
|
42
|
+
})
|
package/LICENSE
DELETED
|
@@ -1,21 +0,0 @@
|
|
|
1
|
-
MIT License
|
|
2
|
-
|
|
3
|
-
Copyright (c) 2024 AI Primitives
|
|
4
|
-
|
|
5
|
-
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
6
|
-
of this software and associated documentation files (the "Software"), to deal
|
|
7
|
-
in the Software without restriction, including without limitation the rights
|
|
8
|
-
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
9
|
-
copies of the Software, and to permit persons to whom the Software is
|
|
10
|
-
furnished to do so, subject to the following conditions:
|
|
11
|
-
|
|
12
|
-
The above copyright notice and this permission notice shall be included in all
|
|
13
|
-
copies or substantial portions of the Software.
|
|
14
|
-
|
|
15
|
-
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
16
|
-
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
17
|
-
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
18
|
-
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
19
|
-
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
20
|
-
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
21
|
-
SOFTWARE.
|
package/bin/cli.js
DELETED
package/dist/cli/index.d.ts
DELETED
|
@@ -1,10 +0,0 @@
|
|
|
1
|
-
interface CliOptions {
|
|
2
|
-
version?: boolean;
|
|
3
|
-
help?: boolean;
|
|
4
|
-
}
|
|
5
|
-
export declare function parseArgs(args: string[]): CliOptions;
|
|
6
|
-
export declare function showHelp(): void;
|
|
7
|
-
export declare function showVersion(): void;
|
|
8
|
-
export declare function cli(args?: string[]): void;
|
|
9
|
-
export {};
|
|
10
|
-
//# sourceMappingURL=index.d.ts.map
|
package/dist/cli/index.d.ts.map
DELETED
|
@@ -1 +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
DELETED
|
@@ -1,38 +0,0 @@
|
|
|
1
|
-
import { version } from '../../package.json';
|
|
2
|
-
export function parseArgs(args) {
|
|
3
|
-
const options = {};
|
|
4
|
-
for (const arg of args) {
|
|
5
|
-
if (arg === '-v' || arg === '--version') {
|
|
6
|
-
options.version = true;
|
|
7
|
-
}
|
|
8
|
-
else if (arg === '-h' || arg === '--help') {
|
|
9
|
-
options.help = true;
|
|
10
|
-
}
|
|
11
|
-
}
|
|
12
|
-
return options;
|
|
13
|
-
}
|
|
14
|
-
export function showHelp() {
|
|
15
|
-
console.log(`
|
|
16
|
-
Usage: ai-functions [options]
|
|
17
|
-
|
|
18
|
-
Options:
|
|
19
|
-
-v, --version Show version number
|
|
20
|
-
-h, --help Show help
|
|
21
|
-
`);
|
|
22
|
-
}
|
|
23
|
-
export function showVersion() {
|
|
24
|
-
console.log(`v${version}`);
|
|
25
|
-
}
|
|
26
|
-
export function cli(args = process.argv.slice(2)) {
|
|
27
|
-
const options = parseArgs(args);
|
|
28
|
-
if (options.version) {
|
|
29
|
-
showVersion();
|
|
30
|
-
}
|
|
31
|
-
else if (options.help) {
|
|
32
|
-
showHelp();
|
|
33
|
-
}
|
|
34
|
-
else {
|
|
35
|
-
showHelp();
|
|
36
|
-
}
|
|
37
|
-
}
|
|
38
|
-
//# sourceMappingURL=index.js.map
|
package/dist/cli/index.js.map
DELETED
|
@@ -1 +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
DELETED
|
@@ -1 +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
DELETED
|
@@ -1,35 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, vi } from 'vitest';
|
|
2
|
-
import { parseArgs, showHelp, showVersion, cli } from './index';
|
|
3
|
-
describe('CLI', () => {
|
|
4
|
-
it('should parse version flag correctly', () => {
|
|
5
|
-
const options = parseArgs(['-v']);
|
|
6
|
-
expect(options.version).toBe(true);
|
|
7
|
-
const longOptions = parseArgs(['--version']);
|
|
8
|
-
expect(longOptions.version).toBe(true);
|
|
9
|
-
});
|
|
10
|
-
it('should parse help flag correctly', () => {
|
|
11
|
-
const options = parseArgs(['-h']);
|
|
12
|
-
expect(options.help).toBe(true);
|
|
13
|
-
const longOptions = parseArgs(['--help']);
|
|
14
|
-
expect(longOptions.help).toBe(true);
|
|
15
|
-
});
|
|
16
|
-
it('should show version number', () => {
|
|
17
|
-
const consoleSpy = vi.spyOn(console, 'log');
|
|
18
|
-
showVersion();
|
|
19
|
-
expect(consoleSpy).toHaveBeenCalled();
|
|
20
|
-
consoleSpy.mockRestore();
|
|
21
|
-
});
|
|
22
|
-
it('should show help message', () => {
|
|
23
|
-
const consoleSpy = vi.spyOn(console, 'log');
|
|
24
|
-
showHelp();
|
|
25
|
-
expect(consoleSpy).toHaveBeenCalled();
|
|
26
|
-
consoleSpy.mockRestore();
|
|
27
|
-
});
|
|
28
|
-
it('should default to help when no arguments provided', () => {
|
|
29
|
-
const consoleSpy = vi.spyOn(console, 'log');
|
|
30
|
-
cli([]);
|
|
31
|
-
expect(consoleSpy).toHaveBeenCalled();
|
|
32
|
-
consoleSpy.mockRestore();
|
|
33
|
-
});
|
|
34
|
-
});
|
|
35
|
-
//# sourceMappingURL=index.test.js.map
|
|
@@ -1 +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"}
|
|
@@ -1,10 +0,0 @@
|
|
|
1
|
-
export declare const MODEL_BEST = "gpt-4o";
|
|
2
|
-
export declare const MODEL_FAST = "gpt-4o-mini";
|
|
3
|
-
export declare const DEFAULT_MODEL = "gpt-4o";
|
|
4
|
-
export declare const FALLBACK_MODEL = "gpt-4o-mini";
|
|
5
|
-
export declare const MODELS: {
|
|
6
|
-
readonly BEST: "gpt-4o";
|
|
7
|
-
readonly FAST: "gpt-4o-mini";
|
|
8
|
-
};
|
|
9
|
-
export type ModelType = typeof MODELS[keyof typeof MODELS];
|
|
10
|
-
//# sourceMappingURL=models.d.ts.map
|
|
@@ -1 +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
DELETED
|
@@ -1,12 +0,0 @@
|
|
|
1
|
-
// Model identifiers
|
|
2
|
-
export const MODEL_BEST = 'gpt-4o';
|
|
3
|
-
export const MODEL_FAST = 'gpt-4o-mini';
|
|
4
|
-
// Default model configuration
|
|
5
|
-
export const DEFAULT_MODEL = MODEL_BEST;
|
|
6
|
-
export const FALLBACK_MODEL = MODEL_FAST;
|
|
7
|
-
// Model constants for external use
|
|
8
|
-
export const MODELS = {
|
|
9
|
-
BEST: MODEL_BEST,
|
|
10
|
-
FAST: MODEL_FAST
|
|
11
|
-
};
|
|
12
|
-
//# sourceMappingURL=models.js.map
|
|
@@ -1 +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"}
|
|
@@ -1 +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
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/converters/index.ts"],"names":[],"mappings":"AAAA,cAAc,SAAS,CAAA;AACvB,cAAc,UAAU,CAAA"}
|
|
@@ -1 +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
DELETED
|
@@ -1,19 +0,0 @@
|
|
|
1
|
-
import { createOpenAI } from '@ai-sdk/openai';
|
|
2
|
-
// Create a singleton OpenAI client
|
|
3
|
-
let clientInstance;
|
|
4
|
-
export function getOpenAIClient() {
|
|
5
|
-
if (!clientInstance) {
|
|
6
|
-
clientInstance = createOpenAI({
|
|
7
|
-
baseURL: process.env.AI_GATEWAY,
|
|
8
|
-
});
|
|
9
|
-
}
|
|
10
|
-
return clientInstance;
|
|
11
|
-
}
|
|
12
|
-
export function convertToOpenAIModel(model) {
|
|
13
|
-
if (typeof model === 'string') {
|
|
14
|
-
const client = getOpenAIClient();
|
|
15
|
-
return client(model);
|
|
16
|
-
}
|
|
17
|
-
return model;
|
|
18
|
-
}
|
|
19
|
-
//# sourceMappingURL=model.js.map
|
|
@@ -1 +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"}
|
|
@@ -1 +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"}
|
|
@@ -1,25 +0,0 @@
|
|
|
1
|
-
import { z } from 'zod';
|
|
2
|
-
function convertValue(value) {
|
|
3
|
-
if (Array.isArray(value)) {
|
|
4
|
-
return z.array(z.string().describe(value[0] || ''));
|
|
5
|
-
}
|
|
6
|
-
else if (typeof value === 'string') {
|
|
7
|
-
if (value.includes('|')) {
|
|
8
|
-
const enumValues = value.split('|').map(v => v.trim());
|
|
9
|
-
return z.enum(enumValues);
|
|
10
|
-
}
|
|
11
|
-
return z.string().describe(value);
|
|
12
|
-
}
|
|
13
|
-
else if (typeof value === 'object') {
|
|
14
|
-
return convertToZodSchema(value);
|
|
15
|
-
}
|
|
16
|
-
return z.string();
|
|
17
|
-
}
|
|
18
|
-
export function convertToZodSchema(schema) {
|
|
19
|
-
const zodSchema = {};
|
|
20
|
-
for (const [key, value] of Object.entries(schema)) {
|
|
21
|
-
zodSchema[key] = convertValue(value);
|
|
22
|
-
}
|
|
23
|
-
return z.object(zodSchema);
|
|
24
|
-
}
|
|
25
|
-
//# sourceMappingURL=schema.js.map
|
|
@@ -1 +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
DELETED
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
import type { GenerateTextResult, StreamingResult } from 'ai';
|
|
2
|
-
export declare function createJsonResponse(result: GenerateTextResult): Response;
|
|
3
|
-
export declare function createStreamResponse(result: StreamingResult): Response;
|
|
4
|
-
export declare function createTextResponse(result: GenerateTextResult): Response;
|
|
5
|
-
//# sourceMappingURL=responses.d.ts.map
|
|
@@ -1 +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
DELETED
|
@@ -1,16 +0,0 @@
|
|
|
1
|
-
export function createJsonResponse(result) {
|
|
2
|
-
return new Response(JSON.stringify(result), {
|
|
3
|
-
headers: { 'Content-Type': 'application/json' }
|
|
4
|
-
});
|
|
5
|
-
}
|
|
6
|
-
export function createStreamResponse(result) {
|
|
7
|
-
return new Response(result.experimental_stream, {
|
|
8
|
-
headers: { 'Content-Type': 'text/event-stream' }
|
|
9
|
-
});
|
|
10
|
-
}
|
|
11
|
-
export function createTextResponse(result) {
|
|
12
|
-
return new Response(result.text, {
|
|
13
|
-
headers: { 'Content-Type': 'text/plain' },
|
|
14
|
-
});
|
|
15
|
-
}
|
|
16
|
-
//# sourceMappingURL=responses.js.map
|
|
@@ -1 +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"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"responses.test.d.ts","sourceRoot":"","sources":["../../src/core/responses.test.ts"],"names":[],"mappings":""}
|