From ea0e6884b0471970f3a82b3121156e813ad6534f Mon Sep 17 00:00:00 2001
From: Stefan Streichsbier <stefan@streichsbier.at>
Date: Tue, 12 Mar 2024 08:20:57 +0800
Subject: [PATCH] Add supported Claude models

---
 installer/client/cli/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/installer/client/cli/utils.py b/installer/client/cli/utils.py
index c78ef46..cb76ccd 100644
--- a/installer/client/cli/utils.py
+++ b/installer/client/cli/utils.py
@@ -251,7 +251,7 @@ class Standalone:
     def fetch_available_models(self):
         gptlist = []
         fullOllamaList = []
-        claudeList = ['claude-3-opus-20240229']
+        claudeList = ['claude-3-opus-20240229', 'claude-3-sonnet-20240229', 'claude-2.1']
         try:
             headers = {
                 "Authorization": f"Bearer {self.client.api_key}"