From 3ea8c6071c7ae6a808a5113f98af65763b5edf73 Mon Sep 17 00:00:00 2001
From: Louis <louis@jan.ai>
Date: Sun, 18 Feb 2024 20:55:31 +0700
Subject: [PATCH] fix: check for updates should show no update are availale on
 the latest build (#2075)

---
 electron/handlers/update.ts            |  3 ++-
 electron/package.json                  |  1 +
 electron/utils/menu.ts                 | 22 +++++++++++++++++++---
 models/mistral-ins-7b-q4/model.json    |  2 +-
 models/openhermes-neural-7b/model.json |  2 +-
 models/trinity-v1.2-7b/model.json      |  2 +-
 6 files changed, 25 insertions(+), 7 deletions(-)

diff --git a/electron/handlers/update.ts b/electron/handlers/update.ts
index cfd738f78..0d8cc4cc0 100644
--- a/electron/handlers/update.ts
+++ b/electron/handlers/update.ts
@@ -11,7 +11,8 @@ export function handleAppUpdates() {
   /* New Update Available */
   autoUpdater.on('update-available', async (_info: any) => {
     const action = await dialog.showMessageBox({
-      message: `Update available. Do you want to download the latest update?`,
+      title: 'Update Available',
+      message: 'Would you like to download and install it now?',
       buttons: ['Download', 'Later'],
     })
     if (action.response === 0) await autoUpdater.downloadUpdate()
diff --git a/electron/package.json b/electron/package.json
index deff3826a..a89803077 100644
--- a/electron/package.json
+++ b/electron/package.json
@@ -4,6 +4,7 @@
   "main": "./build/main.js",
   "author": "Jan <service@jan.ai>",
   "license": "MIT",
+  "productName": "Jan",
   "homepage": "https://github.com/janhq/jan/tree/main/electron",
   "description": "Use offline LLMs with your own data. Run open source models like Llama2 or Falcon on your internal computers/servers.",
   "build": {
diff --git a/electron/utils/menu.ts b/electron/utils/menu.ts
index 4825991ee..da6d52801 100644
--- a/electron/utils/menu.ts
+++ b/electron/utils/menu.ts
@@ -1,7 +1,8 @@
 // @ts-nocheck
-import { app, Menu, shell } from 'electron'
-const isMac = process.platform === 'darwin'
+import { app, Menu, shell, dialog } from 'electron'
 import { autoUpdater } from 'electron-updater'
+import { log } from '@janhq/core/node'
+const isMac = process.platform === 'darwin'
 
 const template: (Electron.MenuItemConstructorOptions | Electron.MenuItem)[] = [
   {
@@ -12,7 +13,22 @@ const template: (Electron.MenuItemConstructorOptions | Electron.MenuItem)[] = [
         label: 'Check for Updates...',
         click: () =>
           // Check for updates and notify user if there are any
-          autoUpdater.checkForUpdatesAndNotify(),
+          autoUpdater
+            .checkForUpdatesAndNotify()
+            .then((updateCheckResult) => {
+              if (
+                !updateCheckResult?.updateInfo ||
+                updateCheckResult?.updateInfo.version === app.getVersion()
+              ) {
+                dialog.showMessageBox({
+                  message: `No updates available.`,
+                })
+                return
+              }
+            })
+            .catch((error) => {
+              log('Error checking for updates:' + JSON.stringify(error))
+            }),
       },
       { type: 'separator' },
       { role: 'services' },
diff --git a/models/mistral-ins-7b-q4/model.json b/models/mistral-ins-7b-q4/model.json
index bfdaffa90..75e0cbf9f 100644
--- a/models/mistral-ins-7b-q4/model.json
+++ b/models/mistral-ins-7b-q4/model.json
@@ -29,7 +29,7 @@
     "author": "MistralAI, The Bloke",
     "tags": ["Featured", "7B", "Foundational Model"],
     "size": 4370000000,
-    "cover": "https://raw.githubusercontent.com/janhq/jan/main/models/mistral-ins-7b-q4/cover.png"
+    "cover": "https://raw.githubusercontent.com/janhq/jan/dev/models/mistral-ins-7b-q4/cover.png"
   },
   "engine": "nitro"
 }
diff --git a/models/openhermes-neural-7b/model.json b/models/openhermes-neural-7b/model.json
index 87e1df143..a13a0f2b8 100644
--- a/models/openhermes-neural-7b/model.json
+++ b/models/openhermes-neural-7b/model.json
@@ -28,7 +28,7 @@
     "author": "Intel, Jan",
     "tags": ["7B", "Merged", "Featured"],
     "size": 4370000000,
-    "cover": "https://raw.githubusercontent.com/janhq/jan/main/models/openhermes-neural-7b/cover.png"
+    "cover": "https://raw.githubusercontent.com/janhq/jan/dev/models/openhermes-neural-7b/cover.png"
   },
   "engine": "nitro"
 }
diff --git a/models/trinity-v1.2-7b/model.json b/models/trinity-v1.2-7b/model.json
index 2dda120e6..947629642 100644
--- a/models/trinity-v1.2-7b/model.json
+++ b/models/trinity-v1.2-7b/model.json
@@ -28,7 +28,7 @@
     "author": "Jan",
     "tags": ["7B", "Merged", "Featured"],
     "size": 4370000000,
-    "cover": "https://raw.githubusercontent.com/janhq/jan/main/models/trinity-v1.2-7b/cover.png"
+    "cover": "https://raw.githubusercontent.com/janhq/jan/dev/models/trinity-v1.2-7b/cover.png"
   },
   "engine": "nitro"
 }