@hassangameryt/freeflow 0.7.2 → 0.7.3

package/pi-extension/tests/context-virtualization/extension.test.js

@@ -169,7 +169,7 @@ function renderComponent(component) {
   return component.render(120).join("\\n");
 }
 
-test("disabled Freeflow Context does not publish an empty oneOf schema", async () => {
+test("disabled Freeflow Context retains valid full operation schemas", async () => {
   const cwd = await mkdtemp(join(tmpdir(), "freeflow-context-disabled-schema-"));
   try {
     await writeConfig(cwd, {});
@@ -178,8 +178,17 @@ test("disabled Freeflow Context does not publish an empty oneOf schema", async (
 
     const contextTool = harness.tools.find((tool) => tool.name === "freeflow_context");
     assert.ok(contextTool);
-    assert.deepEqual(contextTool.parameters.properties, {});
-    assert.equal(Object.hasOwn(contextTool.parameters, "oneOf"), false);
+    assert.equal(contextTool.parameters.type, "object");
+    assert.deepEqual(
+      contextTool.parameters.oneOf.map((v) => v.properties.operation.const),
+      ["archive", "restore", "search", "retrieve"],
+    );
+    const denied = await contextTool.execute(
+      "disabled",
+      { operation: "archive", targets: [{ ref: "ctx:tool-1" }] },
+      undefined,
+    );
+    assert.equal(denied.details.result.status, "rejected");
   } finally {
     await rm(cwd, { recursive: true, force: true });
   }
@@ -205,7 +214,7 @@ test("enabled Context Virtualization registers, projects, archives, restores, an
 
     const contextTool = harness.tools.find((tool) => tool.name === "freeflow_context");
     assert.ok(contextTool);
-    assert.doesNotMatch(contextTool.description, /conversation history/i);
+    assert.match(contextTool.description, /conversation history/i);
     const staleSearch = await contextTool.execute(
       "context-stale-search",
       { operation: "search", query: "not enabled" },
@@ -291,7 +300,7 @@ test("enabled Context Virtualization registers, projects, archives, restores, an
   }
 });
 
-test("enabled Conversation History exposes only search and retrieve operations", async () => {
+test("enabled Conversation History permits search and retrieve within the stable catalog", async () => {
   const cwd = await mkdtemp(join(tmpdir(), "freeflow-conversation-history-extension-"));
   try {
     await writeConfig(cwd, { conversationHistory: true });
@@ -303,7 +312,7 @@ test("enabled Conversation History exposes only search and retrieve operations",
     assert.ok(contextTool);
     assert.deepEqual(
       contextTool.parameters.oneOf.map((variant) => variant.properties.operation.const),
-      ["search", "retrieve"],
+      ["archive", "restore", "search", "retrieve"],
     );
     assert.equal(typeof contextTool.renderCall, "function");
     assert.equal(typeof contextTool.renderResult, "function");
@@ -365,7 +374,7 @@ test("enabled Conversation History exposes only search and retrieve operations",
     assert.match(renderComponent(expandedResult), /ctx:render-source/);
     assert.match(renderComponent(expandedResult), /database timeout was recovered/);
     assert.match(contextTool.description, /hidden conversation history/i);
-    assert.doesNotMatch(contextTool.description, /future context projections/i);
+    assert.match(contextTool.description, /future context projections/i);
     const before = await harness.handlers.get("before_agent_start")({ systemPrompt: "base" }, harness.ctx);
     assert.match(before.systemPrompt, /## Conversation History Cue/);
     assert.doesNotMatch(before.systemPrompt, /# Conversation History$/m);
@@ -744,20 +753,21 @@ test("visible invalid sources do not make an empty hidden corpus partial", async
   }
 });
 
-test("disabled Context Virtualization hides the tool and leaves tool results unchanged", async () => {
+test("disabled Context Virtualization blocks execution while preserving definitions and reference labels", async () => {
   const cwd = await mkdtemp(join(tmpdir(), "freeflow-context-disabled-"));
   try {
     await writeConfig(cwd, false);
     const harness = createHarness(cwd, [toolResultEntry()]);
     await harness.handlers.get("session_start")({ reason: "startup" }, harness.ctx);
 
-    assert.ok(!harness.activeToolNames().includes("freeflow_context"));
+    assert.ok(harness.activeToolNames().includes("freeflow_context"));
     const contextResult = await harness.handlers.get("context")(
       { messages: [harness.entries[0].message] },
       harness.ctx,
     );
     assert.ok(contextResult);
-    assert.deepEqual(contextResult.messages[0], harness.entries[0].message);
+    assert.deepEqual(contextResult.messages[0].content.slice(0, -1), harness.entries[0].message.content);
+    assert.match(contextResult.messages[0].content.at(-1).text, /context-ref:/);
     assert.equal(contextResult.messages.at(-1).customType, "freeflow-runtime-state");
   } finally {
     await rm(cwd, { recursive: true, force: true });

package/pi-extension/tests/context-virtualization/runtime.test.js

@@ -208,9 +208,23 @@ test("mutation persistence failure returns an error without publishing projectio
   assert.match(next.messages[0].content.at(-1).text, /ctx:tool-1/);
 });
 
-test("disabled projection bypasses stored state and adds no marker", async () => {
+test("disabled projection keeps original content and the stable reference marker", async () => {
   const { runtime } = createFixture();
   const result = await runtime.project([toolMessage()], false);
-  assert.equal(result.changed, false);
-  assert.deepEqual(result.messages[0].content, [{ type: "text", text: "secret raw output" }]);
+  assert.equal(result.changed, true);
+  assert.deepEqual(result.messages[0].content.slice(0, -1), [{ type: "text", text: "secret raw output" }]);
+  assert.match(result.messages[0].content.at(-1).text, /context-ref: ctx:tool-1/);
+});
+
+test("unavailable decoration leaves the already projected view intact", () => {
+  const { runtime } = createFixture();
+  runtime.setContext({
+    sessionManager: {
+      getBranch() {
+        throw Error("unavailable");
+      },
+    },
+  });
+  const current = [{ role: "custom", customType: "blocked", content: "Only this safe view" }];
+  assert.equal(runtime.decorate(current), current);
 });

package/pi-extension/tests/fixtures/routing-native.js

@@ -69,7 +69,7 @@ export async function fixture(script, projection = true, after, withUI = true, o
   await writeFile(
     join(cwd, ".freeflow/config.json"),
     JSON.stringify({
-      cognitiveRouting: {
+      cognitiveRouting: options.cognitiveRouting ?? {
         enabled: true,
         projection,
         profiles: {
@@ -81,6 +81,8 @@ export async function fixture(script, projection = true, after, withUI = true, o
   );
   await writeFile(join(cwd, "evidence.txt"), "EXACT_EVIDENCE_BODY_81");
   await writeFile(join(cwd, "unselected.txt"), "UNSELECTED_PRIVATE_BODY_93");
+  await writeFile(join(cwd, "recovery.txt"), "FRESH_RECOVERY_BODY_57");
+  await writeFile(join(cwd, "@recovery.txt"), "LITERAL_AT_FILE_BODY_68");
   await writeFile(
     join(agentDir, "models.json"),
     JSON.stringify({ providers: { openai: { baseUrl: "https://fixture.invalid/v1" } } }),
@@ -166,6 +168,7 @@ export async function fixture(script, projection = true, after, withUI = true, o
       ...(withUI ? { uiContext: { notify: (...args) => notices.push(args), setStatus: () => {} } } : {}),
     });
     await options.onSession?.(session);
+    await options.beforePrompt?.({ session, manager, requests, contexts, cwd, notices });
     await session.prompt("Complete the fixture assignment.");
     await session.waitForIdle();
     assert.deepEqual(errors, [], "extension lifecycle errors");

package/pi-extension/tests/integration/mode-free-core.test.js

@@ -138,7 +138,7 @@ test("Pi exposes base skills without mode or Skills controls", async () => {
     const { handlers, commands } = loadExtension();
     const ctx = context(cwd);
     const resources = await handlers.get("resources_discover")({ cwd }, ctx);
-    assert.equal(resources.skillPaths.length, 24);
+    assert.equal(resources.skillPaths.length, 27);
     assert.ok(resources.skillPaths.some((path) => path.endsWith("/skills/workflow/SKILL.md")));
     assert.ok(!resources.skillPaths.some((path) => path.endsWith("/skills/tdd/SKILL.md")));
     assert.ok(!resources.skillPaths.some((path) => path.endsWith("/skills/mode-contract/SKILL.md")));
@@ -181,9 +181,10 @@ test("mandatory prompt readiness gates Runtime State, discovery, and direct skil
     const { handlers, commands, sentMessages } = loadExtension(extension);
     const ctx = context(cwd);
     const before = await handlers.get("before_agent_start")({ systemPrompt: "base prompt" }, ctx);
-    assert.equal(before.systemPrompt, "base prompt");
+    assert.ok(before.systemPrompt.startsWith("base prompt\n\n"));
+    assert.match(before.systemPrompt, /guidance is dormant/);
     const resources = await handlers.get("resources_discover")({ cwd }, ctx);
-    assert.deepEqual(resources.skillPaths, []);
+    assert.equal(resources.skillPaths.length, 27);
 
     const discuss = commands.find((command) => command.name === "discuss");
     assert.ok(discuss);