fix: previous fix

HeyPuter · Nov 22, 2024 · ee7bedd · ee7bedd
1 parent 56710e1
commit ee7bedd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/backend/src/modules/puterai/XAIService.js b/src/backend/src/modules/puterai/XAIService.js
@@ -21,14 +21,18 @@ class XAIService extends BaseService {
     get_system_prompt () {
         return PUTER_PROMPT;
     }
+
+    adapt_model (model) {
+        return model;
+    }
 
     async _init () {
         this.anthropic = new Anthropic({
             apiKey: this.global_config.services.xai.apiKey,
             baseURL: 'https://api.x.ai'
         });
     }
-    
+
     static IMPLEMENTS = {
         ['puter-chat-completion']: {
             async list () {
@@ -37,6 +41,7 @@ class XAIService extends BaseService {
                 ];
             },
             async complete ({ messages, stream, model }) {
+                model = this.adapt_model(model);
                 const adapted_messages = [];
 
                 const system_prompts = [];