@llumiverse/drivers 0.8.0

package/lib/cjs/test/index.js

@@ -0,0 +1,109 @@
+"use strict";
+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.");
+    }
+    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

package/lib/cjs/test/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/test/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,2CAA+K;AAC/K,iFAA2E;AAC3E,qGAA+F;AAC/F,yCAAgF;AAEhF,iEAA+C;AAC/C,2EAAyD;AAEzD,IAAY,gBAGX;AAHD,WAAY,gBAAgB;IACxB,sDAAkC,CAAA;IAClC,wDAAoC,CAAA;AACxC,CAAC,EAHW,gBAAgB,gCAAhB,gBAAgB,QAG3B;AAED,MAAa,UAAU;IACnB,QAAQ,GAAG,MAAM,CAAC;IAElB,oBAAoB;QAChB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAED,aAAa;QACT,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAED,cAAc;QACV,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAED,cAAc,CAAC,MAAc;QACzB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAED,YAAY,CAAC,QAAyB,EAAE,KAAoB;QACxD,OAAO,QAAQ,CAAC;IACpB,CAAC;IACD,OAAO,CAAC,QAAyB,EAAE,OAAyB;QACxD,QAAQ,OAAO,CAAC,KAAK,EAAE,CAAC;YACpB,KAAK,gBAAgB,CAAC,cAAc;gBAChC,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAChD,KAAK,gBAAgB,CAAC,eAAe;gBACjC,OAAO,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC1D;gBACI,IAAA,qBAAU,EAAC,+BAA+B,GAAG,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAA;QAC7E,CAAC;IACL,CAAC;IACD,KAAK,CAAC,MAAM,CAAC,QAAyB,EAAE,OAAyB;QAC7D,QAAQ,OAAO,CAAC,KAAK,EAAE,CAAC;YACpB,KAAK,gBAAgB,CAAC,cAAc;gBAChC,OAAO,IAAI,wDAAyB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC5D,KAAK,gBAAgB,CAAC,eAAe;gBACjC,OAAO,IAAI,4EAAmC,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACtE;gBACI,IAAA,qBAAU,EAAC,+BAA+B,GAAG,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAA;QAC7E,CAAC;IACL,CAAC;IAED,KAAK,CAAC,mBAAmB;QACrB,OAAO,EAAE,CAAC;IACd,CAAC;IAED,KAAK,CAAC,UAAU;QACZ,OAAO;YACH;gBACI,EAAE,EAAE,gBAAgB,CAAC,cAAc;gBACnC,IAAI,EAAE,iBAAiB;gBACvB,IAAI,EAAE,gBAAS,CAAC,IAAI;gBACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,MAAM,EAAE,oBAAa,CAAC,SAAS;gBAC/B,WAAW,EAAE,uBAAuB;gBACpC,IAAI,EAAE,EAAE;aACX;YACD;gBACI,EAAE,EAAE,gBAAgB,CAAC,eAAe;gBACpC,IAAI,EAAE,kBAAkB;gBACxB,IAAI,EAAE,gBAAS,CAAC,IAAI;gBACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,MAAM,EAAE,oBAAa,CAAC,SAAS;gBAC/B,WAAW,EAAE,wBAAwB;gBACrC,IAAI,EAAE,EAAE;aACX;SACJ,CAAA;IACL,CAAC;IACD,kBAAkB;QACd,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IACD,kBAAkB;QACd,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAED,oDAAoD;IACpD,KAAK,CAAC,YAAY,CAAC,QAAyB,EAAE,QAA0B;QACpE,MAAM,IAAA,gBAAK,EAAC,IAAI,CAAC,CAAC;QAClB,IAAA,qBAAU,EAAC,kCAAkC,EAAE,QAAQ,CAAC,CAAC;IAC7D,CAAC;IACD,qDAAqD;IACrD,KAAK,CAAC,sBAAsB,CAAC,QAAyB,EAAE,QAA0B;QAC9E,MAAM,IAAA,gBAAK,EAAC,IAAI,CAAC,CAAC;QAClB,OAAO,IAAA,0CAA+B,EAAC,QAAQ,CAAC,CAAC;IACrD,CAAC;CAEJ;AAvFD,gCAuFC"}

package/lib/cjs/test/utils.js

@@ -0,0 +1,31 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.sleep = exports.createValidationErrorCompletion = exports.throwError = void 0;
+function throwError(message, prompt) {
+    const err = new Error(message);
+    err.prompt = prompt;
+    throw err;
+}
+exports.throwError = throwError;
+function createValidationErrorCompletion(segments) {
+    return {
+        result: "An invalid result",
+        prompt: segments,
+        execution_time: 3000,
+        error: {
+            code: "validation_error",
+            message: "Result cannot be validated!",
+        },
+        token_usage: {
+            result: 10,
+            prompt: 10,
+            total: 20,
+        },
+    };
+}
+exports.createValidationErrorCompletion = createValidationErrorCompletion;
+function sleep(ms) {
+    return new Promise(resolve => setTimeout(resolve, ms));
+}
+exports.sleep = sleep;
+//# sourceMappingURL=utils.js.map

package/lib/cjs/test/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/test/utils.ts"],"names":[],"mappings":";;;AAEA,SAAgB,UAAU,CAAC,OAAe,EAAE,MAAuB;IAC/D,MAAM,GAAG,GAAG,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC;IAC9B,GAAW,CAAC,MAAM,GAAG,MAAM,CAAC;IAC7B,MAAM,GAAG,CAAC;AACd,CAAC;AAJD,gCAIC;AAED,SAAgB,+BAA+B,CAAC,QAAyB;IACrE,OAAO;QACH,MAAM,EAAE,mBAAmB;QAC3B,MAAM,EAAE,QAAQ;QAChB,cAAc,EAAE,IAAI;QACpB,KAAK,EAAE;YACH,IAAI,EAAE,kBAAkB;YACxB,OAAO,EAAE,6BAA6B;SACzC;QACD,WAAW,EAAE;YACT,MAAM,EAAE,EAAE;YACV,MAAM,EAAE,EAAE;YACV,KAAK,EAAE,EAAE;SACZ;KACkC,CAAC;AAC5C,CAAC;AAfD,0EAeC;AAED,SAAgB,KAAK,CAAC,EAAU;IAC5B,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC;AAFD,sBAEC"}

package/lib/cjs/togetherai/index.js

@@ -0,0 +1,92 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TogetherAIDriver = void 0;
+const core_1 = require("@llumiverse/core");
+const api_fetch_client_1 = require("api-fetch-client");
+class TogetherAIDriver extends core_1.AbstractDriver {
+    provider;
+    apiKey;
+    defaultFormat;
+    fetchClient;
+    constructor(options) {
+        super(options);
+        this.provider = "togetherai";
+        this.defaultFormat = core_1.PromptFormats.genericTextLLM;
+        this.apiKey = options.apiKey;
+        this.fetchClient = new api_fetch_client_1.FetchClient('https://api.together.xyz').withHeaders({
+            authorization: `Bearer ${this.apiKey}`
+        });
+    }
+    async requestCompletion(prompt, options) {
+        const res = await this.fetchClient.post('/v1/completions', {
+            payload: {
+                model: options.model,
+                prompt: prompt,
+                max_tokens: options.max_tokens ?? 1024,
+                temperature: options.temperature ?? 0.7,
+            }
+        });
+        const text = res.choices[0]?.text ?? '';
+        const usage = res.usage || {};
+        return {
+            result: text,
+            token_usage: {
+                prompt: usage.prompt_tokens,
+                result: usage.completion_tokens,
+                total: usage.total_tokens,
+            },
+        };
+    }
+    async requestCompletionStream(prompt, options) {
+        const stream = await this.fetchClient.post('/v1/completions', {
+            payload: {
+                model: options.model,
+                prompt: prompt,
+                max_tokens: options.max_tokens ?? 1024,
+                temperature: options.temperature ?? 0.7,
+                stream: true,
+            },
+            reader: 'sse'
+        });
+        return stream.pipeThrough(new TransformStream({
+            transform(event, controller) {
+                if (event.type === 'event' && event.data && event.data !== '[DONE]') {
+                    try {
+                        const data = JSON.parse(event.data);
+                        controller.enqueue(data.choices[0]?.text ?? '');
+                    }
+                    catch (err) {
+                        // double check for the last event whicb is not a JSON - at this time togetherai returrns the string [DONE]
+                        // do nothing - happens if data is not a JSON - the last event data is the [DONE] string
+                    }
+                }
+            }
+        }));
+    }
+    async listModels() {
+        const models = await this.fetchClient.get("/models/info");
+        //        logObject('#### LIST MODELS RESULT IS', models[0]);
+        const aimodels = models.map(m => {
+            return {
+                id: m.name,
+                name: m.display_name,
+                description: m.description,
+                provider: this.provider,
+                formats: [core_1.PromptFormats.genericTextLLM],
+            };
+        });
+        return aimodels;
+    }
+    listTrainableModels() {
+        throw new Error("Method not implemented.");
+    }
+    validateConnection() {
+        throw new Error("Method not implemented.");
+    }
+    //@ts-ignore
+    generateEmbeddings(content, model) {
+        throw new Error("Method not implemented.");
+    }
+}
+exports.TogetherAIDriver = TogetherAIDriver;
+//# sourceMappingURL=index.js.map

package/lib/cjs/togetherai/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/togetherai/index.ts"],"names":[],"mappings":";;;AAAA,2CAAuH;AACvH,uDAAgE;AAOhE,MAAa,gBAAiB,SAAQ,qBAA+C;IACjF,QAAQ,CAAS;IACjB,MAAM,CAAS;IACf,aAAa,CAAgB;IAC7B,WAAW,CAAc;IAEzB,YAAY,OAAgC;QACxC,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,QAAQ,GAAG,YAAY,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG,oBAAa,CAAC,cAAc,CAAC;QAClD,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,WAAW,GAAG,IAAI,8BAAW,CAAC,0BAA0B,CAAC,CAAC,WAAW,CAAC;YACvE,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;SACzC,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,MAAc,EAAE,OAAyB;QAC7D,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,iBAAiB,EAAE;YACvD,OAAO,EAAE;gBACL,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,MAAM,EAAE,MAAM;gBACd,UAAU,EAAE,OAAO,CAAC,UAAU,IAAI,IAAI;gBACtC,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;aAC1C;SACJ,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;QACxC,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;QAC9B,OAAO;YACH,MAAM,EAAE,IAAI;YACZ,WAAW,EAAE;gBACT,MAAM,EAAE,KAAK,CAAC,aAAa;gBAC3B,MAAM,EAAE,KAAK,CAAC,iBAAiB;gBAC/B,KAAK,EAAE,KAAK,CAAC,YAAY;aAC5B;SACJ,CAAA;IACL,CAAC;IAED,KAAK,CAAC,uBAAuB,CAAC,MAAc,EAAE,OAAyB;QAEnE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,iBAAiB,EAAE;YAC1D,OAAO,EAAE;gBACL,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,MAAM,EAAE,MAAM;gBACd,UAAU,EAAE,OAAO,CAAC,UAAU,IAAI,IAAI;gBACtC,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;gBACvC,MAAM,EAAE,IAAI;aACf;YACD,MAAM,EAAE,KAAK;SAChB,CAAC,CAAA;QAEF,OAAO,MAAM,CAAC,WAAW,CAAC,IAAI,eAAe,CAA0B;YACnE,SAAS,CAAC,KAAsB,EAAE,UAAU;gBACxC,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBAClE,IAAI,CAAC;wBACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBACpC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;oBACpD,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACX,2GAA2G;wBAC3G,wFAAwF;oBAC5F,CAAC;gBACL,CAAC;YACL,CAAC;SACJ,CAAC,CAAC,CAAC;IAER,CAAC;IAED,KAAK,CAAC,UAAU;QACZ,MAAM,MAAM,GAAwB,MAAM,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAC/E,6DAA6D;QAE7D,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;YAC5B,OAAO;gBACH,EAAE,EAAE,CAAC,CAAC,IAAI;gBACV,IAAI,EAAE,CAAC,CAAC,YAAY;gBACpB,WAAW,EAAE,CAAC,CAAC,WAAW;gBAC1B,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,OAAO,EAAE,CAAC,oBAAa,CAAC,cAAc,CAAC;aAC1C,CAAA;QACL,CAAC,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IAEpB,CAAC;IAED,mBAAmB;QACf,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IACD,kBAAkB;QACd,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IACD,YAAY;IACZ,kBAAkB,CAAC,OAAe,EAAE,KAA0B;QAC1D,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;CAEJ;AAhGD,4CAgGC"}

package/lib/cjs/togetherai/interfaces.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=interfaces.js.map

package/lib/cjs/togetherai/interfaces.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"interfaces.js","sourceRoot":"","sources":["../../../src/togetherai/interfaces.ts"],"names":[],"mappings":""}

package/lib/cjs/vertexai/debug.js

@@ -0,0 +1,13 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.logObject = void 0;
+const util_1 = __importDefault(require("util"));
+function logObject(prefix, obj) {
+    const fullObj = util_1.default.inspect(obj, { showHidden: false, depth: null, colors: true });
+    console.log(prefix, fullObj);
+}
+exports.logObject = logObject;
+//# sourceMappingURL=debug.js.map

package/lib/cjs/vertexai/debug.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"debug.js","sourceRoot":"","sources":["../../../src/vertexai/debug.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAwB;AAExB,SAAgB,SAAS,CAAC,MAAc,EAAE,GAAQ;IAC9C,MAAM,OAAO,GAAG,cAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IACpF,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;AAChC,CAAC;AAHD,8BAGC"}

package/lib/cjs/vertexai/index.js

@@ -0,0 +1,80 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.VertexAIDriver = void 0;
+//import { v1 } from "@google-cloud/aiplatform";
+const vertexai_1 = require("@google-cloud/vertexai");
+const core_1 = require("@llumiverse/core");
+const api_fetch_client_1 = require("api-fetch-client");
+const models_js_1 = require("./models.js");
+class VertexAIDriver extends core_1.AbstractDriver {
+    provider = core_1.BuiltinProviders.vertexai;
+    defaultFormat = core_1.PromptFormats.genericTextLLM;
+    //aiplatform: v1.ModelServiceClient;
+    vertexai;
+    fetchClient;
+    constructor(options) {
+        super(options);
+        //this.aiplatform = new v1.ModelServiceClient();
+        this.vertexai = new vertexai_1.VertexAI({
+            project: this.options.project,
+            location: this.options.region,
+        });
+        this.fetchClient = createFetchClient({
+            region: this.options.region,
+            project: this.options.project,
+        }).withAuthCallback(async () => {
+            const token = await this.vertexai.preview.token;
+            return `Bearer ${token}`;
+        });
+        // this.aiplatform = new v1.ModelServiceClient({
+        //     projectId: this.options.project,
+        //     apiEndpoint: `${this.options.region}-${API_BASE_PATH}`,
+        // });
+    }
+    canStream(options) {
+        return Promise.resolve((0, models_js_1.getModelDefinition)(options.model).model.canStream === true);
+    }
+    createPrompt(segments, options) {
+        return (0, models_js_1.getModelDefinition)(options.model).createPrompt(this, segments, options);
+    }
+    async requestCompletion(prompt, options) {
+        return (0, models_js_1.getModelDefinition)(options.model).requestCompletion(this, prompt, options);
+    }
+    async requestCompletionStream(prompt, options) {
+        return (0, models_js_1.getModelDefinition)(options.model).requestCompletionStream(this, prompt, options);
+    }
+    async listModels(_params) {
+        return models_js_1.BuiltinModels;
+        // try {
+        //     const response = await this.fetchClient.get('/publishers/google/models/gemini-pro');
+        //     console.log('>>>>>>>>', response);
+        // } catch (err: any) {
+        //     console.error('+++++VETREX ERROR++++++', err);
+        //     throw err;
+        // }
+        // TODO uncomment this to use apiplatform instead of the fetch client
+        // const response = await this.aiplatform.listModels({
+        //     parent: `projects/${this.options.project}/locations/${this.options.region}`,
+        // });
+        return []; //TODO
+    }
+    listTrainableModels() {
+        throw new Error("Method not implemented.");
+    }
+    validateConnection() {
+        throw new Error("Method not implemented.");
+    }
+    generateEmbeddings(_content, _model) {
+        throw new Error("Method not implemented.");
+    }
+}
+exports.VertexAIDriver = VertexAIDriver;
+//'us-central1-aiplatform.googleapis.com',
+const API_BASE_PATH = 'aiplatform.googleapis.com';
+function createFetchClient({ region, project, apiEndpoint, apiVersion = 'v1' }) {
+    const vertexBaseEndpoint = apiEndpoint ?? `${region}-${API_BASE_PATH}`;
+    return new api_fetch_client_1.FetchClient(`https://${vertexBaseEndpoint}/${apiVersion}/projects/${project}/locations/${region}`).withHeaders({
+        'Content-Type': 'application/json',
+    });
+}
+//# sourceMappingURL=index.js.map

package/lib/cjs/vertexai/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/vertexai/index.ts"],"names":[],"mappings":";;;AAAA,gDAAgD;AAChD,qDAA0E;AAC1E,2CAA2L;AAC3L,uDAA+C;AAC/C,2CAAgE;AAQhE,MAAa,cAAe,SAAQ,qBAA6D;IAC7F,QAAQ,GAAG,uBAAgB,CAAC,QAAQ,CAAC;IACrC,aAAa,GAAG,oBAAa,CAAC,cAAc,CAAC;IAE7C,oCAAoC;IACpC,QAAQ,CAAW;IACnB,WAAW,CAAc;IAEzB,YACI,OAA8B;QAE9B,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,gDAAgD;QAChD,IAAI,CAAC,QAAQ,GAAG,IAAI,mBAAQ,CAAC;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;YAC7B,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;SAChC,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,iBAAiB,CAAC;YACjC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;YAC3B,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;SAChC,CAAC,CAAC,gBAAgB,CAAC,KAAK,IAAI,EAAE;YAC3B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC;YAChD,OAAO,UAAU,KAAK,EAAE,CAAC;QAC7B,CAAC,CAAC,CAAC;QACH,gDAAgD;QAChD,uCAAuC;QACvC,8DAA8D;QAC9D,MAAM;IACV,CAAC;IAES,SAAS,CAAC,OAAyB;QACzC,OAAO,OAAO,CAAC,OAAO,CAAC,IAAA,8BAAkB,EAAC,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,SAAS,KAAK,IAAI,CAAC,CAAC;IACvF,CAAC;IAEM,YAAY,CAAC,QAAyB,EAAE,OAAsB;QACjE,OAAO,IAAA,8BAAkB,EAAC,OAAO,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;IACnF,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,MAA8B,EAAE,OAAyB;QAC7E,OAAO,IAAA,8BAAkB,EAAC,OAAO,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IACtF,CAAC;IACD,KAAK,CAAC,uBAAuB,CAAC,MAA8B,EAAE,OAAyB;QACnF,OAAO,IAAA,8BAAkB,EAAC,OAAO,CAAC,KAAK,CAAC,CAAC,uBAAuB,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IAC5F,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAA4B;QACzC,OAAO,yBAAa,CAAC;QACrB,QAAQ;QACR,2FAA2F;QAC3F,yCAAyC;QACzC,uBAAuB;QACvB,qDAAqD;QACrD,iBAAiB;QACjB,IAAI;QAEJ,qEAAqE;QACrE,sDAAsD;QACtD,mFAAmF;QACnF,MAAM;QAEN,OAAO,EAAE,CAAC,CAAC,MAAM;IACrB,CAAC;IACD,mBAAmB;QACf,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IACD,kBAAkB;QACd,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IACD,kBAAkB,CAAC,QAAgB,EAAE,MAA2B;QAC5D,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;CAEJ;AAxED,wCAwEC;AAED,0CAA0C;AAC1C,MAAM,aAAa,GAAG,2BAA2B,CAAC;AAClD,SAAS,iBAAiB,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,UAAU,GAAG,IAAI,EAK3E;IACG,MAAM,kBAAkB,GAAG,WAAW,IAAI,GAAG,MAAM,IAAI,aAAa,EAAE,CAAC;IACvE,OAAO,IAAI,8BAAW,CAAC,WAAW,kBAAkB,IAAI,UAAU,aAAa,OAAO,cAAc,MAAM,EAAE,CAAC,CAAC,WAAW,CAAC;QACtH,cAAc,EAAE,kBAAkB;KACrC,CAAC,CAAC;AACP,CAAC"}

package/lib/cjs/vertexai/models/codey-chat.js

@@ -0,0 +1,65 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CodeyChatDefinition = void 0;
+const core_1 = require("@llumiverse/core");
+const prompts_js_1 = require("../utils/prompts.js");
+const palm_model_base_js_1 = require("./palm-model-base.js");
+class CodeyChatDefinition extends palm_model_base_js_1.AbstractPalmModelDefinition {
+    versions = [];
+    model = {
+        id: "codechat-bison",
+        name: "Codey for Code Chat",
+        provider: "vertexai",
+        owner: "google",
+        type: core_1.ModelType.Chat,
+        canStream: true,
+    };
+    createNonStreamingPrompt(_driver, segments, opts) {
+        const system = [];
+        const messages = [];
+        const safety = [];
+        for (const segment of segments) {
+            switch (segment.role) {
+                case core_1.PromptRole.user:
+                    messages.push({ author: 'user', content: segment.content });
+                    break;
+                case core_1.PromptRole.assistant:
+                    messages.push({ author: 'assistant', content: segment.content });
+                    break;
+                case core_1.PromptRole.system:
+                    system.push(segment.content);
+                    break;
+                case core_1.PromptRole.safety:
+                    safety.push(segment.content);
+                    break;
+            }
+        }
+        if (opts.resultSchema) {
+            safety.push((0, prompts_js_1.getJSONSafetyNotice)(opts.resultSchema));
+        }
+        const context = [];
+        if (system.length > 0) {
+            context.push(system.join('\n'));
+        }
+        if (safety.length > 0) {
+            context.push('IMPORTANT: ' + safety.join('\n'));
+        }
+        return {
+            instances: [{
+                    context: context.length > 0 ? context.join('\n') : undefined,
+                    messages
+                }],
+            parameters: {
+            // put defauilts here
+            }
+        };
+    }
+    extractContentFromResponse(response) {
+        return response.predictions[0].candidates[0].content ?? '';
+    }
+    extractContentFromResponseChunk(chunk) {
+        return chunk.outputs[0]?.structVal.candidates.listVal[0].structVal.content.stringVal || '';
+    }
+}
+exports.CodeyChatDefinition = CodeyChatDefinition;
+//# sourceMappingURL=codey-chat.js.map

package/lib/cjs/vertexai/models/codey-chat.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codey-chat.js","sourceRoot":"","sources":["../../../../src/vertexai/models/codey-chat.ts"],"names":[],"mappings":";;;AAAA,2CAAgG;AAEhG,oDAA0D;AAC1D,6DAAsI;AAgDtI,MAAa,mBAAoB,SAAQ,gDAAsE;IAC3G,QAAQ,GAAa,EAAE,CAAC;IACxB,KAAK,GAAY;QACb,EAAE,EAAE,gBAAgB;QACpB,IAAI,EAAE,qBAAqB;QAC3B,QAAQ,EAAE,UAAU;QACpB,KAAK,EAAE,QAAQ;QACf,IAAI,EAAE,gBAAS,CAAC,IAAI;QACpB,SAAS,EAAE,IAAI;KAClB,CAAA;IAED,wBAAwB,CAAC,OAAuB,EAAE,QAAyB,EAAE,IAAmB;QAC5F,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,MAAM,QAAQ,GAAuB,EAAE,CAAC;QACxC,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC7B,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;gBACnB,KAAK,iBAAU,CAAC,IAAI;oBAChB,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;oBAC5D,MAAM;gBACV,KAAK,iBAAU,CAAC,SAAS;oBACrB,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;oBACjE,MAAM;gBACV,KAAK,iBAAU,CAAC,MAAM;oBAClB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;oBAC7B,MAAM;gBACV,KAAK,iBAAU,CAAC,MAAM;oBAClB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;oBAC7B,MAAM;YACd,CAAC;QACL,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACpB,MAAM,CAAC,IAAI,CAAC,IAAA,gCAAmB,EAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;QACxD,CAAC;QAED,MAAM,OAAO,GAAG,EAAE,CAAA;QAClB,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpB,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACpC,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpB,OAAO,CAAC,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACpD,CAAC;QAED,OAAO;YACH,SAAS,EAAE,CAAC;oBACR,OAAO,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;oBAC5D,QAAQ;iBACX,CAAC;YACF,UAAU,EAAE;YACR,qBAAqB;aACxB;SACe,CAAC;IACzB,CAAC;IAED,0BAA0B,CAAC,QAA2B;QAClD,OAAO,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC;IAC/D,CAAC;IAED,+BAA+B,CAAC,KAAU;QACtC,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,IAAI,EAAE,CAAC;IAC/F,CAAC;CAEJ;AA/DD,kDA+DC"}

package/lib/cjs/vertexai/models/codey-text.js

@@ -0,0 +1,35 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CodeyTextDefinition = void 0;
+const core_1 = require("@llumiverse/core");
+const prompts_js_1 = require("../utils/prompts.js");
+const palm_model_base_js_1 = require("./palm-model-base.js");
+class CodeyTextDefinition extends palm_model_base_js_1.AbstractPalmModelDefinition {
+    versions = [];
+    model = {
+        id: "code-bison",
+        name: "Codey for Code Generation",
+        provider: "vertexai",
+        owner: "google",
+        type: core_1.ModelType.Text,
+        canStream: true,
+    };
+    createNonStreamingPrompt(_driver, segments, opts) {
+        return {
+            instances: [{
+                    prefix: (0, prompts_js_1.getPromptAsText)(segments, opts)
+                }],
+            parameters: {
+            // put defauilts here
+            }
+        };
+    }
+    extractContentFromResponse(response) {
+        return response.predictions[0].content ?? '';
+    }
+    extractContentFromResponseChunk(chunk) {
+        return chunk.outputs[0]?.structVal.content.stringVal || '';
+    }
+}
+exports.CodeyTextDefinition = CodeyTextDefinition;
+//# sourceMappingURL=codey-text.js.map

package/lib/cjs/vertexai/models/codey-text.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codey-text.js","sourceRoot":"","sources":["../../../../src/vertexai/models/codey-text.ts"],"names":[],"mappings":";;;AAAA,2CAAoF;AAEpF,oDAAsD;AACtD,6DAAsI;AAmCtI,MAAa,mBAAoB,SAAQ,gDAAsE;IAC3G,QAAQ,GAAa,EAAE,CAAC;IACxB,KAAK,GAAY;QACb,EAAE,EAAE,YAAY;QAChB,IAAI,EAAE,2BAA2B;QACjC,QAAQ,EAAE,UAAU;QACpB,KAAK,EAAE,QAAQ;QACf,IAAI,EAAE,gBAAS,CAAC,IAAI;QACpB,SAAS,EAAE,IAAI;KAClB,CAAA;IAED,wBAAwB,CAAC,OAAuB,EAAE,QAAyB,EAAE,IAAmB;QAC5F,OAAO;YACH,SAAS,EAAE,CAAC;oBACR,MAAM,EAAE,IAAA,4BAAe,EAAC,QAAQ,EAAE,IAAI,CAAC;iBAC1C,CAAC;YACF,UAAU,EAAE;YACR,qBAAqB;aACxB;SACe,CAAC;IACzB,CAAC;IAED,0BAA0B,CAAC,QAA2B;QAClD,OAAO,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC;IACjD,CAAC;IAED,+BAA+B,CAAC,KAAU;QACtC,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,OAAO,CAAC,SAAS,IAAI,EAAE,CAAC;IAC/D,CAAC;CAEJ;AA9BD,kDA8BC"}

package/lib/cjs/vertexai/models/gemini.js

@@ -0,0 +1,140 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GeminiModelDefinition = void 0;
+const vertexai_1 = require("@google-cloud/vertexai");
+const core_1 = require("@llumiverse/core");
+const async_1 = require("@llumiverse/core/async");
+function getGenerativeModel(driver, options) {
+    return driver.vertexai.preview.getGenerativeModel({
+        model: options.model,
+        //TODO pass in the options
+        safety_settings: [{
+                category: vertexai_1.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+                threshold: vertexai_1.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE
+            }],
+        generation_config: {
+            temperature: options.temperature,
+            max_output_tokens: options.max_tokens
+        },
+    });
+}
+function collectTextParts(content) {
+    const out = [];
+    const parts = content.parts;
+    if (parts) {
+        for (const part of parts) {
+            if (part.text) {
+                out.push(part.text);
+            }
+        }
+    }
+    return out.join('\n');
+}
+class GeminiModelDefinition {
+    model = {
+        id: "gemini-pro",
+        name: "Gemini Pro",
+        provider: "vertexai",
+        owner: "google",
+        type: core_1.ModelType.Text,
+        canStream: true,
+    };
+    createPrompt(_driver, segments, options) {
+        const schema = options.resultSchema;
+        const contents = [];
+        const safety = [];
+        let lastUserContent = undefined;
+        for (const msg of segments) {
+            if (msg.role === core_1.PromptRole.safety) {
+                safety.push(msg.content);
+            }
+            else {
+                const role = msg.role === core_1.PromptRole.assistant ? "model" : "user";
+                if (lastUserContent && lastUserContent.role === role) {
+                    lastUserContent.parts.push({ text: msg.content });
+                }
+                else {
+                    const content = {
+                        role,
+                        parts: [{ text: msg.content }],
+                    };
+                    if (role === 'user') {
+                        lastUserContent = content;
+                    }
+                    contents.push(content);
+                }
+            }
+        }
+        let tools = undefined;
+        if (schema) {
+            // tools = [{
+            //     function_declarations: [{
+            //         name: "validate_json_response",
+            //         description: "Validate the given JSON response",
+            //         parameters: schema as any,
+            //     }]
+            // } as Tool];
+            safety.push("The answer must be a JSON object using the following JSON Schema:\n" + JSON.stringify(schema));
+        }
+        if (safety.length > 0) {
+            const content = safety.join('\n');
+            if (lastUserContent) {
+                lastUserContent.parts.push({ text: content });
+            }
+            else {
+                contents.push({
+                    role: 'user',
+                    parts: [{ text: content }],
+                });
+            }
+        }
+        // put system mesages first and safety last
+        return { contents, tools };
+    }
+    async requestCompletion(driver, prompt, options) {
+        const model = getGenerativeModel(driver, options);
+        const r = await model.generateContent(prompt);
+        const response = await r.response;
+        const usage = response.usageMetadata;
+        const token_usage = {
+            prompt: usage?.prompt_token_count,
+            result: usage?.candidates_token_count,
+            total: usage?.totalTokenCount,
+        };
+        let result;
+        const candidate = response.candidates[0];
+        if (candidate) {
+            const content = candidate.content;
+            if (content) {
+                result = collectTextParts(content);
+                // if (options.resultSchema) {
+                //     result = candidate.;
+                // } else {
+                // }
+            }
+        }
+        return {
+            result: result ?? '',
+            token_usage
+        };
+    }
+    async requestCompletionStream(driver, prompt, options) {
+        const model = getGenerativeModel(driver, options);
+        const streamingResp = await model.generateContentStream(prompt);
+        const stream = (0, async_1.asyncMap)(streamingResp.stream, async (item) => {
+            if (item.candidates.length > 0) {
+                for (const candidate of item.candidates) {
+                    if (candidate.content?.role === 'model') {
+                        const text = collectTextParts(candidate.content);
+                        if (text)
+                            return text;
+                    }
+                }
+            }
+            return '';
+        });
+        return stream;
+    }
+}
+exports.GeminiModelDefinition = GeminiModelDefinition;
+//# sourceMappingURL=gemini.js.map

package/lib/cjs/vertexai/models/gemini.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../../../src/vertexai/models/gemini.ts"],"names":[],"mappings":";;;AAAA,qDAAsI;AACtI,2CAAmJ;AACnJ,kDAAkD;AAIlD,SAAS,kBAAkB,CAAC,MAAsB,EAAE,OAAyB;IACzE,OAAO,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,kBAAkB,CAAC;QAC9C,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,0BAA0B;QAC1B,eAAe,EAAE,CAAC;gBACd,QAAQ,EAAE,uBAAY,CAAC,+BAA+B;gBACtD,SAAS,EAAE,6BAAkB,CAAC,sBAAsB;aACvD,CAAC;QACF,iBAAiB,EAAE;YACf,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,iBAAiB,EAAE,OAAO,CAAC,UAAU;SACxC;KACJ,CAAC,CAAC;AACP,CAAC;AAGD,SAAS,gBAAgB,CAAC,OAAgB;IACtC,MAAM,GAAG,GAAG,EAAE,CAAC;IACf,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IAC5B,IAAI,KAAK,EAAE,CAAC;QACR,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACvB,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACZ,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;QACL,CAAC;IACL,CAAC;IACD,OAAO,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,MAAa,qBAAqB;IAE9B,KAAK,GAAY;QACb,EAAE,EAAE,YAAY;QAChB,IAAI,EAAE,YAAY;QAClB,QAAQ,EAAE,UAAU;QACpB,KAAK,EAAE,QAAQ;QACf,IAAI,EAAE,gBAAS,CAAC,IAAI;QACpB,SAAS,EAAE,IAAI;KAClB,CAAA;IAED,YAAY,CAAC,OAAuB,EAAE,QAAyB,EAAE,OAAsB;QACnF,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;QACpC,MAAM,QAAQ,GAAc,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAa,EAAE,CAAC;QAE5B,IAAI,eAAe,GAAwB,SAAS,CAAC;QACrD,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YACzB,IAAI,GAAG,CAAC,IAAI,KAAK,iBAAU,CAAC,MAAM,EAAE,CAAC;gBACjC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC7B,CAAC;iBAAM,CAAC;gBACJ,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,KAAK,iBAAU,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC;gBAClE,IAAI,eAAe,IAAI,eAAe,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;oBACnD,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAc,CAAC,CAAC;gBAClE,CAAC;qBAAM,CAAC;oBACJ,MAAM,OAAO,GAAY;wBACrB,IAAI;wBACJ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAc,CAAC;qBAC7C,CAAA;oBACD,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;wBAClB,eAAe,GAAG,OAAO,CAAC;oBAC9B,CAAC;oBACD,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC3B,CAAC;YACL,CAAC;QACL,CAAC;QAED,IAAI,KAAK,GAAQ,SAAS,CAAC;QAC3B,IAAI,MAAM,EAAE,CAAC;YAET,aAAa;YACb,gCAAgC;YAChC,0CAA0C;YAC1C,2DAA2D;YAC3D,qCAAqC;YACrC,SAAS;YACT,cAAc;YAEd,MAAM,CAAC,IAAI,CAAC,qEAAqE,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;QAChH,CAAC;QAED,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpB,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,eAAe,EAAE,CAAC;gBAClB,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAc,CAAC,CAAC;YAC9D,CAAC;iBAAM,CAAC;gBACJ,QAAQ,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAc,CAAC;iBACzC,CAAC,CAAA;YACN,CAAC;QACL,CAAC;QAED,2CAA2C;QAC3C,OAAO,EAAE,QAAQ,EAAE,KAAK,EAA4B,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,MAAsB,EAAE,MAA8B,EAAE,OAAyB;QACrG,MAAM,KAAK,GAAG,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAC,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,QAAQ,CAAC;QAClC,MAAM,KAAK,GAAG,QAAQ,CAAC,aAAa,CAAC;QACrC,MAAM,WAAW,GAAwB;YACrC,MAAM,EAAE,KAAK,EAAE,kBAAkB;YACjC,MAAM,EAAE,KAAK,EAAE,sBAAsB;YACrC,KAAK,EAAE,KAAK,EAAE,eAAe;SAChC,CAAA;QAED,IAAI,MAAW,CAAC;QAChB,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QACzC,IAAI,SAAS,EAAE,CAAC;YACZ,MAAM,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC;YAClC,IAAI,OAAO,EAAE,CAAC;gBACV,MAAM,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;gBACnC,8BAA8B;gBAC9B,2BAA2B;gBAC3B,WAAW;gBACX,IAAI;YACR,CAAC;QACL,CAAC;QAED,OAAO;YACH,MAAM,EAAE,MAAM,IAAI,EAAE;YACpB,WAAW;SACd,CAAC;IACN,CAAC;IAED,KAAK,CAAC,uBAAuB,CAAC,MAAsB,EAAE,MAA8B,EAAE,OAAyB;QAC3G,MAAM,KAAK,GAAG,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,aAAa,GAAG,MAAM,KAAK,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,IAAA,gBAAQ,EAAC,aAAa,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACzD,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC7B,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBACtC,IAAI,SAAS,CAAC,OAAO,EAAE,IAAI,KAAK,OAAO,EAAE,CAAC;wBACtC,MAAM,IAAI,GAAG,gBAAgB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;wBACjD,IAAI,IAAI;4BAAE,OAAO,IAAI,CAAC;oBAC1B,CAAC;gBACL,CAAC;YACL,CAAC;YACD,OAAO,EAAE,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAClB,CAAC;CAEJ;AApHD,sDAoHC"}