promptfoo 0.59.1 → 0.61.0

package/dist/src/providers/http.js

@@ -0,0 +1,64 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HttpProvider = void 0;
+const tiny_invariant_1 = __importDefault(require("tiny-invariant"));
+const logger_1 = __importDefault(require("../logger"));
+const cache_1 = require("../cache");
+const shared_1 = require("./shared");
+const util_1 = require("../util");
+class HttpProvider {
+    constructor(url, options) {
+        this.url = url;
+        this.config = options.config;
+        this.responseParser = createResponseParser(this.config.responseParser);
+    }
+    id() {
+        return this.url;
+    }
+    toString() {
+        return `[HTTP Provider ${this.url}]`;
+    }
+    async callApi(prompt, context) {
+        // Render all nested strings
+        const nunjucks = (0, util_1.getNunjucksEngine)();
+        const stringifiedConfig = (0, util_1.safeJsonStringify)(this.config);
+        const renderedConfig = JSON.parse(nunjucks.renderString(stringifiedConfig, {
+            prompt,
+            ...context?.vars,
+        }));
+        const method = renderedConfig.method || 'POST';
+        const headers = renderedConfig.headers || { 'Content-Type': 'application/json' };
+        (0, tiny_invariant_1.default)(typeof method === 'string', 'Expected method to be a string');
+        (0, tiny_invariant_1.default)(typeof headers === 'object', 'Expected headers to be an object');
+        logger_1.default.debug(`Calling HTTP provider: ${this.url} with config: ${stringifiedConfig}`);
+        let response;
+        try {
+            response = await (0, cache_1.fetchWithCache)(this.url, {
+                method,
+                headers,
+                body: JSON.stringify(renderedConfig.body),
+            }, shared_1.REQUEST_TIMEOUT_MS, 'json');
+        }
+        catch (err) {
+            return {
+                error: `HTTP call error: ${String(err)}`,
+            };
+        }
+        logger_1.default.debug(`\tHTTP response: ${JSON.stringify(response.data)}`);
+        return this.responseParser(response.data);
+    }
+}
+exports.HttpProvider = HttpProvider;
+function createResponseParser(parser) {
+    if (typeof parser === 'function') {
+        return parser;
+    }
+    if (typeof parser === 'string') {
+        return new Function('json', `return ${parser}`);
+    }
+    return (data) => ({ output: data });
+}
+//# sourceMappingURL=http.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"http.js","sourceRoot":"","sources":["../../../src/providers/http.ts"],"names":[],"mappings":";;;;;;AAAA,oEAAuC;AAEvC,uDAA+B;AAC/B,oCAA0C;AAC1C,qCAA8C;AAC9C,kCAA+D;AAS/D,MAAa,YAAY;IAKvB,YAAY,GAAW,EAAE,OAAwB;QAC/C,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,cAAc,GAAG,oBAAoB,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;IACzE,CAAC;IAED,EAAE;QACA,OAAO,IAAI,CAAC,GAAG,CAAC;IAClB,CAAC;IAED,QAAQ;QACN,OAAO,kBAAkB,IAAI,CAAC,GAAG,GAAG,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,OAA8B;QAC1D,4BAA4B;QAC5B,MAAM,QAAQ,GAAG,IAAA,wBAAiB,GAAE,CAAC;QACrC,MAAM,iBAAiB,GAAG,IAAA,wBAAiB,EAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzD,MAAM,cAAc,GAClB,IAAI,CAAC,KAAK,CACR,QAAQ,CAAC,YAAY,CAAC,iBAAiB,EAAE;YACvC,MAAM;YACN,GAAG,OAAO,EAAE,IAAI;SACjB,CAAC,CACH,CAAC;QAEJ,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,IAAI,MAAM,CAAC;QAC/C,MAAM,OAAO,GAAG,cAAc,CAAC,OAAO,IAAI,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC;QACjF,IAAA,wBAAS,EAAC,OAAO,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QACxE,IAAA,wBAAS,EAAC,OAAO,OAAO,KAAK,QAAQ,EAAE,kCAAkC,CAAC,CAAC;QAE3E,gBAAM,CAAC,KAAK,CAAC,0BAA0B,IAAI,CAAC,GAAG,iBAAiB,iBAAiB,EAAE,CAAC,CAAC;QACrF,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,QAAQ,GAAG,MAAM,IAAA,sBAAc,EAC7B,IAAI,CAAC,GAAG,EACR;gBACE,MAAM;gBACN,OAAO;gBACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC;aAC1C,EACD,2BAAkB,EAClB,MAAM,CACP,CAAC;SACH;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,oBAAoB,MAAM,CAAC,GAAG,CAAC,EAAE;aACzC,CAAC;SACH;QACD,gBAAM,CAAC,KAAK,CAAC,oBAAoB,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAElE,OAAO,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC5C,CAAC;CACF;AA1DD,oCA0DC;AAED,SAAS,oBAAoB,CAAC,MAAW;IACvC,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE;QAChC,OAAO,MAAM,CAAC;KACf;IACD,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE;QAC9B,OAAO,IAAI,QAAQ,CAAC,MAAM,EAAE,UAAU,MAAM,EAAE,CAAoC,CAAC;KACpF;IACD,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;AACtC,CAAC"}

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

@@ -1,7 +1,9 @@
 import { ApiProvider, ApiSimilarityProvider, ProviderClassificationResponse, ProviderEmbeddingResponse, ProviderResponse, ProviderSimilarityResponse } from '../types';
-interface HuggingfaceTextGenerationOptions {
+interface HuggingfaceProviderOptions {
     apiKey?: string;
     apiEndpoint?: string;
+}
+type HuggingfaceTextGenerationOptions = HuggingfaceProviderOptions & {
     top_k?: number;
     top_p?: number;
     temperature?: number;
@@ -13,7 +15,7 @@ interface HuggingfaceTextGenerationOptions {
     do_sample?: boolean;
     use_cache?: boolean;
     wait_for_model?: boolean;
-}
+};
 export declare class HuggingfaceTextGenerationProvider implements ApiProvider {
     modelName: string;
     config: HuggingfaceTextGenerationOptions;
@@ -24,13 +26,10 @@ export declare class HuggingfaceTextGenerationProvider implements ApiProvider {
     id(): string;
     toString(): string;
     getApiKey(): string | undefined;
-    getConfig(): Partial<Record<keyof HuggingfaceTextGenerationOptions, string | number | boolean | undefined>>;
+    getConfig(): Partial<Record<"temperature" | "max_new_tokens" | "top_p" | "top_k" | "repetition_penalty" | keyof HuggingfaceProviderOptions | "max_time" | "return_full_text" | "num_return_sequences" | "do_sample" | "use_cache" | "wait_for_model", string | number | boolean | undefined>>;
     callApi(prompt: string): Promise<ProviderResponse>;
 }
-interface HuggingfaceTextClassificationOptions {
-    apiKey?: string;
-    apiEndpoint?: string;
-}
+type HuggingfaceTextClassificationOptions = HuggingfaceProviderOptions;
 export declare class HuggingfaceTextClassificationProvider implements ApiProvider {
     modelName: string;
     config: HuggingfaceTextClassificationOptions;
@@ -40,15 +39,14 @@ export declare class HuggingfaceTextClassificationProvider implements ApiProvide
     });
     id(): string;
     toString(): string;
+    getApiKey(): string | undefined;
     callClassificationApi(prompt: string): Promise<ProviderClassificationResponse>;
     callApi(prompt: string): Promise<ProviderResponse>;
 }
-interface HuggingfaceFeatureExtractionOptions {
-    apiKey?: string;
-    apiEndpoint?: string;
+type HuggingfaceFeatureExtractionOptions = HuggingfaceProviderOptions & {
     use_cache?: boolean;
     wait_for_model?: boolean;
-}
+};
 export declare class HuggingfaceFeatureExtractionProvider implements ApiProvider {
     modelName: string;
     config: HuggingfaceFeatureExtractionOptions;
@@ -58,13 +56,14 @@ export declare class HuggingfaceFeatureExtractionProvider implements ApiProvider
     });
     id(): string;
     toString(): string;
+    getApiKey(): string | undefined;
     callApi(): Promise<ProviderResponse>;
     callEmbeddingApi(text: string): Promise<ProviderEmbeddingResponse>;
 }
-interface HuggingfaceSentenceSimilarityOptions {
+type HuggingfaceSentenceSimilarityOptions = HuggingfaceProviderOptions & {
     use_cache?: boolean;
     wait_for_model?: boolean;
-}
+};
 export declare class HuggingfaceSentenceSimilarityProvider implements ApiSimilarityProvider {
     modelName: string;
     config: HuggingfaceSentenceSimilarityOptions;
@@ -73,17 +72,16 @@ export declare class HuggingfaceSentenceSimilarityProvider implements ApiSimilar
         config?: HuggingfaceSentenceSimilarityOptions;
     });
     id(): string;
+    getApiKey(): string | undefined;
     toString(): string;
     callApi(): Promise<ProviderResponse>;
     callSimilarityApi(expected: string, input: string): Promise<ProviderSimilarityResponse>;
 }
-interface HuggingfaceTokenClassificationOptions {
-    apiKey?: string;
-    apiEndpoint?: string;
+type HuggingfaceTokenClassificationOptions = HuggingfaceProviderOptions & {
     aggregation_strategy?: string;
     use_cache?: boolean;
     wait_for_model?: boolean;
-}
+};
 export declare class HuggingfaceTokenExtractionProvider implements ApiProvider {
     modelName: string;
     config: HuggingfaceTokenClassificationOptions;

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

@@ -1 +1 @@
-{"version":3,"file":"huggingface.d.ts","sourceRoot":"","sources":["../../../src/providers/huggingface.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,WAAW,EACX,qBAAqB,EACrB,8BAA8B,EAC9B,yBAAyB,EACzB,gBAAgB,EAChB,0BAA0B,EAC3B,MAAM,UAAU,CAAC;AAGlB,UAAU,gCAAgC;IACxC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,OAAO,CAAC;CAC1B;AAgBD,qBAAa,iCAAkC,YAAW,WAAW;IACnE,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,gCAAgC,CAAC;gBAGvC,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,gCAAgC,CAAA;KAAO;IAQ1E,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIlB,SAAS,IAAI,MAAM,GAAG,SAAS;IAI/B,SAAS;IAUH,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAoDzD;AAED,UAAU,oCAAoC;IAC5C,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,qBAAa,qCAAsC,YAAW,WAAW;IACvE,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,oCAAoC,CAAC;gBAG3C,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,oCAAoC,CAAA;KAAO;IAQ9E,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIZ,qBAAqB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,8BAA8B,CAAC;IAoD9E,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAOzD;AAED,UAAU,mCAAmC;IAC3C,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,OAAO,CAAC;CAC1B;AAED,qBAAa,oCAAqC,YAAW,WAAW;IACtE,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,mCAAmC,CAAC;gBAG1C,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,mCAAmC,CAAA;KAAO;IAQ7E,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIZ,OAAO,IAAI,OAAO,CAAC,gBAAgB,CAAC;IAIpC,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,yBAAyB,CAAC;CAmDzE;AAED,UAAU,oCAAoC;IAC5C,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,OAAO,CAAC;CAC1B;AAED,qBAAa,qCAAsC,YAAW,qBAAqB;IACjF,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,oCAAoC,CAAC;gBAG3C,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,oCAAoC,CAAA;KAAO;IAQ9E,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIZ,OAAO,IAAI,OAAO,CAAC,gBAAgB,CAAC;IAIpC,iBAAiB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,0BAA0B,CAAC;CAkD9F;AAED,UAAU,qCAAqC;IAC7C,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,OAAO,CAAC;CAC1B;AAED,qBAAa,kCAAmC,YAAW,WAAW;IACpE,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,qCAAqC,CAAC;gBAG5C,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,qCAAqC,CAAA;KAAO;IAQ/E,EAAE,IAAI,MAAM;IAIN,qBAAqB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,8BAA8B,CAAC;IA0D7E,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAOzD"}
+{"version":3,"file":"huggingface.d.ts","sourceRoot":"","sources":["../../../src/providers/huggingface.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,WAAW,EACX,qBAAqB,EACrB,8BAA8B,EAC9B,yBAAyB,EACzB,gBAAgB,EAChB,0BAA0B,EAC3B,MAAM,UAAU,CAAC;AAGlB,UAAU,0BAA0B;IAClC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,KAAK,gCAAgC,GAAG,0BAA0B,GAAG;IACnE,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,OAAO,CAAC;CAC1B,CAAC;AAgBF,qBAAa,iCAAkC,YAAW,WAAW;IACnE,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,gCAAgC,CAAC;gBAGvC,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,gCAAgC,CAAA;KAAO;IAQ1E,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIlB,SAAS,IAAI,MAAM,GAAG,SAAS;IAI/B,SAAS;IAUH,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAoDzD;AAED,KAAK,oCAAoC,GAAG,0BAA0B,CAAC;AAEvE,qBAAa,qCAAsC,YAAW,WAAW;IACvE,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,oCAAoC,CAAC;gBAG3C,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,oCAAoC,CAAA;KAAO;IAQ9E,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIlB,SAAS,IAAI,MAAM,GAAG,SAAS;IAIzB,qBAAqB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,8BAA8B,CAAC;IAkD9E,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAOzD;AAED,KAAK,mCAAmC,GAAG,0BAA0B,GAAG;IACtE,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,OAAO,CAAC;CAC1B,CAAC;AAEF,qBAAa,oCAAqC,YAAW,WAAW;IACtE,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,mCAAmC,CAAC;gBAG1C,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,mCAAmC,CAAA;KAAO;IAQ7E,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIlB,SAAS,IAAI,MAAM,GAAG,SAAS;IAIzB,OAAO,IAAI,OAAO,CAAC,gBAAgB,CAAC;IAIpC,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,yBAAyB,CAAC;CAiDzE;AAED,KAAK,oCAAoC,GAAG,0BAA0B,GAAG;IACvE,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,OAAO,CAAC;CAC1B,CAAC;AAEF,qBAAa,qCAAsC,YAAW,qBAAqB;IACjF,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,oCAAoC,CAAC;gBAG3C,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,oCAAoC,CAAA;KAAO;IAQ9E,EAAE,IAAI,MAAM;IAIZ,SAAS,IAAI,MAAM,GAAG,SAAS;IAI/B,QAAQ,IAAI,MAAM;IAIZ,OAAO,IAAI,OAAO,CAAC,gBAAgB,CAAC;IAIpC,iBAAiB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,0BAA0B,CAAC;CAmD9F;AAED,KAAK,qCAAqC,GAAG,0BAA0B,GAAG;IACxE,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,OAAO,CAAC;CAC1B,CAAC;AAEF,qBAAa,kCAAmC,YAAW,WAAW;IACpE,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,qCAAqC,CAAC;gBAG5C,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,qCAAqC,CAAA;KAAO;IAQ/E,EAAE,IAAI,MAAM;IAIN,qBAAqB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,8BAA8B,CAAC;IA0D7E,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAOzD"}

package/dist/src/providers/huggingface.js

@@ -104,6 +104,9 @@ class HuggingfaceTextClassificationProvider {
     toString() {
         return `[Huggingface Text Classification Provider ${this.modelName}]`;
     }
+    getApiKey() {
+        return this.config.apiKey || process.env.HF_API_TOKEN;
+    }
     async callClassificationApi(prompt) {
         const params = {
             inputs: prompt,
@@ -118,9 +121,7 @@ class HuggingfaceTextClassificationProvider {
                 method: 'POST',
                 headers: {
                     'Content-Type': 'application/json',
-                    ...(this.config.apiKey || process.env.HF_API_TOKEN
-                        ? { Authorization: `Bearer ${process.env.HF_API_TOKEN}` }
-                        : {}),
+                    ...(this.getApiKey() ? { Authorization: `Bearer ${this.getApiKey()}` } : {}),
                 },
                 body: JSON.stringify(params),
             }, shared_1.REQUEST_TIMEOUT_MS);
@@ -170,6 +171,9 @@ class HuggingfaceFeatureExtractionProvider {
     toString() {
         return `[Huggingface Feature Extraction Provider ${this.modelName}]`;
     }
+    getApiKey() {
+        return this.config.apiKey || process.env.HF_API_TOKEN;
+    }
     async callApi() {
         throw new Error('Cannot use a feature extraction provider for text generation');
     }
@@ -192,9 +196,7 @@ class HuggingfaceFeatureExtractionProvider {
                 method: 'POST',
                 headers: {
                     'Content-Type': 'application/json',
-                    ...(this.config.apiKey || process.env.HF_API_TOKEN
-                        ? { Authorization: `Bearer ${process.env.HF_API_TOKEN}` }
-                        : {}),
+                    ...(this.getApiKey() ? { Authorization: `Bearer ${this.getApiKey()}` } : {}),
                 },
                 body: JSON.stringify(params),
             }, shared_1.REQUEST_TIMEOUT_MS);
@@ -230,6 +232,9 @@ class HuggingfaceSentenceSimilarityProvider {
     id() {
         return `huggingface:sentence-similarity:${this.modelName}`;
     }
+    getApiKey() {
+        return this.config.apiKey || process.env.HF_API_TOKEN;
+    }
     toString() {
         return `[Huggingface Sentence Similarity Provider ${this.modelName}]`;
     }
@@ -250,13 +255,14 @@ class HuggingfaceSentenceSimilarityProvider {
         };
         let response;
         try {
-            response = await (0, cache_1.fetchWithCache)(`https://api-inference.huggingface.co/models/${this.modelName}`, {
+            const url = this.config.apiEndpoint
+                ? this.config.apiEndpoint
+                : `https://api-inference.huggingface.co/models/${this.modelName}`;
+            response = await (0, cache_1.fetchWithCache)(url, {
                 method: 'POST',
                 headers: {
                     'Content-Type': 'application/json',
-                    ...(process.env.HF_API_TOKEN
-                        ? { Authorization: `Bearer ${process.env.HF_API_TOKEN}` }
-                        : {}),
+                    ...(this.getApiKey() ? { Authorization: `Bearer ${this.getApiKey()}` } : {}),
                 },
                 body: JSON.stringify(params),
             }, shared_1.REQUEST_TIMEOUT_MS);

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

@@ -1 +1 @@
-{"version":3,"file":"huggingface.js","sourceRoot":"","sources":["../../../src/providers/huggingface.ts"],"names":[],"mappings":";;;;;;AAAA,uDAA+B;AAC/B,oCAA0C;AAU1C,qCAA8C;AAkB9C,MAAM,6BAA6B,GAAG,IAAI,GAAG,CAAyC;IACpF,OAAO;IACP,OAAO;IACP,aAAa;IACb,oBAAoB;IACpB,gBAAgB;IAChB,UAAU;IACV,kBAAkB;IAClB,sBAAsB;IACtB,WAAW;IACX,WAAW;IACX,gBAAgB;CACjB,CAAC,CAAC;AAEH,MAAa,iCAAiC;IAI5C,YACE,SAAiB,EACjB,UAAsE,EAAE;QAExE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,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,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,EAAE;QACA,OAAO,+BAA+B,IAAI,CAAC,SAAS,EAAE,CAAC;IACzD,CAAC;IAED,QAAQ;QACN,OAAO,yCAAyC,IAAI,CAAC,SAAS,GAAG,CAAC;IACpE,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;IACxD,CAAC;IAED,SAAS;QACP,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE;YACtD,MAAM,UAAU,GAAG,GAA6C,CAAC;YACjE,IAAI,6BAA6B,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;gBACjD,OAAO,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;aAC/C;YACD,OAAO,OAAO,CAAC;QACjB,CAAC,EAAE,EAAoG,CAAC,CAAC;IAC3G,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,MAAM,GAAG;YACb,MAAM,EAAE,MAAM;YACd,UAAU,EAAE;gBACV,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,KAAK;gBACvD,GAAG,IAAI,CAAC,SAAS,EAAE;aACpB;SACF,CAAC;QAEF,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;YACjC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW;YACzB,CAAC,CAAC,+CAA+C,IAAI,CAAC,SAAS,EAAE,CAAC;QACpE,gBAAM,CAAC,KAAK,CAAC,4BAA4B,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAE1E,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,QAAQ,GAAG,MAAM,IAAA,sBAAc,EAC7B,GAAG,EACH;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,MAAM,EAAE,kBAAkB;oBAC1B,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC7E;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC7B,EACD,2BAAkB,CACnB,CAAC;YAEF,gBAAM,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAE3E,IAAI,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;gBACvB,OAAO;oBACL,KAAK,EAAE,mBAAmB,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;iBAChD,CAAC;aACH;YACD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,cAAc,EAAE;gBACtD,OAAO;oBACL,KAAK,EAAE,4BAA4B,QAAQ,CAAC,IAAI,EAAE;iBACnD,CAAC;aACH;YAED,OAAO;gBACL,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,cAAc,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,cAAc;aACzE,CAAC;SACH;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,cAAc,QAAQ,EAAE,IAAI,EAAE;aACpE,CAAC;SACH;IACH,CAAC;CACF;AAxFD,8EAwFC;AAOD,MAAa,qCAAqC;IAIhD,YACE,SAAiB,EACjB,UAA0E,EAAE;QAE5E,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,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,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,EAAE;QACA,OAAO,mCAAmC,IAAI,CAAC,SAAS,EAAE,CAAC;IAC7D,CAAC;IAED,QAAQ;QACN,OAAO,6CAA6C,IAAI,CAAC,SAAS,GAAG,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,MAAc;QACxC,MAAM,MAAM,GAAG;YACb,MAAM,EAAE,MAAM;YACd,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;gBACjC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW;gBACzB,CAAC,CAAC,+CAA+C,IAAI,CAAC,SAAS,EAAE,CAAC;YACpE,QAAQ,GAAG,MAAM,IAAA,sBAAc,EAC7B,GAAG,EACH;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY;wBAChD,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE;wBACzD,CAAC,CAAC,EAAE,CAAC;iBACR;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC7B,EACD,2BAAkB,CACnB,CAAC;YAEF,IAAI,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;gBACvB,OAAO;oBACL,KAAK,EAAE,mBAAmB,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;iBAChD,CAAC;aACH;YACD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE;gBACzD,OAAO;oBACL,KAAK,EAAE,4BAA4B,QAAQ,CAAC,IAAI,EAAE;iBACnD,CAAC;aACH;YAED,MAAM,MAAM,GAA2B,EAAE,CAAC;YAC1C,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBAChC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;YAClC,CAAC,CAAC,CAAC;YAEH,OAAO;gBACL,cAAc,EAAE,MAAM;aACvB,CAAC;SACH;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,cAAc,QAAQ,EAAE,IAAI,EAAE;aACpE,CAAC;SACH;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QACrD,OAAO;YACL,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,cAAc,CAAC;SAC3C,CAAC;IACJ,CAAC;CACF;AAjFD,sFAiFC;AASD,MAAa,oCAAoC;IAI/C,YACE,SAAiB,EACjB,UAAyE,EAAE;QAE3E,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,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,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,EAAE;QACA,OAAO,kCAAkC,IAAI,CAAC,SAAS,EAAE,CAAC;IAC5D,CAAC;IAED,QAAQ;QACN,OAAO,4CAA4C,IAAI,CAAC,SAAS,GAAG,CAAC;IACvE,CAAC;IAED,KAAK,CAAC,OAAO;QACX,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;IAClF,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,IAAY;QACjC,wFAAwF;QACxF,MAAM,MAAM,GAAG;YACb,MAAM,EAAE,IAAI;YACZ,OAAO,EAAE;gBACP,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;gBAChC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;aAC3C;SACF,CAAC;QAEF,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;gBACjC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW;gBACzB,CAAC,CAAC,+CAA+C,IAAI,CAAC,SAAS,EAAE,CAAC;YACpE,gBAAM,CAAC,KAAK,CAAC,4BAA4B,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC1E,QAAQ,GAAG,MAAM,IAAA,sBAAc,EAC7B,GAAG,EACH;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY;wBAChD,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE;wBACzD,CAAC,CAAC,EAAE,CAAC;iBACR;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC7B,EACD,2BAAkB,CACnB,CAAC;YAEF,IAAI,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;gBACvB,OAAO;oBACL,KAAK,EAAE,mBAAmB,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;iBAChD,CAAC;aACH;YACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;gBACjC,OAAO;oBACL,KAAK,EAAE,4BAA4B,QAAQ,CAAC,IAAI,EAAE;iBACnD,CAAC;aACH;YAED,OAAO;gBACL,SAAS,EAAE,QAAQ,CAAC,IAAI;aACzB,CAAC;SACH;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,cAAc,QAAQ,EAAE,IAAI,EAAE;aACpE,CAAC;SACH;IACH,CAAC;CACF;AA7ED,oFA6EC;AAOD,MAAa,qCAAqC;IAIhD,YACE,SAAiB,EACjB,UAA0E,EAAE;QAE5E,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,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,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,EAAE;QACA,OAAO,mCAAmC,IAAI,CAAC,SAAS,EAAE,CAAC;IAC7D,CAAC;IAED,QAAQ;QACN,OAAO,6CAA6C,IAAI,CAAC,SAAS,GAAG,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,OAAO;QACX,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;IACnF,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,QAAgB,EAAE,KAAa;QACrD,yFAAyF;QACzF,MAAM,MAAM,GAAG;YACb,MAAM,EAAE;gBACN,eAAe,EAAE,QAAQ;gBACzB,SAAS,EAAE,CAAC,KAAK,CAAC;aACnB;YACD,OAAO,EAAE;gBACP,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;gBAChC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;aAC3C;SACF,CAAC;QAEF,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,QAAQ,GAAG,MAAM,IAAA,sBAAc,EAC7B,+CAA+C,IAAI,CAAC,SAAS,EAAE,EAC/D;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY;wBAC1B,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE;wBACzD,CAAC,CAAC,EAAE,CAAC;iBACR;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC7B,EACD,2BAAkB,CACnB,CAAC;YAEF,IAAI,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;gBACvB,OAAO;oBACL,KAAK,EAAE,mBAAmB,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;iBAChD,CAAC;aACH;YACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;gBACjC,OAAO;oBACL,KAAK,EAAE,4BAA4B,QAAQ,CAAC,IAAI,EAAE;iBACnD,CAAC;aACH;YAED,OAAO;gBACL,UAAU,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;aAC7B,CAAC;SACH;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,cAAc,QAAQ,EAAE,IAAI,EAAE;aACpE,CAAC;SACH;IACH,CAAC;CACF;AA5ED,sFA4EC;AAUD,MAAa,kCAAkC;IAI7C,YACE,SAAiB,EACjB,UAA2E,EAAE;QAE7E,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,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,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,EAAE;QACA,OAAO,oCAAoC,IAAI,CAAC,SAAS,EAAE,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,KAAa;QACvC,MAAM,MAAM,GAAG;YACb,MAAM,EAAE,KAAK;YACb,UAAU,EAAE;gBACV,oBAAoB,EAAE,IAAI,CAAC,MAAM,CAAC,oBAAoB,IAAI,QAAQ;aACnE;YACD,OAAO,EAAE;gBACP,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI;gBAC7E,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,IAAI,KAAK;aACpD;SACF,CAAC;QAEF,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;gBACjC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW;gBACzB,CAAC,CAAC,+CAA+C,IAAI,CAAC,SAAS,EAAE,CAAC;YACpE,QAAQ,GAAG,MAAM,IAAA,sBAAc,EAC7B,GAAG,EACH;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY;wBAC1B,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE;wBACzD,CAAC,CAAC,EAAE,CAAC;iBACR;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC7B,EACD,2BAAkB,CACnB,CAAC;YAEF,IAAI,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;gBACvB,OAAO;oBACL,KAAK,EAAE,mBAAmB,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;iBAChD,CAAC;aACH;YACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;gBACjC,OAAO;oBACL,KAAK,EAAE,4BAA4B,QAAQ,CAAC,IAAI,EAAE;iBACnD,CAAC;aACH;YAED,8CAA8C;YAC9C,MAAM,cAAc,GAA2B,EAAE,CAAC;YAClD,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,IAAI,EAAE;gBAChC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,cAAc,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,IAAI,CAAC,KAAK,EAAE;oBACxF,cAAc,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;iBAChD;aACF;YACD,OAAO,EAAE,cAAc,EAAE,CAAC;SAC3B;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,cAAc,QAAQ,EAAE,IAAI,EAAE;aACpE,CAAC;SACH;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QACrD,OAAO;YACL,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,cAAc,CAAC;SAC3C,CAAC;IACJ,CAAC;CACF;AAnFD,gFAmFC"}
+{"version":3,"file":"huggingface.js","sourceRoot":"","sources":["../../../src/providers/huggingface.ts"],"names":[],"mappings":";;;;;;AAAA,uDAA+B;AAC/B,oCAA0C;AAU1C,qCAA8C;AAqB9C,MAAM,6BAA6B,GAAG,IAAI,GAAG,CAAyC;IACpF,OAAO;IACP,OAAO;IACP,aAAa;IACb,oBAAoB;IACpB,gBAAgB;IAChB,UAAU;IACV,kBAAkB;IAClB,sBAAsB;IACtB,WAAW;IACX,WAAW;IACX,gBAAgB;CACjB,CAAC,CAAC;AAEH,MAAa,iCAAiC;IAI5C,YACE,SAAiB,EACjB,UAAsE,EAAE;QAExE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,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,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,EAAE;QACA,OAAO,+BAA+B,IAAI,CAAC,SAAS,EAAE,CAAC;IACzD,CAAC;IAED,QAAQ;QACN,OAAO,yCAAyC,IAAI,CAAC,SAAS,GAAG,CAAC;IACpE,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;IACxD,CAAC;IAED,SAAS;QACP,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE;YACtD,MAAM,UAAU,GAAG,GAA6C,CAAC;YACjE,IAAI,6BAA6B,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;gBACjD,OAAO,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;aAC/C;YACD,OAAO,OAAO,CAAC;QACjB,CAAC,EAAE,EAAoG,CAAC,CAAC;IAC3G,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,MAAM,GAAG;YACb,MAAM,EAAE,MAAM;YACd,UAAU,EAAE;gBACV,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,KAAK;gBACvD,GAAG,IAAI,CAAC,SAAS,EAAE;aACpB;SACF,CAAC;QAEF,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;YACjC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW;YACzB,CAAC,CAAC,+CAA+C,IAAI,CAAC,SAAS,EAAE,CAAC;QACpE,gBAAM,CAAC,KAAK,CAAC,4BAA4B,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAE1E,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,QAAQ,GAAG,MAAM,IAAA,sBAAc,EAC7B,GAAG,EACH;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,MAAM,EAAE,kBAAkB;oBAC1B,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC7E;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC7B,EACD,2BAAkB,CACnB,CAAC;YAEF,gBAAM,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAE3E,IAAI,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;gBACvB,OAAO;oBACL,KAAK,EAAE,mBAAmB,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;iBAChD,CAAC;aACH;YACD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,cAAc,EAAE;gBACtD,OAAO;oBACL,KAAK,EAAE,4BAA4B,QAAQ,CAAC,IAAI,EAAE;iBACnD,CAAC;aACH;YAED,OAAO;gBACL,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,cAAc,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,cAAc;aACzE,CAAC;SACH;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,cAAc,QAAQ,EAAE,IAAI,EAAE;aACpE,CAAC;SACH;IACH,CAAC;CACF;AAxFD,8EAwFC;AAID,MAAa,qCAAqC;IAIhD,YACE,SAAiB,EACjB,UAA0E,EAAE;QAE5E,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,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,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,EAAE;QACA,OAAO,mCAAmC,IAAI,CAAC,SAAS,EAAE,CAAC;IAC7D,CAAC;IAED,QAAQ;QACN,OAAO,6CAA6C,IAAI,CAAC,SAAS,GAAG,CAAC;IACxE,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;IACxD,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,MAAc;QACxC,MAAM,MAAM,GAAG;YACb,MAAM,EAAE,MAAM;YACd,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;gBACjC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW;gBACzB,CAAC,CAAC,+CAA+C,IAAI,CAAC,SAAS,EAAE,CAAC;YACpE,QAAQ,GAAG,MAAM,IAAA,sBAAc,EAC7B,GAAG,EACH;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC7E;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC7B,EACD,2BAAkB,CACnB,CAAC;YAEF,IAAI,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;gBACvB,OAAO;oBACL,KAAK,EAAE,mBAAmB,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;iBAChD,CAAC;aACH;YACD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE;gBACzD,OAAO;oBACL,KAAK,EAAE,4BAA4B,QAAQ,CAAC,IAAI,EAAE;iBACnD,CAAC;aACH;YAED,MAAM,MAAM,GAA2B,EAAE,CAAC;YAC1C,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBAChC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;YAClC,CAAC,CAAC,CAAC;YAEH,OAAO;gBACL,cAAc,EAAE,MAAM;aACvB,CAAC;SACH;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,cAAc,QAAQ,EAAE,IAAI,EAAE;aACpE,CAAC;SACH;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QACrD,OAAO;YACL,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,cAAc,CAAC;SAC3C,CAAC;IACJ,CAAC;CACF;AAnFD,sFAmFC;AAOD,MAAa,oCAAoC;IAI/C,YACE,SAAiB,EACjB,UAAyE,EAAE;QAE3E,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,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,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,EAAE;QACA,OAAO,kCAAkC,IAAI,CAAC,SAAS,EAAE,CAAC;IAC5D,CAAC;IAED,QAAQ;QACN,OAAO,4CAA4C,IAAI,CAAC,SAAS,GAAG,CAAC;IACvE,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;IACxD,CAAC;IAED,KAAK,CAAC,OAAO;QACX,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;IAClF,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,IAAY;QACjC,wFAAwF;QACxF,MAAM,MAAM,GAAG;YACb,MAAM,EAAE,IAAI;YACZ,OAAO,EAAE;gBACP,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;gBAChC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;aAC3C;SACF,CAAC;QAEF,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;gBACjC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW;gBACzB,CAAC,CAAC,+CAA+C,IAAI,CAAC,SAAS,EAAE,CAAC;YACpE,gBAAM,CAAC,KAAK,CAAC,4BAA4B,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC1E,QAAQ,GAAG,MAAM,IAAA,sBAAc,EAC7B,GAAG,EACH;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC7E;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC7B,EACD,2BAAkB,CACnB,CAAC;YAEF,IAAI,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;gBACvB,OAAO;oBACL,KAAK,EAAE,mBAAmB,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;iBAChD,CAAC;aACH;YACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;gBACjC,OAAO;oBACL,KAAK,EAAE,4BAA4B,QAAQ,CAAC,IAAI,EAAE;iBACnD,CAAC;aACH;YAED,OAAO;gBACL,SAAS,EAAE,QAAQ,CAAC,IAAI;aACzB,CAAC;SACH;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,cAAc,QAAQ,EAAE,IAAI,EAAE;aACpE,CAAC;SACH;IACH,CAAC;CACF;AA/ED,oFA+EC;AAOD,MAAa,qCAAqC;IAIhD,YACE,SAAiB,EACjB,UAA0E,EAAE;QAE5E,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,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,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,EAAE;QACA,OAAO,mCAAmC,IAAI,CAAC,SAAS,EAAE,CAAC;IAC7D,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;IACxD,CAAC;IAED,QAAQ;QACN,OAAO,6CAA6C,IAAI,CAAC,SAAS,GAAG,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,OAAO;QACX,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;IACnF,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,QAAgB,EAAE,KAAa;QACrD,yFAAyF;QACzF,MAAM,MAAM,GAAG;YACb,MAAM,EAAE;gBACN,eAAe,EAAE,QAAQ;gBACzB,SAAS,EAAE,CAAC,KAAK,CAAC;aACnB;YACD,OAAO,EAAE;gBACP,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;gBAChC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;aAC3C;SACF,CAAC;QAEF,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;gBACjC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW;gBACzB,CAAC,CAAC,+CAA+C,IAAI,CAAC,SAAS,EAAE,CAAC;YACpE,QAAQ,GAAG,MAAM,IAAA,sBAAc,EAC7B,GAAG,EACH;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC7E;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC7B,EACD,2BAAkB,CACnB,CAAC;YAEF,IAAI,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;gBACvB,OAAO;oBACL,KAAK,EAAE,mBAAmB,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;iBAChD,CAAC;aACH;YACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;gBACjC,OAAO;oBACL,KAAK,EAAE,4BAA4B,QAAQ,CAAC,IAAI,EAAE;iBACnD,CAAC;aACH;YAED,OAAO;gBACL,UAAU,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;aAC7B,CAAC;SACH;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,cAAc,QAAQ,EAAE,IAAI,EAAE;aACpE,CAAC;SACH;IACH,CAAC;CACF;AAjFD,sFAiFC;AAQD,MAAa,kCAAkC;IAI7C,YACE,SAAiB,EACjB,UAA2E,EAAE;QAE7E,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,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,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,EAAE;QACA,OAAO,oCAAoC,IAAI,CAAC,SAAS,EAAE,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,KAAa;QACvC,MAAM,MAAM,GAAG;YACb,MAAM,EAAE,KAAK;YACb,UAAU,EAAE;gBACV,oBAAoB,EAAE,IAAI,CAAC,MAAM,CAAC,oBAAoB,IAAI,QAAQ;aACnE;YACD,OAAO,EAAE;gBACP,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI;gBAC7E,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,IAAI,KAAK;aACpD;SACF,CAAC;QAEF,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;gBACjC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW;gBACzB,CAAC,CAAC,+CAA+C,IAAI,CAAC,SAAS,EAAE,CAAC;YACpE,QAAQ,GAAG,MAAM,IAAA,sBAAc,EAC7B,GAAG,EACH;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY;wBAC1B,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE;wBACzD,CAAC,CAAC,EAAE,CAAC;iBACR;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC7B,EACD,2BAAkB,CACnB,CAAC;YAEF,IAAI,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;gBACvB,OAAO;oBACL,KAAK,EAAE,mBAAmB,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;iBAChD,CAAC;aACH;YACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;gBACjC,OAAO;oBACL,KAAK,EAAE,4BAA4B,QAAQ,CAAC,IAAI,EAAE;iBACnD,CAAC;aACH;YAED,8CAA8C;YAC9C,MAAM,cAAc,GAA2B,EAAE,CAAC;YAClD,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,IAAI,EAAE;gBAChC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,cAAc,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,IAAI,CAAC,KAAK,EAAE;oBACxF,cAAc,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;iBAChD;aACF;YACD,OAAO,EAAE,cAAc,EAAE,CAAC;SAC3B;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,cAAc,QAAQ,EAAE,IAAI,EAAE;aACpE,CAAC;SACH;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QACrD,OAAO;YACL,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,cAAc,CAAC;SAC3C,CAAC;IACJ,CAAC;CACF;AAnFD,gFAmFC"}

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

@@ -1 +1 @@
-{"version":3,"file":"mistral.d.ts","sourceRoot":"","sources":["../../../src/providers/mistral.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,EAAc,MAAM,UAAU,CAAC;AAGnF,UAAU,4BAA4B;IACpC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE;QAAE,IAAI,EAAE,aAAa,CAAA;KAAE,CAAC;IAC1C,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAuCD,qBAAa,6BAA8B,YAAW,WAAW;IAC/D,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,4BAA4B,CAAC;IACrC,GAAG,CAAC,EAAE,YAAY,CAAC;IAEnB,MAAM,CAAC,mBAAmB;;;;;;QA2CxB;IAEF,MAAM,CAAC,yBAAyB,WAE9B;gBAGA,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,4BAA4B,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAY1F,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIlB,gBAAgB,IAAI,MAAM;IAI1B,SAAS,IAAI,MAAM;IAcnB,SAAS,IAAI,MAAM,GAAG,SAAS;IAazB,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAsEzD"}
+{"version":3,"file":"mistral.d.ts","sourceRoot":"","sources":["../../../src/providers/mistral.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,EAAc,MAAM,UAAU,CAAC;AAGnF,UAAU,4BAA4B;IACpC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE;QAAE,IAAI,EAAE,aAAa,CAAA;KAAE,CAAC;IAC1C,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAuCD,qBAAa,6BAA8B,YAAW,WAAW;IAC/D,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,4BAA4B,CAAC;IACrC,GAAG,CAAC,EAAE,YAAY,CAAC;IAEnB,MAAM,CAAC,mBAAmB;;;;;;QAkDxB;IAEF,MAAM,CAAC,yBAAyB,WAE9B;gBAGA,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,4BAA4B,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAY1F,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIlB,gBAAgB,IAAI,MAAM;IAI1B,SAAS,IAAI,MAAM;IAcnB,SAAS,IAAI,MAAM,GAAG,SAAS;IAazB,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAsEzD"}

package/dist/src/providers/mistral.js

@@ -170,6 +170,13 @@ MistralChatCompletionProvider.MISTRAL_CHAT_MODELS = [
             output: 0.024 / 1000,
         },
     })),
+    ...['codestral-latest'].map((model) => ({
+        id: model,
+        cost: {
+            input: 0.002 / 1000,
+            output: 0.006 / 1000,
+        },
+    })),
 ];
 MistralChatCompletionProvider.MISTRAL_CHAT_MODELS_NAMES = MistralChatCompletionProvider.MISTRAL_CHAT_MODELS.map((model) => model.id);
 //# sourceMappingURL=mistral.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"mistral.js","sourceRoot":"","sources":["../../../src/providers/mistral.ts"],"names":[],"mappings":";;;;;;AAAA,uDAA+B;AAC/B,oCAA0C;AAG1C,qCAA+D;AAgB/D,SAAS,aAAa,CAAC,IAAS,EAAE,MAAe;IAC/C,IAAI,IAAI,CAAC,KAAK,EAAE;QACd,IAAI,MAAM,EAAE;YACV,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;SAC5E;aAAM;YACL,OAAO;gBACL,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;gBAC9B,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC;gBACrC,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC;aAC9C,CAAC;SACH;KACF;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,aAAa,CACpB,SAAiB,EACjB,MAAoC,EACpC,YAAqB,EACrB,gBAAyB;IAEzB,IAAI,CAAC,YAAY,IAAI,CAAC,gBAAgB,EAAE;QACtC,OAAO,SAAS,CAAC;KAClB;IAED,MAAM,KAAK,GAAG,CAAC,GAAG,6BAA6B,CAAC,mBAAmB,CAAC,CAAC,IAAI,CACvE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,SAAS,CAC1B,CAAC;IACF,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE;QACzB,OAAO,SAAS,CAAC;KAClB;IAED,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC;IAClD,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;IACpD,OAAO,SAAS,GAAG,YAAY,GAAG,UAAU,GAAG,gBAAgB,IAAI,SAAS,CAAC;AAC/E,CAAC;AAED,MAAa,6BAA6B;IAsDxC,YACE,SAAiB,EACjB,UAAsF,EAAE;QAExF,IAAI,CAAC,6BAA6B,CAAC,yBAAyB,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE;YAChF,gBAAM,CAAC,IAAI,CAAC,qCAAqC,SAAS,EAAE,CAAC,CAAC;SAC/D;QACD,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC;QACpC,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,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,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,EAAE;QACA,OAAO,WAAW,IAAI,CAAC,SAAS,EAAE,CAAC;IACrC,CAAC;IAED,QAAQ;QACN,OAAO,qBAAqB,IAAI,CAAC,SAAS,GAAG,CAAC;IAChD,CAAC;IAED,gBAAgB;QACd,OAAO,2BAA2B,CAAC;IACrC,CAAC;IAED,SAAS;QACP,MAAM,OAAO,GACX,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,GAAG,EAAE,gBAAgB,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;QACpF,IAAI,OAAO,EAAE;YACX,OAAO,WAAW,OAAO,KAAK,CAAC;SAChC;QACD,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,UAAU;YACtB,IAAI,CAAC,GAAG,EAAE,oBAAoB;YAC9B,OAAO,CAAC,GAAG,CAAC,oBAAoB;YAChC,IAAI,CAAC,gBAAgB,EAAE,CACxB,CAAC;IACJ,CAAC;IAED,SAAS;QACP,gBAAM,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;QAChE,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,MAAM;YAClB,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW;gBACvB,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;oBACpC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,WAAiC,CAAC;gBAC3D,CAAC,CAAC,SAAS,CAAC;YACd,IAAI,CAAC,GAAG,EAAE,eAAe;YACzB,OAAO,CAAC,GAAG,CAAC,eAAe,CAC5B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE;YACrB,MAAM,IAAI,KAAK,CACb,mIAAmI,CACpI,CAAC;SACH;QAED,MAAM,QAAQ,GAAG,IAAA,wBAAe,EAAC,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QAE9E,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,QAAQ,EAAE,QAAQ;YAClB,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,WAAW;YACrC,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC;YAC9B,UAAU,EAAE,IAAI,CAAC,MAAM,EAAE,UAAU,IAAI,IAAI;YAC3C,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,WAAW,IAAI,KAAK;YAC9C,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,WAAW,IAAI,IAAI;YAC7C,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACzF,CAAC;QAEF,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,SAAS,EAAE,mBAAmB,CAAC;QACnD,gBAAM,CAAC,KAAK,CAAC,wBAAwB,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEpE,IAAI,IAAI,EACN,MAAM,GAAG,KAAK,CAAC;QAEjB,IAAI;YACF,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,IAAA,sBAAc,EACvC,GAAG,EACH;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE,EAAE;iBAC5C;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,EACD,2BAAkB,CACnB,CAA8C,CAAC,CAAC;SAClD;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;SACH;QAED,gBAAM,CAAC,KAAK,CAAC,yBAAyB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE9D,IAAI,IAAI,CAAC,KAAK,EAAE;YACd,OAAO;gBACL,KAAK,EAAE,mBAAmB,IAAI,CAAC,KAAK,EAAE;aACvC,CAAC;SACH;QACD,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE;YACxD,OAAO;gBACL,KAAK,EAAE,4BAA4B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;aAC1D,CAAC;SACH;QAED,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO;YACvC,UAAU,EAAE,aAAa,CAAC,IAAI,EAAE,MAAM,CAAC;YACvC,MAAM;YACN,IAAI,EAAE,aAAa,CACjB,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,KAAK,EAAE,aAAa,EACzB,IAAI,CAAC,KAAK,EAAE,iBAAiB,CAC9B;SACF,CAAC;IACJ,CAAC;;AAhLH,sEAiLC;AA5KQ,iDAAmB,GAAG;IAC3B,GAAG,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACrC,EAAE,EAAE,KAAK;QACT,IAAI,EAAE;YACJ,KAAK,EAAE,OAAO,GAAG,IAAI;YACrB,MAAM,EAAE,OAAO,GAAG,IAAI;SACvB;KACF,CAAC,CAAC;IACH,GAAG,CAAC,mBAAmB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACvC,EAAE,EAAE,KAAK;QACT,IAAI,EAAE;YACJ,KAAK,EAAE,MAAM,GAAG,IAAI;YACpB,MAAM,EAAE,MAAM,GAAG,IAAI;SACtB;KACF,CAAC,CAAC;IACH,GAAG,CAAC,oBAAoB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACxC,EAAE,EAAE,KAAK;QACT,IAAI,EAAE;YACJ,KAAK,EAAE,KAAK,GAAG,IAAI;YACnB,MAAM,EAAE,KAAK,GAAG,IAAI;SACrB;KACF,CAAC,CAAC;IACH,GAAG,CAAC,sBAAsB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC1C,EAAE,EAAE,KAAK;QACT,IAAI,EAAE;YACJ,KAAK,EAAE,KAAK,GAAG,IAAI;YACnB,MAAM,EAAE,KAAK,GAAG,IAAI;SACrB;KACF,CAAC,CAAC;IACH,GAAG,CAAC,uBAAuB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC3C,EAAE,EAAE,KAAK;QACT,IAAI,EAAE;YACJ,KAAK,EAAE,MAAM,GAAG,IAAI;YACpB,MAAM,EAAE,MAAM,GAAG,IAAI;SACtB;KACF,CAAC,CAAC;IACH,GAAG,CAAC,sBAAsB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC1C,EAAE,EAAE,KAAK;QACT,IAAI,EAAE;YACJ,KAAK,EAAE,KAAK,GAAG,IAAI;YACnB,MAAM,EAAE,KAAK,GAAG,IAAI;SACrB;KACF,CAAC,CAAC;CACJ,CAAC;AAEK,uDAAyB,GAAG,6BAA6B,CAAC,mBAAmB,CAAC,GAAG,CACtF,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CACpB,CAAC"}
+{"version":3,"file":"mistral.js","sourceRoot":"","sources":["../../../src/providers/mistral.ts"],"names":[],"mappings":";;;;;;AAAA,uDAA+B;AAC/B,oCAA0C;AAG1C,qCAA+D;AAgB/D,SAAS,aAAa,CAAC,IAAS,EAAE,MAAe;IAC/C,IAAI,IAAI,CAAC,KAAK,EAAE;QACd,IAAI,MAAM,EAAE;YACV,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;SAC5E;aAAM;YACL,OAAO;gBACL,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;gBAC9B,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC;gBACrC,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC;aAC9C,CAAC;SACH;KACF;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,aAAa,CACpB,SAAiB,EACjB,MAAoC,EACpC,YAAqB,EACrB,gBAAyB;IAEzB,IAAI,CAAC,YAAY,IAAI,CAAC,gBAAgB,EAAE;QACtC,OAAO,SAAS,CAAC;KAClB;IAED,MAAM,KAAK,GAAG,CAAC,GAAG,6BAA6B,CAAC,mBAAmB,CAAC,CAAC,IAAI,CACvE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,SAAS,CAC1B,CAAC;IACF,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE;QACzB,OAAO,SAAS,CAAC;KAClB;IAED,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC;IAClD,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;IACpD,OAAO,SAAS,GAAG,YAAY,GAAG,UAAU,GAAG,gBAAgB,IAAI,SAAS,CAAC;AAC/E,CAAC;AAED,MAAa,6BAA6B;IA6DxC,YACE,SAAiB,EACjB,UAAsF,EAAE;QAExF,IAAI,CAAC,6BAA6B,CAAC,yBAAyB,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE;YAChF,gBAAM,CAAC,IAAI,CAAC,qCAAqC,SAAS,EAAE,CAAC,CAAC;SAC/D;QACD,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC;QACpC,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,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,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,EAAE;QACA,OAAO,WAAW,IAAI,CAAC,SAAS,EAAE,CAAC;IACrC,CAAC;IAED,QAAQ;QACN,OAAO,qBAAqB,IAAI,CAAC,SAAS,GAAG,CAAC;IAChD,CAAC;IAED,gBAAgB;QACd,OAAO,2BAA2B,CAAC;IACrC,CAAC;IAED,SAAS;QACP,MAAM,OAAO,GACX,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,GAAG,EAAE,gBAAgB,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;QACpF,IAAI,OAAO,EAAE;YACX,OAAO,WAAW,OAAO,KAAK,CAAC;SAChC;QACD,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,UAAU;YACtB,IAAI,CAAC,GAAG,EAAE,oBAAoB;YAC9B,OAAO,CAAC,GAAG,CAAC,oBAAoB;YAChC,IAAI,CAAC,gBAAgB,EAAE,CACxB,CAAC;IACJ,CAAC;IAED,SAAS;QACP,gBAAM,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;QAChE,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,MAAM;YAClB,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW;gBACvB,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;oBACpC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,WAAiC,CAAC;gBAC3D,CAAC,CAAC,SAAS,CAAC;YACd,IAAI,CAAC,GAAG,EAAE,eAAe;YACzB,OAAO,CAAC,GAAG,CAAC,eAAe,CAC5B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE;YACrB,MAAM,IAAI,KAAK,CACb,mIAAmI,CACpI,CAAC;SACH;QAED,MAAM,QAAQ,GAAG,IAAA,wBAAe,EAAC,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QAE9E,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,QAAQ,EAAE,QAAQ;YAClB,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,WAAW;YACrC,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC;YAC9B,UAAU,EAAE,IAAI,CAAC,MAAM,EAAE,UAAU,IAAI,IAAI;YAC3C,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,WAAW,IAAI,KAAK;YAC9C,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,WAAW,IAAI,IAAI;YAC7C,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACzF,CAAC;QAEF,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,SAAS,EAAE,mBAAmB,CAAC;QACnD,gBAAM,CAAC,KAAK,CAAC,wBAAwB,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEpE,IAAI,IAAI,EACN,MAAM,GAAG,KAAK,CAAC;QAEjB,IAAI;YACF,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,IAAA,sBAAc,EACvC,GAAG,EACH;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE,EAAE;iBAC5C;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,EACD,2BAAkB,CACnB,CAA8C,CAAC,CAAC;SAClD;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;SACH;QAED,gBAAM,CAAC,KAAK,CAAC,yBAAyB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE9D,IAAI,IAAI,CAAC,KAAK,EAAE;YACd,OAAO;gBACL,KAAK,EAAE,mBAAmB,IAAI,CAAC,KAAK,EAAE;aACvC,CAAC;SACH;QACD,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE;YACxD,OAAO;gBACL,KAAK,EAAE,4BAA4B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;aAC1D,CAAC;SACH;QAED,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO;YACvC,UAAU,EAAE,aAAa,CAAC,IAAI,EAAE,MAAM,CAAC;YACvC,MAAM;YACN,IAAI,EAAE,aAAa,CACjB,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,KAAK,EAAE,aAAa,EACzB,IAAI,CAAC,KAAK,EAAE,iBAAiB,CAC9B;SACF,CAAC;IACJ,CAAC;;AAvLH,sEAwLC;AAnLQ,iDAAmB,GAAG;IAC3B,GAAG,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACrC,EAAE,EAAE,KAAK;QACT,IAAI,EAAE;YACJ,KAAK,EAAE,OAAO,GAAG,IAAI;YACrB,MAAM,EAAE,OAAO,GAAG,IAAI;SACvB;KACF,CAAC,CAAC;IACH,GAAG,CAAC,mBAAmB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACvC,EAAE,EAAE,KAAK;QACT,IAAI,EAAE;YACJ,KAAK,EAAE,MAAM,GAAG,IAAI;YACpB,MAAM,EAAE,MAAM,GAAG,IAAI;SACtB;KACF,CAAC,CAAC;IACH,GAAG,CAAC,oBAAoB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACxC,EAAE,EAAE,KAAK;QACT,IAAI,EAAE;YACJ,KAAK,EAAE,KAAK,GAAG,IAAI;YACnB,MAAM,EAAE,KAAK,GAAG,IAAI;SACrB;KACF,CAAC,CAAC;IACH,GAAG,CAAC,sBAAsB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC1C,EAAE,EAAE,KAAK;QACT,IAAI,EAAE;YACJ,KAAK,EAAE,KAAK,GAAG,IAAI;YACnB,MAAM,EAAE,KAAK,GAAG,IAAI;SACrB;KACF,CAAC,CAAC;IACH,GAAG,CAAC,uBAAuB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC3C,EAAE,EAAE,KAAK;QACT,IAAI,EAAE;YACJ,KAAK,EAAE,MAAM,GAAG,IAAI;YACpB,MAAM,EAAE,MAAM,GAAG,IAAI;SACtB;KACF,CAAC,CAAC;IACH,GAAG,CAAC,sBAAsB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC1C,EAAE,EAAE,KAAK;QACT,IAAI,EAAE;YACJ,KAAK,EAAE,KAAK,GAAG,IAAI;YACnB,MAAM,EAAE,KAAK,GAAG,IAAI;SACrB;KACF,CAAC,CAAC;IACH,GAAG,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACtC,EAAE,EAAE,KAAK;QACT,IAAI,EAAE;YACJ,KAAK,EAAE,KAAK,GAAG,IAAI;YACnB,MAAM,EAAE,KAAK,GAAG,IAAI;SACrB;KACF,CAAC,CAAC;CACJ,CAAC;AAEK,uDAAyB,GAAG,6BAA6B,CAAC,mBAAmB,CAAC,GAAG,CACtF,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CACpB,CAAC"}

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

@@ -1,6 +1,6 @@
 import OpenAI from 'openai';
 import { OpenAiFunction, OpenAiTool } from './openaiUtil';
-import type { ApiProvider, CallApiContextParams, CallApiOptionsParams, EnvOverrides, ProviderEmbeddingResponse, ProviderResponse } from '../types.js';
+import type { ApiModerationProvider, ApiProvider, CallApiContextParams, CallApiOptionsParams, EnvOverrides, ProviderEmbeddingResponse, ProviderModerationResponse, ProviderResponse } from '../types.js';
 interface OpenAiSharedOptions {
     apiKey?: string;
     apiKeyEnvar?: string;
@@ -136,9 +136,14 @@ export declare class OpenAiImageProvider extends OpenAiGenericProvider {
     });
     callApi(prompt: string, context?: CallApiContextParams, callApiOptions?: CallApiOptionsParams): Promise<ProviderResponse>;
 }
+export declare class OpenAiModerationProvider extends OpenAiGenericProvider implements ApiModerationProvider {
+    callModerationApi(userPrompt: string, // userPrompt is not supported by OpenAI moderation API
+    assistantResponse: string): Promise<ProviderModerationResponse>;
+}
 export declare const DefaultEmbeddingProvider: OpenAiEmbeddingProvider;
 export declare const DefaultGradingProvider: OpenAiChatCompletionProvider;
 export declare const DefaultGradingJsonProvider: OpenAiChatCompletionProvider;
 export declare const DefaultSuggestionsProvider: OpenAiChatCompletionProvider;
+export declare const DefaultModerationProvider: OpenAiModerationProvider;
 export {};
 //# sourceMappingURL=openai.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/providers/openai.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAM5B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAG1D,OAAO,KAAK,EACV,WAAW,EACX,oBAAoB,EACpB,oBAAoB,EACpB,YAAY,EACZ,yBAAyB,EACzB,gBAAgB,EAEjB,MAAM,aAAa,CAAC;AAErB,UAAU,mBAAmB;IAC3B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAAA;KAAE,CAAC;CACrC;AAED,KAAK,uBAAuB,GAAG,mBAAmB,GAAG;IACnD,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,cAAc,EAAE,CAAC;IAC7B,aAAa,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IACnD,KAAK,CAAC,EAAE,UAAU,EAAE,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG;QAAE,IAAI,EAAE,UAAU,CAAC;QAAC,QAAQ,CAAC,EAAE;YAAE,IAAI,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE,CAAC;IAClF,eAAe,CAAC,EAAE;QAAE,IAAI,EAAE,aAAa,CAAA;KAAE,CAAC;IAC1C,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AA4BF,qBAAa,qBAAsB,YAAW,WAAW;IACvD,SAAS,EAAE,MAAM,CAAC;IAElB,MAAM,EAAE,mBAAmB,CAAC;IAC5B,GAAG,CAAC,EAAE,YAAY,CAAC;gBAGjB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,mBAAmB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IASjF,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIlB,eAAe,IAAI,MAAM,GAAG,SAAS;IAMrC,gBAAgB,IAAI,MAAM;IAI1B,SAAS,IAAI,MAAM;IAenB,SAAS,IAAI,MAAM,GAAG,SAAS;IAazB,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CAG7B;AAED,qBAAa,uBAAwB,SAAQ,qBAAqB;IAC1D,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,yBAAyB,CAAC;CA8CzE;AAED,qBAAa,wBAAyB,SAAQ,qBAAqB;IACjE,MAAM,CAAC,wBAAwB;;;;;;;;;SA8B7B;IAEF,MAAM,CAAC,6BAA6B,WAElC;IAEF,MAAM,EAAE,uBAAuB,CAAC;gBAG9B,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;IAY/E,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CA8E7B;AAED,qBAAa,4BAA6B,SAAQ,qBAAqB;IACrE,MAAM,CAAC,kBAAkB;;;;;;QAkDvB;IAEF,MAAM,CAAC,uBAAuB,WAE5B;IAEF,MAAM,EAAE,uBAAuB,CAAC;gBAG9B,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;IAS/E,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CAgG7B;AAmDD,KAAK,sBAAsB,GAAG,mBAAmB,GAAG;IAClD,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC;IAC9D;;;OAGG;IACH,qBAAqB,CAAC,EAAE,MAAM,CAC5B,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,EACjC,CAAC,GAAG,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CACjC,CAAC;IACF,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EACT,MAAM,GACN,MAAM,GACN;QAAE,IAAI,EAAE,UAAU,CAAC;QAAC,QAAQ,CAAC,EAAE;YAAE,IAAI,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE,GACjD;QAAE,IAAI,EAAE,aAAa,CAAA;KAAE,CAAC;CAC3B,CAAC;AAEF,qBAAa,uBAAwB,SAAQ,qBAAqB;IAChE,WAAW,EAAE,MAAM,CAAC;IACpB,eAAe,EAAE,sBAAsB,CAAC;gBAGtC,WAAW,EAAE,MAAM,EACnB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,sBAAsB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAO9E,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CA8L7B;AAED,KAAK,kBAAkB,GAAG,mBAAmB,GAAG;IAC9C,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,qBAAa,mBAAoB,SAAQ,qBAAqB;IAC5D,MAAM,EAAE,kBAAkB,CAAC;gBAGzB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,kBAAkB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAM1E,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CA0E7B;AAED,eAAO,MAAM,wBAAwB,yBAAwD,CAAC;AAC9F,eAAO,MAAM,sBAAsB,8BAAyD,CAAC;AAC7F,eAAO,MAAM,0BAA0B,8BAIrC,CAAC;AACH,eAAO,MAAM,0BAA0B,8BAAyD,CAAC"}
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/providers/openai.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAM5B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAK1D,OAAO,KAAK,EACV,qBAAqB,EACrB,WAAW,EACX,oBAAoB,EACpB,oBAAoB,EACpB,YAAY,EAEZ,yBAAyB,EACzB,0BAA0B,EAC1B,gBAAgB,EAEjB,MAAM,aAAa,CAAC;AAErB,UAAU,mBAAmB;IAC3B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAAA;KAAE,CAAC;CACrC;AAED,KAAK,uBAAuB,GAAG,mBAAmB,GAAG;IACnD,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,cAAc,EAAE,CAAC;IAC7B,aAAa,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IACnD,KAAK,CAAC,EAAE,UAAU,EAAE,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG;QAAE,IAAI,EAAE,UAAU,CAAC;QAAC,QAAQ,CAAC,EAAE;YAAE,IAAI,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE,CAAC;IAClF,eAAe,CAAC,EAAE;QAAE,IAAI,EAAE,aAAa,CAAA;KAAE,CAAC;IAC1C,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AA4BF,qBAAa,qBAAsB,YAAW,WAAW;IACvD,SAAS,EAAE,MAAM,CAAC;IAElB,MAAM,EAAE,mBAAmB,CAAC;IAC5B,GAAG,CAAC,EAAE,YAAY,CAAC;gBAGjB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,mBAAmB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IASjF,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIlB,eAAe,IAAI,MAAM,GAAG,SAAS;IAMrC,gBAAgB,IAAI,MAAM;IAI1B,SAAS,IAAI,MAAM;IAenB,SAAS,IAAI,MAAM,GAAG,SAAS;IAazB,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CAG7B;AAED,qBAAa,uBAAwB,SAAQ,qBAAqB;IAC1D,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,yBAAyB,CAAC;CA8CzE;AAED,qBAAa,wBAAyB,SAAQ,qBAAqB;IACjE,MAAM,CAAC,wBAAwB;;;;;;;;;SA8B7B;IAEF,MAAM,CAAC,6BAA6B,WAElC;IAEF,MAAM,EAAE,uBAAuB,CAAC;gBAG9B,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;IAY/E,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CA8E7B;AAED,qBAAa,4BAA6B,SAAQ,qBAAqB;IACrE,MAAM,CAAC,kBAAkB;;;;;;QAkDvB;IAEF,MAAM,CAAC,uBAAuB,WAE5B;IAEF,MAAM,EAAE,uBAAuB,CAAC;gBAG9B,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;IAS/E,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CAwF7B;AAuCD,KAAK,sBAAsB,GAAG,mBAAmB,GAAG;IAClD,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC;IAC9D;;;OAGG;IACH,qBAAqB,CAAC,EAAE,MAAM,CAC5B,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,EACjC,CAAC,GAAG,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CACjC,CAAC;IACF,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EACP,MAAM,GACN,MAAM,GACN;QAAE,IAAI,EAAE,UAAU,CAAC;QAAC,QAAQ,CAAC,EAAE;YAAE,IAAI,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE,GACjD;QAAE,IAAI,EAAE,aAAa,CAAA;KAAE,CAAC;CAC7B,CAAC;AAEF,qBAAa,uBAAwB,SAAQ,qBAAqB;IAChE,WAAW,EAAE,MAAM,CAAC;IACpB,eAAe,EAAE,sBAAsB,CAAC;gBAGtC,WAAW,EAAE,MAAM,EACnB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,sBAAsB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAO9E,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CA6L7B;AAED,KAAK,kBAAkB,GAAG,mBAAmB,GAAG;IAC9C,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,qBAAa,mBAAoB,SAAQ,qBAAqB;IAC5D,MAAM,EAAE,kBAAkB,CAAC;gBAGzB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,kBAAkB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAM1E,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CAyE7B;AAED,qBAAa,wBACX,SAAQ,qBACR,YAAW,qBAAqB;IAE1B,iBAAiB,CACrB,UAAU,EAAE,MAAM,EAAE,uDAAuD;IAC3E,iBAAiB,EAAE,MAAM,GACxB,OAAO,CAAC,0BAA0B,CAAC;CAqFvC;AAED,eAAO,MAAM,wBAAwB,yBAAwD,CAAC;AAC9F,eAAO,MAAM,sBAAsB,8BAAyD,CAAC;AAC7F,eAAO,MAAM,0BAA0B,8BAIrC,CAAC;AACH,eAAO,MAAM,0BAA0B,8BAAyD,CAAC;AACjG,eAAO,MAAM,yBAAyB,0BAAyD,CAAC"}

package/dist/src/providers/openai.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.DefaultSuggestionsProvider = exports.DefaultGradingJsonProvider = exports.DefaultGradingProvider = exports.DefaultEmbeddingProvider = exports.OpenAiImageProvider = exports.OpenAiAssistantProvider = exports.OpenAiChatCompletionProvider = exports.OpenAiCompletionProvider = exports.OpenAiEmbeddingProvider = exports.OpenAiGenericProvider = void 0;
+exports.DefaultModerationProvider = exports.DefaultSuggestionsProvider = exports.DefaultGradingJsonProvider = exports.DefaultGradingProvider = exports.DefaultEmbeddingProvider = exports.OpenAiModerationProvider = exports.OpenAiImageProvider = exports.OpenAiAssistantProvider = exports.OpenAiChatCompletionProvider = exports.OpenAiCompletionProvider = exports.OpenAiEmbeddingProvider = exports.OpenAiGenericProvider = void 0;
 const openai_1 = __importDefault(require("openai"));
 const logger_1 = __importDefault(require("../logger"));
 const cache_1 = require("../cache");
@@ -248,15 +248,6 @@ class OpenAiChatCompletionProvider extends OpenAiGenericProvider {
             throw new Error('OpenAI API key is not set. Set the OPENAI_API_KEY environment variable or add `apiKey` to the provider config.');
         }
         const messages = (0, shared_1.parseChatPrompt)(prompt, [{ role: 'user', content: prompt }]);
-        let stop;
-        try {
-            stop = process.env.OPENAI_STOP
-                ? JSON.parse(process.env.OPENAI_STOP)
-                : this.config?.stop || [];
-        }
-        catch (err) {
-            throw new Error(`OPENAI_STOP is not a valid JSON string: ${err}`);
-        }
         const body = {
             model: this.modelName,
             messages: messages,
@@ -413,7 +404,6 @@ class OpenAiAssistantProvider extends OpenAiGenericProvider {
         const openai = new openai_1.default({
             apiKey: this.getApiKey(),
             organization: this.getOrganization(),
-            // Unfortunate, but the OpenAI SDK's implementation of base URL is different from how we treat base URL elsewhere.
             baseURL: this.getApiUrl(),
             maxRetries: 3,
             timeout: shared_1.REQUEST_TIMEOUT_MS,
@@ -581,7 +571,6 @@ class OpenAiImageProvider extends OpenAiGenericProvider {
         const openai = new openai_1.default({
             apiKey: this.getApiKey(),
             organization: this.getOrganization(),
-            // Unfortunate, but the OpenAI SDK's implementation of base URL is different from how we treat base URL elsewhere.
             baseURL: this.getApiUrl(),
             maxRetries: 3,
             timeout: shared_1.REQUEST_TIMEOUT_MS,
@@ -632,6 +621,80 @@ class OpenAiImageProvider extends OpenAiGenericProvider {
     }
 }
 exports.OpenAiImageProvider = OpenAiImageProvider;
+class OpenAiModerationProvider extends OpenAiGenericProvider {
+    async callModerationApi(userPrompt, // userPrompt is not supported by OpenAI moderation API
+    assistantResponse) {
+        if (!this.getApiKey()) {
+            throw new Error('OpenAI API key is not set. Set the OPENAI_API_KEY environment variable or add `apiKey` to the provider config.');
+        }
+        const openai = new openai_1.default({
+            apiKey: this.getApiKey(),
+            organization: this.getOrganization(),
+            baseURL: this.getApiUrl(),
+            maxRetries: 3,
+            timeout: shared_1.REQUEST_TIMEOUT_MS,
+        });
+        let cache;
+        let cacheKey;
+        if ((0, cache_1.isCacheEnabled)()) {
+            cache = await (0, cache_1.getCache)();
+            cacheKey = `openai:${this.modelName}:${JSON.stringify(this.config)}:${userPrompt}:${assistantResponse}`;
+            // Try to get the cached response
+            const cachedResponse = await cache.get(cacheKey);
+            if (cachedResponse) {
+                logger_1.default.debug(`Returning cached response for ${userPrompt}: ${cachedResponse}`);
+                return JSON.parse(cachedResponse);
+            }
+        }
+        logger_1.default.debug(`Calling OpenAI moderation API: prompt [${userPrompt}] assistant [${assistantResponse}]`);
+        let moderation;
+        try {
+            moderation = await openai.moderations.create({
+                model: this.modelName,
+                input: assistantResponse,
+            });
+        }
+        catch (err) {
+            return {
+                error: `API call error: ${String(err)}`,
+            };
+        }
+        logger_1.default.debug(`\tOpenAI moderation API response: ${JSON.stringify(moderation)}`);
+        try {
+            const { results } = moderation;
+            const flags = [];
+            if (!results) {
+                throw new Error('API response error: no results');
+            }
+            if (cache && cacheKey) {
+                await cache.set(cacheKey, JSON.stringify(moderation));
+            }
+            if (results.length === 0) {
+                return { flags };
+            }
+            for (const result of results) {
+                if (result.flagged) {
+                    for (const [category, flagged] of Object.entries(result.categories)) {
+                        if (flagged) {
+                            flags.push({
+                                code: category,
+                                description: category,
+                                confidence: result.category_scores[category],
+                            });
+                        }
+                    }
+                }
+            }
+            return { flags };
+        }
+        catch (err) {
+            return {
+                error: `API response error: ${String(err)}: ${JSON.stringify(moderation)}`,
+            };
+        }
+    }
+}
+exports.OpenAiModerationProvider = OpenAiModerationProvider;
 exports.DefaultEmbeddingProvider = new OpenAiEmbeddingProvider('text-embedding-3-large');
 exports.DefaultGradingProvider = new OpenAiChatCompletionProvider('gpt-4-0125-preview');
 exports.DefaultGradingJsonProvider = new OpenAiChatCompletionProvider('gpt-4-0125-preview', {
@@ -640,4 +703,5 @@ exports.DefaultGradingJsonProvider = new OpenAiChatCompletionProvider('gpt-4-012
     },
 });
 exports.DefaultSuggestionsProvider = new OpenAiChatCompletionProvider('gpt-4-0125-preview');
+exports.DefaultModerationProvider = new OpenAiModerationProvider('text-moderation-latest');
 //# sourceMappingURL=openai.js.map