winidows specific fixes

Author: ZahinTapadar

ollama-search

@@ -161,6 +161,74 @@ export async function listOllamaModels(): Promise<string[]> {
   }
 }
 
+// Add periodic model refresh and Windows-specific fixes
+export async function listOllamaModelsWithRetry(): Promise<string[]> {
+  const maxRetries = 5; // Increased from 3 for Windows
+  const retryDelay = 800; // Slightly reduced delay
+  
+  for (let attempt = 1; attempt <= maxRetries; attempt++) {
+    try {
+      const models = await listOllamaModels();
+      if (models.length > 0) {
+        return models;
+      }
+      
+      // If no models found and not the last attempt, wait and retry
+      if (attempt < maxRetries) {
+        // On Windows, add a small random jitter to avoid timing issues
+        const jitter = Math.random() * 200; // 0-200ms
+        await new Promise(resolve => setTimeout(resolve, (retryDelay * attempt) + jitter));
+      }
+    } catch (error) {
+      console.warn(`Model listing attempt ${attempt} failed:`, error);
+      
+      // If not the last attempt, wait and retry
+      if (attempt < maxRetries) {
+        const jitter = Math.random() * 200; // 0-200ms
+        await new Promise(resolve => setTimeout(resolve, (retryDelay * attempt) + jitter));
+      }
+    }
+  }
+  
+  return [];
+}
+
+// Windows-specific model refresh with force reload
+export async function forceRefreshModels(): Promise<string[]> {
+  try {
+    // On Windows, sometimes we need to force a refresh of the Ollama connection
+    // First, try to "wake up" Ollama by making a simple API call
+    await fetch("http://localhost:11434/api/tags", {
+      method: "GET",
+      signal: AbortSignal.timeout(3000),
+    }).catch(() => {}); // Ignore errors, this is just to wake up the service
+    
+    // Wait a moment for the service to respond
+    await new Promise(resolve => setTimeout(resolve, 500));
+    
+    // Now try to get the models with retry logic
+    const models = await listOllamaModelsWithRetry();
+    
+    // If still no models on Windows, try one more aggressive approach
+    if (models.length === 0) {
+      try {
+        // Try the backend command directly as a last resort
+        const backendModels = await invoke('list_installed_models') as string[];
+        if (backendModels && backendModels.length > 0) {
+          return backendModels;
+        }
+      } catch (backendError) {
+        console.warn('Backend fallback failed:', backendError);
+      }
+    }
+    
+    return models;
+  } catch (error) {
+    console.error('Force refresh failed:', error);
+    return [];
+  }
+}
+
 export interface OllamaStatus {
   isInstalled: boolean;
   isRunning: boolean;

src/app/services/ollamaService.ts

@@ -2,7 +2,7 @@
 
 import { useState, useEffect, useRef, useCallback } from "react";
 import { useTheme } from "next-themes";
-import { askOllama, listOllamaModels, checkOllamaStatus, askOllamaVerbose, askOllamaStreaming, searchWeb, type OllamaStatus } from "@/app/services/ollamaService";
+import { askOllama, listOllamaModels, forceRefreshModels, checkOllamaStatus, askOllamaVerbose, askOllamaStreaming, searchWeb, type OllamaStatus } from "@/app/services/ollamaService";
 import { PlaceholdersAndVanishInput } from "@/components/ui/placeholders-and-vanish-input";
 import { TextGenerateEffect } from "@/components/ui/enhanced-text-generate-effect";
 import { StreamingTextEffect } from "@/components/ui/streaming-text-effect";
@@ -170,19 +170,45 @@ export default function Chat() {
     };
   }, [clearConfirmation.isOpen, clearConfirmation.isClearing]);
 
-  const refreshModels = useCallback(async () => {
+  const refreshModels = useCallback(async (forceRefresh: boolean = false) => {
     if (!ollamaStatus.isRunning) return;
 
     setRefreshingModels(true);
     try {
-      const availableModels = await listOllamaModels();
+      // Use force refresh on Windows or when explicitly requested
+      const isWindows = typeof navigator !== 'undefined' && navigator.userAgent.includes('Windows');
+      const availableModels = forceRefresh || isWindows 
+        ? await forceRefreshModels() 
+        : await listOllamaModels();
+      
       setModels(availableModels);
 
       // Set default model if none selected or current model not available
       if (availableModels.length > 0) {
         if (!model || !availableModels.includes(model)) {
           setModel(availableModels[0]);
         }
+      } else if (isWindows) {
+        // On Windows, if we still don't have models, try multiple times with increasing delays
+        const retryAttempts = [1000, 3000, 5000]; // 1s, 3s, 5s delays
+        
+        for (let i = 0; i < retryAttempts.length; i++) {
+          setTimeout(async () => {
+            try {
+              const retryModels = await forceRefreshModels();
+              if (retryModels.length > 0) {
+                setModels(retryModels);
+                if (!model || !retryModels.includes(model)) {
+                  setModel(retryModels[0]);
+                }
+                // Success, exit early
+                return;
+              }
+            } catch (error) {
+              console.error(`Windows model retry ${i + 1} failed:`, error);
+            }
+          }, retryAttempts[i]);
+        }
       }
     } catch (error) {
       console.error('Error loading models:', error);
@@ -192,6 +218,16 @@ export default function Chat() {
     }
   }, [ollamaStatus.isRunning, model]);
 
+  // Event handler wrapper for refresh button clicks
+  const handleRefreshModels = useCallback(() => {
+    refreshModels(false);
+  }, [refreshModels]);
+
+  // Event handler wrapper for force refresh
+  const handleForceRefreshModels = useCallback(() => {
+    refreshModels(true);
+  }, [refreshModels]);
+
   useEffect(() => {
     // Check Ollama status first, then load models if available
     const checkStatus = async () => {
@@ -228,6 +264,28 @@ export default function Chat() {
     }
   }, [refreshModels]);
 
+  // Windows-specific: Periodic background model refresh
+  useEffect(() => {
+    const isWindows = typeof navigator !== 'undefined' && navigator.userAgent.includes('Windows');
+    
+    if (!isWindows || !ollamaStatus.isRunning) return;
+    
+    // Set up periodic refresh for Windows every 30 seconds
+    const interval = setInterval(async () => {
+      try {
+        // Only refresh if we don't have models or there's been a long time since last refresh
+        if (models.length === 0) {
+          console.log('Windows: Background model refresh - no models detected');
+          await refreshModels(true);
+        }
+      } catch (error) {
+        console.error('Windows background model refresh failed:', error);
+      }
+    }, 30000); // 30 seconds
+
+    return () => clearInterval(interval);
+  }, [ollamaStatus.isRunning, models.length, refreshModels]);
+
   const createNewConversation = () => {
     const newConversation: Conversation = {
       id: Date.now().toString(),
@@ -921,7 +979,7 @@ export default function Chat() {
                 <Button
                   variant="ghost"
                   size="sm"
-                  onClick={refreshModels}
+                  onClick={handleRefreshModels}
                   disabled={!ollamaStatus.isRunning || refreshingModels}
                   title="Refresh model list"
                 >
@@ -974,7 +1032,7 @@ export default function Chat() {
                     </Button>
                     <Button
                       variant="outline"
-                      onClick={refreshModels}
+                      onClick={handleRefreshModels}
                       disabled={refreshingModels}
                     >
                       {refreshingModels ? (