diff --git a/src/puter-js/src/modules/AI.js b/src/puter-js/src/modules/AI.js
index cf42af34f..45f0c64c6 100644
--- a/src/puter-js/src/modules/AI.js
+++ b/src/puter-js/src/modules/AI.js
@@ -226,6 +226,12 @@ class AI{
         if (settings.model) {
             options.model = settings.model;
         }
+        if (settings.temperature) {
+            options.temperature = settings.temperature;
+        }
+        if (settings.max_tokens) {
+            options.max_tokens = settings.max_tokens;
+        }
 
         // convert to the correct model name if necessary
         if( options.model === 'claude-3-5-sonnet'){