drizzle-cube 0.4.16 → 0.4.18

package/dist/server/index.d.ts

@@ -39,6 +39,8 @@ export declare interface AgentConfig {
     maxTokens?: number;
     /** Allow X-Agent-Api-Key header to override server apiKey */
     allowClientApiKey?: boolean;
+    /** Optional observability hooks for tracing */
+    observability?: AgentObservabilityHooks;
 }
 
 /**
@@ -61,6 +63,48 @@ declare interface AgentHistoryToolCall {
     status: 'running' | 'complete' | 'error';
 }
 
+/**
+ * Observability hooks for tracing agent chat lifecycle events.
+ * Called synchronously (fire-and-forget) — implementations should not throw.
+ */
+export declare interface AgentObservabilityHooks {
+    /** Called once when agent chat starts */
+    onChatStart?: (event: {
+        traceId: string;
+        sessionId?: string;
+        message: string;
+        model: string;
+        historyLength: number;
+    }) => void;
+    /** Called after each Anthropic API call completes */
+    onGenerationEnd?: (event: {
+        traceId: string;
+        turn: number;
+        model: string;
+        stopReason: string;
+        inputTokens?: number;
+        outputTokens?: number;
+        durationMs: number;
+    }) => void;
+    /** Called after each tool execution */
+    onToolEnd?: (event: {
+        traceId: string;
+        turn: number;
+        toolName: string;
+        toolUseId: string;
+        isError: boolean;
+        durationMs: number;
+    }) => void;
+    /** Called when agent chat completes */
+    onChatEnd?: (event: {
+        traceId: string;
+        sessionId?: string;
+        totalTurns: number;
+        durationMs: number;
+        error?: string;
+    }) => void;
+}
+
 /**
  * SSE events emitted by the agent handler
  */
@@ -98,6 +142,7 @@ export declare type AgentSSEEvent = {
     type: 'done';
     data: {
         sessionId: string;
+        traceId?: string;
     };
 } | {
     type: 'error';