From cc48e02957212c9572a80bd87bc16d3202c62cd1 Mon Sep 17 00:00:00 2001
From: Nicolas Bonamy <nicolas@Bonamy.fr>
Date: Thu, 16 May 2024 13:12:30 -0500
Subject: [PATCH] fixes

---
 src/screens/CommandDefaults.vue | 1 +
 tests/unit/engine.test.ts       | 6 +++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/screens/CommandDefaults.vue b/src/screens/CommandDefaults.vue
index 23c0199..eb24b2d 100644
--- 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
index bbf1d96..03b0395 100644
--- 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 () => {