diff --git a/src/backend/src/modules/puterai/ClaudeService.js b/src/backend/src/modules/puterai/ClaudeService.js
index 61e3ac29..b746af22 100644
--- a/src/backend/src/modules/puterai/ClaudeService.js
+++ b/src/backend/src/modules/puterai/ClaudeService.js
@@ -147,7 +147,9 @@ class ClaudeService extends BaseService {
         return [
             {
                 id: 'claude-3-5-sonnet-20241022',
+                name: 'Claude 3.5 Sonnet',
                 aliases: ['claude-3-5-sonnet-latest'],
+                context: 200000,
                 cost: {
                     currency: 'usd-cents',
                     tokens: 1_000_000,
@@ -161,6 +163,7 @@ class ClaudeService extends BaseService {
             {
                 id: 'claude-3-5-sonnet-20240620',
                 succeeded_by: 'claude-3-5-sonnet-20241022',
+                context: 200000, // might be wrong
                 cost: {
                     currency: 'usd-cents',
                     tokens: 1_000_000,
@@ -171,6 +174,7 @@ class ClaudeService extends BaseService {
             {
                 id: 'claude-3-haiku-20240307',
                 // aliases: ['claude-3-haiku-latest'],
+                context: 200000,
                 cost: {
                     currency: 'usd-cents',
                     tokens: 1_000_000,
diff --git a/src/backend/src/modules/puterai/XAIService.js b/src/backend/src/modules/puterai/XAIService.js
index 44f58426..aaaac0b2 100644
--- a/src/backend/src/modules/puterai/XAIService.js
+++ b/src/backend/src/modules/puterai/XAIService.js
@@ -142,12 +142,25 @@ class XAIService extends BaseService {
             {
                 id: 'grok-beta',
                 name: 'Grok Beta',
+                context: 131072,
                 cost: {
                     currency: 'usd-cents',
                     tokens: 1_000_000,
                     input: 500,
                     output: 1500,
                 },
+            },
+            {
+                id: 'grok-vision-beta',
+                name: 'Grok Vision Beta',
+                context: 8192,
+                cost: {
+                    currency: 'usd-cents',
+                    tokens: 1_000_000,
+                    input: 500,
+                    output: 1500,
+                    image: 1000,
+                },
             }
         ];
     }