@townco/debugger 0.1.27 → 0.1.29

package/src/components/SessionAnalysisDialog.tsx

@@ -0,0 +1,155 @@
+import type { SessionAnalysis } from "../analysis/types";
+import { Dialog, DialogContent, DialogHeader, DialogTitle } from "./ui/dialog";
+
+interface Props {
+  open: boolean;
+  onClose: () => void;
+  analysis: SessionAnalysis;
+}
+
+function formatDate(isoString: string): string {
+  return new Date(isoString).toLocaleString();
+}
+
+function calculateDuration(start: string, end: string): string {
+  const startTime = new Date(start).getTime();
+  const endTime = new Date(end).getTime();
+  const durationMs = endTime - startTime;
+
+  const seconds = Math.floor(durationMs / 1000);
+  const minutes = Math.floor(seconds / 60);
+  const hours = Math.floor(minutes / 60);
+
+  if (hours > 0) {
+    return `${hours}h ${minutes % 60}m ${seconds % 60}s`;
+  }
+  if (minutes > 0) {
+    return `${minutes}m ${seconds % 60}s`;
+  }
+  return `${seconds}s`;
+}
+
+function Section({
+  title,
+  children,
+}: {
+  title: string;
+  children: React.ReactNode;
+}) {
+  return (
+    <div className="space-y-3">
+      <h3 className="text-sm font-semibold text-foreground">{title}</h3>
+      <div className="space-y-2">{children}</div>
+    </div>
+  );
+}
+
+function Field({ label, value }: { label: string; value: string }) {
+  return (
+    <div className="space-y-1">
+      <div className="text-xs font-medium text-muted-foreground">{label}</div>
+      <div className="text-sm text-foreground whitespace-pre-wrap break-words">
+        {value}
+      </div>
+    </div>
+  );
+}
+
+function Badge({ label, value }: { label: string; value: string }) {
+  return (
+    <div className="space-y-1">
+      <div className="text-xs font-medium text-muted-foreground">{label}</div>
+      <div className="inline-block px-3 py-1 bg-primary/10 text-primary rounded-full text-sm font-medium">
+        {value}
+      </div>
+    </div>
+  );
+}
+
+function Metric({ label, value }: { label: string; value: number }) {
+  return (
+    <div className="space-y-1">
+      <div className="text-xs font-medium text-muted-foreground">{label}</div>
+      <div className="text-2xl font-bold text-foreground">{value}</div>
+    </div>
+  );
+}
+
+export function SessionAnalysisDialog({ open, onClose, analysis }: Props) {
+  return (
+    <Dialog open={open} onOpenChange={onClose}>
+      <DialogContent className="max-w-3xl max-h-[85vh] overflow-y-auto">
+        <DialogHeader>
+          <DialogTitle>Session Analysis</DialogTitle>
+        </DialogHeader>
+
+        <div className="space-y-6">
+          {/* Task Section */}
+          <Section title="Task">
+            <Field label="User Query" value={analysis.task.user_query} />
+            <Field label="Summary" value={analysis.task.task_summary} />
+            <Badge label="Intent" value={analysis.task.intent_type} />
+          </Section>
+
+          {/* Trajectory Section */}
+          <Section title="Trajectory">
+            <Field
+              label="High Level Plan"
+              value={analysis.trajectory.high_level_plan}
+            />
+            <div className="grid grid-cols-2 gap-4 pt-2">
+              <Metric label="Steps" value={analysis.trajectory.num_steps} />
+              <Metric
+                label="Tool Calls"
+                value={analysis.trajectory.num_tool_calls}
+              />
+            </div>
+            {analysis.trajectory.tools_used.length > 0 && (
+              <div className="space-y-2 pt-2">
+                <div className="text-xs font-medium text-muted-foreground">
+                  Tools Used
+                </div>
+                <div className="flex flex-wrap gap-2">
+                  {analysis.trajectory.tools_used.map((tool) => (
+                    <span
+                      key={tool}
+                      className="px-2 py-1 bg-secondary text-secondary-foreground rounded text-xs font-medium"
+                    >
+                      {tool}
+                    </span>
+                  ))}
+                </div>
+              </div>
+            )}
+          </Section>
+
+          {/* Outcome Section */}
+          <Section title="Outcome">
+            <div className="flex gap-4">
+              <Badge label="Status" value={analysis.outcome.status} />
+              <Badge label="Answer Type" value={analysis.outcome.answer_type} />
+            </div>
+            <Field label="Assessment" value={analysis.outcome.assessment} />
+          </Section>
+
+          {/* Metadata Section */}
+          <Section title="Metadata">
+            <div className="grid grid-cols-2 gap-4">
+              <Field label="Started" value={formatDate(analysis.started_at)} />
+              <Field label="Ended" value={formatDate(analysis.ended_at)} />
+            </div>
+            <Field
+              label="Duration"
+              value={calculateDuration(analysis.started_at, analysis.ended_at)}
+            />
+            <Field label="Agent" value={analysis.agent_name} />
+            <Field
+              label="Session ID"
+              value={analysis.session_id.slice(0, 16)}
+            />
+          </Section>
+        </div>
+      </DialogContent>
+    </Dialog>
+  );
+}

package/src/components/UnifiedTimeline.tsx

@@ -231,7 +231,7 @@ export function UnifiedTimeline({
         {/* Timeline container */}
         <div
           ref={timelineRef}
-          className="border border-border rounded-lg overflow-hidden bg-card flex"
+          className="border border-border rounded-lg overflow-hidden bg-card flex max-h-[calc(100vh-250px)]"
           onWheel={handleWheel}
         >
           {/* Fixed left column - processes labels */}
@@ -245,7 +245,7 @@ export function UnifiedTimeline({
             </div>
 
             {/* Span labels */}
-            <div className="flex-1 bg-white dark:bg-gray-900">
+            <div className="flex-1 bg-white dark:bg-gray-900 overflow-y-auto">
               {flatSpans.map(({ node, depth }) => {
                 const span = spans.find((s) => s.span_id === node.span_id);
                 // This should never be null now due to filtering
@@ -310,7 +310,7 @@ export function UnifiedTimeline({
 
           {/* Scrollable right side - message bubbles + timeline */}
           <div
-            className="flex-1 overflow-x-auto overflow-y-hidden"
+            className="flex-1 overflow-x-auto overflow-y-auto"
             style={{ background: "#404040" }}
           >
             <div

package/src/components/ui/dialog.tsx

@@ -0,0 +1,120 @@
+import * as DialogPrimitive from "@radix-ui/react-dialog";
+import { X } from "lucide-react";
+import * as React from "react";
+
+import { cn } from "@/lib/utils";
+
+const Dialog = DialogPrimitive.Root;
+
+const DialogTrigger = DialogPrimitive.Trigger;
+
+const DialogPortal = DialogPrimitive.Portal;
+
+const DialogClose = DialogPrimitive.Close;
+
+const DialogOverlay = React.forwardRef<
+  React.ElementRef<typeof DialogPrimitive.Overlay>,
+  React.ComponentPropsWithoutRef<typeof DialogPrimitive.Overlay>
+>(({ className, ...props }, ref) => (
+  <DialogPrimitive.Overlay
+    ref={ref}
+    className={cn(
+      "fixed inset-0 z-50 bg-black/80 data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0",
+      className,
+    )}
+    {...props}
+  />
+));
+DialogOverlay.displayName = DialogPrimitive.Overlay.displayName;
+
+const DialogContent = React.forwardRef<
+  React.ElementRef<typeof DialogPrimitive.Content>,
+  React.ComponentPropsWithoutRef<typeof DialogPrimitive.Content>
+>(({ className, children, ...props }, ref) => (
+  <DialogPortal>
+    <DialogOverlay />
+    <DialogPrimitive.Content
+      ref={ref}
+      className={cn(
+        "fixed left-[50%] top-[50%] z-50 grid w-full max-w-lg translate-x-[-50%] translate-y-[-50%] gap-4 border bg-background p-6 shadow-lg duration-200 data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 data-[state=closed]:slide-out-to-left-1/2 data-[state=closed]:slide-out-to-top-[48%] data-[state=open]:slide-in-from-left-1/2 data-[state=open]:slide-in-from-top-[48%] sm:rounded-lg",
+        className,
+      )}
+      {...props}
+    >
+      {children}
+      <DialogPrimitive.Close className="absolute right-4 top-4 rounded-sm opacity-70 ring-offset-background transition-opacity hover:opacity-100 focus:outline-none focus:ring-2 focus:ring-ring focus:ring-offset-2 disabled:pointer-events-none data-[state=open]:bg-accent data-[state=open]:text-muted-foreground">
+        <X className="size-4" />
+        <span className="sr-only">Close</span>
+      </DialogPrimitive.Close>
+    </DialogPrimitive.Content>
+  </DialogPortal>
+));
+DialogContent.displayName = DialogPrimitive.Content.displayName;
+
+const DialogHeader = ({
+  className,
+  ...props
+}: React.HTMLAttributes<HTMLDivElement>) => (
+  <div
+    className={cn(
+      "flex flex-col space-y-1.5 text-center sm:text-left",
+      className,
+    )}
+    {...props}
+  />
+);
+DialogHeader.displayName = "DialogHeader";
+
+const DialogFooter = ({
+  className,
+  ...props
+}: React.HTMLAttributes<HTMLDivElement>) => (
+  <div
+    className={cn(
+      "flex flex-col-reverse sm:flex-row sm:justify-end sm:space-x-2",
+      className,
+    )}
+    {...props}
+  />
+);
+DialogFooter.displayName = "DialogFooter";
+
+const DialogTitle = React.forwardRef<
+  React.ElementRef<typeof DialogPrimitive.Title>,
+  React.ComponentPropsWithoutRef<typeof DialogPrimitive.Title>
+>(({ className, ...props }, ref) => (
+  <DialogPrimitive.Title
+    ref={ref}
+    className={cn(
+      "text-lg font-semibold leading-none tracking-tight",
+      className,
+    )}
+    {...props}
+  />
+));
+DialogTitle.displayName = DialogPrimitive.Title.displayName;
+
+const DialogDescription = React.forwardRef<
+  React.ElementRef<typeof DialogPrimitive.Description>,
+  React.ComponentPropsWithoutRef<typeof DialogPrimitive.Description>
+>(({ className, ...props }, ref) => (
+  <DialogPrimitive.Description
+    ref={ref}
+    className={cn("text-sm text-muted-foreground", className)}
+    {...props}
+  />
+));
+DialogDescription.displayName = DialogPrimitive.Description.displayName;
+
+export {
+  Dialog,
+  DialogPortal,
+  DialogOverlay,
+  DialogClose,
+  DialogTrigger,
+  DialogContent,
+  DialogHeader,
+  DialogFooter,
+  DialogTitle,
+  DialogDescription,
+};

package/src/db.ts

@@ -89,15 +89,16 @@ export class DebuggerDb {
   getSpansBySessionAttribute(sessionId: string): Span[] {
     // Use JSON extract to find spans where agent.session_id matches
     return this.db
-      .query<Span, [string]>(
+      .query<Span, [string, string]>(
         `
         SELECT *
         FROM spans
         WHERE json_extract(attributes, '$."agent.session_id"') = ?
+           OR json_extract(attributes, '$."session.id"') = ?
         ORDER BY start_time_unix_nano ASC
         `,
       )
-      .all(sessionId);
+      .all(sessionId, sessionId);
   }
 
   /**

package/src/lib/metrics.ts

@@ -1,4 +1,4 @@
-import type { SessionMetrics, Span, Trace } from "../types";
+import type { SessionMetrics, Span, ToolCall, Trace } from "../types";
 import { getModelPricing } from "./pricing";
 
 /**
@@ -28,7 +28,7 @@ export function extractMetricsFromSpans(
 ): Omit<SessionMetrics, "durationMs"> {
   let inputTokens = 0;
   let outputTokens = 0;
-  let toolCallCount = 0;
+  const toolCalls: ToolCall[] = [];
 
   for (const span of spans) {
     const attrs = span.attributes ? JSON.parse(span.attributes) : {};
@@ -41,14 +41,109 @@ export function extractMetricsFromSpans(
       outputTokens += attrs["gen_ai.usage.output_tokens"];
     }
 
-    // Count tool calls
-    if (span.name.includes("tool_call") || span.name.startsWith("tool:")) {
-      toolCallCount++;
+    // Count tool calls from dedicated tool spans
+    const isToolSpan =
+      span.name.includes("tool_call") || span.name.startsWith("tool:");
+    if (isToolSpan) {
+      const name =
+        attrs["tool.name"] ||
+        attrs["tool.type"] ||
+        attrs["gen_ai.tool_call.name"] ||
+        span.name;
+
+      const rawInput =
+        attrs["tool.input"] || attrs["gen_ai.tool_call.input"] || null;
+      const rawOutput = attrs["tool.output"] || null;
+
+      const parseMaybeJson = (val: unknown) => {
+        if (typeof val !== "string") return val;
+        try {
+          return JSON.parse(val);
+        } catch {
+          return val;
+        }
+      };
+
+      toolCalls.push({
+        name: typeof name === "string" ? name : String(name),
+        input: parseMaybeJson(rawInput),
+        output: parseMaybeJson(rawOutput),
+        startTimeUnixNano: span.start_time_unix_nano,
+        endTimeUnixNano: span.end_time_unix_nano,
+      });
+    }
+
+    // Also inspect model outputs for declared tool calls (e.g. Anthropic tool_use)
+    const messagesRaw =
+      attrs["gen_ai.output.messages"] ?? attrs["gen_ai.output.tool_calls"];
+    if (messagesRaw) {
+      try {
+        const messages =
+          typeof messagesRaw === "string"
+            ? JSON.parse(messagesRaw)
+            : messagesRaw;
+        if (Array.isArray(messages)) {
+          for (const msg of messages) {
+            if (msg && typeof msg === "object") {
+              const toolCallsArray =
+                (msg as Record<string, unknown>).tool_calls ?? [];
+              if (Array.isArray(toolCallsArray)) {
+                for (const call of toolCallsArray) {
+                  if (call && typeof call === "object") {
+                    toolCalls.push({
+                      name:
+                        (call as Record<string, unknown>).name?.toString() ||
+                        "tool_call",
+                      input: (call as Record<string, unknown>).args,
+                      output: null,
+                      startTimeUnixNano: span.start_time_unix_nano,
+                      endTimeUnixNano: span.end_time_unix_nano,
+                    });
+                  }
+                }
+              }
+
+              const content = (msg as Record<string, unknown>).content;
+              if (Array.isArray(content)) {
+                for (const c of content) {
+                  if (
+                    c &&
+                    typeof c === "object" &&
+                    (c as Record<string, unknown>).type === "tool_use"
+                  ) {
+                    toolCalls.push({
+                      name:
+                        ((c as Record<string, unknown>).name as
+                          | string
+                          | undefined) || "tool_use",
+                      input: (c as Record<string, unknown>).input,
+                      output: null,
+                      startTimeUnixNano: span.start_time_unix_nano,
+                      endTimeUnixNano: span.end_time_unix_nano,
+                    });
+                  }
+                }
+              }
+            }
+          }
+        }
+      } catch {
+        // ignore parse errors
+      }
     }
   }
 
   const totalTokens = inputTokens + outputTokens;
   const estimatedCost = calculateCost(model, inputTokens, outputTokens);
+  // Dedupe tool calls using name + start time to avoid double counting when captured in multiple places
+  const deduped = new Map<string, ToolCall>();
+  for (const call of toolCalls) {
+    const key = `${call.name}-${call.startTimeUnixNano ?? ""}`;
+    if (!deduped.has(key)) {
+      deduped.set(key, call);
+    }
+  }
+  const toolCallCount = deduped.size;
 
   return {
     inputTokens,
@@ -56,6 +151,7 @@ export function extractMetricsFromSpans(
     totalTokens,
     estimatedCost,
     toolCallCount,
+    toolCalls: Array.from(deduped.values()),
   };
 }