chore: add gemini and claude to UI

Author: Reterics

api_server.py

@@ -578,6 +578,14 @@ def get_user_settings(user_id: str, _: str = Depends(verify_api_key)):
             "gpt-4",
             "gpt-4-turbo",
             "gpt-4o"
+        ],
+        "gemini": [
+            "gemini-1.5-flash",
+            "gemini-1.5-pro"
+        ],
+        "claude": [
+            "claude-3-opus-20240229",
+            "claude-3-5-sonnet-20240620"
         ]
     }
     return {"settings": current_settings, "llm_options": llm_options}

engine/backends/ollama.py

@@ -174,6 +174,8 @@ def stream_generator():
         resp = requests.post("http://localhost:11434/api/generate", json={"model": model, "prompt": text, "stream": False})
         if resp.status_code == 404:
             models_list = get_downloaded_models()
+            if not models_list:
+                return "[404] No models are available. Please download a model using Ollama CLI (e.g., 'ollama pull mistral') or configure a different AI API."
             models_str = "\n".join(f"- {m}" for m in models_list)
             return f"[404] Model '{model}' not available. Choose one of the downloaded models:\n{models_str}"
 

tauri-ui/src/components/Settings.tsx

@@ -263,10 +263,12 @@ export function Settings({ onClose }: SettingsProps) {
                   >
                     <option value="ollama">Ollama</option>
                     <option value="openai">OpenAI</option>
+                    <option value="gemini">Gemini</option>
+                    <option value="claude">Claude</option>
                   </select>
                 </div>
 
-                {settingsForm.settings.llm_backend === 'ollama' ? (
+                {settingsForm.settings.llm_backend === 'ollama' && (
                   <div>
                     <label
                       htmlFor="ollama-model"
@@ -289,7 +291,9 @@ export function Settings({ onClose }: SettingsProps) {
                       ))}
                     </select>
                   </div>
-                ) : (
+                )}
+
+                {settingsForm.settings.llm_backend === 'openai' && (
                   <>
                     <div>
                       <label
@@ -335,6 +339,98 @@ export function Settings({ onClose }: SettingsProps) {
                   </>
                 )}
 
+                {settingsForm.settings.llm_backend === 'gemini' && (
+                  <>
+                    <div>
+                      <label
+                        htmlFor="gemini-model"
+                        className="block text-sm font-medium text-gray-700 mb-1"
+                      >
+                        Gemini Model
+                      </label>
+                      <select
+                        id="gemini-model"
+                        value={settingsForm.settings.gemini_model || ''}
+                        onChange={e =>
+                          handleChange('gemini_model', e.currentTarget.value)
+                        }
+                        className="w-full border border-gray-300 rounded-md px-3 py-2"
+                      >
+                        {(settings.llm_options.gemini || []).map(model => (
+                          <option key={model} value={model}>
+                            {model}
+                          </option>
+                        ))}
+                      </select>
+                    </div>
+
+                    <div>
+                      <label
+                        htmlFor="gemini-api-key"
+                        className="block text-sm font-medium text-gray-700 mb-1"
+                      >
+                        Gemini API Key
+                      </label>
+                      <input
+                        id="gemini-api-key"
+                        type="password"
+                        value={settingsForm.settings.gemini_api_key || ''}
+                        onChange={e =>
+                          handleChange('gemini_api_key', e.currentTarget.value)
+                        }
+                        className="w-full border border-gray-300 rounded-md px-3 py-2"
+                        placeholder="AIza..."
+                      />
+                    </div>
+                  </>
+                )}
+
+                {settingsForm.settings.llm_backend === 'claude' && (
+                  <>
+                    <div>
+                      <label
+                        htmlFor="claude-model"
+                        className="block text-sm font-medium text-gray-700 mb-1"
+                      >
+                        Claude Model
+                      </label>
+                      <select
+                        id="claude-model"
+                        value={settingsForm.settings.claude_model || ''}
+                        onChange={e =>
+                          handleChange('claude_model', e.currentTarget.value)
+                        }
+                        className="w-full border border-gray-300 rounded-md px-3 py-2"
+                      >
+                        {(settings.llm_options.claude || []).map(model => (
+                          <option key={model} value={model}>
+                            {model}
+                          </option>
+                        ))}
+                      </select>
+                    </div>
+
+                    <div>
+                      <label
+                        htmlFor="claude-api-key"
+                        className="block text-sm font-medium text-gray-700 mb-1"
+                      >
+                        Claude API Key
+                      </label>
+                      <input
+                        id="claude-api-key"
+                        type="password"
+                        value={settingsForm.settings.claude_api_key || ''}
+                        onChange={e =>
+                          handleChange('claude_api_key', e.currentTarget.value)
+                        }
+                        className="w-full border border-gray-300 rounded-md px-3 py-2"
+                        placeholder="sk-ant-..."
+                      />
+                    </div>
+                  </>
+                )}
+
                 <div className="flex items-center gap-2">
                   <input
                     type="checkbox"

tauri-ui/src/context/ChatContext.tsx

@@ -19,10 +19,14 @@ export type SettingsType = {
   llm_backend: string;
   llm_model: string;
   openai_model: string;
+  gemini_model?: string;
+  claude_model?: string;
   voice: boolean;
   streaming: boolean;
   secured_streaming: boolean;
-  openai_api_key: string;
+  openai_api_key?: string;
+  gemini_api_key?: string;
+  claude_api_key?: string;
 }
 
 export type AppSettings = {
@@ -71,6 +75,8 @@ export interface UserPrivacy {
 export interface LLMOptions {
   ollama: string[];
   openai: string[];
+  gemini?: string[];
+  claude?: string[];
 }
 
 export interface UserSession {
@@ -311,18 +317,19 @@ export function ChatProvider({
 
   const sendMessage = async (
     input: string,
-    backend = 'ollama',
+    backend?: string,
     mode?: string,
     project_path?: string
   ) => {
+    const effectiveBackend = backend ?? settings?.settings.llm_backend ?? 'ollama';
     const data = await fetchWithKey(`${BASE_URL}/query`, apiKey, setError, {
       method: 'POST',
       headers: {
         'Content-Type': 'application/json',
       },
       body: JSON.stringify({
         input,
-        backend,
+        backend: effectiveBackend,
         conversation_id: conversationId,
         mode,
         project_path
@@ -337,10 +344,11 @@ export function ChatProvider({
   const sendStreamingMessage = async (
     input: string,
     onToken: (token: string) => void,
-    backend = 'ollama',
+    backend?: string,
     mode?: string,
     project_path?: string
   ) => {
+    const effectiveBackend = backend ?? settings?.settings.llm_backend ?? 'ollama';
     const res = await fetch(`${BASE_URL}/query/stream`, {
       method: 'POST',
       headers: {
@@ -349,7 +357,7 @@ export function ChatProvider({
       },
       body: JSON.stringify({
         input,
-        backend,
+        backend: effectiveBackend,
         conversation_id: conversationId,
         mode,
         project_path
@@ -370,21 +378,22 @@ export function ChatProvider({
   const sendSecuredStreamingMessage = (
     input: string,
     onToken: (token: string) => void,
-    backend = 'ollama',
+    backend?: string,
     mode?: string,
     project_path?: string
   )=> {
     if (messages.length === 0) {
       addTemporaryConversation(input)
     }
+    const effectiveBackend = backend ?? settings?.settings.llm_backend ?? 'ollama';
     return consumeEncryptedStream(
       `${BASE_URL}/query/secure_stream`,
       apiKey,
       conversationId,
       onToken,
       JSON.stringify({
         input,
-        backend,
+        backend: effectiveBackend,
         conversation_id: conversationId,
         mode,
         project_path