Update api/config.py

api/config.py

@@ -1,3 +1,5 @@
+# api/config.py
+
 import os
 from dotenv import load_dotenv
 
@@ -55,6 +57,7 @@ ALLOWED_MODELS = [
 MODEL_MAPPING = {
     "blackboxai": "blackboxai",
     "blackboxai-pro": "blackboxai-pro",
+    "flux": "flux",
     "ImageGeneration": "flux",
     "llama-3.1-8b": "llama-3.1-8b",
     "llama-3.1-70b": "llama-3.1-70b",
@@ -80,6 +83,7 @@ MODEL_MAPPING = {
     # Additional mappings
     "gemini-flash": "gemini-1.5-flash",
     "claude-3.5-sonnet": "claude-sonnet-3.5",
+    "flux": "flux",
     "gemini-1.5-pro-latest": "gemini-pro",
     "gemini-1.5-pro": "gemini-1.5-pro",
     "claude-3-5-sonnet-20240620": "claude-sonnet-3.5",
@@ -114,6 +118,7 @@ TRENDING_AGENT_MODE = {
     'RepoMap': {'mode': True, 'id': "repomap"},
 }
 
+# Model prefixes
 MODEL_PREFIXES = {
     "blackboxai": "@blackboxai",
     "blackboxai-pro": "@blackboxai-pro",
@@ -139,4 +144,34 @@ MODEL_PREFIXES = {
     "XcodeAgent": "@XcodeAgent",
     "AngularJSAgent": "@AngularJSAgent",
     "RepoMap": "@RepoMap",
-}
+    # Add any additional prefixes if necessary
+}
+
+# Model referers
+MODEL_REFERERS = {
+    "blackboxai": "/?model=blackboxai",
+    "blackboxai-pro": "/?model=blackboxai-pro",
+    "flux": "/?model=flux",
+    "llama-3.1-8b": "/?model=llama-3.1-8b",
+    "llama-3.1-70b": "/?model=llama-3.1-70b",
+    "llama-3.1-405b": "/?model=llama-3.1-405b",
+    "gpt-4o": "/?model=gpt-4o",
+    "gemini-pro": "/?model=gemini-pro",
+    "gemini-1.5-flash": "/?model=gemini-1.5-flash",
+    "claude-sonnet-3.5": "/?model=claude-sonnet-3.5",
+    "PythonAgent": "/?model=PythonAgent",
+    "JavaAgent": "/?model=JavaAgent",
+    "JavaScriptAgent": "/?model=JavaScriptAgent",
+    "HTMLAgent": "/?model=HTMLAgent",
+    "GoogleCloudAgent": "/?model=GoogleCloudAgent",
+    "AndroidDeveloper": "/?model=AndroidDeveloper",
+    "SwiftDeveloper": "/?model=SwiftDeveloper",
+    "Next.jsAgent": "/?model=Next.jsAgent",
+    "MongoDBAgent": "/?model=MongoDBAgent",
+    "PyTorchAgent": "/?model=PyTorchAgent",
+    "ReactAgent": "/?model=ReactAgent",
+    "XcodeAgent": "/?model=XcodeAgent",
+    "AngularJSAgent": "/?model=AngularJSAgent",
+    "RepoMap": "/?model=RepoMap",
+    # Add any additional referers if necessary
+}