@hassangameryt/freeflow 0.5.0 → 0.6.0

package/pi-extension/src/index.ts

@@ -1,38 +1,210 @@
-import { handleNativeToolSafetyNet } from "./output-router/native-safety-net.js";
-import { handleObservedToolRouting } from "./output-router/observed-tool-routing.js";
-import { registerRouterTools } from "./output-router/router-tools.js";
-import { handleFreeflowCommand, handleOutputRouterCommand } from "./settings/settings-ui.js";
+import {
+  COGNITIVE_ROUTING_CONTROL_ENTRY,
+  COGNITIVE_ROUTING_INTENT_ENTRY,
+  CognitiveRoutingController,
+} from "./cognitive-routing/controller.js";
+import {
+  cognitiveRoutingProfileCompletions,
+  handleCognitiveRoutingProfileCommand,
+} from "./cognitive-routing/commands.js";
+import { registerCognitiveRoutingHistoryTool, registerCognitiveRoutingTool } from "./cognitive-routing/tool.js";
+import { readCognitiveRoutingHistory } from "./cognitive-routing/history.js";
+import { ConversationHistoryRuntime } from "./conversation-history/runtime.js";
+import { FreeflowContextRuntime } from "./freeflow-context/runtime.js";
+import { CONTEXT_VIRTUALIZATION_TOOL_NAME, registerFreeflowContextTool } from "./freeflow-context/tool.js";
+import { handleContextCommand } from "./context-virtualization/commands.js";
+import { ContextVirtualizationRuntime } from "./context-virtualization/runtime.js";
+import { handleFreeflowCommand } from "./settings/settings-ui.js";
+import { isPiFlowHost } from "./runtime/runtime-identity.js";
 import {
   CONTRIBUTOR_COMMANDS,
+  COGNITIVE_ROUTING_SWITCH_TOOL_NAME,
   WORKFLOW_COMMANDS,
-  WORKFLOW_BOOTSTRAP_MESSAGE_TYPE,
-  FREEFLOW_STATUS_TOOL_NAME,
   freeflowModelSkillPaths,
   freeflowSkillPath,
   getRuntimeContext,
-  OUTPUT_ROUTER_TOOL_NAMES,
   readCapabilityState,
   readModeState,
-  readOutputRouterConfig,
   refreshRuntimeContext,
   restoreModeOverride,
   runtimeContext,
-  workflowBootstrapMessage,
+  filterBootstrapMessage,
   setModeStatus,
   skillPrompt,
-  notifyRouterConfigWarnings,
+  withFreeflowRuntimeState,
 } from "./runtime/runtime-context.js";
 
-function isOutputRouterToolName(name: string): boolean {
-  return OUTPUT_ROUTER_TOOL_NAMES.includes(name);
+function startupSelectionSuppressesCognitiveRouting(ctx: any): boolean {
+  return ctx?.modelStateProvenance?.explicitModel === true || ctx?.modelStateProvenance?.explicitThinking === true;
 }
 
-async function applyCapabilityToolVisibility(pi: any, ctx: any, capabilityState = undefined): Promise<void> {
+function unavailableCapability(capability: any, code: string, message: string): any {
+  return {
+    ...capability,
+    effective: false,
+    blockingReason: { code, message },
+  };
+}
+
+function modelFacingCapabilityState(
+  capabilityState: any,
+  ctx: any,
+  freeflowContext: any,
+  cognitiveRoutingController: CognitiveRoutingController | undefined,
+  cognitiveRoutingActivationFailed = false,
+): any {
+  const surfaceState = {
+    ...capabilityState,
+    interactionContract: { ...capabilityState?.interactionContract },
+    skills: { ...capabilityState?.skills },
+    contextVirtualization: { ...capabilityState?.contextVirtualization },
+    conversationHistory: { ...capabilityState?.conversationHistory },
+    cognitiveRouting: { ...capabilityState?.cognitiveRouting },
+  };
+  if (surfaceState.enabled !== true) return surfaceState;
+
+  const markUnavailable = (key: string, message: string) => {
+    if (surfaceState[key]?.effective === true) {
+      surfaceState[key] = unavailableCapability(surfaceState[key], "unavailable", message);
+    }
+  };
+
+  if (!freeflowContext?.corePrompt) {
+    for (const key of [
+      "interactionContract",
+      "skills",
+      "contextVirtualization",
+      "conversationHistory",
+      "cognitiveRouting",
+    ]) {
+      markUnavailable(key, "Freeflow core prompt is unavailable.");
+    }
+    return surfaceState;
+  }
+
+  if (surfaceState.interactionContract?.effective === true && !freeflowContext.interactionContractPrompt) {
+    markUnavailable("interactionContract", "Interaction Contract prompt is unavailable.");
+  }
+  if (surfaceState.skills?.effective === true && !freeflowContext.skillsPrompt) {
+    markUnavailable("skills", "Skills prompt is unavailable.");
+  }
+
+  if (surfaceState.skills?.effective === true) {
+    if (surfaceState.contextVirtualization?.effective === true && !freeflowContext.contextVirtualizationPrompt) {
+      markUnavailable("contextVirtualization", "Context Virtualization prompt is unavailable.");
+    }
+    if (surfaceState.conversationHistory?.effective === true && !freeflowContext.conversationHistoryPrompt) {
+      markUnavailable("conversationHistory", "Conversation History prompt is unavailable.");
+    }
+  } else {
+    for (const key of ["contextVirtualization", "conversationHistory", "cognitiveRouting"]) {
+      markUnavailable(key, "Skills prompt is unavailable.");
+    }
+  }
+
+  if (surfaceState.cognitiveRouting?.effective === true) {
+    if (startupSelectionSuppressesCognitiveRouting(ctx)) {
+      surfaceState.cognitiveRouting = unavailableCapability(
+        surfaceState.cognitiveRouting,
+        "runtime_inactive",
+        "Cognitive Routing is inactive because the host selected the startup model or thinking level.",
+      );
+    } else if (cognitiveRoutingActivationFailed || cognitiveRoutingController?.state()?.effective === false) {
+      surfaceState.cognitiveRouting = unavailableCapability(
+        surfaceState.cognitiveRouting,
+        "unavailable",
+        "Cognitive Routing could not activate for this session.",
+      );
+    } else if (!freeflowContext.cognitiveRoutingPrompt) {
+      markUnavailable("cognitiveRouting", "Cognitive Routing prompt is unavailable.");
+    }
+  }
+
+  return surfaceState;
+}
+
+function modelFacingModeState(modeState: any, capabilityState: any): any {
+  if (capabilityState?.skills?.effective === true) return modeState;
+  return { ...modeState, active: false, effectiveMode: null };
+}
+
+type SessionEntriesContext = {
+  sessionManager?: {
+    getBranch?: () => readonly unknown[];
+    getEntries?: () => readonly unknown[];
+  };
+};
+
+function sessionHasConversationOrRoutingState(ctx: SessionEntriesContext | undefined): boolean {
+  try {
+    const entries = ctx?.sessionManager?.getBranch?.() ?? ctx?.sessionManager?.getEntries?.() ?? [];
+    if (!Array.isArray(entries)) return false;
+    return entries.some((entry: unknown) => {
+      if (!entry || typeof entry !== "object") return false;
+      const record = entry as { type?: unknown; customType?: unknown };
+      if (
+        typeof record.type === "string" &&
+        ["message", "custom_message", "compaction", "branch_summary"].includes(record.type)
+      ) {
+        return true;
+      }
+      return (
+        record.type === "custom" &&
+        (record.customType === COGNITIVE_ROUTING_INTENT_ENTRY || record.customType === COGNITIVE_ROUTING_CONTROL_ENTRY)
+      );
+    });
+  } catch {
+    return false;
+  }
+}
+
+function hasCognitiveRoutingHost(pi: any, ctx: any): boolean {
+  return (
+    isPiFlowHost(pi?.host) &&
+    typeof pi?.appendEntryDurable === "function" &&
+    typeof pi?.acquireModelStateControl === "function" &&
+    typeof ctx?.modelRegistry?.getApiKeyAndHeaders === "function"
+  );
+}
+
+async function reconcileCognitiveRoutingController(
+  pi: any,
+  ctx: any,
+  capabilityState: any,
+  previous: CognitiveRoutingController | undefined,
+): Promise<CognitiveRoutingController | undefined> {
+  const cognitiveRouting = capabilityState?.cognitiveRouting;
+  if (
+    !isPiFlowHost(pi?.host) ||
+    !cognitiveRouting?.effective ||
+    startupSelectionSuppressesCognitiveRouting(ctx) ||
+    !hasCognitiveRoutingHost(pi, ctx)
+  ) {
+    if (previous) await previous.deactivate();
+    return undefined;
+  }
+
+  const controller = new CognitiveRoutingController({ capabilityState: cognitiveRouting, pi, ctx });
+  const recovered = await controller.recover();
+  if (recovered.status === "pending") return controller;
+  if (recovered.status !== "active") {
+    const activated = await controller.activate();
+    return activated.status === "active" ? controller : undefined;
+  }
+  return controller;
+}
+
+async function applyCapabilityToolVisibility(
+  pi: any,
+  ctx: any,
+  capabilityState = undefined,
+  cognitiveRoutingController: CognitiveRoutingController | undefined = undefined,
+): Promise<void> {
   if (typeof pi?.setActiveTools !== "function" || typeof pi?.getAllTools !== "function") {
     return;
   }
 
-  const state = capabilityState ?? (await readCapabilityState(ctx.cwd));
+  const state = capabilityState ?? (await readCapabilityState(ctx.cwd, ctx, pi?.host));
   const allToolNames = pi
     .getAllTools()
     .map((tool: any) => tool?.name)
@@ -43,25 +215,49 @@ async function applyCapabilityToolVisibility(pi: any, ctx: any, capabilityState
     (Array.isArray(currentActive) ? currentActive : allToolNames).filter((name: string) => allToolNameSet.has(name)),
   );
 
-  if (allToolNameSet.has(FREEFLOW_STATUS_TOOL_NAME)) {
-    if (state.configured && state.enabled) active.add(FREEFLOW_STATUS_TOOL_NAME);
-    else active.delete(FREEFLOW_STATUS_TOOL_NAME);
+  if (allToolNameSet.has(COGNITIVE_ROUTING_SWITCH_TOOL_NAME)) {
+    const controllerState = cognitiveRoutingController?.state();
+    const routingToolActive =
+      state.cognitiveRouting?.effective === true &&
+      controllerState?.effective === true &&
+      controllerState.controlMode === "automatic" &&
+      !startupSelectionSuppressesCognitiveRouting(ctx);
+    if (routingToolActive) active.add(COGNITIVE_ROUTING_SWITCH_TOOL_NAME);
+    else active.delete(COGNITIVE_ROUTING_SWITCH_TOOL_NAME);
   }
 
-  for (const toolName of OUTPUT_ROUTER_TOOL_NAMES) {
-    if (!allToolNameSet.has(toolName)) continue;
-    if (state.outputRouter.enabled) active.add(toolName);
-    else active.delete(toolName);
+  if (allToolNameSet.has(CONTEXT_VIRTUALIZATION_TOOL_NAME)) {
+    if (state.contextVirtualization?.effective === true || state.conversationHistory?.effective === true) {
+      active.add(CONTEXT_VIRTUALIZATION_TOOL_NAME);
+    } else {
+      active.delete(CONTEXT_VIRTUALIZATION_TOOL_NAME);
+    }
   }
 
   pi.setActiveTools([...active]);
 }
 
-async function applyLiveCapabilityState(pi: any, ctx: any): Promise<void> {
-  const [modeState, capabilityState] = await Promise.all([readModeState(ctx.cwd), readCapabilityState(ctx.cwd)]);
+async function applyLiveCapabilityState(
+  pi: any,
+  ctx: any,
+  cognitiveRoutingController: CognitiveRoutingController | undefined,
+  options: { reconcileCognitiveRouting?: boolean } = {},
+): Promise<CognitiveRoutingController | undefined> {
+  const [modeState, capabilityState] = await Promise.all([
+    readModeState(ctx.cwd),
+    readCapabilityState(ctx.cwd, ctx, pi?.host),
+  ]);
   await refreshRuntimeContext(capabilityState);
-  setModeStatus(ctx, modeState, capabilityState);
-  await applyCapabilityToolVisibility(pi, ctx, capabilityState);
+  let nextController = cognitiveRoutingController;
+  if (
+    options.reconcileCognitiveRouting &&
+    (nextController === undefined || capabilityState.cognitiveRouting?.effective !== true)
+  ) {
+    nextController = await reconcileCognitiveRoutingController(pi, ctx, capabilityState, nextController);
+  }
+  setModeStatus(ctx, modeState, capabilityState, nextController?.state());
+  await applyCapabilityToolVisibility(pi, ctx, capabilityState, nextController);
+  return nextController;
 }
 
 function disabledToolCall(toolName: string, capability: string) {
@@ -72,17 +268,7 @@ function disabledToolCall(toolName: string, capability: string) {
   };
 }
 
-function capabilityCompletions(prefix: string | undefined) {
-  const query = prefix ?? "";
-  return [
-    { value: "settings", label: "settings", description: "Open repository Output Router settings" },
-    { value: "status", label: "status", description: "Show effective Output Router state" },
-    { value: "enable", label: "enable", description: "Enable Output Router for this repository" },
-    { value: "disable", label: "disable", description: "Disable Output Router for this repository" },
-  ].filter((item) => item.value.startsWith(query));
-}
-
-function freeflowCompletions(prefix: string | undefined) {
+function freeflowCompletions(prefix: string | undefined, hostInfo = undefined) {
   const query = prefix ?? "";
   if (query.startsWith("settings ")) {
     const settingsQuery = query.slice("settings ".length);
@@ -94,6 +280,12 @@ function freeflowCompletions(prefix: string | undefined) {
       .filter((item) => item.value.startsWith(settingsQuery))
       .map((item) => ({ ...item, value: `settings ${item.value}` }));
   }
+  if (isPiFlowHost(hostInfo) && query.startsWith("profile ")) {
+    return cognitiveRoutingProfileCompletions(query.slice("profile ".length)).map((item) => ({
+      ...item,
+      value: `profile ${item.value}`,
+    }));
+  }
   if (query.startsWith("mode ")) {
     const modeQuery = query.slice("mode ".length);
     return [
@@ -109,10 +301,25 @@ function freeflowCompletions(prefix: string | undefined) {
       .filter((item) => item.value.startsWith(modeQuery))
       .map((item) => ({ ...item, value: `mode ${item.value}` }));
   }
+  if (query.startsWith("context ")) {
+    const contextQuery = query.slice("context ".length);
+    return [
+      { value: "status", label: "status", description: "Show Freeflow Context state" },
+      { value: "list", label: "list", description: "List archived context projections" },
+      { value: "restore", label: "restore", description: "Restore one or more context references" },
+      { value: "reset all", label: "reset all", description: "Reset projection decisions on the active branch" },
+    ]
+      .filter((item) => item.value.startsWith(contextQuery))
+      .map((item) => ({ ...item, value: `context ${item.value}` }));
+  }
   return [
     { value: "settings", label: "settings", description: "Open personal override settings" },
     { value: "status", label: "status", description: "Show effective Freeflow state" },
+    { value: "context", label: "context", description: "Inspect Freeflow Context" },
     { value: "mode", label: "mode", description: "Select a temporary session mode" },
+    ...(isPiFlowHost(hostInfo)
+      ? [{ value: "profile", label: "profile", description: "Hold or release Cognitive Routing profile control" }]
+      : []),
     { value: "enable", label: "enable", description: "Enable Freeflow for this repository" },
     { value: "disable", label: "disable", description: "Disable Freeflow for this repository" },
   ].filter((item) => item.value.startsWith(query));
@@ -127,7 +334,7 @@ function bypassCompletions(prefix: string | undefined) {
 }
 
 async function sendSkillCommand(pi: any, ctx: any, skill: string, args: string | undefined) {
-  const state = await readCapabilityState(ctx.cwd);
+  const state = await readCapabilityState(ctx.cwd, ctx, pi?.host);
   if (skill === "setup-freeflow" && !state.configured) {
     await pi.sendUserMessage(skillPrompt(skill, args));
     return;
@@ -151,102 +358,322 @@ async function sendSkillCommand(pi: any, ctx: any, skill: string, args: string |
 }
 
 export default function freeflow(pi) {
-  registerRouterTools(pi);
+  let cognitiveRoutingController: CognitiveRoutingController | undefined;
+  let latestCognitiveRoutingContext: any;
+  let providerSurfaceSnapshot: { context: any; value: any } | undefined;
+  let freeflowContextRuntime: FreeflowContextRuntime | undefined;
+  let contextVirtualizationRuntime: ContextVirtualizationRuntime | undefined;
+  let conversationHistoryRuntime: ConversationHistoryRuntime | undefined;
+  const registerContextToolForState = (capabilityState: any) => {
+    registerFreeflowContextTool(
+      pi,
+      () => contextVirtualizationRuntime,
+      () => conversationHistoryRuntime,
+      {
+        contextVirtualization: capabilityState?.contextVirtualization?.effective === true,
+        conversationHistory: capabilityState?.conversationHistory?.effective === true,
+      },
+    );
+  };
+  const buildProviderSurface = async (ctx: any, activateCognitiveRouting = false): Promise<any> => {
+    const [modeState, capabilityState] = await Promise.all([
+      readModeState(ctx.cwd),
+      readCapabilityState(ctx.cwd, ctx, pi?.host),
+    ]);
+    const freeflowContext = await getRuntimeContext(capabilityState);
+    const promptCapabilityState = modelFacingCapabilityState(
+      capabilityState,
+      ctx,
+      freeflowContext,
+      cognitiveRoutingController,
+    );
+    let activationAttempted = false;
+    if (
+      activateCognitiveRouting &&
+      (cognitiveRoutingController === undefined || promptCapabilityState.cognitiveRouting?.effective !== true)
+    ) {
+      activationAttempted =
+        promptCapabilityState.cognitiveRouting?.effective === true && !startupSelectionSuppressesCognitiveRouting(ctx);
+      try {
+        cognitiveRoutingController = await reconcileCognitiveRoutingController(
+          pi,
+          ctx,
+          promptCapabilityState,
+          cognitiveRoutingController,
+        );
+      } catch {
+        cognitiveRoutingController = undefined;
+      }
+    }
+    const surfaceCapabilityState = modelFacingCapabilityState(
+      promptCapabilityState,
+      ctx,
+      freeflowContext,
+      cognitiveRoutingController,
+      activationAttempted && !cognitiveRoutingController,
+    );
+    return {
+      modeState: modelFacingModeState(modeState, surfaceCapabilityState),
+      capabilityState: surfaceCapabilityState,
+      freeflowContext,
+      cognitiveRoutingRuntime: cognitiveRoutingController?.state(),
+      configuredCapabilityState: capabilityState,
+    };
+  };
+  const applyLiveCapabilityStateForSession = async (
+    ctx: any,
+    options: { reconcileCognitiveRouting?: boolean } = {},
+  ): Promise<void> => {
+    cognitiveRoutingController = await applyLiveCapabilityState(pi, ctx, cognitiveRoutingController, options);
+    registerContextToolForState(await readCapabilityState(ctx.cwd, ctx, pi?.host));
+  };
+
+  if (isPiFlowHost(pi?.host)) {
+    registerCognitiveRoutingTool(pi, () => cognitiveRoutingController);
+    registerCognitiveRoutingHistoryTool(pi, (options, context) => {
+      if (cognitiveRoutingController) return cognitiveRoutingController.history(options);
+      return readCognitiveRoutingHistory(context ?? latestCognitiveRoutingContext, options);
+    });
+  }
+  registerFreeflowContextTool(
+    pi,
+    () => contextVirtualizationRuntime,
+    () => conversationHistoryRuntime,
+    { contextVirtualization: false, conversationHistory: false },
+  );
+
+  if (isPiFlowHost(pi?.host) && typeof pi.registerShortcut === "function") {
+    pi.registerShortcut("ctrl+shift+r", {
+      description: "Cycle the Cognitive Routing manual standard/reasoning hold",
+      handler: async (ctx) => {
+        if (typeof ctx?.isIdle === "function" && !ctx.isIdle()) {
+          ctx.ui?.notify?.("Freeflow settings and profile changes are available only while Pi is idle.", "warning");
+          return;
+        }
+        if (!cognitiveRoutingController) {
+          await applyLiveCapabilityStateForSession(ctx, { reconcileCognitiveRouting: true });
+        }
+        const controller = cognitiveRoutingController;
+        if (!controller) {
+          ctx.ui?.notify?.("Cognitive Routing is unavailable for this session.", "warning");
+          return;
+        }
+        const current = controller.state().activeProfile;
+        const target = current === "reasoning" ? "standard" : "reasoning";
+        const result = await controller.setManualProfile(target, "profile-shortcut");
+        if (result.status === "active") {
+          ctx.ui?.notify?.(`Cognitive Routing manual hold set to ${target}.`, "info");
+        } else {
+          ctx.ui?.notify?.(`Cognitive Routing could not set the manual profile: ${result.reason}.`, "warning");
+        }
+        await applyLiveCapabilityStateForSession(ctx);
+      },
+    });
+    pi.registerShortcut("ctrl+shift+a", {
+      description: "Cycle the Cognitive Routing automatic standard/reasoning profile",
+      handler: async (ctx) => {
+        if (typeof ctx?.isIdle === "function" && !ctx.isIdle()) {
+          ctx.ui?.notify?.("Freeflow settings and profile changes are available only while Pi is idle.", "warning");
+          return;
+        }
+        if (!cognitiveRoutingController) {
+          await applyLiveCapabilityStateForSession(ctx, { reconcileCognitiveRouting: true });
+        }
+        const controller = cognitiveRoutingController;
+        if (!controller) {
+          ctx.ui?.notify?.("Cognitive Routing is unavailable for this session.", "warning");
+          return;
+        }
+        const state = controller.state();
+        const target = state.activeProfile === "reasoning" ? "standard" : "reasoning";
+        const result =
+          state.controlMode === "automatic" && state.activeProfile
+            ? await controller.switchAutomaticProfile(
+                target,
+                `Cycle automatic profile to ${target}.`,
+                "user",
+                "profile-shortcut",
+              )
+            : await controller.setAutomaticControl("profile-shortcut");
+        if (result.status === "automatic") {
+          ctx.ui?.notify?.("Cognitive Routing automatic control active.", "info");
+        } else if (result.status === "active") {
+          ctx.ui?.notify?.(`Cognitive Routing automatic profile set to ${result.profile}.`, "info");
+        } else {
+          ctx.ui?.notify?.(`Cognitive Routing could not cycle the automatic profile: ${result.reason}.`, "warning");
+        }
+        await applyLiveCapabilityStateForSession(ctx);
+      },
+    });
+  }
 
   pi.on("resources_discover", async (event, ctx) => {
     const cwd = ctx?.cwd ?? event?.cwd ?? process.cwd();
-    const state = await readCapabilityState(cwd);
+    const snapshot =
+      providerSurfaceSnapshot?.context === ctx
+        ? providerSurfaceSnapshot.value
+        : await buildProviderSurface(ctx ?? { cwd }, false);
+    const state = snapshot.capabilityState;
     if (!state.configured) {
       return { skillPaths: [freeflowSkillPath("setup-freeflow")] };
     }
     if (!state.enabled || !state.skills.effective) {
       return { skillPaths: [] };
     }
-    return { skillPaths: freeflowModelSkillPaths() };
+    return { skillPaths: freeflowModelSkillPaths(state) };
   });
 
   pi.on("session_start", async (_event, ctx) => {
+    latestCognitiveRoutingContext = ctx;
+    providerSurfaceSnapshot = undefined;
     restoreModeOverride(ctx);
-    const [modeState, routerConfigResult, capabilityState] = await Promise.all([
-      readModeState(ctx.cwd),
-      readOutputRouterConfig(ctx.cwd),
-      readCapabilityState(ctx.cwd),
-    ]);
+    freeflowContextRuntime = new FreeflowContextRuntime(ctx);
+    contextVirtualizationRuntime = new ContextVirtualizationRuntime(pi, ctx, freeflowContextRuntime);
+    conversationHistoryRuntime = new ConversationHistoryRuntime(
+      ctx,
+      (entryId) => contextVirtualizationRuntime?.isSourceFullyProjected(entryId) ?? true,
+      freeflowContextRuntime,
+    );
+    await contextVirtualizationRuntime.recover(ctx);
+    const capabilityState = await readCapabilityState(ctx.cwd, ctx, pi?.host);
     await refreshRuntimeContext(capabilityState);
-    setModeStatus(ctx, modeState, capabilityState);
-    notifyRouterConfigWarnings(ctx, routerConfigResult);
-    await applyCapabilityToolVisibility(pi, ctx, capabilityState);
+    const hasSessionState = sessionHasConversationOrRoutingState(ctx);
+    const snapshot = await buildProviderSurface(ctx, hasSessionState);
+    const cognitiveRoutingStartupPending =
+      snapshot.capabilityState?.cognitiveRouting?.effective === true &&
+      !hasSessionState &&
+      !startupSelectionSuppressesCognitiveRouting(ctx);
+    registerContextToolForState(snapshot.capabilityState);
+    setModeStatus(ctx, snapshot.modeState, snapshot.capabilityState, snapshot.cognitiveRoutingRuntime, {
+      cognitiveRoutingStartupPending,
+    });
+    await applyCapabilityToolVisibility(pi, ctx, snapshot.capabilityState, cognitiveRoutingController);
+  });
+
+  pi.on("agent_settled", async (_event, ctx) => {
+    if (!cognitiveRoutingController) return undefined;
+    await applyLiveCapabilityStateForSession(ctx);
+    return undefined;
+  });
+
+  pi.on("session_shutdown", async (event) => {
+    providerSurfaceSnapshot = undefined;
+    const controller = cognitiveRoutingController;
+    cognitiveRoutingController = undefined;
+    freeflowContextRuntime = undefined;
+    contextVirtualizationRuntime = undefined;
+    conversationHistoryRuntime = undefined;
+    if (controller) await controller.shutdown(event?.reason);
   });
 
   pi.on("session_tree", async (_event, ctx) => {
+    latestCognitiveRoutingContext = ctx;
+    providerSurfaceSnapshot = undefined;
     restoreModeOverride(ctx);
-    await applyLiveCapabilityState(pi, ctx);
+    const controller = cognitiveRoutingController;
+    if (controller) await controller.reconcileBranch();
+    if (contextVirtualizationRuntime) {
+      contextVirtualizationRuntime.setContext(ctx);
+      await contextVirtualizationRuntime.recover(ctx);
+    }
+    conversationHistoryRuntime?.setContext(ctx);
+    await applyLiveCapabilityStateForSession(ctx);
   });
 
   pi.on("session_compact", async (_event, ctx) => {
-    const [modeState, routerConfigResult, capabilityState] = await Promise.all([
+    latestCognitiveRoutingContext = ctx;
+    providerSurfaceSnapshot = undefined;
+    const controller = cognitiveRoutingController;
+    if (controller) await controller.reconcileBranch("session-compact");
+    if (contextVirtualizationRuntime) {
+      contextVirtualizationRuntime.setContext(ctx);
+      await contextVirtualizationRuntime.recover(ctx);
+    }
+    conversationHistoryRuntime?.setContext(ctx);
+    const [modeState, capabilityState] = await Promise.all([
       readModeState(ctx.cwd),
-      readOutputRouterConfig(ctx.cwd),
-      readCapabilityState(ctx.cwd),
+      readCapabilityState(ctx.cwd, ctx, pi?.host),
     ]);
     await refreshRuntimeContext(capabilityState);
-    setModeStatus(ctx, modeState, capabilityState);
-    notifyRouterConfigWarnings(ctx, routerConfigResult);
-    await applyCapabilityToolVisibility(pi, ctx, capabilityState);
+    registerContextToolForState(capabilityState);
+    setModeStatus(ctx, modeState, capabilityState, cognitiveRoutingController?.state());
+    await applyCapabilityToolVisibility(pi, ctx, capabilityState, cognitiveRoutingController);
   });
 
   pi.on("before_agent_start", async (event, ctx) => {
-    const [modeState, routerConfigResult, capabilityState] = await Promise.all([
-      readModeState(ctx.cwd),
-      readOutputRouterConfig(ctx.cwd),
-      readCapabilityState(ctx.cwd),
-    ]);
-    const freeflowContext = await getRuntimeContext(capabilityState);
-    setModeStatus(ctx, modeState, capabilityState);
-    notifyRouterConfigWarnings(ctx, routerConfigResult);
-    await applyCapabilityToolVisibility(pi, ctx, capabilityState);
-    const freeflowRuntimeContext = runtimeContext(modeState, freeflowContext, routerConfigResult, capabilityState);
-    const workflowMessage = workflowBootstrapMessage(freeflowContext, capabilityState, ctx.sessionManager);
+    latestCognitiveRoutingContext = ctx;
+    const snapshot = await buildProviderSurface(ctx, true);
+    providerSurfaceSnapshot = { context: ctx, value: snapshot };
+    registerContextToolForState(snapshot.capabilityState);
+    setModeStatus(ctx, snapshot.modeState, snapshot.capabilityState, snapshot.cognitiveRoutingRuntime);
+    await applyCapabilityToolVisibility(pi, ctx, snapshot.capabilityState, cognitiveRoutingController);
+    const freeflowRuntimeContext = runtimeContext(snapshot.freeflowContext, snapshot.capabilityState);
     const systemPrompt = freeflowRuntimeContext
       ? `${event.systemPrompt}\n\n${freeflowRuntimeContext}`
       : event.systemPrompt;
-    return { message: workflowMessage, systemPrompt };
+    return { systemPrompt };
   });
 
   pi.on("context", async (event, ctx) => {
-    const capabilityState = await readCapabilityState(ctx.cwd);
-    if (capabilityState.skills.effective) {
-      return undefined;
+    const snapshot =
+      providerSurfaceSnapshot?.context === ctx ? providerSurfaceSnapshot.value : await buildProviderSurface(ctx, false);
+    if (providerSurfaceSnapshot?.context === ctx) providerSurfaceSnapshot = undefined;
+    const cognitiveRoutingRuntime = snapshot.cognitiveRoutingRuntime;
+    const surfaceCapabilityState = snapshot.capabilityState;
+    const surfaceModeState = snapshot.modeState;
+    let changed = false;
+    let messages = event.messages
+      .map((message) => {
+        const filtered = filterBootstrapMessage(message);
+        if (filtered !== message) changed = true;
+        return filtered;
+      })
+      .filter((message) => message !== undefined);
+    if (contextVirtualizationRuntime) {
+      contextVirtualizationRuntime.setContext(ctx);
+      const projected = await contextVirtualizationRuntime.project(
+        messages,
+        surfaceCapabilityState.contextVirtualization?.effective === true,
+      );
+      if (projected.changed) {
+        changed = true;
+        messages = projected.messages;
+      }
     }
-
-    const messages = event.messages.filter(
-      (message) => !(message?.role === "custom" && message?.customType === WORKFLOW_BOOTSTRAP_MESSAGE_TYPE),
+    if (conversationHistoryRuntime && surfaceCapabilityState.conversationHistory?.effective === true) {
+      conversationHistoryRuntime.setContext(ctx);
+      conversationHistoryRuntime.capture(surfaceCapabilityState.contextVirtualization?.effective === true);
+    }
+    const nextMessages = withFreeflowRuntimeState(
+      messages,
+      surfaceModeState,
+      surfaceCapabilityState,
+      cognitiveRoutingRuntime,
     );
-    return messages.length === event.messages.length ? undefined : { messages };
+    changed = true;
+    messages = nextMessages;
+    return changed ? { messages } : undefined;
   });
 
   pi.on("tool_call", async (event, ctx) => {
-    const capabilityState = await readCapabilityState(ctx.cwd);
+    const capabilityState = await readCapabilityState(ctx.cwd, ctx, pi?.host);
     const toolName = typeof event?.toolName === "string" ? event.toolName : "";
-    if ((!capabilityState.configured || !capabilityState.enabled) && toolName === FREEFLOW_STATUS_TOOL_NAME) {
-      return disabledToolCall(toolName, "freeflow");
-    }
-    if (!capabilityState.outputRouter.enabled && isOutputRouterToolName(toolName)) {
-      return disabledToolCall(toolName, "output-router");
+    if (
+      capabilityState.contextVirtualization?.effective !== true &&
+      capabilityState.conversationHistory?.effective !== true &&
+      toolName === CONTEXT_VIRTUALIZATION_TOOL_NAME
+    ) {
+      return disabledToolCall(toolName, "context");
     }
     return undefined;
   });
 
   pi.on("tool_result", async (event, ctx) => {
-    const capabilityState = await readCapabilityState(ctx.cwd);
-    if (!capabilityState.outputRouter.enabled) {
-      return undefined;
-    }
-    const observed = await handleObservedToolRouting(event, ctx);
-    if (observed) {
-      return observed;
+    const toolName = typeof event?.toolName === "string" ? event.toolName : "";
+    if (isPiFlowHost(pi?.host) && toolName === COGNITIVE_ROUTING_SWITCH_TOOL_NAME) {
+      await applyLiveCapabilityStateForSession(ctx);
     }
-    return handleNativeToolSafetyNet(event, ctx);
+    return undefined;
   });
 
   for (const { command, skill } of WORKFLOW_COMMANDS) {
@@ -270,26 +697,51 @@ export default function freeflow(pi) {
 
   pi.registerCommand("freeflow", {
     description: "Open unified Freeflow settings or print compact status",
-    getArgumentCompletions: freeflowCompletions,
+    getArgumentCompletions: (prefix) => freeflowCompletions(prefix, pi?.host),
     handler: async (args, ctx) => {
+      const contextInput = (args ?? "").trim();
+      if (contextInput === "context" || contextInput.startsWith("context ")) {
+        const capabilityState = await readCapabilityState(ctx.cwd, ctx, pi?.host);
+        await handleContextCommand(
+          contextInput.slice("context".length).trim(),
+          ctx,
+          contextVirtualizationRuntime,
+          capabilityState.contextVirtualization?.effective === true,
+          capabilityState.conversationHistory?.effective === true,
+        );
+        return;
+      }
+      if (isPiFlowHost(pi?.host) && /^profile(?:\s|$)/i.test((args ?? "").trim())) {
+        const profileCommandContext = {
+          isIdle: ctx?.isIdle,
+          ui: ctx?.ui,
+          history: (options) =>
+            cognitiveRoutingController?.history(options) ?? readCognitiveRoutingHistory(ctx, options),
+        };
+        const historyCommand = /^profile\s+history(?:\s|$)/i.test((args ?? "").trim());
+        if (historyCommand) {
+          await handleCognitiveRoutingProfileCommand(args, profileCommandContext, cognitiveRoutingController);
+          return;
+        }
+        if (!cognitiveRoutingController) {
+          await applyLiveCapabilityStateForSession(ctx, { reconcileCognitiveRouting: true });
+        }
+        if (await handleCognitiveRoutingProfileCommand(args, profileCommandContext, cognitiveRoutingController)) {
+          await applyLiveCapabilityStateForSession(ctx);
+          return;
+        }
+      }
       await handleFreeflowCommand(
         args,
         ctx,
-        async () => {
-          await applyLiveCapabilityState(pi, ctx);
+        async (_changed, options = {}) => {
+          await applyLiveCapabilityStateForSession(ctx, {
+            reconcileCognitiveRouting: options.reconcileCognitiveRouting ?? true,
+          });
         },
         pi,
+        cognitiveRoutingController,
       );
     },
   });
-
-  pi.registerCommand("output-router", {
-    description: "Open Freeflow Output Router settings or print compact status",
-    getArgumentCompletions: capabilityCompletions,
-    handler: async (args, ctx) => {
-      await handleOutputRouterCommand(args, ctx, async () => {
-        await applyLiveCapabilityState(pi, ctx);
-      });
-    },
-  });
 }