From f742b9972ec768e5b5076debc747d01d3fd77b2e Mon Sep 17 00:00:00 2001
From: Danny Avila <danny@librechat.ai>
Date: Tue, 27 Aug 2024 09:07:04 -0400
Subject: [PATCH] =?UTF-8?q?=F0=9F=8F=B7=EF=B8=8F=20chore:=20Add=20Unoffici?=
 =?UTF-8?q?al=20Naming=20Variation=20for=20Claude-3.5-Sonnet=20(#3800)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/models/tx.js      | 2 ++
 api/models/tx.spec.js | 7 +++++++
 api/utils/tokens.js   | 1 +
 3 files changed, 10 insertions(+)

diff --git a/api/models/tx.js b/api/models/tx.js
index adbedd82c3..1b515cca2a 100644
--- a/api/models/tx.js
+++ b/api/models/tx.js
@@ -55,6 +55,7 @@ const tokenValues = Object.assign(
     'claude-3-opus': { prompt: 15, completion: 75 },
     'claude-3-sonnet': { prompt: 3, completion: 15 },
     'claude-3-5-sonnet': { prompt: 3, completion: 15 },
+    'claude-3.5-sonnet': { prompt: 3, completion: 15 },
     'claude-3-haiku': { prompt: 0.25, completion: 1.25 },
     'claude-2.1': { prompt: 8, completion: 24 },
     'claude-2': { prompt: 8, completion: 24 },
@@ -77,6 +78,7 @@ const tokenValues = Object.assign(
  * @type {Object.<string, {write: number, read: number }>}
  */
 const cacheTokenValues = {
+  'claude-3.5-sonnet': { write: 3.75, read: 0.3 },
   'claude-3-5-sonnet': { write: 3.75, read: 0.3 },
   'claude-3-haiku': { write: 0.3, read: 0.03 },
 };
diff --git a/api/models/tx.spec.js b/api/models/tx.spec.js
index db0251dbb6..f0a118c012 100644
--- a/api/models/tx.spec.js
+++ b/api/models/tx.spec.js
@@ -82,6 +82,13 @@ describe('getValueKey', () => {
     expect(getValueKey('claude-3-5-sonnet-turbo')).toBe('claude-3-5-sonnet');
     expect(getValueKey('claude-3-5-sonnet-0125')).toBe('claude-3-5-sonnet');
   });
+
+  it('should return "claude-3.5-sonnet" for model type of "claude-3.5-sonnet-"', () => {
+    expect(getValueKey('claude-3.5-sonnet-20240620')).toBe('claude-3.5-sonnet');
+    expect(getValueKey('anthropic/claude-3.5-sonnet')).toBe('claude-3.5-sonnet');
+    expect(getValueKey('claude-3.5-sonnet-turbo')).toBe('claude-3.5-sonnet');
+    expect(getValueKey('claude-3.5-sonnet-0125')).toBe('claude-3.5-sonnet');
+  });
 });
 
 describe('getMultiplier', () => {
diff --git a/api/utils/tokens.js b/api/utils/tokens.js
index 122c3c3f0a..f236fa8f4d 100644
--- a/api/utils/tokens.js
+++ b/api/utils/tokens.js
@@ -60,6 +60,7 @@ const anthropicModels = {
   'claude-3-sonnet': 200000,
   'claude-3-opus': 200000,
   'claude-3-5-sonnet': 200000,
+  'claude-3.5-sonnet': 200000,
 };
 
 const aggregateModels = { ...openAIModels, ...googleModels, ...anthropicModels, ...cohereModels };