@posthog/ai 8.14.0 → 8.14.2

package/dist/anthropic/index.mjs

@@ -2,6 +2,11 @@ import AnthropicOriginal from "@anthropic-ai/sdk";
 import { v4 } from "uuid";
 import { toJsonSafeValue, uuidv7 } from "@posthog/core";
 import { Stream } from "@anthropic-ai/sdk/streaming";
+//#region src/typeGuards.ts
+const isObject = (value) => {
+	return value !== null && typeof value === "object" && !Array.isArray(value);
+};
+//#endregion
 //#region src/captureAiEvent.ts
 /** @internal */
 function isFullAiCaptureEnabled(client) {
@@ -167,17 +172,26 @@ var BinaryContentRedactor = class {
 		this.visited = /* @__PURE__ */ new WeakSet();
 		return this.walk(value, mediaType ? new MediaTypeContext(void 0, void 0, mediaType) : MediaTypeContext.EMPTY);
 	}
+	/** Redact one leaf while a caller owns bounded, type-preserving traversal. @internal */
+	redactLeaf(value, context) {
+		return typeof value === "string" ? this.redactString(value, context) : this.placeholderFor(context.inferMediaType());
+	}
 	walk(value, ctx) {
 		if (value === null || value === void 0) return value;
-		if (typeof value === "string") return this.redactString(value, ctx);
+		if (typeof value === "string") return this.redactLeaf(value, ctx);
 		if (typeof value !== "object") return value;
-		if (typeof Uint8Array !== "undefined" && value instanceof Uint8Array) return this.placeholderFor(ctx.inferMediaType());
+		if (typeof Uint8Array !== "undefined" && value instanceof Uint8Array) return this.redactLeaf(value, ctx);
 		if (this.visited.has(value)) return null;
 		this.visited.add(value);
 		if (Array.isArray(value)) return value.map((item) => this.walk(item, ctx));
 		const obj = value;
 		const out = {};
-		for (const k of Object.keys(obj)) out[k] = this.walk(obj[k], new MediaTypeContext(obj, k));
+		for (const k of Object.keys(obj)) Object.defineProperty(out, k, {
+			value: this.walk(obj[k], new MediaTypeContext(obj, k)),
+			enumerable: true,
+			writable: true,
+			configurable: true
+		});
 		return out;
 	}
 	redactString(value, ctx) {
@@ -199,8 +213,8 @@ var BinaryContentRedactor = class {
 };
 //#endregion
 //#region src/sanitization.ts
-const redactor = new BinaryContentRedactor();
-const sanitize = (data, client) => isFullAiCaptureEnabled(client) ? data : redactor.redact(data);
+const redactor$1 = new BinaryContentRedactor();
+const sanitize = (data, client) => isFullAiCaptureEnabled(client) ? data : redactor$1.redact(data);
 const sanitizeAnthropic = (data, client) => sanitize(data, client);
 //#endregion
 //#region src/utils.ts
@@ -222,9 +236,10 @@ function hasTokenOverrides(posthogProperties) {
 function getTokensSource(posthogProperties) {
 	return hasTokenOverrides(posthogProperties) ? "passthrough" : "sdk";
 }
+const MAX_OUTPUT_SIZE = 2e5;
 const STRING_FORMAT = "utf8";
-new TextEncoder();
-new TextDecoder(STRING_FORMAT, { fatal: false });
+const sharedTextEncoder = new TextEncoder();
+const sharedTextDecoder = new TextDecoder(STRING_FORMAT, { fatal: false });
 const getModelParams = (params, responseServiceTier) => {
 	if (!params) return {};
 	const modelParams = {};
@@ -278,8 +293,27 @@ const mergeSystemPrompt = (params, provider) => {
 	}
 	return params.messages;
 };
-const withPrivacyMode = (client, privacyMode, input) => {
-	return client.privacy_mode || privacyMode ? null : input;
+const isPrivacyModeEnabled = (client, privacyMode) => Boolean(client.privacy_mode || privacyMode);
+function toSafeString(input) {
+	if (input === void 0 || input === null) return "";
+	if (typeof input === "string") return input;
+	try {
+		return JSON.stringify(input);
+	} catch {
+		console.warn("Failed to stringify input", input);
+		return "";
+	}
+}
+const truncate = (input, client, maxBytes = MAX_OUTPUT_SIZE) => {
+	const str = toSafeString(input);
+	if (str === "") return "";
+	if (isFullAiCaptureEnabled(client)) return str;
+	const buffer = sharedTextEncoder.encode(str);
+	if (buffer.length <= maxBytes) return sharedTextDecoder.decode(buffer);
+	const truncatedBuffer = buffer.slice(0, maxBytes);
+	let truncatedStr = sharedTextDecoder.decode(truncatedBuffer);
+	if (truncatedStr.endsWith("�")) truncatedStr = truncatedStr.slice(0, -1);
+	return `${truncatedStr}... [truncated]`;
 };
 /**
 * Extract available tool calls from the request parameters.
@@ -340,7 +374,7 @@ function addDefaults(params) {
 }
 //#endregion
 //#region package.json
-var version = "8.14.0";
+var version = "8.14.2";
 //#endregion
 //#region src/serializeError.ts
 const DEFAULT_MAX_DEPTH = 3;
@@ -427,7 +461,7 @@ const captureAiGeneration = async (client, options) => {
 		const eventType = options.eventType ?? "$ai_generation";
 		const privacyMode = options.privacyMode ?? false;
 		const usage = options.usage ?? {};
-		const shouldRedact = withPrivacyMode(client, privacyMode, false) === null;
+		const shouldRedact = isPrivacyModeEnabled(client, privacyMode);
 		const safeInput = shouldRedact ? null : toJsonSafeValue(options.input);
 		const safeOutput = shouldRedact ? null : toJsonSafeValue(options.output);
 		let httpStatus = options.httpStatus;
@@ -745,6 +779,89 @@ function monitoredStreamTee(source, createStream) {
 	}), controller)];
 }
 //#endregion
+//#region src/toolResult.ts
+const MAX_DEPTH = 20;
+const MAX_ITEMS = 1e3;
+const MAX_NODES = 1e4;
+const TRUNCATED = "[Truncated]";
+const UNSERIALIZABLE = "[Unserializable]";
+const redactor = new BinaryContentRedactor();
+const propertyIsEnumerable = Object.prototype.propertyIsEnumerable;
+/** @internal Preserve tool-result structure while bounding each string value. */
+function formatToolResult(content, client, maxBytes = 5e3) {
+	try {
+		if (isFullAiCaptureEnabled(client)) return toJsonSafeValue(content);
+		const ancestors = /* @__PURE__ */ new WeakSet();
+		let remainingNodes = MAX_NODES;
+		const convert = (value, depth, context) => {
+			if (remainingNodes <= 0) return TRUNCATED;
+			remainingNodes--;
+			try {
+				if (value === null || typeof value !== "object") {
+					const safe = toJsonSafeValue(value);
+					return typeof safe === "string" ? truncate(redactor.redactLeaf(safe, context), void 0, maxBytes) : safe;
+				}
+				if (depth >= MAX_DEPTH) return TRUNCATED;
+				if (typeof Uint8Array !== "undefined" && value instanceof Uint8Array) return redactor.redactLeaf(value, context);
+				if (ancestors.has(value)) return "[Circular]";
+				ancestors.add(value);
+				try {
+					if (value instanceof Date) {
+						const safe = toJsonSafeValue(value);
+						return typeof safe === "string" ? truncate(redactor.redactLeaf(safe, context), void 0, maxBytes) : safe;
+					}
+					let hasToJSONResult = false;
+					let toJSONResult;
+					try {
+						const toJSON = value.toJSON;
+						if (typeof toJSON === "function") {
+							toJSONResult = toJSON.call(value);
+							hasToJSONResult = true;
+						}
+					} catch {}
+					if (hasToJSONResult) return convert(toJSONResult, depth + 1, context);
+					if (Array.isArray(value)) {
+						const output = [];
+						const count = Math.min(value.length, MAX_ITEMS);
+						let index = 0;
+						for (; index < count && remainingNodes > 0; index++) output.push(convert(value[index], depth + 1, context));
+						if (value.length > index) output.push(TRUNCATED);
+						return output;
+					}
+					const input = value;
+					const output = {};
+					let count = 0;
+					let truncated = false;
+					for (const key in input) {
+						if (!propertyIsEnumerable.call(input, key)) break;
+						if (count >= MAX_ITEMS || remainingNodes <= 0) {
+							truncated = true;
+							break;
+						}
+						const converted = convert(input[key], depth + 1, new MediaTypeContext(input, key));
+						Object.defineProperty(output, key, {
+							value: converted,
+							enumerable: true,
+							writable: true,
+							configurable: true
+						});
+						count++;
+					}
+					if (truncated) output[TRUNCATED] = "Additional properties omitted";
+					return output;
+				} finally {
+					ancestors.delete(value);
+				}
+			} catch {
+				return UNSERIALIZABLE;
+			}
+		};
+		return convert(content, 0, MediaTypeContext.EMPTY);
+	} catch {
+		return UNSERIALIZABLE;
+	}
+}
+//#endregion
 //#region src/anthropic/index.ts
 var PostHogAnthropic = class extends AnthropicOriginal {
 	constructor(config) {
@@ -760,6 +877,30 @@ var WrappedMessages = class extends AnthropicOriginal.Messages {
 		this.phClient = phClient;
 		this.baseURL = parentClient.baseURL;
 	}
+	formatInputForPostHog(params, privacyMode) {
+		try {
+			if (isPrivacyModeEnabled(this.phClient, privacyMode)) return null;
+			const input = mergeSystemPrompt(params, "anthropic");
+			const messages = Array.isArray(input) ? input.map((message) => {
+				if (!isObject(message) || !Array.isArray(message.content)) return message;
+				const originalContent = message.content;
+				const content = originalContent.map((block) => {
+					if (isObject(block) && typeof block.type === "string" && (block.type === "tool_result" || block.type.endsWith("_tool_result")) && Object.hasOwn(block, "content")) return {
+						...block,
+						content: formatToolResult(block.content, this.phClient, Infinity)
+					};
+					return block;
+				});
+				return content.some((block, index) => block !== originalContent[index]) ? {
+					...message,
+					content
+				} : message;
+			}) : input;
+			return sanitizeAnthropic(messages, this.phClient);
+		} catch {
+			return "[Unserializable]";
+		}
+	}
 	create(body, options) {
 		const { providerParams: anthropicParams, posthogParams } = extractPosthogParams(body);
 		const startTime = Date.now();
@@ -866,7 +1007,7 @@ var WrappedMessages = class extends AnthropicOriginal.Messages {
 								...posthogParams,
 								model: anthropicParams.model,
 								provider: "anthropic",
-								input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams, "anthropic"), this.phClient),
+								input: this.formatInputForPostHog(anthropicParams, posthogParams.privacyMode),
 								output: formattedOutput,
 								latency,
 								timeToFirstToken,
@@ -883,7 +1024,7 @@ var WrappedMessages = class extends AnthropicOriginal.Messages {
 								...posthogParams,
 								model: anthropicParams.model,
 								provider: "anthropic",
-								input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams, "anthropic"), this.phClient),
+								input: this.formatInputForPostHog(anthropicParams, posthogParams.privacyMode),
 								output: [],
 								latency: (Date.now() - startTime) / 1e3,
 								baseURL: this.baseURL,
@@ -911,7 +1052,7 @@ var WrappedMessages = class extends AnthropicOriginal.Messages {
 						...posthogParams,
 						model: anthropicParams.model,
 						provider: "anthropic",
-						input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams, "anthropic"), this.phClient),
+						input: this.formatInputForPostHog(anthropicParams, posthogParams.privacyMode),
 						output: formatResponseAnthropic(result),
 						latency,
 						baseURL: this.baseURL,
@@ -935,7 +1076,7 @@ var WrappedMessages = class extends AnthropicOriginal.Messages {
 					...posthogParams,
 					model: anthropicParams.model,
 					provider: "anthropic",
-					input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams, "anthropic"), this.phClient),
+					input: this.formatInputForPostHog(anthropicParams, posthogParams.privacyMode),
 					output: [],
 					latency: (Date.now() - startTime) / 1e3,
 					baseURL: this.baseURL,