promptfoo 0.120.27 → 0.121.2

package/dist/src/{sagemaker-gmskuyre.js → sagemaker-CNBxx5CJ.js}

@@ -1,7 +1,12 @@
-import { L as getEnvFloat, P as logger_default, R as getEnvInt, S as telemetry_default, c as transform, z as getEnvString } from "./server/index.js";
+import { b as getEnvString, i as logger, v as getEnvFloat, y as getEnvInt } from "./logger-DLcq4dWf.js";
+import "./fetch-HaqdX7U1.js";
+import "./accounts-B0pgC1oV.js";
+import "./esm-CaIwzWR5.js";
+import { n as telemetry } from "./telemetry-Cps3mIU-.js";
+import "./pythonUtils-D6fwaDSg.js";
+import { n as transform } from "./transform-aa6tmVpZ.js";
 import { z } from "zod";
 import crypto from "crypto";
-
 //#region src/providers/sagemaker.ts
 /**
 * Sleep utility function for implementing delays
@@ -59,13 +64,13 @@ var SageMakerGenericProvider = class {
 		try {
 			SageMakerConfigSchema.parse(config);
 		} catch (error) {
-			logger_default.warn(`Error validating SageMaker config\nConfig: ${JSON.stringify(config)}\n${error instanceof z.ZodError ? z.prettifyError(error) : error}`);
+			logger.warn(`Error validating SageMaker config\nConfig: ${JSON.stringify(config)}\n${error instanceof z.ZodError ? z.prettifyError(error) : error}`);
 		}
 		this.config = config ?? {};
 		this.delay = delay || this.config.delay;
 		this.transform = transform || this.config.transform;
 		this.providerId = id;
-		telemetry_default.record("feature_used", { feature: "sagemaker" });
+		telemetry.record("feature_used", { feature: "sagemaker" });
 	}
 	id() {
 		return this.providerId || `sagemaker:${this.endpointName}`;
@@ -78,7 +83,7 @@ var SageMakerGenericProvider = class {
 	*/
 	async getCredentials() {
 		if (this.config.accessKeyId && this.config.secretAccessKey) {
-			logger_default.debug("Using explicit credentials from config");
+			logger.debug("Using explicit credentials from config");
 			return {
 				accessKeyId: this.config.accessKeyId,
 				secretAccessKey: this.config.secretAccessKey,
@@ -86,7 +91,7 @@ var SageMakerGenericProvider = class {
 			};
 		}
 		if (this.config.profile) {
-			logger_default.debug(`Using AWS profile: ${this.config.profile}`);
+			logger.debug(`Using AWS profile: ${this.config.profile}`);
 			try {
 				const { fromSSO } = await import("@aws-sdk/credential-provider-sso");
 				return fromSSO({ profile: this.config.profile });
@@ -94,7 +99,7 @@ var SageMakerGenericProvider = class {
 				throw new Error(`Failed to load AWS SSO profile. Please install @aws-sdk/credential-provider-sso`);
 			}
 		}
-		logger_default.debug("Using default AWS credentials from environment");
+		logger.debug("Using default AWS credentials from environment");
 	}
 	/**
 	* Initialize and return the SageMaker runtime client
@@ -109,7 +114,7 @@ var SageMakerGenericProvider = class {
 				retryMode: "adaptive",
 				...credentials ? { credentials } : {}
 			});
-			logger_default.debug(`SageMaker client initialized for region ${this.getRegion()}`);
+			logger.debug(`SageMaker client initialized for region ${this.getRegion()}`);
 		} catch {
 			throw new Error("The @aws-sdk/client-sagemaker-runtime package is required. Please install it with: npm install @aws-sdk/client-sagemaker-runtime");
 		}
@@ -155,12 +160,12 @@ var SageMakerGenericProvider = class {
 			};
 			const transformFn = this.transform || context?.originalProvider?.transform;
 			if (!transformFn) return prompt;
-			logger_default.debug(`Applying transform to prompt for SageMaker endpoint ${this.getEndpointName()}`);
+			logger.debug(`Applying transform to prompt for SageMaker endpoint ${this.getEndpointName()}`);
 			if (typeof transformFn === "string" && !transformFn.startsWith("file://")) try {
 				if (transformFn.includes("=>")) {
 					const result = new Function("prompt", "context", `try { return (${transformFn})(prompt, context); } catch(e) { throw new Error("Transform function error: " + e.message); }`)(prompt, transformContext);
 					if (result === void 0 || result === null) {
-						logger_default.debug("Transform function returned null or undefined, using original prompt");
+						logger.debug("Transform function returned null or undefined, using original prompt");
 						return prompt;
 					}
 					if (typeof result === "string") return result;
@@ -169,7 +174,7 @@ var SageMakerGenericProvider = class {
 				} else {
 					const result = new Function("prompt", "context", `try { ${transformFn} } catch(e) { throw new Error("Transform function error: " + e.message); }`)(prompt, transformContext);
 					if (result === void 0 || result === null) {
-						logger_default.debug("Transform function returned null or undefined, using original prompt");
+						logger.debug("Transform function returned null or undefined, using original prompt");
 						return prompt;
 					}
 					if (typeof result === "string") return result;
@@ -177,25 +182,25 @@ var SageMakerGenericProvider = class {
 					else return String(result);
 				}
 			} catch (transformError) {
-				logger_default.error(`Error executing inline transform: ${transformError}`);
+				logger.error(`Error executing inline transform: ${transformError}`);
 			}
 			else try {
-				const { TransformInputType } = await import("./server/index.js");
+				const { TransformInputType } = await import("./transform-aa6tmVpZ.js").then((n) => n.r);
 				const transformed = await transform(transformFn, prompt, transformContext, false, TransformInputType.OUTPUT);
 				if (transformed === void 0 || transformed === null) {
-					logger_default.debug("Transform function returned null or undefined, using original prompt");
+					logger.debug("Transform function returned null or undefined, using original prompt");
 					return prompt;
 				}
 				if (typeof transformed === "string") return transformed;
 				else if (typeof transformed === "object") return JSON.stringify(transformed);
 				else return String(transformed);
 			} catch (transformError) {
-				logger_default.error(`Error using transform utility: ${transformError}`);
+				logger.error(`Error using transform utility: ${transformError}`);
 			}
-			logger_default.warn(`Transform did not produce a valid result, using original prompt`);
+			logger.warn(`Transform did not produce a valid result, using original prompt`);
 			return prompt;
 		} catch (_) {
-			logger_default.error(`Error applying transform to prompt: ${_}`);
+			logger.error(`Error applying transform to prompt: ${_}`);
 			return prompt;
 		}
 	}
@@ -207,28 +212,28 @@ var SageMakerGenericProvider = class {
 		if (!pathExpression) return responseJson;
 		try {
 			if (pathExpression.startsWith("file://")) try {
-				const { TransformInputType } = await import("./server/index.js");
+				const { TransformInputType } = await import("./transform-aa6tmVpZ.js").then((n) => n.r);
 				const transformedResult = await transform(pathExpression, responseJson, { prompt: {} }, false, TransformInputType.OUTPUT);
 				return transformedResult !== void 0 && transformedResult !== null ? transformedResult : responseJson;
 			} catch (error) {
-				logger_default.warn(`Failed to transform response using file: ${error}`);
+				logger.warn(`Failed to transform response using file: ${error}`);
 				return responseJson;
 			}
 			try {
 				const result = new Function("json", `try { return ${pathExpression}; } catch(e) { return undefined; }`)(responseJson);
 				if (result === void 0) {
-					logger_default.warn(`Path expression "${pathExpression}" did not match any data in the response`);
-					logger_default.debug(`Response JSON structure: ${JSON.stringify(responseJson).substring(0, 200)}...`);
+					logger.warn(`Path expression "${pathExpression}" did not match any data in the response`);
+					logger.debug(`Response JSON structure: ${JSON.stringify(responseJson).substring(0, 200)}...`);
 					return responseJson;
 				}
 				return result;
 			} catch (error) {
-				logger_default.warn(`Failed to evaluate expression "${pathExpression}": ${error}`);
+				logger.warn(`Failed to evaluate expression "${pathExpression}": ${error}`);
 				return responseJson;
 			}
 		} catch (error) {
-			logger_default.warn(`Failed to extract data using path expression "${pathExpression}": ${error}`);
-			logger_default.debug(`Response JSON structure: ${JSON.stringify(responseJson).substring(0, 200)}...`);
+			logger.warn(`Failed to extract data using path expression "${pathExpression}": ${error}`);
+			logger.debug(`Response JSON structure: ${JSON.stringify(responseJson).substring(0, 200)}...`);
 			return responseJson;
 		}
 	}
@@ -265,7 +270,7 @@ var SageMakerCompletionProvider = class extends SageMakerGenericProvider {
 		const topP = typeof this.config.topP === "number" ? this.config.topP : getEnvFloat("AWS_SAGEMAKER_TOP_P") ?? 1;
 		const stopSequences = this.config.stopSequences || [];
 		let payload;
-		logger_default.debug(`Formatting payload for model type: ${this.modelType}`);
+		logger.debug(`Formatting payload for model type: ${this.modelType}`);
 		switch (this.modelType) {
 			case "openai":
 				try {
@@ -351,23 +356,23 @@ var SageMakerCompletionProvider = class extends SageMakerGenericProvider {
 	*/
 	async parseResponse(responseBody) {
 		let responseJson;
-		logger_default.debug(`Parsing response for model type: ${this.modelType}`);
+		logger.debug(`Parsing response for model type: ${this.modelType}`);
 		try {
 			responseJson = JSON.parse(responseBody);
 		} catch {
-			logger_default.debug("Response is not JSON, returning as-is");
+			logger.debug("Response is not JSON, returning as-is");
 			return responseBody;
 		}
 		if (this.config.responseFormat?.path) try {
 			const pathExpression = this.config.responseFormat.path;
 			return await this.extractFromPath(responseJson, pathExpression);
 		} catch (error) {
-			logger_default.warn(`Failed to extract from path: ${this.config.responseFormat.path}, Error: ${error}`);
-			logger_default.debug(`Response JSON structure: ${JSON.stringify(responseJson).substring(0, 200)}...`);
+			logger.warn(`Failed to extract from path: ${this.config.responseFormat.path}, Error: ${error}`);
+			logger.debug(`Response JSON structure: ${JSON.stringify(responseJson).substring(0, 200)}...`);
 			return responseJson;
 		}
 		if (responseJson.generated_text) {
-			logger_default.debug("Detected JumpStart model response format with generated_text field");
+			logger.debug("Detected JumpStart model response format with generated_text field");
 			return responseJson.generated_text;
 		}
 		switch (this.modelType) {
@@ -402,21 +407,21 @@ var SageMakerCompletionProvider = class extends SageMakerGenericProvider {
 	* Invoke SageMaker endpoint for text generation with caching, delay support, and transformations
 	*/
 	async callApi(prompt, context, _options) {
-		const { isCacheEnabled, getCache } = await import("./server/index.js");
+		const { isCacheEnabled, getCache } = await import("./cache-i1P6crbO.js").then((n) => n.t);
 		const delayMs = context?.originalProvider?.delay || this.delay;
 		const transformedPrompt = await this.applyTransformation(prompt, context);
 		const isTransformed = transformedPrompt !== prompt;
 		if (isTransformed) {
-			logger_default.debug(`Prompt transformed for SageMaker endpoint ${this.getEndpointName()}`);
-			logger_default.debug(`Original: ${prompt.substring(0, 100)}${prompt.length > 100 ? "..." : ""}`);
-			logger_default.debug(`Transformed: ${transformedPrompt.substring(0, 100)}${transformedPrompt.length > 100 ? "..." : ""}`);
+			logger.debug(`Prompt transformed for SageMaker endpoint ${this.getEndpointName()}`);
+			logger.debug(`Original: ${prompt.substring(0, 100)}${prompt.length > 100 ? "..." : ""}`);
+			logger.debug(`Transformed: ${transformedPrompt.substring(0, 100)}${transformedPrompt.length > 100 ? "..." : ""}`);
 		}
 		const bustCache = context?.bustCache ?? context?.debug === true;
 		if (isCacheEnabled() && !bustCache) {
 			const cacheKey = this.getCacheKey(transformedPrompt);
-			const cachedResult = await (getCache ? getCache() : await import("./server/index.js").then((m) => m.getCache())).get(cacheKey);
+			const cachedResult = await (getCache ? getCache() : await import("./cache-i1P6crbO.js").then((n) => n.t).then((m) => m.getCache())).get(cacheKey);
 			if (cachedResult) {
-				logger_default.debug(`Using cached SageMaker response for ${this.getEndpointName()}`);
+				logger.debug(`Using cached SageMaker response for ${this.getEndpointName()}`);
 				try {
 					const parsedResult = JSON.parse(cachedResult);
 					if (parsedResult.tokenUsage) parsedResult.tokenUsage.cached = parsedResult.tokenUsage.total || 0;
@@ -429,18 +434,18 @@ var SageMakerCompletionProvider = class extends SageMakerGenericProvider {
 						cached: true
 					};
 				} catch (_) {
-					logger_default.warn(`Failed to parse cached SageMaker response: ${_}`);
+					logger.warn(`Failed to parse cached SageMaker response: ${_}`);
 				}
 			}
 		}
 		if (delayMs && delayMs > 0) {
-			logger_default.debug(`Applying delay of ${delayMs}ms before calling SageMaker endpoint ${this.getEndpointName()}`);
+			logger.debug(`Applying delay of ${delayMs}ms before calling SageMaker endpoint ${this.getEndpointName()}`);
 			await sleep(delayMs);
 		}
 		const runtime = await this.getSageMakerRuntimeInstance();
 		const payload = this.formatPayload(transformedPrompt);
-		logger_default.debug(`Calling SageMaker endpoint ${this.getEndpointName()}`);
-		logger_default.debug(`With payload: ${payload.length > 1e3 ? payload.substring(0, 1e3) + "..." : payload}`);
+		logger.debug(`Calling SageMaker endpoint ${this.getEndpointName()}`);
+		logger.debug(`With payload: ${payload.length > 1e3 ? payload.substring(0, 1e3) + "..." : payload}`);
 		try {
 			const { InvokeEndpointCommand } = await import("@aws-sdk/client-sagemaker-runtime");
 			const command = new InvokeEndpointCommand({
@@ -453,17 +458,17 @@ var SageMakerCompletionProvider = class extends SageMakerGenericProvider {
 			const response = await runtime.send(command);
 			const _latency = Date.now() - startTime;
 			if (!response.Body) {
-				logger_default.error("No response body returned from SageMaker endpoint");
+				logger.error("No response body returned from SageMaker endpoint");
 				return { error: "No response body returned from SageMaker endpoint" };
 			}
 			const responseBody = new TextDecoder().decode(response.Body);
-			logger_default.debug(`SageMaker response (truncated): ${responseBody.length > 1e3 ? responseBody.substring(0, 1e3) + "..." : responseBody}`);
+			logger.debug(`SageMaker response (truncated): ${responseBody.length > 1e3 ? responseBody.substring(0, 1e3) + "..." : responseBody}`);
 			const output = await this.parseResponse(responseBody);
 			if (typeof output === "object" && output !== null && "code" in output) {
 				const code = output.code;
 				if (Number.isInteger(code) && code === 424) {
 					const errorMessage = `API Error: 424${output?.message ? ` ${output.message}` : ""}\n${JSON.stringify(output)}`;
-					logger_default.error(errorMessage);
+					logger.error(errorMessage);
 					return { error: errorMessage };
 				}
 			}
@@ -488,18 +493,18 @@ var SageMakerCompletionProvider = class extends SageMakerGenericProvider {
 			};
 			if (isCacheEnabled() && !bustCache && result.output && !result.error) {
 				const cacheKey = this.getCacheKey(transformedPrompt);
-				const cache = getCache ? getCache() : await import("./server/index.js").then((m) => m.getCache());
+				const cache = getCache ? getCache() : await import("./cache-i1P6crbO.js").then((n) => n.t).then((m) => m.getCache());
 				const resultToCache = JSON.stringify(result);
 				try {
 					await cache.set(cacheKey, resultToCache);
-					logger_default.debug(`Stored SageMaker response in cache with key: ${cacheKey.substring(0, 100)}...`);
+					logger.debug(`Stored SageMaker response in cache with key: ${cacheKey.substring(0, 100)}...`);
 				} catch (_) {
-					logger_default.warn(`Failed to store SageMaker response in cache: ${_}`);
+					logger.warn(`Failed to store SageMaker response in cache: ${_}`);
 				}
 			}
 			return result;
 		} catch (error) {
-			logger_default.error(`SageMaker API error: ${error}`);
+			logger.error(`SageMaker API error: ${error}`);
 			return { error: `SageMaker API error: ${error.message || String(error)}` };
 		}
 	}
@@ -533,21 +538,21 @@ var SageMakerEmbeddingProvider = class extends SageMakerGenericProvider {
 	* Invoke SageMaker endpoint for embeddings with caching, delay support, and transformations
 	*/
 	async callEmbeddingApi(text, context) {
-		const { isCacheEnabled, getCache } = await import("./server/index.js");
+		const { isCacheEnabled, getCache } = await import("./cache-i1P6crbO.js").then((n) => n.t);
 		const delayMs = context?.originalProvider?.delay || this.delay;
 		const transformedText = await this.applyTransformation(text, context);
 		const isTransformed = transformedText !== text;
 		if (isTransformed) {
-			logger_default.debug(`Text transformed for SageMaker embedding endpoint ${this.getEndpointName()}`);
-			logger_default.debug(`Original: ${text.substring(0, 100)}${text.length > 100 ? "..." : ""}`);
-			logger_default.debug(`Transformed: ${transformedText.substring(0, 100)}${transformedText.length > 100 ? "..." : ""}`);
+			logger.debug(`Text transformed for SageMaker embedding endpoint ${this.getEndpointName()}`);
+			logger.debug(`Original: ${text.substring(0, 100)}${text.length > 100 ? "..." : ""}`);
+			logger.debug(`Transformed: ${transformedText.substring(0, 100)}${transformedText.length > 100 ? "..." : ""}`);
 		}
 		const bustCache = context?.debug === true;
 		if (isCacheEnabled() && !bustCache) {
 			const cacheKey = this.getCacheKey(transformedText);
-			const cachedResult = await (await getCache ? await getCache() : await import("./server/index.js").then((m) => m.getCache())).get(cacheKey);
+			const cachedResult = await (await getCache ? await getCache() : await import("./cache-i1P6crbO.js").then((n) => n.t).then((m) => m.getCache())).get(cacheKey);
 			if (cachedResult) {
-				logger_default.debug(`Using cached SageMaker embedding response for ${this.getEndpointName()}`);
+				logger.debug(`Using cached SageMaker embedding response for ${this.getEndpointName()}`);
 				try {
 					const parsedResult = JSON.parse(cachedResult);
 					if (parsedResult.tokenUsage) parsedResult.tokenUsage.cached = parsedResult.tokenUsage.prompt || 0;
@@ -556,18 +561,18 @@ var SageMakerEmbeddingProvider = class extends SageMakerGenericProvider {
 						cached: true
 					};
 				} catch (_) {
-					logger_default.warn(`Failed to parse cached SageMaker embedding response: ${_}`);
+					logger.warn(`Failed to parse cached SageMaker embedding response: ${_}`);
 				}
 			}
 		}
 		if (delayMs && delayMs > 0) {
-			logger_default.debug(`Applying delay of ${delayMs}ms before calling SageMaker embedding endpoint ${this.getEndpointName()}`);
+			logger.debug(`Applying delay of ${delayMs}ms before calling SageMaker embedding endpoint ${this.getEndpointName()}`);
 			await sleep(delayMs);
 		}
 		const runtime = await this.getSageMakerRuntimeInstance();
 		let payload;
 		const modelType = this.config.modelType || "custom";
-		logger_default.debug(`Formatting embedding payload for model type: ${modelType}`);
+		logger.debug(`Formatting embedding payload for model type: ${modelType}`);
 		switch (modelType) {
 			case "openai":
 				payload = JSON.stringify({
@@ -586,8 +591,8 @@ var SageMakerEmbeddingProvider = class extends SageMakerGenericProvider {
 				});
 				break;
 		}
-		logger_default.debug(`Calling SageMaker embedding endpoint ${this.getEndpointName()}`);
-		logger_default.debug(`With payload: ${payload}`);
+		logger.debug(`Calling SageMaker embedding endpoint ${this.getEndpointName()}`);
+		logger.debug(`With payload: ${payload}`);
 		try {
 			const { InvokeEndpointCommand } = await import("@aws-sdk/client-sagemaker-runtime");
 			const command = new InvokeEndpointCommand({
@@ -600,11 +605,11 @@ var SageMakerEmbeddingProvider = class extends SageMakerGenericProvider {
 			const response = await runtime.send(command);
 			Date.now() - startTime;
 			if (!response.Body) {
-				logger_default.error("No response body returned from SageMaker embedding endpoint");
+				logger.error("No response body returned from SageMaker embedding endpoint");
 				return { error: "No response body returned from SageMaker embedding endpoint" };
 			}
 			const responseBody = new TextDecoder().decode(response.Body);
-			logger_default.debug(`SageMaker embedding response: ${responseBody.substring(0, 200)}...`);
+			logger.debug(`SageMaker embedding response: ${responseBody.substring(0, 200)}...`);
 			let responseJson;
 			try {
 				responseJson = JSON.parse(responseBody);
@@ -630,10 +635,10 @@ var SageMakerEmbeddingProvider = class extends SageMakerGenericProvider {
 					};
 					await this.cacheEmbeddingResult(result, transformedText, context, isTransformed, isTransformed ? text : void 0);
 					return result;
-				} else logger_default.warn("Extracted data is not a valid embedding array, trying other extraction methods");
+				} else logger.warn("Extracted data is not a valid embedding array, trying other extraction methods");
 			} catch (error) {
-				logger_default.warn(`Failed to extract embedding from path expression: ${this.config.responseFormat.path}, Error: ${error}`);
-				logger_default.debug(`Response JSON structure: ${JSON.stringify(responseJson).substring(0, 200)}...`);
+				logger.warn(`Failed to extract embedding from path expression: ${this.config.responseFormat.path}, Error: ${error}`);
+				logger.debug(`Response JSON structure: ${JSON.stringify(responseJson).substring(0, 200)}...`);
 			}
 			if (!embedding || !Array.isArray(embedding)) return { error: `Invalid embedding response format. Could not find embedding array in: ${JSON.stringify(responseJson).substring(0, 100)}...` };
 			const result = {
@@ -651,7 +656,7 @@ var SageMakerEmbeddingProvider = class extends SageMakerGenericProvider {
 			await this.cacheEmbeddingResult(result, transformedText, context, isTransformed, isTransformed ? text : void 0);
 			return result;
 		} catch (error) {
-			logger_default.error(`SageMaker embedding API error: ${error}`);
+			logger.error(`SageMaker embedding API error: ${error}`);
 			return { error: `SageMaker embedding API error: ${error.message || String(error)}` };
 		}
 	}
@@ -659,11 +664,11 @@ var SageMakerEmbeddingProvider = class extends SageMakerGenericProvider {
 	* Helper method to cache embedding results
 	*/
 	async cacheEmbeddingResult(result, text, context, isTransformed = false, originalText) {
-		const { isCacheEnabled, getCache } = await import("./server/index.js");
+		const { isCacheEnabled, getCache } = await import("./cache-i1P6crbO.js").then((n) => n.t);
 		const bustCache = context?.debug === true;
 		if (isCacheEnabled() && !bustCache && result.embedding && !result.error) {
 			const cacheKey = this.getCacheKey(text);
-			const cache = await getCache ? await getCache() : await import("./server/index.js").then((m) => m.getCache());
+			const cache = await getCache ? await getCache() : await import("./cache-i1P6crbO.js").then((n) => n.t).then((m) => m.getCache());
 			if (isTransformed && originalText && !result.metadata) result.metadata = {
 				transformed: true,
 				originalText
@@ -675,14 +680,14 @@ var SageMakerEmbeddingProvider = class extends SageMakerGenericProvider {
 			const resultToCache = JSON.stringify(result);
 			try {
 				await cache.set(cacheKey, resultToCache);
-				logger_default.debug(`Stored SageMaker embedding response in cache with key: ${cacheKey.substring(0, 100)}...`);
+				logger.debug(`Stored SageMaker embedding response in cache with key: ${cacheKey.substring(0, 100)}...`);
 			} catch (_) {
-				logger_default.warn(`Failed to store SageMaker embedding response in cache: ${_}`);
+				logger.warn(`Failed to store SageMaker embedding response in cache: ${_}`);
 			}
 		}
 	}
 };
-
 //#endregion
 export { SageMakerCompletionProvider, SageMakerEmbeddingProvider };
-//# sourceMappingURL=sagemaker-gmskuyre.js.map
+
+//# sourceMappingURL=sagemaker-CNBxx5CJ.js.map