@taylorsatula/pi-timestamp 0.2.0 → 0.2.1

package/README.md

@@ -12,13 +12,20 @@ They're injected poorly if they're that annoying.
 [11:34pm - 9/1/26]
 ```
 
-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.
+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.
 
 ## Why plain-text tail append (design notes)
 
@@ -35,6 +42,9 @@ Injection happens only at **turn boundaries**, never mid-tool-loop:
 - **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
 
@@ -61,6 +71,9 @@ 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,6 +14,15 @@ 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*$/;
 
@@ -30,40 +39,73 @@ function formatTimestamp(): string {
   return `[${hours}:${String(minutes).padStart(2, "0")}${ampm} - ${month}/${day}/${year}]`;
 }
 
-function injectable(msgs: Array<Record<string, unknown>>): boolean {
-  const last = msgs[msgs.length - 1];
-  if (!last) return false;
-
-  if (last.role === "user") return true;
+// 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 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", ... }
 
-  return false; // tool results → mid-loop, skip
+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
 }
 
 export default function (pi: ExtensionAPI) {
   pi.on("before_provider_request", (event) => {
-    const payload = event.payload as Record<string, unknown>;
-    const msgs = payload.messages as Array<Record<string, unknown>>;
-    if (!injectable(msgs)) return;
+    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 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 (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>>)];
+      // 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[])];
       const idx = blocks.length - 1;
-      const lastBlock = blocks[idx] as { type?: string; text?: string } | undefined;
-      if (lastBlock && lastBlock.type === "text" && typeof lastBlock.text === "string") {
+      const lastBlock = blocks[idx] as { text?: unknown } | undefined;
+      if (lastBlock && 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.0",
+  "version": "0.2.1",
   "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"