fix: auto save tool & bump app version (#1100)

* remove ref
dependencies

* ensure not triggering when saving tool

* only make optimistic updates in messages once

* bump app version

Author: paulclindo

apps/shinkai-desktop/src/components/chat/websocket-message.tsx

@@ -191,25 +191,6 @@ export const useWebSocketMessage = ({
             }
           }),
         );
-
-        if (parseData.metadata?.is_done === true) {
-          queryClient.setQueryData(
-            queryKey,
-            produce((draft: ChatConversationInfiniteData | undefined) => {
-              if (!draft?.pages?.[0]) return;
-              const lastMessage = draft.pages.at(-1)?.at(-1);
-              if (
-                lastMessage &&
-                lastMessage.role === 'assistant' &&
-                lastMessage.messageId === OPTIMISTIC_ASSISTANT_MESSAGE_ID
-              ) {
-                lastMessage.status = { type: 'complete', reason: 'unknown' };
-              }
-            }),
-          );
-
-          return;
-        }
       } catch (error) {
         console.error('Failed to parse ws message', error);
       }
@@ -262,7 +243,6 @@ export const useWebSocketTools = ({
   const { inboxId: encodedInboxId = '' } = useParams();
   const inboxId = defaultInboxId || decodeURIComponent(encodedInboxId);
   const queryClient = useQueryClient();
-  const isToolReceived = useRef(false);
 
   const setWidget = useToolsStore((state) => state.setWidget);
 
@@ -277,23 +257,10 @@ export const useWebSocketTools = ({
         const parseData: WsMessage = JSON.parse(lastMessage.data);
         if (parseData.inbox !== inboxId) return;
 
-        if (
-          parseData.message_type === 'ShinkaiMessage' &&
-          isToolReceived.current
-        ) {
-          isToolReceived.current = false;
-          const paginationKey = [
-            FunctionKeyV2.GET_CHAT_CONVERSATION_PAGINATION,
-            { inboxId: inboxId as string },
-          ];
-          void queryClient.invalidateQueries({ queryKey: paginationKey });
-          return;
-        }
         if (
           parseData.message_type === 'Widget' &&
           parseData?.widget?.ToolRequest
         ) {
-          isToolReceived.current = true;
           const tool = parseData.widget.ToolRequest;
           queryClient.setQueryData(
             queryKey,

apps/shinkai-desktop/src/components/playground-tool/hooks/use-tool-flow.ts

@@ -451,13 +451,17 @@ export const useToolFlow = ({
         }
       }
     }
+    // eslint-disable-next-line react-hooks/exhaustive-deps
   }, [
     metadataConversation?.pages,
     setToolMetadata,
     setToolMetadataStatus,
     setToolMetadataError,
     isPlaygroundMode,
     toolMetadataStatus,
+    sendMetadataGenerationCompleteNotification,
+    setCurrentStep,
+    setToolCreationError,
   ]);
 
   useEffect(() => {
@@ -592,7 +596,8 @@ export const useToolFlow = ({
       toolMetadata &&
       toolCodeStatus === 'success' &&
       toolMetadataStatus === 'success' &&
-      forceAutoSave.current
+      forceAutoSave.current &&
+      !isSavingTool
     ) {
       forceAutoSave.current = false;
       forceGenerateMetadata.current = false;
@@ -623,9 +628,7 @@ export const useToolFlow = ({
     handleSaveTool,
     form,
     setCurrentStep,
-    forceAutoSave,
-    forceGenerateMetadata,
-    forceGenerateCode,
+    isSavingTool,
   ]);
 
   const handleApplyChangesCodeSubmit = useCallback(

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "@shinkai/source",
-  "version": "1.1.14",
+  "version": "1.1.15",
   "license": "SEE LICENSE IN LICENSE",
   "files": [
     "LICENSE"