From c73226395872b5cc7351e487e6b9f064dcee123f Mon Sep 17 00:00:00 2001
From: jelveh <nj@puter.com>
Date: Tue, 15 Apr 2025 23:24:53 -0700
Subject: [PATCH] Update AI.js

---
 src/puter-js/src/modules/AI.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/puter-js/src/modules/AI.js b/src/puter-js/src/modules/AI.js
index e212eaf48..d347833f1 100644
--- a/src/puter-js/src/modules/AI.js
+++ b/src/puter-js/src/modules/AI.js
@@ -326,7 +326,7 @@ class AI{
         }
 
         // map model to the appropriate driver
-        if (!requestParams.model || requestParams.model === 'gpt-4o' || requestParams.model === 'gpt-4o-mini') {
+        if (!requestParams.model || requestParams.model.startsWith('gpt-')) {
             driver = 'openai-completion';
         }else if(
             requestParams.model === 'claude-3-haiku-20240307' ||