omnirush 0.7.1 → 0.8.1

package/assets/extensions/omnirush/trace-format.ts

@@ -1,311 +0,0 @@
-// Trace artifact format (team spec, 2026-09-24) — converts a pi session
-// transcript into the reference trace shape: ONE single-line JSON object
-// carrying the run header and the FULL conversation (system prompt,
-// every user/assistant message with tool calls + full results — no
-// truncation of message bodies). Pure module: no imports, injectable
-// redaction, so node:test covers it directly.
-//
-// Reference (buffcode-fable-5 export) header keys, in order:
-//   session_id, user_id, client_id, agent_id, model, started_at,
-//   ended_at, duration_minutes, run_count, incomplete, context_pruned,
-//   unexported_tail_messages, system_prompt_chars, messages, summary
-// Message shapes (OpenAI-ish flat):
-//   {role:"system", content}
-//   {role:"user", content}
-//   {role:"assistant", content, tool_calls?:[{id,type,function:{name,
-//    arguments}}], reasoning?}
-//   {role:"tool", tool_call_id, content, is_error?}
-// Summary:
-//   {message_count, roles:{system,user,assistant,tool}, tool_call_count,
-//    reasoning_message_count, non_text_part_count, ends_on_assistant}
-
-/** One `type:"message"` record from a pi session JSONL (loosely typed). */
-export interface PiRecord {
-  type?: string;
-  id?: string;
-  timestamp?: string;
-  message?: {
-    role?: string;
-    content?: unknown;
-    sections?: Record<string, string>;
-    toolCallId?: string;
-    toolName?: string;
-    isError?: boolean;
-    provider?: string;
-    model?: string;
-    stopReason?: string;
-    usage?: Record<string, unknown>;
-    [key: string]: unknown;
-  };
-  provider?: string;
-  modelId?: string;
-  thinkingLevel?: string;
-  cwd?: string;
-}
-
-export interface ParsedPiSession {
-  sessionId: string | null;
-  cwd: string | null;
-  startedAt: string | null;
-  endedAt: string | null;
-  /** provider/modelId from the model_change record, when present. */
-  model: string | null;
-  messageRecords: PiRecord[];
-  sawCompaction: boolean;
-  /** `omnirush.plan` custom entries (the plan tool's session record),
-   *  in file order: {at, items} snapshots of the task list. Additive —
-   *  the artifact carries them as plan_records so buyers see how the
-   *  agent structured the work. */
-  planRecords: Array<{ at: string | null; items: unknown }>;
-}
-
-function asRecord(line: string): PiRecord | null {
-  try {
-    const parsed = JSON.parse(line);
-    return parsed && typeof parsed === "object" ? (parsed as PiRecord) : null;
-  } catch {
-    return null;
-  }
-}
-
-/** Parse a pi session JSONL (one JSON record per line). */
-export function parsePiSession(text: string): ParsedPiSession {
-  const result: ParsedPiSession = {
-    sessionId: null,
-    cwd: null,
-    startedAt: null,
-    endedAt: null,
-    model: null,
-    messageRecords: [],
-    sawCompaction: false,
-    planRecords: [],
-  };
-  let firstTimestamp: string | null = null;
-  let lastTimestamp: string | null = null;
-  for (const line of text.split("\n")) {
-    if (!line.trim()) continue;
-    const record = asRecord(line);
-    if (!record) continue;
-    if (record.type === "session") {
-      result.sessionId = typeof record.id === "string" ? record.id : result.sessionId;
-      result.cwd = typeof record.cwd === "string" ? record.cwd : result.cwd;
-    } else if (record.type === "model_change") {
-      if (typeof record.provider === "string" && typeof record.modelId === "string") {
-        result.model = `${record.provider}/${record.modelId}`;
-      }
-    } else if (record.type && /compact/i.test(record.type)) {
-      result.sawCompaction = true;
-    } else if (record.type === "custom" && (record as any).customType === "omnirush.plan") {
-      const data = (record as any).data as { items?: unknown } | undefined;
-      result.planRecords.push({
-        at: typeof record.timestamp === "string" ? record.timestamp : null,
-        items: data?.items ?? [],
-      });
-    }
-    if (typeof record.timestamp === "string") {
-      firstTimestamp ??= record.timestamp;
-      lastTimestamp = record.timestamp;
-    }
-    if (record.type === "message" && record.message) {
-      result.messageRecords.push(record);
-    }
-  }
-  result.startedAt = firstTimestamp;
-  result.endedAt = lastTimestamp ?? firstTimestamp;
-  return result;
-}
-
-/**
- * Flatten pi message content into display text: string passes through;
- * part arrays join their text parts with a blank line. Non-text parts
- * (images) are COUNTED, not embedded — the reference format counts them
- * in summary.non_text_part_count and carries no base64 payloads.
- */
-export function contentToText(content: unknown): { text: string; nonTextParts: number } {
-  if (typeof content === "string") return { text: content, nonTextParts: 0 };
-  if (!Array.isArray(content)) {
-    return { text: content === null || content === undefined ? "" : String(content), nonTextParts: 0 };
-  }
-  const texts: string[] = [];
-  let nonTextParts = 0;
-  for (const part of content as Array<any>) {
-    if (part && typeof part === "object" && part.type === "text" && typeof part.text === "string") {
-      texts.push(part.text);
-    } else {
-      nonTextParts += 1;
-    }
-  }
-  return { text: texts.join("\n\n"), nonTextParts };
-}
-
-/** Reconstruct the system prompt from a pi system message record. */
-function systemPromptOf(message: PiRecord["message"]): string {
-  if (!message) return "";
-  if (message.sections && typeof message.sections === "object") {
-    return Object.values(message.sections)
-      .filter((value) => typeof value === "string")
-      .join("\n\n");
-  }
-  return contentToText(message.content).text;
-}
-
-function durationMinutes(startedAt: string | null, endedAt: string | null): number {
-  if (!startedAt || !endedAt) return 0;
-  const start = Date.parse(startedAt);
-  const end = Date.parse(endedAt);
-  if (!Number.isFinite(start) || !Number.isFinite(end) || end <= start) return 0;
-  return Math.round(((end - start) / 60_000) * 10) / 10;
-}
-
-export interface BuildTraceArtifactInput {
-  sessionId: string;
-  /** Set when this artifact belongs to a subagent session. */
-  parentSessionId?: string | null;
-  userId?: string | null;
-  clientId?: string | null;
-  agentId?: string | null;
-  parsed: ParsedPiSession;
-  /** Scrubber applied to every produced string (existing redaction). */
-  redact: (text: string) => string;
-}
-
-/**
- * Build the trace artifact object (reference shape; key order mirrored).
- * Returns the plain object — serialize with JSON.stringify for the
- * single-line .jsonl form.
- */
-export function buildTraceArtifact(input: BuildTraceArtifactInput): Record<string, unknown> {
-  const { parsed, redact } = input;
-  const messages: Array<Record<string, unknown>> = [];
-  const roles: Record<string, number> = {};
-  let toolCallCount = 0;
-  let reasoningMessageCount = 0;
-  let nonTextPartCount = 0;
-  let systemPrompt = "";
-  let model = input.parsed.model ?? null;
-
-  for (const record of parsed.messageRecords) {
-    const message = record.message!;
-    const role = typeof message.role === "string" ? message.role : "unknown";
-    roles[role] = (roles[role] ?? 0) + 1;
-
-    if (role === "system") {
-      systemPrompt = systemPromptOf(message);
-      messages.push({ role: "system", content: redact(systemPrompt) });
-      continue;
-    }
-
-    if (role === "assistant") {
-      const text = contentToText(message.content);
-      const reasoningParts: string[] = [];
-      const toolCalls: Array<Record<string, unknown>> = [];
-      if (Array.isArray(message.content)) {
-        for (const part of message.content as Array<any>) {
-          if (part && part.type === "thinking" && typeof part.thinking === "string") {
-            reasoningParts.push(part.thinking);
-          } else if (part && part.type === "toolCall") {
-            toolCallCount += 1;
-            toolCalls.push({
-              id: typeof part.id === "string" ? part.id : `call_${toolCalls.length}`,
-              type: "function",
-              function: {
-                name: typeof part.name === "string" ? part.name : "unknown",
-                arguments: JSON.stringify(part.arguments ?? {}),
-              },
-            });
-          } else if (part && part.type === "text") {
-            // already folded into `content` via contentToText
-          } else {
-            // Images and any other non-text parts: counted in the
-            // summary, payload not embedded.
-            nonTextPartCount += 1;
-          }
-        }
-      }
-      const assistant: Record<string, unknown> = {
-        role: "assistant",
-        content: redact(text.text),
-      };
-      if (toolCalls.length > 0) assistant.tool_calls = toolCalls;
-      if (reasoningParts.length > 0) assistant.reasoning = redact(reasoningParts.join("\n\n"));
-      if (!model && typeof message.provider === "string" && typeof message.model === "string") {
-        model = `${message.provider}/${message.model}`;
-      }
-      messages.push(assistant);
-      if (reasoningParts.length > 0) reasoningMessageCount += 1;
-      continue;
-    }
-
-    if (role === "toolResult") {
-      const text = contentToText(message.content);
-      nonTextPartCount += text.nonTextParts;
-      const tool: Record<string, unknown> = {
-        role: "tool",
-        tool_call_id: typeof message.toolCallId === "string" ? message.toolCallId : "",
-        content: redact(text.text),
-      };
-      if (message.isError === true) tool.is_error = true;
-      messages.push(tool);
-      continue;
-    }
-
-    // user and anything else: full-fidelity text.
-    const text = contentToText(message.content);
-    nonTextPartCount += text.nonTextParts;
-    messages.push({ role, content: redact(text.text) });
-  }
-
-  const endsOnAssistant = messages.length > 0 && messages[messages.length - 1].role === "assistant";
-  const runCount = roles.user ?? 0;
-  const startedAt = parsed.startedAt;
-  const endedAt = parsed.endedAt ?? parsed.startedAt;
-
-  const artifact: Record<string, unknown> = {
-    session_id: input.sessionId,
-    user_id: input.userId ?? null,
-    client_id: input.clientId ?? null,
-    agent_id: input.agentId ?? null,
-    model: model ?? "unknown",
-    started_at: startedAt,
-    ended_at: endedAt,
-    duration_minutes: durationMinutes(startedAt, endedAt),
-    run_count: runCount,
-    incomplete: !endsOnAssistant,
-    context_pruned: parsed.sawCompaction,
-    unexported_tail_messages: 0,
-    system_prompt_chars: systemPrompt.length,
-    messages,
-    summary: {
-      message_count: messages.length,
-      roles: {
-        system: roles.system ?? 0,
-        user: roles.user ?? 0,
-        assistant: roles.assistant ?? 0,
-        tool: roles.toolResult ?? 0,
-      },
-      tool_call_count: toolCallCount,
-      reasoning_message_count: reasoningMessageCount,
-      non_text_part_count: nonTextPartCount,
-      ends_on_assistant: endsOnAssistant,
-    },
-  };
-  if (input.parentSessionId && input.parentSessionId !== input.sessionId) {
-    artifact.parent_session_id = input.parentSessionId;
-  }
-  // Additive: the plan tool's task-list snapshots, in order. Absent
-  // entirely for sessions that never used the plan tool.
-  if (parsed.planRecords.length > 0) {
-    artifact.plan_records = parsed.planRecords;
-  }
-  return artifact;
-}
-
-/**
- * pi stores sessions under <agentDir>/sessions/<cwd-slug>/<ts>_<id>.jsonl
- * (e.g. /home/dan/proj -> --home-dan-proj--). Advisory only: the
- * collector discovers the right dir by scanning for the session id, so
- * this does not have to be perfect.
- */
-export function sessionDirSlug(cwd: string): string {
-  return `${cwd.replace(/\\//g, "-")}-`;
-}