dev: enable tool use for Claude

HeyPuter · Jan 24, 2025 · 1fde01a · 1fde01a
1 parent e9f6266
commit 1fde01a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/puter-js/src/modules/AI.js b/src/puter-js/src/modules/AI.js
@@ -237,6 +237,9 @@ class AI{
         if ( options.model === 'groq' ) {
             options.model = 'llama3-8b-8192';
         }
+        if ( options.model === 'deepseek' ) {
+            options.model = 'deepseek-chat';
+        }
 
         // map model to the appropriate driver
         if (!options.model || options.model === 'gpt-4o' || options.model === 'gpt-4o-mini') {
@@ -270,6 +273,12 @@ class AI{
         }else if(options.model === 'grok-beta') {
             driver = 'xai';
         }
+        else if(
+            options.model === 'deepseek-chat' ||
+            options.model === 'deepseek-reasoner'
+        ){
+            driver = 'deepseek';
+        }
 
         // stream flag from settings
         if(settings.stream !== undefined && typeof settings.stream === 'boolean'){