@llumiverse/drivers
Version:
LLM driver implementations. Currently supported are: openai, huggingface, bedrock, replicate.
109 lines • 4.42 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __exportStar = (this && this.__exportStar) || function(m, exports) {
for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.TestDriver = exports.TestDriverModels = void 0;
const core_1 = require("@llumiverse/core");
const TestErrorCompletionStream_js_1 = require("./TestErrorCompletionStream.js");
const TestValidationErrorCompletionStream_js_1 = require("./TestValidationErrorCompletionStream.js");
const utils_js_1 = require("./utils.js");
__exportStar(require("./TestErrorCompletionStream.js"), exports);
__exportStar(require("./TestValidationErrorCompletionStream.js"), exports);
var TestDriverModels;
(function (TestDriverModels) {
TestDriverModels["executionError"] = "execution-error";
TestDriverModels["validationError"] = "validation-error";
})(TestDriverModels || (exports.TestDriverModels = TestDriverModels = {}));
class TestDriver {
provider = "test";
createTrainingPrompt() {
throw new Error("Method not implemented.");
}
startTraining() {
throw new Error("Method not implemented.");
}
cancelTraining() {
throw new Error("Method not implemented.");
}
getTrainingJob(_jobId) {
throw new Error("Method not implemented.");
}
async createPrompt(segments, _opts) {
return segments;
}
execute(segments, options) {
switch (options.model) {
case TestDriverModels.executionError:
return this.executeError(segments, options);
case TestDriverModels.validationError:
return this.executeValidationError(segments, options);
default:
(0, utils_js_1.throwError)("[test driver] Unknown model: " + options.model, segments);
}
}
async stream(segments, options) {
switch (options.model) {
case TestDriverModels.executionError:
return new TestErrorCompletionStream_js_1.TestErrorCompletionStream(segments, options);
case TestDriverModels.validationError:
return new TestValidationErrorCompletionStream_js_1.TestValidationErrorCompletionStream(segments, options);
default:
(0, utils_js_1.throwError)("[test driver] Unknown model: " + options.model, segments);
}
}
async listTrainableModels() {
return [];
}
async listModels() {
return [
{
id: TestDriverModels.executionError,
name: "Execution Error",
type: core_1.ModelType.Test,
provider: this.provider,
status: core_1.AIModelStatus.Available,
description: "Test execution errors",
tags: [],
},
{
id: TestDriverModels.validationError,
name: "Validation Error",
type: core_1.ModelType.Test,
provider: this.provider,
status: core_1.AIModelStatus.Available,
description: "Test validation errors",
tags: [],
},
];
}
validateConnection() {
throw new Error("Method not implemented.");
}
generateEmbeddings() {
throw new Error("Method not implemented.");
}
// ============== execution error ==================
async executeError(segments, _options) {
await (0, utils_js_1.sleep)(1000);
(0, utils_js_1.throwError)("Testing stream completion error.", segments);
}
// ============== validation error ==================
async executeValidationError(segments, _options) {
await (0, utils_js_1.sleep)(3000);
return (0, utils_js_1.createValidationErrorCompletion)(segments);
}
}
exports.TestDriver = TestDriver;
//# sourceMappingURL=index.js.map