promptfoo 0.47.0 → 0.49.0

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

@@ -1,3 +1,4 @@
+import { AssistantCreationOptions, AssistantsClient, FunctionDefinition } from '@azure/openai-assistants';
 import type { ApiProvider, CallApiContextParams, CallApiOptionsParams, EnvOverrides, ProviderEmbeddingResponse, ProviderResponse } from '../types';
 interface AzureOpenAiCompletionOptions {
     azureClientId?: string;
@@ -11,6 +12,7 @@ interface AzureOpenAiCompletionOptions {
     apiBaseUrl?: string;
     apiKey?: string;
     apiVersion?: string;
+    max_tokens?: number;
     temperature?: number;
     top_p?: number;
     frequency_penalty?: number;
@@ -42,6 +44,7 @@ interface AzureOpenAiCompletionOptions {
         type: 'json_object';
     };
     stop?: string[];
+    seed?: number;
     passthrough?: object;
 }
 declare class AzureOpenAiGenericProvider implements ApiProvider {
@@ -70,5 +73,25 @@ export declare class AzureOpenAiCompletionProvider extends AzureOpenAiGenericPro
 export declare class AzureOpenAiChatCompletionProvider extends AzureOpenAiGenericProvider {
     callApi(prompt: string, context?: CallApiContextParams, callApiOptions?: CallApiOptionsParams): Promise<ProviderResponse>;
 }
+type AzureOpenAiAssistantOptions = AzureOpenAiCompletionOptions & Partial<AssistantCreationOptions> & {
+    /**
+     * If set, automatically call these functions when the assistant activates
+     * these function tools.
+     */
+    functionToolCallbacks?: Record<FunctionDefinition['name'], (arg: string) => Promise<string>>;
+};
+export declare class AzureOpenAiAssistantProvider extends AzureOpenAiGenericProvider {
+    assistantConfig: AzureOpenAiAssistantOptions;
+    assistantsClient: AssistantsClient | undefined;
+    private initializationPromise;
+    constructor(deploymentName: string, options?: {
+        config?: AzureOpenAiAssistantOptions;
+        id?: string;
+        env?: EnvOverrides;
+    });
+    initialize(): Promise<void>;
+    ensureInitialized(): Promise<void>;
+    callApi(prompt: string, context?: CallApiContextParams, callApiOptions?: CallApiOptionsParams): Promise<ProviderResponse>;
+}
 export {};
 //# sourceMappingURL=azureopenai.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"azureopenai.d.ts","sourceRoot":"","sources":["../../../src/providers/azureopenai.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EACV,WAAW,EACX,oBAAoB,EACpB,oBAAoB,EACpB,YAAY,EACZ,yBAAyB,EACzB,gBAAgB,EACjB,MAAM,UAAU,CAAC;AAElB,UAAU,4BAA4B;IAEpC,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,eAAe,CAAC,EAAE,MAAM,CAAC;IAGzB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,WAAW,CAAC,EAAE,GAAG,CAAC;IAGlB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IAGpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,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;QACV,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,UAAU,EAAE,GAAG,CAAC;KACjB,EAAE,CAAC;IACJ,aAAa,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IACnD,KAAK,CAAC,EAAE;QACN,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAC;YACT,IAAI,EAAE,MAAM,CAAC;YACb,WAAW,CAAC,EAAE,MAAM,CAAC;YACrB,UAAU,EAAE,GAAG,CAAC;SACf,CAAA;KACF,EAAE,CAAC;IACJ,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;IAEhB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,cAAM,0BAA2B,YAAW,WAAW;IACrD,cAAc,EAAE,MAAM,CAAC;IACvB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB,MAAM,EAAE,4BAA4B,CAAC;gBAGnC,cAAc,EAAE,MAAM,EACtB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,4BAA4B,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAe1F,aAAa,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,IAAI,OAAO,CAAC,MAAM,CAAC;IA+BlC,aAAa,IAAI,MAAM;IAIvB,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAKZ,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CAG7B;AAED,qBAAa,4BAA6B,SAAQ,0BAA0B;IACpE,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,yBAAyB,CAAC;CA0EzE;AAED,qBAAa,6BAA8B,SAAQ,0BAA0B;IACrE,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CA8E7B;AAED,qBAAa,iCAAkC,SAAQ,0BAA0B;IACzE,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CAyG7B"}
+{"version":3,"file":"azureopenai.d.ts","sourceRoot":"","sources":["../../../src/providers/azureopenai.ts"],"names":[],"mappings":"AACA,OAAO,EACL,wBAAwB,EACxB,gBAAgB,EAEhB,kBAAkB,EAGnB,MAAM,0BAA0B,CAAC;AAMlC,OAAO,KAAK,EACV,WAAW,EACX,oBAAoB,EACpB,oBAAoB,EACpB,YAAY,EACZ,yBAAyB,EACzB,gBAAgB,EACjB,MAAM,UAAU,CAAC;AAElB,UAAU,4BAA4B;IAEpC,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,eAAe,CAAC,EAAE,MAAM,CAAC;IAGzB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,WAAW,CAAC,EAAE,GAAG,CAAC;IAGlB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IAGpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,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;QACV,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,UAAU,EAAE,GAAG,CAAC;KACjB,EAAE,CAAC;IACJ,aAAa,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IACnD,KAAK,CAAC,EAAE;QACN,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAE;YACR,IAAI,EAAE,MAAM,CAAC;YACb,WAAW,CAAC,EAAE,MAAM,CAAC;YACrB,UAAU,EAAE,GAAG,CAAC;SACjB,CAAC;KACH,EAAE,CAAC;IACJ,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;IAEd,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,cAAM,0BAA2B,YAAW,WAAW;IACrD,cAAc,EAAE,MAAM,CAAC;IACvB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB,MAAM,EAAE,4BAA4B,CAAC;gBAGnC,cAAc,EAAE,MAAM,EACtB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,4BAA4B,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAe1F,aAAa,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,IAAI,OAAO,CAAC,MAAM,CAAC;IA+BlC,aAAa,IAAI,MAAM;IAIvB,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAKZ,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CAG7B;AAED,qBAAa,4BAA6B,SAAQ,0BAA0B;IACpE,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,yBAAyB,CAAC;CA0EzE;AAED,qBAAa,6BAA8B,SAAQ,0BAA0B;IACrE,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CA+E7B;AAED,qBAAa,iCAAkC,SAAQ,0BAA0B;IACzE,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CA0G7B;AAED,KAAK,2BAA2B,GAAG,4BAA4B,GAC7D,OAAO,CAAC,wBAAwB,CAAC,GAAG;IAClC;;;OAGG;IACH,qBAAqB,CAAC,EAAE,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;CAC9F,CAAC;AAGJ,qBAAa,4BAA6B,SAAQ,0BAA0B;IAC1E,eAAe,EAAE,2BAA2B,CAAC;IAC7C,gBAAgB,EAAE,gBAAgB,GAAG,SAAS,CAAC;IAC/C,OAAO,CAAC,qBAAqB,CAA8B;gBAGzD,cAAc,EAAE,MAAM,EACtB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,2BAA2B,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAQnF,UAAU;IAaV,iBAAiB;IAMjB,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,cAAc,CAAC,EAAE,oBAAoB,GACpC,OAAO,CAAC,gBAAgB,CAAC;CA+I7B"}

package/dist/src/providers/azureopenai.js

@@ -26,7 +26,9 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.AzureOpenAiChatCompletionProvider = exports.AzureOpenAiCompletionProvider = exports.AzureOpenAiEmbeddingProvider = void 0;
+exports.AzureOpenAiAssistantProvider = exports.AzureOpenAiChatCompletionProvider = exports.AzureOpenAiCompletionProvider = exports.AzureOpenAiEmbeddingProvider = void 0;
+const tiny_invariant_1 = __importDefault(require("tiny-invariant"));
+const openai_assistants_1 = require("@azure/openai-assistants");
 const logger_1 = __importDefault(require("../logger"));
 const shared_1 = require("./shared");
 const cache_1 = require("../cache");
@@ -166,12 +168,13 @@ class AzureOpenAiCompletionProvider extends AzureOpenAiGenericProvider {
         const body = {
             model: this.deploymentName,
             prompt,
-            max_tokens: parseInt(process.env.OPENAI_MAX_TOKENS || '1024'),
+            max_tokens: this.config.max_tokens ?? parseInt(process.env.OPENAI_MAX_TOKENS || '1024'),
             temperature: this.config.temperature ?? parseFloat(process.env.OPENAI_TEMPERATURE || '0'),
             top_p: this.config.top_p ?? parseFloat(process.env.OPENAI_TOP_P || '1'),
             presence_penalty: this.config.presence_penalty ?? parseFloat(process.env.OPENAI_PRESENCE_PENALTY || '0'),
             frequency_penalty: this.config.frequency_penalty ?? parseFloat(process.env.OPENAI_FREQUENCY_PENALTY || '0'),
             best_of: this.config.best_of ?? parseInt(process.env.OPENAI_BEST_OF || '1'),
+            ...(this.config.seed !== undefined ? { seed: this.config.seed } : {}),
             ...(this.config.deployment_id ? { deployment_id: this.config.deployment_id } : {}),
             ...(this.config.dataSources ? { dataSources: this.config.dataSources } : {}),
             ...(this.config.response_format ? { response_format: this.config.response_format } : {}),
@@ -237,13 +240,14 @@ class AzureOpenAiChatCompletionProvider extends AzureOpenAiGenericProvider {
         const body = {
             model: this.deploymentName,
             messages: messages,
-            max_tokens: parseInt(process.env.OPENAI_MAX_TOKENS || '1024'),
+            max_tokens: this.config.max_tokens ?? parseInt(process.env.OPENAI_MAX_TOKENS || '1024'),
             temperature: this.config.temperature ?? parseFloat(process.env.OPENAI_TEMPERATURE || '0'),
             top_p: this.config.top_p ?? parseFloat(process.env.OPENAI_TOP_P || '1'),
             presence_penalty: this.config.presence_penalty ?? parseFloat(process.env.OPENAI_PRESENCE_PENALTY || '0'),
             frequency_penalty: this.config.frequency_penalty ?? parseFloat(process.env.OPENAI_FREQUENCY_PENALTY || '0'),
             functions: this.config.functions || undefined,
             function_call: this.config.function_call || undefined,
+            ...(this.config.seed !== undefined ? { seed: this.config.seed } : {}),
             ...(this.config.tools ? { tools: this.config.tools } : {}),
             ...(this.config.tool_choice ? { tool_choice: this.config.tool_choice } : {}),
             ...(this.config.deployment_id ? { deployment_id: this.config.deployment_id } : {}),
@@ -306,4 +310,138 @@ class AzureOpenAiChatCompletionProvider extends AzureOpenAiGenericProvider {
     }
 }
 exports.AzureOpenAiChatCompletionProvider = AzureOpenAiChatCompletionProvider;
+// See https://learn.microsoft.com/en-us/javascript/api/overview/azure/openai-assistants-readme?view=azure-node-preview
+class AzureOpenAiAssistantProvider extends AzureOpenAiGenericProvider {
+    constructor(deploymentName, options = {}) {
+        super(deploymentName, options);
+        this.initializationPromise = null;
+        this.assistantConfig = options.config || {};
+        this.initializationPromise = this.initialize();
+    }
+    async initialize() {
+        const apiKey = await this.getApiKey();
+        if (!apiKey) {
+            throw new Error('Azure OpenAI API key must be set');
+        }
+        this.assistantsClient = new openai_assistants_1.AssistantsClient(this.getApiBaseUrl(), new openai_assistants_1.AzureKeyCredential(apiKey));
+        this.initializationPromise = null;
+    }
+    async ensureInitialized() {
+        if (this.initializationPromise) {
+            await this.initializationPromise;
+        }
+    }
+    async callApi(prompt, context, callApiOptions) {
+        await this.ensureInitialized();
+        (0, tiny_invariant_1.default)(this.assistantsClient, 'Assistants client not initialized');
+        if (!this.getApiBaseUrl()) {
+            throw new Error('Azure OpenAI API host must be set');
+        }
+        const assistantId = this.deploymentName;
+        const assistantThread = await this.assistantsClient.createThread();
+        await this.assistantsClient.createMessage(assistantThread.id, 'user', prompt);
+        let run = await this.assistantsClient.createRun(assistantThread.id, {
+            assistantId,
+        });
+        logger_1.default.debug(`\tAzure thread run API response: ${JSON.stringify(run)}`);
+        while (run.status === 'in_progress' ||
+            run.status === 'queued' ||
+            run.status === 'requires_action') {
+            if (run.status === 'requires_action') {
+                const requiredAction = run.requiredAction;
+                (0, tiny_invariant_1.default)(requiredAction, 'Run requires action but no action is provided');
+                if (requiredAction === null || requiredAction.type !== 'submit_tool_outputs') {
+                    break;
+                }
+                const functionCallsWithCallbacks = requiredAction.submitToolOutputs?.toolCalls.filter((toolCall) => {
+                    return (toolCall.type === 'function' &&
+                        toolCall.function.name in (this.assistantConfig.functionToolCallbacks ?? {}));
+                });
+                if (!functionCallsWithCallbacks || functionCallsWithCallbacks.length === 0) {
+                    break;
+                }
+                logger_1.default.debug(`Calling functionToolCallbacks for functions: ${functionCallsWithCallbacks.map(({ function: { name } }) => name)}`);
+                const toolOutputs = await Promise.all(functionCallsWithCallbacks.map(async (toolCall) => {
+                    logger_1.default.debug(`Calling functionToolCallbacks[${toolCall.function.name}]('${toolCall.function.arguments}')`);
+                    const result = await this.assistantConfig.functionToolCallbacks[toolCall.function.name](toolCall.function.arguments);
+                    return {
+                        tool_call_id: toolCall.id,
+                        output: result,
+                    };
+                }));
+                logger_1.default.debug(`Calling Azure API, submitting tool outputs for ${run.threadId}: ${JSON.stringify(toolOutputs)}`);
+                run = await this.assistantsClient.submitToolOutputsToRun(run.threadId, run.id, toolOutputs);
+            }
+            await new Promise((resolve) => setTimeout(resolve, 1000));
+            logger_1.default.debug(`Calling Azure API, getting thread run ${run.id} status`);
+            run = await this.assistantsClient.getRun(run.threadId, run.id);
+            logger_1.default.debug(`\tAzure thread run API response: ${JSON.stringify(run)}`);
+        }
+        if (run.status !== 'completed' && run.status !== 'requires_action') {
+            if (run.lastError) {
+                return {
+                    error: `Thread run failed: ${run.lastError.message}`,
+                };
+            }
+            return {
+                error: `Thread run failed: ${run.status}`,
+            };
+        }
+        logger_1.default.debug(`Calling Azure API, getting thread run steps for ${run.threadId}`);
+        const steps = await this.assistantsClient.listRunSteps(run.threadId, run.id, { order: 'asc' });
+        logger_1.default.debug(`\tAzure thread run steps API response: ${JSON.stringify(steps)}`);
+        const outputBlocks = [];
+        for (const step of steps.data) {
+            if (step.type === 'message_creation') {
+                logger_1.default.debug(`Calling Azure API, getting message ${step.id}`);
+                const stepDetails = step.stepDetails;
+                // Bug in the API: the field is currently `message_id` even though it's documented as `messageId`
+                const messageId = stepDetails.messageCreation.message_id || stepDetails.messageCreation.messageId;
+                const message = await this.assistantsClient.getMessage(run.threadId, messageId);
+                logger_1.default.debug(`\tAzure thread run step message API response: ${JSON.stringify(message)}`);
+                const content = message.content
+                    .map((content) => content.type === 'text' ? content.text.value : `<${content.type} output>`)
+                    .join('\n');
+                outputBlocks.push(`[${(0, shared_1.toTitleCase)(message.role)}] ${content}`);
+            }
+            else if (step.type === 'tool_calls') {
+                for (const toolCall of step.stepDetails.toolCalls) {
+                    if (toolCall.type === 'function') {
+                        outputBlocks.push(`[Call function ${toolCall.function.name} with arguments ${toolCall.function.arguments}]`);
+                        outputBlocks.push(`[Function output: ${toolCall.function.output}]`);
+                    }
+                    else if (toolCall.type === 'retrieval') {
+                        outputBlocks.push(`[Ran retrieval]`);
+                    }
+                    else if (toolCall.type === 'code_interpreter') {
+                        const output = toolCall.codeInterpreter.outputs
+                            .map((output) => (output.type === 'logs' ? output.logs : `<${output.type} output>`))
+                            .join('\n');
+                        outputBlocks.push(`[Code interpreter input]`);
+                        outputBlocks.push(toolCall.codeInterpreter.input);
+                        outputBlocks.push(`[Code interpreter output]`);
+                        outputBlocks.push(output);
+                    }
+                    else {
+                        outputBlocks.push(`[Unknown tool call type: ${toolCall.type}]`);
+                    }
+                }
+            }
+            else {
+                outputBlocks.push(`[Unknown step type: ${step.type}]`);
+            }
+        }
+        return {
+            output: outputBlocks.join('\n\n').trim(),
+            /*
+            tokenUsage: {
+              total: data.usage.total_tokens,
+              prompt: data.usage.prompt_tokens,
+              completion: data.usage.completion_tokens,
+            },
+            */
+        };
+    }
+}
+exports.AzureOpenAiAssistantProvider = AzureOpenAiAssistantProvider;
 //# sourceMappingURL=azureopenai.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"azureopenai.js","sourceRoot":"","sources":["../../../src/providers/azureopenai.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,uDAA+B;AAC/B,qCAA+D;AAC/D,oCAA0C;AAwD1C,MAAM,0BAA0B;IAO9B,YACE,cAAsB,EACtB,UAAsF,EAAE;QAExF,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC;QAEpC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QAErC,IAAI,CAAC,OAAO;YACV,MAAM,EAAE,OAAO,IAAI,GAAG,EAAE,qBAAqB,IAAI,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;QACrF,IAAI,CAAC,UAAU;YACb,MAAM,EAAE,UAAU,IAAI,GAAG,EAAE,yBAAyB,IAAI,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC;QAEhG,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;IAGD,KAAK,CAAC,SAAS;QACb,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE;YACvB,IACE,IAAI,CAAC,MAAM,EAAE,iBAAiB;gBAC9B,IAAI,CAAC,MAAM,EAAE,aAAa;gBAC1B,IAAI,CAAC,MAAM,EAAE,aAAa,EAC1B;gBACA,MAAM,EAAE,sBAAsB,EAAE,GAAG,wDAAa,iBAAiB,GAAC,CAAC;gBACnE,MAAM,UAAU,GAAG,IAAI,sBAAsB,CAC3C,IAAI,CAAC,MAAM,CAAC,aAAa,EACzB,IAAI,CAAC,MAAM,CAAC,aAAa,EACzB,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAC7B;oBACE,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,kBAAkB,IAAI,mCAAmC;iBACrF,CACF,CAAC;gBACF,IAAI,CAAC,aAAa,GAAG,CACnB,MAAM,UAAU,CAAC,QAAQ,CACvB,IAAI,CAAC,MAAM,CAAC,eAAe,IAAI,8CAA8C,CAC9E,CACF,CAAC,KAAK,CAAC;aACT;iBAAM;gBACL,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,MAAM,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;gBAC7E,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE;oBACvB,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;iBACrD;aACF;SACF;QACD,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,UAAU,IAAI,WAAW,IAAI,CAAC,OAAO,EAAE,CAAC;IACtD,CAAC;IAED,EAAE;QACA,OAAO,eAAe,IAAI,CAAC,cAAc,EAAE,CAAC;IAC9C,CAAC;IAED,QAAQ;QACN,OAAO,0BAA0B,IAAI,CAAC,cAAc,GAAG,CAAC;IAC1D,CAAC;IAED,yDAAyD;IACzD,KAAK,CAAC,OAAO,CACX,MAAc,EACd,OAA8B,EAC9B,cAAqC;QAErC,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF;AAED,MAAa,4BAA6B,SAAQ,0BAA0B;IAC1E,KAAK,CAAC,gBAAgB,CAAC,IAAY;QACjC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;SAC/E;QACD,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE;YACzB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;SACtD;QAED,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,IAAI;YACX,KAAK,EAAE,IAAI,CAAC,cAAc;SAC3B,CAAC;QACF,IAAI,IAAI,EACN,MAAM,GAAG,KAAK,CAAC;QACjB,IAAI;YACF,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,IAAA,sBAAc,EACvC,GAAG,IAAI,CAAC,aAAa,EAAE,uBAAuB,IAAI,CAAC,cAAc,2BAC/D,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,oBAC5B,EAAE,EACF;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,SAAS,EAAE,MAAM;iBAClB;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,EACD,2BAAkB,CACnB,CAAmB,CAAC,CAAC;SACvB;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;gBACvC,UAAU,EAAE;oBACV,KAAK,EAAE,CAAC;oBACR,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;aACF,CAAC;SACH;QACD,gBAAM,CAAC,KAAK,CAAC,6CAA6C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAElF,IAAI;YACF,MAAM,SAAS,GAAG,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC;YAC7C,IAAI,CAAC,SAAS,EAAE;gBACd,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;aAC1C;YACD,MAAM,GAAG,GAAG;gBACV,SAAS;gBACT,UAAU,EAAE,MAAM;oBAChB,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE;oBACrE,CAAC,CAAC;wBACE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;wBAC9B,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa;wBAChC,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB;qBACzC;aACN,CAAC;YACF,OAAO,GAAG,CAAC;SACZ;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,uBAAuB,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;gBACpE,UAAU,EAAE,MAAM;oBAChB,CAAC,CAAC;wBACE,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;wBAC/B,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;qBAC/B;oBACH,CAAC,CAAC;wBACE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY;wBAChC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,aAAa;wBAClC,UAAU,EAAE,IAAI,EAAE,KAAK,EAAE,iBAAiB;qBAC3C;aACN,CAAC;SACH;IACH,CAAC;CACF;AA3ED,oEA2EC;AAED,MAAa,6BAA8B,SAAQ,0BAA0B;IAC3E,KAAK,CAAC,OAAO,CACX,MAAc,EACd,OAA8B,EAC9B,cAAqC;QAErC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CACb,8HAA8H,CAC/H,CAAC;SACH;QACD,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE;YACzB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;SACtD;QAED,IAAI,IAAY,CAAC;QACjB,IAAI;YACF,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW;gBAC5B,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;gBACrC,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,IAAI,CAAC,YAAY,EAAE,eAAe,CAAC,CAAC;SAC1D;QAAC,OAAO,GAAG,EAAE;YACZ,MAAM,IAAI,KAAK,CAAC,2CAA2C,GAAG,EAAE,CAAC,CAAC;SACnE;QACD,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,IAAI,CAAC,cAAc;YAC1B,MAAM;YACN,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,MAAM,CAAC;YAC7D,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,GAAG,CAAC;YACzF,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC;YACvE,gBAAgB,EACd,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,GAAG,CAAC;YACxF,iBAAiB,EACf,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,IAAI,GAAG,CAAC;YAC1F,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,GAAG,CAAC;YAC3E,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAClF,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxF,GAAG,CAAC,cAAc,EAAE,eAAe,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,cAAc,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxF,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzB,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;SACnC,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClE,IAAI,IAAI,EACN,MAAM,GAAG,KAAK,CAAC;QACjB,IAAI;YACF,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,IAAA,sBAAc,EACvC,GAAG,IAAI,CAAC,aAAa,EAAE,uBACrB,IAAI,CAAC,cACP,4BAA4B,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,oBAAoB,EAAE,EAC5E;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,SAAS,EAAE,MAAM;iBAClB;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,EACD,2BAAkB,CACnB,CAAmB,CAAC,CAAC;SACvB;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;SACH;QACD,gBAAM,CAAC,KAAK,CAAC,gCAAgC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACrE,IAAI;YACF,OAAO;gBACL,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI;gBAC5B,UAAU,EAAE,MAAM;oBAChB,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE;oBACrE,CAAC,CAAC;wBACE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;wBAC9B,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa;wBAChC,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB;qBACzC;aACN,CAAC;SACH;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;CACF;AAnFD,sEAmFC;AAED,MAAa,iCAAkC,SAAQ,0BAA0B;IAC/E,KAAK,CAAC,OAAO,CACX,MAAc,EACd,OAA8B,EAC9B,cAAqC;QAErC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CACb,4HAA4H,CAC7H,CAAC;SACH;QACD,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE;YACzB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;SACtD;QAED,MAAM,QAAQ,GAAG,IAAA,wBAAe,EAAC,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QAE9E,IAAI,IAAY,CAAC;QACjB,IAAI;YACF,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC;SAC1F;QAAC,OAAO,GAAG,EAAE;YACZ,MAAM,IAAI,KAAK,CAAC,2CAA2C,GAAG,EAAE,CAAC,CAAC;SACnE;QACD,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,IAAI,CAAC,cAAc;YAC1B,QAAQ,EAAE,QAAQ;YAClB,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,MAAM,CAAC;YAC7D,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,GAAG,CAAC;YACzF,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC;YACvE,gBAAgB,EACd,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,GAAG,CAAC;YACxF,iBAAiB,EACf,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,IAAI,GAAG,CAAC;YAC1F,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,SAAS;YAC7C,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,SAAS;YACrD,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1D,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAClF,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxF,GAAG,CAAC,cAAc,EAAE,eAAe,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,cAAc,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxF,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACvD,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;SACnC,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAElE,IAAI,IAAI,EACN,MAAM,GAAG,KAAK,CAAC;QACjB,IAAI;YACF,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;gBACjC,CAAC,CAAC,GAAG,IAAI,CAAC,aAAa,EAAE,uBACrB,IAAI,CAAC,cACP,4CACE,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,oBAC5B,EAAE;gBACJ,CAAC,CAAC,GAAG,IAAI,CAAC,aAAa,EAAE,uBACrB,IAAI,CAAC,cACP,iCAAiC,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,oBAAoB,EAAE,CAAC;YAEtF,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,SAAS,EAAE,MAAM;iBAClB;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,EACD,2BAAkB,CACnB,CAAmB,CAAC,CAAC;SACvB;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;SACH;QAED,gBAAM,CAAC,KAAK,CAAC,gCAAgC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACrE,IAAI;YACF,IAAI,IAAI,CAAC,KAAK,EAAE;gBACd,OAAO;oBACL,KAAK,EAAE,uBAAuB,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE;iBACtE,CAAC;aACH;YACD,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;gBACrC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAqB,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,WAAW,CAAC;gBACpF,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAC5B,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;YACjF,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,CACrD,CAAC,UAA8C,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,CACvE,CAAC;YACF,OAAO;gBACL,MAAM;gBACN,UAAU,EAAE,MAAM;oBAChB,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE;oBACxE,CAAC,CAAC;wBACE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,KAAK,EAAE,aAAa;wBACjC,UAAU,EAAE,IAAI,CAAC,KAAK,EAAE,iBAAiB;qBAC1C;gBACL,MAAM;gBACN,QAAQ;aACT,CAAC;SACH;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;CACF;AA9GD,8EA8GC"}
+{"version":3,"file":"azureopenai.js","sourceRoot":"","sources":["../../../src/providers/azureopenai.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,oEAAuC;AACvC,gEAOkC;AAElC,uDAA+B;AAC/B,qCAA4E;AAC5E,oCAA0C;AA0D1C,MAAM,0BAA0B;IAO9B,YACE,cAAsB,EACtB,UAAsF,EAAE;QAExF,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC;QAEpC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QAErC,IAAI,CAAC,OAAO;YACV,MAAM,EAAE,OAAO,IAAI,GAAG,EAAE,qBAAqB,IAAI,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;QACrF,IAAI,CAAC,UAAU;YACb,MAAM,EAAE,UAAU,IAAI,GAAG,EAAE,yBAAyB,IAAI,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC;QAEhG,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;IAGD,KAAK,CAAC,SAAS;QACb,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE;YACvB,IACE,IAAI,CAAC,MAAM,EAAE,iBAAiB;gBAC9B,IAAI,CAAC,MAAM,EAAE,aAAa;gBAC1B,IAAI,CAAC,MAAM,EAAE,aAAa,EAC1B;gBACA,MAAM,EAAE,sBAAsB,EAAE,GAAG,wDAAa,iBAAiB,GAAC,CAAC;gBACnE,MAAM,UAAU,GAAG,IAAI,sBAAsB,CAC3C,IAAI,CAAC,MAAM,CAAC,aAAa,EACzB,IAAI,CAAC,MAAM,CAAC,aAAa,EACzB,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAC7B;oBACE,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,kBAAkB,IAAI,mCAAmC;iBACrF,CACF,CAAC;gBACF,IAAI,CAAC,aAAa,GAAG,CACnB,MAAM,UAAU,CAAC,QAAQ,CACvB,IAAI,CAAC,MAAM,CAAC,eAAe,IAAI,8CAA8C,CAC9E,CACF,CAAC,KAAK,CAAC;aACT;iBAAM;gBACL,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,MAAM,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;gBAC7E,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE;oBACvB,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;iBACrD;aACF;SACF;QACD,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,UAAU,IAAI,WAAW,IAAI,CAAC,OAAO,EAAE,CAAC;IACtD,CAAC;IAED,EAAE;QACA,OAAO,eAAe,IAAI,CAAC,cAAc,EAAE,CAAC;IAC9C,CAAC;IAED,QAAQ;QACN,OAAO,0BAA0B,IAAI,CAAC,cAAc,GAAG,CAAC;IAC1D,CAAC;IAED,yDAAyD;IACzD,KAAK,CAAC,OAAO,CACX,MAAc,EACd,OAA8B,EAC9B,cAAqC;QAErC,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF;AAED,MAAa,4BAA6B,SAAQ,0BAA0B;IAC1E,KAAK,CAAC,gBAAgB,CAAC,IAAY;QACjC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;SAC/E;QACD,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE;YACzB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;SACtD;QAED,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,IAAI;YACX,KAAK,EAAE,IAAI,CAAC,cAAc;SAC3B,CAAC;QACF,IAAI,IAAI,EACN,MAAM,GAAG,KAAK,CAAC;QACjB,IAAI;YACF,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,IAAA,sBAAc,EACvC,GAAG,IAAI,CAAC,aAAa,EAAE,uBAAuB,IAAI,CAAC,cAAc,2BAC/D,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,oBAC5B,EAAE,EACF;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,SAAS,EAAE,MAAM;iBAClB;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,EACD,2BAAkB,CACnB,CAAmB,CAAC,CAAC;SACvB;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;gBACvC,UAAU,EAAE;oBACV,KAAK,EAAE,CAAC;oBACR,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;aACF,CAAC;SACH;QACD,gBAAM,CAAC,KAAK,CAAC,6CAA6C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAElF,IAAI;YACF,MAAM,SAAS,GAAG,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC;YAC7C,IAAI,CAAC,SAAS,EAAE;gBACd,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;aAC1C;YACD,MAAM,GAAG,GAAG;gBACV,SAAS;gBACT,UAAU,EAAE,MAAM;oBAChB,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE;oBACrE,CAAC,CAAC;wBACE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;wBAC9B,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa;wBAChC,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB;qBACzC;aACN,CAAC;YACF,OAAO,GAAG,CAAC;SACZ;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,uBAAuB,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;gBACpE,UAAU,EAAE,MAAM;oBAChB,CAAC,CAAC;wBACE,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;wBAC/B,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;qBAC/B;oBACH,CAAC,CAAC;wBACE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY;wBAChC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,aAAa;wBAClC,UAAU,EAAE,IAAI,EAAE,KAAK,EAAE,iBAAiB;qBAC3C;aACN,CAAC;SACH;IACH,CAAC;CACF;AA3ED,oEA2EC;AAED,MAAa,6BAA8B,SAAQ,0BAA0B;IAC3E,KAAK,CAAC,OAAO,CACX,MAAc,EACd,OAA8B,EAC9B,cAAqC;QAErC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CACb,8HAA8H,CAC/H,CAAC;SACH;QACD,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE;YACzB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;SACtD;QAED,IAAI,IAAY,CAAC;QACjB,IAAI;YACF,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW;gBAC5B,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;gBACrC,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,IAAI,CAAC,YAAY,EAAE,eAAe,CAAC,CAAC;SAC1D;QAAC,OAAO,GAAG,EAAE;YACZ,MAAM,IAAI,KAAK,CAAC,2CAA2C,GAAG,EAAE,CAAC,CAAC;SACnE;QACD,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,IAAI,CAAC,cAAc;YAC1B,MAAM;YACN,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,MAAM,CAAC;YACvF,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,GAAG,CAAC;YACzF,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC;YACvE,gBAAgB,EACd,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,GAAG,CAAC;YACxF,iBAAiB,EACf,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,IAAI,GAAG,CAAC;YAC1F,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,GAAG,CAAC;YAC3E,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACrE,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAClF,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxF,GAAG,CAAC,cAAc,EAAE,eAAe,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,cAAc,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxF,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzB,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;SACnC,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClE,IAAI,IAAI,EACN,MAAM,GAAG,KAAK,CAAC;QACjB,IAAI;YACF,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,IAAA,sBAAc,EACvC,GAAG,IAAI,CAAC,aAAa,EAAE,uBACrB,IAAI,CAAC,cACP,4BAA4B,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,oBAAoB,EAAE,EAC5E;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,SAAS,EAAE,MAAM;iBAClB;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,EACD,2BAAkB,CACnB,CAAmB,CAAC,CAAC;SACvB;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;SACH;QACD,gBAAM,CAAC,KAAK,CAAC,gCAAgC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACrE,IAAI;YACF,OAAO;gBACL,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI;gBAC5B,UAAU,EAAE,MAAM;oBAChB,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE;oBACrE,CAAC,CAAC;wBACE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;wBAC9B,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa;wBAChC,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB;qBACzC;aACN,CAAC;SACH;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;CACF;AApFD,sEAoFC;AAED,MAAa,iCAAkC,SAAQ,0BAA0B;IAC/E,KAAK,CAAC,OAAO,CACX,MAAc,EACd,OAA8B,EAC9B,cAAqC;QAErC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CACb,4HAA4H,CAC7H,CAAC;SACH;QACD,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE;YACzB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;SACtD;QAED,MAAM,QAAQ,GAAG,IAAA,wBAAe,EAAC,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QAE9E,IAAI,IAAY,CAAC;QACjB,IAAI;YACF,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC;SAC1F;QAAC,OAAO,GAAG,EAAE;YACZ,MAAM,IAAI,KAAK,CAAC,2CAA2C,GAAG,EAAE,CAAC,CAAC;SACnE;QACD,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,IAAI,CAAC,cAAc;YAC1B,QAAQ,EAAE,QAAQ;YAClB,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,MAAM,CAAC;YACvF,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,GAAG,CAAC;YACzF,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC;YACvE,gBAAgB,EACd,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,GAAG,CAAC;YACxF,iBAAiB,EACf,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,IAAI,GAAG,CAAC;YAC1F,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,SAAS;YAC7C,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,SAAS;YACrD,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACrE,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1D,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAClF,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxF,GAAG,CAAC,cAAc,EAAE,eAAe,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,cAAc,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxF,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACvD,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;SACnC,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAElE,IAAI,IAAI,EACN,MAAM,GAAG,KAAK,CAAC;QACjB,IAAI;YACF,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;gBACjC,CAAC,CAAC,GAAG,IAAI,CAAC,aAAa,EAAE,uBACrB,IAAI,CAAC,cACP,4CACE,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,oBAC5B,EAAE;gBACJ,CAAC,CAAC,GAAG,IAAI,CAAC,aAAa,EAAE,uBACrB,IAAI,CAAC,cACP,iCAAiC,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,oBAAoB,EAAE,CAAC;YAEtF,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,SAAS,EAAE,MAAM;iBAClB;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,EACD,2BAAkB,CACnB,CAAmB,CAAC,CAAC;SACvB;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;SACH;QAED,gBAAM,CAAC,KAAK,CAAC,gCAAgC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACrE,IAAI;YACF,IAAI,IAAI,CAAC,KAAK,EAAE;gBACd,OAAO;oBACL,KAAK,EAAE,uBAAuB,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE;iBACtE,CAAC;aACH;YACD,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;gBACrC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAqB,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,WAAW,CAAC;gBACpF,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAC5B,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;YACjF,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,CACrD,CAAC,UAA8C,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,CACvE,CAAC;YACF,OAAO;gBACL,MAAM;gBACN,UAAU,EAAE,MAAM;oBAChB,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE;oBACxE,CAAC,CAAC;wBACE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,KAAK,EAAE,aAAa;wBACjC,UAAU,EAAE,IAAI,CAAC,KAAK,EAAE,iBAAiB;qBAC1C;gBACL,MAAM;gBACN,QAAQ;aACT,CAAC;SACH;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;CACF;AA/GD,8EA+GC;AAWD,uHAAuH;AACvH,MAAa,4BAA6B,SAAQ,0BAA0B;IAK1E,YACE,cAAsB,EACtB,UAAqF,EAAE;QAEvF,KAAK,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QANzB,0BAAqB,GAAyB,IAAI,CAAC;QAOzD,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC;QAE5C,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;SACrD;QAED,IAAI,CAAC,gBAAgB,GAAG,IAAI,oCAAgB,CAC1C,IAAI,CAAC,aAAa,EAAE,EACpB,IAAI,sCAAkB,CAAC,MAAM,CAAC,CAC/B,CAAC;QACF,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,iBAAiB;QACrB,IAAI,IAAI,CAAC,qBAAqB,EAAE;YAC9B,MAAM,IAAI,CAAC,qBAAqB,CAAC;SAClC;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CACX,MAAc,EACd,OAA8B,EAC9B,cAAqC;QAErC,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC/B,IAAA,wBAAS,EAAC,IAAI,CAAC,gBAAgB,EAAE,mCAAmC,CAAC,CAAC;QACtE,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE;YACzB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;SACtD;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC;QAExC,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE,CAAC;QACnE,MAAM,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,eAAe,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QAE9E,IAAI,GAAG,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,eAAe,CAAC,EAAE,EAAE;YAClE,WAAW;SACZ,CAAC,CAAC;QAEH,gBAAM,CAAC,KAAK,CAAC,oCAAoC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAExE,OACE,GAAG,CAAC,MAAM,KAAK,aAAa;YAC5B,GAAG,CAAC,MAAM,KAAK,QAAQ;YACvB,GAAG,CAAC,MAAM,KAAK,iBAAiB,EAChC;YACA,IAAI,GAAG,CAAC,MAAM,KAAK,iBAAiB,EAAE;gBACpC,MAAM,cAAc,GAAG,GAAG,CAAC,cAAc,CAAC;gBAC1C,IAAA,wBAAS,EAAC,cAAc,EAAE,+CAA+C,CAAC,CAAC;gBAC3E,IAAI,cAAc,KAAK,IAAI,IAAI,cAAc,CAAC,IAAI,KAAK,qBAAqB,EAAE;oBAC5E,MAAM;iBACP;gBACD,MAAM,0BAA0B,GAAG,cAAc,CAAC,iBAAiB,EAAE,SAAS,CAAC,MAAM,CACnF,CAAC,QAAQ,EAAE,EAAE;oBACX,OAAO,CACL,QAAQ,CAAC,IAAI,KAAK,UAAU;wBAC5B,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,qBAAqB,IAAI,EAAE,CAAC,CAC7E,CAAC;gBACJ,CAAC,CACF,CAAC;gBACF,IAAI,CAAC,0BAA0B,IAAI,0BAA0B,CAAC,MAAM,KAAK,CAAC,EAAE;oBAC1E,MAAM;iBACP;gBACD,gBAAM,CAAC,KAAK,CACV,gDAAgD,0BAA0B,CAAC,GAAG,CAC5E,CAAC,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC,IAAI,CACjC,EAAE,CACJ,CAAC;gBACF,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,GAAG,CACnC,0BAA0B,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE;oBAChD,gBAAM,CAAC,KAAK,CACV,iCAAiC,QAAQ,CAAC,QAAQ,CAAC,IAAI,MAAM,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,CAC7F,CAAC;oBACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,qBAAsB,CAC9D,QAAQ,CAAC,QAAQ,CAAC,IAAI,CACvB,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;oBAC/B,OAAO;wBACL,YAAY,EAAE,QAAQ,CAAC,EAAE;wBACzB,MAAM,EAAE,MAAM;qBACf,CAAC;gBACJ,CAAC,CAAC,CACH,CAAC;gBACF,gBAAM,CAAC,KAAK,CACV,kDAAkD,GAAG,CAAC,QAAQ,KAAK,IAAI,CAAC,SAAS,CAC/E,WAAW,CACZ,EAAE,CACJ,CAAC;gBACF,GAAG,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;aAC7F;YAED,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;YAE1D,gBAAM,CAAC,KAAK,CAAC,yCAAyC,GAAG,CAAC,EAAE,SAAS,CAAC,CAAC;YACvE,GAAG,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;YAC/D,gBAAM,CAAC,KAAK,CAAC,oCAAoC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;SACzE;QAED,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,IAAI,GAAG,CAAC,MAAM,KAAK,iBAAiB,EAAE;YAClE,IAAI,GAAG,CAAC,SAAS,EAAE;gBACjB,OAAO;oBACL,KAAK,EAAE,sBAAsB,GAAG,CAAC,SAAS,CAAC,OAAO,EAAE;iBACrD,CAAC;aACH;YACD,OAAO;gBACL,KAAK,EAAE,sBAAsB,GAAG,CAAC,MAAM,EAAE;aAC1C,CAAC;SACH;QAED,gBAAM,CAAC,KAAK,CAAC,mDAAmD,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;QAChF,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/F,gBAAM,CAAC,KAAK,CAAC,0CAA0C,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAEhF,MAAM,YAAY,GAAG,EAAE,CAAC;QACxB,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,IAAI,EAAE;YAC7B,IAAI,IAAI,CAAC,IAAI,KAAK,kBAAkB,EAAE;gBACpC,gBAAM,CAAC,KAAK,CAAC,sCAAsC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC9D,MAAM,WAAW,GAAG,IAAI,CAAC,WAA4C,CAAC;gBACtE,iGAAiG;gBACjG,MAAM,SAAS,GACZ,WAAW,CAAC,eAAuB,CAAC,UAAU,IAAI,WAAW,CAAC,eAAe,CAAC,SAAS,CAAC;gBAC3F,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;gBAChF,gBAAM,CAAC,KAAK,CAAC,iDAAiD,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAEzF,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO;qBAC5B,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CACf,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,IAAI,UAAU,CAC1E;qBACA,IAAI,CAAC,IAAI,CAAC,CAAC;gBACd,YAAY,CAAC,IAAI,CAAC,IAAI,IAAA,oBAAW,EAAC,OAAO,CAAC,IAAI,CAAC,KAAK,OAAO,EAAE,CAAC,CAAC;aAChE;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE;gBACrC,KAAK,MAAM,QAAQ,IAAK,IAAI,CAAC,WAAsC,CAAC,SAAS,EAAE;oBAC7E,IAAI,QAAQ,CAAC,IAAI,KAAK,UAAU,EAAE;wBAChC,YAAY,CAAC,IAAI,CACf,kBAAkB,QAAQ,CAAC,QAAQ,CAAC,IAAI,mBAAmB,QAAQ,CAAC,QAAQ,CAAC,SAAS,GAAG,CAC1F,CAAC;wBACF,YAAY,CAAC,IAAI,CAAC,qBAAqB,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;qBACrE;yBAAM,IAAI,QAAQ,CAAC,IAAI,KAAK,WAAW,EAAE;wBACxC,YAAY,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;qBACtC;yBAAM,IAAI,QAAQ,CAAC,IAAI,KAAK,kBAAkB,EAAE;wBAC/C,MAAM,MAAM,GAAG,QAAQ,CAAC,eAAe,CAAC,OAAO;6BAC5C,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,IAAI,UAAU,CAAC,CAAC;6BACnF,IAAI,CAAC,IAAI,CAAC,CAAC;wBACd,YAAY,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;wBAC9C,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;wBAClD,YAAY,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;wBAC/C,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;qBAC3B;yBAAM;wBACL,YAAY,CAAC,IAAI,CAAC,4BAA6B,QAAgB,CAAC,IAAI,GAAG,CAAC,CAAC;qBAC1E;iBACF;aACF;iBAAM;gBACL,YAAY,CAAC,IAAI,CAAC,uBAAuB,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC;aACxD;SACF;QAED,OAAO;YACL,MAAM,EAAE,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE;YACxC;;;;;;cAME;SACH,CAAC;IACJ,CAAC;CACF;AArLD,oEAqLC"}

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

@@ -1 +1 @@
-{"version":3,"file":"bedrock.d.ts","sourceRoot":"","sources":["../../../src/providers/bedrock.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iCAAiC,CAAC;AAEtE,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAE/E,UAAU,cAAc;IACtB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAgED,qBAAa,4BAA6B,YAAW,WAAW;IAC9D,MAAM,CAAC,6BAA6B,WAAmC;IAEvE,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,cAAc,CAAC;IACvB,GAAG,CAAC,EAAE,YAAY,CAAC;IACnB,OAAO,CAAC,EAAE,cAAc,CAAC;gBAGvB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,cAAc,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAS5E,EAAE,IAAI,MAAM;IAIN,kBAAkB;IAcxB,QAAQ,IAAI,MAAM;IAIlB,SAAS,IAAI,MAAM;IASb,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAmEzD"}
+{"version":3,"file":"bedrock.d.ts","sourceRoot":"","sources":["../../../src/providers/bedrock.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iCAAiC,CAAC;AAEtE,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAG/E,UAAU,cAAc;IACtB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAoGD,qBAAa,4BAA6B,YAAW,WAAW;IAC9D,MAAM,CAAC,6BAA6B,WAAmC;IAEvE,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,cAAc,CAAC;IACvB,GAAG,CAAC,EAAE,YAAY,CAAC;IACnB,OAAO,CAAC,EAAE,cAAc,CAAC;gBAGvB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,cAAc,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAS5E,EAAE,IAAI,MAAM;IAIN,kBAAkB;IAcxB,QAAQ,IAAI,MAAM;IAIlB,SAAS,IAAI,MAAM;IASb,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAqEzD"}

package/dist/src/providers/bedrock.js

@@ -30,8 +30,9 @@ exports.AwsBedrockCompletionProvider = void 0;
 const sdk_1 = __importDefault(require("@anthropic-ai/sdk"));
 const logger_1 = __importDefault(require("../logger"));
 const cache_1 = require("../cache");
+const anthropic_1 = require("./anthropic");
 const BEDROCK_MODEL = {
-    CLAUDE: {
+    CLAUDE_COMPLETION: {
         params: (config, prompt, stop) => ({
             prompt: `${sdk_1.default.HUMAN_PROMPT} ${prompt} ${sdk_1.default.AI_PROMPT}`,
             max_tokens_to_sample: config?.max_tokens_to_sample || parseInt(process.env.AWS_BEDROCK_MAX_TOKENS || '1024'),
@@ -42,29 +43,56 @@ const BEDROCK_MODEL = {
             return responseJson?.completion;
         },
     },
+    CLAUDE_MESSAGES: {
+        params: (config, prompt, stop) => {
+            const { system, extractedMessages } = (0, anthropic_1.parseMessages)(prompt);
+            return {
+                max_tokens: config?.max_tokens || parseInt(process.env.AWS_BEDROCK_MAX_TOKENS || '1024'),
+                temperature: config.temperature || 0,
+                anthropic_version: config.anthropic_version || 'bedrock-2023-05-31',
+                messages: extractedMessages,
+                ...(system ? { system } : {}),
+            };
+        },
+        output: (responseJson) => {
+            return responseJson?.content[0].text;
+        }
+    },
     TITAN_TEXT: {
         params: (config, prompt, stop) => ({
             inputText: prompt,
             textGenerationConfig: {
-                maxTokenCount: config?.textGenerationConfig?.maxTokenCount || parseInt(process.env.AWS_BEDROCK_MAX_TOKENS || '1024'),
-                temperature: config?.textGenerationConfig?.temperature || parseFloat(process.env.AWS_BEDROCK_TEMPERATURE || '0'),
+                maxTokenCount: config?.textGenerationConfig?.maxTokenCount ||
+                    parseInt(process.env.AWS_BEDROCK_MAX_TOKENS || '1024'),
+                temperature: config?.textGenerationConfig?.temperature ||
+                    parseFloat(process.env.AWS_BEDROCK_TEMPERATURE || '0'),
                 topP: config?.textGenerationConfig?.topP || parseFloat(process.env.AWS_BEDROCK_TOP_P || '1'),
                 stopSequences: config?.textGenerationConfig?.stopSequences || stop,
-            }
+            },
         }),
         output: (responseJson) => {
             return responseJson?.results[0]?.outputText;
         },
-    }
+    },
 };
 const AWS_BEDROCK_MODELS = {
-    'anthropic.claude-instant-v1': BEDROCK_MODEL.CLAUDE,
-    'anthropic.claude-v1': BEDROCK_MODEL.CLAUDE,
-    'anthropic.claude-v2': BEDROCK_MODEL.CLAUDE,
-    'anthropic.claude-v2.1': BEDROCK_MODEL.CLAUDE,
+    'anthropic.claude-instant-v1': BEDROCK_MODEL.CLAUDE_COMPLETION,
+    'anthropic.claude-v1': BEDROCK_MODEL.CLAUDE_COMPLETION,
+    'anthropic.claude-v2': BEDROCK_MODEL.CLAUDE_COMPLETION,
+    'anthropic.claude-v2:1': BEDROCK_MODEL.CLAUDE_COMPLETION,
     'amazon.titan-text-lite-v1': BEDROCK_MODEL.TITAN_TEXT,
     'amazon.titan-text-express-v1': BEDROCK_MODEL.TITAN_TEXT,
 };
+function getHandlerForModel(modelName) {
+    const ret = AWS_BEDROCK_MODELS[modelName];
+    if (ret) {
+        return ret;
+    }
+    if (modelName.startsWith('anthropic.claude')) {
+        return BEDROCK_MODEL.CLAUDE_MESSAGES;
+    }
+    throw new Error(`Unknown Amazon Bedrock model: ${modelName}`);
+}
 class AwsBedrockCompletionProvider {
     constructor(modelName, options = {}) {
         const { config, id, env } = options;
@@ -107,10 +135,10 @@ class AwsBedrockCompletionProvider {
         catch (err) {
             throw new Error(`BEDROCK_STOP is not a valid JSON string: ${err}`);
         }
-        let model = AWS_BEDROCK_MODELS[this.modelName];
+        let model = getHandlerForModel(this.modelName);
         if (!model) {
             logger_1.default.warn(`Unknown Amazon Bedrock model: ${this.modelName}. Assuming its API is Claude-like.`);
-            model = BEDROCK_MODEL.CLAUDE;
+            model = BEDROCK_MODEL.CLAUDE_MESSAGES;
         }
         const params = model.params(this.config, prompt, stop);
         logger_1.default.debug(`Calling Amazon Bedrock API: ${JSON.stringify(params)}`);
@@ -131,10 +159,10 @@ class AwsBedrockCompletionProvider {
         let response;
         try {
             response = await bedrockInstance.invokeModel({
-                body: JSON.stringify(params),
                 modelId: this.modelName,
                 accept: 'application/json',
                 contentType: 'application/json',
+                body: JSON.stringify(params),
             });
         }
         catch (err) {

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

@@ -1 +1 @@
-{"version":3,"file":"bedrock.js","sourceRoot":"","sources":["../../../src/providers/bedrock.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4DAA0C;AAE1C,uDAA+B;AAC/B,oCAAoD;AAiCpD,MAAM,aAAa,GAAG;IACpB,MAAM,EAAE;QACN,MAAM,EAAE,CAAC,MAAsC,EAAE,MAAc,EAAE,IAAc,EAAE,EAAE,CAAC,CAAC;YACnF,MAAM,EAAE,GAAG,aAAS,CAAC,YAAY,IAAI,MAAM,IAAI,aAAS,CAAC,SAAS,EAAE;YACpE,oBAAoB,EAClB,MAAM,EAAE,oBAAoB,IAAI,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,IAAI,MAAM,CAAC;YACxF,WAAW,EACT,MAAM,CAAC,WAAW,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,GAAG,CAAC;YAC9E,cAAc,EAAE,IAAI;SACrB,CAAC;QACF,MAAM,EAAE,CAAC,YAAiB,EAAE,EAAE;YAC5B,OAAO,YAAY,EAAE,UAAU,CAAC;QAClC,CAAC;KACF;IACD,UAAU,EAAE;QACV,MAAM,EAAE,CAAC,MAAoC,EAAE,MAAc,EAAE,IAAc,EAAE,EAAE,CAAC,CAAC;YACjF,SAAS,EAAE,MAAM;YACjB,oBAAoB,EAAE;gBACpB,aAAa,EAAE,MAAM,EAAE,oBAAoB,EAAE,aAAa,IAAI,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,IAAI,MAAM,CAAC;gBACpH,WAAW,EAAE,MAAM,EAAE,oBAAoB,EAAE,WAAW,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,GAAG,CAAC;gBAChH,IAAI,EAAE,MAAM,EAAE,oBAAoB,EAAE,IAAI,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,GAAG,CAAC;gBAC5F,aAAa,EAAE,MAAM,EAAE,oBAAoB,EAAE,aAAa,IAAI,IAAI;aACnE;SACF,CAAC;QACF,MAAM,EAAE,CAAC,YAAiB,EAAE,EAAE;YAC5B,OAAO,YAAY,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC;QAC9C,CAAC;KACF;CACF,CAAA;AAED,MAAM,kBAAkB,GAAkC;IACxD,6BAA6B,EAAE,aAAa,CAAC,MAAM;IACnD,qBAAqB,EAAE,aAAa,CAAC,MAAM;IAC3C,qBAAqB,EAAE,aAAa,CAAC,MAAM;IAC3C,uBAAuB,EAAE,aAAa,CAAC,MAAM;IAC7C,2BAA2B,EAAE,aAAa,CAAC,UAAU;IACrD,8BAA8B,EAAE,aAAa,CAAC,UAAU;CACzD,CAAC;AAEF,MAAa,4BAA4B;IAQvC,YACE,SAAiB,EACjB,UAAwE,EAAE;QAE1E,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,WAAW,IAAI,CAAC,SAAS,EAAE,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,kBAAkB;QACtB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE;YACjB,IAAI;gBACF,MAAM,EAAE,cAAc,EAAE,GAAG,wDAAa,iCAAiC,GAAC,CAAC;gBAC3E,IAAI,CAAC,OAAO,GAAG,IAAI,cAAc,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;aACjE;YAAC,OAAO,GAAG,EAAE;gBACZ,MAAM,IAAI,KAAK,CACb,8HAA8H,CAC/H,CAAC;aACH;SACF;QACD,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,QAAQ;QACN,OAAO,4BAA4B,IAAI,CAAC,SAAS,GAAG,CAAC;IACvD,CAAC;IAED,SAAS;QACP,OAAO,CACL,IAAI,CAAC,MAAM,EAAE,MAAM;YACnB,IAAI,CAAC,GAAG,EAAE,kBAAkB;YAC5B,OAAO,CAAC,GAAG,CAAC,kBAAkB;YAC9B,WAAW,CACZ,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,IAAI,IAAc,CAAC;QACnB,IAAI;YACF,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB;gBACjC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;gBAC1C,CAAC,CAAC,CAAC,YAAY,EAAE,eAAe,CAAC,CAAC;SACrC;QAAC,OAAO,GAAG,EAAE;YACZ,MAAM,IAAI,KAAK,CAAC,4CAA4C,GAAG,EAAE,CAAC,CAAC;SACpE;QAED,IAAI,KAAK,GAAG,kBAAkB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK,EAAE;YACV,gBAAM,CAAC,IAAI,CAAC,iCAAiC,IAAI,CAAC,SAAS,oCAAoC,CAAC,CAAC;YACjG,KAAK,GAAG,aAAa,CAAC,MAAM,CAAC;SAC9B;QACD,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;QAEvD,gBAAM,CAAC,KAAK,CAAC,+BAA+B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAEtE,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,WAAW,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;QAErD,IAAI,IAAA,sBAAc,GAAE,EAAE;YACpB,iCAAiC;YACjC,MAAM,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACjD,IAAI,cAAc,EAAE;gBAClB,gBAAM,CAAC,KAAK,CAAC,iCAAiC,MAAM,KAAK,cAAc,EAAE,CAAC,CAAC;gBAC3E,OAAO;oBACL,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,cAAwB,CAAC,CAAC;oBAC1D,UAAU,EAAE,EAAE;iBACf,CAAC;aACH;SACF;QAED,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxD,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,QAAQ,GAAG,MAAM,eAAe,CAAC,WAAW,CAAC;gBAC3C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;gBAC5B,OAAO,EAAE,IAAI,CAAC,SAAS;gBACvB,MAAM,EAAE,kBAAkB;gBAC1B,WAAW,EAAE,kBAAkB;aAChC,CAAC,CAAC;SACJ;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;SACH;QACD,gBAAM,CAAC,KAAK,CAAC,kCAAkC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,EAAE,CAAC,CAAC;QAChG,IAAI,IAAA,sBAAc,GAAE,EAAE;YACpB,IAAI;gBACF,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,CAAC;aAC9D;YAAC,OAAO,GAAG,EAAE;gBACZ,gBAAM,CAAC,KAAK,CAAC,6BAA6B,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;aAC1D;SACF;QACD,IAAI;YACF,OAAO;gBACL,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,CAAC;gBACnE,UAAU,EAAE,EAAE,EAAE,4DAA4D;aAC7E,CAAC;SACH;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;;AApHH,oEAqHC;AApHQ,0DAA6B,GAAG,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC"}
+{"version":3,"file":"bedrock.js","sourceRoot":"","sources":["../../../src/providers/bedrock.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4DAA0C;AAE1C,uDAA+B;AAC/B,oCAAoD;AAKpD,2CAA4C;AAmC5C,MAAM,aAAa,GAAG;IACpB,iBAAiB,EAAE;QACjB,MAAM,EAAE,CAAC,MAA4C,EAAE,MAAc,EAAE,IAAc,EAAE,EAAE,CAAC,CAAC;YACzF,MAAM,EAAE,GAAG,aAAS,CAAC,YAAY,IAAI,MAAM,IAAI,aAAS,CAAC,SAAS,EAAE;YACpE,oBAAoB,EAClB,MAAM,EAAE,oBAAoB,IAAI,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,IAAI,MAAM,CAAC;YACxF,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,GAAG,CAAC;YACzF,cAAc,EAAE,IAAI;SACrB,CAAC;QACF,MAAM,EAAE,CAAC,YAAiB,EAAE,EAAE;YAC5B,OAAO,YAAY,EAAE,UAAU,CAAC;QAClC,CAAC;KACF;IACD,eAAe,EAAE;QACf,MAAM,EAAE,CAAC,MAA8C,EAAE,MAAc,EAAE,IAAc,EAAE,EAAE;YACzF,MAAM,EAAE,MAAM,EAAE,iBAAiB,EAAE,GAAG,IAAA,yBAAa,EAAC,MAAM,CAAC,CAAC;YAC5D,OAAO;gBACL,UAAU,EAAE,MAAM,EAAE,UAAU,IAAI,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,IAAI,MAAM,CAAC;gBACxF,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,CAAC;gBACpC,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,oBAAoB;gBACnE,QAAQ,EAAE,iBAAiB;gBAC3B,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC9B,CAAC;QACJ,CAAC;QACD,MAAM,EAAE,CAAC,YAAiB,EAAE,EAAE;YAC5B,OAAO,YAAY,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACvC,CAAC;KACF;IACD,UAAU,EAAE;QACV,MAAM,EAAE,CAAC,MAAoC,EAAE,MAAc,EAAE,IAAc,EAAE,EAAE,CAAC,CAAC;YACjF,SAAS,EAAE,MAAM;YACjB,oBAAoB,EAAE;gBACpB,aAAa,EACX,MAAM,EAAE,oBAAoB,EAAE,aAAa;oBAC3C,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,IAAI,MAAM,CAAC;gBACxD,WAAW,EACT,MAAM,EAAE,oBAAoB,EAAE,WAAW;oBACzC,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,GAAG,CAAC;gBACxD,IAAI,EACF,MAAM,EAAE,oBAAoB,EAAE,IAAI,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,GAAG,CAAC;gBACxF,aAAa,EAAE,MAAM,EAAE,oBAAoB,EAAE,aAAa,IAAI,IAAI;aACnE;SACF,CAAC;QACF,MAAM,EAAE,CAAC,YAAiB,EAAE,EAAE;YAC5B,OAAO,YAAY,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC;QAC9C,CAAC;KACF;CACF,CAAC;AAEF,MAAM,kBAAkB,GAAkC;IACxD,6BAA6B,EAAE,aAAa,CAAC,iBAAiB;IAC9D,qBAAqB,EAAE,aAAa,CAAC,iBAAiB;IACtD,qBAAqB,EAAE,aAAa,CAAC,iBAAiB;IACtD,uBAAuB,EAAE,aAAa,CAAC,iBAAiB;IACxD,2BAA2B,EAAE,aAAa,CAAC,UAAU;IACrD,8BAA8B,EAAE,aAAa,CAAC,UAAU;CACzD,CAAC;AAEF,SAAS,kBAAkB,CAAC,SAAiB;IAC3C,MAAM,GAAG,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;IAC1C,IAAI,GAAG,EAAE;QACP,OAAO,GAAG,CAAC;KACZ;IACD,IAAI,SAAS,CAAC,UAAU,CAAC,kBAAkB,CAAC,EAAE;QAC5C,OAAO,aAAa,CAAC,eAAe,CAAC;KACtC;IACD,MAAM,IAAI,KAAK,CAAC,iCAAiC,SAAS,EAAE,CAAC,CAAC;AAChE,CAAC;AAED,MAAa,4BAA4B;IAQvC,YACE,SAAiB,EACjB,UAAwE,EAAE;QAE1E,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,WAAW,IAAI,CAAC,SAAS,EAAE,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,kBAAkB;QACtB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE;YACjB,IAAI;gBACF,MAAM,EAAE,cAAc,EAAE,GAAG,wDAAa,iCAAiC,GAAC,CAAC;gBAC3E,IAAI,CAAC,OAAO,GAAG,IAAI,cAAc,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;aACjE;YAAC,OAAO,GAAG,EAAE;gBACZ,MAAM,IAAI,KAAK,CACb,8HAA8H,CAC/H,CAAC;aACH;SACF;QACD,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,QAAQ;QACN,OAAO,4BAA4B,IAAI,CAAC,SAAS,GAAG,CAAC;IACvD,CAAC;IAED,SAAS;QACP,OAAO,CACL,IAAI,CAAC,MAAM,EAAE,MAAM;YACnB,IAAI,CAAC,GAAG,EAAE,kBAAkB;YAC5B,OAAO,CAAC,GAAG,CAAC,kBAAkB;YAC9B,WAAW,CACZ,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,IAAI,IAAc,CAAC;QACnB,IAAI;YACF,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB;gBACjC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;gBAC1C,CAAC,CAAC,CAAC,YAAY,EAAE,eAAe,CAAC,CAAC;SACrC;QAAC,OAAO,GAAG,EAAE;YACZ,MAAM,IAAI,KAAK,CAAC,4CAA4C,GAAG,EAAE,CAAC,CAAC;SACpE;QAED,IAAI,KAAK,GAAG,kBAAkB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK,EAAE;YACV,gBAAM,CAAC,IAAI,CACT,iCAAiC,IAAI,CAAC,SAAS,oCAAoC,CACpF,CAAC;YACF,KAAK,GAAG,aAAa,CAAC,eAAe,CAAC;SACvC;QACD,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;QAEvD,gBAAM,CAAC,KAAK,CAAC,+BAA+B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAEtE,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,WAAW,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;QAErD,IAAI,IAAA,sBAAc,GAAE,EAAE;YACpB,iCAAiC;YACjC,MAAM,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACjD,IAAI,cAAc,EAAE;gBAClB,gBAAM,CAAC,KAAK,CAAC,iCAAiC,MAAM,KAAK,cAAc,EAAE,CAAC,CAAC;gBAC3E,OAAO;oBACL,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,cAAwB,CAAC,CAAC;oBAC1D,UAAU,EAAE,EAAE;iBACf,CAAC;aACH;SACF;QAED,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxD,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,QAAQ,GAAG,MAAM,eAAe,CAAC,WAAW,CAAC;gBAC3C,OAAO,EAAE,IAAI,CAAC,SAAS;gBACvB,MAAM,EAAE,kBAAkB;gBAC1B,WAAW,EAAE,kBAAkB;gBAC/B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC7B,CAAC,CAAC;SACJ;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;SACH;QACD,gBAAM,CAAC,KAAK,CAAC,kCAAkC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,EAAE,CAAC,CAAC;QAChG,IAAI,IAAA,sBAAc,GAAE,EAAE;YACpB,IAAI;gBACF,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,CAAC;aAC9D;YAAC,OAAO,GAAG,EAAE;gBACZ,gBAAM,CAAC,KAAK,CAAC,6BAA6B,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;aAC1D;SACF;QACD,IAAI;YACF,OAAO;gBACL,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,CAAC;gBACnE,UAAU,EAAE,EAAE,EAAE,4DAA4D;aAC7E,CAAC;SACH;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;;AAtHH,oEAuHC;AAtHQ,0DAA6B,GAAG,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC"}

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

@@ -77,5 +77,23 @@ export declare class HuggingfaceSentenceSimilarityProvider implements ApiSimilar
     callApi(): Promise<ProviderResponse>;
     callSimilarityApi(expected: string, input: string): Promise<ProviderSimilarityResponse>;
 }
+interface HuggingfaceTokenClassificationOptions {
+    apiKey?: string;
+    apiEndpoint?: string;
+    aggregation_strategy?: string;
+    use_cache?: boolean;
+    wait_for_model?: boolean;
+}
+export declare class HuggingfaceTokenExtractionProvider implements ApiProvider {
+    modelName: string;
+    config: HuggingfaceTokenClassificationOptions;
+    constructor(modelName: string, options?: {
+        id?: string;
+        config?: HuggingfaceTokenClassificationOptions;
+    });
+    id(): string;
+    callClassificationApi(input: string): Promise<ProviderClassificationResponse>;
+    callApi(prompt: string): Promise<ProviderResponse>;
+}
 export {};
 //# sourceMappingURL=huggingface.d.ts.map

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;CAmDzD;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"}
+{"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;CAmDzD;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"}

package/dist/src/providers/huggingface.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.HuggingfaceSentenceSimilarityProvider = exports.HuggingfaceFeatureExtractionProvider = exports.HuggingfaceTextClassificationProvider = exports.HuggingfaceTextGenerationProvider = void 0;
+exports.HuggingfaceTokenExtractionProvider = exports.HuggingfaceSentenceSimilarityProvider = exports.HuggingfaceFeatureExtractionProvider = exports.HuggingfaceTextClassificationProvider = exports.HuggingfaceTextGenerationProvider = void 0;
 const logger_1 = __importDefault(require("../logger"));
 const cache_1 = require("../cache");
 const shared_1 = require("./shared");
@@ -281,4 +281,74 @@ class HuggingfaceSentenceSimilarityProvider {
     }
 }
 exports.HuggingfaceSentenceSimilarityProvider = HuggingfaceSentenceSimilarityProvider;
+class HuggingfaceTokenExtractionProvider {
+    constructor(modelName, options = {}) {
+        const { id, config } = options;
+        this.modelName = modelName;
+        this.id = id ? () => id : this.id;
+        this.config = config || {};
+    }
+    id() {
+        return `huggingface:token-classification:${this.modelName}`;
+    }
+    async callClassificationApi(input) {
+        const params = {
+            inputs: input,
+            parameters: {
+                aggregation_strategy: this.config.aggregation_strategy || 'simple',
+            },
+            options: {
+                use_cache: this.config.use_cache !== undefined ? this.config.use_cache : true,
+                wait_for_model: this.config.wait_for_model || false,
+            },
+        };
+        let response;
+        try {
+            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}` }
+                        : {}),
+                },
+                body: JSON.stringify(params),
+            }, shared_1.REQUEST_TIMEOUT_MS);
+            if (response.data.error) {
+                return {
+                    error: `API call error: ${response.data.error}`,
+                };
+            }
+            if (!Array.isArray(response.data)) {
+                return {
+                    error: `Malformed response data: ${response.data}`,
+                };
+            }
+            // Take the highest score of each entity group
+            const classification = {};
+            for (const item of response.data) {
+                if (!classification[item.entity_group] || classification[item.entity_group] < item.score) {
+                    classification[item.entity_group] = item.score;
+                }
+            }
+            return { classification };
+        }
+        catch (err) {
+            return {
+                error: `API call error: ${String(err)}. Output:\n${response?.data}`,
+            };
+        }
+    }
+    async callApi(prompt) {
+        const ret = await this.callClassificationApi(prompt);
+        return {
+            error: ret.error,
+            output: JSON.stringify(ret.classification),
+        };
+    }
+}
+exports.HuggingfaceTokenExtractionProvider = HuggingfaceTokenExtractionProvider;
 //# sourceMappingURL=huggingface.js.map

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,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;AAvFD,8EAuFC;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"}
+{"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,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;AAvFD,8EAuFC;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"}

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

@@ -1,6 +1,9 @@
 import { ApiProvider, EnvOverrides, ProviderResponse } from '../types';
 interface MistralChatCompletionOptions {
     apiKey?: string;
+    apiKeyEnvar?: string;
+    apiHost?: string;
+    apiBaseUrl?: string;
     temperature?: number;
     top_p?: number;
     max_tokens?: number;
@@ -30,6 +33,8 @@ export declare class MistralChatCompletionProvider implements ApiProvider {
     });
     id(): string;
     toString(): string;
+    getApiUrlDefault(): string;
+    getApiUrl(): string;
     getApiKey(): string | undefined;
     callApi(prompt: string): Promise<ProviderResponse>;
 }