@cad0p/pi-timestamps 0.1.1 → 0.1.4

package/extensions/message-timestamps/index.ts

@@ -1,17 +1,12 @@
 /**
  * message-timestamps
  *
- * Appends a dim-styled UTC ISO timestamp to every user/assistant message.
+ * Appends a dim-styled UTC ISO timestamp to every assistant message.
  * Uses ANSI dim escape codes for visual parity with thinking blocks.
  * Strips at `context` so the LLM never sees them.
  */
 
-import type {
-  AssistantMessage,
-  Message as PiAiMessage,
-  TextContent,
-  UserMessage,
-} from '@earendil-works/pi-ai';
+import type { AssistantMessage, Message as PiAiMessage, TextContent } from '@earendil-works/pi-ai';
 import type { ExtensionAPI } from '@earendil-works/pi-coding-agent';
 
 const DIM = '\x1b[2m';
@@ -49,7 +44,7 @@ export function stripTimestamp(content: PiAiMessage['content']): any {
   return content;
 }
 
-function getTimestamp(msg: UserMessage | AssistantMessage): number {
+function getTimestamp(msg: AssistantMessage): number {
   return msg.timestamp ?? Date.now();
 }
 
@@ -63,11 +58,12 @@ export default function (pi: ExtensionAPI) {
   pi.on('message_end' as any, async (event: MessageEndEvent, _ctx: any) => {
     const msg = event.message;
 
-    if (msg.role !== 'user' && msg.role !== 'assistant') {
+    // Only add timestamps to assistant messages
+    if (msg.role !== 'assistant') {
       return;
     }
 
-    const ts = getTimestamp(msg as UserMessage | AssistantMessage);
+    const ts = getTimestamp(msg as AssistantMessage);
     const suffix = fmtUtc(ts);
 
     return {
@@ -81,11 +77,17 @@ export default function (pi: ExtensionAPI) {
 
   // biome-ignore lint/suspicious/noExplicitAny: pi event types are opaque
   pi.on('context' as any, async (event: { messages: any[] }, _ctx: any) => {
-    const cleanMessages = event.messages.map((m) => ({
-      ...m,
-      // biome-ignore lint/suspicious/noExplicitAny: message content is opaque
-      content: stripTimestamp((m as any).content),
-    }));
+    const cleanMessages = event.messages.map((m) => {
+      // Only strip timestamps from assistant messages that have content
+      if (m.role === 'assistant' && m.content != null) {
+        return {
+          ...m,
+          // biome-ignore lint/suspicious/noExplicitAny: message content is opaque
+          content: stripTimestamp((m as any).content),
+        };
+      }
+      return m;
+    });
     return { messages: cleanMessages };
   });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cad0p/pi-timestamps",
-  "version": "0.1.1",
+  "version": "0.1.4",
   "description": "Inline UTC timestamps for every message in pi coding agent — dim-styled, LLM-safe, zero-config",
   "publishConfig": {
     "access": "public",