From f3b876887e21775b9cf140dc2759252a8290faa1 Mon Sep 17 00:00:00 2001
From: nai-degen <khoners@protonmail.com>
Date: Wed, 23 Oct 2024 13:59:10 -0500
Subject: [PATCH] fixes issue with AWS model name reassignment

---
 src/proxy/aws-claude.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/proxy/aws-claude.ts b/src/proxy/aws-claude.ts
index 5eaedc6..85de7dc 100644
--- a/src/proxy/aws-claude.ts
+++ b/src/proxy/aws-claude.ts
@@ -187,7 +187,7 @@ function maybeReassignModel(req: Request) {
     throw new Error(`Provided model name (${model}) doesn't resemble a Claude model ID.`);
   }
 
-  const [_, _cl, instant, _v, major, _sep, minor, _ctx, name, rev] = match;
+  const [_, _cl, instant, _v, major, _sep, minor, _ctx, rawName, rev] = match;
 
   if (instant) {
     req.body.model = "anthropic.claude-instant-v1";
@@ -195,6 +195,8 @@ function maybeReassignModel(req: Request) {
   }
 
   const ver = minor ? `${major}.${minor}` : major;
+  const name = rawName.match(/([a-z]+)/)?.[1];
+
   switch (ver) {
     case "1":
     case "1.0":
@@ -220,6 +222,7 @@ function maybeReassignModel(req: Request) {
           return;
       }
       break;
+    case "3-5":
     case "3.5":
       switch (name) {
         case "sonnet":