fixes issue with AWS model name reassignment

This commit is contained in:
nai-degen
2024-10-23 13:59:10 -05:00
parent 49c578f4dc
commit f3b876887e
+4 -1
View File
@@ -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":