fix some typing errors

Reapor-Yurnero · Feb 1, 2026 · Feb 1, 2026 · Feb 1, 2026 · Feb 1, 2026 · Feb 1, 2026
commit 1ae9c115f1a965167a0f474ef8a5714c95b7ee11
diff --git a/src/agents/cache-trace.ts b/src/agents/cache-trace.ts
@@ -39,6 +39,7 @@ export type CacheTraceEvent = {
   messageFingerprints?: string[];
   messagesDigest?: string;
   systemDigest?: string;
+  guardrailBlocked?: string;
   note?: string;
   error?: string;
 };

diff --git a/src/agents/guardrails.ts b/src/agents/guardrails.ts
@@ -330,6 +330,13 @@ const GRAYSWAN_DEFAULT_TIMEOUT_MS = 30_000;
 
 const grayswanLogger = createSubsystemLogger("guardrails/grayswan");
 
+function makeGrayswanMessage(
+  role: GrayswanMonitorMessage["role"],
+  content: string,
+): GrayswanMonitorMessage {
+  return { role, content };
+}
+
 function resolveGrayswanConfig(context: GuardrailContext): GrayswanGuardrailConfig | undefined {
   const cfg = context.config?.guardrails?.grayswan;
   if (!cfg) {
@@ -711,9 +718,9 @@ function createGrayswanGuardrail(): Guardrail {
         return;
       }
       const includeHistory = stageCfg?.includeHistory !== false;
-      const messages = includeHistory
-        ? [...toGrayswanMessages(input.messages), { role: "user", content: prompt }]
-        : [{ role: "user", content: prompt }];
+      const messages: GrayswanMonitorMessage[] = includeHistory
+        ? [...toGrayswanMessages(input.messages), makeGrayswanMessage("user", prompt)]
+        : [makeGrayswanMessage("user", prompt)];
       if (messages.length === 0) {
         return;
       }
@@ -765,9 +772,9 @@ function createGrayswanGuardrail(): Guardrail {
       }
       const toolSummary = buildToolCallSummary(input);
       const includeHistory = stageCfg?.includeHistory !== false;
-      const messages = includeHistory
-        ? [...toGrayswanMessages(input.messages), { role: "assistant", content: toolSummary }]
-        : [{ role: "assistant", content: toolSummary }];
+      const messages: GrayswanMonitorMessage[] = includeHistory
+        ? [...toGrayswanMessages(input.messages), makeGrayswanMessage("assistant", toolSummary)]
+        : [makeGrayswanMessage("assistant", toolSummary)];
       if (messages.length === 0) {
         return;
       }
@@ -819,9 +826,9 @@ function createGrayswanGuardrail(): Guardrail {
       }
       const toolText = extractToolResultText(input.result).trim();
       const includeHistory = stageCfg?.includeHistory !== false;
-      const messages = includeHistory
-        ? [...toGrayswanMessages(input.messages), { role: "tool", content: toolText }]
-        : [{ role: "tool", content: toolText }];
+      const messages: GrayswanMonitorMessage[] = includeHistory
+        ? [...toGrayswanMessages(input.messages), makeGrayswanMessage("tool", toolText)]
+        : [makeGrayswanMessage("tool", toolText)];
       if (!toolText || messages.length === 0) {
         return;
       }
@@ -886,7 +893,10 @@ function createGrayswanGuardrail(): Guardrail {
       }
       const includeHistory = stageCfg?.includeHistory !== false;
       const historyMessages = includeHistory ? toGrayswanMessages(input.messages) : [];
-      const messages = [...historyMessages, { role: "assistant", content: assistantText }];
+      const messages: GrayswanMonitorMessage[] = [
+        ...historyMessages,
+        makeGrayswanMessage("assistant", assistantText),
+      ];
       let response: GrayswanMonitorResponse | null = null;
       try {
         response = await callGrayswanMonitor({ cfg, messages });

diff --git a/src/agents/pi-embedded-runner/run.ts b/src/agents/pi-embedded-runner/run.ts
@@ -370,6 +370,13 @@ export async function runEmbeddedPiAgent(
 
           const { aborted, promptError, timedOut, sessionIdUsed, lastAssistant, guardrailBlock } =
             attempt;
+          const guardrailBlockSummary = guardrailBlock
+            ? {
+                stage: guardrailBlock.stage,
+                guardrailId: guardrailBlock.guardrailId,
+                reason: guardrailBlock.reason,
+              }
+            : undefined;
 
           if (guardrailBlock) {
             const isBeforeRequest = guardrailBlock.stage === "before_request";
@@ -420,11 +427,7 @@ export async function runEmbeddedPiAgent(
                 agentMeta,
                 aborted,
                 systemPromptReport: attempt.systemPromptReport,
-                guardrailBlock: {
-                  stage: guardrailBlock.stage,
-                  guardrailId: guardrailBlock.guardrailId,
-                  reason: guardrailBlock.reason,
-                },
+                guardrailBlock: guardrailBlockSummary,
               },
               didSendViaMessagingTool: attempt.didSendViaMessagingTool,
               messagingToolSentTexts: attempt.messagingToolSentTexts,
@@ -730,13 +733,7 @@ export async function runEmbeddedPiAgent(
               agentMeta,
               aborted,
               systemPromptReport: attempt.systemPromptReport,
-              guardrailBlock: guardrailBlock
-                ? {
-                    stage: guardrailBlock.stage,
-                    guardrailId: guardrailBlock.guardrailId,
-                    reason: guardrailBlock.reason,
-                  }
-                : undefined,
+              guardrailBlock: guardrailBlockSummary,
               // Handle client tool calls (OpenResponses hosted tools)
               stopReason: attempt.clientToolCall ? "tool_calls" : undefined,
               pendingToolCalls: attempt.clientToolCall