From aee382c84e010c561c9a31bb35441a9d3c1dc5f6 Mon Sep 17 00:00:00 2001
From: nai-degen <khoners@protonmail.com>
Date: Tue, 11 Jul 2023 09:14:46 -0500
Subject: [PATCH] adds claude-2 to supported models

---
 src/key-management/anthropic/provider.ts | 2 ++
 src/proxy/anthropic.ts                   | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/src/key-management/anthropic/provider.ts b/src/key-management/anthropic/provider.ts
index 28e8617..e177835 100644
--- a/src/key-management/anthropic/provider.ts
+++ b/src/key-management/anthropic/provider.ts
@@ -3,11 +3,13 @@ import { Key, KeyProvider } from "..";
 import { config } from "../../config";
 import { logger } from "../../logger";
 
+// https://docs.anthropic.com/claude/reference/selecting-a-model
 export const ANTHROPIC_SUPPORTED_MODELS = [
   "claude-instant-v1",
   "claude-instant-v1-100k",
   "claude-v1",
   "claude-v1-100k",
+  "claude-2",
 ] as const;
 export type AnthropicModel = (typeof ANTHROPIC_SUPPORTED_MODELS)[number];
 
diff --git a/src/proxy/anthropic.ts b/src/proxy/anthropic.ts
index c4efd6b..109ef9e 100644
--- a/src/proxy/anthropic.ts
+++ b/src/proxy/anthropic.ts
@@ -43,6 +43,8 @@ const getModelsResponse = () => {
     "claude-instant-v1.1",
     "claude-instant-v1.1-100k",
     "claude-instant-v1.0",
+    "claude-2", // claude-2 is 100k by default it seems
+    "claude-2.0",
   ];
 
   const models = claudeVariants.map((id) => ({