ai
Version:
AI SDK by Vercel - The AI Toolkit for TypeScript and JavaScript
121 lines (114 loc) • 3.74 kB
JavaScript
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
;