fix: model stats fallback (#46)

louisgv · Jun 22, 2023 · 6b8087b · 6b8087b · vercel · Jun 22, 2023
1 parent 850dfb7
commit 6b8087b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 8 deletions.
diff --git a/apps/desktop/src-tauri/src/model/stats.rs b/apps/desktop/src-tauri/src/model/stats.rs
@@ -10,7 +10,7 @@ pub struct ModelStats {
 crate::macros::bucket_state::make!(ModelStats, "model_stats", "v1");
 impl State {
   pub fn increment_load_count(&self, path: &str) -> Result<(), String> {
-    let current_value = self.get(path)?;
+    let current_value = self.get(path).unwrap_or_default();
     let bucket = self.0.lock();
     let file_path = String::from(path);
 

diff --git a/apps/desktop/src/features/inference-server/model-config.tsx b/apps/desktop/src/features/inference-server/model-config.tsx
@@ -48,8 +48,7 @@ export const ModelConfig = () => {
   const {
     model,
     modelLoadState,
-    launchCount,
-    incrementLaunchCount,
+    modelStats,
     loadModel,
     downloadState,
     modelConfig
@@ -137,7 +136,7 @@ export const ModelConfig = () => {
                 "text-xs rounded-lg bg-gray-6 py-2 px-3",
                 className
               )}>
-              {launchCount}
+              {modelStats.launchCount}
             </code>
           )}
           isSpinning={modelLoadState === ModelLoadState.Loading}
@@ -147,7 +146,7 @@ export const ModelConfig = () => {
               downloadState !== DownloadState.Completed)
           }
           onClick={() => {
-            incrementLaunchCount()
+            modelStats.incrementLaunchCount()
             loadModel()
           }}>
           {modelLoadState === ModelLoadState.Loaded ? "Loaded" : "Load Model"}

diff --git a/apps/desktop/src/providers/model.ts b/apps/desktop/src/providers/model.ts
@@ -57,7 +57,7 @@ const useModelProvider = ({ model }: { model: ModelMetadata }) => {
   const { downloadState, pauseDownload, progress, resumeDownload, modelSize } =
     useModelDownload(model)
 
-  const { launchCount, incrementLaunchCount } = useModelStats(model)
+  const modelStats = useModelStats(model)
 
   const knownModelInfo = useMemo(
     () => modelMap?.[integrity?.blake3],
@@ -137,8 +137,7 @@ const useModelProvider = ({ model }: { model: ModelMetadata }) => {
     progress,
     pauseDownload,
     resumeDownload,
-    launchCount,
-    incrementLaunchCount
+    modelStats
   }
 }