1 files changed, 17 insertions, 3 deletions

diff --git a/g4f/Provider/Blackbox.py b/g4f/Provider/Blackbox.py
index 3ba7abf4..fd84875e 100644
--- a/g4f/Provider/Blackbox.py
+++ b/g4f/Provider/Blackbox.py
@@ -14,7 +14,13 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
     url = "https://www.blackbox.ai"
     working = True
     default_model = 'blackbox'
-
+    models = [
+        default_model,
+        "gemini-1.5-flash",
+        "llama-3.1-8b",
+        'llama-3.1-70b',
+        'llama-3.1-405b',
+    ]
     @classmethod
     async def create_async_generator(
         cls,
@@ -28,7 +34,8 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
         if image is not None:
             messages[-1]["data"] = {
                 "fileText": image_name,
-                "imageBase64": to_data_uri(image)
+                "imageBase64": to_data_uri(image),
+                "title": str(uuid.uuid4())
             }
 
         headers = {
@@ -48,7 +55,13 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
         async with ClientSession(headers=headers) as session:
             random_id = secrets.token_hex(16)
             random_user_id = str(uuid.uuid4())
-
+            model_id_map = {
+                "blackbox": {},
+                "gemini-1.5-flash": {'mode': True, 'id': 'Gemini'},
+                "llama-3.1-8b": {'mode': True, 'id': "llama-3.1-8b"},
+                'llama-3.1-70b': {'mode': True, 'id': "llama-3.1-70b"},
+                'llama-3.1-405b': {'mode': True, 'id': "llama-3.1-405b"}
+            }
             data = {
                 "messages": messages,
                 "id": random_id,
@@ -62,6 +75,7 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
                 "webSearchMode": False,
                 "userSystemPrompt": "",
                 "githubToken": None,
+                "trendingAgentModel": model_id_map[model], # if you actually test this on the site, just ask each model "yo", weird behavior imo
                 "maxTokens": None
             }