chore: add GPU driver and toolkit status (#2628)

janhq · Apr 8, 2024 · f8cf93a · f8cf93a
1 parent c5c3156
commit f8cf93a
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 101 deletions.
diff --git a/extensions/monitoring-extension/src/node/index.ts b/extensions/monitoring-extension/src/node/index.ts
@@ -221,6 +221,7 @@ const updateGpuInfo = async () =>
 
             data = updateCudaExistence(data)
             writeFileSync(GPU_INFO_FILE, JSON.stringify(data, null, 2))
+            log(`[APP]::${JSON.stringify(data)}`)
             resolve({})
           } else {
             reject(error)
@@ -263,6 +264,7 @@ const updateGpuInfo = async () =>
 
           data = updateCudaExistence(data)
           writeFileSync(GPU_INFO_FILE, JSON.stringify(data, null, 2))
+          log(`[APP]::${JSON.stringify(data)}`)
           resolve({})
         }
       )

diff --git a/web/containers/GPUDriverPromptModal/index.tsx b/web/containers/GPUDriverPromptModal/index.tsx
diff --git a/web/containers/Providers/index.tsx b/web/containers/Providers/index.tsx
@@ -6,7 +6,6 @@ import { Toaster } from 'react-hot-toast'
 
 import { TooltipProvider } from '@janhq/uikit'
 
-import GPUDriverPrompt from '@/containers/GPUDriverPromptModal'
 import EventListenerWrapper from '@/containers/Providers/EventListener'
 import JotaiWrapper from '@/containers/Providers/Jotai'
 import ThemeWrapper from '@/containers/Providers/Theme'
@@ -81,7 +80,6 @@ const Providers = ({ children }: PropsWithChildren) => {
               <TooltipProvider delayDuration={0}>
                 <DataLoader>{children}</DataLoader>
               </TooltipProvider>
-              {!isMac && <GPUDriverPrompt />}
             </EventListenerWrapper>
             <Toaster />
           </KeyListener>

diff --git a/web/hooks/useSettings.ts b/web/hooks/useSettings.ts
@@ -1,9 +1,6 @@
 import { useCallback, useEffect, useState } from 'react'
 
 import { fs, joinPath } from '@janhq/core'
-import { atom, useAtom } from 'jotai'
-
-export const isShowNotificationAtom = atom<boolean>(false)
 
 export type AppSettings = {
   run_mode: 'cpu' | 'gpu' | undefined
@@ -15,9 +12,6 @@ export type AppSettings = {
 
 export const useSettings = () => {
   const [isGPUModeEnabled, setIsGPUModeEnabled] = useState(false) // New state for GPU mode
-  const [showNotification, setShowNotification] = useAtom(
-    isShowNotificationAtom
-  )
   const [settings, setSettings] = useState<AppSettings>()
 
   useEffect(() => {
@@ -29,15 +23,6 @@ export const useSettings = () => {
 
   const validateSettings = async () => {
     readSettings().then((settings) => {
-      if (
-        settings &&
-        settings.notify &&
-        ((settings.nvidia_driver?.exist && !settings.cuda?.exist) ||
-          !settings.nvidia_driver?.exist)
-      ) {
-        setShowNotification(false)
-      }
-
       // Check if run_mode is 'gpu' or 'cpu' and update state accordingly
       setIsGPUModeEnabled(settings?.run_mode === 'gpu')
     })
@@ -84,11 +69,9 @@ export const useSettings = () => {
   }
 
   return {
-    showNotification,
     isGPUModeEnabled,
     readSettings,
     saveSettings,
-    setShowNotification,
     validateSettings,
     settings,
   }

diff --git a/web/screens/Settings/Advanced/index.tsx b/web/screens/Settings/Advanced/index.tsx
@@ -67,8 +67,7 @@ const Advanced = () => {
   const [gpuList, setGpuList] = useState<GPU[]>([])
   const [gpusInUse, setGpusInUse] = useState<string[]>([])
 
-  const { readSettings, saveSettings, validateSettings, setShowNotification } =
-    useSettings()
+  const { readSettings, saveSettings, validateSettings } = useSettings()
   const { stopModel } = useActiveModel()
 
   const selectedGpu = gpuList
@@ -273,7 +272,6 @@ const Advanced = () => {
                       if (e === true) {
                         saveSettings({ runMode: 'gpu' })
                         setGpuEnabled(true)
-                        setShowNotification(false)
                         snackbar({
                           description:
                             'Successfully turned on GPU Acceleration',