@contractspec/lib.ai-agent 1.61.0 → 2.0.0

package/dist/node/telemetry/index.js

@@ -76,11 +76,127 @@ var init_adapter = __esm(() => {
   };
 });
 
+// src/telemetry/posthog.ts
+var exports_posthog = {};
+__export(exports_posthog, {
+  createPostHogTracedModel: () => createPostHogTracedModel,
+  createPostHogTelemetryCollector: () => createPostHogTelemetryCollector,
+  createCompositeTelemetryCollector: () => createCompositeTelemetryCollector,
+  PostHogTelemetryCollector: () => PostHogTelemetryCollector,
+  CompositeTelemetryCollector: () => CompositeTelemetryCollector
+});
+async function createPostHogTracedModel(model, config, overrides) {
+  const { withTracing } = await importPostHogAI();
+  const phClient = await resolvePostHogClient(config);
+  const tracingOptions = {
+    ...config.defaults,
+    ...overrides
+  };
+  return withTracing(model, phClient, tracingOptions);
+}
+
+class PostHogTelemetryCollector {
+  phClient;
+  config;
+  initPromise;
+  constructor(config) {
+    this.config = config;
+    this.phClient = config.client;
+  }
+  async collect(sample) {
+    const client = await this.getClient();
+    const distinctId = this.config.defaults?.posthogDistinctId ?? sample.metadata?.["actorId"] ?? "system";
+    client.capture({
+      distinctId,
+      event: "$ai_generation",
+      properties: {
+        $ai_model: sample.operation.name,
+        $ai_provider: "contractspec",
+        $ai_latency: sample.durationMs / 1000,
+        $ai_is_error: !sample.success,
+        $ai_trace_id: this.config.defaults?.posthogTraceId,
+        ...sample.metadata?.["tokenUsage"] ? mapTokenUsage(sample.metadata["tokenUsage"]) : {},
+        ...this.config.defaults?.posthogProperties,
+        contractspec_operation: sample.operation.name,
+        contractspec_version: sample.operation.version,
+        contractspec_agent_id: sample.metadata?.["agentId"],
+        contractspec_finish_reason: sample.metadata?.["finishReason"],
+        contractspec_tool_count: sample.metadata?.["toolCallCount"]
+      },
+      groups: this.config.defaults?.posthogGroups
+    });
+  }
+  async shutdown() {
+    if (this.phClient?.shutdown) {
+      await this.phClient.shutdown();
+    }
+  }
+  async getClient() {
+    if (this.phClient)
+      return this.phClient;
+    if (!this.initPromise) {
+      this.initPromise = resolvePostHogClient(this.config).then((client) => {
+        this.phClient = client;
+        return client;
+      });
+    }
+    return this.initPromise;
+  }
+}
+function createPostHogTelemetryCollector(config) {
+  return new PostHogTelemetryCollector(config);
+}
+
+class CompositeTelemetryCollector {
+  collectors;
+  constructor(collectors) {
+    this.collectors = collectors;
+  }
+  async collect(sample) {
+    await Promise.all(this.collectors.map((c) => c.collect(sample)));
+  }
+}
+function createCompositeTelemetryCollector(collectors) {
+  return new CompositeTelemetryCollector(collectors);
+}
+async function importPostHogAI() {
+  try {
+    return await import("@posthog/ai");
+  } catch {
+    throw new Error("PostHog LLM Analytics requires @posthog/ai to be installed. Run: npm install @posthog/ai posthog-node");
+  }
+}
+async function resolvePostHogClient(config) {
+  if (config.client)
+    return config.client;
+  if (!config.apiKey) {
+    throw new Error("PostHog LLM Analytics requires either a client instance or an apiKey.");
+  }
+  try {
+    const { PostHog } = await import("posthog-node");
+    return new PostHog(config.apiKey, {
+      host: config.host ?? "https://us.i.posthog.com"
+    });
+  } catch {
+    throw new Error("PostHog LLM Analytics requires posthog-node to be installed. Run: npm install posthog-node");
+  }
+}
+function mapTokenUsage(usage) {
+  return {
+    $ai_input_tokens: usage["promptTokens"],
+    $ai_output_tokens: usage["completionTokens"]
+  };
+}
 // src/telemetry/index.ts
 init_adapter();
 export {
   trackAgentStep,
   noopTelemetryCollector,
+  createPostHogTracedModel,
+  createPostHogTelemetryCollector,
   createInMemoryTelemetryCollector,
-  InMemoryTelemetryCollector
+  createCompositeTelemetryCollector,
+  PostHogTelemetryCollector,
+  InMemoryTelemetryCollector,
+  CompositeTelemetryCollector
 };

package/dist/node/telemetry/posthog.js

@@ -0,0 +1,132 @@
+import { createRequire } from "node:module";
+var __defProp = Object.defineProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, {
+      get: all[name],
+      enumerable: true,
+      configurable: true,
+      set: (newValue) => all[name] = () => newValue
+    });
+};
+var __esm = (fn, res) => () => (fn && (res = fn(fn = 0)), res);
+var __require = /* @__PURE__ */ createRequire(import.meta.url);
+
+// src/telemetry/posthog.ts
+var exports_posthog = {};
+__export(exports_posthog, {
+  createPostHogTracedModel: () => createPostHogTracedModel,
+  createPostHogTelemetryCollector: () => createPostHogTelemetryCollector,
+  createCompositeTelemetryCollector: () => createCompositeTelemetryCollector,
+  PostHogTelemetryCollector: () => PostHogTelemetryCollector,
+  CompositeTelemetryCollector: () => CompositeTelemetryCollector
+});
+async function createPostHogTracedModel(model, config, overrides) {
+  const { withTracing } = await importPostHogAI();
+  const phClient = await resolvePostHogClient(config);
+  const tracingOptions = {
+    ...config.defaults,
+    ...overrides
+  };
+  return withTracing(model, phClient, tracingOptions);
+}
+
+class PostHogTelemetryCollector {
+  phClient;
+  config;
+  initPromise;
+  constructor(config) {
+    this.config = config;
+    this.phClient = config.client;
+  }
+  async collect(sample) {
+    const client = await this.getClient();
+    const distinctId = this.config.defaults?.posthogDistinctId ?? sample.metadata?.["actorId"] ?? "system";
+    client.capture({
+      distinctId,
+      event: "$ai_generation",
+      properties: {
+        $ai_model: sample.operation.name,
+        $ai_provider: "contractspec",
+        $ai_latency: sample.durationMs / 1000,
+        $ai_is_error: !sample.success,
+        $ai_trace_id: this.config.defaults?.posthogTraceId,
+        ...sample.metadata?.["tokenUsage"] ? mapTokenUsage(sample.metadata["tokenUsage"]) : {},
+        ...this.config.defaults?.posthogProperties,
+        contractspec_operation: sample.operation.name,
+        contractspec_version: sample.operation.version,
+        contractspec_agent_id: sample.metadata?.["agentId"],
+        contractspec_finish_reason: sample.metadata?.["finishReason"],
+        contractspec_tool_count: sample.metadata?.["toolCallCount"]
+      },
+      groups: this.config.defaults?.posthogGroups
+    });
+  }
+  async shutdown() {
+    if (this.phClient?.shutdown) {
+      await this.phClient.shutdown();
+    }
+  }
+  async getClient() {
+    if (this.phClient)
+      return this.phClient;
+    if (!this.initPromise) {
+      this.initPromise = resolvePostHogClient(this.config).then((client) => {
+        this.phClient = client;
+        return client;
+      });
+    }
+    return this.initPromise;
+  }
+}
+function createPostHogTelemetryCollector(config) {
+  return new PostHogTelemetryCollector(config);
+}
+
+class CompositeTelemetryCollector {
+  collectors;
+  constructor(collectors) {
+    this.collectors = collectors;
+  }
+  async collect(sample) {
+    await Promise.all(this.collectors.map((c) => c.collect(sample)));
+  }
+}
+function createCompositeTelemetryCollector(collectors) {
+  return new CompositeTelemetryCollector(collectors);
+}
+async function importPostHogAI() {
+  try {
+    return await import("@posthog/ai");
+  } catch {
+    throw new Error("PostHog LLM Analytics requires @posthog/ai to be installed. Run: npm install @posthog/ai posthog-node");
+  }
+}
+async function resolvePostHogClient(config) {
+  if (config.client)
+    return config.client;
+  if (!config.apiKey) {
+    throw new Error("PostHog LLM Analytics requires either a client instance or an apiKey.");
+  }
+  try {
+    const { PostHog } = await import("posthog-node");
+    return new PostHog(config.apiKey, {
+      host: config.host ?? "https://us.i.posthog.com"
+    });
+  } catch {
+    throw new Error("PostHog LLM Analytics requires posthog-node to be installed. Run: npm install posthog-node");
+  }
+}
+function mapTokenUsage(usage) {
+  return {
+    $ai_input_tokens: usage["promptTokens"],
+    $ai_output_tokens: usage["completionTokens"]
+  };
+}
+export {
+  createPostHogTracedModel,
+  createPostHogTelemetryCollector,
+  createCompositeTelemetryCollector,
+  PostHogTelemetryCollector,
+  CompositeTelemetryCollector
+};