From 05ced4e625b7121909b5674113ee25a8de23c67f Mon Sep 17 00:00:00 2001
From: KernelDeimos <eric.alex.dube@gmail.com>
Date: Thu, 30 Jan 2025 12:18:26 -0500
Subject: [PATCH] Revert "dev: remove max_tokens from Claude"

This reverts commit 3bce60893e0557656551b48f7a7b64568ba2fdc5.
---
 src/backend/src/modules/puterai/ClaudeService.js | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/backend/src/modules/puterai/ClaudeService.js b/src/backend/src/modules/puterai/ClaudeService.js
index 4b11ffedc..1721510ef 100644
--- a/src/backend/src/modules/puterai/ClaudeService.js
+++ b/src/backend/src/modules/puterai/ClaudeService.js
@@ -251,6 +251,7 @@ class ClaudeService extends BaseService {
 
                 const msg = await this.anthropic.messages.create({
                     model: model ?? this.get_default_model(),
+                    max_tokens: (model === 'claude-3-5-sonnet-20241022' || model === 'claude-3-5-sonnet-20240620') ? 8192 : 4096,
                     temperature: 0,
                     system: PUTER_PROMPT + JSON.stringify(system_prompts),
                     messages: adapted_messages,