Update main.py

main.py

@@ -173,13 +173,6 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
                 "webSearchMode": False,
             }
 
-            if model in cls.agentMode:
-                data["agentMode"] = cls.agentMode[model]
-            elif model in cls.trendingAgentMode:
-                data["trendingAgentMode"] = cls.trendingAgentMode[model]
-            elif model in cls.userSelectedModel:
-                data["userSelectedModel"] = cls.userSelectedModel[model]
-            
             async with session.post(cls.api_endpoint, json=data, proxy=proxy) as response:
                 response.raise_for_status()
                 if model == 'ImageGenerationLV45LJp':
@@ -191,12 +184,19 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
                     else:
                         raise Exception("Image URL not found in the response")
                 else:
+                    response_content = ""
                     async for chunk in response.content.iter_any():
                         if chunk:
-                            decoded_chunk = chunk.decode(errors='ignore')  # Handle decoding errors
+                            decoded_chunk = chunk.decode(errors='ignore')
                             decoded_chunk = re.sub(r'\$@\$v=[^$]+\$@\$', '', decoded_chunk)
                             if decoded_chunk.strip():
-                                yield decoded_chunk
+                                response_content += decoded_chunk
+
+                    # Check if the response content is empty
+                    if not response_content.strip():
+                        raise ModelNotWorkingException(model)
+
+                    yield response_content
 
 # FastAPI app setup
 app = FastAPI()
@@ -240,7 +240,8 @@ async def chat_completions(request: ChatRequest):
             model=request.model,
             messages=messages,
             image=None,  # Pass the image if required
-            image_name=None  # Pass image name if required
+            image_name=None
+  # Pass image name if required
         )
     except ModelNotWorkingException as e:
         raise HTTPException(status_code=503, detail=str(e))
@@ -249,7 +250,6 @@ async def chat_completions(request: ChatRequest):
         async def generate():
             async for chunk in async_generator:
                 if isinstance(chunk, ImageResponse):
-                    # Format the response as a Markdown image
                     image_markdown = f"![image]({chunk.url})"
                     yield f"data: {json.dumps(create_response(image_markdown, request.model))}\n\n"
                 else:
@@ -263,7 +263,7 @@ async def chat_completions(request: ChatRequest):
             if isinstance(chunk, ImageResponse):
                 response_content += f"![image]({chunk.url})\n"
             else:
-                response_content += chunk  # Concatenate text responses
+                response_content += chunk
 
         return {
             "id": f"chatcmpl-{uuid.uuid4()}",
@@ -286,3 +286,4 @@ async def chat_completions(request: ChatRequest):
 @app.get("/niansuhai/v1/models")
 async def get_models():
     return {"models": Blackbox.models}
+