@llumiverse/drivers 0.15.0 → 0.16.0

package/lib/cjs/vertexai/index.js

@@ -1,22 +1,32 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.VertexAIDriver = void 0;
+exports.trimModelName = trimModelName;
 const vertexai_1 = require("@google-cloud/vertexai");
 const core_1 = require("@llumiverse/core");
 const api_fetch_client_1 = require("api-fetch-client");
 const google_auth_library_1 = require("google-auth-library");
 const embeddings_text_js_1 = require("./embeddings/embeddings-text.js");
 const models_js_1 = require("./models.js");
+const embeddings_image_js_1 = require("./embeddings/embeddings-image.js");
+const aiplatform_1 = require("@google-cloud/aiplatform");
+const vertex_sdk_1 = require("@anthropic-ai/vertex-sdk");
+const imagen_js_1 = require("./models/imagen.js");
+function trimModelName(model) {
+    const i = model.lastIndexOf('@');
+    return i > -1 ? model.substring(0, i) : model;
+}
 class VertexAIDriver extends core_1.AbstractDriver {
     static PROVIDER = "vertexai";
     provider = VertexAIDriver.PROVIDER;
-    //aiplatform: v1.ModelServiceClient;
+    aiplatform;
     vertexai;
     fetchClient;
     authClient;
+    anthropicClient;
     constructor(options) {
         super(options);
-        //this.aiplatform = new v1.ModelServiceClient();
+        this.anthropicClient = undefined;
         this.authClient = options.googleAuthOptions?.authClient ?? new google_auth_library_1.GoogleAuth(options.googleAuthOptions);
         this.vertexai = new vertexai_1.VertexAI({
             project: this.options.project,
@@ -31,43 +41,97 @@ class VertexAIDriver extends core_1.AbstractDriver {
             const token = await this.authClient.getAccessToken();
             return `Bearer ${token}`;
         });
-        // this.aiplatform = new v1.ModelServiceClient({
-        //     projectId: this.options.project,
-        //     apiEndpoint: `${this.options.region}-${API_BASE_PATH}`,
-        // });
+        this.aiplatform = new aiplatform_1.v1beta1.ModelServiceClient({
+            projectId: this.options.project,
+            apiEndpoint: `${this.options.region}-${API_BASE_PATH}`,
+        });
+    }
+    getAnthropicClient() {
+        //Lazy initialisation
+        if (!this.anthropicClient) {
+            this.anthropicClient = new vertex_sdk_1.AnthropicVertex({ region: "us-east5", projectId: process.env.GOOGLE_PROJECT_ID });
+        }
+        return this.anthropicClient;
     }
     canStream(options) {
+        if (options.output_modality == core_1.Modalities.image) {
+            return Promise.resolve(false);
+        }
         return Promise.resolve((0, models_js_1.getModelDefinition)(options.model).model.can_stream === true);
     }
     createPrompt(segments, options) {
+        if (options.model.includes("imagen")) {
+            return new imagen_js_1.ImagenModelDefinition(options.model).createPrompt(this, 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 requestTextCompletion(prompt, options) {
+        return (0, models_js_1.getModelDefinition)(options.model).requestTextCompletion(this, prompt, options);
     }
-    async requestCompletionStream(prompt, options) {
-        return (0, models_js_1.getModelDefinition)(options.model).requestCompletionStream(this, prompt, options);
+    async requestTextCompletionStream(prompt, options) {
+        return (0, models_js_1.getModelDefinition)(options.model).requestTextCompletionStream(this, prompt, options);
+    }
+    async requestImageGeneration(_prompt, _options) {
+        const splits = _options.model.split("/");
+        const modelName = trimModelName(splits[splits.length - 1]);
+        return new imagen_js_1.ImagenModelDefinition(modelName).requestImageGeneration(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
+        let models = [];
+        const modelGarden = new aiplatform_1.v1beta1.ModelGardenServiceClient({
+            projectId: this.options.project,
+            apiEndpoint: `${this.options.region}-${API_BASE_PATH}`,
+        });
+        //Project specific deployed models
+        const [response] = await this.aiplatform.listModels({
+            parent: `projects/${this.options.project}/locations/${this.options.region}`,
+        });
+        models = models.concat(response.map(model => ({
+            id: model.name?.split('/').pop() ?? '',
+            name: model.displayName ?? '',
+            provider: 'vertexai',
+        })));
+        //Model Garden Publisher models - Pretrained models
+        const publishers = ['google', 'anthropic'];
+        const supportedModels = { google: ['gemini', 'imagen'], anthropic: ['claude'] };
+        for (const publisher of publishers) {
+            const [response] = await modelGarden.listPublisherModels({
+                parent: `publishers/${publisher}`,
+                orderBy: 'name',
+                //filter: `name eq name`,
+                listAllVersions: true,
+            });
+            models = models.concat(response.map(model => ({
+                id: model.name ?? '',
+                name: model.name?.split('/').pop() ?? '',
+                provider: 'vertexai',
+                owner: publisher,
+            })).filter(model => {
+                const modelFamily = supportedModels[publisher];
+                for (const family of modelFamily) {
+                    if (model.name.includes(family)) {
+                        return true;
+                    }
+                }
+            }));
+        }
+        return models;
     }
     validateConnection() {
         throw new Error("Method not implemented.");
     }
     async generateEmbeddings(options) {
-        return (0, embeddings_text_js_1.getEmbeddingsForText)(this, options);
+        if (options.image || options.model?.includes("multimodal")) {
+            if (options.text && options.image) {
+                throw new Error("Text and Image simultaneous embedding not implemented. Submit seperately");
+            }
+            return (0, embeddings_image_js_1.getEmbeddingsForImages)(this, options);
+        }
+        const text_options = {
+            content: options.text ?? '',
+            model: options.model,
+        };
+        return (0, embeddings_text_js_1.getEmbeddingsForText)(this, text_options);
     }
 }
 exports.VertexAIDriver = VertexAIDriver;

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

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/vertexai/index.ts"],"names":[],"mappings":";;;AAAA,qDAA0E;AAC1E,2CAA4K;AAC5K,uDAA+C;AAC/C,6DAAoE;AAEpE,wEAA8F;AAC9F,2CAAgE;AAShE,MAAa,cAAe,SAAQ,qBAA6D;IAC7F,MAAM,CAAC,QAAQ,GAAG,UAAU,CAAC;IAC7B,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC;IAEnC,oCAAoC;IACpC,QAAQ,CAAW;IACnB,WAAW,CAAc;IACzB,UAAU,CAAsC;IAEhD,YAAa,OAA8B;QACvC,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,gDAAgD;QAEhD,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,iBAAiB,EAAE,UAAU,IAAI,IAAI,gCAAU,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAErG,IAAI,CAAC,QAAQ,GAAG,IAAI,mBAAQ,CAAC;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;YAC7B,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;YAC7B,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB;SACpD,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,YAAY;YACZ,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC;YACrD,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,UAAU,KAAK,IAAI,CAAC,CAAC;IACxF,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;IAED,kBAAkB;QACd,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,OAA8B;QACnD,OAAO,IAAA,yCAAoB,EAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC/C,CAAC;;AAzEL,wCA2EC;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"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/vertexai/index.ts"],"names":[],"mappings":";;;AAuBA,sCAGC;AA1BD,qDAA0E;AAC1E,2CAAiN;AACjN,uDAA+C;AAC/C,6DAAoE;AAEpE,wEAA8F;AAC9F,2CAAiD;AAEjD,0EAA0E;AAC1E,yDAAmD;AACnD,yDAA2D;AAC3D,kDAAyE;AAYzE,SAAgB,aAAa,CAAC,KAAa;IACvC,MAAM,CAAC,GAAG,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IACjC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;AAClD,CAAC;AAED,MAAa,cAAe,SAAQ,qBAAqD;IACrF,MAAM,CAAC,QAAQ,GAAG,UAAU,CAAC;IAC7B,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC;IAEnC,UAAU,CAA6B;IACvC,QAAQ,CAAW;IACnB,WAAW,CAAc;IACzB,UAAU,CAAsC;IAChD,eAAe,CAA8B;IAE7C,YAAa,OAA8B;QACvC,KAAK,CAAC,OAAO,CAAC,CAAC;QAEf,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QAEjC,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,iBAAiB,EAAE,UAAU,IAAI,IAAI,gCAAU,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAErG,IAAI,CAAC,QAAQ,GAAG,IAAI,mBAAQ,CAAC;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;YAC7B,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;YAC7B,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB;SACpD,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,YAAY;YACZ,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC;YACrD,OAAO,UAAU,KAAK,EAAE,CAAC;QAC7B,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,UAAU,GAAG,IAAI,oBAAO,CAAC,kBAAkB,CAAC;YAC7C,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;YAC/B,WAAW,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,aAAa,EAAE;SACzD,CAAC,CAAC;IACP,CAAC;IAEM,kBAAkB;QACrB,qBAAqB;QACrB,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YACxB,IAAI,CAAC,eAAe,GAAG,IAAI,4BAAe,CAAC,EAAC,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAC,CAAC,CAAC;QAC/G,CAAC;QACD,OAAO,IAAI,CAAC,eAAe,CAAC;IAChC,CAAC;IAES,SAAS,CAAC,OAAyB;QACzC,IAAI,OAAO,CAAC,eAAe,IAAI,iBAAU,CAAC,KAAK,EAAE,CAAC;YAC9C,OAAO,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,OAAO,CAAC,OAAO,CAAC,IAAA,8BAAkB,EAAC,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,UAAU,KAAK,IAAI,CAAC,CAAC;IACxF,CAAC;IAEM,YAAY,CAAC,QAAyB,EAAE,OAAyB;QACpE,IAAI,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACnC,OAAO,IAAI,iCAAqB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC1F,CAAC;QACD,OAAO,IAAA,8BAAkB,EAAC,OAAO,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;IACnF,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,MAAsB,EAAE,OAAyB;QACzE,OAAO,IAAA,8BAAkB,EAAC,OAAO,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IAC1F,CAAC;IACD,KAAK,CAAC,2BAA2B,CAAC,MAAsB,EAAE,OAAyB;QAC/E,OAAO,IAAA,8BAAkB,EAAC,OAAO,CAAC,KAAK,CAAC,CAAC,2BAA2B,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IAChG,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,OAAqB,EAAE,QAA0B;QAC1E,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACzC,MAAM,SAAS,GAAG,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QAC3D,OAAO,IAAI,iCAAqB,CAAC,SAAS,CAAC,CAAC,sBAAsB,CAAC,IAAI,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;IAChG,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAA4B;QACzC,IAAI,MAAM,GAAsB,EAAE,CAAC;QACnC,MAAM,WAAW,GAAG,IAAI,oBAAO,CAAC,wBAAwB,CAAC;YACrD,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;YAC/B,WAAW,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,aAAa,EAAE;SACzD,CAAC,CAAC;QAEH,kCAAkC;QAClC,MAAM,CAAC,QAAQ,CAAC,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;YAChD,MAAM,EAAE,YAAY,IAAI,CAAC,OAAO,CAAC,OAAO,cAAc,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;SAC9E,CAAC,CAAC;QACH,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC1C,EAAE,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE;YACtC,IAAI,EAAE,KAAK,CAAC,WAAW,IAAI,EAAE;YAC7B,QAAQ,EAAE,UAAU;SACvB,CAAC,CAAC,CAAC,CAAC;QAEL,mDAAmD;QACnD,MAAM,UAAU,GAAG,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAA;QAC1C,MAAM,eAAe,GAAG,EAAC,MAAM,EAAE,CAAC,QAAQ,EAAC,QAAQ,CAAC,EAAE,SAAS,EAAE,CAAC,QAAQ,CAAC,EAAC,CAAA;QAE5E,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACjC,MAAM,CAAC,QAAQ,CAAC,GAAG,MAAM,WAAW,CAAC,mBAAmB,CAAC;gBACrD,MAAM,EAAE,cAAc,SAAS,EAAE;gBACjC,OAAO,EAAE,MAAM;gBACf,yBAAyB;gBACzB,eAAe,EAAE,IAAI;aACxB,CAAC,CAAC;YAEH,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBAC1C,EAAE,EAAE,KAAK,CAAC,IAAI,IAAI,EAAE;gBACpB,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE;gBACxC,QAAQ,EAAE,UAAU;gBACpB,KAAK,EAAE,SAAS;aACnB,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;gBACf,MAAM,WAAW,GAAG,eAAe,CAAC,SAAyC,CAAC,CAAC;gBAC/E,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE,CAAC;oBAC/B,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC9B,OAAO,IAAI,CAAC;oBAChB,CAAC;gBACL,CAAC;YACL,CAAC,CAAC,CAAC,CAAC;QACR,CAAC;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,kBAAkB;QACd,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,OAA0B;QAC/C,IAAI,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YACzD,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBAChC,MAAM,IAAI,KAAK,CAAC,0EAA0E,CAAC,CAAC;YAChG,CAAC;YACD,OAAO,IAAA,4CAAsB,EAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACjD,CAAC;QACD,MAAM,YAAY,GAA0B;YACxC,OAAO,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE;YAC3B,KAAK,EAAE,OAAO,CAAC,KAAK;SACvB,CAAA;QACD,OAAO,IAAA,yCAAoB,EAAC,IAAI,EAAE,YAAY,CAAC,CAAC;IACpD,CAAC;;AAtIL,wCAwIC;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/claude.js

@@ -0,0 +1,252 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ClaudeModelDefinition = void 0;
+exports.collectTools = collectTools;
+const core_1 = require("@llumiverse/core");
+const async_1 = require("@llumiverse/core/async");
+function getFullModelName(model) {
+    if (model.includes("claude-3-5-sonnet-v2")) {
+        return "claude-3-5-sonnet-v2@20241022";
+    }
+    else if (model.includes("claude-3-5-sonnet")) {
+        return "claude-3-5-sonnet@20240620";
+    }
+    else if (model.includes("claude-3-5-haiku")) {
+        return "claude-3-5-haiku@20241022";
+    }
+    else if (model.includes("claude-3-opus")) {
+        return "claude-3-opus@20240229";
+    }
+    else if (model.includes("claude-3-sonnet")) {
+        return "claude-3-sonnet@20240229";
+    }
+    else if (model.includes("claude-3-haike")) {
+        return "claude-3-haiku@20240307";
+    }
+    return model;
+}
+function claudeFinishReason(reason) {
+    if (!reason)
+        return undefined;
+    switch (reason) {
+        case 'end_turn': return "stop";
+        case 'max_tokens': return "length";
+        default: return reason; //stop_sequence
+    }
+}
+function collectTextParts(content) {
+    const out = [];
+    for (const block of content) {
+        if (block?.text) {
+            out.push(block.text);
+        }
+    }
+    return out.join('\n');
+}
+function maxToken(max_tokens, model) {
+    const contains = (str, substr) => str.indexOf(substr) !== -1;
+    if (max_tokens) {
+        return max_tokens;
+    }
+    else if (contains(model, "claude-3-5")) {
+        return 8192;
+    }
+    else {
+        return 4096;
+    }
+}
+class ClaudeModelDefinition {
+    model;
+    constructor(modelId) {
+        this.model = {
+            id: modelId,
+            name: modelId,
+            provider: 'vertexai',
+            type: core_1.ModelType.Text,
+            can_stream: true,
+        };
+    }
+    async createPrompt(_driver, segments, options) {
+        // Convert the prompt to the format expected by the Claude API
+        const systemSegments = segments
+            .filter(segment => segment.role === core_1.PromptRole.system)
+            .map(segment => ({
+            text: segment.content,
+            type: 'text'
+        }));
+        const safetySegments = segments
+            .filter(segment => segment.role === core_1.PromptRole.safety)
+            .map(segment => ({
+            text: segment.content,
+            type: 'text'
+        }));
+        if (options.result_schema) {
+            const schemaSegments = {
+                text: "The answer must be a JSON object using the following JSON Schema:\n" + JSON.stringify(options.result_schema),
+                type: 'text'
+            };
+            safetySegments.push(schemaSegments);
+        }
+        const messages = segments.filter(segment => segment.role == core_1.PromptRole.user
+            || segment.role == core_1.PromptRole.assistant
+            || segment.role === core_1.PromptRole.tool)
+            .map(segment => {
+            if (segment.role === core_1.PromptRole.tool) {
+                if (!segment.tool_use_id) {
+                    throw new Error("Tool prompt segment must have a tool_use_id");
+                }
+                return {
+                    role: 'user',
+                    content: [
+                        {
+                            type: 'tool_result',
+                            tool_use_id: segment.tool_use_id,
+                            content: segment.content || undefined
+                        }
+                    ]
+                };
+            }
+            else {
+                return {
+                    role: segment.role !== core_1.PromptRole.user ? 'assistant' : 'user',
+                    content: segment.content
+                };
+            }
+        });
+        const system = systemSegments.concat(safetySegments);
+        return {
+            messages: messages,
+            system: system
+        };
+    }
+    async requestTextCompletion(driver, prompt, options) {
+        const client = driver.getAnthropicClient();
+        const splits = options.model.split("/");
+        const modelName = splits[splits.length - 1];
+        options = { ...options, model: modelName };
+        options.model_options = options.model_options;
+        if (options.model_options?._option_id !== "vertexai-claude") {
+            driver.logger.warn("Invalid model options", { options: options.model_options });
+        }
+        let conversation = updateConversation(options.conversation, prompt);
+        const result = await client.messages.create({
+            ...conversation, // messages, system,
+            tools: options.tools, // we are using the same shape as claude for tools
+            temperature: options.model_options?.temperature,
+            model: modelName,
+            max_tokens: maxToken(options.model_options?.max_tokens, modelName),
+            top_p: options.model_options?.top_p,
+            top_k: options.model_options?.top_k,
+            stop_sequences: options.model_options?.stop_sequence,
+            thinking: options.model_options?.thinking_mode ?
+                {
+                    budget_tokens: options.model_options?.thinking_budget_tokens ?? 1024,
+                    type: "enabled"
+                } : {
+                type: "disabled"
+            }
+        });
+        const text = collectTextParts(result.content);
+        const tool_use = collectTools(result.content);
+        conversation = updateConversation(options.conversation, createPromptFromResponse(result));
+        return {
+            chat: [prompt, { role: result.role, content: result.content }],
+            result: text ?? '',
+            tool_use,
+            token_usage: {
+                prompt: result?.usage.input_tokens,
+                result: result?.usage.output_tokens,
+                total: result?.usage.input_tokens + result?.usage.output_tokens
+            },
+            // make sure we set finish_reason to the correct value (claude is normally setting this by itself)
+            finish_reason: tool_use ? "tool_use" : claudeFinishReason(result?.stop_reason ?? ''),
+            conversation
+        };
+    }
+    async requestTextCompletionStream(driver, prompt, options) {
+        const client = driver.getAnthropicClient();
+        const splits = options.model.split("/");
+        const modelName = splits[splits.length - 1];
+        options = { ...options, model: modelName };
+        options.model_options = options.model_options;
+        if (options.model_options?._option_id !== "vertexai-claude") {
+            driver.logger.warn("Invalid model options", { options: options.model_options });
+        }
+        const response_stream = await client.messages.stream({
+            ...prompt, // messages, system,
+            tools: options.tools, // we are using the same shape as claude for tools
+            temperature: options.model_options?.temperature,
+            model: modelName,
+            max_tokens: maxToken(options.model_options?.max_tokens, modelName),
+            top_p: options.model_options?.top_p,
+            top_k: options.model_options?.top_k,
+            stop_sequences: options.model_options?.stop_sequence,
+            thinking: options.model_options?.thinking_mode ?
+                {
+                    budget_tokens: options.model_options?.thinking_budget_tokens ?? 1024,
+                    type: "enabled"
+                } : {
+                type: "disabled"
+            }
+        });
+        //Streaming does not give information on the input tokens,
+        //So we use a seperate call to get the input tokens.
+        //Non-critical and model name sensitive so we put it in a try catch block
+        let count_tokens = { input_tokens: 0 };
+        try {
+            count_tokens = await client.messages.countTokens({
+                ...prompt, // messages, system
+                model: getFullModelName(modelName),
+            });
+        }
+        catch (e) {
+            driver.logger.warn("Failed to get token count for model " + modelName);
+        }
+        const stream = (0, async_1.asyncMap)(response_stream, async (item) => {
+            return {
+                result: item?.delta?.text ?? '',
+                token_usage: { prompt: count_tokens.input_tokens, result: item?.usage?.output_tokens },
+                finish_reason: claudeFinishReason(item?.delta?.stop_reason ?? ''),
+            };
+        });
+        return stream;
+    }
+}
+exports.ClaudeModelDefinition = ClaudeModelDefinition;
+function collectTools(content) {
+    const out = [];
+    for (const block of content) {
+        if (block?.type === "tool_use") {
+            out.push({
+                id: block.id,
+                name: block.name,
+                input: block.input,
+            });
+        }
+    }
+    return out.length > 0 ? out : undefined;
+}
+function createPromptFromResponse(response) {
+    return {
+        messages: [{
+                role: core_1.PromptRole.assistant,
+                content: response.content,
+            }],
+        system: []
+    };
+}
+/**
+ * Update the converatation messages
+ * @param prompt
+ * @param response
+ * @returns
+ */
+function updateConversation(conversation, prompt) {
+    const baseSystemMessages = conversation ? conversation.system : [];
+    const baseMessages = conversation ? conversation.messages : [];
+    return {
+        messages: baseMessages.concat(prompt.messages || []),
+        system: baseSystemMessages.concat(prompt.system || [])
+    };
+}
+//# sourceMappingURL=claude.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"claude.js","sourceRoot":"","sources":["../../../../src/vertexai/models/claude.ts"],"names":[],"mappings":";;;AAiPA,oCAcC;AA7PD,2CAA0K;AAC1K,kDAAkD;AAYlD,SAAS,gBAAgB,CAAC,KAAa;IACnC,IAAI,KAAK,CAAC,QAAQ,CAAC,sBAAsB,CAAC,EAAE,CAAC;QACzC,OAAO,+BAA+B,CAAA;IAC1C,CAAC;SAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;QAC7C,OAAO,4BAA4B,CAAA;IACvC,CAAC;SAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC;QAC5C,OAAO,2BAA2B,CAAA;IACtC,CAAC;SAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;QACzC,OAAO,wBAAwB,CAAA;IACnC,CAAC;SAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,iBAAiB,CAAC,EAAE,CAAC;QAC3C,OAAO,0BAA0B,CAAA;IACrC,CAAC;SAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;QAC1C,OAAO,yBAAyB,CAAA;IACpC,CAAC;IACD,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,SAAS,kBAAkB,CAAC,MAA0B;IAClD,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAC9B,QAAQ,MAAM,EAAE,CAAC;QACb,KAAK,UAAU,CAAC,CAAC,OAAO,MAAM,CAAC;QAC/B,KAAK,YAAY,CAAC,CAAC,OAAO,QAAQ,CAAC;QACnC,OAAO,CAAC,CAAC,OAAO,MAAM,CAAC,CAAC,eAAe;IAC3C,CAAC;AACL,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAY;IAClC,MAAM,GAAG,GAAG,EAAE,CAAC;IAEf,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC1B,IAAI,KAAK,EAAE,IAAI,EAAE,CAAC;YACd,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACzB,CAAC;IACL,CAAC;IACD,OAAO,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,SAAS,QAAQ,CAAC,UAA8B,EAAE,KAAa;IAC3D,MAAM,QAAQ,GAAG,CAAC,GAAW,EAAE,MAAc,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAC7E,IAAI,UAAU,EAAE,CAAC;QACb,OAAO,UAAU,CAAC;IACtB,CAAC;SAAM,IAAI,QAAQ,CAAC,KAAK,EAAE,YAAY,CAAC,EAAE,CAAC;QACvC,OAAO,IAAI,CAAC;IAChB,CAAC;SAAM,CAAC;QACJ,OAAO,IAAI,CAAA;IACf,CAAC;AACL,CAAC;AAED,MAAa,qBAAqB;IAE9B,KAAK,CAAS;IAEd,YAAY,OAAe;QACvB,IAAI,CAAC,KAAK,GAAG;YACT,EAAE,EAAE,OAAO;YACX,IAAI,EAAE,OAAO;YACb,QAAQ,EAAE,UAAU;YACpB,IAAI,EAAE,gBAAS,CAAC,IAAI;YACpB,UAAU,EAAE,IAAI;SACR,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,OAAuB,EAAE,QAAyB,EAAE,OAAsB;QACzF,8DAA8D;QAC9D,MAAM,cAAc,GAAqB,QAAQ;aAC5C,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,iBAAU,CAAC,MAAM,CAAC;aACrD,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACb,IAAI,EAAE,OAAO,CAAC,OAAO;YACrB,IAAI,EAAE,MAAM;SACf,CAAC,CAAC,CAAC;QAER,MAAM,cAAc,GAAqB,QAAQ;aAC5C,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,iBAAU,CAAC,MAAM,CAAC;aACrD,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACb,IAAI,EAAE,OAAO,CAAC,OAAO;YACrB,IAAI,EAAE,MAAM;SACf,CAAC,CAAC,CAAC;QAER,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,MAAM,cAAc,GAAmB;gBACnC,IAAI,EAAE,qEAAqE,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,aAAa,CAAC;gBACnH,IAAI,EAAE,MAAM;aACf,CAAA;YACD,cAAc,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACxC,CAAC;QAED,MAAM,QAAQ,GAAmB,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CACvD,OAAO,CAAC,IAAI,IAAI,iBAAU,CAAC,IAAI;eAC5B,OAAO,CAAC,IAAI,IAAI,iBAAU,CAAC,SAAS;eACpC,OAAO,CAAC,IAAI,KAAK,iBAAU,CAAC,IAAI,CAAC;aACnC,GAAG,CAAC,OAAO,CAAC,EAAE;YACX,IAAI,OAAO,CAAC,IAAI,KAAK,iBAAU,CAAC,IAAI,EAAE,CAAC;gBACnC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;oBACvB,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;gBACnE,CAAC;gBACD,OAAO;oBACH,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE;wBACL;4BACI,IAAI,EAAE,aAAa;4BACnB,WAAW,EAAE,OAAO,CAAC,WAAW;4BAChC,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,SAAS;yBACxC;qBACJ;iBACJ,CAAA;YACL,CAAC;iBAAM,CAAC;gBACJ,OAAO;oBACH,IAAI,EAAE,OAAO,CAAC,IAAI,KAAK,iBAAU,CAAC,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM;oBAC7D,OAAO,EAAE,OAAO,CAAC,OAAO;iBAC3B,CAAA;YACL,CAAC;QACL,CAAC,CAAC,CAAC;QAEP,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QAErD,OAAO;YACH,QAAQ,EAAE,QAAQ;YAClB,MAAM,EAAE,MAAM;SACjB,CAAA;IACL,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,MAAsB,EAAE,MAAoB,EAAE,OAAyB;QAC/F,MAAM,MAAM,GAAG,MAAM,CAAC,kBAAkB,EAAE,CAAC;QAC3C,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC5C,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAC3C,OAAO,CAAC,aAAa,GAAG,OAAO,CAAC,aAAsC,CAAC;QAEvE,IAAI,OAAO,CAAC,aAAa,EAAE,UAAU,KAAK,iBAAiB,EAAE,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,OAAO,EAAE,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;QACpF,CAAC;QAED,IAAI,YAAY,GAAG,kBAAkB,CAAC,OAAO,CAAC,YAA4B,EAAE,MAAM,CAAC,CAAC;QAEpF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;YACxC,GAAG,YAAY,EAAE,oBAAoB;YACrC,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,kDAAkD;YACxE,WAAW,EAAE,OAAO,CAAC,aAAa,EAAE,WAAW;YAC/C,KAAK,EAAE,SAAS;YAChB,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,aAAa,EAAE,UAAU,EAAE,SAAS,CAAC;YAClE,KAAK,EAAE,OAAO,CAAC,aAAa,EAAE,KAAK;YACnC,KAAK,EAAE,OAAO,CAAC,aAAa,EAAE,KAAK;YACnC,cAAc,EAAE,OAAO,CAAC,aAAa,EAAE,aAAa;YACpD,QAAQ,EAAE,OAAO,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;gBAC5C;oBACI,aAAa,EAAE,OAAO,CAAC,aAAa,EAAE,sBAAsB,IAAI,IAAI;oBACpE,IAAI,EAAE,SAAS;iBAClB,CAAC,CAAC,CAAC;gBACA,IAAI,EAAE,UAAU;aACnB;SACR,CAAY,CAAC;QAEd,MAAM,IAAI,GAAG,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAE9C,YAAY,GAAG,kBAAkB,CAAC,OAAO,CAAC,YAA4B,EAAE,wBAAwB,CAAC,MAAM,CAAC,CAAC,CAAC;QAE1G,OAAO;YACH,IAAI,EAAE,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;YAC9D,MAAM,EAAE,IAAI,IAAI,EAAE;YAClB,QAAQ;YACR,WAAW,EAAE;gBACT,MAAM,EAAE,MAAM,EAAE,KAAK,CAAC,YAAY;gBAClC,MAAM,EAAE,MAAM,EAAE,KAAK,CAAC,aAAa;gBACnC,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,YAAY,GAAG,MAAM,EAAE,KAAK,CAAC,aAAa;aAClE;YACD,kGAAkG;YAClG,aAAa,EAAE,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,kBAAkB,CAAC,MAAM,EAAE,WAAW,IAAI,EAAE,CAAC;YACpF,YAAY;SACD,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,2BAA2B,CAAC,MAAsB,EAAE,MAAoB,EAAE,OAAyB;QACrG,MAAM,MAAM,GAAG,MAAM,CAAC,kBAAkB,EAAE,CAAC;QAC3C,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC5C,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAC3C,OAAO,CAAC,aAAa,GAAG,OAAO,CAAC,aAAsC,CAAC;QAEvE,IAAI,OAAO,CAAC,aAAa,EAAE,UAAU,KAAK,iBAAiB,EAAE,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,OAAO,EAAE,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;QACpF,CAAC;QAED,MAAM,eAAe,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;YACjD,GAAG,MAAM,EAAE,oBAAoB;YAC/B,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,kDAAkD;YACxE,WAAW,EAAE,OAAO,CAAC,aAAa,EAAE,WAAW;YAC/C,KAAK,EAAE,SAAS;YAChB,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,aAAa,EAAE,UAAU,EAAE,SAAS,CAAC;YAClE,KAAK,EAAE,OAAO,CAAC,aAAa,EAAE,KAAK;YACnC,KAAK,EAAE,OAAO,CAAC,aAAa,EAAE,KAAK;YACnC,cAAc,EAAE,OAAO,CAAC,aAAa,EAAE,aAAa;YACpD,QAAQ,EAAE,OAAO,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;gBAC5C;oBACI,aAAa,EAAE,OAAO,CAAC,aAAa,EAAE,sBAAsB,IAAI,IAAI;oBACpE,IAAI,EAAE,SAAS;iBAClB,CAAC,CAAC,CAAC;gBACA,IAAI,EAAE,UAAU;aACnB;SACR,CAAC,CAAC;QAEH,0DAA0D;QAC1D,oDAAoD;QACpD,yEAAyE;QACzE,IAAI,YAAY,GAAG,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;QACvC,IAAI,CAAC;YACD,YAAY,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC;gBAC7C,GAAG,MAAM,EAAG,mBAAmB;gBAC/B,KAAK,EAAE,gBAAgB,CAAC,SAAS,CAAC;aACrC,CAAC,CAAC;QACP,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,sCAAsC,GAAG,SAAS,CAAC,CAAC;QAC3E,CAAC;QAED,MAAM,MAAM,GAAG,IAAA,gBAAQ,EAAC,eAAe,EAAE,KAAK,EAAE,IAAS,EAAE,EAAE;YACzD,OAAO;gBACH,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,IAAI,EAAE;gBAC/B,WAAW,EAAE,EAAE,MAAM,EAAE,YAAY,CAAC,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,aAAa,EAAE;gBACtF,aAAa,EAAE,kBAAkB,CAAC,IAAI,EAAE,KAAK,EAAE,WAAW,IAAI,EAAE,CAAC;aACpE,CAAA;QACL,CAAC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAClB,CAAC;CACJ;AAhLD,sDAgLC;AAED,SAAgB,YAAY,CAAC,OAAuB;IAChD,MAAM,GAAG,GAAc,EAAE,CAAC;IAE1B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC1B,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;YAC7B,GAAG,CAAC,IAAI,CAAC;gBACL,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,KAAK,EAAE,KAAK,CAAC,KAAmB;aACnC,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAED,OAAO,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC;AAC5C,CAAC;AAED,SAAS,wBAAwB,CAAC,QAAiB;IAC/C,OAAO;QACH,QAAQ,EAAE,CAAC;gBACP,IAAI,EAAE,iBAAU,CAAC,SAAS;gBAC1B,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC5B,CAAC;QACF,MAAM,EAAE,EAAE;KACb,CAAA;AACL,CAAC;AAED;;;;;GAKG;AACH,SAAS,kBAAkB,CAAC,YAA6C,EAAE,MAAoB;IAC3F,MAAM,kBAAkB,GAAG,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;IACnE,MAAM,YAAY,GAAG,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAA;IAC9D,OAAO;QACH,QAAQ,EAAE,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC;QACpD,MAAM,EAAE,kBAAkB,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;KACzD,CAAC;AACN,CAAC"}