@agent-marshal/marshal-prime-agent 0.49.4 → 0.49.5

package/README.md

@@ -5,9 +5,12 @@
 `marshal-pi` package and targets Prime Agent's conversation lifecycle and canonical
 session identities.
 
-The extension runs inside each Prime Agent conversation. It:
+The extension runs inside each top-level Prime Agent conversation. Prime Agent
+RLM sub-agents stay part of their parent's Marshal session and are not added to the
+roster independently. It:
 
-- registers the conversation on the Marshal roster using Prime Agent's session ID;
+- registers the top-level conversation on the Marshal roster using Prime Agent's
+  session ID;
 - exposes native `marshal_*` tools to the model;
 - subscribes to Marshal's live notification stream; and
 - injects incoming direct messages with

package/package.json

@@ -46,8 +46,9 @@
     "url": "https://github.com/ignition-is-go/marshal"
   },
   "scripts": {
-    "gen": "cargo run -p marshal-entities --no-default-features --features codegen --bin typegen -- src/generated"
+    "gen": "cargo run -p marshal-entities --no-default-features --features codegen --bin typegen -- src/generated",
+    "test": "bun test"
   },
   "type": "module",
-  "version": "0.49.4"
+  "version": "0.49.5"
 }

package/src/index.ts

@@ -25,6 +25,7 @@ import { Type } from "typebox";
 import { MarshalDaemon } from "./daemon.ts";
 import { resolveIdentity, type Identity } from "./identity.ts";
 import type { NotifyChannelMeta, SessionItem } from "./entities.ts";
+import { shouldRegisterPrimeAgentSession } from "./runtime.ts";
 
 const DEFAULT_ADDRESS = "ws://localhost:6155";
 
@@ -218,6 +219,15 @@ async function init(pi: ExtensionAPI) {
   }
 
   pi.on("session_start", async (_event, ctx) => {
+    // Prime Agent loads extensions for RLM children too. The persisted header's
+    // RLM depth is the canonical distinction; parentSession alone is not enough
+    // because ordinary top-level forks also have a parent.
+    const header = ctx.sessionManager.getHeader() as { rlmDepth?: unknown } | null;
+    if (!shouldRegisterPrimeAgentSession(header)) {
+      log("RLM sub-agent detected; skipping Marshal session registration");
+      return;
+    }
+
     sessionCtx = ctx;
     sessionId = ctx.sessionManager.getSessionId();
 
@@ -309,6 +319,10 @@ async function init(pi: ExtensionAPI) {
   // ── Per-turn context injection ─────────────────────────────────────────
 
   pi.on("before_agent_start", async (event, _ctx) => {
+    // A skipped RLM child has no Marshal identity. Do not tell it that it is
+    // connected or inject any coordination context intended for its parent.
+    if (!sessionId) return;
+
     const result: { systemPrompt?: string; message?: { customType: string; content: string; display: boolean } } = {};
 
     // Add marshal system prompt context (once per turn, before the LLM call).

package/src/runtime.ts

@@ -0,0 +1,19 @@
+/** The small part of Prime Agent's session header used for registration. */
+export interface PrimeAgentSessionHeader {
+  rlmDepth?: unknown;
+  [key: string]: unknown;
+}
+
+function isPositiveIntegerDepth(value: unknown): boolean {
+  return typeof value === "number" && Number.isInteger(value) && value > 0;
+}
+
+/** Return whether a canonical Prime Agent session depth belongs to an RLM child. */
+export function isPrimeAgentSubagent(rlmDepth: unknown): boolean {
+  return isPositiveIntegerDepth(rlmDepth);
+}
+
+/** Public-API guard for the session_start registration boundary. */
+export function shouldRegisterPrimeAgentSession(header: PrimeAgentSessionHeader | null): boolean {
+  return !isPrimeAgentSubagent(header?.rlmDepth);
+}