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

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.8"
 }

package/src/generated/index.ts

@@ -164,11 +164,6 @@ export type { SetSessionLastActivityAtArgs } from "./SetSessionLastActivityAtArg
 export type { SetSessionLastToolArgs } from "./SetSessionLastToolArgs";
 export type { SetSessionLastToolAtArgs } from "./SetSessionLastToolAtArgs";
 export type { SetSessionSessionNameArgs } from "./SetSessionSessionNameArgs";
-export type { StartProfile } from "./StartProfile";
-export type { StartProfileArgs } from "./StartProfileArgs";
-export type { StopProfile } from "./StopProfile";
-export type { StopProfileArgs } from "./StopProfileArgs";
-export type { StopProfileOutput } from "./StopProfileOutput";
 export type { ViewError } from "./ViewError";
 export type { ViewWindowUpdate } from "./ViewWindowUpdate";
 export type { WakeStatus } from "./WakeStatus";
@@ -277,6 +272,84 @@ import type { SetSessionLastToolAt as _SetSessionLastToolAt } from "./SetSession
 import type { SetSessionSessionName as _SetSessionSessionName } from "./SetSessionSessionName";
 
 // Query classes
+export class GetSessionsByQuery {
+  static readonly queryId = "GetSessionsByQuery" as const;
+  static readonly queryItemType = "Session" as const;
+  readonly queryId = "GetSessionsByQuery" as const;
+  readonly queryItemType = "Session" as const;
+  readonly query: Omit<_GetSessionsByQuery, "tx" | "createdAt">;
+  declare readonly $res: () => Session[];
+
+  constructor(args: Omit<_GetSessionsByQuery, "tx" | "createdAt">) {
+    this.query = args;
+  }
+}
+
+export class GetSessionsByIds {
+  static readonly queryId = "GetSessionsByIds" as const;
+  static readonly queryItemType = "Session" as const;
+  readonly queryId = "GetSessionsByIds" as const;
+  readonly queryItemType = "Session" as const;
+  readonly query: Omit<_GetSessionsByIds, "tx" | "createdAt">;
+  declare readonly $res: () => Session[];
+
+  constructor(args: Omit<_GetSessionsByIds, "tx" | "createdAt">) {
+    this.query = args;
+  }
+}
+
+export class GetAllSessions {
+  static readonly queryId = "GetAllSessions" as const;
+  static readonly queryItemType = "Session" as const;
+  readonly queryId = "GetAllSessions" as const;
+  readonly queryItemType = "Session" as const;
+  readonly query: Omit<_GetAllSessions, "tx" | "createdAt">;
+  declare readonly $res: () => Session[];
+
+  constructor(args: Omit<_GetAllSessions, "tx" | "createdAt">) {
+    this.query = args;
+  }
+}
+
+export class GetRoomsByQuery {
+  static readonly queryId = "GetRoomsByQuery" as const;
+  static readonly queryItemType = "Room" as const;
+  readonly queryId = "GetRoomsByQuery" as const;
+  readonly queryItemType = "Room" as const;
+  readonly query: Omit<_GetRoomsByQuery, "tx" | "createdAt">;
+  declare readonly $res: () => Room[];
+
+  constructor(args: Omit<_GetRoomsByQuery, "tx" | "createdAt">) {
+    this.query = args;
+  }
+}
+
+export class GetRoomsByIds {
+  static readonly queryId = "GetRoomsByIds" as const;
+  static readonly queryItemType = "Room" as const;
+  readonly queryId = "GetRoomsByIds" as const;
+  readonly queryItemType = "Room" as const;
+  readonly query: Omit<_GetRoomsByIds, "tx" | "createdAt">;
+  declare readonly $res: () => Room[];
+
+  constructor(args: Omit<_GetRoomsByIds, "tx" | "createdAt">) {
+    this.query = args;
+  }
+}
+
+export class GetAllRooms {
+  static readonly queryId = "GetAllRooms" as const;
+  static readonly queryItemType = "Room" as const;
+  readonly queryId = "GetAllRooms" as const;
+  readonly queryItemType = "Room" as const;
+  readonly query: Omit<_GetAllRooms, "tx" | "createdAt">;
+  declare readonly $res: () => Room[];
+
+  constructor(args: Omit<_GetAllRooms, "tx" | "createdAt">) {
+    this.query = args;
+  }
+}
+
 export class GetMessageReadsByQuery {
   static readonly queryId = "GetMessageReadsByQuery" as const;
   static readonly queryItemType = "MessageRead" as const;
@@ -355,45 +428,6 @@ export class GetAllRoomMembers {
   }
 }
 
-export class GetSessionsByQuery {
-  static readonly queryId = "GetSessionsByQuery" as const;
-  static readonly queryItemType = "Session" as const;
-  readonly queryId = "GetSessionsByQuery" as const;
-  readonly queryItemType = "Session" as const;
-  readonly query: Omit<_GetSessionsByQuery, "tx" | "createdAt">;
-  declare readonly $res: () => Session[];
-
-  constructor(args: Omit<_GetSessionsByQuery, "tx" | "createdAt">) {
-    this.query = args;
-  }
-}
-
-export class GetSessionsByIds {
-  static readonly queryId = "GetSessionsByIds" as const;
-  static readonly queryItemType = "Session" as const;
-  readonly queryId = "GetSessionsByIds" as const;
-  readonly queryItemType = "Session" as const;
-  readonly query: Omit<_GetSessionsByIds, "tx" | "createdAt">;
-  declare readonly $res: () => Session[];
-
-  constructor(args: Omit<_GetSessionsByIds, "tx" | "createdAt">) {
-    this.query = args;
-  }
-}
-
-export class GetAllSessions {
-  static readonly queryId = "GetAllSessions" as const;
-  static readonly queryItemType = "Session" as const;
-  readonly queryId = "GetAllSessions" as const;
-  readonly queryItemType = "Session" as const;
-  readonly query: Omit<_GetAllSessions, "tx" | "createdAt">;
-  declare readonly $res: () => Session[];
-
-  constructor(args: Omit<_GetAllSessions, "tx" | "createdAt">) {
-    this.query = args;
-  }
-}
-
 export class GetMessagesByQuery {
   static readonly queryId = "GetMessagesByQuery" as const;
   static readonly queryItemType = "Message" as const;
@@ -433,45 +467,6 @@ export class GetAllMessages {
   }
 }
 
-export class GetRoomsByQuery {
-  static readonly queryId = "GetRoomsByQuery" as const;
-  static readonly queryItemType = "Room" as const;
-  readonly queryId = "GetRoomsByQuery" as const;
-  readonly queryItemType = "Room" as const;
-  readonly query: Omit<_GetRoomsByQuery, "tx" | "createdAt">;
-  declare readonly $res: () => Room[];
-
-  constructor(args: Omit<_GetRoomsByQuery, "tx" | "createdAt">) {
-    this.query = args;
-  }
-}
-
-export class GetRoomsByIds {
-  static readonly queryId = "GetRoomsByIds" as const;
-  static readonly queryItemType = "Room" as const;
-  readonly queryId = "GetRoomsByIds" as const;
-  readonly queryItemType = "Room" as const;
-  readonly query: Omit<_GetRoomsByIds, "tx" | "createdAt">;
-  declare readonly $res: () => Room[];
-
-  constructor(args: Omit<_GetRoomsByIds, "tx" | "createdAt">) {
-    this.query = args;
-  }
-}
-
-export class GetAllRooms {
-  static readonly queryId = "GetAllRooms" as const;
-  static readonly queryItemType = "Room" as const;
-  readonly queryId = "GetAllRooms" as const;
-  readonly queryItemType = "Room" as const;
-  readonly query: Omit<_GetAllRooms, "tx" | "createdAt">;
-  declare readonly $res: () => Room[];
-
-  constructor(args: Omit<_GetAllRooms, "tx" | "createdAt">) {
-    this.query = args;
-  }
-}
-
 export class GetSessionNicknamesByQuery {
   static readonly queryId = "GetSessionNicknamesByQuery" as const;
   static readonly queryItemType = "SessionNickname" as const;
@@ -514,6 +509,72 @@ export class GetAllSessionNicknames {
 // View classes
 
 // Report classes
+export class GetSessionById {
+  static readonly reportId = "GetSessionById" as const;
+  readonly reportId = "GetSessionById" as const;
+  readonly report: Omit<_GetSessionById, "tx">;
+  declare readonly $res: () => Session | null;
+
+  constructor(args: Omit<_GetSessionById, "tx">) {
+    this.report = args;
+  }
+}
+
+export class CountSessions {
+  static readonly reportId = "CountSessions" as const;
+  readonly reportId = "CountSessions" as const;
+  readonly report: Omit<_CountSessions, "tx">;
+  declare readonly $res: () => SessionCount;
+
+  constructor(args: Omit<_CountSessions, "tx">) {
+    this.report = args;
+  }
+}
+
+export class CountAllSessions {
+  static readonly reportId = "CountAllSessions" as const;
+  readonly reportId = "CountAllSessions" as const;
+  readonly report: Omit<_CountAllSessions, "tx">;
+  declare readonly $res: () => SessionCount;
+
+  constructor(args: Omit<_CountAllSessions, "tx">) {
+    this.report = args;
+  }
+}
+
+export class GetRoomById {
+  static readonly reportId = "GetRoomById" as const;
+  readonly reportId = "GetRoomById" as const;
+  readonly report: Omit<_GetRoomById, "tx">;
+  declare readonly $res: () => Room | null;
+
+  constructor(args: Omit<_GetRoomById, "tx">) {
+    this.report = args;
+  }
+}
+
+export class CountRooms {
+  static readonly reportId = "CountRooms" as const;
+  readonly reportId = "CountRooms" as const;
+  readonly report: Omit<_CountRooms, "tx">;
+  declare readonly $res: () => RoomCount;
+
+  constructor(args: Omit<_CountRooms, "tx">) {
+    this.report = args;
+  }
+}
+
+export class CountAllRooms {
+  static readonly reportId = "CountAllRooms" as const;
+  readonly reportId = "CountAllRooms" as const;
+  readonly report: Omit<_CountAllRooms, "tx">;
+  declare readonly $res: () => RoomCount;
+
+  constructor(args: Omit<_CountAllRooms, "tx">) {
+    this.report = args;
+  }
+}
+
 export class GetMessageReadById {
   static readonly reportId = "GetMessageReadById" as const;
   readonly reportId = "GetMessageReadById" as const;
@@ -580,39 +641,6 @@ export class CountAllRoomMembers {
   }
 }
 
-export class GetSessionById {
-  static readonly reportId = "GetSessionById" as const;
-  readonly reportId = "GetSessionById" as const;
-  readonly report: Omit<_GetSessionById, "tx">;
-  declare readonly $res: () => Session | null;
-
-  constructor(args: Omit<_GetSessionById, "tx">) {
-    this.report = args;
-  }
-}
-
-export class CountSessions {
-  static readonly reportId = "CountSessions" as const;
-  readonly reportId = "CountSessions" as const;
-  readonly report: Omit<_CountSessions, "tx">;
-  declare readonly $res: () => SessionCount;
-
-  constructor(args: Omit<_CountSessions, "tx">) {
-    this.report = args;
-  }
-}
-
-export class CountAllSessions {
-  static readonly reportId = "CountAllSessions" as const;
-  readonly reportId = "CountAllSessions" as const;
-  readonly report: Omit<_CountAllSessions, "tx">;
-  declare readonly $res: () => SessionCount;
-
-  constructor(args: Omit<_CountAllSessions, "tx">) {
-    this.report = args;
-  }
-}
-
 export class GetMessageById {
   static readonly reportId = "GetMessageById" as const;
   readonly reportId = "GetMessageById" as const;
@@ -646,39 +674,6 @@ export class CountAllMessages {
   }
 }
 
-export class GetRoomById {
-  static readonly reportId = "GetRoomById" as const;
-  readonly reportId = "GetRoomById" as const;
-  readonly report: Omit<_GetRoomById, "tx">;
-  declare readonly $res: () => Room | null;
-
-  constructor(args: Omit<_GetRoomById, "tx">) {
-    this.report = args;
-  }
-}
-
-export class CountRooms {
-  static readonly reportId = "CountRooms" as const;
-  readonly reportId = "CountRooms" as const;
-  readonly report: Omit<_CountRooms, "tx">;
-  declare readonly $res: () => RoomCount;
-
-  constructor(args: Omit<_CountRooms, "tx">) {
-    this.report = args;
-  }
-}
-
-export class CountAllRooms {
-  static readonly reportId = "CountAllRooms" as const;
-  readonly reportId = "CountAllRooms" as const;
-  readonly report: Omit<_CountAllRooms, "tx">;
-  declare readonly $res: () => RoomCount;
-
-  constructor(args: Omit<_CountAllRooms, "tx">) {
-    this.report = args;
-  }
-}
-
 export class GetSessionNicknameById {
   static readonly reportId = "GetSessionNicknameById" as const;
   readonly reportId = "GetSessionNicknameById" as const;
@@ -713,72 +708,6 @@ export class CountAllSessionNicknames {
 }
 
 // Command classes
-export class SendMessage {
-  static readonly commandId = "SendMessage" as const;
-  readonly commandId = "SendMessage" as const;
-  readonly command: Omit<_SendMessage, "tx" | "createdAt">;
-  declare readonly $res: () => SendMessageResult;
-
-  constructor(args: Omit<_SendMessage, "tx" | "createdAt">) {
-    this.command = args;
-  }
-}
-
-export class DeleteMessageReads {
-  static readonly commandId = "DeleteMessageReads" as const;
-  readonly commandId = "DeleteMessageReads" as const;
-  readonly command: Omit<_DeleteMessageReads, "tx" | "createdAt">;
-  declare readonly $res: () => DeleteMessageReadsResult;
-
-  constructor(args: Omit<_DeleteMessageReads, "tx" | "createdAt">) {
-    this.command = args;
-  }
-}
-
-export class DeleteMessageRead {
-  static readonly commandId = "DeleteMessageRead" as const;
-  readonly commandId = "DeleteMessageRead" as const;
-  readonly command: Omit<_DeleteMessageRead, "tx" | "createdAt">;
-  declare readonly $res: () => DeleteMessageReadResult;
-
-  constructor(args: Omit<_DeleteMessageRead, "tx" | "createdAt">) {
-    this.command = args;
-  }
-}
-
-export class NotifyChannel {
-  static readonly commandId = "NotifyChannel" as const;
-  readonly commandId = "NotifyChannel" as const;
-  readonly command: Omit<_NotifyChannel, "tx" | "createdAt">;
-  declare readonly $res: () => void;
-
-  constructor(args: Omit<_NotifyChannel, "tx" | "createdAt">) {
-    this.command = args;
-  }
-}
-
-export class DeleteRoomMembers {
-  static readonly commandId = "DeleteRoomMembers" as const;
-  readonly commandId = "DeleteRoomMembers" as const;
-  readonly command: Omit<_DeleteRoomMembers, "tx" | "createdAt">;
-  declare readonly $res: () => DeleteRoomMembersResult;
-
-  constructor(args: Omit<_DeleteRoomMembers, "tx" | "createdAt">) {
-    this.command = args;
-  }
-}
-
-export class DeleteRoomMember {
-  static readonly commandId = "DeleteRoomMember" as const;
-  readonly commandId = "DeleteRoomMember" as const;
-  readonly command: Omit<_DeleteRoomMember, "tx" | "createdAt">;
-  declare readonly $res: () => DeleteRoomMemberResult;
-
-  constructor(args: Omit<_DeleteRoomMember, "tx" | "createdAt">) {
-    this.command = args;
-  }
-}
-
 export class DeleteSessions {
   static readonly commandId = "DeleteSessions" as const;
   readonly commandId = "DeleteSessions" as const;
@@ -878,39 +807,6 @@ export class SetSessionGitBranch {
   }
 }
 
-export class BroadcastMessage {
-  static readonly commandId = "BroadcastMessage" as const;
-  readonly commandId = "BroadcastMessage" as const;
-  readonly command: Omit<_BroadcastMessage, "tx" | "createdAt">;
-  declare readonly $res: () => BroadcastMessageResult;
-
-  constructor(args: Omit<_BroadcastMessage, "tx" | "createdAt">) {
-    this.command = args;
-  }
-}
-
-export class DeleteMessages {
-  static readonly commandId = "DeleteMessages" as const;
-  readonly commandId = "DeleteMessages" as const;
-  readonly command: Omit<_DeleteMessages, "tx" | "createdAt">;
-  declare readonly $res: () => DeleteMessagesResult;
-
-  constructor(args: Omit<_DeleteMessages, "tx" | "createdAt">) {
-    this.command = args;
-  }
-}
-
-export class DeleteMessage {
-  static readonly commandId = "DeleteMessage" as const;
-  readonly commandId = "DeleteMessage" as const;
-  readonly command: Omit<_DeleteMessage, "tx" | "createdAt">;
-  declare readonly $res: () => DeleteMessageResult;
-
-  constructor(args: Omit<_DeleteMessage, "tx" | "createdAt">) {
-    this.command = args;
-  }
-}
-
 export class LeaveRoom {
   static readonly commandId = "LeaveRoom" as const;
   readonly commandId = "LeaveRoom" as const;
@@ -977,6 +873,116 @@ export class SetRoomDescription {
   }
 }
 
+export class BroadcastMessage {
+  static readonly commandId = "BroadcastMessage" as const;
+  readonly commandId = "BroadcastMessage" as const;
+  readonly command: Omit<_BroadcastMessage, "tx" | "createdAt">;
+  declare readonly $res: () => BroadcastMessageResult;
+
+  constructor(args: Omit<_BroadcastMessage, "tx" | "createdAt">) {
+    this.command = args;
+  }
+}
+
+export class DeleteMessageReads {
+  static readonly commandId = "DeleteMessageReads" as const;
+  readonly commandId = "DeleteMessageReads" as const;
+  readonly command: Omit<_DeleteMessageReads, "tx" | "createdAt">;
+  declare readonly $res: () => DeleteMessageReadsResult;
+
+  constructor(args: Omit<_DeleteMessageReads, "tx" | "createdAt">) {
+    this.command = args;
+  }
+}
+
+export class DeleteMessageRead {
+  static readonly commandId = "DeleteMessageRead" as const;
+  readonly commandId = "DeleteMessageRead" as const;
+  readonly command: Omit<_DeleteMessageRead, "tx" | "createdAt">;
+  declare readonly $res: () => DeleteMessageReadResult;
+
+  constructor(args: Omit<_DeleteMessageRead, "tx" | "createdAt">) {
+    this.command = args;
+  }
+}
+
+export class NotifyChannel {
+  static readonly commandId = "NotifyChannel" as const;
+  readonly commandId = "NotifyChannel" as const;
+  readonly command: Omit<_NotifyChannel, "tx" | "createdAt">;
+  declare readonly $res: () => void;
+
+  constructor(args: Omit<_NotifyChannel, "tx" | "createdAt">) {
+    this.command = args;
+  }
+}
+
+export class DeleteRoomMembers {
+  static readonly commandId = "DeleteRoomMembers" as const;
+  readonly commandId = "DeleteRoomMembers" as const;
+  readonly command: Omit<_DeleteRoomMembers, "tx" | "createdAt">;
+  declare readonly $res: () => DeleteRoomMembersResult;
+
+  constructor(args: Omit<_DeleteRoomMembers, "tx" | "createdAt">) {
+    this.command = args;
+  }
+}
+
+export class DeleteRoomMember {
+  static readonly commandId = "DeleteRoomMember" as const;
+  readonly commandId = "DeleteRoomMember" as const;
+  readonly command: Omit<_DeleteRoomMember, "tx" | "createdAt">;
+  declare readonly $res: () => DeleteRoomMemberResult;
+
+  constructor(args: Omit<_DeleteRoomMember, "tx" | "createdAt">) {
+    this.command = args;
+  }
+}
+
+export class DeleteMessages {
+  static readonly commandId = "DeleteMessages" as const;
+  readonly commandId = "DeleteMessages" as const;
+  readonly command: Omit<_DeleteMessages, "tx" | "createdAt">;
+  declare readonly $res: () => DeleteMessagesResult;
+
+  constructor(args: Omit<_DeleteMessages, "tx" | "createdAt">) {
+    this.command = args;
+  }
+}
+
+export class DeleteMessage {
+  static readonly commandId = "DeleteMessage" as const;
+  readonly commandId = "DeleteMessage" as const;
+  readonly command: Omit<_DeleteMessage, "tx" | "createdAt">;
+  declare readonly $res: () => DeleteMessageResult;
+
+  constructor(args: Omit<_DeleteMessage, "tx" | "createdAt">) {
+    this.command = args;
+  }
+}
+
+export class ReadMessages {
+  static readonly commandId = "ReadMessages" as const;
+  readonly commandId = "ReadMessages" as const;
+  readonly command: Omit<_ReadMessages, "tx" | "createdAt">;
+  declare readonly $res: () => ReadMessagesResult;
+
+  constructor(args: Omit<_ReadMessages, "tx" | "createdAt">) {
+    this.command = args;
+  }
+}
+
+export class SendMessage {
+  static readonly commandId = "SendMessage" as const;
+  readonly commandId = "SendMessage" as const;
+  readonly command: Omit<_SendMessage, "tx" | "createdAt">;
+  declare readonly $res: () => SendMessageResult;
+
+  constructor(args: Omit<_SendMessage, "tx" | "createdAt">) {
+    this.command = args;
+  }
+}
+
 export class DeleteSessionNicknames {
   static readonly commandId = "DeleteSessionNicknames" as const;
   readonly commandId = "DeleteSessionNicknames" as const;
@@ -999,17 +1005,6 @@ export class DeleteSessionNickname {
   }
 }
 
-export class ReadMessages {
-  static readonly commandId = "ReadMessages" as const;
-  readonly commandId = "ReadMessages" as const;
-  readonly command: Omit<_ReadMessages, "tx" | "createdAt">;
-  declare readonly $res: () => ReadMessagesResult;
-
-  constructor(args: Omit<_ReadMessages, "tx" | "createdAt">) {
-    this.command = args;
-  }
-}
-
 export class AckMessages {
   static readonly commandId = "AckMessages" as const;
   readonly commandId = "AckMessages" as const;
@@ -1023,11 +1018,11 @@ export class AckMessages {
 
 // Item constructors
 export const items = {
+  Session: (args: Session) => ({ item: args, itemType: "Session" }),
+  Room: (args: Room) => ({ item: args, itemType: "Room" }),
   MessageRead: (args: MessageRead) => ({ item: args, itemType: "MessageRead" }),
   RoomMember: (args: RoomMember) => ({ item: args, itemType: "RoomMember" }),
-  Session: (args: Session) => ({ item: args, itemType: "Session" }),
   Message: (args: Message) => ({ item: args, itemType: "Message" }),
-  Room: (args: Room) => ({ item: args, itemType: "Room" }),
   SessionNickname: (args: SessionNickname) => ({ item: args, itemType: "SessionNickname" }),
 };
 

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);
+}

package/src/generated/StartProfile.ts

@@ -1,10 +0,0 @@
-// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
-
-/**
- * Start CPU sampling of this server process.
- */
-export type StartProfile = { 
-/**
- * Sampling frequency in Hz. Defaults to 99 when omitted.
- */
-frequencyHz: number | null, };

package/src/generated/StartProfileArgs.ts

@@ -1,7 +0,0 @@
-// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
-
-export type StartProfileArgs = { 
-/**
- * Sampling frequency in Hz. Defaults to 99 when omitted.
- */
-frequencyHz: number | null, };

package/src/generated/StopProfile.ts

@@ -1,6 +0,0 @@
-// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
-
-/**
- * Stop CPU sampling and return the collected profile.
- */
-export type StopProfile = Record<string, never>;

package/src/generated/StopProfileArgs.ts

@@ -1,3 +0,0 @@
-// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
-
-export type StopProfileArgs = Record<string, never>;

package/src/generated/StopProfileOutput.ts

@@ -1,14 +0,0 @@
-// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
-
-/**
- * Result of `StopProfile`. `folded` is the agent-readable primary artifact.
- */
-export type StopProfileOutput = { 
-/**
- * Collapsed stacks, one per line: `root;child;leaf <count>`. Sortable/greppable.
- */
-folded: string, 
-/**
- * Absolute path to a flamegraph `.svg` written to disk for humans (if writable).
- */
-svgPath: string | null, sampleCount: number, durationMs: bigint, frequencyHz: number, };