Update api/config.py
Browse files- api/config.py +57 -0
api/config.py
CHANGED
|
@@ -60,6 +60,63 @@ ALLOWED_MODELS = [
|
|
| 60 |
{"id": "builderAgent", "name": "builderAgent"},
|
| 61 |
]
|
| 62 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
# Agent modes
|
| 64 |
AGENT_MODE = {
|
| 65 |
'flux': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "flux"},
|
|
|
|
| 60 |
{"id": "builderAgent", "name": "builderAgent"},
|
| 61 |
]
|
| 62 |
|
| 63 |
+
MODEL_MAPPING = {
|
| 64 |
+
"blackboxai": "blackboxai",
|
| 65 |
+
"blackboxai-pro": "blackboxai-pro",
|
| 66 |
+
"flux": "flux",
|
| 67 |
+
"ImageGeneration": "flux",
|
| 68 |
+
"llama-3.1-8b": "llama-3.1-8b",
|
| 69 |
+
"llama-3.1-70b": "llama-3.1-70b",
|
| 70 |
+
"llama-3.1-405b": "llama-3.1-405",
|
| 71 |
+
"gpt-4o": "gpt-4o",
|
| 72 |
+
"gemini-pro": "gemini-pro",
|
| 73 |
+
"gemini-1.5-flash": "gemini-1.5-flash",
|
| 74 |
+
"claude-sonnet-3.5": "claude-sonnet-3.5",
|
| 75 |
+
"PythonAgent": "PythonAgent",
|
| 76 |
+
"JavaAgent": "JavaAgent",
|
| 77 |
+
"JavaScriptAgent": "JavaScriptAgent",
|
| 78 |
+
"HTMLAgent": "HTMLAgent",
|
| 79 |
+
"GoogleCloudAgent": "GoogleCloudAgent",
|
| 80 |
+
"AndroidDeveloper": "AndroidDeveloper",
|
| 81 |
+
"SwiftDeveloper": "SwiftDeveloper",
|
| 82 |
+
"Next.jsAgent": "Next.jsAgent",
|
| 83 |
+
"MongoDBAgent": "MongoDBAgent",
|
| 84 |
+
"PyTorchAgent": "PyTorchAgent",
|
| 85 |
+
"ReactAgent": "ReactAgent",
|
| 86 |
+
"XcodeAgent": "XcodeAgent",
|
| 87 |
+
"AngularJSAgent": "AngularJSAgent",
|
| 88 |
+
"HerokuAgent": "HerokuAgent",
|
| 89 |
+
"GodotAgent": "GodotAgent",
|
| 90 |
+
"GoAgent": "GoAgent",
|
| 91 |
+
"GitlabAgent": "GitlabAgent",
|
| 92 |
+
"GitAgent": "GitAgent",
|
| 93 |
+
"RepoMap": "RepoMap",
|
| 94 |
+
# Additional mappings
|
| 95 |
+
"gemini-flash": "gemini-1.5-flash",
|
| 96 |
+
"claude-3.5-sonnet": "claude-sonnet-3.5",
|
| 97 |
+
"gemini-1.5-pro-latest": "gemini-pro",
|
| 98 |
+
"gemini-1.5-pro": "gemini-1.5-pro",
|
| 99 |
+
"claude-3-5-sonnet-20240620": "claude-sonnet-3.5",
|
| 100 |
+
"claude-3-5-sonnet": "claude-sonnet-3.5",
|
| 101 |
+
"Niansuh": "Niansuh",
|
| 102 |
+
"o1-preview": "o1-preview",
|
| 103 |
+
"claude-3-5-sonnet-20241022": "claude-3-5-sonnet-20241022",
|
| 104 |
+
"claude-3-5-sonnet-x": "claude-3-5-sonnet-x",
|
| 105 |
+
# Added New Agents
|
| 106 |
+
"FlaskAgent": "FlaskAgent",
|
| 107 |
+
"FirebaseAgent": "FirebaseAgent",
|
| 108 |
+
"FastAPIAgent": "FastAPIAgent",
|
| 109 |
+
"ErlangAgent": "ErlangAgent",
|
| 110 |
+
"ElectronAgent": "ElectronAgent",
|
| 111 |
+
"DockerAgent": "DockerAgent",
|
| 112 |
+
"DigitalOceanAgent": "DigitalOceanAgent",
|
| 113 |
+
"BitbucketAgent": "BitbucketAgent",
|
| 114 |
+
"AzureAgent": "AzureAgent",
|
| 115 |
+
"FlutterAgent": "FlutterAgent",
|
| 116 |
+
"YoutubeAgent": "YoutubeAgent",
|
| 117 |
+
"builderAgent": "builderAgent",
|
| 118 |
+
}
|
| 119 |
+
|
| 120 |
# Agent modes
|
| 121 |
AGENT_MODE = {
|
| 122 |
'flux': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "flux"},
|