@contractspec/lib.ai-agent 1.62.0 → 2.1.0

package/dist/session/store.js

@@ -15,10 +15,19 @@ var __require = import.meta.require;
 // src/session/store.ts
 class InMemorySessionStore {
   sessions = new Map;
+  maxSessions;
+  maxMessagesPerSession;
+  maxStepsPerSession;
+  constructor(options = {}) {
+    this.maxSessions = options.maxSessions ?? 500;
+    this.maxMessagesPerSession = options.maxMessagesPerSession ?? 200;
+    this.maxStepsPerSession = options.maxStepsPerSession ?? 200;
+  }
   async get(sessionId) {
     return this.sessions.get(sessionId) ?? null;
   }
   async create(session) {
+    this.evictIfNeeded();
     const now = new Date;
     const fullSession = {
       ...session,
@@ -32,6 +41,7 @@ class InMemorySessionStore {
     const session = this.sessions.get(sessionId);
     if (session) {
       session.steps.push(step);
+      this.trimArray(session.steps, this.maxStepsPerSession);
       session.updatedAt = new Date;
     }
   }
@@ -39,6 +49,7 @@ class InMemorySessionStore {
     const session = this.sessions.get(sessionId);
     if (session) {
       session.messages.push(message);
+      this.trimArray(session.messages, this.maxMessagesPerSession);
       session.updatedAt = new Date;
     }
   }
@@ -76,9 +87,33 @@ class InMemorySessionStore {
   clear() {
     this.sessions.clear();
   }
+  evictIfNeeded() {
+    if (this.sessions.size < this.maxSessions) {
+      return;
+    }
+    let oldestSessionId = null;
+    let oldestUpdatedAt = Number.POSITIVE_INFINITY;
+    for (const [sessionId, session] of this.sessions.entries()) {
+      const updatedAt = session.updatedAt.getTime();
+      if (updatedAt < oldestUpdatedAt) {
+        oldestUpdatedAt = updatedAt;
+        oldestSessionId = sessionId;
+      }
+    }
+    if (oldestSessionId) {
+      this.sessions.delete(oldestSessionId);
+    }
+  }
+  trimArray(items, maxItems) {
+    if (items.length <= maxItems) {
+      return;
+    }
+    const overflow = items.length - maxItems;
+    items.splice(0, overflow);
+  }
 }
-function createInMemorySessionStore() {
-  return new InMemorySessionStore;
+function createInMemorySessionStore(options) {
+  return new InMemorySessionStore(options);
 }
 function generateSessionId() {
   return `sess_${Date.now()}_${Math.random().toString(36).slice(2, 11)}`;

package/dist/spec/index.d.ts

@@ -1,3 +1,2 @@
 export * from './spec';
 export * from './registry';
-//# sourceMappingURL=index.d.ts.map

package/dist/spec/index.js

@@ -13,7 +13,7 @@ var __esm = (fn, res) => () => (fn && (res = fn(fn = 0)), res);
 var __require = import.meta.require;
 
 // src/i18n/catalogs/en.ts
-import { defineTranslation } from "@contractspec/lib.contracts/translations";
+import { defineTranslation } from "@contractspec/lib.contracts-spec/translations";
 var enMessages;
 var init_en = __esm(() => {
   enMessages = defineTranslation({
@@ -609,7 +609,7 @@ var init_en = __esm(() => {
 });
 
 // src/i18n/catalogs/fr.ts
-import { defineTranslation as defineTranslation2 } from "@contractspec/lib.contracts/translations";
+import { defineTranslation as defineTranslation2 } from "@contractspec/lib.contracts-spec/translations";
 var frMessages;
 var init_fr = __esm(() => {
   frMessages = defineTranslation2({
@@ -1220,7 +1220,7 @@ var init_fr = __esm(() => {
 });
 
 // src/i18n/catalogs/es.ts
-import { defineTranslation as defineTranslation3 } from "@contractspec/lib.contracts/translations";
+import { defineTranslation as defineTranslation3 } from "@contractspec/lib.contracts-spec/translations";
 var esMessages;
 var init_es = __esm(() => {
   esMessages = defineTranslation3({
@@ -1885,7 +1885,7 @@ var init_locale = __esm(() => {
 });
 
 // src/i18n/messages.ts
-import { TranslationRegistry } from "@contractspec/lib.contracts/translations";
+import { TranslationRegistry } from "@contractspec/lib.contracts-spec/translations";
 function getRegistry() {
   if (!sharedRegistry) {
     sharedRegistry = new TranslationRegistry([
@@ -2140,7 +2140,7 @@ var init_spec = __esm(() => {
 // src/spec/registry.ts
 init_i18n();
 import { compareVersions } from "compare-versions";
-import { SpecContractRegistry } from "@contractspec/lib.contracts/registry";
+import { SpecContractRegistry } from "@contractspec/lib.contracts-spec/registry";
 
 class AgentRegistry extends SpecContractRegistry {
   constructor(items) {

package/dist/spec/registry.d.ts

@@ -1,5 +1,5 @@
 import type { AgentSpec } from './spec';
-import { SpecContractRegistry } from '@contractspec/lib.contracts/registry';
+import { SpecContractRegistry } from '@contractspec/lib.contracts-spec/registry';
 /**
  * Registry for managing agent specifications.
  *
@@ -39,4 +39,3 @@ export declare class AgentRegistry extends SpecContractRegistry<'agent', AgentSp
  * Create a new agent registry.
  */
 export declare function createAgentRegistry(): AgentRegistry;
-//# sourceMappingURL=registry.d.ts.map

package/dist/spec/registry.js

@@ -13,7 +13,7 @@ var __esm = (fn, res) => () => (fn && (res = fn(fn = 0)), res);
 var __require = import.meta.require;
 
 // src/i18n/catalogs/en.ts
-import { defineTranslation } from "@contractspec/lib.contracts/translations";
+import { defineTranslation } from "@contractspec/lib.contracts-spec/translations";
 var enMessages;
 var init_en = __esm(() => {
   enMessages = defineTranslation({
@@ -609,7 +609,7 @@ var init_en = __esm(() => {
 });
 
 // src/i18n/catalogs/fr.ts
-import { defineTranslation as defineTranslation2 } from "@contractspec/lib.contracts/translations";
+import { defineTranslation as defineTranslation2 } from "@contractspec/lib.contracts-spec/translations";
 var frMessages;
 var init_fr = __esm(() => {
   frMessages = defineTranslation2({
@@ -1220,7 +1220,7 @@ var init_fr = __esm(() => {
 });
 
 // src/i18n/catalogs/es.ts
-import { defineTranslation as defineTranslation3 } from "@contractspec/lib.contracts/translations";
+import { defineTranslation as defineTranslation3 } from "@contractspec/lib.contracts-spec/translations";
 var esMessages;
 var init_es = __esm(() => {
   esMessages = defineTranslation3({
@@ -1885,7 +1885,7 @@ var init_locale = __esm(() => {
 });
 
 // src/i18n/messages.ts
-import { TranslationRegistry } from "@contractspec/lib.contracts/translations";
+import { TranslationRegistry } from "@contractspec/lib.contracts-spec/translations";
 function getRegistry() {
   if (!sharedRegistry) {
     sharedRegistry = new TranslationRegistry([
@@ -2106,7 +2106,7 @@ var init_i18n = __esm(() => {
 // src/spec/registry.ts
 init_i18n();
 import { compareVersions } from "compare-versions";
-import { SpecContractRegistry } from "@contractspec/lib.contracts/registry";
+import { SpecContractRegistry } from "@contractspec/lib.contracts-spec/registry";
 
 class AgentRegistry extends SpecContractRegistry {
   constructor(items) {

package/dist/spec/spec.d.ts

@@ -1,6 +1,6 @@
-import type { OwnerShipMeta } from '@contractspec/lib.contracts/ownership';
-import type { KnowledgeCategory } from '@contractspec/lib.contracts/knowledge/spec';
-import type { PolicyRef } from '@contractspec/lib.contracts/policy/spec';
+import type { OwnerShipMeta } from '@contractspec/lib.contracts-spec/ownership';
+import type { KnowledgeCategory } from '@contractspec/lib.contracts-spec/knowledge/spec';
+import type { PolicyRef } from '@contractspec/lib.contracts-spec/policy/spec';
 /**
  * Metadata for an agent specification.
  */
@@ -121,4 +121,3 @@ export declare function defineAgent(spec: AgentSpec): AgentSpec;
  * Generate a unique key for an agent spec.
  */
 export declare function agentKey(meta: AgentMeta): string;
-//# sourceMappingURL=spec.d.ts.map

package/dist/spec/spec.js

@@ -13,7 +13,7 @@ var __esm = (fn, res) => () => (fn && (res = fn(fn = 0)), res);
 var __require = import.meta.require;
 
 // src/i18n/catalogs/en.ts
-import { defineTranslation } from "@contractspec/lib.contracts/translations";
+import { defineTranslation } from "@contractspec/lib.contracts-spec/translations";
 var enMessages;
 var init_en = __esm(() => {
   enMessages = defineTranslation({
@@ -609,7 +609,7 @@ var init_en = __esm(() => {
 });
 
 // src/i18n/catalogs/fr.ts
-import { defineTranslation as defineTranslation2 } from "@contractspec/lib.contracts/translations";
+import { defineTranslation as defineTranslation2 } from "@contractspec/lib.contracts-spec/translations";
 var frMessages;
 var init_fr = __esm(() => {
   frMessages = defineTranslation2({
@@ -1220,7 +1220,7 @@ var init_fr = __esm(() => {
 });
 
 // src/i18n/catalogs/es.ts
-import { defineTranslation as defineTranslation3 } from "@contractspec/lib.contracts/translations";
+import { defineTranslation as defineTranslation3 } from "@contractspec/lib.contracts-spec/translations";
 var esMessages;
 var init_es = __esm(() => {
   esMessages = defineTranslation3({
@@ -1885,7 +1885,7 @@ var init_locale = __esm(() => {
 });
 
 // src/i18n/messages.ts
-import { TranslationRegistry } from "@contractspec/lib.contracts/translations";
+import { TranslationRegistry } from "@contractspec/lib.contracts-spec/translations";
 function getRegistry() {
   if (!sharedRegistry) {
     sharedRegistry = new TranslationRegistry([

package/dist/telemetry/adapter.d.ts

@@ -26,6 +26,14 @@ export interface TelemetryCollector {
      */
     collect(sample: OperationMetricSample): Promise<void>;
 }
+interface TrackAgentStepContext {
+    sessionId?: string;
+    tenantId?: string;
+    actorId?: string;
+    traceId?: string;
+    stepIndex?: number;
+    stepStartedAt?: Date;
+}
 /**
  * Track an agent step for telemetry.
  *
@@ -37,7 +45,7 @@ export interface TelemetryCollector {
  * @param step - AI SDK step result
  * @param durationMs - Optional step duration in milliseconds
  */
-export declare function trackAgentStep(collector: TelemetryCollector, agentId: string, step: StepResult<ToolSet>, durationMs?: number): Promise<void>;
+export declare function trackAgentStep(collector: TelemetryCollector, agentId: string, step: StepResult<ToolSet>, durationMs?: number, context?: TrackAgentStepContext): Promise<void>;
 /**
  * In-memory telemetry collector for testing.
  */
@@ -65,4 +73,4 @@ export declare function createInMemoryTelemetryCollector(): InMemoryTelemetryCol
  * No-op telemetry collector that discards all metrics.
  */
 export declare const noopTelemetryCollector: TelemetryCollector;
-//# sourceMappingURL=adapter.d.ts.map
+export {};

package/dist/telemetry/adapter.js

@@ -14,24 +14,78 @@ var __require = import.meta.require;
 
 // src/telemetry/adapter.ts
 function parseAgentId(agentId) {
-  const match = agentId.match(/^(.+)\.v(\s+)$/);
+  const match = agentId.match(/^(.+)\.v(\d+)$/);
   if (match) {
     return { name: match[1], version: match[2] };
   }
   return { name: agentId, version: "1.0.0" };
 }
-async function trackAgentStep(collector, agentId, step, durationMs) {
+function getRecord(value) {
+  return value && typeof value === "object" ? value : undefined;
+}
+function getStepResponse(step) {
+  const response = getRecord(step.response);
+  if (!response)
+    return;
+  return {
+    id: response["id"],
+    modelId: response["modelId"],
+    timestamp: response["timestamp"],
+    headers: response["headers"],
+    body: response["body"],
+    messages: response["messages"]
+  };
+}
+function getRequestBodyValue(request, key) {
+  const body = request?.["body"];
+  if (!body)
+    return;
+  if (typeof body === "string") {
+    try {
+      const parsed = JSON.parse(body);
+      return typeof parsed[key] === "string" ? parsed[key] : undefined;
+    } catch {
+      return;
+    }
+  }
+  const record = getRecord(body);
+  return typeof record?.[key] === "string" ? record[key] : undefined;
+}
+async function trackAgentStep(collector, agentId, step, durationMs, context) {
   const { name, version } = parseAgentId(agentId);
+  const response = getStepResponse(step);
+  const providerMetadata = step.providerMetadata;
+  const warnings = step.warnings;
+  const rawFinishReason = step.rawFinishReason;
+  const request = getRecord(step.request);
+  const traceId = context?.traceId ?? getRequestBodyValue(request, "$ai_trace_id");
+  const sessionId = context?.sessionId ?? getRequestBodyValue(request, "$ai_session_id");
+  const parentSpanId = response?.id;
   for (const toolCall of step.toolCalls ?? []) {
+    const toolResult = step.toolResults?.find((r) => r.toolCallId === toolCall.toolCallId);
+    const toolError = getRecord(toolResult?.output)?.["error"];
     const toolSample = {
       operation: { name: `${name}.${toolCall.toolName}`, version },
       durationMs: durationMs ?? 0,
       success: step.toolResults?.some((r) => r.toolCallId === toolCall.toolCallId && r.output !== undefined) ?? false,
       timestamp: new Date,
       metadata: {
+        telemetryEvent: "span",
+        traceId,
+        sessionId,
+        spanId: toolCall.toolCallId,
+        parentSpanId,
+        spanName: `tool.${toolCall.toolName}`,
         agentId,
+        actorId: context?.actorId,
+        tenantId: context?.tenantId,
+        stepIndex: context?.stepIndex,
         toolName: toolCall.toolName,
-        finishReason: step.finishReason
+        toolCallArgs: toolCall.input,
+        toolResultOutput: toolResult?.output,
+        errorMessage: typeof toolError === "string" ? toolError : typeof toolResult?.output === "string" ? toolResult.output : undefined,
+        finishReason: step.finishReason,
+        rawFinishReason
       }
     };
     await collector.collect(toolSample);
@@ -42,10 +96,36 @@ async function trackAgentStep(collector, agentId, step, durationMs) {
     success: step.finishReason !== "error",
     timestamp: new Date,
     metadata: {
+      telemetryEvent: "generation",
+      traceId,
+      sessionId,
+      spanId: response?.id,
+      spanName: `agent.${name}.step`,
       agentId,
+      actorId: context?.actorId,
+      tenantId: context?.tenantId,
+      stepIndex: context?.stepIndex,
+      stepStartedAt: context?.stepStartedAt,
       finishReason: step.finishReason,
+      rawFinishReason,
       tokenUsage: step.usage,
-      toolCallCount: step.toolCalls?.length ?? 0
+      totalUsage: step.totalUsage,
+      providerMetadata,
+      warnings,
+      stepText: step.text,
+      stepReasoningText: step.reasoningText,
+      responseId: response?.id,
+      responseModelId: response?.modelId,
+      responseTimestamp: response?.timestamp,
+      responseHeaders: response?.headers,
+      responseBody: response?.body,
+      responseMessages: response?.messages,
+      requestBody: request?.["body"],
+      requestHeaders: request?.["headers"],
+      toolCallCount: step.toolCalls?.length ?? 0,
+      toolCalls: step.toolCalls,
+      toolResults: step.toolResults,
+      errorMessage: step.finishReason === "error" ? step.text : undefined
     }
   };
   await collector.collect(stepSample);

package/dist/telemetry/adapter.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/telemetry/index.d.ts

@@ -1,4 +1,3 @@
 export * from './adapter';
 export * from './posthog-types';
 export * from './posthog';
-//# sourceMappingURL=index.d.ts.map

package/dist/telemetry/index.js

@@ -14,24 +14,78 @@ var __require = import.meta.require;
 
 // src/telemetry/adapter.ts
 function parseAgentId(agentId) {
-  const match = agentId.match(/^(.+)\.v(\s+)$/);
+  const match = agentId.match(/^(.+)\.v(\d+)$/);
   if (match) {
     return { name: match[1], version: match[2] };
   }
   return { name: agentId, version: "1.0.0" };
 }
-async function trackAgentStep(collector, agentId, step, durationMs) {
+function getRecord(value) {
+  return value && typeof value === "object" ? value : undefined;
+}
+function getStepResponse(step) {
+  const response = getRecord(step.response);
+  if (!response)
+    return;
+  return {
+    id: response["id"],
+    modelId: response["modelId"],
+    timestamp: response["timestamp"],
+    headers: response["headers"],
+    body: response["body"],
+    messages: response["messages"]
+  };
+}
+function getRequestBodyValue(request, key) {
+  const body = request?.["body"];
+  if (!body)
+    return;
+  if (typeof body === "string") {
+    try {
+      const parsed = JSON.parse(body);
+      return typeof parsed[key] === "string" ? parsed[key] : undefined;
+    } catch {
+      return;
+    }
+  }
+  const record = getRecord(body);
+  return typeof record?.[key] === "string" ? record[key] : undefined;
+}
+async function trackAgentStep(collector, agentId, step, durationMs, context) {
   const { name, version } = parseAgentId(agentId);
+  const response = getStepResponse(step);
+  const providerMetadata = step.providerMetadata;
+  const warnings = step.warnings;
+  const rawFinishReason = step.rawFinishReason;
+  const request = getRecord(step.request);
+  const traceId = context?.traceId ?? getRequestBodyValue(request, "$ai_trace_id");
+  const sessionId = context?.sessionId ?? getRequestBodyValue(request, "$ai_session_id");
+  const parentSpanId = response?.id;
   for (const toolCall of step.toolCalls ?? []) {
+    const toolResult = step.toolResults?.find((r) => r.toolCallId === toolCall.toolCallId);
+    const toolError = getRecord(toolResult?.output)?.["error"];
     const toolSample = {
       operation: { name: `${name}.${toolCall.toolName}`, version },
       durationMs: durationMs ?? 0,
       success: step.toolResults?.some((r) => r.toolCallId === toolCall.toolCallId && r.output !== undefined) ?? false,
       timestamp: new Date,
       metadata: {
+        telemetryEvent: "span",
+        traceId,
+        sessionId,
+        spanId: toolCall.toolCallId,
+        parentSpanId,
+        spanName: `tool.${toolCall.toolName}`,
         agentId,
+        actorId: context?.actorId,
+        tenantId: context?.tenantId,
+        stepIndex: context?.stepIndex,
         toolName: toolCall.toolName,
-        finishReason: step.finishReason
+        toolCallArgs: toolCall.input,
+        toolResultOutput: toolResult?.output,
+        errorMessage: typeof toolError === "string" ? toolError : typeof toolResult?.output === "string" ? toolResult.output : undefined,
+        finishReason: step.finishReason,
+        rawFinishReason
       }
     };
     await collector.collect(toolSample);
@@ -42,10 +96,36 @@ async function trackAgentStep(collector, agentId, step, durationMs) {
     success: step.finishReason !== "error",
     timestamp: new Date,
     metadata: {
+      telemetryEvent: "generation",
+      traceId,
+      sessionId,
+      spanId: response?.id,
+      spanName: `agent.${name}.step`,
       agentId,
+      actorId: context?.actorId,
+      tenantId: context?.tenantId,
+      stepIndex: context?.stepIndex,
+      stepStartedAt: context?.stepStartedAt,
       finishReason: step.finishReason,
+      rawFinishReason,
       tokenUsage: step.usage,
-      toolCallCount: step.toolCalls?.length ?? 0
+      totalUsage: step.totalUsage,
+      providerMetadata,
+      warnings,
+      stepText: step.text,
+      stepReasoningText: step.reasoningText,
+      responseId: response?.id,
+      responseModelId: response?.modelId,
+      responseTimestamp: response?.timestamp,
+      responseHeaders: response?.headers,
+      responseBody: response?.body,
+      responseMessages: response?.messages,
+      requestBody: request?.["body"],
+      requestHeaders: request?.["headers"],
+      toolCallCount: step.toolCalls?.length ?? 0,
+      toolCalls: step.toolCalls,
+      toolResults: step.toolResults,
+      errorMessage: step.finishReason === "error" ? step.text : undefined
     }
   };
   await collector.collect(stepSample);
@@ -105,23 +185,60 @@ class PostHogTelemetryCollector {
   }
   async collect(sample) {
     const client = await this.getClient();
-    const distinctId = this.config.defaults?.posthogDistinctId ?? sample.metadata?.["actorId"] ?? "system";
+    const metadata = asRecord(sample.metadata);
+    const distinctId = this.config.defaults?.posthogDistinctId ?? asString(metadata["actorId"]) ?? "system";
+    const traceId = asString(metadata["traceId"]) ?? this.config.defaults?.posthogTraceId;
+    const sessionId = asString(metadata["sessionId"]);
+    const telemetryEvent = asString(metadata["telemetryEvent"]);
+    const event = telemetryEvent === "span" ? "$ai_span" : "$ai_generation";
+    const tokenUsage = metadata["tokenUsage"];
+    const totalUsage = metadata["totalUsage"];
+    const errorMessage = asString(metadata["errorMessage"]);
     client.capture({
       distinctId,
-      event: "$ai_generation",
+      event,
       properties: {
-        $ai_model: sample.operation.name,
-        $ai_provider: "contractspec",
+        $ai_model: asString(metadata["responseModelId"]) ?? sample.operation.name,
+        $ai_provider: asString(metadata["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"]) : {},
+        $ai_error: !sample.success ? errorMessage : undefined,
+        $ai_trace_id: traceId,
+        $ai_session_id: sessionId,
+        $ai_span_id: asString(metadata["spanId"]),
+        $ai_parent_id: asString(metadata["parentSpanId"]),
+        $ai_span_name: asString(metadata["spanName"]) ?? sample.operation.name,
+        ...tokenUsage ? mapTokenUsage(tokenUsage) : {},
+        ...totalUsage ? mapTotalUsage(totalUsage) : {},
+        $ai_http_status: asNumber(metadata["httpStatus"]),
+        $ai_request_url: asString(metadata["requestUrl"]),
+        $ai_base_url: asString(metadata["baseUrl"]),
         ...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"]
+        contractspec_agent_id: asString(metadata["agentId"]),
+        contractspec_tenant_id: asString(metadata["tenantId"]),
+        contractspec_actor_id: asString(metadata["actorId"]),
+        contractspec_step_index: asNumber(metadata["stepIndex"]),
+        contractspec_step_started_at: asDateIso(metadata["stepStartedAt"]),
+        contractspec_finish_reason: asString(metadata["finishReason"]),
+        contractspec_finish_reason_raw: asString(metadata["rawFinishReason"]),
+        contractspec_tool_count: asNumber(metadata["toolCallCount"]),
+        contractspec_tool_name: asString(metadata["toolName"]),
+        contractspec_tool_call_args: metadata["toolCallArgs"],
+        contractspec_tool_result_output: metadata["toolResultOutput"],
+        contractspec_provider_metadata: metadata["providerMetadata"],
+        contractspec_step_warnings: metadata["warnings"],
+        contractspec_response_id: asString(metadata["responseId"]),
+        contractspec_response_model_id: asString(metadata["responseModelId"]),
+        contractspec_response_timestamp: asDateIso(metadata["responseTimestamp"]),
+        contractspec_response_headers: metadata["responseHeaders"],
+        contractspec_response_body: this.config.defaults?.posthogPrivacyMode ? undefined : metadata["responseBody"],
+        contractspec_response_messages: this.config.defaults?.posthogPrivacyMode ? undefined : metadata["responseMessages"],
+        contractspec_request_headers: metadata["requestHeaders"],
+        contractspec_request_body: this.config.defaults?.posthogPrivacyMode ? undefined : metadata["requestBody"],
+        contractspec_step_text: this.config.defaults?.posthogPrivacyMode ? undefined : metadata["stepText"],
+        contractspec_step_reasoning_text: this.config.defaults?.posthogPrivacyMode ? undefined : metadata["stepReasoningText"]
       },
       groups: this.config.defaults?.posthogGroups
     });
@@ -182,11 +299,44 @@ async function resolvePostHogClient(config) {
   }
 }
 function mapTokenUsage(usage) {
+  const usageRecord = asRecord(usage);
+  const inputTokenDetails = asRecord(usageRecord["inputTokenDetails"]);
+  const outputTokenDetails = asRecord(usageRecord["outputTokenDetails"]);
   return {
-    $ai_input_tokens: usage["promptTokens"],
-    $ai_output_tokens: usage["completionTokens"]
+    $ai_input_tokens: asNumber(usageRecord["inputTokens"]) ?? asNumber(usageRecord["promptTokens"]),
+    $ai_output_tokens: asNumber(usageRecord["outputTokens"]) ?? asNumber(usageRecord["completionTokens"]),
+    $ai_reasoning_tokens: asNumber(outputTokenDetails["reasoningTokens"]) ?? asNumber(usageRecord["reasoningTokens"]),
+    $ai_cache_read_input_tokens: asNumber(inputTokenDetails["cacheReadTokens"]) ?? asNumber(usageRecord["cachedInputTokens"]),
+    $ai_cache_creation_input_tokens: asNumber(inputTokenDetails["cacheWriteTokens"]),
+    $ai_usage: maybeRecord(usageRecord["raw"]) ?? usageRecord
   };
 }
+function mapTotalUsage(usage) {
+  const usageRecord = asRecord(usage);
+  return {
+    contractspec_total_input_tokens: asNumber(usageRecord["inputTokens"]),
+    contractspec_total_output_tokens: asNumber(usageRecord["outputTokens"]),
+    contractspec_total_tokens: asNumber(usageRecord["totalTokens"]),
+    contractspec_total_usage_raw: maybeRecord(usageRecord["raw"]) ?? usageRecord
+  };
+}
+function asRecord(value) {
+  return value && typeof value === "object" ? value : {};
+}
+function maybeRecord(value) {
+  return value && typeof value === "object" ? value : undefined;
+}
+function asString(value) {
+  return typeof value === "string" && value.length > 0 ? value : undefined;
+}
+function asNumber(value) {
+  return typeof value === "number" ? value : undefined;
+}
+function asDateIso(value) {
+  if (value instanceof Date)
+    return value.toISOString();
+  return;
+}
 // src/telemetry/index.ts
 init_adapter();
 export {

package/dist/telemetry/posthog-types.d.ts

@@ -15,12 +15,24 @@ export interface PostHogTracingOptions {
     posthogDistinctId?: string;
     /** Trace ID for grouping related generations */
     posthogTraceId?: string;
+    /** Optional model id override */
+    posthogModelOverride?: string;
+    /** Optional provider override */
+    posthogProviderOverride?: string;
+    /** Optional explicit token pricing override */
+    posthogCostOverride?: PostHogCostOverride;
     /** Custom properties attached to every generation event */
     posthogProperties?: Record<string, unknown>;
     /** When true, input/output content is redacted */
     posthogPrivacyMode?: boolean;
     /** Group analytics (e.g., { company: "companyId" }) */
     posthogGroups?: Record<string, string>;
+    /** Sends events immediately (useful for short-lived/serverless contexts) */
+    posthogCaptureImmediate?: boolean;
+}
+export interface PostHogCostOverride {
+    inputCost: number;
+    outputCost: number;
 }
 /**
  * Minimal interface for posthog-node PostHog client.
@@ -57,4 +69,3 @@ export interface PostHogLLMConfig {
     /** Default tracing options applied to all wrapped models */
     defaults?: PostHogTracingOptions;
 }
-//# sourceMappingURL=posthog-types.d.ts.map

package/dist/telemetry/posthog.d.ts

@@ -78,4 +78,3 @@ export declare class CompositeTelemetryCollector implements TelemetryCollector {
 }
 /** Create a composite telemetry collector. */
 export declare function createCompositeTelemetryCollector(collectors: TelemetryCollector[]): CompositeTelemetryCollector;
-//# sourceMappingURL=posthog.d.ts.map