pi-subagents-lite 0.2.0 → 0.3.1

package/src/context.ts

@@ -1,9 +1,13 @@
 /**
  * context.ts — Extract parent conversation context for subagent inheritance.
  *
- * Keep extractText only. buildParentContext removed (inherit_context is cut).
+ * Contains extractText (for message content) and buildSnapshotMarkdown
+ * (for agent session snapshot viewer). buildParentContext was removed
+ * when inherit_context was cut.
  */
 
+import { summarizeToolArgs } from "./output-file.js";
+
 /** Extract text from a message content block array. */
 export function extractText(content: unknown[]): string {
   return content
@@ -11,3 +15,78 @@ export function extractText(content: unknown[]): string {
     .map((c: any) => c.text ?? "")
     .join("\n");
 }
+
+/**
+ * Build a markdown snapshot of the full agent conversation from session messages.
+ *
+ * Formatting rules:
+ *   - User messages: `> user: text` blockquote
+ *   - Assistant messages: regular markdown text
+ *   - Tool results: `> ToolName: summarized args` blockquote (using summarizeToolArgs)
+ *   - Unrecognized message roles are skipped
+ *
+ * Tool arguments live on ToolCall blocks inside AssistantMessage.content (linked by id),
+ * not on ToolResultMessage. We pre-build a lookup map.
+ *
+ * @param messages - Agent session messages (from AgentSession.messages)
+ * @returns Formatted markdown string
+ */
+export function buildSnapshotMarkdown(messages: readonly any[]): string {
+  // Build toolCallId -> arguments map from assistant messages
+  const argsMap = new Map<string, Record<string, unknown>>();
+  for (const msg of messages) {
+    if (msg.role === "assistant" && Array.isArray(msg.content)) {
+      for (const block of msg.content) {
+        if (block.type === "toolCall" && block.id && block.arguments) {
+          argsMap.set(block.id, block.arguments);
+        }
+      }
+    }
+  }
+
+  const lines: string[] = [];
+
+  for (const msg of messages) {
+    switch (msg.role) {
+      case "user": {
+        const content = msg.content;
+        const text =
+          typeof content === "string"
+            ? content
+            : extractText(content ?? []);
+        lines.push(`> user: ${text}`, "");
+        break;
+      }
+      case "assistant": {
+        const text = extractText(msg.content ?? []);
+        if (text) {
+          lines.push(text, "");
+        }
+        break;
+      }
+      case "toolResult": {
+        const name = msg.toolName ?? "tool";
+        const displayName = name.charAt(0).toUpperCase() + name.slice(1);
+
+        // Args live on the ToolCall (assistant message), looked up by toolCallId
+        const toolArgs = argsMap.get(msg.toolCallId);
+        let summary = summarizeToolArgs(name, toolArgs);
+        if (summary) {
+          // Strip outer parens from summarizeToolArgs output for cleaner display
+          if (summary.startsWith("(") && summary.endsWith(")")) {
+            summary = summary.slice(1, -1);
+          }
+          // Strip leading space from default multi-key output
+          if (summary.startsWith(" ")) {
+            summary = summary.trimStart();
+          }
+        }
+        lines.push(summary ? `> ${displayName}: ${summary}` : `> ${displayName}`, "");
+        break;
+      }
+      // Skip custom/unrecognized message types
+    }
+  }
+
+  return lines.join("\n");
+}