@posthog/agent 2.3.655 → 2.3.657

package/src/utils/gateway.ts

@@ -1,4 +1,34 @@
-export type GatewayProduct = "posthog_code" | "background_agents";
+export type GatewayProduct = "posthog_code" | "background_agents" | "signals";
+
+export function resolveGatewayProduct({
+  isInternal,
+  originProduct,
+}: {
+  isInternal?: boolean;
+  originProduct?: string | null;
+} = {}): GatewayProduct {
+  if (isInternal) {
+    return originProduct === "signal_report" ? "signals" : "background_agents";
+  }
+  return "posthog_code";
+}
+
+/**
+ * Build `x-posthog-property-<name>: <value>` header lines that the LLM
+ * gateway lifts onto the `$ai_generation` event it captures for each call
+ * (see `services/llm-gateway/src/llm_gateway/request_context.py`).
+ *
+ * Returns a newline-joined string ready for `ANTHROPIC_CUSTOM_HEADERS`.
+ * `null`/`undefined` property values are dropped.
+ */
+export function buildGatewayPropertyHeaders(
+  properties: Record<string, string | number | boolean | null | undefined>,
+): string {
+  return Object.entries(properties)
+    .filter(([, value]) => value !== null && value !== undefined)
+    .map(([key, value]) => `x-posthog-property-${key}: ${value}`)
+    .join("\n");
+}
 
 function getGatewayBaseUrl(posthogHost: string): string {
   const url = new URL(posthogHost);