@smart-cloud/ai-kit-ui 1.4.0 → 1.4.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@smart-cloud/ai-kit-ui",
-  "version": "1.4.0",
+  "version": "1.4.2",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",
@@ -20,17 +20,20 @@
     "@emotion/cache": "^11.14.0",
     "@emotion/react": "^11.14.0",
     "@mantine/colors-generator": "^8.3.16",
-    "@smart-cloud/ai-kit-core": "^1.4.2",
+    "@smart-cloud/ai-kit-core": "^1.4.4",
     "@smart-cloud/wpsuite-core": "^2.2.10",
     "@tabler/icons-react": "^3.40.0",
     "chroma-js": "^3.2.0",
     "react-markdown": "^10.1.0",
+    "rehype-parse": "^9.0.1",
     "rehype-raw": "^7.0.0",
+    "rehype-remark": "^10.0.0",
     "rehype-sanitize": "^6.0.0",
     "rehype-stringify": "^10.0.1",
     "remark-gfm": "^4.0.1",
     "remark-parse": "^11.0.0",
     "remark-rehype": "^11.1.2",
+    "remark-stringify": "^11.0.0",
     "unified": "^11.0.5"
   },
   "peerDependencies": {

package/src/ai-feature/AiFeature.tsx

@@ -472,7 +472,31 @@ const AiFeatureBase: FC<AiFeatureProps & AiKitShellInjectedProps> = (props) => {
   }, [text, defaults]);
 
   const canGenerate = useMemo(() => {
-    const text = typeof inputText === "function" ? inputText() : inputText;
+    // If inputText is a function (async or sync getText), we can't determine
+    // if it has content without calling it. Assume it's valid if provided.
+    const input = inputText;
+    if (typeof input === "function") {
+      switch (mode) {
+        case "generateImageMetadata":
+          return Boolean(image);
+        case "translate":
+          // For translate, we need outputLanguage check, but can't check text without calling getText
+          return Boolean(
+            !outputLanguage || detectedLanguage !== outputLanguage,
+          );
+        case "summarize":
+        case "proofread":
+        case "rewrite":
+        case "write":
+        case "generatePostMetadata":
+          return true; // Assume getText will provide valid content
+        default:
+          return false;
+      }
+    }
+
+    // If inputText is a string, check it directly
+    const text = input as string | undefined;
     switch (mode) {
       case "generateImageMetadata":
         return Boolean(image);
@@ -507,8 +531,11 @@ const AiFeatureBase: FC<AiFeatureProps & AiKitShellInjectedProps> = (props) => {
       setError(null);
       setGenerated(null);
 
+      const input = await inputText;
       try {
-        const text = typeof inputText === "function" ? inputText() : inputText;
+        // Support both sync and async getText functions
+        const text =
+          typeof input === "function" ? await Promise.resolve(input()) : input;
         switch (mode) {
           case "summarize": {
             const res = await ai.run(async ({ signal, onStatus }) => {
@@ -524,13 +551,17 @@ const AiFeatureBase: FC<AiFeatureProps & AiKitShellInjectedProps> = (props) => {
                 type: type as SummarizerType,
                 outputLanguage: outLang as SummarizeArgs["outputLanguage"],
               };
-              const out = await summarize(args, {
+
+              const featureOptions: FeatureOptions = {
                 signal,
                 onStatus,
                 context,
                 modeOverride,
                 onDeviceTimeoutOverride: onDeviceTimeout,
-              });
+              };
+
+              // Normal single-pass summarization
+              const out = await summarize(args, featureOptions);
               return out.result;
             });
             setGenerated((res as never) ?? "");
@@ -614,13 +645,17 @@ const AiFeatureBase: FC<AiFeatureProps & AiKitShellInjectedProps> = (props) => {
                 sourceLanguage: inputLang!,
                 targetLanguage: outLang,
               };
-              const out = await translate(args, {
+
+              const featureOptions: FeatureOptions = {
                 signal,
                 onStatus,
                 context,
                 modeOverride,
                 onDeviceTimeoutOverride: onDeviceTimeout,
-              });
+              };
+
+              // Normal single-pass translation
+              const out = await translate(args, featureOptions);
               return out.result;
             });
             setGenerated((res as never) ?? "");
@@ -652,13 +687,17 @@ const AiFeatureBase: FC<AiFeatureProps & AiKitShellInjectedProps> = (props) => {
                 length: length as RewriterLength,
                 outputLanguage: outLang as RewriteArgs["outputLanguage"],
               };
-              const out = await rewrite(args, {
+
+              const featureOptions: FeatureOptions = {
                 signal,
                 onStatus,
                 context,
                 modeOverride,
                 onDeviceTimeoutOverride: onDeviceTimeout,
-              });
+              };
+
+              // Normal single-pass rewrite
+              const out = await rewrite(args, featureOptions);
               return out.result;
             });
             setGenerated((res as never) ?? "");

package/tsup.config.ts

@@ -2,7 +2,8 @@ import { defineConfig } from "tsup";
 
 export default defineConfig({
   // Copy non-hashed global CSS so consumers can import it (like Mantine styles)
-  onSuccess: "node -e \"const fs=require('fs'); const path=require('path'); fs.mkdirSync('dist',{recursive:true}); fs.copyFileSync(path.join('src','styles','ai-kit-ui.css'), path.join('dist','ai-kit-ui.css'));\"",
+  onSuccess:
+    "node -e \"const fs=require('fs'); const path=require('path'); fs.mkdirSync('dist',{recursive:true}); fs.copyFileSync(path.join('src','styles','ai-kit-ui.css'), path.join('dist','ai-kit-ui.css'));\"",
 
   entry: ["src/index.tsx"],
   format: ["cjs", "esm"],