Update api/config.py
Browse files- api/config.py +22 -18
api/config.py
CHANGED
|
@@ -1,3 +1,5 @@
|
|
|
|
|
|
|
|
| 1 |
import os
|
| 2 |
from dotenv import load_dotenv
|
| 3 |
|
|
@@ -95,36 +97,38 @@ MODEL_MAPPING = {
|
|
| 95 |
# Agent modes configuration
|
| 96 |
AGENT_MODE = {
|
| 97 |
'ImageGeneration': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "Image Generation"},
|
|
|
|
| 98 |
}
|
| 99 |
|
|
|
|
| 100 |
TRENDING_AGENT_MODE = {
|
| 101 |
"blackboxai": {},
|
| 102 |
"gemini-1.5-flash": {'mode': True, 'id': 'Gemini'},
|
| 103 |
"llama-3.1-8b": {'mode': True, 'id': "llama-3.1-8b"},
|
| 104 |
-
|
| 105 |
-
|
| 106 |
-
|
| 107 |
-
|
| 108 |
-
|
| 109 |
-
|
| 110 |
-
|
| 111 |
-
|
| 112 |
-
|
| 113 |
-
|
| 114 |
-
|
| 115 |
-
|
| 116 |
-
|
| 117 |
-
|
| 118 |
-
|
| 119 |
-
|
| 120 |
-
|
| 121 |
}
|
| 122 |
|
| 123 |
# User selected models mapping
|
| 124 |
USER_SELECTED_MODEL = {
|
| 125 |
"gpt-4o": "gpt-4o",
|
| 126 |
"gemini-pro": "gemini-pro",
|
| 127 |
-
|
| 128 |
}
|
| 129 |
|
| 130 |
# Model prefixes for prompt formatting
|
|
|
|
| 1 |
+
# api/config.py
|
| 2 |
+
|
| 3 |
import os
|
| 4 |
from dotenv import load_dotenv
|
| 5 |
|
|
|
|
| 97 |
# Agent modes configuration
|
| 98 |
AGENT_MODE = {
|
| 99 |
'ImageGeneration': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "Image Generation"},
|
| 100 |
+
# Add other agent-specific modes if necessary
|
| 101 |
}
|
| 102 |
|
| 103 |
+
# Trending agent modes configuration
|
| 104 |
TRENDING_AGENT_MODE = {
|
| 105 |
"blackboxai": {},
|
| 106 |
"gemini-1.5-flash": {'mode': True, 'id': 'Gemini'},
|
| 107 |
"llama-3.1-8b": {'mode': True, 'id': "llama-3.1-8b"},
|
| 108 |
+
"llama-3.1-70b": {'mode': True, 'id': "llama-3.1-70b"},
|
| 109 |
+
"llama-3.1-405b": {'mode': True, 'id': "llama-3.1-405b"},
|
| 110 |
+
"blackboxai-pro": {'mode': True, 'id': "BLACKBOXAI-PRO"},
|
| 111 |
+
"PythonAgent": {'mode': True, 'id': "Python Agent"},
|
| 112 |
+
"JavaAgent": {'mode': True, 'id': "Java Agent"},
|
| 113 |
+
"JavaScriptAgent": {'mode': True, 'id': "JavaScript Agent"},
|
| 114 |
+
"HTMLAgent": {'mode': True, 'id': "HTML Agent"},
|
| 115 |
+
"GoogleCloudAgent": {'mode': True, 'id': "Google Cloud Agent"},
|
| 116 |
+
"AndroidDeveloper": {'mode': True, 'id': "Android Developer"},
|
| 117 |
+
"SwiftDeveloper": {'mode': True, 'id': "Swift Developer"},
|
| 118 |
+
"Next.jsAgent": {'mode': True, 'id': "Next.js Agent"},
|
| 119 |
+
"MongoDBAgent": {'mode': True, 'id': "MongoDB Agent"},
|
| 120 |
+
"PyTorchAgent": {'mode': True, 'id': "PyTorch Agent"},
|
| 121 |
+
"ReactAgent": {'mode': True, 'id': "React Agent"},
|
| 122 |
+
"XcodeAgent": {'mode': True, 'id': "Xcode Agent"},
|
| 123 |
+
"AngularJSAgent": {'mode': True, 'id': "AngularJS Agent"},
|
| 124 |
+
"RepoMap": {'mode': True, 'id': "repomap"},
|
| 125 |
}
|
| 126 |
|
| 127 |
# User selected models mapping
|
| 128 |
USER_SELECTED_MODEL = {
|
| 129 |
"gpt-4o": "gpt-4o",
|
| 130 |
"gemini-pro": "gemini-pro",
|
| 131 |
+
"claude-sonnet-3.5": "claude-sonnet-3.5",
|
| 132 |
}
|
| 133 |
|
| 134 |
# Model prefixes for prompt formatting
|