From 8828d03f86005fd32ec46219f84f6a277e88f39e Mon Sep 17 00:00:00 2001
From: KernelDeimos <eric.alex.dube@gmail.com>
Date: Fri, 14 Feb 2025 12:15:10 -0500
Subject: [PATCH] dev: add gemini to puter.js

---
 src/puter-js/src/modules/AI.js | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/puter-js/src/modules/AI.js b/src/puter-js/src/modules/AI.js
index 93afb4be1..ee63e252f 100644
--- a/src/puter-js/src/modules/AI.js
+++ b/src/puter-js/src/modules/AI.js
@@ -279,6 +279,12 @@ class AI{
         ){
             driver = 'deepseek';
         }
+        else if(
+            options.model === 'gemini-1.5-flash' ||
+            options.model === 'gemini-2.0-flash'
+        ){
+            driver = 'gemini';
+        }
 
         // stream flag from settings
         if(settings.stream !== undefined && typeof settings.stream === 'boolean'){