@taylorsatula/pi-timestamp 0.2.1 → 0.2.2

package/README.md

@@ -12,20 +12,13 @@ They're injected poorly if they're that annoying.
 [11:34pm - 9/1/26]
 ```
 
-Injection happens only at **turn boundaries**, never mid-tool-loop, and handles both wire
-formats the hook can deliver (the payload is the provider-serialized request body, which
-differs per API):
-
-- **Chat Completions / Anthropic** (conversation under `messages`): last message is a
-  **user message** → append the timestamp to its content. Last message is an
-  **assistant message with text content** (a vocalization between tool steps) → append
-  to its content. Last message is an assistant message carrying `tool_calls`, a tool
-  result, or an Anthropic user message containing a `tool_result` block (that API's
-  tool-result encoding) → skip.
-- **OpenAI Responses API** (conversation under `input`, typed items): last item is a
-  user item → append to its content blocks. Last item is an assistant `message` item
-  (vocalization) → append to its `output_text` blocks. Last item is a `function_call` or
-  `function_call_output` item → skip.
+Injection happens only at **turn boundaries**, never mid-tool-loop:
+
+- Last message is a **user message** → append the timestamp to its content.
+- Last message is an **assistant message with text content** (a vocalization between tool
+  steps) → append to its content.
+- Last message is an assistant message carrying `tool_calls`, or a tool result (mid-turn
+  churn) → skip.
 
 ## Why plain-text tail append (design notes)
 
@@ -42,9 +35,6 @@ differs per API):
 - **Content-shape aware.** String content concatenates directly; block-array content
   (Anthropic-shaped) appends into or after the last text block; unknown shapes are left
   untouched rather than guessed at.
-- **Wire-format aware.** Chat Completions/Anthropic payloads (`messages`) and OpenAI
-  Responses payloads (`input`, typed items) are both handled; any other payload shape
-  passes through untouched.
 
 ## Installation
 
@@ -71,9 +61,6 @@ Or add to `~/.pi/agent/settings.json` under `packages`:
 - Version 0.2.0 changed the mechanism (plain-text tail append replacing the synthetic
   tool-call pair). Sessions started before the upgrade may still show one old-style pair
   until restart.
-- Version 0.2.1 fixed a crash on OpenAI Responses-API models (`before_provider_request`
-  payloads carry the conversation under `input`, not `messages`) and added
-  Anthropic `tool_result` skipping.
 
 ## Development
 

package/extensions/timestamp.ts

@@ -14,15 +14,6 @@ import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 //     never after tool results (mid-loop churn).
 //   - Idempotent per request: a previously appended tag is stripped before the
 //     fresh one is written.
-//
-// Payload shapes (before_provider_request delivers the provider-serialized
-// request body, which differs per API):
-//   - Chat Completions / Anthropic: conversation lives under `messages`.
-//   - OpenAI Responses API: conversation lives under `input`, with typed items
-//     (input_text / output_text content blocks, function_call and
-//     function_call_output tool traffic) instead of chat messages.
-
-type Rec = Record<string, unknown>;
 
 const TS_STRIP = /\n\[\d{1,2}:\d{2}(am|pm) - \d{1,2}\/\d{1,2}\/\d{2}\]\s*$/;
 
@@ -39,73 +30,44 @@ function formatTimestamp(): string {
   return `[${hours}:${String(minutes).padStart(2, "0")}${ampm} - ${month}/${day}/${year}]`;
 }
 
-// Chat Completions / Anthropic shape: payload.messages
-// user: { role: "user", content: string | blocks }
-// assistant vocalization: { role: "assistant", content: string | blocks, no tool_calls }
-// assistant tool call: { role: "assistant", tool_calls: [...] }
-// Anthropic tool results arrive as user messages containing a tool_result
-// content block; treat them as mid-loop traffic, not turn boundaries.
+function injectable(msgs: Array<Record<string, unknown>> | undefined): boolean {
+  if (!Array.isArray(msgs) || msgs.length === 0) return false;
+  const last = msgs[msgs.length - 1];
+  if (!last) return false;
+
+  if (last.role === "user") return true;
 
-function chatShouldInject(last: Rec): boolean {
-  if (last.role === "user") {
-    const content = last.content;
-    if (Array.isArray(content) && content.some((b) => (b as Rec).type === "tool_result")) {
-      return false;
-    }
-    return true;
-  }
   if (last.role === "assistant") {
     const hasToolCalls = Array.isArray(last.tool_calls) && last.tool_calls.length > 0;
     const hasContent = last.content != null && last.content !== "";
     return !hasToolCalls && hasContent; // vocalization → inject; tool call → never
   }
-  return false;
-}
-
-// OpenAI Responses shape: payload.input
-// user: { role: "user", content: [{ type: "input_text", text }, ...] }
-// vocalization: { type: "message", role: "assistant", content: [{ type: "output_text", ... }] }
-// tool call: { type: "function_call", ... } / result: { type: "function_call_output", ... }
 
-function responsesShouldInject(last: Rec): boolean {
-  if (last.role === "user") return true;
-  if (last.type === "message" && last.role === "assistant") {
-    return Array.isArray(last.content) && last.content.length > 0;
-  }
-  return false; // function_call / function_call_output → mid-loop, skip
+  return false; // tool results → mid-loop, skip
 }
 
 export default function (pi: ExtensionAPI) {
   pi.on("before_provider_request", (event) => {
-    const payload = event.payload as Rec;
-    const isResponses = Array.isArray(payload.input);
-    const items = isResponses
-      ? (payload.input as Rec[])
-      : Array.isArray(payload.messages)
-        ? (payload.messages as Rec[])
-        : undefined;
-    if (!items) return; // unknown payload shape → leave untouched
-
-    const last = items[items.length - 1];
-    if (!last) return;
-    if (!(isResponses ? responsesShouldInject(last) : chatShouldInject(last))) return;
+    const payload = event.payload as Record<string, unknown>;
+    // OpenAI Responses API carries the conversation in `input` instead of `messages`.
+    const key = Array.isArray(payload.messages) ? "messages" : Array.isArray(payload.input) ? "input" : null;
+    if (!key) return;
+    const msgs = payload[key] as Array<Record<string, unknown>>;
+    if (!injectable(msgs)) return;
 
     const ts = `\n${formatTimestamp()}`;
+    const last = msgs[msgs.length - 1];
 
     if (typeof last.content === "string") {
       last.content = last.content.replace(TS_STRIP, "") + ts;
     } else if (Array.isArray(last.content)) {
-      // Provider-shaped block arrays: normalize the last text-bearing block in
-      // place, or append a typed text block when none exists.
-      const blocks = [...(last.content as Rec[])];
+      // Provider-shaped block arrays (e.g. Anthropic): normalize the last
+      // text block in place, or append a text block when none exists.
+      const blocks = [...(last.content as Array<Record<string, unknown>>)];
       const idx = blocks.length - 1;
-      const lastBlock = blocks[idx] as { text?: unknown } | undefined;
-      if (lastBlock && typeof lastBlock.text === "string") {
+      const lastBlock = blocks[idx] as { type?: string; text?: string } | undefined;
+      if (lastBlock && lastBlock.type === "text" && typeof lastBlock.text === "string") {
         blocks[idx] = { ...lastBlock, text: lastBlock.text.replace(TS_STRIP, "") + ts };
-      } else if (isResponses && last.role === "assistant") {
-        blocks.push({ type: "output_text", text: ts, annotations: [] });
-      } else if (isResponses) {
-        blocks.push({ type: "input_text", text: ts });
       } else {
         blocks.push({ type: "text", text: ts });
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@taylorsatula/pi-timestamp",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "description": "Timestamp injection extension for Pi — appends the current wall-clock time to the tail of user messages and vocalizations just-in-time (cache-safe, no synthetic tool calls), so the model always knows what time it is.",
   "keywords": [
     "pi-package"