Update model_logic.py

model_logic.py

@@ -390,11 +390,11 @@ def generate_stream(provider: str, model_display_name: str, api_key_override: st
             byte_buffer = b""
             for chunk in response.iter_content(chunk_size=8192):
                 # Check for potential HTTP errors during streaming
-                 if response.status_code != 200:
-                      error_body = response.text
-                      logger.error(f"HTTP Error during Cohere stream: {response.status_code}, Body: {error_body}")
-                      yield f"API HTTP Error ({response.status_code}) during Cohere stream: {error_body}"
-                      return # Stop streaming on error
+                if response.status_code != 200:
+                    error_body = response.text
+                    logger.error(f"HTTP Error during Cohere stream: {response.status_code}, Body: {error_body}")
+                    yield f"API HTTP Error ({response.status_code}) during Cohere stream: {error_body}"
+                    return # Stop streaming on error
 
                 byte_buffer += chunk
                 while b'\n\n' in byte_buffer: # Cohere uses \n\n as event separator