@axiom-lattice/protocols 4.3.1 → 4.4.0

package/src/__tests__/TrustedRunContextProtocol.test.ts

@@ -82,7 +82,7 @@ describe("parseTrustedRunContext", () => {
     expect(calls).toBe(0);
   });
 
-  it.each(["tenantId", "responsibility"])(
+  it.each(["tenantId", "responsibility", "source"])(
     "rejects a nested %s getter without invoking it",
     (field) => {
       let calls = 0;
@@ -146,7 +146,79 @@ describe("parseTrustedRunContext", () => {
       .toThrow("Invalid trusted agent run context");
   });
 
-  it.each(Object.keys(validProjectTask()))("rejects an empty project task %s", (field) => {
+  it("round-trips an explicit project task provenance source", () => {
+    const parsed = parseTrustedRunContext({
+      projectTask: { ...validProjectTask(), source: "project_task" },
+    });
+
+    expect(parsed.projectTask?.source).toBe("project_task");
+  });
+
+  it("leaves project task provenance source undefined when absent", () => {
+    const parsed = parseTrustedRunContext({ projectTask: validProjectTask() });
+
+    expect(parsed.projectTask?.source).toBeUndefined();
+    expect(Object.prototype.hasOwnProperty.call(parsed.projectTask, "source")).toBe(false);
+  });
+
+  it("treats an explicit project task provenance source of undefined as absent", () => {
+    const parsed = parseTrustedRunContext({
+      projectTask: { ...validProjectTask(), source: undefined },
+    });
+
+    expect(parsed.projectTask?.source).toBeUndefined();
+    expect(Object.prototype.hasOwnProperty.call(parsed.projectTask, "source")).toBe(false);
+  });
+
+  it("rejects a non-string project task provenance source", () => {
+    expect(() => parseTrustedRunContext({
+      projectTask: { ...validProjectTask(), source: 42 },
+    })).toThrow("Invalid trusted agent run context");
+  });
+
+  it("rejects an empty project task provenance source", () => {
+    expect(() => parseTrustedRunContext({
+      projectTask: { ...validProjectTask(), source: "" },
+    })).toThrow("Invalid trusted agent run context");
+  });
+
+  it("round-trips an explicit project room provenance source", () => {
+    const parsed = parseTrustedRunContext({
+      projectRoom: { ...validProjectRoom(), source: "room_message" },
+    });
+
+    expect(parsed.projectRoom?.source).toBe("room_message");
+  });
+
+  it("leaves project room provenance source undefined when absent", () => {
+    const parsed = parseTrustedRunContext({ projectRoom: validProjectRoom() });
+
+    expect(parsed.projectRoom?.source).toBeUndefined();
+    expect(Object.prototype.hasOwnProperty.call(parsed.projectRoom, "source")).toBe(false);
+  });
+
+  it("treats an explicit project room provenance source of undefined as absent", () => {
+    const parsed = parseTrustedRunContext({
+      projectRoom: { ...validProjectRoom(), source: undefined },
+    });
+
+    expect(parsed.projectRoom?.source).toBeUndefined();
+    expect(Object.prototype.hasOwnProperty.call(parsed.projectRoom, "source")).toBe(false);
+  });
+
+  it("rejects a non-string project room provenance source", () => {
+    expect(() => parseTrustedRunContext({
+      projectRoom: { ...validProjectRoom(), source: 42 },
+    })).toThrow("Invalid trusted agent run context");
+  });
+
+  it("rejects an empty project room provenance source", () => {
+    expect(() => parseTrustedRunContext({
+      projectRoom: { ...validProjectRoom(), source: "" },
+    })).toThrow("Invalid trusted agent run context");
+  });
+
+  it.each([...Object.keys(validProjectTask()), "source"])("rejects an empty project task %s", (field) => {
     expect(() => parseTrustedRunContext({
       projectTask: { ...validProjectTask(), [field]: "" },
     })).toThrow("Invalid trusted agent run context");
@@ -193,7 +265,7 @@ describe("parseTrustedRunContext", () => {
     expect(calls).toBe(0);
   });
 
-  it.each(Object.keys(validProjectTask()))(
+  it.each([...Object.keys(validProjectTask()), "source"])(
     "rejects a nested project task %s getter without invoking it",
     (field) => {
       let calls = 0;

package/src/__tests__/open-ui-meta.test.ts

@@ -0,0 +1,23 @@
+import type { PluginMeta, PluginUiResource } from "../PluginProtocol";
+
+describe("plugin inline uiResources contract", () => {
+  it("round-trips inline UI html and csp on a PluginMeta object", () => {
+    const resource: PluginUiResource = {
+      html: "<!doctype html><html><body>hello</body></html>",
+      csp: { connectDomains: ["https://api.example.com"] },
+    };
+
+    const meta: PluginMeta = {
+      type: "demo",
+      name: "Demo",
+      description: "demo plugin",
+      uiResources: { "ui://demo/app": resource },
+    };
+
+    expect(meta.uiResources?.["ui://demo/app"].html).toContain("hello");
+    expect(meta.uiResources?.["ui://demo/app"].csp?.connectDomains).toEqual([
+      "https://api.example.com",
+    ]);
+    expect(resource.mimeType).toBeUndefined();
+  });
+});

package/src/index.ts

@@ -80,6 +80,8 @@ export type {
   PluginDiscoveredResource,
   PluginContext,
   PluginToolMeta,
+  PluginToolUi,
+  PluginUiResource,
   PluginSkillResource,
   PluginSkillDefinition,
   PluginStandardConnectionConfig,
@@ -92,3 +94,4 @@ export * from "./SandboxPluginProtocol";
 export * from "./types";
 export * from "./TrustedRunContextProtocol";
 export * from "./OpenProtocol";
+export * from "./McpUiProtocol";