UNPKG

ai

Version:

AI SDK by Vercel - The AI Toolkit for TypeScript and JavaScript

121 lines (114 loc) 3.74 kB
"use strict"; var __defProp = Object.defineProperty; var __getOwnPropDesc = Object.getOwnPropertyDescriptor; var __getOwnPropNames = Object.getOwnPropertyNames; var __hasOwnProp = Object.prototype.hasOwnProperty; var __export = (target, all) => { for (var name in all) __defProp(target, name, { get: all[name], enumerable: true }); }; var __copyProps = (to, from, except, desc) => { if (from && typeof from === "object" || typeof from === "function") { for (let key of __getOwnPropNames(from)) if (!__hasOwnProp.call(to, key) && key !== except) __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable }); } return to; }; var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod); // test/index.ts var test_exports = {}; __export(test_exports, { MockEmbeddingModelV1: () => MockEmbeddingModelV1, MockLanguageModelV1: () => MockLanguageModelV1, convertArrayToReadableStream: () => import_test.convertArrayToReadableStream, mockId: () => import_test.mockId, mockValues: () => mockValues, simulateReadableStream: () => simulateReadableStream2 }); module.exports = __toCommonJS(test_exports); var import_test = require("@ai-sdk/provider-utils/test"); // core/test/not-implemented.ts function notImplemented() { throw new Error("Not implemented"); } // core/test/mock-embedding-model-v1.ts var MockEmbeddingModelV1 = class { constructor({ provider = "mock-provider", modelId = "mock-model-id", maxEmbeddingsPerCall = 1, supportsParallelCalls = false, doEmbed = notImplemented } = {}) { this.specificationVersion = "v1"; this.provider = provider; this.modelId = modelId; this.maxEmbeddingsPerCall = maxEmbeddingsPerCall != null ? maxEmbeddingsPerCall : void 0; this.supportsParallelCalls = supportsParallelCalls; this.doEmbed = doEmbed; } }; // core/test/mock-language-model-v1.ts var MockLanguageModelV1 = class { constructor({ provider = "mock-provider", modelId = "mock-model-id", supportsUrl = void 0, doGenerate = notImplemented, doStream = notImplemented, defaultObjectGenerationMode = void 0, supportsStructuredOutputs = void 0 } = {}) { this.specificationVersion = "v1"; this.provider = provider; this.modelId = modelId; this.doGenerate = doGenerate; this.doStream = doStream; this.supportsUrl = supportsUrl; this.defaultObjectGenerationMode = defaultObjectGenerationMode; this.supportsStructuredOutputs = supportsStructuredOutputs; } }; // core/test/mock-values.ts function mockValues(...values) { let counter = 0; return () => { var _a; return (_a = values[counter++]) != null ? _a : values[values.length - 1]; }; } // core/util/simulate-readable-stream.ts var import_provider_utils = require("@ai-sdk/provider-utils"); function simulateReadableStream({ chunks, initialDelayInMs = 0, chunkDelayInMs = 0, _internal }) { var _a; const delay = (_a = _internal == null ? void 0 : _internal.delay) != null ? _a : import_provider_utils.delay; let index = 0; return new ReadableStream({ async pull(controller) { if (index < chunks.length) { await delay(index === 0 ? initialDelayInMs : chunkDelayInMs); controller.enqueue(chunks[index++]); } else { controller.close(); } } }); } // test/index.ts var simulateReadableStream2 = simulateReadableStream; // Annotate the CommonJS export names for ESM import in node: 0 && (module.exports = { MockEmbeddingModelV1, MockLanguageModelV1, convertArrayToReadableStream, mockId, mockValues, simulateReadableStream }); //# sourceMappingURL=index.js.map