Update main.py

main.py

@@ -126,7 +126,7 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
             prefix = f"@{cls.userSelectedModel[model]}"
             if not messages[0]['content'].startswith(prefix):
                 messages[0]['content'] = f"{prefix} {messages[0]['content']}"
-
+        
         async with ClientSession(headers=headers) as session:
             if image is not None:
                 messages[-1]["data"] = {
@@ -147,7 +147,7 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
                 "userSelectedModel": None,
                 "userSystemPrompt": None,
                 "isMicMode": False,
-                "maxTokens": 99999999,
+                "maxTokens": 8192,
                 "playgroundTopP": 0.9,
                 "playgroundTemperature": 0.5,
                 "isChromeExt": False,
@@ -169,12 +169,8 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
             
             async with session.post(cls.api_endpoint, json=data, proxy=proxy) as response:
                 response.raise_for_status()
-                response_text = await response.text()
-                
-                if not response_text:  # Check for empty response
-                    raise ModelNotWorkingException(model)
-
                 if model == 'ImageGenerationLV45LJp':
+                    response_text = await response.text()
                     url_match = re.search(r'https://storage\.googleapis\.com/[^\s\)]+', response_text)
                     if url_match:
                         image_url = url_match.group(0)
@@ -275,4 +271,4 @@ async def chat_completions(request: ChatRequest):
 
 @app.get("/niansuhai/v1/models")
 async def get_models():
-    return {"models": Blackbox.models}
+    return {"models": Blackbox.models}