@pi-harness/pi-harness 0.1.13 → 0.1.14

package/packages/core/src/plugins/context.ts

@@ -2,6 +2,7 @@ import type { Context } from "@deepseek-ai/cordis";
 import type { AgentSessionEvent } from "@earendil-works/pi-coding-agent";
 import { Type } from "@earendil-works/pi-ai";
 import { defineTool, type AgentToolResult } from "@earendil-works/pi-coding-agent";
+import { EmptyConfig } from "../config.js";
 
 type RuntimeEvent = AgentSessionEvent & { readonly type?: string };
 type MessageRole = "user" | "assistant" | "toolResult" | "system" | "other";
@@ -11,16 +12,76 @@ type ContextInsightReport = {
   contextWindow: number | null;
   percent: number | null;
   messages: number;
+  scannedMessages: number;
+  messagesTruncated: boolean;
   events: number;
   compactions: number;
   composition: ContextComposition;
   eventTypes: Record<string, number>;
   recentEvents: { type: string; at: number }[];
 };
+type ContextSessionSnapshot = Pick<
+  ContextInsightReport,
+  "tokens" | "contextWindow" | "percent" | "messages" | "scannedMessages" | "messagesTruncated" | "composition"
+>;
+
+const maxRecentEvents = 50;
+const maxEventTypes = 64;
+const maxEventTypeLength = 128;
+const maxScannedMessages = 10_000;
+const contextRefreshEvents = new Set(["message_end", "tool_execution_end", "agent_end", "agent_settled", "compaction_end", "entry_appended"]);
+
+function boundedCount(value: number): number {
+  return value >= Number.MAX_SAFE_INTEGER ? Number.MAX_SAFE_INTEGER : value + 1;
+}
+
+function normalizedEventType(value: unknown): string {
+  if (typeof value !== "string") return "unknown";
+  const type = value.trim().replaceAll("\0", "�");
+  if (type.length === 0) return "unknown";
+  return type.length <= maxEventTypeLength ? type : "other";
+}
+
+function nonNegativeInteger(value: unknown): number | null {
+  return typeof value === "number" && Number.isSafeInteger(value) && value >= 0 ? value : null;
+}
+
+function validPercent(value: unknown): number | null {
+  return typeof value === "number" && Number.isFinite(value) && value >= 0 ? value : null;
+}
+
+function parseEmptyParams(value: unknown): void {
+  if (typeof value !== "object" || value === null || Array.isArray(value)) throw new Error("Context inspect parameters must be an object");
+  let prototype: unknown;
+  let descriptors: PropertyDescriptorMap;
+  try {
+    prototype = Object.getPrototypeOf(value) as unknown;
+    descriptors = Object.getOwnPropertyDescriptors(value);
+  } catch {
+    throw new Error("Context inspect parameters could not be inspected safely");
+  }
+  if (prototype !== Object.prototype && prototype !== null) throw new Error("Context inspect parameters must be a plain object");
+  if (Reflect.ownKeys(descriptors).length > 0) throw new Error("Context inspect parameters contain an unknown property");
+}
+
+function throwIfAborted(signal: AbortSignal): void {
+  if (!signal.aborted) return;
+  throw signal.reason instanceof Error ? signal.reason : new Error("Context inspection was cancelled", { cause: signal.reason });
+}
+
+function dataProperty(value: unknown, key: PropertyKey): unknown {
+  if (typeof value !== "object" || value === null) return undefined;
+  try {
+    const descriptor = Object.getOwnPropertyDescriptor(value, key);
+    return descriptor !== undefined && "value" in descriptor ? descriptor.value : undefined;
+  } catch {
+    return undefined;
+  }
+}
 
 function messageRole(message: unknown): MessageRole {
   if (typeof message !== "object" || message === null) return "other";
-  const role = (message as { role?: unknown }).role;
+  const role = dataProperty(message, "role");
   if (role === "user" || role === "assistant" || role === "toolResult" || role === "system") return role;
   return "other";
 }
@@ -28,75 +89,139 @@ function messageRole(message: unknown): MessageRole {
 export default {
   name: "pi-context",
   inject: ["piPluginUi", "piTools"],
+  Config: EmptyConfig,
   apply(context: Context) {
+    const lifecycle = new AbortController();
     let eventCount = 0;
     let compactionCount = 0;
     const eventTypes = new Map<string, number>();
     const recentEvents: { type: string; at: number }[] = [];
-    const onEvent = (event: RuntimeEvent) => {
-      eventCount += 1;
-      const type = event.type ?? "unknown";
-      eventTypes.set(type, (eventTypes.get(type) ?? 0) + 1);
-      recentEvents.push({ type, at: Date.now() });
-      if (recentEvents.length > 50) recentEvents.shift();
-      if (type.startsWith("compaction_")) compactionCount += 1;
+    let cachedSession: unknown;
+    let cachedSnapshot: ContextSessionSnapshot = {
+      tokens: null,
+      contextWindow: null,
+      percent: null,
+      messages: 0,
+      scannedMessages: 0,
+      messagesTruncated: false,
+      composition: { user: 0, assistant: 0, toolResult: 0, system: 0, other: 0 },
     };
-    const unsubscribe = context.on("pi/session-event", onEvent);
-    const report = (): ContextInsightReport => {
+    const resetEvents = (): void => {
+      eventCount = 0;
+      compactionCount = 0;
+      eventTypes.clear();
+      recentEvents.length = 0;
+    };
+    const refreshSnapshot = (): void => {
       const runtime = context.get("piRuntime");
-      if (runtime === undefined) {
-        return {
-          tokens: null,
-          contextWindow: null,
-          percent: null,
-          messages: 0,
-          events: eventCount,
-          compactions: compactionCount,
-          composition: { user: 0, assistant: 0, toolResult: 0, system: 0, other: 0 },
-          eventTypes: Object.fromEntries(eventTypes),
-          recentEvents: recentEvents.map((event) => ({ ...event })),
-        };
+      const session = runtime?.session;
+      if (session !== cachedSession) resetEvents();
+      cachedSession = session;
+      let usage: unknown;
+      if (session !== undefined) {
+        try {
+          usage = session.getContextUsage();
+        } catch {
+          usage = undefined;
+        }
       }
-      const usage = runtime.session.getContextUsage();
+      let messages: readonly unknown[] = [];
+      if (session !== undefined) {
+        try {
+          if (Array.isArray(session.messages)) messages = session.messages;
+        } catch {
+          messages = [];
+        }
+      }
+      const rawMessageCount = dataProperty(messages, "length");
+      const messageCount = typeof rawMessageCount === "number" && Number.isSafeInteger(rawMessageCount) && rawMessageCount >= 0 ? rawMessageCount : 0;
+      const scannedMessages = Math.min(messageCount, maxScannedMessages);
+      const start = messageCount - scannedMessages;
       const composition: ContextComposition = { user: 0, assistant: 0, toolResult: 0, system: 0, other: 0 };
-      for (const message of runtime.session.messages) composition[messageRole(message)] += 1;
+      for (let index = start; index < messageCount; index += 1) composition[messageRole(dataProperty(messages, String(index)))] += 1;
+      cachedSnapshot = {
+        tokens: nonNegativeInteger(dataProperty(usage, "tokens")),
+        contextWindow: nonNegativeInteger(dataProperty(usage, "contextWindow")),
+        percent: validPercent(dataProperty(usage, "percent")),
+        messages: messageCount,
+        scannedMessages,
+        messagesTruncated: messageCount > scannedMessages,
+        composition,
+      };
+    };
+    const ensureCurrentSession = (): void => {
+      if (context.get("piRuntime")?.session !== cachedSession) refreshSnapshot();
+    };
+    const report = (): ContextInsightReport => {
+      ensureCurrentSession();
       return {
-        tokens: usage?.tokens ?? null,
-        contextWindow: usage?.contextWindow ?? null,
-        percent: usage?.percent ?? null,
-        messages: runtime.session.messages.length,
+        ...cachedSnapshot,
+        composition: { ...cachedSnapshot.composition },
         events: eventCount,
         compactions: compactionCount,
-        composition,
         eventTypes: Object.fromEntries(eventTypes),
         recentEvents: recentEvents.map((event) => ({ ...event })),
       };
     };
+    const onEvent = (event: RuntimeEvent) => {
+      ensureCurrentSession();
+      eventCount = boundedCount(eventCount);
+      let type = normalizedEventType(dataProperty(event, "type"));
+      if (!eventTypes.has(type) && eventTypes.size >= maxEventTypes - 1) type = "other";
+      eventTypes.set(type, boundedCount(eventTypes.get(type) ?? 0));
+      recentEvents.push({ type, at: Date.now() });
+      if (recentEvents.length > maxRecentEvents) recentEvents.shift();
+      if (type === "compaction_start") compactionCount = boundedCount(compactionCount);
+      if (contextRefreshEvents.has(type)) refreshSnapshot();
+    };
+    const unsubscribe = context.on("pi/session-event", onEvent);
+    context.effect(() => () => {
+      lifecycle.abort(new Error("Context Insights plugin disposed"));
+      unsubscribe();
+    });
+    refreshSnapshot();
     const unregisterTool = context.piTools.register(
       defineTool({
         name: "context_inspect",
         label: "Context inspect",
         description: "Inspect current context usage, message composition, and recent lifecycle events.",
         promptSnippet: "inspect the current context composition and recent events",
-        parameters: Type.Object({}),
-        execute(): Promise<AgentToolResult<ContextInsightReport>> {
-          const details = report();
-          return Promise.resolve({ content: [{ type: "text", text: `${details.messages} messages, ${details.events} context events.` }], details });
+        parameters: Type.Object({}, { additionalProperties: false }),
+        executionMode: "sequential",
+        execute(_toolCallId, params, signal): Promise<AgentToolResult<ContextInsightReport>> {
+          return Promise.resolve().then(() => {
+            parseEmptyParams(params);
+            const actionSignal = signal === undefined ? lifecycle.signal : AbortSignal.any([signal, lifecycle.signal]);
+            throwIfAborted(actionSignal);
+            refreshSnapshot();
+            const details = report();
+            return {
+              content: [
+                { type: "text" as const, text: `${details.messages} messages (${details.scannedMessages} scanned), ${details.events} context events.` },
+              ],
+              details,
+            };
+          });
         },
       }),
     );
+    context.effect(() => unregisterTool);
     const disposePanel = context.piPluginUi.register({
       id: "context-insight-panel",
       pluginId: "@pi-harness/core/plugins/context",
       title: "上下文洞察",
       description: "查看当前上下文占用、消息规模和压缩事件。",
       icon: "◒",
-      read: report,
-    });
-    context.effect(() => () => {
-      unsubscribe();
-      unregisterTool();
-      disposePanel();
+      read: () => ({
+        ...report(),
+        limits: {
+          scannedMessages: maxScannedMessages,
+          recentEvents: maxRecentEvents,
+          eventTypes: maxEventTypes,
+          eventTypeCharacters: maxEventTypeLength,
+        },
+      }),
     });
+    context.effect(() => disposePanel);
   },
 };