Update main.py

main.py

@@ -373,7 +373,6 @@ class Blackbox:
                                     except json.JSONDecodeError as je:
                                         logger.error("Failed to parse search results JSON.")
                                         raise je
-                break  # Exit the retry loop if successful
             except ClientError as ce:
                 logger.error(f"Client error occurred: {ce}. Retrying attempt {attempt + 1}/{retry_attempts}")
                 if attempt == retry_attempts - 1:
@@ -526,7 +525,8 @@ async def chat_completions(request: ChatRequest, req: Request, api_key: str = De
     try:
         if request.stream:
             logger.info("Streaming response")
-            streaming_response = await Blackbox.create_async_generator(
+            # **Removed the 'await' keyword here**
+            streaming_response = Blackbox.create_async_generator(
                 model=modified_request.model,
                 messages=[{"role": msg.role, "content": msg.content} for msg in modified_request.messages],
                 proxy=None,
@@ -610,7 +610,8 @@ async def chat_completions(request: ChatRequest, req: Request, api_key: str = De
             return StreamingResponse(generate_with_analysis(), media_type="text/event-stream")
         else:
             logger.info("Non-streaming response")
-            streaming_response = await Blackbox.create_async_generator(
+            # **Removed the 'await' keyword here as well**
+            streaming_response = Blackbox.create_async_generator(
                 model=modified_request.model,
                 messages=[{"role": msg.role, "content": msg.content} for msg in modified_request.messages],
                 proxy=None,