@markusylisiurunen/tau 0.3.71 → 0.3.73

package/dist/protocol/session_protocol.d.ts

@@ -1,5 +1,6 @@
 import type { AssistantMessage, Message, TextContent, ThinkingContent, Tool, ToolCall } from "@earendil-works/pi-ai";
-export declare const SESSION_PROTOCOL_VERSION: 13;
+import { type IntermediateSystemMessage } from "./system_message.js";
+export declare const SESSION_PROTOCOL_VERSION: 14;
 export declare const SESSION_PROTOCOL_MAX_EXEC_CAPTURE_BYTES: number;
 export declare const SESSION_PROTOCOL_MAX_EXEC_STDIN_BYTES: number;
 export declare const SESSION_PROTOCOL_MAX_CLIENT_TOOL_PRESENTATION_BYTES: number;
@@ -109,7 +110,9 @@ export type SessionProtocolCancelExecParams = SessionProtocolSessionIdParams & {
 };
 export type SessionProtocolSampleContext = {
     systemPrompt: string;
-    messages: Message[];
+    messages: (Exclude<Message, {
+        role: "system";
+    }> | IntermediateSystemMessage)[];
     tools?: Tool[];
 };
 export type SessionProtocolSampleOptions = {
@@ -363,7 +366,9 @@ export type SessionProtocolDraftAssistantMessage = {
     content: (TextContent | ThinkingContent | ToolCall)[];
     timestamp: number;
 };
-export type SessionProtocolMessagePayload = SessionProtocolSystemMessage | Message | SessionProtocolDraftAssistantMessage;
+export type SessionProtocolMessagePayload = SessionProtocolSystemMessage | IntermediateSystemMessage | Exclude<Message, {
+    role: "system";
+}> | SessionProtocolDraftAssistantMessage;
 export type SessionProtocolMessageState = "draft" | "committed" | "interrupted" | "discarded";
 export type SessionProtocolMessage = {
     id: string;
@@ -795,7 +800,7 @@ export type SessionProtocolParsedSuccessResponseMessage = Omit<SessionProtocolSu
 };
 export type SessionProtocolParsedResponseMessage = SessionProtocolParsedSuccessResponseMessage | SessionProtocolErrorResponseMessage;
 export type SessionProtocolDeltaCause = {
-    type: "user-message" | "assistant-stream" | "assistant-message" | "tool-run" | "tool-result" | "notice" | "agent-run" | "maintenance" | "configuration" | "goal";
+    type: "user-message" | "system-message" | "assistant-stream" | "assistant-message" | "tool-run" | "tool-result" | "notice" | "agent-run" | "maintenance" | "configuration" | "goal";
 } | {
     type: "compaction";
     previousEpoch: number;

package/dist/protocol/session_protocol.js

@@ -1,5 +1,6 @@
 import { z } from "zod";
-export const SESSION_PROTOCOL_VERSION = 13;
+import { isIntermediateSystemMessage } from "./system_message.js";
+export const SESSION_PROTOCOL_VERSION = 14;
 export const SESSION_PROTOCOL_MAX_EXEC_CAPTURE_BYTES = 24 * 1024 * 1024;
 export const SESSION_PROTOCOL_MAX_EXEC_STDIN_BYTES = 16 * 1024 * 1024;
 export const SESSION_PROTOCOL_MAX_CLIENT_TOOL_PRESENTATION_BYTES = 1024 * 1024;
@@ -420,6 +421,7 @@ const modelToolResultMessageSchema = z
 })
     .strip();
 const modelMessageSchema = z.union([
+    z.custom(isIntermediateSystemMessage),
     modelUserMessageSchema,
     modelAssistantMessageSchema,
     modelToolResultMessageSchema,
@@ -921,13 +923,11 @@ const sessionProtocolExecutionEnvironmentSnapshotSchema = z.discriminatedUnion("
     sessionProtocolCloudflareSandboxExecutionEnvironmentSnapshotSchema,
     sessionProtocolFlySpriteExecutionEnvironmentSnapshotSchema,
 ]);
-const sessionProtocolSystemMessageSchema = z
-    .object({
+const sessionProtocolSystemMessageSchema = z.strictObject({
     role: z.literal("system"),
     content: z.string(),
     timestamp: z.number().finite(),
-})
-    .strip();
+});
 const sessionProtocolDraftAssistantMessageSchema = z
     .object({
     role: z.literal("assistant"),
@@ -1297,6 +1297,19 @@ const sessionProtocolSnapshotSchema = z
                 message: `duplicate message id '${message.id}'`,
             });
         }
+        if (message.message.role === "system") {
+            const isInitialPrompt = message === snapshot.messages[0];
+            const isIntermediate = "metadata" in message.message;
+            if (isInitialPrompt === isIntermediate) {
+                ctx.addIssue({
+                    code: "custom",
+                    path: ["messages"],
+                    message: isInitialPrompt
+                        ? "the initial system prompt must not have intermediate metadata"
+                        : "intermediate system messages require metadata",
+                });
+            }
+        }
         messagesById.set(message.id, message);
     }
     const checkpoint = snapshot.agentState.usageCheckpoint;
@@ -1544,6 +1557,7 @@ const sessionProtocolDeltaCauseSchema = z.union([
         .object({
         type: z.enum([
             "user-message",
+            "system-message",
             "assistant-stream",
             "assistant-message",
             "tool-run",