fix: Default model is missing in asking (#149)

Author: szl97

src/app/[owner]/[repo]/page.tsx

@@ -1999,6 +1999,7 @@ IMPORTANT:
         excludedFiles={modelExcludedFiles}
         setExcludedFiles={setModelExcludedFiles}
         onApply={confirmRefresh}
+        showWikiType={true}
       />
     </div>
   );

src/components/Ask.tsx

@@ -1,6 +1,6 @@
 'use client';
 
-import React, { useState, useRef, useEffect } from 'react';
+import React, {useState, useRef, useEffect} from 'react';
 import {FaChevronLeft, FaChevronRight } from 'react-icons/fa';
 import Markdown from './Markdown';
 import { useLanguage } from '@/contexts/LanguageContext';
@@ -9,6 +9,18 @@ import getRepoUrl from '@/utils/getRepoUrl';
 import ModelSelectionModal from './ModelSelectionModal';
 import { createChatWebSocket, closeWebSocket, ChatCompletionRequest } from '@/utils/websocketClient';
 
+interface Model {
+  id: string;
+  name: string;
+}
+
+interface Provider {
+  id: string;
+  name: string;
+  models: Model[];
+  supportsCustomModel?: boolean;
+}
+
 interface Message {
   role: 'user' | 'assistant' | 'system';
   content: string;
@@ -93,6 +105,38 @@ const Ask: React.FC<AskProps> = ({
     };
   }, []);
 
+  useEffect(() => {
+    const fetchDefaultModel = async () => {
+      try {
+        setIsLoading(true);
+
+        const response = await fetch('/api/models/config');
+
+        if (!response.ok) {
+          throw new Error(`Error fetching model configurations: ${response.status}`);
+        }
+
+        const data = await response.json();
+
+        setSelectedProvider(data.defaultProvider);
+
+        // Find the default provider and set its default model
+        const selectedProvider = data.providers.find((p:Provider) => p.id === data.defaultProvider);
+        if (selectedProvider && selectedProvider.models.length > 0) {
+          setSelectedModel(selectedProvider.models[0].id);
+        }
+      } catch (err) {
+        console.error('Failed to fetch model configurations:', err);
+      } finally {
+        setIsLoading(false);
+      }
+    };
+
+    if(provider == '' || model == '') {
+      fetchDefaultModel()
+    }
+  }, [provider, model]);
+
   const clearConversation = () => {
     setQuestion('');
     setResponse('');
@@ -824,6 +868,7 @@ const Ask: React.FC<AskProps> = ({
         onApply={() => {
           console.log('Model selection applied:', selectedProvider, selectedModel);
         }}
+        showWikiType={false}
       />
     </div>
   );

src/components/ModelSelectionModal.tsx

@@ -32,6 +32,7 @@ interface ModelSelectionModalProps {
   includedFiles?: string;
   setIncludedFiles?: (value: string) => void;
   showFileFilters?: boolean;
+  showWikiType: boolean;
 }
 
 export default function ModelSelectionModal({
@@ -56,7 +57,8 @@ export default function ModelSelectionModal({
   setIncludedDirs,
   includedFiles = '',
   setIncludedFiles,
-  showFileFilters = false
+  showFileFilters = false,
+  showWikiType = true,
 }: ModelSelectionModalProps) {
   const { messages: t } = useLanguage();
 
@@ -126,10 +128,12 @@ export default function ModelSelectionModal({
           {/* Modal body */}
           <div className="p-6">
             {/* Wiki Type Selector */}
-            <WikiTypeSelector
-              isComprehensiveView={localIsComprehensiveView}
-              setIsComprehensiveView={setLocalIsComprehensiveView}
-            />
+            {
+              showWikiType && <WikiTypeSelector
+                    isComprehensiveView={localIsComprehensiveView}
+                    setIsComprehensiveView={setLocalIsComprehensiveView}
+                />
+            }
 
             {/* Divider */}
             <div className="my-4 border-t border-[var(--border-color)]/30"></div>