@agent-native/creative-context 0.7.25 → 0.8.0-nightly-20260915020720

package/src/server/native-context-clone.ts

@@ -0,0 +1,9 @@
+import { resolveNativeContextCloneReference as resolveFromStore } from "../store/contexts.js";
+import { assertCreativeContextLabEnabled } from "./labs.js";
+
+export async function resolveNativeContextCloneReference(
+  input: Parameters<typeof resolveFromStore>[0],
+) {
+  await assertCreativeContextLabEnabled();
+  return resolveFromStore(input);
+}

package/src/server/prompt-provider.spec.ts

@@ -3,7 +3,10 @@ import { beforeEach, describe, expect, it, vi } from "vitest";
 const mocks = vi.hoisted(() => ({
   readAppState: vi.fn(),
   getBrandProfile: vi.fn(),
+  isCreativeContextLabAvailable: vi.fn(),
+  getCreativeContext: vi.fn(),
   provider: null as null | {
+    failOnError?: boolean;
     load: (context: {
       owner: string;
       compact: boolean;
@@ -29,6 +32,14 @@ vi.mock("../store/brand.js", () => ({
   getBrandProfile: mocks.getBrandProfile,
 }));
 
+vi.mock("./labs.js", () => ({
+  isCreativeContextLabAvailable: mocks.isCreativeContextLabAvailable,
+}));
+
+vi.mock("./context.js", () => ({
+  getCreativeContext: mocks.getCreativeContext,
+}));
+
 import {
   publishedBrandContextInput,
   registerCreativeContextPromptProvider,
@@ -38,6 +49,11 @@ describe("creative context prompt provider", () => {
   beforeEach(() => {
     mocks.readAppState.mockReset();
     mocks.getBrandProfile.mockReset();
+    mocks.isCreativeContextLabAvailable.mockReset();
+    mocks.getCreativeContext.mockReturnValue({
+      labKey: "creative-context.library",
+    });
+    mocks.isCreativeContextLabAvailable.mockResolvedValue(true);
   });
 
   it("maps only structured published payload fields into the compiler", () => {
@@ -75,4 +91,43 @@ describe("creative context prompt provider", () => {
     expect(mocks.getBrandProfile).not.toHaveBeenCalled();
     unregister();
   });
+
+  it("omits published brand context while its Lab is disabled", async () => {
+    mocks.isCreativeContextLabAvailable.mockResolvedValue(false);
+    const unregister = registerCreativeContextPromptProvider();
+
+    await expect(
+      mocks.provider?.load({
+        owner: "user@example.test",
+        compact: false,
+        orgId: null,
+      }),
+    ).resolves.toBeNull();
+    expect(mocks.isCreativeContextLabAvailable).toHaveBeenCalledWith(
+      "user@example.test",
+      "creative-context.library",
+    );
+    expect(mocks.readAppState).not.toHaveBeenCalled();
+    expect(mocks.getBrandProfile).not.toHaveBeenCalled();
+    unregister();
+  });
+
+  it("surfaces an unreadable Lab state instead of loading brand context", async () => {
+    mocks.isCreativeContextLabAvailable.mockRejectedValue(
+      new Error("settings unavailable"),
+    );
+    const unregister = registerCreativeContextPromptProvider();
+    expect(mocks.provider?.failOnError).toBe(true);
+
+    await expect(
+      mocks.provider?.load({
+        owner: "user@example.test",
+        compact: false,
+        orgId: null,
+      }),
+    ).rejects.toThrow("settings unavailable");
+    expect(mocks.readAppState).not.toHaveBeenCalled();
+    expect(mocks.getBrandProfile).not.toHaveBeenCalled();
+    unregister();
+  });
 });

package/src/server/prompt-provider.ts

@@ -7,6 +7,8 @@ import {
   compilePublishedBrandContext,
   type PublishedBrandContextInput,
 } from "./brand-context.js";
+import { getCreativeContext } from "./context.js";
+import { isCreativeContextLabAvailable } from "./labs.js";
 
 const PROVIDER_ID = "creative-context";
 let unregisterProvider: (() => void) | null = null;
@@ -46,8 +48,17 @@ export function registerCreativeContextPromptProvider(): () => void {
   if (unregisterProvider) return unregisterProvider;
   const unregister = registerPromptContextProvider({
     id: PROVIDER_ID,
-    async load() {
-      const state = await readAppState("creative-context").catch(() => null);
+    failOnError: true,
+    async load(context) {
+      if (
+        !(await isCreativeContextLabAvailable(
+          context.owner,
+          getCreativeContext().labKey,
+        ))
+      ) {
+        return null;
+      }
+      const state = await readAppState("creative-context");
       if (state?.contextMode === "off") return null;
       const { profile, dna } = await getBrandProfile({});
       if (!profile || !dna || dna.status !== "published") return null;