fixes

nbonamy · May 16, 2024 · cc48e02 · cc48e02
1 parent 76ad2bf
commit cc48e02
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/screens/CommandDefaults.vue b/src/screens/CommandDefaults.vue
@@ -13,6 +13,7 @@
             <option value="ollama">Ollama</option>
             <option value="anthropic">Anthropic</option>
             <option value="mistralai">MistralAI</option>
+            <option value="google">Google</option>
             <option value="groq">Groq</option>
           </select>
         </div>

diff --git a/tests/unit/engine.test.ts b/tests/unit/engine.test.ts
@@ -118,7 +118,11 @@ test('Ignite Engine', async () => {
 
 test('Has Vision Models', async () => {
   expect(hasVisionModels('openai')).toBe(true)
-  expect(hasVisionModels('anthropic')).toBe(false)
+  expect(hasVisionModels('ollana')).toBe(true)
+  expect(hasVisionModels('mistralai')).toBe(false)
+  expect(hasVisionModels('anthropic')).toBe(true)
+  expect(hasVisionModels('google')).toBe(true)
+  expect(hasVisionModels('groq')).toBe(false)
 })
 
 test('Is Vision Model', async () => {