oc-tweaks 0.8.2 → 0.9.0

package/dist/index.js

@@ -834,12 +834,21 @@ function parseJsonBlob(value) {
   return JSON.parse(String(value));
 }
 function queryMessages(db, sessionId) {
-  const rows = db.query("SELECT data FROM message WHERE session_id = ? ORDER BY time_created ASC").all(sessionId);
-  return rows.map((row) => parseJsonBlob(row.data));
+  const rows = db.query("SELECT id, data FROM message WHERE session_id = ? ORDER BY time_created ASC").all(sessionId);
+  return rows.map((row) => {
+    const parsed = parseJsonBlob(row.data);
+    parsed._messageId = row.id;
+    return parsed;
+  });
 }
 function queryParts(db, sessionId) {
-  const rows = db.query("SELECT data FROM part WHERE session_id = ? ORDER BY id ASC").all(sessionId);
-  return rows.map((row) => parseJsonBlob(row.data));
+  const rows = db.query("SELECT data, message_id FROM part WHERE session_id = ? ORDER BY id ASC").all(sessionId);
+  return rows.map((row) => {
+    const parsed = parseJsonBlob(row.data);
+    if (row.message_id)
+      parsed._messageId = row.message_id;
+    return parsed;
+  });
 }
 function isRecoverableBlobError(error) {
   if (error instanceof SyntaxError)
@@ -1300,31 +1309,62 @@ function formatTranscript(sessionId, messages, parts) {
   lines.push(`Messages: ${messages.length}`);
   lines.push(`Parts: ${parts.length}`);
   lines.push("");
-  for (const message of messages) {
-    const role = typeof message?.role === "string" ? message.role : "unknown";
-    const text = getMessageText(message?.content, role === "assistant" ? 320 : 520);
-    if (text) {
-      lines.push(`[${role === "user" ? "User" : role === "assistant" ? "Assistant" : "Message"}] ${text}`);
+  const hasPartBasedText = parts.some((part) => part.type === "text" && typeof part.text === "string" && part.text.trim().length > 0 || part.type === "reasoning" && typeof part.reasoning === "string" && part.reasoning.trim().length > 0);
+  if (hasPartBasedText) {
+    const roleByMessageId = new Map;
+    for (const message of messages) {
+      if (message._messageId && typeof message.role === "string") {
+        roleByMessageId.set(message._messageId, message.role);
+      }
+    }
+    for (const part of parts) {
+      if (part.type === "text" && typeof part.text === "string" && part.text.trim().length > 0) {
+        const role = part._messageId ? roleByMessageId.get(part._messageId) ?? "unknown" : "unknown";
+        const label = role === "user" ? "User" : role === "assistant" ? "Assistant" : "Message";
+        const maxChars = role === "assistant" ? 320 : 520;
+        lines.push(`[${label}] ${toLimitedText(part.text, maxChars)}`);
+      } else if (part.type === "reasoning" && typeof part.reasoning === "string" && part.reasoning.trim().length > 0) {
+        lines.push(`[Reasoning] ${toLimitedText(part.reasoning, 320)}`);
+      } else if (part.type === "tool" && part.tool) {
+        const toolName = typeof part.tool === "string" ? part.tool : "unknown";
+        const inputText = safeStringify(part.state?.input, 260);
+        const outputText = safeStringify(part.state?.output, 260);
+        const errorText = safeStringify(part.state?.error, 180);
+        lines.push(`[Tool: ${toolName}]`);
+        if (inputText)
+          lines.push(`  input: ${inputText}`);
+        if (outputText)
+          lines.push(`  output: ${outputText}`);
+        if (errorText)
+          lines.push(`  error: ${errorText}`);
+      }
+    }
+  } else {
+    for (const message of messages) {
+      const role = typeof message?.role === "string" ? message.role : "unknown";
+      const text = getMessageText(message?.content, role === "assistant" ? 320 : 520);
+      if (text) {
+        lines.push(`[${role === "user" ? "User" : role === "assistant" ? "Assistant" : "Message"}] ${text}`);
+      }
+    }
+    const toolParts = parts.filter((p) => p?.type === "tool");
+    if (toolParts.length > 0) {
+      lines.push("");
+      lines.push("[Tool Parts]");
+      for (const part of toolParts) {
+        const toolName = typeof part.tool === "string" ? part.tool : "unknown";
+        const inputText = safeStringify(part.state?.input, 260);
+        const outputText = safeStringify(part.state?.output, 260);
+        const errorText = safeStringify(part.state?.error, 180);
+        lines.push(`[Tool: ${toolName}]`);
+        if (inputText)
+          lines.push(`  input: ${inputText}`);
+        if (outputText)
+          lines.push(`  output: ${outputText}`);
+        if (errorText)
+          lines.push(`  error: ${errorText}`);
+      }
     }
-  }
-  if (parts.length > 0) {
-    lines.push("");
-    lines.push("[Tool Parts]");
-  }
-  for (const part of parts) {
-    if (part?.type !== "tool")
-      continue;
-    const toolName = typeof part.tool === "string" ? part.tool : "unknown";
-    const inputText = safeStringify(part.state?.input, 260);
-    const outputText = safeStringify(part.state?.output, 260);
-    const errorText = safeStringify(part.state?.error, 180);
-    lines.push(`[Tool: ${toolName}]`);
-    if (inputText)
-      lines.push(`  input: ${inputText}`);
-    if (outputText)
-      lines.push(`  output: ${outputText}`);
-    if (errorText)
-      lines.push(`  error: ${errorText}`);
   }
   return lines.join(`
 `);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "oc-tweaks",
-  "version": "0.8.2",
+  "version": "0.9.0",
   "type": "module",
   "exports": {
     ".": "./dist/index.js"