@posthog/ai 8.10.2 → 8.10.4

package/dist/langchain/middleware/index.cjs

@@ -244,8 +244,7 @@ const getModelParams = (params, responseServiceTier) => {
 		"streaming",
 		"language",
 		"response_format",
-		"timestamp_granularities",
-		"service_tier"
+		"timestamp_granularities"
 	]) if (key in params && params[key] !== void 0) modelParams[key] = params[key];
 	if (responseServiceTier != null) modelParams.service_tier = responseServiceTier;
 	return modelParams;
@@ -263,7 +262,7 @@ function sanitizeValues(obj) {
 }
 //#endregion
 //#region package.json
-var version = "8.10.2";
+var version = "8.10.4";
 //#endregion
 //#region src/serializeError.ts
 const DEFAULT_MAX_DEPTH = 3;
@@ -608,6 +607,13 @@ var LangChainCallbackHandler = class extends _langchain_core_callbacks_base.Base
 	_captureGeneration(traceId, runId, run, output, parentRunId) {
 		const latency = run.endTime ? (run.endTime - run.startTime) / 1e3 : 0;
 		warnIfPostHogAiGateway(run.baseUrl);
+		let modelParams = run.modelParams;
+		const servedTier = output instanceof Error ? void 0 : this._extractServedServiceTier(output);
+		if (servedTier != null) modelParams = {
+			...modelParams,
+			service_tier: servedTier
+		};
+		const eventPropertiesServedTier = servedTier != null ? { $ai_service_tier: String(servedTier) } : {};
 		const eventProperties = {
 			$ai_lib: "posthog-ai",
 			$ai_lib_version: version,
@@ -616,12 +622,13 @@ var LangChainCallbackHandler = class extends _langchain_core_callbacks_base.Base
 			$ai_span_name: run.name,
 			$ai_provider: run.provider,
 			$ai_model: run.model,
-			$ai_model_parameters: run.modelParams,
+			$ai_model_parameters: modelParams,
 			$ai_input: withPrivacyMode(this.client, this.privacyMode, run.input),
 			$ai_http_status: 200,
 			$ai_latency: latency,
 			$ai_base_url: run.baseUrl,
-			$ai_framework: "langchain"
+			$ai_framework: "langchain",
+			...eventPropertiesServedTier
 		};
 		if (parentRunId) eventProperties["$ai_parent_id"] = parentRunId;
 		if (run.tools) eventProperties["$ai_tools"] = run.tools;
@@ -741,6 +748,13 @@ var LangChainCallbackHandler = class extends _langchain_core_callbacks_base.Base
 		};
 		return sanitizeLangChain(messageDict, this.client);
 	}
+	_extractServedServiceTier(output) {
+		const gen = output.generations?.[output.generations.length - 1]?.[0];
+		const fromResponsesAdapter = gen?.message?.response_metadata?.service_tier;
+		const fromCompletionsAdapter = gen?.generationInfo?.service_tier;
+		const tier = fromResponsesAdapter ?? fromCompletionsAdapter;
+		return tier == null ? void 0 : String(tier);
+	}
 	_extractStopReason(output) {
 		if (!output.generations || !Array.isArray(output.generations)) return;
 		const lastGeneration = output.generations[output.generations.length - 1];