promptfoo 0.60.0 → 0.62.0

package/dist/src/providers/pythonCompletion.js

@@ -20,32 +20,57 @@ class PythonProvider {
     id() {
         return `python:${this.scriptPath}`;
     }
-    async callApi(prompt, context) {
+    async executePythonScript(prompt, context, apiType) {
         const absPath = path_1.default.resolve(path_1.default.join(this.options?.config.basePath || '', this.scriptPath));
         logger_1.default.debug(`Computing file hash for script ${absPath}`);
         const fileHash = (0, util_1.sha256)(fs_1.default.readFileSync(absPath, 'utf-8'));
-        const cacheKey = `python:${this.scriptPath}:${fileHash}:${prompt}:${JSON.stringify(this.options)}`;
+        const cacheKey = `python:${this.scriptPath}:${apiType}:${fileHash}:${prompt}:${JSON.stringify(this.options)}`;
         const cache = await (0, cache_1.getCache)();
         let cachedResult;
         if ((0, cache_1.isCacheEnabled)()) {
             cachedResult = (await cache.get(cacheKey));
         }
         if (cachedResult) {
-            logger_1.default.debug(`Returning cached result for script ${absPath}`);
+            logger_1.default.debug(`Returning cached ${apiType} result for script ${absPath}`);
             return JSON.parse(cachedResult);
         }
         else {
-            // These are not useful in Python
-            delete context?.fetchWithCache;
-            delete context?.getCache;
-            delete context?.logger;
-            const args = [prompt, this.options, context];
+            if (context) {
+                // These are not useful in Python
+                delete context.fetchWithCache;
+                delete context.getCache;
+                delete context.logger;
+            }
+            const args = apiType === 'call_api' ? [prompt, this.options, context] : [prompt, this.options];
             logger_1.default.debug(`Running python script ${absPath} with scriptPath ${this.scriptPath} and args: ${args.join('\n')}`);
-            const result = (await (0, wrapper_1.runPython)(absPath, 'call_api', args, {
-                pythonExecutable: this.config.pythonExecutable,
-            }));
-            if (!('output' in result) && !('error' in result)) {
-                throw new Error('The Python script `call_api` function must return a dict with an `output` string/object or `error` string');
+            let result;
+            switch (apiType) {
+                case 'call_api':
+                    result = (await (0, wrapper_1.runPython)(absPath, apiType, args, {
+                        pythonExecutable: this.config.pythonExecutable,
+                    }));
+                    if (!('output' in result) && !('error' in result)) {
+                        throw new Error(`The Python script \`${apiType}\` function must return a dict with an \`output\` string/object or \`error\` string, instead got: ${JSON.stringify(result)}`);
+                    }
+                    break;
+                case 'call_embedding_api':
+                    result = (await (0, wrapper_1.runPython)(absPath, apiType, args, {
+                        pythonExecutable: this.config.pythonExecutable,
+                    }));
+                    if (!('embedding' in result) && !('error' in result)) {
+                        throw new Error(`The Python script \`${apiType}\` function must return a dict with an \`embedding\` array or \`error\` string, instead got ${JSON.stringify(result)}`);
+                    }
+                    break;
+                case 'call_classification_api':
+                    result = (await (0, wrapper_1.runPython)(absPath, apiType, args, {
+                        pythonExecutable: this.config.pythonExecutable,
+                    }));
+                    if (!('classification' in result) && !('error' in result)) {
+                        throw new Error(`The Python script \`${apiType}\` function must return a dict with a \`classification\` object or \`error\` string, instead of ${JSON.stringify(result)}`);
+                    }
+                    break;
+                default:
+                    throw new Error(`Unsupported apiType: ${apiType}`);
             }
             if ((0, cache_1.isCacheEnabled)() && !('error' in result)) {
                 await cache.set(cacheKey, JSON.stringify(result));
@@ -53,6 +78,15 @@ class PythonProvider {
             return result;
         }
     }
+    async callApi(prompt, context) {
+        return this.executePythonScript(prompt, context, 'call_api');
+    }
+    async callEmbeddingApi(prompt) {
+        return this.executePythonScript(prompt, undefined, 'call_embedding_api');
+    }
+    async callClassificationApi(prompt) {
+        return this.executePythonScript(prompt, undefined, 'call_classification_api');
+    }
 }
 exports.PythonProvider = PythonProvider;
 //# sourceMappingURL=pythonCompletion.js.map

package/dist/src/providers/pythonCompletion.js.map

@@ -1 +1 @@
-{"version":3,"file":"pythonCompletion.js","sourceRoot":"","sources":["../../../src/providers/pythonCompletion.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAwB;AACxB,4CAAoB;AAEpB,uDAA+B;AAC/B,oCAAoD;AACpD,+CAA8C;AAC9C,kCAAiC;AAajC,MAAa,cAAc;IAGzB,YAAoB,UAAkB,EAAU,OAAyB;QAArD,eAAU,GAAV,UAAU,CAAQ;QAAU,YAAO,GAAP,OAAO,CAAkB;QACvE,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,CAAC,OAAO,EAAE,EAAE,IAAI,UAAU,IAAI,CAAC,UAAU,EAAE,CAAC;QAC3D,IAAI,CAAC,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,EAAE,CAAC;IACtC,CAAC;IAED,EAAE;QACA,OAAO,UAAU,IAAI,CAAC,UAAU,EAAE,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,OAA8B;QAC1D,MAAM,OAAO,GAAG,cAAI,CAAC,OAAO,CAAC,cAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,QAAQ,IAAI,EAAE,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;QAC9F,gBAAM,CAAC,KAAK,CAAC,kCAAkC,OAAO,EAAE,CAAC,CAAC;QAC1D,MAAM,QAAQ,GAAG,IAAA,aAAM,EAAC,YAAE,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QAC3D,MAAM,QAAQ,GAAG,UAAU,IAAI,CAAC,UAAU,IAAI,QAAQ,IAAI,MAAM,IAAI,IAAI,CAAC,SAAS,CAChF,IAAI,CAAC,OAAO,CACb,EAAE,CAAC;QACJ,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;QAC/B,IAAI,YAAY,CAAC;QAEjB,IAAI,IAAA,sBAAc,GAAE,EAAE;YACpB,YAAY,GAAG,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAW,CAAC;SACtD;QAED,IAAI,YAAY,EAAE;YAChB,gBAAM,CAAC,KAAK,CAAC,sCAAsC,OAAO,EAAE,CAAC,CAAC;YAC9D,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;SACjC;aAAM;YACL,iCAAiC;YACjC,OAAO,OAAO,EAAE,cAAc,CAAC;YAC/B,OAAO,OAAO,EAAE,QAAQ,CAAC;YACzB,OAAO,OAAO,EAAE,MAAM,CAAC;YAEvB,MAAM,IAAI,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC7C,gBAAM,CAAC,KAAK,CACV,yBAAyB,OAAO,oBAAoB,IAAI,CAAC,UAAU,cAAc,IAAI,CAAC,IAAI,CACxF,IAAI,CACL,EAAE,CACJ,CAAC;YACF,MAAM,MAAM,GAAG,CAAC,MAAM,IAAA,mBAAS,EAAC,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE;gBACzD,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;aAC/C,CAAC,CAGD,CAAC;YACF,IAAI,CAAC,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,EAAE;gBACjD,MAAM,IAAI,KAAK,CACb,2GAA2G,CAC5G,CAAC;aACH;YACD,IAAI,IAAA,sBAAc,GAAE,IAAI,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,EAAE;gBAC5C,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;aACnD;YACD,OAAO,MAAM,CAAC;SACf;IACH,CAAC;CACF;AA1DD,wCA0DC"}
+{"version":3,"file":"pythonCompletion.js","sourceRoot":"","sources":["../../../src/providers/pythonCompletion.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAwB;AACxB,4CAAoB;AAEpB,uDAA+B;AAC/B,oCAAoD;AACpD,+CAA8C;AAC9C,kCAAiC;AAejC,MAAa,cAAc;IAGzB,YAAoB,UAAkB,EAAU,OAAyB;QAArD,eAAU,GAAV,UAAU,CAAQ;QAAU,YAAO,GAAP,OAAO,CAAkB;QACvE,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,CAAC,OAAO,EAAE,EAAE,IAAI,UAAU,IAAI,CAAC,UAAU,EAAE,CAAC;QAC3D,IAAI,CAAC,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,EAAE,CAAC;IACtC,CAAC;IAED,EAAE;QACA,OAAO,UAAU,IAAI,CAAC,UAAU,EAAE,CAAC;IACrC,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,MAAc,EACd,OAAyC,EACzC,OAAsE;QAEtE,MAAM,OAAO,GAAG,cAAI,CAAC,OAAO,CAAC,cAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,QAAQ,IAAI,EAAE,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;QAC9F,gBAAM,CAAC,KAAK,CAAC,kCAAkC,OAAO,EAAE,CAAC,CAAC;QAC1D,MAAM,QAAQ,GAAG,IAAA,aAAM,EAAC,YAAE,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QAC3D,MAAM,QAAQ,GAAG,UAAU,IAAI,CAAC,UAAU,IAAI,OAAO,IAAI,QAAQ,IAAI,MAAM,IAAI,IAAI,CAAC,SAAS,CAC3F,IAAI,CAAC,OAAO,CACb,EAAE,CAAC;QACJ,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;QAC/B,IAAI,YAAY,CAAC;QAEjB,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;YACrB,YAAY,GAAG,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAW,CAAC;QACvD,CAAC;QAED,IAAI,YAAY,EAAE,CAAC;YACjB,gBAAM,CAAC,KAAK,CAAC,oBAAoB,OAAO,sBAAsB,OAAO,EAAE,CAAC,CAAC;YACzE,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAClC,CAAC;aAAM,CAAC;YACN,IAAI,OAAO,EAAE,CAAC;gBACZ,iCAAiC;gBACjC,OAAO,OAAO,CAAC,cAAc,CAAC;gBAC9B,OAAO,OAAO,CAAC,QAAQ,CAAC;gBACxB,OAAO,OAAO,CAAC,MAAM,CAAC;YACxB,CAAC;YAED,MAAM,IAAI,GACR,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YACpF,gBAAM,CAAC,KAAK,CACV,yBAAyB,OAAO,oBAAoB,IAAI,CAAC,UAAU,cAAc,IAAI,CAAC,IAAI,CACxF,IAAI,CACL,EAAE,CACJ,CAAC;YACF,IAAI,MAAM,CAAC;YACX,QAAQ,OAAO,EAAE,CAAC;gBAChB,KAAK,UAAU;oBACb,MAAM,GAAG,CAAC,MAAM,IAAA,mBAAS,EAAC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE;wBAChD,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;qBAC/C,CAAC,CAAqB,CAAC;oBACxB,IAAI,CAAC,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,EAAE,CAAC;wBAClD,MAAM,IAAI,KAAK,CACb,uBAAuB,OAAO,qGAAqG,IAAI,CAAC,SAAS,CAC/I,MAAM,CACP,EAAE,CACJ,CAAC;oBACJ,CAAC;oBACD,MAAM;gBACR,KAAK,oBAAoB;oBACvB,MAAM,GAAG,CAAC,MAAM,IAAA,mBAAS,EAAC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE;wBAChD,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;qBAC/C,CAAC,CAA8B,CAAC;oBACjC,IAAI,CAAC,CAAC,WAAW,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,EAAE,CAAC;wBACrD,MAAM,IAAI,KAAK,CACb,uBAAuB,OAAO,+FAA+F,IAAI,CAAC,SAAS,CACzI,MAAM,CACP,EAAE,CACJ,CAAC;oBACJ,CAAC;oBACD,MAAM;gBACR,KAAK,yBAAyB;oBAC5B,MAAM,GAAG,CAAC,MAAM,IAAA,mBAAS,EAAC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE;wBAChD,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;qBAC/C,CAAC,CAAmC,CAAC;oBACtC,IAAI,CAAC,CAAC,gBAAgB,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,EAAE,CAAC;wBAC1D,MAAM,IAAI,KAAK,CACb,uBAAuB,OAAO,mGAAmG,IAAI,CAAC,SAAS,CAC7I,MAAM,CACP,EAAE,CACJ,CAAC;oBACJ,CAAC;oBACD,MAAM;gBACR;oBACE,MAAM,IAAI,KAAK,CAAC,wBAAwB,OAAO,EAAE,CAAC,CAAC;YACvD,CAAC;YAED,IAAI,IAAA,sBAAc,GAAE,IAAI,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,EAAE,CAAC;gBAC7C,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;YACpD,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,OAA8B;QAC1D,OAAO,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,MAAc;QACnC,OAAO,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,SAAS,EAAE,oBAAoB,CAAC,CAAC;IAC3E,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,MAAc;QACxC,OAAO,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,SAAS,EAAE,yBAAyB,CAAC,CAAC;IAChF,CAAC;CACF;AA5GD,wCA4GC"}

package/dist/src/providers/replicate.d.ts

@@ -1,4 +1,4 @@
-import type { ApiProvider, EnvOverrides, ProviderResponse } from '../types.js';
+import type { ApiModerationProvider, ApiProvider, EnvOverrides, ProviderModerationResponse, ProviderResponse } from '../types.js';
 interface ReplicateCompletionOptions {
     apiKey?: string;
     temperature?: number;
@@ -31,5 +31,9 @@ export declare class ReplicateProvider implements ApiProvider {
     toString(): string;
     callApi(prompt: string): Promise<ProviderResponse>;
 }
+export declare class ReplicateModerationProvider extends ReplicateProvider implements ApiModerationProvider {
+    callModerationApi(prompt: string, assistant: string): Promise<ProviderModerationResponse>;
+}
+export declare const DefaultModerationProvider: ReplicateModerationProvider;
 export {};
 //# sourceMappingURL=replicate.d.ts.map

package/dist/src/providers/replicate.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"replicate.d.ts","sourceRoot":"","sources":["../../../src/providers/replicate.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAE/E,UAAU,0BAA0B;IAClC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd,MAAM,CAAC,EAAE;QACP,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,CAAC;IAGF,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACpB;AAED,qBAAa,iBAAkB,YAAW,WAAW;IACnD,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,GAAG,CAAC;IACf,MAAM,EAAE,0BAA0B,CAAC;gBAGjC,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,0BAA0B,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAcxF,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIZ,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CA6GzD"}
+{"version":3,"file":"replicate.d.ts","sourceRoot":"","sources":["../../../src/providers/replicate.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EACV,qBAAqB,EACrB,WAAW,EACX,YAAY,EAEZ,0BAA0B,EAC1B,gBAAgB,EACjB,MAAM,aAAa,CAAC;AAErB,UAAU,0BAA0B;IAClC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd,MAAM,CAAC,EAAE;QACP,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,CAAC;IAGF,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACpB;AAED,qBAAa,iBAAkB,YAAW,WAAW;IACnD,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,GAAG,CAAC;IACf,MAAM,EAAE,0BAA0B,CAAC;gBAGjC,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,0BAA0B,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAcxF,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIZ,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CA6GzD;AAgBD,qBAAa,2BACX,SAAQ,iBACR,YAAW,qBAAqB;IAE1B,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,0BAA0B,CAAC;CAgFhG;AAED,eAAO,MAAM,yBAAyB,6BAErC,CAAC"}

package/dist/src/providers/replicate.js

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ReplicateProvider = void 0;
+exports.DefaultModerationProvider = exports.ReplicateModerationProvider = exports.ReplicateProvider = void 0;
 const replicate_1 = __importDefault(require("replicate"));
 const node_fetch_1 = __importDefault(require("node-fetch"));
 const logger_1 = __importDefault(require("../logger"));
@@ -125,4 +125,94 @@ class ReplicateProvider {
     }
 }
 exports.ReplicateProvider = ReplicateProvider;
+const LLAMAGUARD_DESCRIPTIONS = {
+    S1: 'Violent Crimes',
+    S2: 'Non-Violent Crimes',
+    S3: 'Sex-Related Crimes',
+    S4: 'Child Sexual Exploitation',
+    S5: 'Specialized Advice',
+    S6: 'Privacy',
+    S7: 'Intellectual Property',
+    S8: 'Indiscriminate Weapons',
+    S9: 'Hate',
+    S10: 'Suicide & Self-Harm',
+    S11: 'Sexual Content',
+};
+class ReplicateModerationProvider extends ReplicateProvider {
+    async callModerationApi(prompt, assistant) {
+        if (!this.apiKey) {
+            throw new Error('Replicate API key is not set. Set the REPLICATE_API_TOKEN environment variable or or add `apiKey` to the provider config.');
+        }
+        let cache;
+        let cacheKey;
+        if ((0, cache_1.isCacheEnabled)()) {
+            cache = await (0, cache_1.getCache)();
+            cacheKey = `replicate:${this.modelName}:${JSON.stringify(this.config)}:${prompt}:${assistant}`;
+            // Try to get the cached response
+            const cachedResponse = await cache.get(cacheKey);
+            if (cachedResponse) {
+                logger_1.default.debug(`Returning cached response for ${prompt}: ${cachedResponse}`);
+                return JSON.parse(cachedResponse);
+            }
+        }
+        const replicate = new replicate_1.default({
+            auth: this.apiKey,
+            fetch: node_fetch_1.default,
+        });
+        logger_1.default.debug(`Calling Replicate moderation API: prompt [${prompt}] assistant [${assistant}]`);
+        let output;
+        try {
+            const data = {
+                input: {
+                    prompt,
+                    assistant,
+                },
+            };
+            const resp = await replicate.run(this.modelName, data);
+            // Replicate SDK seems to be mis-typed for this type of model.
+            output = resp;
+        }
+        catch (err) {
+            return {
+                error: `API call error: ${String(err)}`,
+            };
+        }
+        logger_1.default.debug(`\tReplicate moderation API response: ${JSON.stringify(output)}`);
+        try {
+            if (!output) {
+                throw new Error('API response error: no output');
+            }
+            const [safeString, codes] = output.split('\n');
+            const saveCache = async () => {
+                if (cache && cacheKey) {
+                    await cache.set(cacheKey, JSON.stringify(output));
+                }
+            };
+            const flags = [];
+            if (safeString === 'safe') {
+                await saveCache();
+            }
+            else {
+                const splits = codes.split(',');
+                for (const code of splits) {
+                    if (LLAMAGUARD_DESCRIPTIONS[code]) {
+                        flags.push({
+                            code,
+                            description: `${LLAMAGUARD_DESCRIPTIONS[code]} (${code})`,
+                            confidence: 1,
+                        });
+                    }
+                }
+            }
+            return { flags };
+        }
+        catch (err) {
+            return {
+                error: `API response error: ${String(err)}: ${JSON.stringify(output)}`,
+            };
+        }
+    }
+}
+exports.ReplicateModerationProvider = ReplicateModerationProvider;
+exports.DefaultModerationProvider = new ReplicateModerationProvider('meta/meta-llama-guard-2-8b:b063023ee937f28e922982abdbf97b041ffe34ad3b35a53d33e1d74bb19b36c4');
 //# sourceMappingURL=replicate.js.map

package/dist/src/providers/replicate.js.map

@@ -1 +1 @@
-{"version":3,"file":"replicate.js","sourceRoot":"","sources":["../../../src/providers/replicate.ts"],"names":[],"mappings":";;;;;;AAAA,0DAAkC;AAElC,4DAA+B;AAC/B,uDAA+B;AAC/B,oCAAoD;AA0BpD,MAAa,iBAAiB;IAM5B,YACE,SAAiB,EACjB,UAAoF,EAAE;QAEtF,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC;QACpC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM;YACT,MAAM,EAAE,MAAM;gBACd,GAAG,EAAE,iBAAiB;gBACtB,GAAG,EAAE,mBAAmB;gBACxB,OAAO,CAAC,GAAG,CAAC,mBAAmB;gBAC/B,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QAChC,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;QAC3B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;IACpC,CAAC;IAED,EAAE;QACA,OAAO,aAAa,IAAI,CAAC,SAAS,EAAE,CAAC;IACvC,CAAC;IAED,QAAQ;QACN,OAAO,uBAAuB,IAAI,CAAC,SAAS,GAAG,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YAChB,MAAM,IAAI,KAAK,CACb,2HAA2H,CAC5H,CAAC;SACH;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE;YAC9B,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC;SAC7C;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE;YAC9B,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC;SAC7C;QAED,IAAI,KAAK,CAAC;QACV,IAAI,QAAQ,CAAC;QACb,IAAI,IAAA,sBAAc,GAAE,EAAE;YACpB,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;YACzB,QAAQ,GAAG,aAAa,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;YAElF,iCAAiC;YACjC,MAAM,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAEjD,IAAI,cAAc,EAAE;gBAClB,gBAAM,CAAC,KAAK,CAAC,iCAAiC,MAAM,KAAK,cAAc,EAAE,CAAC,CAAC;gBAC3E,OAAO,IAAI,CAAC,KAAK,CAAC,cAAwB,CAAC,CAAC;aAC7C;SACF;QAED,MAAM,SAAS,GAAG,IAAI,mBAAS,CAAC;YAC9B,IAAI,EAAE,IAAI,CAAC,MAAM;YACjB,KAAK,EAAE,oBAAY;SACpB,CAAC,CAAC;QAEH,gBAAM,CAAC,KAAK,CAAC,sBAAsB,MAAM,EAAE,CAAC,CAAC;QAC7C,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,MAAM,QAAQ,GAAG,CACf,WAAwC,EACxC,MAAc,EACd,YAAY,CAAC,GAAQ,EAAE,EAAE,CAAC,GAAG,EAC7B,EAAE;gBACF,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBACrC,IAAI,WAAW,KAAK,SAAS,EAAE;oBAC7B,OAAO,WAAW,CAAC;iBACpB;qBAAM,IAAI,QAAQ,KAAK,SAAS,EAAE;oBACjC,OAAO,SAAS,CAAC,QAAQ,CAAC,CAAC;iBAC5B;gBACD,OAAO,SAAS,CAAC;YACnB,CAAC,CAAC;YAEF,MAAM,YAAY,GAAG;gBACnB,UAAU,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,sBAAsB,EAAE,QAAQ,CAAC;gBAC9E,cAAc,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,0BAA0B,EAAE,QAAQ,CAAC;gBAC1F,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,uBAAuB,EAAE,UAAU,CAAC;gBACnF,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,iBAAiB,EAAE,UAAU,CAAC;gBACjE,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,iBAAiB,EAAE,QAAQ,CAAC;gBAC/D,kBAAkB,EAAE,QAAQ,CAC1B,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAC9B,8BAA8B,EAC9B,UAAU,CACX;gBACD,aAAa,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,yBAAyB,CAAC;gBAC7E,cAAc,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,0BAA0B,CAAC;gBAChF,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,gBAAgB,EAAE,QAAQ,CAAC;aAC7D,CAAC;YAEF,MAAM,IAAI,GAAG;gBACX,KAAK,EAAE;oBACL,GAAG,IAAI,CAAC,MAAM;oBACd,MAAM;oBACN,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;iBACxF;aACF,CAAC;YACF,QAAQ,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,SAAgB,EAAE,IAAI,CAAC,CAAC;SAC7D;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;SACH;QACD,gBAAM,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QACtE,IAAI;YACF,IAAI,eAAe,CAAC;YACpB,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;gBAC3B,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,OAAO,QAAQ,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE;oBAC5D,eAAe,GAAG,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;iBACrC;qBAAM;oBACL,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;iBAC5C;aACF;iBAAM,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE;gBACvC,eAAe,GAAG,QAAQ,CAAC;aAC5B;iBAAM;gBACL,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;aAC5C;YAED,MAAM,MAAM,GAAG;gBACb,MAAM,EAAE,eAAe;gBACvB,UAAU,EAAE,EAAE,EAAE,uDAAuD;aACxE,CAAC;YACF,IAAI,KAAK,IAAI,QAAQ,EAAE;gBACrB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;aACnD;YACD,OAAO,MAAM,CAAC;SACf;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,uBAAuB,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE;aACzE,CAAC;SACH;IACH,CAAC;CACF;AA3ID,8CA2IC"}
+{"version":3,"file":"replicate.js","sourceRoot":"","sources":["../../../src/providers/replicate.ts"],"names":[],"mappings":";;;;;;AAAA,0DAAkC;AAElC,4DAA+B;AAC/B,uDAA+B;AAC/B,oCAAoD;AAmCpD,MAAa,iBAAiB;IAM5B,YACE,SAAiB,EACjB,UAAoF,EAAE;QAEtF,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC;QACpC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM;YACT,MAAM,EAAE,MAAM;gBACd,GAAG,EAAE,iBAAiB;gBACtB,GAAG,EAAE,mBAAmB;gBACxB,OAAO,CAAC,GAAG,CAAC,mBAAmB;gBAC/B,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QAChC,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;QAC3B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;IACpC,CAAC;IAED,EAAE;QACA,OAAO,aAAa,IAAI,CAAC,SAAS,EAAE,CAAC;IACvC,CAAC;IAED,QAAQ;QACN,OAAO,uBAAuB,IAAI,CAAC,SAAS,GAAG,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CACb,2HAA2H,CAC5H,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;YAC/B,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC;QAC9C,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;YAC/B,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC;QAC9C,CAAC;QAED,IAAI,KAAK,CAAC;QACV,IAAI,QAAQ,CAAC;QACb,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;YACrB,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;YACzB,QAAQ,GAAG,aAAa,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;YAElF,iCAAiC;YACjC,MAAM,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAEjD,IAAI,cAAc,EAAE,CAAC;gBACnB,gBAAM,CAAC,KAAK,CAAC,iCAAiC,MAAM,KAAK,cAAc,EAAE,CAAC,CAAC;gBAC3E,OAAO,IAAI,CAAC,KAAK,CAAC,cAAwB,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,mBAAS,CAAC;YAC9B,IAAI,EAAE,IAAI,CAAC,MAAM;YACjB,KAAK,EAAE,oBAAY;SACpB,CAAC,CAAC;QAEH,gBAAM,CAAC,KAAK,CAAC,sBAAsB,MAAM,EAAE,CAAC,CAAC;QAC7C,IAAI,QAAQ,CAAC;QACb,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,CACf,WAAwC,EACxC,MAAc,EACd,YAAY,CAAC,GAAQ,EAAE,EAAE,CAAC,GAAG,EAC7B,EAAE;gBACF,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBACrC,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;oBAC9B,OAAO,WAAW,CAAC;gBACrB,CAAC;qBAAM,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;oBAClC,OAAO,SAAS,CAAC,QAAQ,CAAC,CAAC;gBAC7B,CAAC;gBACD,OAAO,SAAS,CAAC;YACnB,CAAC,CAAC;YAEF,MAAM,YAAY,GAAG;gBACnB,UAAU,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,sBAAsB,EAAE,QAAQ,CAAC;gBAC9E,cAAc,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,0BAA0B,EAAE,QAAQ,CAAC;gBAC1F,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,uBAAuB,EAAE,UAAU,CAAC;gBACnF,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,iBAAiB,EAAE,UAAU,CAAC;gBACjE,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,iBAAiB,EAAE,QAAQ,CAAC;gBAC/D,kBAAkB,EAAE,QAAQ,CAC1B,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAC9B,8BAA8B,EAC9B,UAAU,CACX;gBACD,aAAa,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,yBAAyB,CAAC;gBAC7E,cAAc,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,0BAA0B,CAAC;gBAChF,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,gBAAgB,EAAE,QAAQ,CAAC;aAC7D,CAAC;YAEF,MAAM,IAAI,GAAG;gBACX,KAAK,EAAE;oBACL,GAAG,IAAI,CAAC,MAAM;oBACd,MAAM;oBACN,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;iBACxF;aACF,CAAC;YACF,QAAQ,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,SAAgB,EAAE,IAAI,CAAC,CAAC;QAC9D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;QACJ,CAAC;QACD,gBAAM,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QACtE,IAAI,CAAC;YACH,IAAI,eAAe,CAAC;YACpB,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC5B,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,OAAO,QAAQ,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;oBAC7D,eAAe,GAAG,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACtC,CAAC;qBAAM,CAAC;oBACN,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;gBAC7C,CAAC;YACH,CAAC;iBAAM,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACxC,eAAe,GAAG,QAAQ,CAAC;YAC7B,CAAC;iBAAM,CAAC;gBACN,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC7C,CAAC;YAED,MAAM,MAAM,GAAG;gBACb,MAAM,EAAE,eAAe;gBACvB,UAAU,EAAE,EAAE,EAAE,uDAAuD;aACxE,CAAC;YACF,IAAI,KAAK,IAAI,QAAQ,EAAE,CAAC;gBACtB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;YACpD,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,uBAAuB,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE;aACzE,CAAC;QACJ,CAAC;IACH,CAAC;CACF;AA3ID,8CA2IC;AAED,MAAM,uBAAuB,GAA8B;IACzD,EAAE,EAAE,gBAAgB;IACpB,EAAE,EAAE,oBAAoB;IACxB,EAAE,EAAE,oBAAoB;IACxB,EAAE,EAAE,2BAA2B;IAC/B,EAAE,EAAE,oBAAoB;IACxB,EAAE,EAAE,SAAS;IACb,EAAE,EAAE,uBAAuB;IAC3B,EAAE,EAAE,wBAAwB;IAC5B,EAAE,EAAE,MAAM;IACV,GAAG,EAAE,qBAAqB;IAC1B,GAAG,EAAE,gBAAgB;CACtB,CAAC;AAEF,MAAa,2BACX,SAAQ,iBAAiB;IAGzB,KAAK,CAAC,iBAAiB,CAAC,MAAc,EAAE,SAAiB;QACvD,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CACb,2HAA2H,CAC5H,CAAC;QACJ,CAAC;QAED,IAAI,KAAwB,CAAC;QAC7B,IAAI,QAA4B,CAAC;QACjC,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;YACrB,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;YACzB,QAAQ,GAAG,aAAa,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CACtD,IAAI,CAAC,MAAM,CACZ,IAAI,MAAM,IAAI,SAAS,EAAE,CAAC;YAE3B,iCAAiC;YACjC,MAAM,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAEjD,IAAI,cAAc,EAAE,CAAC;gBACnB,gBAAM,CAAC,KAAK,CAAC,iCAAiC,MAAM,KAAK,cAAc,EAAE,CAAC,CAAC;gBAC3E,OAAO,IAAI,CAAC,KAAK,CAAC,cAAwB,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,mBAAS,CAAC;YAC9B,IAAI,EAAE,IAAI,CAAC,MAAM;YACjB,KAAK,EAAE,oBAAY;SACpB,CAAC,CAAC;QAEH,gBAAM,CAAC,KAAK,CAAC,6CAA6C,MAAM,gBAAgB,SAAS,GAAG,CAAC,CAAC;QAC9F,IAAI,MAA0B,CAAC;QAC/B,IAAI,CAAC;YACH,MAAM,IAAI,GAAG;gBACX,KAAK,EAAE;oBACL,MAAM;oBACN,SAAS;iBACV;aACF,CAAC;YACF,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,SAAgB,EAAE,IAAI,CAAC,CAAC;YAC9D,8DAA8D;YAC9D,MAAM,GAAG,IAAyB,CAAC;QACrC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;QACJ,CAAC;QACD,gBAAM,CAAC,KAAK,CAAC,wCAAwC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC/E,IAAI,CAAC;YACH,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;YACnD,CAAC;YACD,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC/C,MAAM,SAAS,GAAG,KAAK,IAAI,EAAE;gBAC3B,IAAI,KAAK,IAAI,QAAQ,EAAE,CAAC;oBACtB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;gBACpD,CAAC;YACH,CAAC,CAAC;YAEF,MAAM,KAAK,GAAqB,EAAE,CAAC;YACnC,IAAI,UAAU,KAAK,MAAM,EAAE,CAAC;gBAC1B,MAAM,SAAS,EAAE,CAAC;YACpB,CAAC;iBAAM,CAAC;gBACN,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBAChC,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;oBAC1B,IAAI,uBAAuB,CAAC,IAAI,CAAC,EAAE,CAAC;wBAClC,KAAK,CAAC,IAAI,CAAC;4BACT,IAAI;4BACJ,WAAW,EAAE,GAAG,uBAAuB,CAAC,IAAI,CAAC,KAAK,IAAI,GAAG;4BACzD,UAAU,EAAE,CAAC;yBACd,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;YACD,OAAO,EAAE,KAAK,EAAE,CAAC;QACnB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,uBAAuB,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE;aACvE,CAAC;QACJ,CAAC;IACH,CAAC;CACF;AApFD,kEAoFC;AAEY,QAAA,yBAAyB,GAAG,IAAI,2BAA2B,CACtE,6FAA6F,CAC9F,CAAC"}

package/dist/src/providers/scriptCompletion.js.map

@@ -1 +1 @@
-{"version":3,"file":"scriptCompletion.js","sourceRoot":"","sources":["../../../src/providers/scriptCompletion.ts"],"names":[],"mappings":";;;;;;AAAA,iDAAyC;AAEzC,oEAAuC;AAEvC,uDAA+B;AAC/B,oCAAoD;AACpD,kCAA4C;AAS5C,MAAM,WAAW,GAAG,wCAAwC,CAAC;AAE7D,SAAS,SAAS,CAAC,IAAY;IAC7B,OAAO,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;AACvC,CAAC;AAED,MAAa,wBAAwB;IACnC,YAAoB,UAAkB,EAAU,OAAyB;QAArD,eAAU,GAAV,UAAU,CAAQ;QAAU,YAAO,GAAP,OAAO,CAAkB;IAAG,CAAC;IAE7E,EAAE;QACA,OAAO,QAAQ,IAAI,CAAC,UAAU,EAAE,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,OAA8B;QAC1D,MAAM,QAAQ,GAAG,QAAQ,IAAI,CAAC,UAAU,IAAI,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QACrF,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;QAC/B,IAAI,YAAY,CAAC;QAEjB,IAAI,IAAA,sBAAc,GAAE,EAAE;YACpB,YAAY,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;SAC1C;QAED,IAAI,YAAY,EAAE;YAChB,gBAAM,CAAC,KAAK,CAAC,sCAAsC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;YACtE,OAAO,IAAI,CAAC,KAAK,CAAC,YAAsB,CAAC,CAAC;SAC3C;aAAM;YACL,OAAO,IAAI,OAAO,CAAmB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACvD,MAAM,gBAAgB,GAAG,+BAA+B,CAAC;gBACzD,IAAI,KAAK,CAAC;gBACV,MAAM,WAAW,GAAG,EAAE,CAAC;gBAEvB,OAAO,CAAC,KAAK,GAAG,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,KAAK,IAAI,EAAE;oBAChE,qEAAqE;oBACrE,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE;wBACZ,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;qBAC5B;yBAAM,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE;wBACnB,6EAA6E;wBAC7E,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;qBAC5B;yBAAM;wBACL,kCAAkC;wBAClC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;qBAC5B;iBACF;gBACD,MAAM,OAAO,GAAG,WAAW,CAAC,KAAK,EAAE,CAAC;gBACpC,IAAA,wBAAS,EAAC,OAAO,EAAE,iCAAiC,CAAC,CAAC;gBACtD,oCAAoC;gBACpC,OAAO,OAAO,EAAE,cAAc,CAAC;gBAC/B,OAAO,OAAO,EAAE,QAAQ,CAAC;gBACzB,OAAO,OAAO,EAAE,MAAM,CAAC;gBACvB,MAAM,UAAU,GAAG,WAAW,CAAC,MAAM,CAAC;oBACpC,MAAM;oBACN,IAAA,wBAAiB,EAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC;oBACrC,IAAA,wBAAiB,EAAC,OAAO,IAAI,EAAE,CAAC;iBACjC,CAAC,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAE3F,IAAA,wBAAQ,EAAC,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE;oBACrE,IAAI,KAAK,EAAE;wBACT,gBAAM,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,UAAU,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;wBAC1E,MAAM,CAAC,KAAK,CAAC,CAAC;wBACd,OAAO;qBACR;oBACD,MAAM,cAAc,GAAG,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;oBACxD,MAAM,WAAW,GAAG,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;oBACrD,IAAI,WAAW,EAAE;wBACf,gBAAM,CAAC,KAAK,CAAC,4BAA4B,IAAI,CAAC,UAAU,KAAK,WAAW,EAAE,CAAC,CAAC;wBAC5E,IAAI,CAAC,cAAc,EAAE;4BACnB,MAAM,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;4BAC/B,OAAO;yBACR;qBACF;oBACD,gBAAM,CAAC,KAAK,CAAC,sBAAsB,IAAI,CAAC,UAAU,KAAK,cAAc,EAAE,CAAC,CAAC;oBACzE,MAAM,MAAM,GAAG,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;oBAC1C,IAAI,IAAA,sBAAc,GAAE,EAAE;wBACpB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;qBACnD;oBACD,OAAO,CAAC,MAAM,CAAC,CAAC;gBAClB,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;SACJ;IACH,CAAC;CACF;AA3ED,4DA2EC"}
+{"version":3,"file":"scriptCompletion.js","sourceRoot":"","sources":["../../../src/providers/scriptCompletion.ts"],"names":[],"mappings":";;;;;;AAAA,iDAAyC;AAEzC,oEAAuC;AAEvC,uDAA+B;AAC/B,oCAAoD;AACpD,kCAA4C;AAS5C,MAAM,WAAW,GAAG,wCAAwC,CAAC;AAE7D,SAAS,SAAS,CAAC,IAAY;IAC7B,OAAO,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;AACvC,CAAC;AAED,MAAa,wBAAwB;IACnC,YAAoB,UAAkB,EAAU,OAAyB;QAArD,eAAU,GAAV,UAAU,CAAQ;QAAU,YAAO,GAAP,OAAO,CAAkB;IAAG,CAAC;IAE7E,EAAE;QACA,OAAO,QAAQ,IAAI,CAAC,UAAU,EAAE,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,OAA8B;QAC1D,MAAM,QAAQ,GAAG,QAAQ,IAAI,CAAC,UAAU,IAAI,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QACrF,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;QAC/B,IAAI,YAAY,CAAC;QAEjB,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;YACrB,YAAY,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,YAAY,EAAE,CAAC;YACjB,gBAAM,CAAC,KAAK,CAAC,sCAAsC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;YACtE,OAAO,IAAI,CAAC,KAAK,CAAC,YAAsB,CAAC,CAAC;QAC5C,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,OAAO,CAAmB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACvD,MAAM,gBAAgB,GAAG,+BAA+B,CAAC;gBACzD,IAAI,KAAK,CAAC;gBACV,MAAM,WAAW,GAAG,EAAE,CAAC;gBAEvB,OAAO,CAAC,KAAK,GAAG,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;oBACjE,qEAAqE;oBACrE,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;wBACb,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC7B,CAAC;yBAAM,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;wBACpB,6EAA6E;wBAC7E,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC7B,CAAC;yBAAM,CAAC;wBACN,kCAAkC;wBAClC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC7B,CAAC;gBACH,CAAC;gBACD,MAAM,OAAO,GAAG,WAAW,CAAC,KAAK,EAAE,CAAC;gBACpC,IAAA,wBAAS,EAAC,OAAO,EAAE,iCAAiC,CAAC,CAAC;gBACtD,oCAAoC;gBACpC,OAAO,OAAO,EAAE,cAAc,CAAC;gBAC/B,OAAO,OAAO,EAAE,QAAQ,CAAC;gBACzB,OAAO,OAAO,EAAE,MAAM,CAAC;gBACvB,MAAM,UAAU,GAAG,WAAW,CAAC,MAAM,CAAC;oBACpC,MAAM;oBACN,IAAA,wBAAiB,EAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC;oBACrC,IAAA,wBAAiB,EAAC,OAAO,IAAI,EAAE,CAAC;iBACjC,CAAC,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAE3F,IAAA,wBAAQ,EAAC,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE;oBACrE,IAAI,KAAK,EAAE,CAAC;wBACV,gBAAM,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,UAAU,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;wBAC1E,MAAM,CAAC,KAAK,CAAC,CAAC;wBACd,OAAO;oBACT,CAAC;oBACD,MAAM,cAAc,GAAG,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;oBACxD,MAAM,WAAW,GAAG,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;oBACrD,IAAI,WAAW,EAAE,CAAC;wBAChB,gBAAM,CAAC,KAAK,CAAC,4BAA4B,IAAI,CAAC,UAAU,KAAK,WAAW,EAAE,CAAC,CAAC;wBAC5E,IAAI,CAAC,cAAc,EAAE,CAAC;4BACpB,MAAM,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;4BAC/B,OAAO;wBACT,CAAC;oBACH,CAAC;oBACD,gBAAM,CAAC,KAAK,CAAC,sBAAsB,IAAI,CAAC,UAAU,KAAK,cAAc,EAAE,CAAC,CAAC;oBACzE,MAAM,MAAM,GAAG,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;oBAC1C,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;wBACrB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;oBACpD,CAAC;oBACD,OAAO,CAAC,MAAM,CAAC,CAAC;gBAClB,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;CACF;AA3ED,4DA2EC"}

package/dist/src/providers/shared.js.map

@@ -1 +1 @@
-{"version":3,"file":"shared.js","sourceRoot":"","sources":["../../../src/providers/shared.ts"],"names":[],"mappings":";;;;;;AAAA,sDAA2B;AAEd,QAAA,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB;IAC9D,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,EAAE,CAAC;IAC9C,CAAC,CAAC,MAAO,CAAC;AAEZ,SAAgB,eAAe,CAAI,MAAc,EAAE,YAAe;IAChE,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;IACpC,IAAI,aAAa,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE;QACvC,IAAI;YACF,oDAAoD;YACpD,OAAO,iBAAI,CAAC,IAAI,CAAC,MAAM,CAAM,CAAC;SAC/B;QAAC,OAAO,GAAG,EAAE;YACZ,MAAM,IAAI,KAAK,CAAC,sDAAsD,GAAG,OAAO,MAAM,EAAE,CAAC,CAAC;SAC3F;KACF;SAAM;QACL,IAAI;YACF,WAAW;YACX,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAM,CAAC;SAChC;QAAC,OAAO,GAAG,EAAE;YACZ,IACE,OAAO,CAAC,GAAG,CAAC,8BAA8B;gBAC1C,aAAa,CAAC,UAAU,CAAC,GAAG,CAAC;gBAC7B,aAAa,CAAC,UAAU,CAAC,GAAG,CAAC,EAC7B;gBACA,MAAM,IAAI,KAAK,CAAC,sDAAsD,GAAG,OAAO,MAAM,EAAE,CAAC,CAAC;aAC3F;YACD,0CAA0C;YAC1C,OAAO,YAAY,CAAC;SACrB;KACF;AACH,CAAC;AAzBD,0CAyBC;AAED,SAAgB,WAAW,CAAC,GAAW;IACrC,OAAO,GAAG,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;AACnG,CAAC;AAFD,kCAEC"}
+{"version":3,"file":"shared.js","sourceRoot":"","sources":["../../../src/providers/shared.ts"],"names":[],"mappings":";;;;;;AAAA,sDAA2B;AAEd,QAAA,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB;IAC9D,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,EAAE,CAAC;IAC9C,CAAC,CAAC,MAAO,CAAC;AAEZ,SAAgB,eAAe,CAAI,MAAc,EAAE,YAAe;IAChE,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;IACpC,IAAI,aAAa,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QACxC,IAAI,CAAC;YACH,oDAAoD;YACpD,OAAO,iBAAI,CAAC,IAAI,CAAC,MAAM,CAAM,CAAC;QAChC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,sDAAsD,GAAG,OAAO,MAAM,EAAE,CAAC,CAAC;QAC5F,CAAC;IACH,CAAC;SAAM,CAAC;QACN,IAAI,CAAC;YACH,WAAW;YACX,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAM,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IACE,OAAO,CAAC,GAAG,CAAC,8BAA8B;gBAC1C,aAAa,CAAC,UAAU,CAAC,GAAG,CAAC;gBAC7B,aAAa,CAAC,UAAU,CAAC,GAAG,CAAC,EAC7B,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,sDAAsD,GAAG,OAAO,MAAM,EAAE,CAAC,CAAC;YAC5F,CAAC;YACD,0CAA0C;YAC1C,OAAO,YAAY,CAAC;QACtB,CAAC;IACH,CAAC;AACH,CAAC;AAzBD,0CAyBC;AAED,SAAgB,WAAW,CAAC,GAAW;IACrC,OAAO,GAAG,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;AACnG,CAAC;AAFD,kCAEC"}

package/dist/src/providers/vertex.d.ts

@@ -31,6 +31,44 @@ interface VertexCompletionOptions {
         topP?: number;
         topK?: number;
     };
+    toolConfig?: {
+        functionCallingConfig?: {
+            mode?: 'MODE_UNSPECIFIED' | 'AUTO' | 'ANY' | 'NONE';
+            allowedFunctionNames?: string[];
+        };
+    };
+    systemInstruction?: Content;
+}
+interface Content {
+    parts: Part[];
+    role?: string;
+}
+interface Part {
+    text?: string;
+    inlineData?: Blob;
+    functionCall?: FunctionCall;
+    functionResponse?: FunctionResponse;
+    fileData?: FileData;
+}
+interface Blob {
+    mimeType: string;
+    data: string;
+}
+interface FunctionCall {
+    name: string;
+    args?: {
+        [key: string]: any;
+    };
+}
+interface FunctionResponse {
+    name: string;
+    response: {
+        [key: string]: any;
+    };
+}
+interface FileData {
+    mimeType?: string;
+    fileUri: string;
 }
 declare class VertexGenericProvider implements ApiProvider {
     modelName: string;

package/dist/src/providers/vertex.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"vertex.d.ts","sourceRoot":"","sources":["../../../src/providers/vertex.ts"],"names":[],"mappings":"AAeA,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,EAAc,MAAM,aAAa,CAAC;AAuB3F,UAAU,uBAAuB;IAC/B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAC/C,cAAc,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAC7D,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd,gBAAgB,CAAC,EAAE;QACjB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAA;SAAE,EAAE,CAAC;QAC/C,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;KACf,CAAC;CACH;AAED,cAAM,qBAAsB,YAAW,WAAW;IAChD,SAAS,EAAE,MAAM,CAAC;IAElB,MAAM,EAAE,uBAAuB,CAAC;IAChC,GAAG,CAAC,EAAE,YAAY,CAAC;gBAGjB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,uBAAuB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IASrF,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIlB,UAAU,IAAI,MAAM,GAAG,SAAS;IAS1B,YAAY;IASlB,SAAS,IAAI,MAAM,GAAG,SAAS;IAI/B,SAAS,IAAI,MAAM;IAMnB,YAAY,IAAI,MAAM,GAAG,SAAS;IAU5B,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAGzD;AAED,qBAAa,kBAAmB,SAAQ,qBAAqB;IAG3D,MAAM,CAAC,WAAW,WA0BhB;gBAGA,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,uBAAuB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAQ/E,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAOlD,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA4HxD,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CA0F9D"}
+{"version":3,"file":"vertex.d.ts","sourceRoot":"","sources":["../../../src/providers/vertex.ts"],"names":[],"mappings":"AAcA,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,EAAc,MAAM,aAAa,CAAC;AAuB3F,UAAU,uBAAuB;IAC/B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IAGnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAC/C,cAAc,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAC7D,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd,gBAAgB,CAAC,EAAE;QACjB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAA;SAAE,EAAE,CAAC;QAC/C,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;KACf,CAAC;IAEF,UAAU,CAAC,EAAE;QACX,qBAAqB,CAAC,EAAE;YACtB,IAAI,CAAC,EAAE,kBAAkB,GAAG,MAAM,GAAG,KAAK,GAAG,MAAM,CAAC;YACpD,oBAAoB,CAAC,EAAE,MAAM,EAAE,CAAC;SACjC,CAAC;KACH,CAAC;IAEF,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC7B;AAED,UAAU,OAAO;IACf,KAAK,EAAE,IAAI,EAAE,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,UAAU,IAAI;IACZ,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,IAAI,CAAC;IAClB,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;IACpC,QAAQ,CAAC,EAAE,QAAQ,CAAC;CACrB;AAED,UAAU,IAAI;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;CACd;AAED,UAAU,YAAY;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC;CAC/B;AAED,UAAU,gBAAgB;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC;CAClC;AAED,UAAU,QAAQ;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,cAAM,qBAAsB,YAAW,WAAW;IAChD,SAAS,EAAE,MAAM,CAAC;IAElB,MAAM,EAAE,uBAAuB,CAAC;IAChC,GAAG,CAAC,EAAE,YAAY,CAAC;gBAGjB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,uBAAuB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IASrF,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIlB,UAAU,IAAI,MAAM,GAAG,SAAS;IAS1B,YAAY;IASlB,SAAS,IAAI,MAAM,GAAG,SAAS;IAI/B,SAAS,IAAI,MAAM;IAMnB,YAAY,IAAI,MAAM,GAAG,SAAS;IAU5B,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAGzD;AAED,qBAAa,kBAAmB,SAAQ,qBAAqB;IAG3D,MAAM,CAAC,WAAW,WA0BhB;gBAGA,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,uBAAuB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAQ/E,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAOlD,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAgIxD,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CA0F9D"}

package/dist/src/providers/vertex.js

@@ -132,7 +132,11 @@ class VertexChatProvider extends VertexGenericProvider {
                 topK: this.config.topK,
                 ...this.config.generationConfig,
             },
-            safetySettings: this.config.safetySettings,
+            ...(this.config.safetySettings ? { safetySettings: this.config.safetySettings } : {}),
+            ...(this.config.toolConfig ? { toolConfig: this.config.toolConfig } : {}),
+            ...(this.config.systemInstruction
+                ? { systemInstruction: this.config.systemInstruction }
+                : {}),
         };
         logger_1.default.debug(`Preparing to call Google Vertex API (Gemini) with body: ${JSON.stringify(body)}`);
         const cache = await (0, cache_1.getCache)();

package/dist/src/providers/vertex.js.map

@@ -1 +1 @@
-{"version":3,"file":"vertex.js","sourceRoot":"","sources":["../../../src/providers/vertex.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,uDAA+B;AAE/B,qCAA+D;AAC/D,oCAAoD;AAEpD,6CAMsB;AAMtB,IAAI,UAAkC,CAAC;AACvC,KAAK,UAAU,eAAe;IAC5B,IAAI,CAAC,UAAU,EAAE;QACf,IAAI,UAAU,CAAC;QACf,IAAI;YACF,MAAM,cAAc,GAAG,wDAAa,qBAAqB,GAAC,CAAC;YAC3D,UAAU,GAAG,cAAc,CAAC,UAAU,CAAC;SACxC;QAAC,OAAO,GAAG,EAAE;YACZ,MAAM,IAAI,KAAK,CACb,kHAAkH,CACnH,CAAC;SACH;QACD,UAAU,GAAG,IAAI,UAAU,CAAC;YAC1B,MAAM,EAAE,gDAAgD;SACzD,CAAC,CAAC;KACJ;IACD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,SAAS,EAAE,CAAC;IAC5C,MAAM,SAAS,GAAG,MAAM,UAAU,CAAC,YAAY,EAAE,CAAC;IAClD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;AAC/B,CAAC;AA6BD,MAAM,qBAAqB;IAMzB,YACE,SAAiB,EACjB,UAAiF,EAAE;QAEnF,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC;QACpC,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;QAC3B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;IACpC,CAAC;IAED,EAAE;QACA,OAAO,UAAU,IAAI,CAAC,SAAS,EAAE,CAAC;IACpC,CAAC;IAED,QAAQ;QACN,OAAO,2BAA2B,IAAI,CAAC,SAAS,GAAG,CAAC;IACtD,CAAC;IAED,UAAU;QACR,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,OAAO;YACnB,IAAI,CAAC,GAAG,EAAE,eAAe;YACzB,OAAO,CAAC,GAAG,CAAC,eAAe;YAC3B,GAAG,IAAI,CAAC,SAAS,EAAE,4BAA4B,CAChD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,OAAO,CACL,CAAC,MAAM,eAAe,EAAE,CAAC,CAAC,SAAS;YACnC,IAAI,CAAC,MAAM,CAAC,SAAS;YACrB,IAAI,CAAC,GAAG,EAAE,iBAAiB;YAC3B,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAC9B,CAAC;IACJ,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,cAAc,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;IACtF,CAAC;IAED,SAAS;QACP,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,aAAa,IAAI,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,aAAa,CAC5F,CAAC;IACJ,CAAC;IAED,YAAY;QACV,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,SAAS;YACrB,IAAI,CAAC,GAAG,EAAE,gBAAgB;YAC1B,OAAO,CAAC,GAAG,CAAC,gBAAgB;YAC5B,QAAQ,CACT,CAAC;IACJ,CAAC;IAED,wDAAwD;IACxD,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF;AAED,MAAa,kBAAmB,SAAQ,qBAAqB;IA+B3D,YACE,SAAiB,EACjB,UAAiF,EAAE;QAEnF,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE;YACvD,gBAAM,CAAC,IAAI,CAAC,2CAA2C,SAAS,EAAE,CAAC,CAAC;SACrE;QACD,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE;YACrC,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;SACnC;QACD,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAc;QAChC,0FAA0F;QAC1F,IAAI,QAAQ,GAAG,IAAA,wBAAe,EAAC,MAAM,EAAE;YACrC,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE;gBACL,IAAI,EAAE,MAAM;aACb;SACF,CAAC,CAAC;QACH,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,OAAO,EAAE,GAAG,IAAA,sCAAyB,EAAC,QAAQ,CAAC,CAAC;QACnF,IAAI,OAAO,EAAE;YACX,gBAAM,CAAC,KAAK,CAAC,yCAAyC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAClF,QAAQ,GAAG,eAAe,CAAC;SAC5B;QAED,iEAAiE;QACjE,MAAM,IAAI,GAAG;YACX,QAAQ;YACR,gBAAgB,EAAE;gBAChB,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;gBAC5B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC9B,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;gBACtC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;gBAC5C,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB;aAChC;YACD,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;SAC3C,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,2DAA2D,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEhG,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,iBAAiB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;QAEzD,IAAI,cAAc,CAAC;QACnB,IAAI,IAAA,sBAAc,GAAE,EAAE;YACpB,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC3C,IAAI,cAAc,EAAE;gBAClB,gBAAM,CAAC,KAAK,CAAC,yCAAyC,MAAM,EAAE,CAAC,CAAC;gBAChE,MAAM,oBAAoB,GAAG,IAAI,CAAC,KAAK,CAAC,cAAwB,CAAC,CAAC;gBAClE,MAAM,UAAU,GAAG,oBAAoB,CAAC,UAAwB,CAAC;gBACjE,IAAI,UAAU,EAAE;oBACd,UAAU,CAAC,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC;iBACtC;gBACD,OAAO,EAAE,GAAG,oBAAoB,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;aAClD;SACF;QAED,IAAI,IAAI,CAAC;QACT,IAAI;YACF,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,eAAe,EAAE,CAAC;YACtD,MAAM,GAAG,GAAG,WAAW,IAAI,CAAC,UAAU,EAAE,gBAAgB,SAAS,cAAc,IAAI,CAAC,SAAS,EAAE,eAAe,IAAI,CAAC,YAAY,EAAE,WAC/H,IAAI,CAAC,SACP,wBAAwB,CAAC;YACzB,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC;gBAC/B,GAAG;gBACH,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI;aACX,CAAC,CAAC;YACH,IAAI,GAAG,GAAG,CAAC,IAAyB,CAAC;SACtC;QAAC,OAAO,GAAG,EAAE;YACZ,MAAM,WAAW,GAAG,GAAU,CAAC;YAC/B,IACE,WAAW,CAAC,QAAQ;gBACpB,WAAW,CAAC,QAAQ,CAAC,IAAI;gBACzB,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC5B,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,EAClC;gBACA,MAAM,YAAY,GAAG,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;gBACxD,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC;gBAC/B,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC;gBACrC,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;gBACnC,OAAO;oBACL,KAAK,EAAE,0BAA0B,MAAM,UAAU,IAAI,iBAAiB,OAAO,EAAE;iBAChF,CAAC;aACH;YACD,OAAO;gBACL,KAAK,EAAE,mBAAmB,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE;aACzD,CAAC;SACH;QAED,gBAAM,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC7D,IAAI;YACF,MAAM,aAAa,GAAG,IAA6B,CAAC;YACpD,MAAM,KAAK,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;YACrC,IAAI,KAAK,EAAE;gBACT,OAAO;oBACL,KAAK,EAAE,SAAS,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE;iBAC/C,CAAC;aACH;YACD,MAAM,gBAAgB,GAAG,IAA4B,CAAC;YACtD,MAAM,MAAM,GAAG,gBAAgB;iBAC5B,GAAG,CAAC,CAAC,KAAyB,EAAE,EAAE;gBACjC,MAAM,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAClD,IAAI,MAAM,IAAI,IAAI,EAAE;oBAClB,OAAO,IAAI,CAAC,IAAI,CAAC;iBAClB;gBACD,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAC9B,CAAC,CAAC;iBACD,IAAI,CAAC,EAAE,CAAC,CAAC;YACZ,MAAM,QAAQ,GAAG,gBAAgB,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC/D,MAAM,UAAU,GAAG;gBACjB,KAAK,EAAE,QAAQ,CAAC,aAAa,EAAE,eAAe,IAAI,CAAC;gBACnD,MAAM,EAAE,QAAQ,CAAC,aAAa,EAAE,gBAAgB,IAAI,CAAC;gBACrD,UAAU,EAAE,QAAQ,CAAC,aAAa,EAAE,oBAAoB,IAAI,CAAC;aAC9D,CAAC;YACF,MAAM,QAAQ,GAAG;gBACf,MAAM,EAAE,KAAK;gBACb,MAAM;gBACN,UAAU;aACX,CAAC;YAEF,IAAI,IAAA,sBAAc,GAAE,EAAE;gBACpB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;aACrD;YAED,OAAO,QAAQ,CAAC;SACjB;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,8BAA8B,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;aAC5E,CAAC;SACH;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc;QAC/B,MAAM,SAAS,GAAG,IAAA,wBAAe,EAAC,MAAM,EAAE;YACxC;gBACE,QAAQ,EAAE;oBACR;wBACE,MAAM,EAAE,MAAM;wBACd,OAAO,EAAE,MAAM;qBAChB;iBACF;aACF;SACF,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG;YACX,SAAS;YACT,UAAU,EAAE;gBACV,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;gBAC5B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC9B,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;gBAC1C,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;gBACtC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;gBAC5C,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;aACvB;SACF,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAElE,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,gBAAgB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;QAExD,IAAI,cAAc,CAAC;QACnB,IAAI,IAAA,sBAAc,GAAE,EAAE;YACpB,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC3C,IAAI,cAAc,EAAE;gBAClB,gBAAM,CAAC,KAAK,CAAC,yCAAyC,MAAM,EAAE,CAAC,CAAC;gBAChE,MAAM,oBAAoB,GAAG,IAAI,CAAC,KAAK,CAAC,cAAwB,CAAC,CAAC;gBAClE,MAAM,UAAU,GAAG,oBAAoB,CAAC,UAAwB,CAAC;gBACjE,IAAI,UAAU,EAAE;oBACd,UAAU,CAAC,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC;iBACtC;gBACD,OAAO,EAAE,GAAG,oBAAoB,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;aAClD;SACF;QAED,IAAI,IAAsB,CAAC;QAC3B,IAAI;YACF,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,eAAe,EAAE,CAAC;YACtD,MAAM,GAAG,GAAG,WAAW,IAAI,CAAC,UAAU,EAAE,gBAAgB,SAAS,cAAc,IAAI,CAAC,SAAS,EAAE,eAAe,IAAI,CAAC,YAAY,EAAE,WAC/H,IAAI,CAAC,SACP,UAAU,CAAC;YACX,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAAmB;gBACjD,GAAG;gBACH,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI;aACX,CAAC,CAAC;YACH,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;SACjB;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE;aAChD,CAAC;SACH;QAED,gBAAM,CAAC,KAAK,CAAC,8BAA8B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnE,IAAI;YACF,IAAI,IAAI,CAAC,KAAK,EAAE;gBACd,OAAO;oBACL,KAAK,EAAE,SAAS,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE;iBACzD,CAAC;aACH;YACD,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAE3D,MAAM,QAAQ,GAAG;gBACf,MAAM;gBACN,MAAM,EAAE,KAAK;aACd,CAAC;YAEF,IAAI,IAAA,sBAAc,GAAE,EAAE;gBACpB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;aACrD;YAED,OAAO,QAAQ,CAAC;SACjB;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,uBAAuB,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;aACrE,CAAC;SACH;IACH,CAAC;;AArQH,gDAsQC;AArQC,+BAA+B;AAC/B,qGAAqG;AAC9F,8BAAW,GAAG;IACnB,YAAY;IACZ,gBAAgB;IAChB,gBAAgB;IAChB,gBAAgB;IAChB,oBAAoB;IACpB,oBAAoB;IACpB,gBAAgB;IAChB,oBAAoB;IACpB,oBAAoB;IACpB,oBAAoB;IACpB,wBAAwB;IACxB,wBAAwB;IACxB,YAAY;IACZ,cAAc;IACd,uBAAuB;IACvB,2BAA2B;IAC3B,gBAAgB;IAChB,oBAAoB;IACpB,oBAAoB;IACpB,mBAAmB;IACnB,uBAAuB;IACvB,6BAA6B;IAC7B,6BAA6B;IAC7B,+BAA+B;IAC/B,KAAK;CACN,CAAC"}
+{"version":3,"file":"vertex.js","sourceRoot":"","sources":["../../../src/providers/vertex.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,uDAA+B;AAC/B,qCAA2C;AAC3C,oCAAoD;AAEpD,6CAMsB;AAMtB,IAAI,UAAkC,CAAC;AACvC,KAAK,UAAU,eAAe;IAC5B,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,IAAI,UAAU,CAAC;QACf,IAAI,CAAC;YACH,MAAM,cAAc,GAAG,wDAAa,qBAAqB,GAAC,CAAC;YAC3D,UAAU,GAAG,cAAc,CAAC,UAAU,CAAC;QACzC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CACb,kHAAkH,CACnH,CAAC;QACJ,CAAC;QACD,UAAU,GAAG,IAAI,UAAU,CAAC;YAC1B,MAAM,EAAE,gDAAgD;SACzD,CAAC,CAAC;IACL,CAAC;IACD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,SAAS,EAAE,CAAC;IAC5C,MAAM,SAAS,GAAG,MAAM,UAAU,CAAC,YAAY,EAAE,CAAC;IAClD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;AAC/B,CAAC;AAwED,MAAM,qBAAqB;IAMzB,YACE,SAAiB,EACjB,UAAiF,EAAE;QAEnF,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC;QACpC,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;QAC3B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;IACpC,CAAC;IAED,EAAE;QACA,OAAO,UAAU,IAAI,CAAC,SAAS,EAAE,CAAC;IACpC,CAAC;IAED,QAAQ;QACN,OAAO,2BAA2B,IAAI,CAAC,SAAS,GAAG,CAAC;IACtD,CAAC;IAED,UAAU;QACR,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,OAAO;YACnB,IAAI,CAAC,GAAG,EAAE,eAAe;YACzB,OAAO,CAAC,GAAG,CAAC,eAAe;YAC3B,GAAG,IAAI,CAAC,SAAS,EAAE,4BAA4B,CAChD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,OAAO,CACL,CAAC,MAAM,eAAe,EAAE,CAAC,CAAC,SAAS;YACnC,IAAI,CAAC,MAAM,CAAC,SAAS;YACrB,IAAI,CAAC,GAAG,EAAE,iBAAiB;YAC3B,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAC9B,CAAC;IACJ,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,cAAc,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;IACtF,CAAC;IAED,SAAS;QACP,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,aAAa,IAAI,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,aAAa,CAC5F,CAAC;IACJ,CAAC;IAED,YAAY;QACV,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,SAAS;YACrB,IAAI,CAAC,GAAG,EAAE,gBAAgB;YAC1B,OAAO,CAAC,GAAG,CAAC,gBAAgB;YAC5B,QAAQ,CACT,CAAC;IACJ,CAAC;IAED,wDAAwD;IACxD,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF;AAED,MAAa,kBAAmB,SAAQ,qBAAqB;IA+B3D,YACE,SAAiB,EACjB,UAAiF,EAAE;QAEnF,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YACxD,gBAAM,CAAC,IAAI,CAAC,2CAA2C,SAAS,EAAE,CAAC,CAAC;QACtE,CAAC;QACD,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACtC,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QACpC,CAAC;QACD,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAc;QAChC,0FAA0F;QAC1F,IAAI,QAAQ,GAAG,IAAA,wBAAe,EAAC,MAAM,EAAE;YACrC,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE;gBACL,IAAI,EAAE,MAAM;aACb;SACF,CAAC,CAAC;QACH,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,OAAO,EAAE,GAAG,IAAA,sCAAyB,EAAC,QAAQ,CAAC,CAAC;QACnF,IAAI,OAAO,EAAE,CAAC;YACZ,gBAAM,CAAC,KAAK,CAAC,yCAAyC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAClF,QAAQ,GAAG,eAAe,CAAC;QAC7B,CAAC;QAED,iEAAiE;QACjE,MAAM,IAAI,GAAG;YACX,QAAQ;YACR,gBAAgB,EAAE;gBAChB,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;gBAC5B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC9B,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;gBACtC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;gBAC5C,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB;aAChC;YACD,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACrF,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzE,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,iBAAiB;gBAC/B,CAAC,CAAC,EAAE,iBAAiB,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE;gBACtD,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,2DAA2D,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEhG,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,iBAAiB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;QAEzD,IAAI,cAAc,CAAC;QACnB,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;YACrB,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC3C,IAAI,cAAc,EAAE,CAAC;gBACnB,gBAAM,CAAC,KAAK,CAAC,yCAAyC,MAAM,EAAE,CAAC,CAAC;gBAChE,MAAM,oBAAoB,GAAG,IAAI,CAAC,KAAK,CAAC,cAAwB,CAAC,CAAC;gBAClE,MAAM,UAAU,GAAG,oBAAoB,CAAC,UAAwB,CAAC;gBACjE,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC;gBACvC,CAAC;gBACD,OAAO,EAAE,GAAG,oBAAoB,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;YACnD,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC;QACT,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,eAAe,EAAE,CAAC;YACtD,MAAM,GAAG,GAAG,WAAW,IAAI,CAAC,UAAU,EAAE,gBAAgB,SAAS,cAAc,IAAI,CAAC,SAAS,EAAE,eAAe,IAAI,CAAC,YAAY,EAAE,WAC/H,IAAI,CAAC,SACP,wBAAwB,CAAC;YACzB,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC;gBAC/B,GAAG;gBACH,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI;aACX,CAAC,CAAC;YACH,IAAI,GAAG,GAAG,CAAC,IAAyB,CAAC;QACvC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,WAAW,GAAG,GAAU,CAAC;YAC/B,IACE,WAAW,CAAC,QAAQ;gBACpB,WAAW,CAAC,QAAQ,CAAC,IAAI;gBACzB,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC5B,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,EAClC,CAAC;gBACD,MAAM,YAAY,GAAG,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;gBACxD,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC;gBAC/B,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC;gBACrC,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;gBACnC,OAAO;oBACL,KAAK,EAAE,0BAA0B,MAAM,UAAU,IAAI,iBAAiB,OAAO,EAAE;iBAChF,CAAC;YACJ,CAAC;YACD,OAAO;gBACL,KAAK,EAAE,mBAAmB,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE;aACzD,CAAC;QACJ,CAAC;QAED,gBAAM,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC7D,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,IAA6B,CAAC;YACpD,MAAM,KAAK,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;YACrC,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO;oBACL,KAAK,EAAE,SAAS,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE;iBAC/C,CAAC;YACJ,CAAC;YACD,MAAM,gBAAgB,GAAG,IAA4B,CAAC;YACtD,MAAM,MAAM,GAAG,gBAAgB;iBAC5B,GAAG,CAAC,CAAC,KAAyB,EAAE,EAAE;gBACjC,MAAM,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAClD,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;oBACnB,OAAO,IAAI,CAAC,IAAI,CAAC;gBACnB,CAAC;gBACD,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAC9B,CAAC,CAAC;iBACD,IAAI,CAAC,EAAE,CAAC,CAAC;YACZ,MAAM,QAAQ,GAAG,gBAAgB,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC/D,MAAM,UAAU,GAAG;gBACjB,KAAK,EAAE,QAAQ,CAAC,aAAa,EAAE,eAAe,IAAI,CAAC;gBACnD,MAAM,EAAE,QAAQ,CAAC,aAAa,EAAE,gBAAgB,IAAI,CAAC;gBACrD,UAAU,EAAE,QAAQ,CAAC,aAAa,EAAE,oBAAoB,IAAI,CAAC;aAC9D,CAAC;YACF,MAAM,QAAQ,GAAG;gBACf,MAAM,EAAE,KAAK;gBACb,MAAM;gBACN,UAAU;aACX,CAAC;YAEF,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;gBACrB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;YACtD,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,8BAA8B,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;aAC5E,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc;QAC/B,MAAM,SAAS,GAAG,IAAA,wBAAe,EAAC,MAAM,EAAE;YACxC;gBACE,QAAQ,EAAE;oBACR;wBACE,MAAM,EAAE,MAAM;wBACd,OAAO,EAAE,MAAM;qBAChB;iBACF;aACF;SACF,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG;YACX,SAAS;YACT,UAAU,EAAE;gBACV,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;gBAC5B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC9B,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;gBAC1C,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;gBACtC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;gBAC5C,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;aACvB;SACF,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAElE,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,gBAAgB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;QAExD,IAAI,cAAc,CAAC;QACnB,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;YACrB,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC3C,IAAI,cAAc,EAAE,CAAC;gBACnB,gBAAM,CAAC,KAAK,CAAC,yCAAyC,MAAM,EAAE,CAAC,CAAC;gBAChE,MAAM,oBAAoB,GAAG,IAAI,CAAC,KAAK,CAAC,cAAwB,CAAC,CAAC;gBAClE,MAAM,UAAU,GAAG,oBAAoB,CAAC,UAAwB,CAAC;gBACjE,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC;gBACvC,CAAC;gBACD,OAAO,EAAE,GAAG,oBAAoB,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;YACnD,CAAC;QACH,CAAC;QAED,IAAI,IAAsB,CAAC;QAC3B,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,eAAe,EAAE,CAAC;YACtD,MAAM,GAAG,GAAG,WAAW,IAAI,CAAC,UAAU,EAAE,gBAAgB,SAAS,cAAc,IAAI,CAAC,SAAS,EAAE,eAAe,IAAI,CAAC,YAAY,EAAE,WAC/H,IAAI,CAAC,SACP,UAAU,CAAC;YACX,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAAmB;gBACjD,GAAG;gBACH,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI;aACX,CAAC,CAAC;YACH,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,mBAAmB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE;aAChD,CAAC;QACJ,CAAC;QAED,gBAAM,CAAC,KAAK,CAAC,8BAA8B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnE,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,KAAK,EAAE,SAAS,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE;iBACzD,CAAC;YACJ,CAAC;YACD,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAE3D,MAAM,QAAQ,GAAG;gBACf,MAAM;gBACN,MAAM,EAAE,KAAK;aACd,CAAC;YAEF,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;gBACrB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;YACtD,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,uBAAuB,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;aACrE,CAAC;QACJ,CAAC;IACH,CAAC;;AAzQH,gDA0QC;AAzQC,+BAA+B;AAC/B,qGAAqG;AAC9F,8BAAW,GAAG;IACnB,YAAY;IACZ,gBAAgB;IAChB,gBAAgB;IAChB,gBAAgB;IAChB,oBAAoB;IACpB,oBAAoB;IACpB,gBAAgB;IAChB,oBAAoB;IACpB,oBAAoB;IACpB,oBAAoB;IACpB,wBAAwB;IACxB,wBAAwB;IACxB,YAAY;IACZ,cAAc;IACd,uBAAuB;IACvB,2BAA2B;IAC3B,gBAAgB;IAChB,oBAAoB;IACpB,oBAAoB;IACpB,mBAAmB;IACnB,uBAAuB;IACvB,6BAA6B;IAC7B,6BAA6B;IAC7B,+BAA+B;IAC/B,KAAK;CACN,CAAC"}

package/dist/src/providers/vertexUtil.js.map

@@ -1 +1 @@
-{"version":3,"file":"vertexUtil.js","sourceRoot":"","sources":["../../../src/providers/vertexUtil.ts"],"names":[],"mappings":";;;AAwEA,SAAgB,yBAAyB,CAAC,QAAa;IACrD,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,OAAO,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,KAAK,QAAQ,EAAE;QACtE,4EAA4E;QAC5E,QAAQ,GAAG;YACT,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;aACpD;SACF,CAAC;QACF,OAAO,GAAG,IAAI,CAAC;KAChB;IACD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;AAC/B,CAAC;AAbD,8DAaC"}
+{"version":3,"file":"vertexUtil.js","sourceRoot":"","sources":["../../../src/providers/vertexUtil.ts"],"names":[],"mappings":";;;AAwEA,SAAgB,yBAAyB,CAAC,QAAa;IACrD,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,OAAO,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;QACvE,4EAA4E;QAC5E,QAAQ,GAAG;YACT,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;aACpD;SACF,CAAC;QACF,OAAO,GAAG,IAAI,CAAC;IACjB,CAAC;IACD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;AAC/B,CAAC;AAbD,8DAaC"}

package/dist/src/providers/webhook.js.map

@@ -1 +1 @@
-{"version":3,"file":"webhook.js","sourceRoot":"","sources":["../../../src/providers/webhook.ts"],"names":[],"mappings":";;;;;;AAAA,uDAA+B;AAC/B,oCAA0C;AAE1C,qCAA8C;AAI9C,MAAa,eAAe;IAI1B,YAAY,UAAkB,EAAE,UAA4C,EAAE;QAC5E,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC/B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;QAClC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,EAAE;QACA,OAAO,WAAW,IAAI,CAAC,UAAU,EAAE,CAAC;IACtC,CAAC;IAED,QAAQ;QACN,OAAO,qBAAqB,IAAI,CAAC,UAAU,GAAG,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,MAAM,GAAwC;YAClD,MAAM;SACP,CAAC;QACF,IAAI,IAAI,CAAC,MAAM,EAAE;YACf,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;SAC7B;QAED,gBAAM,CAAC,KAAK,CAAC,oBAAoB,IAAI,CAAC,UAAU,iBAAiB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC3F,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,QAAQ,GAAG,MAAM,IAAA,sBAAc,EAC7B,IAAI,CAAC,UAAU,EACf;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC7B,EACD,2BAAkB,EAClB,MAAM,CACP,CAAC;SACH;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,uBAAuB,MAAM,CAAC,GAAG,CAAC,EAAE;aAC5C,CAAC;SACH;QACD,gBAAM,CAAC,KAAK,CAAC,uBAAuB,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAErE,IAAI,QAAQ,CAAC,IAAI,IAAI,OAAO,QAAQ,CAAC,IAAI,CAAC,MAAM,KAAK,QAAQ,EAAE;YAC7D,OAAO;gBACL,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM;aAC7B,CAAC;SACH;aAAM;YACL,OAAO;gBACL,KAAK,EAAE,uDAAuD,IAAI,CAAC,SAAS,CAC1E,QAAQ,CAAC,IAAI,CACd,EAAE;aACJ,CAAC;SACH;IACH,CAAC;CACF;AA7DD,0CA6DC"}
+{"version":3,"file":"webhook.js","sourceRoot":"","sources":["../../../src/providers/webhook.ts"],"names":[],"mappings":";;;;;;AAAA,uDAA+B;AAC/B,oCAA0C;AAE1C,qCAA8C;AAI9C,MAAa,eAAe;IAI1B,YAAY,UAAkB,EAAE,UAA4C,EAAE;QAC5E,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC/B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;QAClC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,EAAE;QACA,OAAO,WAAW,IAAI,CAAC,UAAU,EAAE,CAAC;IACtC,CAAC;IAED,QAAQ;QACN,OAAO,qBAAqB,IAAI,CAAC,UAAU,GAAG,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,MAAM,GAAwC;YAClD,MAAM;SACP,CAAC;QACF,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC9B,CAAC;QAED,gBAAM,CAAC,KAAK,CAAC,oBAAoB,IAAI,CAAC,UAAU,iBAAiB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC3F,IAAI,QAAQ,CAAC;QACb,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,IAAA,sBAAc,EAC7B,IAAI,CAAC,UAAU,EACf;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC7B,EACD,2BAAkB,EAClB,MAAM,CACP,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,uBAAuB,MAAM,CAAC,GAAG,CAAC,EAAE;aAC5C,CAAC;QACJ,CAAC;QACD,gBAAM,CAAC,KAAK,CAAC,uBAAuB,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAErE,IAAI,QAAQ,CAAC,IAAI,IAAI,OAAO,QAAQ,CAAC,IAAI,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC9D,OAAO;gBACL,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM;aAC7B,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,OAAO;gBACL,KAAK,EAAE,uDAAuD,IAAI,CAAC,SAAS,CAC1E,QAAQ,CAAC,IAAI,CACd,EAAE;aACJ,CAAC;QACJ,CAAC;IACH,CAAC;CACF;AA7DD,0CA6DC"}

package/dist/src/providers.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"providers.d.ts","sourceRoot":"","sources":["../../src/providers.ts"],"names":[],"mappings":"AAOA,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,4BAA4B,EAG7B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,2BAA2B,EAA6B,MAAM,uBAAuB,CAAC;AAC/F,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EACL,yBAAyB,EACzB,mBAAmB,EAEpB,MAAM,qBAAqB,CAAC;AA4B7B,OAAO,EAAE,eAAe,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAGxE,OAAO,KAAK,EACV,WAAW,EACX,YAAY,EACZ,eAAe,EAEf,eAAe,EAChB,MAAM,SAAS,CAAC;AAEjB,wBAAsB,gBAAgB,CACpC,aAAa,EAAE,eAAe,CAAC,WAAW,CAAC,EAC3C,OAAO,GAAE;IACP,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,YAAY,CAAC;CACf,GACL,OAAO,CAAC,WAAW,EAAE,CAAC,CAuCxB;AAGD,wBAAsB,eAAe,CACnC,YAAY,EAAE,MAAM,EACpB,OAAO,GAAE;IACP,OAAO,CAAC,EAAE,eAAe,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,YAAY,CAAC;CACf,GACL,OAAO,CAAC,WAAW,CAAC,CAmOtB;;;;;;;;;;;;;AAED,wBAWE"}
+{"version":3,"file":"providers.d.ts","sourceRoot":"","sources":["../../src/providers.ts"],"names":[],"mappings":"AAOA,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,4BAA4B,EAI7B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,2BAA2B,EAA6B,MAAM,uBAAuB,CAAC;AAC/F,OAAO,EAA+B,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AACvF,OAAO,EACL,yBAAyB,EACzB,mBAAmB,EAEpB,MAAM,qBAAqB,CAAC;AA6B7B,OAAO,EAAE,eAAe,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAKxE,OAAO,KAAK,EACV,WAAW,EACX,YAAY,EACZ,eAAe,EAEf,eAAe,EAChB,MAAM,SAAS,CAAC;AAEjB,wBAAsB,gBAAgB,CACpC,aAAa,EAAE,eAAe,CAAC,WAAW,CAAC,EAC3C,OAAO,GAAE;IACP,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,YAAY,CAAC;CACf,GACL,OAAO,CAAC,WAAW,EAAE,CAAC,CAuCxB;AAGD,wBAAsB,eAAe,CACnC,YAAY,EAAE,MAAM,EACpB,OAAO,GAAE;IACP,OAAO,CAAC,EAAE,eAAe,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,YAAY,CAAC;CACf,GACL,OAAO,CAAC,WAAW,CAAC,CAgRtB;;;;;;;;;;;;;AAED,wBAWE"}

package/dist/src/providers.js

@@ -48,7 +48,10 @@ const huggingface_1 = require("./providers/huggingface");
 const bedrock_1 = require("./providers/bedrock");
 const pythonCompletion_1 = require("./providers/pythonCompletion");
 const cohere_1 = require("./providers/cohere");
+const CloudflareAiProviders = __importStar(require("./providers/cloudflare-ai"));
 const bam_1 = require("./providers/bam");
+const portkey_1 = require("./providers/portkey");
+const http_1 = require("./providers/http");
 const esm_1 = require("./esm");
 async function loadApiProviders(providerPaths, options = {}) {
     const { basePath, env } = options;
@@ -70,7 +73,7 @@ async function loadApiProviders(providerPaths, options = {}) {
             }
             else if (typeof provider === 'function') {
                 return {
-                    id: () => `custom-function-${idx}`,
+                    id: provider.label ? () => provider.label : () => `custom-function-${idx}`,
                     callApi: provider,
                 };
             }
@@ -107,7 +110,8 @@ async function loadApiProvider(providerPath, context = {}) {
         env,
     };
     let ret;
-    if (providerPath.startsWith('file://') && (providerPath.endsWith('.yaml') || providerPath.endsWith('.yml'))) {
+    if (providerPath.startsWith('file://') &&
+        (providerPath.endsWith('.yaml') || providerPath.endsWith('.yml'))) {
         const filePath = providerPath.slice('file://'.length);
         const yamlContent = js_yaml_1.default.load(fs_1.default.readFileSync(filePath, 'utf8'));
         (0, tiny_invariant_1.default)(yamlContent, `Provider config ${filePath} is undefined`);
@@ -139,10 +143,13 @@ async function loadApiProvider(providerPath, context = {}) {
             ret = new openai_1.OpenAiChatCompletionProvider(modelName || 'gpt-3.5-turbo', providerOptions);
         }
         else if (modelType === 'embedding' || modelType === 'embeddings') {
-            ret = new openai_1.OpenAiEmbeddingProvider(modelName || 'text-embedding-ada-002', providerOptions);
+            ret = new openai_1.OpenAiEmbeddingProvider(modelName || 'text-embedding-3-large', providerOptions);
         }
         else if (modelType === 'completion') {
-            ret = new openai_1.OpenAiCompletionProvider(modelName || 'text-davinci-003', providerOptions);
+            ret = new openai_1.OpenAiCompletionProvider(modelName || 'gpt-3.5-turbo-instruct', providerOptions);
+        }
+        else if (modelType === 'moderation') {
+            ret = new openai_1.OpenAiModerationProvider(modelName || 'text-moderation-latest', providerOptions);
         }
         else if (openai_1.OpenAiChatCompletionProvider.OPENAI_CHAT_MODEL_NAMES.includes(modelType)) {
             ret = new openai_1.OpenAiChatCompletionProvider(modelType, providerOptions);
@@ -193,6 +200,11 @@ async function loadApiProvider(providerPath, context = {}) {
             },
         });
     }
+    else if (providerPath.startsWith('portkey:')) {
+        const splits = providerPath.split(':');
+        const modelName = splits.slice(1).join(':');
+        ret = new portkey_1.PortkeyChatCompletionProvider(modelName, providerOptions);
+    }
     else if (providerPath.startsWith('anthropic:')) {
         const splits = providerPath.split(':');
         const modelType = splits[1];
@@ -252,8 +264,15 @@ async function loadApiProvider(providerPath, context = {}) {
     }
     else if (providerPath.startsWith('replicate:')) {
         const splits = providerPath.split(':');
-        const modelName = splits.slice(1).join(':');
-        ret = new replicate_1.ReplicateProvider(modelName, providerOptions);
+        const modelType = splits[1];
+        const modelName = splits.slice(2).join(':');
+        if (modelType === 'moderation') {
+            ret = new replicate_1.ReplicateModerationProvider(modelName, providerOptions);
+        }
+        else {
+            // By default, there is no model type.
+            ret = new replicate_1.ReplicateProvider(modelType + ':' + modelName, providerOptions);
+        }
     }
     else if (providerPath.startsWith('bam:')) {
         const splits = providerPath.split(':');
@@ -266,6 +285,24 @@ async function loadApiProvider(providerPath, context = {}) {
             throw new Error(`Invalid BAM provider: ${providerPath}. Use one of the following providers: bam:chat:<model name>`);
         }
     }
+    else if (providerPath.startsWith('cloudflare-ai:')) {
+        // Load Cloudflare AI
+        const splits = providerPath.split(':');
+        const modelType = splits[1];
+        const deploymentName = splits[2];
+        if (modelType === 'chat') {
+            ret = new CloudflareAiProviders.CloudflareAiChatCompletionProvider(deploymentName, providerOptions);
+        }
+        else if (modelType === 'embedding' || modelType === 'embeddings') {
+            ret = new CloudflareAiProviders.CloudflareAiEmbeddingProvider(deploymentName, providerOptions);
+        }
+        else if (modelType === 'completion') {
+            ret = new CloudflareAiProviders.CloudflareAiCompletionProvider(deploymentName, providerOptions);
+        }
+        else {
+            throw new Error(`Unknown Cloudflare AI model type: ${modelType}. Use one of the following providers: cloudflare-ai:chat:<model name>, cloudflare-ai:completion:<model name>, cloudflare-ai:embedding:`);
+        }
+    }
     else if (providerPath.startsWith('webhook:')) {
         const webhookUrl = providerPath.substring('webhook:'.length);
         ret = new webhook_1.WebhookProvider(webhookUrl, providerOptions);
@@ -329,8 +366,12 @@ async function loadApiProvider(providerPath, context = {}) {
             ret = new localai_1.LocalAiChatProvider(modelType, providerOptions);
         }
     }
+    else if (providerPath.startsWith('http:') || providerPath.startsWith('https:')) {
+        ret = new http_1.HttpProvider(providerPath, providerOptions);
+    }
     else if (providerPath === 'promptfoo:redteam:iterative') {
-        const RedteamIterativeProvider = (await Promise.resolve(`${path_1.default.join(__dirname, './redteam/iterative')}`).then(s => __importStar(require(s)))).default;
+        const RedteamIterativeProvider = (await Promise.resolve(`${path_1.default.join(__dirname, './redteam/iterative')}`).then(s => __importStar(require(s))))
+            .default;
         ret = new RedteamIterativeProvider(providerOptions);
     }
     else {
@@ -341,6 +382,7 @@ async function loadApiProvider(providerPath, context = {}) {
         const modulePath = path_1.default.join(basePath || process.cwd(), providerPath);
         const CustomApiProvider = await (0, esm_1.importModule)(modulePath);
         ret = new CustomApiProvider(options);
+        ret.label = ret.label || options.label;
     }
     ret.transform = options.transform;
     ret.delay = options.delay;