Update api/config.py

api/config.py

@@ -60,37 +60,37 @@ ALLOWED_MODELS = [
     {"id": "gemini-pro", "name": "gemini-pro"},
     {"id": "gemini-1.5-flash", "name": "gemini-1.5-flash"},
     {"id": "claude-sonnet-3.5", "name": "claude-sonnet-3.5"},
-    {"id": "PythonAgent", "name": "Python Agent"},
-    {"id": "JavaAgent", "name": "Java Agent"},
-    {"id": "JavaScriptAgent", "name": "JavaScript Agent"},
-    {"id": "HTMLAgent", "name": "HTML Agent"},
-    {"id": "GoogleCloudAgent", "name": "Google Cloud Agent"},
-    {"id": "AndroidDeveloper", "name": "Android Developer"},
-    {"id": "SwiftDeveloper", "name": "Swift Developer"},
-    {"id": "Next.jsAgent", "name": "Next.js Agent"},
-    {"id": "MongoDBAgent", "name": "MongoDB Agent"},
-    {"id": "PyTorchAgent", "name": "PyTorch Agent"},
-    {"id": "ReactAgent", "name": "React Agent"},
-    {"id": "XcodeAgent", "name": "Xcode Agent"},
-    {"id": "AngularJSAgent", "name": "AngularJS Agent"},
+    {"id": "Python Agent", "name": "Python Agent"},
+    {"id": "Java Agent", "name": "Java Agent"},
+    {"id": "JavaScript Agent", "name": "JavaScript Agent"},
+    {"id": "HTML Agent", "name": "HTML Agent"},
+    {"id": "Google Cloud Agent", "name": "Google Cloud Agent"},
+    {"id": "Android Developer", "name": "Android Developer"},
+    {"id": "Swift Developer", "name": "Swift Developer"},
+    {"id": "Next.js Agent", "name": "Next.js Agent"},
+    {"id": "MongoDB Agent", "name": "MongoDB Agent"},
+    {"id": "PyTorch Agent", "name": "PyTorch Agent"},
+    {"id": "React Agent", "name": "React Agent"},
+    {"id": "Xcode Agent", "name": "Xcode Agent"},
+    {"id": "AngularJS Agent", "name": "AngularJS Agent"},
     {"id": "repomap", "name": "repomap"},
-    {"id": "HerokuAgent", "name": "Heroku Agent"},
-    {"id": "GodotAgent", "name": "Godot Agent"},
-    {"id": "GoAgent", "name": "Go Agent"},
-    {"id": "GitlabAgent", "name": "Gitlab Agent"},
-    {"id": "GitAgent", "name": "Git Agent"},
-    {"id": "FlaskAgent", "name": "Flask Agent"},
-    {"id": "FirebaseAgent", "name": "Firebase Agent"},
-    {"id": "FastAPIAgent", "name": "FastAPI Agent"},
-    {"id": "ErlangAgent", "name": "Erlang Agent"},
-    {"id": "ElectronAgent", "name": "Electron Agent"},
-    {"id": "DockerAgent", "name": "Docker Agent"},
-    {"id": "DigitalOceanAgent", "name": "DigitalOcean Agent"},
-    {"id": "BitbucketAgent", "name": "Bitbucket Agent"},
-    {"id": "AzureAgent", "name": "Azure Agent"},
-    {"id": "FlutterAgent", "name": "Flutter Agent"},
-    {"id": "YoutubeAgent", "name": "Youtube Agent"},
-    {"id": "builderAgent", "name": "builder Agent"},
+    {"id": "Heroku Agent", "name": "Heroku Agent"},
+    {"id": "Godot Agent", "name": "Godot Agent"},
+    {"id": "Go Agent", "name": "Go Agent"},
+    {"id": "Gitlab Agent", "name": "Gitlab Agent"},
+    {"id": "Git Agent", "name": "Git Agent"},
+    {"id": "Flask Agent", "name": "Flask Agent"},
+    {"id": "Firebase Agent", "name": "Firebase Agent"},
+    {"id": "FastAPI Agent", "name": "FastAPI Agent"},
+    {"id": "Erlang Agent", "name": "Erlang Agent"},
+    {"id": "Electron Agent", "name": "Electron Agent"},
+    {"id": "Docker Agent", "name": "Docker Agent"},
+    {"id": "DigitalOcean Agent", "name": "DigitalOcean Agent"},
+    {"id": "Bitbucket Agent", "name": "Bitbucket Agent"},
+    {"id": "Azure Agent", "name": "Azure Agent"},
+    {"id": "Flutter Agent", "name": "Flutter Agent"},
+    {"id": "Youtube Agent", "name": "Youtube Agent"},
+    {"id": "builder Agent", "name": "builder Agent"},
 ]
 
 # Model Aliases
@@ -152,11 +152,9 @@ MODEL_PREFIXES = {
     if mode not in ["gemini-1.5-flash", "llama-3.1-8b", "llama-3.1-70b", "llama-3.1-405b", "repomap"]
 }
 
-# Consolidate all models
+# Consolidate all model IDs
 MODELS = [model["id"] for model in ALLOWED_MODELS]
 
-# Remove MODEL_REFERERS as per your request
-
 # Generate a random ID
 def generate_id(length=7):
     characters = string.ascii_letters + string.digits