skybridge 0.6.0 → 0.8.0

package/dist/src/web/hooks/use-widget-state.js

@@ -1,10 +1,33 @@
 import { useCallback, useEffect, useState } from "react";
 import { useOpenAiGlobal } from "./use-openai-global.js";
+const WIDGET_CONTEXT_KEY = "__widget_context";
+function filterWidgetContext(state) {
+    if (state === null) {
+        return null;
+    }
+    const { [WIDGET_CONTEXT_KEY]: _, ...filteredState } = state;
+    return filteredState;
+}
+function preserveWidgetContext(newState) {
+    if (newState === null) {
+        return null;
+    }
+    const currentWindowState = window.openai?.widgetState;
+    if (currentWindowState !== null &&
+        currentWindowState !== undefined &&
+        WIDGET_CONTEXT_KEY in currentWindowState) {
+        return {
+            ...newState,
+            [WIDGET_CONTEXT_KEY]: currentWindowState[WIDGET_CONTEXT_KEY],
+        };
+    }
+    return newState;
+}
 export function useWidgetState(defaultState) {
     const widgetStateFromWindow = useOpenAiGlobal("widgetState");
     const [widgetState, _setWidgetState] = useState(() => {
         if (widgetStateFromWindow !== null) {
-            return widgetStateFromWindow;
+            return filterWidgetContext(widgetStateFromWindow);
         }
         return typeof defaultState === "function"
             ? defaultState()
@@ -13,18 +36,19 @@ export function useWidgetState(defaultState) {
     useEffect(() => {
         // Fixes openai implementation bug
         if (widgetStateFromWindow !== null) {
-            _setWidgetState(widgetStateFromWindow);
+            _setWidgetState(filterWidgetContext(widgetStateFromWindow));
         }
     }, [widgetStateFromWindow]);
     const setWidgetState = useCallback((state) => {
         _setWidgetState((prevState) => {
             const newState = typeof state === "function" ? state(prevState) : state;
-            if (newState !== null) {
-                window.openai.setWidgetState(newState);
+            const stateToSet = preserveWidgetContext(newState);
+            if (stateToSet !== null) {
+                window.openai.setWidgetState(stateToSet);
             }
-            return newState;
+            return filterWidgetContext(stateToSet);
         });
-    }, [window.openai.setWidgetState]);
+    }, []);
     return [widgetState, setWidgetState];
 }
 //# sourceMappingURL=use-widget-state.js.map

package/dist/src/web/hooks/use-widget-state.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-widget-state.js","sourceRoot":"","sources":["../../../../src/web/hooks/use-widget-state.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,EAAuB,MAAM,OAAO,CAAC;AAE9E,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAQzD,MAAM,UAAU,cAAc,CAC5B,YAA0C;IAE1C,MAAM,qBAAqB,GAAG,eAAe,CAAC,aAAa,CAAM,CAAC;IAElE,MAAM,CAAC,WAAW,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAW,GAAG,EAAE;QAC7D,IAAI,qBAAqB,KAAK,IAAI,EAAE,CAAC;YACnC,OAAO,qBAAqB,CAAC;QAC/B,CAAC;QAED,OAAO,OAAO,YAAY,KAAK,UAAU;YACvC,CAAC,CAAC,YAAY,EAAE;YAChB,CAAC,CAAC,YAAY,IAAI,IAAI,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,kCAAkC;QAClC,IAAI,qBAAqB,KAAK,IAAI,EAAE,CAAC;YACnC,eAAe,CAAC,qBAAqB,CAAC,CAAC;QACzC,CAAC;IACH,CAAC,EAAE,CAAC,qBAAqB,CAAC,CAAC,CAAC;IAE5B,MAAM,cAAc,GAAG,WAAW,CAChC,CAAC,KAA+B,EAAE,EAAE;QAClC,eAAe,CAAC,CAAC,SAAS,EAAE,EAAE;YAC5B,MAAM,QAAQ,GAAG,OAAO,KAAK,KAAK,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;YAExE,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;gBACtB,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;YACzC,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC,CAAC,CAAC;IACL,CAAC,EACD,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAC/B,CAAC;IAEF,OAAO,CAAC,WAAW,EAAE,cAAc,CAAU,CAAC;AAChD,CAAC"}
+{"version":3,"file":"use-widget-state.js","sourceRoot":"","sources":["../../../../src/web/hooks/use-widget-state.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,EAAuB,MAAM,OAAO,CAAC;AAE9E,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAEzD,MAAM,kBAAkB,GAAG,kBAA2B,CAAC;AAEvD,SAAS,mBAAmB,CAC1B,KAAgB;IAEhB,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACnB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,EAAE,CAAC,kBAAkB,CAAC,EAAE,CAAC,EAAE,GAAG,aAAa,EAAE,GAAG,KAErD,CAAC;IAEF,OAAO,aAAkB,CAAC;AAC5B,CAAC;AAED,SAAS,qBAAqB,CAC5B,QAAkB;IAElB,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,kBAAkB,GAAG,MAAM,CAAC,MAAM,EAAE,WAG7B,CAAC;IAEd,IACE,kBAAkB,KAAK,IAAI;QAC3B,kBAAkB,KAAK,SAAS;QAChC,kBAAkB,IAAI,kBAAkB,EACxC,CAAC;QACD,OAAO;YACL,GAAG,QAAQ;YACX,CAAC,kBAAkB,CAAC,EAAE,kBAAkB,CAAC,kBAAkB,CAAC;SACxD,CAAC;IACT,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAQD,MAAM,UAAU,cAAc,CAC5B,YAA0C;IAE1C,MAAM,qBAAqB,GAAG,eAAe,CAAC,aAAa,CAAa,CAAC;IAEzE,MAAM,CAAC,WAAW,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAW,GAAG,EAAE;QAC7D,IAAI,qBAAqB,KAAK,IAAI,EAAE,CAAC;YACnC,OAAO,mBAAmB,CAAC,qBAAqB,CAAC,CAAC;QACpD,CAAC;QAED,OAAO,OAAO,YAAY,KAAK,UAAU;YACvC,CAAC,CAAC,YAAY,EAAE;YAChB,CAAC,CAAC,YAAY,IAAI,IAAI,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,kCAAkC;QAClC,IAAI,qBAAqB,KAAK,IAAI,EAAE,CAAC;YACnC,eAAe,CAAC,mBAAmB,CAAC,qBAAqB,CAAC,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC,EAAE,CAAC,qBAAqB,CAAC,CAAC,CAAC;IAE5B,MAAM,cAAc,GAAG,WAAW,CAAC,CAAC,KAA+B,EAAE,EAAE;QACrE,eAAe,CAAC,CAAC,SAAS,EAAE,EAAE;YAC5B,MAAM,QAAQ,GAAG,OAAO,KAAK,KAAK,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;YACxE,MAAM,UAAU,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YAEnD,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;gBACxB,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;YAC3C,CAAC;YAED,OAAO,mBAAmB,CAAC,UAAU,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,CAAC,WAAW,EAAE,cAAc,CAAU,CAAC;AAChD,CAAC"}

package/dist/src/web/index.d.ts

@@ -1,4 +1,6 @@
 export * from "./types.js";
+export * from "./data-llm.js";
 export { mountWidget } from "./mount-widget.js";
-export { skybridge } from "./plugin.js";
+export { skybridge } from "./plugin/plugin.js";
 export * from "./hooks/index.js";
+export { generateHelpers } from "./generate-helpers.js";

package/dist/src/web/index.js

@@ -1,5 +1,7 @@
 export * from "./types.js";
+export * from "./data-llm.js";
 export { mountWidget } from "./mount-widget.js";
-export { skybridge } from "./plugin.js";
+export { skybridge } from "./plugin/plugin.js";
 export * from "./hooks/index.js";
+export { generateHelpers } from "./generate-helpers.js";
 //# sourceMappingURL=index.js.map

package/dist/src/web/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/web/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAC3B,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,cAAc,kBAAkB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/web/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAC3B,cAAc,eAAe,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,cAAc,kBAAkB,CAAC;AACjC,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC"}

package/dist/src/web/plugin/data-llm.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/src/web/plugin/data-llm.test.js

@@ -0,0 +1,81 @@
+import { describe, expect, it } from "vitest";
+import { transform } from "./transform-data-llm.js";
+describe("data-llm plugin", () => {
+    it("should transform JSX element with data-llm string attribute", async () => {
+        const code = `
+      function Component() {
+        return <div data-llm="Test description">Content</div>;
+      }
+    `;
+        const result = await transform(code, "test.tsx");
+        expect(result).not.toBeNull();
+        expect(result?.code).toContain("DataLLM");
+        expect(result?.code).toContain('content="Test description"');
+        expect(result?.code).not.toContain("data-llm");
+    });
+    it("should transform JSX element with data-llm expression attribute", async () => {
+        const code = `
+      function Component() {
+        const desc = "Dynamic description";
+        return <div data-llm={desc}>Content</div>;
+      }
+    `;
+        const result = await transform(code, "test.tsx");
+        expect(result).not.toBeNull();
+        expect(result?.code).toContain("DataLLM");
+        expect(result?.code).toContain("content={desc}");
+        expect(result?.code).not.toContain("data-llm");
+    });
+    it("should add import for DataLLM when not present", async () => {
+        const code = `
+      function Component() {
+        return <div data-llm="Test">Content</div>;
+      }
+    `;
+        const result = await transform(code, "test.tsx");
+        expect(result).not.toBeNull();
+        expect(result?.code).toContain('import { DataLLM } from "skybridge/web"');
+    });
+    it("should handle DataLLM imports correctly", async () => {
+        // No duplicate import
+        const codeWithImport = `
+      import { DataLLM } from "skybridge/web";
+      function Component() {
+        return <div data-llm="Test">Content</div>;
+      }
+    `;
+        const result1 = await transform(codeWithImport, "test.tsx");
+        expect(result1?.code.match(/import.*DataLLM.*from.*skybridge\/web/g)).toHaveLength(1);
+        // Preserve other imports and add missing DataLLM
+        const codeWithOthers = `
+      import React from "react";
+      import { useState } from "react";
+      function Component() {
+        return <div data-llm="Test">Content</div>;
+      }
+    `;
+        const result2 = await transform(codeWithOthers, "test.tsx");
+        expect(result2?.code).toContain('import React from "react"');
+        expect(result2?.code).toContain('import { useState } from "react"');
+        expect(result2?.code).toContain('import { DataLLM } from "skybridge/web"');
+    });
+    it("should handle complex JSX with multiple data-llm attributes", async () => {
+        const code = `
+      function Component() {
+        return (
+          <div>
+            <section data-llm="Section 1">
+              <p>Content 1</p>
+            </section>
+            <section data-llm="Section 2">
+              <p>Content 2</p>
+            </section>
+          </div>
+        );
+      }
+    `;
+        const result = await transform(code, "test.tsx");
+        expect(result).toMatchSnapshot();
+    });
+});
+//# sourceMappingURL=data-llm.test.js.map

package/dist/src/web/plugin/data-llm.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"data-llm.test.js","sourceRoot":"","sources":["../../../../src/web/plugin/data-llm.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AAEpD,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAEjD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC9B,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAC;QAC7D,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,MAAM,IAAI,GAAG;;;;;KAKZ,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAEjD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC9B,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAEjD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC9B,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,yCAAyC,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,sBAAsB;QACtB,MAAM,cAAc,GAAG;;;;;KAKtB,CAAC;QACF,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;QAC5D,MAAM,CACJ,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAC9D,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAElB,iDAAiD;QACjD,MAAM,cAAc,GAAG;;;;;;KAMtB,CAAC;QACF,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;QAC5D,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAC;QAC7D,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,kCAAkC,CAAC,CAAC;QACpE,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,yCAAyC,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,IAAI,GAAG;;;;;;;;;;;;;KAaZ,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAEjD,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/web/{plugin.js → plugin/plugin.js}

@@ -1,3 +1,4 @@
+import { transform as dataLlmTransform } from "./transform-data-llm.js";
 export function skybridge() {
     return {
         name: "skybridge",
@@ -23,6 +24,10 @@ export function skybridge() {
                 },
             };
         },
+        enforce: "pre",
+        async transform(code, id) {
+            return await dataLlmTransform(code, id);
+        },
     };
 }
 //# sourceMappingURL=plugin.js.map

package/dist/src/web/plugin/plugin.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"plugin.js","sourceRoot":"","sources":["../../../../src/web/plugin/plugin.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,IAAI,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAExE,MAAM,UAAU,SAAS;IACvB,OAAO;QACL,IAAI,EAAE,WAAW;QAEjB,KAAK,CAAC,MAAM,CAAC,MAAM;YACjB,4EAA4E;YAC5E,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;YAC7C,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;YAE9C,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;YACjD,MAAM,cAAc,GAAG,OAAO,CAC5B,WAAW,EACX,oCAAoC,CACrC,CAAC;YAEF,MAAM,KAAK,GAAG,MAAM,CAAC,WAAW,CAC9B,QAAQ,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;gBACrC,IAAI,CAAC,KAAK,CAAC,0BAA0B,CAAC,EAAE,CAAC,CAAC,CAAC;gBAC3C,IAAI;aACL,CAAC,CACH,CAAC;YAEF,OAAO;gBACL,IAAI,EAAE,SAAS;gBACf,KAAK,EAAE;oBACL,QAAQ,EAAE,IAAI;oBACd,MAAM,EAAE,IAAI;oBACZ,YAAY,EAAE,KAAK;oBACnB,aAAa,EAAE;wBACb,KAAK;qBACN;iBACF;aACF,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,KAAK;QACd,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE;YACtB,OAAO,MAAM,gBAAgB,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC1C,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/src/web/plugin/transform-data-llm.d.ts

@@ -0,0 +1,12 @@
+export declare const transform: (code: string, id: string) => Promise<{
+    code: string;
+    map: {
+        version: number;
+        sources: string[];
+        names: string[];
+        sourceRoot?: string | undefined;
+        sourcesContent?: string[] | undefined;
+        mappings: string;
+        file: string;
+    } | null;
+} | null>;

package/dist/src/web/plugin/transform-data-llm.js

@@ -0,0 +1,93 @@
+const LLM_IMPORT_SOURCE = "skybridge/web";
+function createBabelPlugin(t) {
+    return {
+        name: "data-llm-babel",
+        visitor: {
+            Program: {
+                enter(path, state) {
+                    state.hasDataLLMImport = false;
+                    state.needsDataLLMImport = false;
+                    for (const node of path.node.body) {
+                        if (!t.isImportDeclaration(node))
+                            continue;
+                        if (node.source.value !== LLM_IMPORT_SOURCE)
+                            continue;
+                        const hasSpecifier = node.specifiers.some((s) => t.isImportSpecifier(s) &&
+                            t.isIdentifier(s.imported, { name: "DataLLM" }));
+                        if (hasSpecifier) {
+                            state.hasDataLLMImport = true;
+                            break;
+                        }
+                    }
+                },
+                exit(path, state) {
+                    if (state.needsDataLLMImport && !state.hasDataLLMImport) {
+                        const importDecl = t.importDeclaration([
+                            t.importSpecifier(t.identifier("DataLLM"), t.identifier("DataLLM")),
+                        ], t.stringLiteral(LLM_IMPORT_SOURCE));
+                        path.node.body.unshift(importDecl);
+                    }
+                },
+            },
+            JSXElement(path, state) {
+                const opening = path.node.openingElement;
+                const attributes = opening.attributes;
+                const llmAttributeIndex = attributes.findIndex((attribute) => t.isJSXAttribute(attribute) &&
+                    t.isJSXIdentifier(attribute.name, { name: "data-llm" }));
+                if (llmAttributeIndex === -1)
+                    return;
+                const llmAttribute = attributes[llmAttributeIndex];
+                let contentExpression;
+                if (t.isStringLiteral(llmAttribute.value)) {
+                    contentExpression = llmAttribute.value;
+                }
+                else if (t.isJSXExpressionContainer(llmAttribute.value)) {
+                    contentExpression = llmAttribute.value.expression;
+                }
+                else {
+                    return;
+                }
+                const contentAttr = t.jsxAttribute(t.jsxIdentifier("content"), t.isStringLiteral(contentExpression)
+                    ? contentExpression
+                    : t.jsxExpressionContainer(contentExpression));
+                const filteredAttributes = attributes.filter((_, index) => index !== llmAttributeIndex);
+                const newOpening = t.jsxOpeningElement(opening.name, filteredAttributes, opening.selfClosing);
+                const elementWithoutLlm = t.jsxElement(newOpening, path.node.closingElement, path.node.children, path.node.selfClosing);
+                const llmOpening = t.jsxOpeningElement(t.jsxIdentifier("DataLLM"), [
+                    contentAttr,
+                ]);
+                const llmClosing = t.jsxClosingElement(t.jsxIdentifier("DataLLM"));
+                const wrapped = t.jsxElement(llmOpening, llmClosing, [elementWithoutLlm], false);
+                state.needsDataLLMImport = true;
+                path.replaceWith(wrapped);
+            },
+        },
+    };
+}
+export const transform = async (code, id) => {
+    if (!/\.(jsx|tsx)$/.test(id)) {
+        return null;
+    }
+    if (id.includes("node_modules")) {
+        return null;
+    }
+    // Dynamic import to ensure @babel/core is only loaded in Node.js context
+    const { types: t, transformSync } = await import("@babel/core");
+    const babelOptions = {
+        plugins: [createBabelPlugin(t)],
+        parserOpts: {
+            plugins: ["jsx", "typescript"],
+        },
+        filename: id,
+        sourceFileName: id,
+    };
+    const result = transformSync(code, babelOptions);
+    if (!result || !result.code) {
+        return null;
+    }
+    return {
+        code: result.code,
+        map: result.map || null,
+    };
+};
+//# sourceMappingURL=transform-data-llm.js.map

package/dist/src/web/plugin/transform-data-llm.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transform-data-llm.js","sourceRoot":"","sources":["../../../../src/web/plugin/transform-data-llm.ts"],"names":[],"mappings":"AAEA,MAAM,iBAAiB,GAAG,eAAe,CAAC;AAO1C,SAAS,iBAAiB,CAAC,CAAe;IACxC,OAAO;QACL,IAAI,EAAE,gBAAgB;QAEtB,OAAO,EAAE;YACP,OAAO,EAAE;gBACP,KAAK,CAAC,IAAI,EAAE,KAAK;oBACf,KAAK,CAAC,gBAAgB,GAAG,KAAK,CAAC;oBAC/B,KAAK,CAAC,kBAAkB,GAAG,KAAK,CAAC;oBAEjC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;wBAClC,IAAI,CAAC,CAAC,CAAC,mBAAmB,CAAC,IAAI,CAAC;4BAAE,SAAS;wBAC3C,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,KAAK,iBAAiB;4BAAE,SAAS;wBAEtD,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CACvC,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC;4BACtB,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAClD,CAAC;wBAEF,IAAI,YAAY,EAAE,CAAC;4BACjB,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC;4BAC9B,MAAM;wBACR,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,IAAI,CAAC,IAAI,EAAE,KAAK;oBACd,IAAI,KAAK,CAAC,kBAAkB,IAAI,CAAC,KAAK,CAAC,gBAAgB,EAAE,CAAC;wBACxD,MAAM,UAAU,GAAG,CAAC,CAAC,iBAAiB,CACpC;4BACE,CAAC,CAAC,eAAe,CACf,CAAC,CAAC,UAAU,CAAC,SAAS,CAAC,EACvB,CAAC,CAAC,UAAU,CAAC,SAAS,CAAC,CACxB;yBACF,EACD,CAAC,CAAC,aAAa,CAAC,iBAAiB,CAAC,CACnC,CAAC;wBAEF,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;oBACrC,CAAC;gBACH,CAAC;aACF;YAED,UAAU,CAAC,IAAI,EAAE,KAAK;gBACpB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC;gBACzC,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;gBAEtC,MAAM,iBAAiB,GAAG,UAAU,CAAC,SAAS,CAC5C,CAAC,SAAS,EAAE,EAAE,CACZ,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC;oBAC3B,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAC1D,CAAC;gBAEF,IAAI,iBAAiB,KAAK,CAAC,CAAC;oBAAE,OAAO;gBAErC,MAAM,YAAY,GAAG,UAAU,CAC7B,iBAAiB,CACI,CAAC;gBAExB,IAAI,iBAAmC,CAAC;gBAExC,IAAI,CAAC,CAAC,eAAe,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC1C,iBAAiB,GAAG,YAAY,CAAC,KAAK,CAAC;gBACzC,CAAC;qBAAM,IAAI,CAAC,CAAC,wBAAwB,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC1D,iBAAiB,GAAG,YAAY,CAAC,KAAK,CAAC,UAA8B,CAAC;gBACxE,CAAC;qBAAM,CAAC;oBACN,OAAO;gBACT,CAAC;gBAED,MAAM,WAAW,GAAG,CAAC,CAAC,YAAY,CAChC,CAAC,CAAC,aAAa,CAAC,SAAS,CAAC,EAC1B,CAAC,CAAC,eAAe,CAAC,iBAAiB,CAAC;oBAClC,CAAC,CAAC,iBAAiB;oBACnB,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,iBAAiB,CAAC,CAChD,CAAC;gBAEF,MAAM,kBAAkB,GAAG,UAAU,CAAC,MAAM,CAC1C,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,iBAAiB,CAC1C,CAAC;gBACF,MAAM,UAAU,GAAG,CAAC,CAAC,iBAAiB,CACpC,OAAO,CAAC,IAAI,EACZ,kBAAkB,EAClB,OAAO,CAAC,WAAW,CACpB,CAAC;gBAEF,MAAM,iBAAiB,GAAG,CAAC,CAAC,UAAU,CACpC,UAAU,EACV,IAAI,CAAC,IAAI,CAAC,cAAc,EACxB,IAAI,CAAC,IAAI,CAAC,QAAQ,EAClB,IAAI,CAAC,IAAI,CAAC,WAAW,CACtB,CAAC;gBAEF,MAAM,UAAU,GAAG,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,aAAa,CAAC,SAAS,CAAC,EAAE;oBACjE,WAAW;iBACZ,CAAC,CAAC;gBACH,MAAM,UAAU,GAAG,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC;gBAEnE,MAAM,OAAO,GAAG,CAAC,CAAC,UAAU,CAC1B,UAAU,EACV,UAAU,EACV,CAAC,iBAAiB,CAAC,EACnB,KAAK,CACN,CAAC;gBAEF,KAAK,CAAC,kBAAkB,GAAG,IAAI,CAAC;gBAChC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAC5B,CAAC;SACF;KACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,SAAS,GAAG,KAAK,EAAE,IAAY,EAAE,EAAU,EAAE,EAAE;IAC1D,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;QAChC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,yEAAyE;IACzE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;IAEhE,MAAM,YAAY,GAAqB;QACrC,OAAO,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC;QAC/B,UAAU,EAAE;YACV,OAAO,EAAE,CAAC,KAAK,EAAE,YAAY,CAAC;SAC/B;QACD,QAAQ,EAAE,EAAE;QACZ,cAAc,EAAE,EAAE;KACnB,CAAC;IAEF,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;IAEjD,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAC5B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO;QACL,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,GAAG,EAAE,MAAM,CAAC,GAAG,IAAI,IAAI;KACxB,CAAC;AACJ,CAAC,CAAC"}

package/dist/src/web/plugin/transform-data-llm.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/src/web/plugin/transform-data-llm.test.js

@@ -0,0 +1,81 @@
+import { describe, expect, it } from "vitest";
+import { transform } from "./transform-data-llm.js";
+describe("data-llm plugin", () => {
+    it("should transform JSX element with data-llm string attribute", async () => {
+        const code = `
+      function Component() {
+        return <div data-llm="Test description">Content</div>;
+      }
+    `;
+        const result = await transform(code, "test.tsx");
+        expect(result).not.toBeNull();
+        expect(result?.code).toContain("DataLLM");
+        expect(result?.code).toContain('content="Test description"');
+        expect(result?.code).not.toContain("data-llm");
+    });
+    it("should transform JSX element with data-llm expression attribute", async () => {
+        const code = `
+      function Component() {
+        const desc = "Dynamic description";
+        return <div data-llm={desc}>Content</div>;
+      }
+    `;
+        const result = await transform(code, "test.tsx");
+        expect(result).not.toBeNull();
+        expect(result?.code).toContain("DataLLM");
+        expect(result?.code).toContain("content={desc}");
+        expect(result?.code).not.toContain("data-llm");
+    });
+    it("should add import for DataLLM when not present", async () => {
+        const code = `
+      function Component() {
+        return <div data-llm="Test">Content</div>;
+      }
+    `;
+        const result = await transform(code, "test.tsx");
+        expect(result).not.toBeNull();
+        expect(result?.code).toContain('import { DataLLM } from "skybridge/web"');
+    });
+    it("should handle DataLLM imports correctly", async () => {
+        // No duplicate import
+        const codeWithImport = `
+      import { DataLLM } from "skybridge/web";
+      function Component() {
+        return <div data-llm="Test">Content</div>;
+      }
+    `;
+        const result1 = await transform(codeWithImport, "test.tsx");
+        expect(result1?.code.match(/import.*DataLLM.*from.*skybridge\/web/g)).toHaveLength(1);
+        // Preserve other imports and add missing DataLLM
+        const codeWithOthers = `
+      import React from "react";
+      import { useState } from "react";
+      function Component() {
+        return <div data-llm="Test">Content</div>;
+      }
+    `;
+        const result2 = await transform(codeWithOthers, "test.tsx");
+        expect(result2?.code).toContain('import React from "react"');
+        expect(result2?.code).toContain('import { useState } from "react"');
+        expect(result2?.code).toContain('import { DataLLM } from "skybridge/web"');
+    });
+    it("should handle complex JSX with multiple data-llm attributes", async () => {
+        const code = `
+      function Component() {
+        return (
+          <div>
+            <section data-llm="Section 1">
+              <p>Content 1</p>
+            </section>
+            <section data-llm="Section 2">
+              <p>Content 2</p>
+            </section>
+          </div>
+        );
+      }
+    `;
+        const result = await transform(code, "test.tsx");
+        expect(result).toMatchSnapshot();
+    });
+});
+//# sourceMappingURL=transform-data-llm.test.js.map

package/dist/src/web/plugin/transform-data-llm.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transform-data-llm.test.js","sourceRoot":"","sources":["../../../../src/web/plugin/transform-data-llm.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AAEpD,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAEjD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC9B,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAC;QAC7D,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,MAAM,IAAI,GAAG;;;;;KAKZ,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAEjD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC9B,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAEjD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC9B,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,yCAAyC,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,sBAAsB;QACtB,MAAM,cAAc,GAAG;;;;;KAKtB,CAAC;QACF,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;QAC5D,MAAM,CACJ,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAC9D,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAElB,iDAAiD;QACjD,MAAM,cAAc,GAAG;;;;;;KAMtB,CAAC;QACF,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;QAC5D,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAC;QAC7D,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,kCAAkC,CAAC,CAAC;QACpE,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,yCAAyC,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,IAAI,GAAG;;;;;;;;;;;;;KAaZ,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAEjD,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/web/types.d.ts

@@ -1,4 +1,14 @@
+import "react";
+declare module "react" {
+    interface HTMLAttributes<T extends Element = HTMLElement> {
+        "data-llm"?: string;
+    }
+}
 export type UnknownObject = Record<string, unknown>;
+export type Prettify<T> = {
+    [K in keyof T]: T[K];
+} & {};
+export type Objectify<T> = T & UnknownObject;
 type WidgetState = UnknownObject;
 type FileMetadata = {
     fileId: string;

package/dist/src/web/types.js

@@ -1,3 +1,4 @@
+import "react";
 export const TOOL_RESPONSE_EVENT_TYPE = "openai:tool_response";
 export class ToolResponseEvent extends CustomEvent {
     type = TOOL_RESPONSE_EVENT_TYPE;

package/dist/src/web/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/web/types.ts"],"names":[],"mappings":"AAMA,MAAM,CAAC,MAAM,wBAAwB,GAAG,sBAAsB,CAAC;AAC/D,MAAM,OAAO,iBAAkB,SAAQ,WAErC;IACkB,IAAI,GAAG,wBAAwB,CAAC;CACnD;AA8FD,sDAAsD;AACtD,MAAM,CAAC,MAAM,sBAAsB,GAAG,oBAAoB,CAAC;AAC3D,MAAM,OAAO,eAAgB,SAAQ,WAEnC;IACkB,IAAI,GAAG,sBAAsB,CAAC;CACjD"}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/web/types.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,CAAC;AAiBf,MAAM,CAAC,MAAM,wBAAwB,GAAG,sBAAsB,CAAC;AAC/D,MAAM,OAAO,iBAAkB,SAAQ,WAErC;IACkB,IAAI,GAAG,wBAAwB,CAAC;CACnD;AA8FD,sDAAsD;AACtD,MAAM,CAAC,MAAM,sBAAsB,GAAG,oBAAoB,CAAC;AAC3D,MAAM,OAAO,eAAgB,SAAQ,WAEnC;IACkB,IAAI,GAAG,sBAAsB,CAAC;CACjD"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "skybridge",
-  "version": "0.6.0",
+  "version": "0.8.0",
   "description": "Skybridge is a framework for building ChatGPT apps",
   "type": "module",
   "files": [
@@ -20,7 +20,7 @@
     "build": "tsc && pnpm run build:templates",
     "build:templates": "cp -r src/server/templates dist/src/server/",
     "test": "pnpm run test:unit && pnpm run test:type",
-    "test:unit": "vitest run --silent",
+    "test:unit": "vitest run",
     "test:type": "tsc --noEmit",
     "docs:dev": "pnpm --filter @skybridge/docs start",
     "docs:build": "pnpm --filter @skybridge/docs build",
@@ -39,6 +39,7 @@
     "react-dom": ">=18.0.0"
   },
   "dependencies": {
+    "@babel/core": "^7.28.5",
     "@modelcontextprotocol/sdk": "^1.20.0",
     "cors": "^2.8.5",
     "express": "^5.1.0",
@@ -50,6 +51,7 @@
     "@testing-library/dom": "^10.4.1",
     "@testing-library/react": "^16.3.0",
     "@total-typescript/tsconfig": "^1.0.4",
+    "@types/babel__core": "^7.20.5",
     "@types/cors": "^2.8.19",
     "@types/express": "^5.0.3",
     "@types/jsdom": "^21.1.6",

package/dist/src/web/plugin.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"plugin.js","sourceRoot":"","sources":["../../../src/web/plugin.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,SAAS;IACvB,OAAO;QACL,IAAI,EAAE,WAAW;QAEjB,KAAK,CAAC,MAAM,CAAC,MAAM;YACjB,4EAA4E;YAC5E,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;YAC7C,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;YAE9C,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;YACjD,MAAM,cAAc,GAAG,OAAO,CAC5B,WAAW,EACX,oCAAoC,CACrC,CAAC;YAEF,MAAM,KAAK,GAAG,MAAM,CAAC,WAAW,CAC9B,QAAQ,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;gBACrC,IAAI,CAAC,KAAK,CAAC,0BAA0B,CAAC,EAAE,CAAC,CAAC,CAAC;gBAC3C,IAAI;aACL,CAAC,CACH,CAAC;YAEF,OAAO;gBACL,IAAI,EAAE,SAAS;gBACf,KAAK,EAAE;oBACL,QAAQ,EAAE,IAAI;oBACd,MAAM,EAAE,IAAI;oBACZ,YAAY,EAAE,KAAK;oBACnB,aAAa,EAAE;wBACb,KAAK;qBACN;iBACF;aACF,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC"}