ai-functions 0.2.19 → 0.3.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/LICENSE +1 -1
- package/README.md +112 -21
- package/bin/cli.js +5 -0
- package/dist/cli/index.d.ts +10 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +38 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/index.test.d.ts +2 -0
- package/dist/cli/index.test.d.ts.map +1 -0
- package/dist/cli/index.test.js +35 -0
- package/dist/cli/index.test.js.map +1 -0
- package/dist/constants/models.d.ts +10 -0
- package/dist/constants/models.d.ts.map +1 -0
- package/dist/constants/models.js +12 -0
- package/dist/constants/models.js.map +1 -0
- package/dist/converters/index.d.ts +3 -0
- package/dist/converters/index.d.ts.map +1 -0
- package/dist/converters/index.js +3 -0
- package/dist/converters/index.js.map +1 -0
- package/dist/converters/model.d.ts +4 -0
- package/dist/converters/model.d.ts.map +1 -0
- package/dist/converters/model.js +19 -0
- package/dist/converters/model.js.map +1 -0
- package/dist/converters/schema.d.ts +4 -0
- package/dist/converters/schema.d.ts.map +1 -0
- package/dist/converters/schema.js +25 -0
- package/dist/converters/schema.js.map +1 -0
- package/dist/core/responses.d.ts +5 -0
- package/dist/core/responses.d.ts.map +1 -0
- package/dist/core/responses.js +16 -0
- package/dist/core/responses.js.map +1 -0
- package/dist/core/responses.test.d.ts +2 -0
- package/dist/core/responses.test.d.ts.map +1 -0
- package/dist/core/responses.test.js +31 -0
- package/dist/core/responses.test.js.map +1 -0
- package/dist/errors.d.ts +6 -0
- package/dist/errors.d.ts.map +1 -0
- package/dist/errors.js +9 -0
- package/dist/errors.js.map +1 -0
- package/dist/examples/streaming.test.d.ts +2 -0
- package/dist/examples/streaming.test.d.ts.map +1 -0
- package/dist/examples/streaming.test.js +176 -0
- package/dist/examples/streaming.test.js.map +1 -0
- package/dist/factory/__tests__/index.test.d.ts +2 -0
- package/dist/factory/__tests__/index.test.d.ts.map +1 -0
- package/dist/factory/__tests__/index.test.js +430 -0
- package/dist/factory/__tests__/index.test.js.map +1 -0
- package/dist/factory/__tests__/list.test.d.ts +2 -0
- package/dist/factory/__tests__/list.test.d.ts.map +1 -0
- package/dist/factory/__tests__/list.test.js +92 -0
- package/dist/factory/__tests__/list.test.js.map +1 -0
- package/dist/factory/index.d.ts +20 -0
- package/dist/factory/index.d.ts.map +1 -0
- package/dist/factory/index.js +287 -0
- package/dist/factory/index.js.map +1 -0
- package/dist/factory/index.test.d.ts +2 -0
- package/dist/factory/index.test.d.ts.map +1 -0
- package/dist/factory/index.test.js +287 -0
- package/dist/factory/index.test.js.map +1 -0
- package/dist/factory/list.d.ts +3 -0
- package/dist/factory/list.d.ts.map +1 -0
- package/dist/factory/list.js +221 -0
- package/dist/factory/list.js.map +1 -0
- package/dist/factory/list.test.d.ts +2 -0
- package/dist/factory/list.test.d.ts.map +1 -0
- package/dist/factory/list.test.js +84 -0
- package/dist/factory/list.test.js.map +1 -0
- package/dist/generate/index.d.ts +5 -0
- package/dist/generate/index.d.ts.map +1 -0
- package/dist/generate/index.js +17 -0
- package/dist/generate/index.js.map +1 -0
- package/dist/index.d.ts +5 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +53 -0
- package/dist/index.js.map +1 -0
- package/dist/index.test.d.ts +2 -0
- package/dist/index.test.d.ts.map +1 -0
- package/dist/index.test.js +59 -0
- package/dist/index.test.js.map +1 -0
- package/dist/list/await.d.ts +3 -0
- package/dist/list/await.d.ts.map +1 -0
- package/dist/list/await.js +28 -0
- package/dist/list/await.js.map +1 -0
- package/dist/list/constants.d.ts +4 -0
- package/dist/list/constants.d.ts.map +1 -0
- package/dist/list/constants.js +5 -0
- package/dist/list/constants.js.map +1 -0
- package/dist/list/create-function.d.ts +3 -0
- package/dist/list/create-function.d.ts.map +1 -0
- package/dist/list/create-function.js +11 -0
- package/dist/list/create-function.js.map +1 -0
- package/dist/list/index.d.ts +4 -0
- package/dist/list/index.d.ts.map +1 -0
- package/dist/list/index.js +5 -0
- package/dist/list/index.js.map +1 -0
- package/dist/list/prompt.d.ts +3 -0
- package/dist/list/prompt.d.ts.map +1 -0
- package/dist/list/prompt.js +6 -0
- package/dist/list/prompt.js.map +1 -0
- package/dist/list/schemas.d.ts +4 -0
- package/dist/list/schemas.d.ts.map +1 -0
- package/dist/list/schemas.js +8 -0
- package/dist/list/schemas.js.map +1 -0
- package/dist/list/stream.d.ts +3 -0
- package/dist/list/stream.d.ts.map +1 -0
- package/dist/list/stream.js +33 -0
- package/dist/list/stream.js.map +1 -0
- package/dist/list/types.d.ts +11 -0
- package/dist/list/types.d.ts.map +1 -0
- package/dist/list/types.js +2 -0
- package/dist/list/types.js.map +1 -0
- package/dist/list/validation.d.ts +3 -0
- package/dist/list/validation.d.ts.map +1 -0
- package/dist/list/validation.js +12 -0
- package/dist/list/validation.js.map +1 -0
- package/dist/providers/config.d.ts +4 -0
- package/dist/providers/config.d.ts.map +1 -0
- package/dist/providers/config.js +21 -0
- package/dist/providers/config.js.map +1 -0
- package/dist/providers/config.test.d.ts +2 -0
- package/dist/providers/config.test.d.ts.map +1 -0
- package/dist/providers/config.test.js +37 -0
- package/dist/providers/config.test.js.map +1 -0
- package/dist/proxy/constants.d.ts +4 -0
- package/dist/proxy/constants.d.ts.map +1 -0
- package/dist/proxy/constants.js +5 -0
- package/dist/proxy/constants.js.map +1 -0
- package/dist/proxy/create-function.d.ts +4 -0
- package/dist/proxy/create-function.d.ts.map +1 -0
- package/dist/proxy/create-function.js +24 -0
- package/dist/proxy/create-function.js.map +1 -0
- package/dist/proxy/create-proxy.d.ts +2 -0
- package/dist/proxy/create-proxy.d.ts.map +1 -0
- package/dist/proxy/create-proxy.js +11 -0
- package/dist/proxy/create-proxy.js.map +1 -0
- package/dist/proxy/function-generator.d.ts +9 -0
- package/dist/proxy/function-generator.d.ts.map +1 -0
- package/dist/proxy/function-generator.js +29 -0
- package/dist/proxy/function-generator.js.map +1 -0
- package/dist/proxy/index.d.ts +4 -0
- package/dist/proxy/index.d.ts.map +1 -0
- package/dist/proxy/index.js +4 -0
- package/dist/proxy/index.js.map +1 -0
- package/dist/proxy/prompt.d.ts +2 -0
- package/dist/proxy/prompt.d.ts.map +1 -0
- package/dist/proxy/prompt.js +6 -0
- package/dist/proxy/prompt.js.map +1 -0
- package/dist/proxy/types.d.ts +7 -0
- package/dist/proxy/types.d.ts.map +1 -0
- package/dist/proxy/types.js +2 -0
- package/dist/proxy/types.js.map +1 -0
- package/dist/queue/manager.d.ts +5 -0
- package/dist/queue/manager.d.ts.map +1 -0
- package/dist/queue/manager.js +37 -0
- package/dist/queue/manager.js.map +1 -0
- package/dist/queue/manager.test.d.ts +2 -0
- package/dist/queue/manager.test.d.ts.map +1 -0
- package/dist/queue/manager.test.js +52 -0
- package/dist/queue/manager.test.js.map +1 -0
- package/dist/schema-converter.d.ts +4 -0
- package/dist/schema-converter.d.ts.map +1 -0
- package/dist/schema-converter.js +30 -0
- package/dist/schema-converter.js.map +1 -0
- package/dist/stream/index.d.ts +7 -0
- package/dist/stream/index.d.ts.map +1 -0
- package/dist/stream/index.js +23 -0
- package/dist/stream/index.js.map +1 -0
- package/dist/streaming/utils.d.ts +4 -0
- package/dist/streaming/utils.d.ts.map +1 -0
- package/dist/streaming/utils.js +131 -0
- package/dist/streaming/utils.js.map +1 -0
- package/dist/streaming/utils.test.d.ts +2 -0
- package/dist/streaming/utils.test.d.ts.map +1 -0
- package/dist/streaming/utils.test.js +84 -0
- package/dist/streaming/utils.test.js.map +1 -0
- package/dist/templates/result.d.ts +7 -0
- package/dist/templates/result.d.ts.map +1 -0
- package/dist/templates/result.js +40 -0
- package/dist/templates/result.js.map +1 -0
- package/dist/templates/result.test.d.ts +2 -0
- package/dist/templates/result.test.d.ts.map +1 -0
- package/dist/templates/result.test.js +75 -0
- package/dist/templates/result.test.js.map +1 -0
- package/dist/test/setup.d.ts +2 -0
- package/dist/test/setup.d.ts.map +1 -0
- package/dist/test/setup.js +21 -0
- package/dist/test/setup.js.map +1 -0
- package/dist/test-types.d.ts +13 -0
- package/dist/test-types.d.ts.map +1 -0
- package/dist/test-types.js +55 -0
- package/dist/test-types.js.map +1 -0
- package/dist/types/index.d.ts +4 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +4 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/list.d.ts +10 -0
- package/dist/types/list.d.ts.map +1 -0
- package/dist/types/list.js +2 -0
- package/dist/types/list.js.map +1 -0
- package/dist/types/model.d.ts +7 -0
- package/dist/types/model.d.ts.map +1 -0
- package/dist/types/model.js +2 -0
- package/dist/types/model.js.map +1 -0
- package/dist/types/options.d.ts +25 -0
- package/dist/types/options.d.ts.map +1 -0
- package/dist/types/options.js +2 -0
- package/dist/types/options.js.map +1 -0
- package/dist/types/schema.d.ts +5 -0
- package/dist/types/schema.d.ts.map +1 -0
- package/dist/types/schema.js +2 -0
- package/dist/types/schema.js.map +1 -0
- package/dist/types.d.ts +113 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +9 -0
- package/dist/types.js.map +1 -0
- package/dist/utils/__tests__/request-handler.test.d.ts +2 -0
- package/dist/utils/__tests__/request-handler.test.d.ts.map +1 -0
- package/dist/utils/__tests__/request-handler.test.js +134 -0
- package/dist/utils/__tests__/request-handler.test.js.map +1 -0
- package/dist/utils/__tests__/schema.test.d.ts +2 -0
- package/dist/utils/__tests__/schema.test.d.ts.map +1 -0
- package/dist/utils/__tests__/schema.test.js +49 -0
- package/dist/utils/__tests__/schema.test.js.map +1 -0
- package/dist/utils/__tests__/stream-progress.test.d.ts +2 -0
- package/dist/utils/__tests__/stream-progress.test.d.ts.map +1 -0
- package/dist/utils/__tests__/stream-progress.test.js +85 -0
- package/dist/utils/__tests__/stream-progress.test.js.map +1 -0
- package/dist/utils/index.d.ts +2 -0
- package/dist/utils/index.d.ts.map +1 -0
- package/dist/utils/index.js +2 -0
- package/dist/utils/index.js.map +1 -0
- package/dist/utils/request-handler.d.ts +17 -0
- package/dist/utils/request-handler.d.ts.map +1 -0
- package/dist/utils/request-handler.js +105 -0
- package/dist/utils/request-handler.js.map +1 -0
- package/dist/utils/schema.d.ts +11 -0
- package/dist/utils/schema.d.ts.map +1 -0
- package/dist/utils/schema.js +51 -0
- package/dist/utils/schema.js.map +1 -0
- package/dist/utils/stream-progress.d.ts +17 -0
- package/dist/utils/stream-progress.d.ts.map +1 -0
- package/dist/utils/stream-progress.js +86 -0
- package/dist/utils/stream-progress.js.map +1 -0
- package/dist/utils/validation.d.ts +3 -0
- package/dist/utils/validation.d.ts.map +1 -0
- package/dist/utils/validation.js +30 -0
- package/dist/utils/validation.js.map +1 -0
- package/package.json +56 -40
- package/db/cache.ts +0 -6
- package/db/mongo.ts +0 -75
- package/dist/mjs/db/cache.d.ts +0 -1
- package/dist/mjs/db/cache.js +0 -5
- package/dist/mjs/db/mongo.d.ts +0 -31
- package/dist/mjs/db/mongo.js +0 -48
- package/dist/mjs/examples/data.d.ts +0 -1105
- package/dist/mjs/examples/data.js +0 -1105
- package/dist/mjs/functions/ai.d.ts +0 -20
- package/dist/mjs/functions/ai.js +0 -83
- package/dist/mjs/functions/ai.test.d.ts +0 -1
- package/dist/mjs/functions/ai.test.js +0 -29
- package/dist/mjs/functions/gpt.d.ts +0 -4
- package/dist/mjs/functions/gpt.js +0 -10
- package/dist/mjs/functions/list.d.ts +0 -7
- package/dist/mjs/functions/list.js +0 -72
- package/dist/mjs/index.d.ts +0 -3
- package/dist/mjs/index.js +0 -3
- package/dist/mjs/queue/kafka.d.ts +0 -0
- package/dist/mjs/queue/kafka.js +0 -1
- package/dist/mjs/queue/memory.d.ts +0 -0
- package/dist/mjs/queue/memory.js +0 -1
- package/dist/mjs/queue/mongo.d.ts +0 -30
- package/dist/mjs/queue/mongo.js +0 -42
- package/dist/mjs/streams/kafka.d.ts +0 -0
- package/dist/mjs/streams/kafka.js +0 -1
- package/dist/mjs/streams/memory.d.ts +0 -0
- package/dist/mjs/streams/memory.js +0 -1
- package/dist/mjs/streams/mongo.d.ts +0 -0
- package/dist/mjs/streams/mongo.js +0 -1
- package/dist/mjs/streams/types.d.ts +0 -0
- package/dist/mjs/streams/types.js +0 -1
- package/dist/mjs/types.d.ts +0 -11
- package/dist/mjs/types.js +0 -1
- package/dist/mjs/utils/completion.d.ts +0 -9
- package/dist/mjs/utils/completion.js +0 -20
- package/dist/mjs/utils/schema.d.ts +0 -10
- package/dist/mjs/utils/schema.js +0 -72
- package/dist/mjs/utils/schema.test.d.ts +0 -1
- package/dist/mjs/utils/schema.test.js +0 -60
- package/dist/mjs/utils/state.d.ts +0 -1
- package/dist/mjs/utils/state.js +0 -19
- package/examples/data.ts +0 -1105
- package/fixup +0 -11
- package/functions/ai.test.ts +0 -41
- package/functions/ai.ts +0 -115
- package/functions/gpt.ts +0 -12
- package/functions/list.ts +0 -84
- package/index.ts +0 -3
- package/queue/kafka.ts +0 -0
- package/queue/memory.ts +0 -0
- package/queue/mongo.ts +0 -88
- package/streams/kafka.ts +0 -0
- package/streams/memory.ts +0 -0
- package/streams/mongo.ts +0 -0
- package/streams/types.ts +0 -0
- package/tsconfig-backup.json +0 -105
- package/tsconfig-base.json +0 -26
- package/tsconfig-cjs.json +0 -8
- package/tsconfig.json +0 -8
- package/types.ts +0 -12
- package/utils/completion.ts +0 -28
- package/utils/schema.test.ts +0 -69
- package/utils/schema.ts +0 -74
- package/utils/state.ts +0 -23
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import { generateObject as aiGenerateObject } from 'ai';
|
|
2
|
+
import { convertToZodSchema, convertToOpenAIModel } from '@/converters';
|
|
3
|
+
import { validateSchema } from '@/utils';
|
|
4
|
+
export async function generateObject(options) {
|
|
5
|
+
validateSchema(options.schema);
|
|
6
|
+
const zodSchema = convertToZodSchema(options.schema);
|
|
7
|
+
const model = options.model ? convertToOpenAIModel(options.model) : convertToOpenAIModel('gpt-4-turbo');
|
|
8
|
+
return aiGenerateObject({
|
|
9
|
+
...options,
|
|
10
|
+
model,
|
|
11
|
+
schema: zodSchema,
|
|
12
|
+
temperature: options.temperature ?? 0.7,
|
|
13
|
+
maxTokens: options.maxTokens ?? 1000,
|
|
14
|
+
seed: typeof options.seed === 'number' ? options.seed : undefined
|
|
15
|
+
});
|
|
16
|
+
}
|
|
17
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/generate/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,IAAI,gBAAgB,EAAE,MAAM,IAAI,CAAA;AACvD,OAAO,EAAE,kBAAkB,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAA;AACvE,OAAO,EAAE,cAAc,EAAE,MAAM,SAAS,CAAA;AAGxC,MAAM,CAAC,KAAK,UAAU,cAAc,CAAyB,OAAiC;IAC5F,cAAc,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;IAC9B,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;IACpD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAA;IAEvG,OAAO,gBAAgB,CAAC;QACtB,GAAG,OAAO;QACV,KAAK;QACL,MAAM,EAAE,SAAS;QACjB,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;QACvC,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;QACpC,IAAI,EAAE,OAAO,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;KAClE,CAAC,CAAA;AACJ,CAAC"}
|
package/dist/index.d.ts
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,EAAE,EAAE,YAAY,EAA2D,MAAM,SAAS,CAAA;AA0DnG,eAAO,MAAM,EAAE,EAA4D,EAAE,CAAA;AAG7E,eAAO,MAAM,IAAI,EAA4C,YAAY,CAAA;AAGzE,cAAc,SAAS,CAAA"}
|
package/dist/index.js
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
import { createAIFunction, createTemplateFunction } from './factory';
|
|
2
|
+
import { createListFunction } from './factory/list';
|
|
3
|
+
import { z } from 'zod';
|
|
4
|
+
// Create the main template function with async iteration support
|
|
5
|
+
const templateFn = createTemplateFunction();
|
|
6
|
+
// Create the list function with async iteration support
|
|
7
|
+
const listFn = createListFunction();
|
|
8
|
+
// Create the categorizeProduct function with proper schema
|
|
9
|
+
const categorizeProduct = createAIFunction(z.object({
|
|
10
|
+
productType: z.enum(['App', 'API', 'Marketplace', 'Platform', 'Packaged Service', 'Professional Service', 'Website']),
|
|
11
|
+
customer: z.string().describe('ideal customer profile in 3-5 words'),
|
|
12
|
+
solution: z.string().describe('describe the offer in 4-10 words'),
|
|
13
|
+
description: z.string().describe('website meta description'),
|
|
14
|
+
}));
|
|
15
|
+
// Create the main AI object with template literal and async iteration support
|
|
16
|
+
function isTemplateStringsArray(value) {
|
|
17
|
+
return Array.isArray(value) && 'raw' in value;
|
|
18
|
+
}
|
|
19
|
+
function createWrappedTemplateFunction(baseFn) {
|
|
20
|
+
const wrappedFn = function (stringsOrOptions, ...values) {
|
|
21
|
+
if (!isTemplateStringsArray(stringsOrOptions)) {
|
|
22
|
+
return baseFn(values.shift(), ...values, stringsOrOptions);
|
|
23
|
+
}
|
|
24
|
+
return baseFn(stringsOrOptions, ...values);
|
|
25
|
+
};
|
|
26
|
+
// Make wrappedFn async iterable by delegating to baseFn
|
|
27
|
+
Object.defineProperty(wrappedFn, Symbol.asyncIterator, {
|
|
28
|
+
value: function () {
|
|
29
|
+
return baseFn[Symbol.asyncIterator]();
|
|
30
|
+
},
|
|
31
|
+
writable: true,
|
|
32
|
+
configurable: true,
|
|
33
|
+
});
|
|
34
|
+
// Add withOptions method
|
|
35
|
+
Object.defineProperty(wrappedFn, 'withOptions', {
|
|
36
|
+
value: (opts) => baseFn.withOptions(opts),
|
|
37
|
+
writable: true,
|
|
38
|
+
configurable: true,
|
|
39
|
+
});
|
|
40
|
+
// Add queue property
|
|
41
|
+
Object.defineProperty(wrappedFn, 'queue', {
|
|
42
|
+
get: () => baseFn.queue,
|
|
43
|
+
configurable: true,
|
|
44
|
+
});
|
|
45
|
+
return wrappedFn;
|
|
46
|
+
}
|
|
47
|
+
const aiFn = createWrappedTemplateFunction(templateFn);
|
|
48
|
+
export const ai = Object.assign(aiFn, { categorizeProduct });
|
|
49
|
+
// Create the list function with template literal and async iteration support
|
|
50
|
+
export const list = createWrappedTemplateFunction(listFn);
|
|
51
|
+
// Export types for consumers
|
|
52
|
+
export * from './types';
|
|
53
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,MAAM,WAAW,CAAA;AACpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAA;AAEnD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,iEAAiE;AACjE,MAAM,UAAU,GAAG,sBAAsB,EAAE,CAAA;AAE3C,wDAAwD;AACxD,MAAM,MAAM,GAAG,kBAAkB,EAAE,CAAA;AAEnC,2DAA2D;AAC3D,MAAM,iBAAiB,GAAG,gBAAgB,CACxC,CAAC,CAAC,MAAM,CAAC;IACP,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,kBAAkB,EAAE,sBAAsB,EAAE,SAAS,CAAC,CAAC;IACrH,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qCAAqC,CAAC;IACpE,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kCAAkC,CAAC;IACjE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,0BAA0B,CAAC;CAC7D,CAAC,CACH,CAAA;AAED,8EAA8E;AAC9E,SAAS,sBAAsB,CAAC,KAAc;IAC5C,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,KAAK,CAAA;AAC/C,CAAC;AAED,SAAS,6BAA6B,CAAC,MAA4B;IACjE,MAAM,SAAS,GAAG,UAAU,gBAA2D,EAAE,GAAG,MAAiB;QAC3G,IAAI,CAAC,sBAAsB,CAAC,gBAAgB,CAAC,EAAE,CAAC;YAC9C,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,EAA0B,EAAE,GAAG,MAAM,EAAE,gBAAgB,CAAC,CAAA;QACpF,CAAC;QACD,OAAO,MAAM,CAAC,gBAAgB,EAAE,GAAG,MAAM,CAAC,CAAA;IAC5C,CAAC,CAAA;IAED,wDAAwD;IACxD,MAAM,CAAC,cAAc,CAAC,SAAS,EAAE,MAAM,CAAC,aAAa,EAAE;QACrD,KAAK,EAAE;YACL,OAAO,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAA;QACvC,CAAC;QACD,QAAQ,EAAE,IAAI;QACd,YAAY,EAAE,IAAI;KACnB,CAAC,CAAA;IAEF,yBAAyB;IACzB,MAAM,CAAC,cAAc,CAAC,SAAS,EAAE,aAAa,EAAE;QAC9C,KAAK,EAAE,CAAC,IAAwB,EAAE,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC;QAC7D,QAAQ,EAAE,IAAI;QACd,YAAY,EAAE,IAAI;KACnB,CAAC,CAAA;IAEF,qBAAqB;IACrB,MAAM,CAAC,cAAc,CAAC,SAAS,EAAE,OAAO,EAAE;QACxC,GAAG,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK;QACvB,YAAY,EAAE,IAAI;KACnB,CAAC,CAAA;IAEF,OAAO,SAAiC,CAAA;AAC1C,CAAC;AAED,MAAM,IAAI,GAAG,6BAA6B,CAAC,UAAU,CAAC,CAAA;AACtD,MAAM,CAAC,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,iBAAiB,EAAE,CAAkB,CAAA;AAE7E,6EAA6E;AAC7E,MAAM,CAAC,MAAM,IAAI,GAAG,6BAA6B,CAAC,MAAM,CAAiB,CAAA;AAEzE,6BAA6B;AAC7B,cAAc,SAAS,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.test.d.ts","sourceRoot":"","sources":["../src/index.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,59 @@
|
|
|
1
|
+
import { describe, expect, it, beforeEach } from 'vitest';
|
|
2
|
+
import { ai } from './index';
|
|
3
|
+
import { z } from 'zod';
|
|
4
|
+
import { openai } from '@ai-sdk/openai';
|
|
5
|
+
describe('ai template tag', () => {
|
|
6
|
+
beforeEach(() => {
|
|
7
|
+
process.env.OPENAI_API_KEY = process.env.OPENAI_API_KEY || 'test-key';
|
|
8
|
+
});
|
|
9
|
+
const model = openai(process.env.OPENAI_DEFAULT_MODEL || 'gpt-4o', { structuredOutputs: true });
|
|
10
|
+
it('should support basic template literals', async () => {
|
|
11
|
+
const name = 'World';
|
|
12
|
+
const result = await ai `Hello ${name}`({ model });
|
|
13
|
+
expect(typeof result).toBe('string');
|
|
14
|
+
expect(result.length).toBeGreaterThan(0);
|
|
15
|
+
});
|
|
16
|
+
it('should support configuration object', async () => {
|
|
17
|
+
const name = 'World';
|
|
18
|
+
const result = await ai `Hello ${name}`({ model });
|
|
19
|
+
expect(typeof result).toBe('string');
|
|
20
|
+
expect(result.length).toBeGreaterThan(0);
|
|
21
|
+
});
|
|
22
|
+
it('should support JSON output with schema', async () => {
|
|
23
|
+
const schema = z.object({
|
|
24
|
+
greeting: z.string(),
|
|
25
|
+
timestamp: z.number(),
|
|
26
|
+
});
|
|
27
|
+
const result = await ai `Generate a greeting with the current timestamp`({
|
|
28
|
+
model,
|
|
29
|
+
outputFormat: 'json',
|
|
30
|
+
schema
|
|
31
|
+
});
|
|
32
|
+
const parsed = schema.parse(JSON.parse(result));
|
|
33
|
+
expect(parsed).toBeDefined();
|
|
34
|
+
expect(typeof parsed.greeting).toBe('string');
|
|
35
|
+
expect(typeof parsed.timestamp).toBe('number');
|
|
36
|
+
});
|
|
37
|
+
it('should support streaming responses', async () => {
|
|
38
|
+
const streamingModel = openai('gpt-4o', {
|
|
39
|
+
structuredOutputs: true
|
|
40
|
+
});
|
|
41
|
+
const response = ai `List some items`({
|
|
42
|
+
model: streamingModel,
|
|
43
|
+
outputFormat: 'json',
|
|
44
|
+
schema: z.array(z.string()),
|
|
45
|
+
streaming: {
|
|
46
|
+
onProgress: () => { },
|
|
47
|
+
enableTokenCounting: true
|
|
48
|
+
}
|
|
49
|
+
});
|
|
50
|
+
expect(response[Symbol.asyncIterator]).toBeDefined();
|
|
51
|
+
const chunks = [];
|
|
52
|
+
for await (const chunk of response) {
|
|
53
|
+
chunks.push(chunk);
|
|
54
|
+
}
|
|
55
|
+
expect(chunks.length).toBeGreaterThan(0);
|
|
56
|
+
expect(JSON.parse(chunks.join(''))).toBeInstanceOf(Array);
|
|
57
|
+
});
|
|
58
|
+
});
|
|
59
|
+
//# sourceMappingURL=index.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.test.js","sourceRoot":"","sources":["../src/index.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAA;AACzD,OAAO,EAAE,EAAE,EAAE,MAAM,SAAS,CAAA;AAC5B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAA;AAEvC,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,UAAU,CAAA;IACvE,CAAC,CAAC,CAAA;IAEF,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,QAAQ,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAA;IAE/F,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,IAAI,GAAG,OAAO,CAAA;QACpB,MAAM,MAAM,GAAG,MAAM,EAAE,CAAA,SAAS,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;QACjD,MAAM,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACpC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;IAC1C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,IAAI,GAAG,OAAO,CAAA;QACpB,MAAM,MAAM,GAAG,MAAM,EAAE,CAAA,SAAS,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;QACjD,MAAM,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACpC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;IAC1C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;YACtB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;YACpB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;SACtB,CAAC,CAAA;QAEF,MAAM,MAAM,GAAG,MAAM,EAAE,CAAA,gDAAgD,CAAC;YACtE,KAAK;YACL,YAAY,EAAE,MAAM;YACpB,MAAM;SACP,CAAC,CAAA;QACF,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAA;QAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAA;QAC5B,MAAM,CAAC,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC7C,MAAM,CAAC,OAAO,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;IAChD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,cAAc,GAAG,MAAM,CAAC,QAAQ,EAAE;YACtC,iBAAiB,EAAE,IAAI;SACxB,CAAC,CAAA;QACF,MAAM,QAAQ,GAAG,EAAE,CAAA,iBAAiB,CAAC;YACnC,KAAK,EAAE,cAAc;YACrB,YAAY,EAAE,MAAM;YACpB,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;YAC3B,SAAS,EAAE;gBACT,UAAU,EAAE,GAAG,EAAE,GAAE,CAAC;gBACpB,mBAAmB,EAAE,IAAI;aAC1B;SACF,CAAiC,CAAA;QAElC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,WAAW,EAAE,CAAA;QACpD,MAAM,MAAM,GAAa,EAAE,CAAA;QAC3B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;YACnC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACpB,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;QACxC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAA;IAC3D,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"await.d.ts","sourceRoot":"","sources":["../../src/list/await.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,iBAAiB,EAAE,WAAW,EAAE,MAAM,SAAS,CAAA;AAE7D,eAAO,MAAM,eAAe,YAAa,WAAW,KAAG,iBAuBtD,CAAA"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import { generateObject } from 'ai';
|
|
2
|
+
import { convertToOpenAIModel } from '@/converters';
|
|
3
|
+
import { DEFAULT_LIST_OPTIONS } from './constants';
|
|
4
|
+
import { generateListPrompt } from './prompt';
|
|
5
|
+
import { createArraySchema } from './schemas';
|
|
6
|
+
export const createListAwait = (options) => {
|
|
7
|
+
return async function listAwait(strings, ...values) {
|
|
8
|
+
const prompt = generateListPrompt(strings, ...values);
|
|
9
|
+
const model = convertToOpenAIModel(options.model || DEFAULT_LIST_OPTIONS.model);
|
|
10
|
+
const schema = createArraySchema();
|
|
11
|
+
const { object } = await generateObject({
|
|
12
|
+
model,
|
|
13
|
+
schema,
|
|
14
|
+
prompt,
|
|
15
|
+
system: options.system,
|
|
16
|
+
seed: options.seed,
|
|
17
|
+
temperature: 0.7,
|
|
18
|
+
maxTokens: 2000, // Increased for better completion
|
|
19
|
+
retry: {
|
|
20
|
+
attempts: 3,
|
|
21
|
+
delay: 1000,
|
|
22
|
+
factor: 2 // Exponential backoff
|
|
23
|
+
}
|
|
24
|
+
});
|
|
25
|
+
return object.items || [];
|
|
26
|
+
};
|
|
27
|
+
};
|
|
28
|
+
//# sourceMappingURL=await.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"await.js","sourceRoot":"","sources":["../../src/list/await.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,IAAI,CAAA;AACnC,OAAO,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAA;AACnD,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAA;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAA;AAC7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAA;AAG7C,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,OAAoB,EAAqB,EAAE;IACzE,OAAO,KAAK,UAAU,SAAS,CAAC,OAAO,EAAE,GAAG,MAAM;QAChD,MAAM,MAAM,GAAG,kBAAkB,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,CAAA;QACrD,MAAM,KAAK,GAAG,oBAAoB,CAAC,OAAO,CAAC,KAAK,IAAI,oBAAoB,CAAC,KAAK,CAAC,CAAA;QAC/E,MAAM,MAAM,GAAG,iBAAiB,EAAE,CAAA;QAElC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,cAAc,CAAC;YACtC,KAAK;YACL,MAAM;YACN,MAAM;YACN,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,WAAW,EAAE,GAAG;YAChB,SAAS,EAAE,IAAI,EAAE,kCAAkC;YACnD,KAAK,EAAE;gBACL,QAAQ,EAAE,CAAC;gBACX,KAAK,EAAE,IAAI;gBACX,MAAM,EAAE,CAAC,CAAC,sBAAsB;aACjC;SACF,CAAC,CAAA;QAEF,OAAO,MAAM,CAAC,KAAK,IAAI,EAAE,CAAA;IAC3B,CAAC,CAAA;AACH,CAAC,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../src/list/constants.ts"],"names":[],"mappings":"AAEA,eAAO,MAAM,oBAAoB;;CAEvB,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../src/list/constants.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAA;AAE5C,MAAM,CAAC,MAAM,oBAAoB,GAAG;IAClC,KAAK,EAAE,MAAM,CAAC,IAAI;CACV,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"create-function.d.ts","sourceRoot":"","sources":["../../src/list/create-function.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,SAAS,CAAA;AAExD,wBAAgB,kBAAkB,CAAC,OAAO,GAAE,WAAkC,GAAG,YAAY,CAK5F"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { validateListOptions } from './validation';
|
|
2
|
+
import { createListStream } from './stream';
|
|
3
|
+
import { createListAwait } from './await';
|
|
4
|
+
import { DEFAULT_LIST_OPTIONS } from './constants';
|
|
5
|
+
export function createListFunction(options = DEFAULT_LIST_OPTIONS) {
|
|
6
|
+
validateListOptions(options);
|
|
7
|
+
const listStream = createListStream(options);
|
|
8
|
+
const listAwait = createListAwait(options);
|
|
9
|
+
return Object.assign(listStream, { await: listAwait });
|
|
10
|
+
}
|
|
11
|
+
//# sourceMappingURL=create-function.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"create-function.js","sourceRoot":"","sources":["../../src/list/create-function.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAA;AAClD,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAA;AAC3C,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAA;AACzC,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAA;AAGlD,MAAM,UAAU,kBAAkB,CAAC,UAAuB,oBAAoB;IAC5E,mBAAmB,CAAC,OAAO,CAAC,CAAA;IAC5B,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAA;IAC5C,MAAM,SAAS,GAAG,eAAe,CAAC,OAAO,CAAC,CAAA;IAC1C,OAAO,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;AACxD,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/list/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAA;AAGtD,eAAO,MAAM,IAAI,cAA2C,CAAA;AAC5D,OAAO,EAAE,kBAAkB,EAAE,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/list/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAA;AACtD,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAA;AAElD,MAAM,CAAC,MAAM,IAAI,GAAG,kBAAkB,CAAC,oBAAoB,CAAC,CAAA;AAC5D,OAAO,EAAE,kBAAkB,EAAE,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"prompt.d.ts","sourceRoot":"","sources":["../../src/list/prompt.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAA;AAE7C,eAAO,MAAM,kBAAkB,EAAE,mBAIhC,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"prompt.js","sourceRoot":"","sources":["../../src/list/prompt.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,kBAAkB,GAAwB,CAAC,OAAO,EAAE,GAAG,MAAM,EAAE,EAAE;IAC5E,OAAO,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;QAC3B,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAClB,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,GAAG,MAAM,CAAC,CAAA;AAC7C,CAAC,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"schemas.d.ts","sourceRoot":"","sources":["../../src/list/schemas.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,SAAS,CAAA;AAErD,eAAO,MAAM,kBAAkB,QAAO,UAGlC,CAAA;AAEJ,eAAO,MAAM,iBAAiB,QAAO,eAKjC,CAAA"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
export const createStreamSchema = () => z.object({
|
|
3
|
+
item: z.string().describe('A single item from the requested list')
|
|
4
|
+
});
|
|
5
|
+
export const createArraySchema = () => z.object({
|
|
6
|
+
items: z.array(z.string().describe('An item from the requested list')).describe('List of requested items')
|
|
7
|
+
});
|
|
8
|
+
//# sourceMappingURL=schemas.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"schemas.js","sourceRoot":"","sources":["../../src/list/schemas.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAGvB,MAAM,CAAC,MAAM,kBAAkB,GAAG,GAAe,EAAE,CACjD,CAAC,CAAC,MAAM,CAAC;IACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uCAAuC,CAAC;CACnE,CAAC,CAAA;AAEJ,MAAM,CAAC,MAAM,iBAAiB,GAAG,GAAoB,EAAE,CACrD,CAAC,CAAC,MAAM,CAAC;IACP,KAAK,EAAE,CAAC,CAAC,KAAK,CACZ,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC,CACvD,CAAC,QAAQ,CAAC,yBAAyB,CAAC;CACtC,CAAC,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../src/list/stream.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,kBAAkB,EAAE,WAAW,EAAE,MAAM,SAAS,CAAA;AAE9D,eAAO,MAAM,gBAAgB,YAAa,WAAW,KAAG,kBA4BvD,CAAA"}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
import { streamObject } from 'ai';
|
|
2
|
+
import { convertToOpenAIModel } from '@/converters';
|
|
3
|
+
import { DEFAULT_LIST_OPTIONS } from './constants';
|
|
4
|
+
import { generateListPrompt } from './prompt';
|
|
5
|
+
import { createStreamSchema } from './schemas';
|
|
6
|
+
export const createListStream = (options) => {
|
|
7
|
+
return async function* listStream(strings, ...values) {
|
|
8
|
+
const prompt = generateListPrompt(strings, ...values);
|
|
9
|
+
const model = convertToOpenAIModel(options.model || DEFAULT_LIST_OPTIONS.model);
|
|
10
|
+
const schema = createStreamSchema();
|
|
11
|
+
const { elementStream } = await streamObject({
|
|
12
|
+
model,
|
|
13
|
+
output: 'array',
|
|
14
|
+
schema,
|
|
15
|
+
prompt,
|
|
16
|
+
system: options.system,
|
|
17
|
+
seed: options.seed,
|
|
18
|
+
temperature: 0.7,
|
|
19
|
+
maxTokens: 2000, // Increased for better completion
|
|
20
|
+
retry: {
|
|
21
|
+
attempts: 3,
|
|
22
|
+
delay: 1000,
|
|
23
|
+
factor: 2 // Exponential backoff
|
|
24
|
+
}
|
|
25
|
+
});
|
|
26
|
+
for await (const element of elementStream) {
|
|
27
|
+
if (element?.item) {
|
|
28
|
+
yield element.item;
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
};
|
|
32
|
+
};
|
|
33
|
+
//# sourceMappingURL=stream.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../src/list/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAA;AACjC,OAAO,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAA;AACnD,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAA;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAA;AAC7C,OAAO,EAAE,kBAAkB,EAAE,MAAM,WAAW,CAAA;AAG9C,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,OAAoB,EAAsB,EAAE;IAC3E,OAAO,KAAK,SAAS,CAAC,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,MAAM;QAClD,MAAM,MAAM,GAAG,kBAAkB,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,CAAA;QACrD,MAAM,KAAK,GAAG,oBAAoB,CAAC,OAAO,CAAC,KAAK,IAAI,oBAAoB,CAAC,KAAK,CAAC,CAAA;QAC/E,MAAM,MAAM,GAAG,kBAAkB,EAAE,CAAA;QAEnC,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,YAAY,CAAC;YAC3C,KAAK;YACL,MAAM,EAAE,OAAO;YACf,MAAM;YACN,MAAM;YACN,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,WAAW,EAAE,GAAG;YAChB,SAAS,EAAE,IAAI,EAAE,kCAAkC;YACnD,KAAK,EAAE;gBACL,QAAQ,EAAE,CAAC;gBACX,KAAK,EAAE,IAAI;gBACX,MAAM,EAAE,CAAC,CAAC,sBAAsB;aACjC;SACF,CAAC,CAAA;QAEF,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,aAAa,EAAE,CAAC;YAC1C,IAAI,OAAO,EAAE,IAAI,EAAE,CAAC;gBAClB,MAAM,OAAO,CAAC,IAAI,CAAA;YACpB,CAAC;QACH,CAAC;IACH,CAAC,CAAA;AACH,CAAC,CAAA"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
export type ListStreamFunction = (strings: string[] | TemplateStringsArray, ...values: any[]) => AsyncIterableIterator<any>;
|
|
3
|
+
export type ListAwaitFunction = (strings: string[] | TemplateStringsArray, ...values: any[]) => Promise<string[]>;
|
|
4
|
+
export type ListSchema = z.ZodObject<{
|
|
5
|
+
item: z.ZodString;
|
|
6
|
+
}>;
|
|
7
|
+
export type ListArraySchema = z.ZodObject<{
|
|
8
|
+
items: z.ZodArray<z.ZodString>;
|
|
9
|
+
}>;
|
|
10
|
+
export type ListPromptGenerator = (strings: string[] | TemplateStringsArray, ...values: any[]) => string;
|
|
11
|
+
//# sourceMappingURL=types.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/list/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAGvB,MAAM,MAAM,kBAAkB,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,oBAAoB,EAAE,GAAG,MAAM,EAAE,GAAG,EAAE,KAAK,qBAAqB,CAAC,GAAG,CAAC,CAAA;AAC3H,MAAM,MAAM,iBAAiB,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,oBAAoB,EAAE,GAAG,MAAM,EAAE,GAAG,EAAE,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC,CAAA;AAEjH,MAAM,MAAM,UAAU,GAAG,CAAC,CAAC,SAAS,CAAC;IACnC,IAAI,EAAE,CAAC,CAAC,SAAS,CAAA;CAClB,CAAC,CAAA;AAEF,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,SAAS,CAAC;IACxC,KAAK,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAA;CAC/B,CAAC,CAAA;AAEF,MAAM,MAAM,mBAAmB,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,oBAAoB,EAAE,GAAG,MAAM,EAAE,GAAG,EAAE,KAAK,MAAM,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/list/types.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"validation.d.ts","sourceRoot":"","sources":["../../src/list/validation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAA;AAE1C,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,WAAW,GAAG,IAAI,CAY9D"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
export function validateListOptions(options) {
|
|
2
|
+
if (options.seed !== undefined && !Number.isInteger(options.seed)) {
|
|
3
|
+
throw new Error('List options seed must be an integer');
|
|
4
|
+
}
|
|
5
|
+
if (options.model !== undefined && typeof options.model !== 'string') {
|
|
6
|
+
throw new Error('List options model must be a string');
|
|
7
|
+
}
|
|
8
|
+
if (options.system !== undefined && typeof options.system !== 'string') {
|
|
9
|
+
throw new Error('List options system must be a string');
|
|
10
|
+
}
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=validation.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"validation.js","sourceRoot":"","sources":["../../src/list/validation.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,mBAAmB,CAAC,OAAoB;IACtD,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAClE,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAA;IACzD,CAAC;IAED,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,IAAI,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrE,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;IACxD,CAAC;IAED,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,IAAI,OAAO,OAAO,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QACvE,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAA;IACzD,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/providers/config.ts"],"names":[],"mappings":"AAIA,wBAAgB,WAAW,YAmBV,MAAM,YAAY;IAAE,iBAAiB,CAAC,EAAE,OAAO,CAAA;CAAE,kCAEjE"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import { openai } from '@ai-sdk/openai';
|
|
2
|
+
import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
|
|
3
|
+
export function getProvider() {
|
|
4
|
+
const gateway = process.env.AI_GATEWAY;
|
|
5
|
+
const apiKey = process.env.OPENAI_API_KEY;
|
|
6
|
+
if (!apiKey) {
|
|
7
|
+
throw new Error('OPENAI_API_KEY environment variable is required');
|
|
8
|
+
}
|
|
9
|
+
const provider = gateway
|
|
10
|
+
? createOpenAICompatible({
|
|
11
|
+
baseURL: gateway,
|
|
12
|
+
name: 'openai',
|
|
13
|
+
headers: {
|
|
14
|
+
'Authorization': `Bearer ${apiKey}`,
|
|
15
|
+
'Content-Type': 'application/json'
|
|
16
|
+
}
|
|
17
|
+
})
|
|
18
|
+
: openai;
|
|
19
|
+
return (model, options) => provider(model, { structuredOutputs: options?.structuredOutputs ?? false });
|
|
20
|
+
}
|
|
21
|
+
//# sourceMappingURL=config.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/providers/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAA;AACvC,OAAO,EAAE,sBAAsB,EAAE,MAAM,2BAA2B,CAAA;AAGlE,MAAM,UAAU,WAAW;IACzB,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,CAAA;IACtC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAA;IAEzC,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAA;IACpE,CAAC;IAED,MAAM,QAAQ,GAAG,OAAO;QACtB,CAAC,CAAC,sBAAsB,CAAC;YACrB,OAAO,EAAE,OAAO;YAChB,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE;gBACP,eAAe,EAAE,UAAU,MAAM,EAAE;gBACnC,cAAc,EAAE,kBAAkB;aACnC;SACyC,CAAC;QAC/C,CAAC,CAAC,MAAM,CAAA;IAEV,OAAO,CAAC,KAAa,EAAE,OAAyC,EAAE,EAAE,CAClE,QAAQ,CAAC,KAAK,EAAE,EAAE,iBAAiB,EAAE,OAAO,EAAE,iBAAiB,IAAI,KAAK,EAAE,CAAC,CAAA;AAC/E,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.test.d.ts","sourceRoot":"","sources":["../../src/providers/config.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
import { describe, expect, it, beforeEach, afterEach } from 'vitest';
|
|
2
|
+
import { getProvider } from './config';
|
|
3
|
+
describe('Provider Configuration', () => {
|
|
4
|
+
const originalEnv = process.env;
|
|
5
|
+
beforeEach(() => {
|
|
6
|
+
process.env = { ...originalEnv };
|
|
7
|
+
process.env.OPENAI_API_KEY = 'test-key';
|
|
8
|
+
});
|
|
9
|
+
afterEach(() => {
|
|
10
|
+
process.env = originalEnv;
|
|
11
|
+
});
|
|
12
|
+
it('should throw error when API key is missing', () => {
|
|
13
|
+
delete process.env.OPENAI_API_KEY;
|
|
14
|
+
expect(() => getProvider()).toThrow('OPENAI_API_KEY environment variable is required');
|
|
15
|
+
});
|
|
16
|
+
it('should use custom gateway when AI_GATEWAY is set', () => {
|
|
17
|
+
process.env.AI_GATEWAY = 'https://custom-gateway.com';
|
|
18
|
+
const provider = getProvider();
|
|
19
|
+
expect(provider).toBeDefined();
|
|
20
|
+
});
|
|
21
|
+
it('should use default OpenAI when AI_GATEWAY is not set', () => {
|
|
22
|
+
delete process.env.AI_GATEWAY;
|
|
23
|
+
const provider = getProvider();
|
|
24
|
+
expect(provider).toBeDefined();
|
|
25
|
+
});
|
|
26
|
+
it('should return a function that accepts model and options', () => {
|
|
27
|
+
const provider = getProvider();
|
|
28
|
+
const modelFn = provider('gpt-4o-mini');
|
|
29
|
+
expect(modelFn).toBeDefined();
|
|
30
|
+
});
|
|
31
|
+
it('should handle structured outputs option', () => {
|
|
32
|
+
const provider = getProvider();
|
|
33
|
+
const modelFn = provider('gpt-4o-mini', { structuredOutputs: true });
|
|
34
|
+
expect(modelFn).toBeDefined();
|
|
35
|
+
});
|
|
36
|
+
});
|
|
37
|
+
//# sourceMappingURL=config.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.test.js","sourceRoot":"","sources":["../../src/providers/config.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAA;AACpE,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAA;AAEtC,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAA;IAE/B,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,CAAC,GAAG,GAAG,EAAE,GAAG,WAAW,EAAE,CAAA;QAChC,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAA;IACzC,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,CAAC,GAAG,GAAG,WAAW,CAAA;IAC3B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,CAAA;QACjC,MAAM,CAAC,GAAG,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC,OAAO,CAAC,iDAAiD,CAAC,CAAA;IACxF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,OAAO,CAAC,GAAG,CAAC,UAAU,GAAG,4BAA4B,CAAA;QACrD,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAA;QAC9B,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;IAChC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,OAAO,OAAO,CAAC,GAAG,CAAC,UAAU,CAAA;QAC7B,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAA;QAC9B,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;IAChC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAA;QAC9B,MAAM,OAAO,GAAG,QAAQ,CAAC,aAAa,CAAC,CAAA;QACvC,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAA;IAC/B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAA;QAC9B,MAAM,OAAO,GAAG,QAAQ,CAAC,aAAa,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAA;QACpE,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAA;IAC/B,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../src/proxy/constants.ts"],"names":[],"mappings":"AAEA,eAAO,MAAM,qBAAqB;;CAExB,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../src/proxy/constants.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAA;AAE5C,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACnC,KAAK,EAAE,MAAM,CAAC,IAAI;CACV,CAAA"}
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
import { SimpleSchema, AIFunctionOptions } from '@/types';
|
|
2
|
+
import { AIFunction } from './types';
|
|
3
|
+
export declare function createAIFunction<T extends SimpleSchema>(functionName: string, schema: T, options?: AIFunctionOptions): AIFunction<T>;
|
|
4
|
+
//# sourceMappingURL=create-function.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"create-function.d.ts","sourceRoot":"","sources":["../../src/proxy/create-function.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAA;AAGzD,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAA;AAEpC,wBAAgB,gBAAgB,CAAC,CAAC,SAAS,YAAY,EACrD,YAAY,EAAE,MAAM,EACpB,MAAM,EAAE,CAAC,EACT,OAAO,GAAE,iBAAsB,GAC9B,UAAU,CAAC,CAAC,CAAC,CAqBf"}
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
import { generateObject, streamObject } from '@/index';
|
|
2
|
+
import { DEFAULT_PROXY_OPTIONS } from './constants';
|
|
3
|
+
import { generatePrompt } from './prompt';
|
|
4
|
+
export function createAIFunction(functionName, schema, options = {}) {
|
|
5
|
+
return async function (input) {
|
|
6
|
+
const { seed, ...restInput } = input;
|
|
7
|
+
const prompt = generatePrompt(functionName, restInput);
|
|
8
|
+
const baseOptions = {
|
|
9
|
+
schema,
|
|
10
|
+
prompt,
|
|
11
|
+
model: options.model || DEFAULT_PROXY_OPTIONS.model,
|
|
12
|
+
system: options.system,
|
|
13
|
+
seed: parseInt(seed || '1', 10) // Convert seed to integer
|
|
14
|
+
};
|
|
15
|
+
if (options.stream) {
|
|
16
|
+
return streamObject(baseOptions);
|
|
17
|
+
}
|
|
18
|
+
return generateObject({
|
|
19
|
+
...baseOptions,
|
|
20
|
+
output: 'single'
|
|
21
|
+
});
|
|
22
|
+
};
|
|
23
|
+
}
|
|
24
|
+
//# sourceMappingURL=create-function.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"create-function.js","sourceRoot":"","sources":["../../src/proxy/create-function.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,SAAS,CAAA;AAEtD,OAAO,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAA;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAA;AAGzC,MAAM,UAAU,gBAAgB,CAC9B,YAAoB,EACpB,MAAS,EACT,UAA6B,EAAE;IAE/B,OAAO,KAAK,WAAU,KAAU;QAC9B,MAAM,EAAE,IAAI,EAAE,GAAG,SAAS,EAAE,GAAG,KAAK,CAAA;QACpC,MAAM,MAAM,GAAG,cAAc,CAAC,YAAY,EAAE,SAAS,CAAC,CAAA;QACtD,MAAM,WAAW,GAAG;YAClB,MAAM;YACN,MAAM;YACN,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,qBAAqB,CAAC,KAAK;YACnD,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,GAAG,EAAE,EAAE,CAAC,CAAC,0BAA0B;SAC3D,CAAA;QAED,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,OAAO,YAAY,CAAC,WAAW,CAAC,CAAA;QAClC,CAAC;QAED,OAAO,cAAc,CAAC;YACpB,GAAG,WAAW;YACd,MAAM,EAAE,QAAQ;SACjB,CAAC,CAAA;IACJ,CAAC,CAAA;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"create-proxy.d.ts","sourceRoot":"","sources":["../../src/proxy/create-proxy.ts"],"names":[],"mappings":"AAGA,wBAAgB,aAAa,OAQ5B"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { createAIFunction } from './create-function';
|
|
2
|
+
export function createAIProxy() {
|
|
3
|
+
return new Proxy({}, {
|
|
4
|
+
get: (_, functionName) => {
|
|
5
|
+
return (schema, options = {}) => {
|
|
6
|
+
return createAIFunction(functionName, schema, options);
|
|
7
|
+
};
|
|
8
|
+
}
|
|
9
|
+
});
|
|
10
|
+
}
|
|
11
|
+
//# sourceMappingURL=create-proxy.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"create-proxy.js","sourceRoot":"","sources":["../../src/proxy/create-proxy.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAA;AAEpD,MAAM,UAAU,aAAa;IAC3B,OAAO,IAAI,KAAK,CAAC,EAAE,EAAE;QACnB,GAAG,EAAE,CAAC,CAAC,EAAE,YAAoB,EAAE,EAAE;YAC/B,OAAO,CAAyB,MAAS,EAAE,UAA6B,EAAE,EAAE,EAAE;gBAC5E,OAAO,gBAAgB,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;YACxD,CAAC,CAAA;QACH,CAAC;KACF,CAAC,CAAA;AACJ,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import { SimpleSchema, AIFunctionOptions } from '@/types';
|
|
2
|
+
export declare function createAIFunction<T extends SimpleSchema>(functionName: string, schema: T, options?: AIFunctionOptions): (input: any) => Promise<import("ai").GenerateObjectResult<{
|
|
3
|
+
[x: string]: any;
|
|
4
|
+
}> | import("ai").StreamObjectResult<{
|
|
5
|
+
[x: string]: any;
|
|
6
|
+
}, {
|
|
7
|
+
[x: string]: any;
|
|
8
|
+
}, never>>;
|
|
9
|
+
//# sourceMappingURL=function-generator.d.ts.map
|