@posthog/ai 8.14.0 → 8.14.2

package/dist/claude-agent-sdk/index.cjs

@@ -3,7 +3,7 @@ let _anthropic_ai_claude_agent_sdk = require("@anthropic-ai/claude-agent-sdk");
 let uuid = require("uuid");
 let _posthog_core = require("@posthog/core");
 //#region package.json
-var version = "8.14.0";
+var version = "8.14.2";
 //#endregion
 //#region src/captureAiEvent.ts
 /** @internal */
@@ -170,17 +170,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) {
@@ -200,11 +209,7 @@ var BinaryContentRedactor = class {
 		return `[base64 ${mediaType} redacted]`;
 	}
 };
-//#endregion
-//#region src/sanitization.ts
-const redactor = new BinaryContentRedactor();
-const sanitize = (data, client) => isFullAiCaptureEnabled(client) ? data : redactor.redact(data);
-const sanitizeAnthropic = (data, client) => sanitize(data, client);
+new BinaryContentRedactor();
 //#endregion
 //#region src/utils.ts
 const TOKEN_PROPERTY_KEYS = /* @__PURE__ */ new Set([
@@ -253,9 +258,7 @@ function toContentString(content) {
 	}
 	return String(content);
 }
-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;
@@ -371,7 +374,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 : (0, _posthog_core.toJsonSafeValue)(options.input);
 		const safeOutput = shouldRedact ? null : (0, _posthog_core.toJsonSafeValue)(options.output);
 		let httpStatus = options.httpStatus;
@@ -453,22 +456,92 @@ const captureAiGeneration = async (client, options) => {
 	}
 };
 //#endregion
-//#region src/claude-agent-sdk/formatting.ts
-/**
-* Tool results in an agent run carry whole files and command output, and the
-* same result is replayed as input on every later turn, so each string is
-* capped before it reaches the event.
-*/
-const TOOL_RESULT_MAX_STRING_LENGTH = 5e3;
-function capStrings(value, max) {
-	if (typeof value === "string") return truncate(value, void 0, max);
-	if (Array.isArray(value)) return value.map((item) => capStrings(item, max));
-	if (value && typeof value === "object") return Object.fromEntries(Object.entries(value).map(([key, item]) => [key, capStrings(item, max)]));
-	return value;
+//#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 (0, _posthog_core.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 = (0, _posthog_core.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 = (0, _posthog_core.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/claude-agent-sdk/formatting.ts
 function formatContent(content, client, maxBytes = MAX_OUTPUT_SIZE) {
-	const redacted = sanitizeAnthropic((0, _posthog_core.toJsonSafeValue)(content), client);
-	return isFullAiCaptureEnabled(client) ? redacted : capStrings(redacted, maxBytes);
+	return formatToolResult(content, client, maxBytes);
 }
 /** Read the system prompt out of the SDK options, whatever shape it takes. */
 function extractSystemPrompt(options) {
@@ -507,7 +580,8 @@ function formatAssistantBlocks(blocks, client) {
 	return content;
 }
 /** Convert an Anthropic user message body into PostHog content items. */
-function formatUserContent(content, client) {
+function formatUserContent(content, client, privacyMode = false) {
+	if (isPrivacyModeEnabled(client, privacyMode)) return [];
 	if (typeof content === "string") return content;
 	if (!Array.isArray(content)) return [];
 	const formatted = [];
@@ -516,7 +590,7 @@ function formatUserContent(content, client) {
 		if (block.type === "tool_result") formatted.push({
 			type: "tool_result",
 			tool_use_id: block.tool_use_id,
-			content: formatContent(block.content, client, TOOL_RESULT_MAX_STRING_LENGTH),
+			content: formatToolResult(block.content, client),
 			...typeof block.is_error === "boolean" ? { is_error: block.is_error } : {}
 		});
 		else if (typeof block.text === "string") formatted.push({
@@ -820,7 +894,7 @@ var PostHogClaudeAgentProcessor = class {
 			yield message;
 		}
 	}
-	_beginTurn(state, userMessageUuid) {
+	_beginTurn(state, trace, userMessageUuid) {
 		if (state.turnActive) return;
 		state.turnActive = true;
 		let end = userMessageUuid ? state.pendingPrompts.findIndex((message) => message.uuid === userMessageUuid) : -1;
@@ -834,7 +908,7 @@ var PostHogClaudeAgentProcessor = class {
 		while (start > 0 && state.pendingPrompts[start - 1].shouldQuery === false) start -= 1;
 		for (const message of state.pendingPrompts.splice(start, end - start + 1)) state.tracker.appendPendingInput([{
 			role: "user",
-			content: formatUserContent(message.message.content, this._client)
+			content: formatUserContent(message.message.content, this._client, trace.privacyMode)
 		}]);
 	}
 	async _handleMessage(message, state, trace) {
@@ -846,11 +920,11 @@ var PostHogClaudeAgentProcessor = class {
 		} else if (message.type === "stream_event") {
 			if (message.parent_tool_use_id) return;
 			if (message.user_message_uuid) state.userMessageUuid = message.user_message_uuid;
-			if (message.event.type === "message_start") this._beginTurn(state, message.user_message_uuid);
+			if (message.event.type === "message_start") this._beginTurn(state, trace, message.user_message_uuid);
 			state.tracker.processStreamEvent(message.event, this._client, message.ttft_ms);
 			await this._captureCompletedGenerations(state, trace);
 		} else if (message.type === "assistant") {
-			if (!message.parent_tool_use_id) this._beginTurn(state, message.user_message_uuid);
+			if (!message.parent_tool_use_id) this._beginTurn(state, trace, message.user_message_uuid);
 			this._handleAssistantMessage(message, state, trace);
 		} else if (message.type === "user") {
 			if ("isReplay" in message && message.isReplay) return;
@@ -858,13 +932,13 @@ var PostHogClaudeAgentProcessor = class {
 				for (const block of message.message.content) if (block.type === "tool_result") await this._finishToolSpan(block.tool_use_id, state, trace, block);
 			}
 			if (message.parent_tool_use_id) return;
-			const content = formatUserContent(message.message?.content, this._client);
+			const content = formatUserContent(message.message?.content, this._client, trace.privacyMode);
 			if (typeof content === "string" || Array.isArray(content) && content.length > 0) state.tracker.appendPendingInput([{
 				role: "user",
 				content
 			}]);
 		} else if (message.type === "result") try {
-			this._beginTurn(state, message.user_message_uuid);
+			this._beginTurn(state, trace, message.user_message_uuid);
 			state.tracker.finishCurrent();
 			await this._captureCompletedGenerations(state, trace, resultError(message), message);
 			if (message.total_cost_usd != null) {
@@ -954,7 +1028,7 @@ var PostHogClaudeAgentProcessor = class {
 			...parentSpanId ? { $ai_parent_id: parentSpanId } : {},
 			$ai_span_name: block.name,
 			$ai_span_type: "tool",
-			$ai_input_state: withPrivacyMode(this._client, trace.privacyMode ?? false, formatContent(block.input ?? {}, this._client))
+			$ai_input_state: isPrivacyModeEnabled(this._client, trace.privacyMode ?? false) ? null : formatContent(block.input ?? {}, this._client)
 		};
 		state.pendingTools.set(properties.$ai_span_id, {
 			startTime: performance.now(),
@@ -969,7 +1043,7 @@ var PostHogClaudeAgentProcessor = class {
 			...tool.properties,
 			$ai_latency: (performance.now() - tool.startTime) / 1e3,
 			...result ? {
-				$ai_output_state: withPrivacyMode(this._client, trace.privacyMode ?? false, formatContent(result.content, this._client)),
+				$ai_output_state: isPrivacyModeEnabled(this._client, trace.privacyMode ?? false) ? null : formatContent(result.content, this._client),
 				$ai_is_error: result.is_error === true
 			} : {}
 		};