Chore: Make release 1.0.113

goldmansachs · Aug 19, 2024 · e5f6c7f · e5f6c7f
1 parent 74181e0
commit e5f6c7f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/gs_quant/session.py b/gs_quant/session.py
@@ -476,9 +476,11 @@ async def _put_async(self, path: str, payload: Optional[Union[dict, Base]] = Non
                                          return_request_id=return_request_id)
         return ret
 
-    def _connect_websocket(self, path: str, headers: Optional[dict] = None, include_version=True):
+    def _connect_websocket(self, path: str, headers: Optional[dict] = None, include_version=True,
+                           domain: Optional[str] = None):
         import websockets
-        url = 'ws{}{}{}'.format(self.domain[4:], '/' + self.api_version if include_version else '', path)
+        version_path = '/' + self.api_version if include_version else ''
+        url = f'{domain}{version_path}{path}' if domain else f'ws{self.domain[4:]}{version_path}{path}'
         extra_headers = self._headers() + list((headers or {}).items())
         return websockets.connect(url,
                                   extra_headers=extra_headers,