@smart-cloud/ai-kit-ui 1.1.30 → 1.1.31

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@smart-cloud/ai-kit-ui",
-  "version": "1.1.30",
+  "version": "1.1.31",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",
@@ -20,7 +20,7 @@
     "@emotion/cache": "^11.14.0",
     "@emotion/react": "^11.14.0",
     "@mantine/colors-generator": "^8.3.14",
-    "@smart-cloud/ai-kit-core": "^1.1.14",
+    "@smart-cloud/ai-kit-core": "^1.1.16",
     "@smart-cloud/wpsuite-core": "^2.1.2",
     "@tabler/icons-react": "^3.36.1",
     "chroma-js": "^3.2.0",

package/src/ai-feature/AiFeature.tsx

@@ -333,6 +333,7 @@ const AiFeatureBase: FC<AiFeatureProps & AiKitShellInjectedProps> = (props) => {
     );
   }, [allowOverride]);
 
+  const [state, setState] = useState<string>();
   const [featureOpen, setFeatureOpen] = useState<boolean>(!showOpenButton);
   const [optionsOpen, setOptionsOpen] = useState<boolean>(false);
   const [backendConfigured, setBackendConfigured] = useState(false);
@@ -396,7 +397,7 @@ const AiFeatureBase: FC<AiFeatureProps & AiKitShellInjectedProps> = (props) => {
 
   const formatAiKitStatus = useCallback(
     (e: AiKitStatusEvent | null): string | null => {
-      if (!e) return null;
+      if (!e || e.silent) return null;
 
       const step = e.step;
       const msg = I18n.get((e.message ?? "").trim());
@@ -420,11 +421,21 @@ const AiFeatureBase: FC<AiFeatureProps & AiKitShellInjectedProps> = (props) => {
         case "on-device:run":
           return msg || I18n.get("Generating...");
         case "backend:request":
-          return msg || I18n.get("Sending request to backend...");
         case "backend:waiting":
-          return msg || I18n.get("Waiting for backend response...");
         case "backend:response":
-          return msg || I18n.get("Received backend response.");
+          switch (mode) {
+            case "rewrite":
+              return I18n.get("Rewriting text");
+            case "summarize":
+              return I18n.get("Summarizing");
+            case "proofread":
+              return I18n.get("Proofreading");
+            case "write":
+            case "generatePostMetadata":
+            case "generateImageMetadata":
+            default:
+              return I18n.get("Generating text");
+          }
         case "done":
           return msg || I18n.get("Done.");
         case "error":
@@ -433,7 +444,7 @@ const AiFeatureBase: FC<AiFeatureProps & AiKitShellInjectedProps> = (props) => {
           return msg || I18n.get("Working...");
       }
     },
-    [language],
+    [language, mode],
   );
 
   const inputText = useMemo(() => {
@@ -463,7 +474,8 @@ const AiFeatureBase: FC<AiFeatureProps & AiKitShellInjectedProps> = (props) => {
   }, [inputText, mode, image, inputLanguage, outputLanguage]);
 
   const ai = useAiRun();
-  const statusText = formatAiKitStatus(ai.statusEvent);
+  const statusText =
+    formatAiKitStatus(ai.statusEvent) ?? (state ? I18n.get(state) : null);
 
   const runGenerate = useCallback(
     async (modeOverride?: AiModePreference) => {
@@ -516,6 +528,7 @@ const AiFeatureBase: FC<AiFeatureProps & AiKitShellInjectedProps> = (props) => {
                     context,
                     modeOverride,
                     onDeviceTimeoutOverride: onDeviceTimeout,
+                    silent: true,
                   },
                 );
                 const langCodes = res.result?.candidates
@@ -545,11 +558,13 @@ const AiFeatureBase: FC<AiFeatureProps & AiKitShellInjectedProps> = (props) => {
             const res = await ai.run(async ({ signal, onStatus }) => {
               let inputLang = inputLanguage ?? "auto";
               if (inputLang === "auto") {
+                setState("Detecting input language...");
                 inputLang = await detectTopLanguage(text!.trim(), {
                   signal,
                   context,
                   modeOverride,
                   onDeviceTimeoutOverride: onDeviceTimeout,
+                  silent: true,
                 });
                 setInputLanguage(inputLang);
               }
@@ -589,11 +604,13 @@ const AiFeatureBase: FC<AiFeatureProps & AiKitShellInjectedProps> = (props) => {
                   ? outputLanguage
                   : null) || readDefaultOutputLanguage();
               if (outputLanguage === "auto") {
+                setState("Detecting input language...");
                 outLang = await detectTopLanguage(text!.trim(), {
                   signal,
                   context,
                   modeOverride,
                   onDeviceTimeoutOverride: onDeviceTimeout,
+                  silent: true,
                 });
                 setOutputLanguage(outLang);
               }
@@ -637,12 +654,15 @@ const AiFeatureBase: FC<AiFeatureProps & AiKitShellInjectedProps> = (props) => {
                 context,
                 modeOverride,
                 onDeviceTimeoutOverride: onDeviceTimeout,
+                silent: true,
               };
+              setState("Detecting input language...");
               const inLang = await detectTopLanguage(
                 text!.trim() + "\n" + (instructions?.trim() || ""),
                 langFeatureOptions,
               );
               if (inLang !== outLang && inLang !== "en") {
+                setState("Translating instructions...");
                 args.prompt = (
                   await translate(
                     {
@@ -727,6 +747,7 @@ Follow these additional instructions: ${instructions}`
                 ? outputLanguage
                 : null) || readDefaultOutputLanguage();
             try {
+              setState("Translating result...");
               const parsed = (await ai.run(async ({ signal }) => {
                 return await parsePostMetadataFromPromptResult(
                   cleaned,
@@ -736,6 +757,7 @@ Follow these additional instructions: ${instructions}`
                     context,
                     modeOverride,
                     onDeviceTimeoutOverride: onDeviceTimeout,
+                    silent: true,
                   },
                 );
               })) as string | null;
@@ -811,6 +833,7 @@ Follow these additional instructions: ${instructions}`
 
               const cleaned = stripCodeFence(res).trim();
               try {
+                setState("Translating result...");
                 const parsed = (await ai.run(async ({ signal }) => {
                   return await parseImageMetadataFromPromptResult(
                     cleaned,
@@ -820,6 +843,7 @@ Follow these additional instructions: ${instructions}`
                       context,
                       modeOverride,
                       onDeviceTimeoutOverride: onDeviceTimeout,
+                      silent: true,
                     },
                   );
                 })) as string | null;
@@ -840,6 +864,7 @@ Follow these additional instructions: ${instructions}`
             : I18n.get("An unknown error occurred."),
         );
       }
+      setState(undefined);
     },
     [
       language,