diff --git a/Cargo.lock b/Cargo.lock
index c42611e902..f912769820 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -3714,7 +3714,7 @@ dependencies = [
 
 [[package]]
 name = "desktop"
-version = "0.0.78"
+version = "0.0.77"
 dependencies = [
  "anyhow",
  "bytes",
diff --git a/apps/desktop/src-tauri/Cargo.toml b/apps/desktop/src-tauri/Cargo.toml
index f44ae727e3..1354a11494 100644
--- a/apps/desktop/src-tauri/Cargo.toml
+++ b/apps/desktop/src-tauri/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "desktop"
-version = "0.0.78"
+version = "0.0.77"
 authors = ["you"]
 edition = "2021"
 description = "Hyprnote Desktop App"
diff --git a/apps/desktop/src-tauri/tauri.conf.json b/apps/desktop/src-tauri/tauri.conf.json
index 0a4a4898e6..1ce6398d26 100644
--- a/apps/desktop/src-tauri/tauri.conf.json
+++ b/apps/desktop/src-tauri/tauri.conf.json
@@ -1,6 +1,6 @@
 {
   "$schema": "https://schema.tauri.app/config/2",
-  "version": "0.0.78",
+  "version": "0.0.77",
   "productName": "Hyprnote Dev",
   "mainBinaryName": "Hyprnote Dev",
   "identifier": "com.hyprnote.dev",
diff --git a/apps/desktop/src/components/right-panel/hooks/useChatLogic.ts b/apps/desktop/src/components/right-panel/hooks/useChatLogic.ts
index a2c1aaab10..8759f6c4d0 100644
--- a/apps/desktop/src/components/right-panel/hooks/useChatLogic.ts
+++ b/apps/desktop/src/components/right-panel/hooks/useChatLogic.ts
@@ -19,6 +19,7 @@ import {
   smoothStream,
   stepCountIs,
   streamText,
+  tool,
 } from "@hypr/utils/ai";
 import { useSessions } from "@hypr/utils/contexts";
 import { useQueryClient } from "@tanstack/react-query";
@@ -330,6 +331,7 @@ export function useChatLogic({
         tools: {
           ...(shouldUseTools && { ...hyprMcpTools, ...newMcpTools }),
           ...(shouldUseTools && baseTools),
+          ...(type === "HyprLocal" && { progress_update: tool({ inputSchema: z.any() }) }),
         },
         onError: (error) => {
           console.error("On Error Catch:", error);
@@ -408,6 +410,7 @@ export function useChatLogic({
 
         if (chunk.type === "tool-call" && !(chunk.toolName === "progress_update" && type === "HyprLocal")) {
           // Save accumulated AI text before processing tool
+
           if (currentAiTextMessageId && aiResponse.trim()) {
             const saveAiText = async () => {
               try {
diff --git a/apps/desktop/src/components/right-panel/utils/chat-utils.ts b/apps/desktop/src/components/right-panel/utils/chat-utils.ts
index 924fd82d1c..3fb3bb7877 100644
--- a/apps/desktop/src/components/right-panel/utils/chat-utils.ts
+++ b/apps/desktop/src/components/right-panel/utils/chat-utils.ts
@@ -77,7 +77,6 @@ export const prepareMessageHistory = async (
     || modelId === "gpt-4o"
     || modelId === "openai/gpt-5"
     || (apiBase && apiBase.includes("pro.hyprnote.com"))
-    || type === "HyprLocal"
   );
 
   const systemContent = await templateCommands.render("chat.system", {