topchester-ai 0.36.0 → 0.37.0

package/dist/bin.mjs

@@ -1,5 +1,5 @@
 #!/usr/bin/env node
-import { t as runTopchesterCli } from "./cli-BWT2aSBz.mjs";
+import { t as runTopchesterCli } from "./cli-CKfn9zSQ.mjs";
 //#region src/bin.ts
 await runTopchesterCli();
 //#endregion

package/dist/{cli-BWT2aSBz.mjs → cli-CKfn9zSQ.mjs}

@@ -4872,7 +4872,8 @@ var ModelGateway = class ModelGateway {
 		const usage = normalizeUsage(result.usage, {
 			providerId: resolved.providerId,
 			providerConfig: resolved.providerConfig,
-			responseBody: result.response.body
+			responseBody: result.response.body,
+			responseHeaders: result.response.headers
 		});
 		return {
 			text: result.text,
@@ -4962,7 +4963,8 @@ var ModelGateway = class ModelGateway {
 		const usage = normalizeUsage(result.usage, {
 			providerId: resolved.providerId,
 			providerConfig: resolved.providerConfig,
-			responseBody: result.response.body
+			responseBody: result.response.body,
+			responseHeaders: result.response.headers
 		});
 		const toolCalls = result.toolCalls.map((call, index) => {
 			const parsed = parseNativeToolCall(call.toolName, call.input);
@@ -5003,7 +5005,8 @@ var ModelGateway = class ModelGateway {
 		const usage = normalizeUsage(result.usage, {
 			providerId: resolved.providerId,
 			providerConfig: resolved.providerConfig,
-			responseBody: result.response.body
+			responseBody: result.response.body,
+			responseHeaders: result.response.headers
 		});
 		const parsed = parseToolCallWithSource(result.text, allowedSources);
 		const defaultProtocol = allowedSources.length === 1 && allowedSources[0] === "text-xml" ? "text-xml" : "text-json";
@@ -5221,7 +5224,7 @@ function extractWarningMessages(warnings) {
 	return warnings.map((warning) => formatErrorMessage$2(warning));
 }
 function normalizeUsage(usage, context) {
-	const costUsd = extractResponseCostUsd(context?.responseBody);
+	const costUsd = extractResponseCostUsd(context?.responseBody, context?.responseHeaders);
 	if (!usage) return costUsd === void 0 ? void 0 : { costUsd };
 	const normalized = {
 		...typeof usage.inputTokens === "number" ? { inputTokens: usage.inputTokens } : {},
@@ -5231,21 +5234,29 @@ function normalizeUsage(usage, context) {
 	};
 	return Object.keys(normalized).length > 0 ? normalized : void 0;
 }
-function extractResponseCostUsd(responseBody) {
+function extractResponseCostUsd(responseBody, responseHeaders) {
+	return extractResponseBodyCostUsd(responseBody) ?? extractResponseHeaderCostUsd(responseHeaders);
+}
+function extractResponseBodyCostUsd(responseBody) {
 	if (!responseBody || typeof responseBody !== "object") return;
 	const usage = responseBody.usage;
 	const hiddenParams = responseBody.hidden_params;
 	const privateHiddenParams = responseBody._hidden_params;
 	return firstFiniteNumber(getObjectNumber(responseBody, "response_cost"), getObjectNumber(responseBody, "responseCost"), getObjectNumber(responseBody, "cost"), getObjectNumber(responseBody, "cost_usd"), getObjectNumber(responseBody, "costUsd"), getObjectNumber(usage, "cost"), getObjectNumber(usage, "response_cost"), getObjectNumber(usage, "responseCost"), getObjectNumber(usage, "cost_usd"), getObjectNumber(usage, "costUsd"), getObjectNumber(hiddenParams, "response_cost"), getObjectNumber(privateHiddenParams, "response_cost"));
 }
+function extractResponseHeaderCostUsd(responseHeaders) {
+	if (!responseHeaders) return;
+	for (const [key, value] of Object.entries(responseHeaders)) if (key.toLowerCase() === "x-litellm-response-cost") return parseFiniteNumber(value);
+}
 function getObjectNumber(value, key) {
 	if (!value || typeof value !== "object") return;
 	const field = value[key];
 	if (typeof field === "number" && Number.isFinite(field)) return field;
-	if (typeof field === "string" && field.trim().length > 0) {
-		const parsed = Number(field);
-		return Number.isFinite(parsed) ? parsed : void 0;
-	}
+	if (typeof field === "string" && field.trim().length > 0) return parseFiniteNumber(field);
+}
+function parseFiniteNumber(value) {
+	const parsed = Number(value.trim());
+	return Number.isFinite(parsed) ? parsed : void 0;
 }
 function firstFiniteNumber(...values) {
 	return values.find((value) => value !== void 0);
@@ -13704,4 +13715,4 @@ function formatDryRunSyncStatus(status) {
 //#endregion
 export { runTopchesterCli as t };
 
-//# sourceMappingURL=cli-BWT2aSBz.mjs.map
+//# sourceMappingURL=cli-CKfn9zSQ.mjs.map