Fix session disconnects

Author: Breta01

brain/api.py

@@ -58,6 +58,8 @@ def _new_session(self):
                         "sec-fetch-mode": "cors",
                         "sec-fetch-site": "same-site",
                         "Referer": "https://platform.worldquantbrain.com/",
+                        "connection": "keep-alive",
+                        "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/136.0.0.0 Safari/537.36",
                     }
                 )
                 self._authenticate()

brain/search_algorithm.py

@@ -106,8 +106,9 @@ def monitor_alpha(response, alpha_config):
         )
     except Exception as e:
         print(f"Error during obtaining results: {e}")
-        # TODO: Remote end closed connection without response
-        # Need to refresh session
+        if isinstance(e, (ConnectionError)):
+            BrainAPI._new_session()
+
         return {
             "alpha_id": None,
             "simulate_data": alpha_config,