Update api/config.py

api/config.py

@@ -15,7 +15,9 @@ headers = {
     'sec-fetch-dest': 'empty',
     'sec-fetch-mode': 'cors',
     'sec-fetch-site': 'same-origin',
-    'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36',
+    'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) '
+                  'AppleWebKit/537.36 (KHTML, like Gecko) '
+                  'Chrome/129.0.0.0 Safari/537.36',
 }
 APP_SECRET = os.getenv("APP_SECRET")
 
@@ -45,6 +47,10 @@ ALLOWED_MODELS = [
     {"id": "XcodeAgent", "name": "XcodeAgent"},
     {"id": "AngularJSAgent", "name": "AngularJSAgent"},
     {"id": "RepoMap", "name": "RepoMap"},
+    {"id": "gemini-1.5-pro-latest", "name": "gemini-pro"},
+    {"id": "gemini-1.5-pro", "name": "gemini-1.5-pro"},
+    {"id": "claude-3-5-sonnet-20240620", "name": "claude-sonnet-3.5"},
+    {"id": "claude-3-5-sonnet", "name": "claude-sonnet-3.5"},
 ]
 
 MODEL_MAPPING = {
@@ -72,9 +78,14 @@ MODEL_MAPPING = {
     "XcodeAgent": "XcodeAgent",
     "AngularJSAgent": "AngularJSAgent",
     "RepoMap": "RepoMap",
+    # Additional mappings
     "gemini-flash": "gemini-1.5-flash",
     "claude-3.5-sonnet": "claude-sonnet-3.5",
     "flux": "ImageGeneration",
+    "gemini-1.5-pro-latest": "gemini-pro",
+    "gemini-1.5-pro": "gemini-1.5-pro",
+    "claude-3-5-sonnet-20240620": "claude-sonnet-3.5",
+    "claude-3-5-sonnet": "claude-sonnet-3.5",
 }
 
 # Agent modes