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'){