From 27a67a042d268ee5f2fa0334e8eb5da0f35dc837 Mon Sep 17 00:00:00 2001 From: Danglewood <85772166+deeleeramone@users.noreply.github.com> Date: Thu, 2 Jan 2025 01:29:03 -0800 Subject: [PATCH] not ready for uvloop --- .../core/openbb_core/provider/utils/websockets/broadcast.py | 2 ++ .../core/openbb_core/provider/utils/websockets/database.py | 3 --- .../providers/polygon/openbb_polygon/utils/websocket_client.py | 3 --- .../providers/tiingo/openbb_tiingo/utils/websocket_client.py | 3 --- 4 files changed, 2 insertions(+), 9 deletions(-) diff --git a/openbb_platform/core/openbb_core/provider/utils/websockets/broadcast.py b/openbb_platform/core/openbb_core/provider/utils/websockets/broadcast.py index 9e230c538a4..0d53eddba84 100644 --- a/openbb_platform/core/openbb_core/provider/utils/websockets/broadcast.py +++ b/openbb_platform/core/openbb_core/provider/utils/websockets/broadcast.py @@ -10,6 +10,7 @@ from pathlib import Path from typing import Optional import uvicorn + from fastapi import FastAPI, WebSocket, WebSocketDisconnect from openbb_core.provider.utils.websockets.database import ( CHECK_FOR, @@ -23,6 +24,7 @@ from openbb_core.provider.utils.websockets.helpers import ( ) from starlette.websockets import WebSocketState + kwargs = parse_kwargs() HOST = kwargs.pop("host", None) or "localhost" diff --git a/openbb_platform/core/openbb_core/provider/utils/websockets/database.py b/openbb_platform/core/openbb_core/provider/utils/websockets/database.py index 168eed69c73..32441de2382 100644 --- a/openbb_platform/core/openbb_core/provider/utils/websockets/database.py +++ b/openbb_platform/core/openbb_core/provider/utils/websockets/database.py @@ -8,13 +8,10 @@ from contextlib import asynccontextmanager from datetime import timedelta from typing import TYPE_CHECKING, Any, Iterable, Optional, Union -import uvloop from openbb_core.app.model.abstract.error import OpenBBError from openbb_core.provider.utils.helpers import run_async from openbb_core.provider.utils.websockets.helpers import kill_thread -asyncio.set_event_loop_policy(uvloop.EventLoopPolicy()) - if TYPE_CHECKING: import logging from pathlib import Path diff --git a/openbb_platform/providers/polygon/openbb_polygon/utils/websocket_client.py b/openbb_platform/providers/polygon/openbb_polygon/utils/websocket_client.py index d12153f3eea..af0924f1d5d 100644 --- a/openbb_platform/providers/polygon/openbb_polygon/utils/websocket_client.py +++ b/openbb_platform/providers/polygon/openbb_polygon/utils/websocket_client.py @@ -38,7 +38,6 @@ import sys import time import orjson -import uvloop import websockets from openbb_core.provider.utils.websockets.database import Database, DatabaseWriter from openbb_core.provider.utils.websockets.helpers import ( @@ -56,8 +55,6 @@ from pydantic import ValidationError from websockets.asyncio.client import connect from websockets.extensions import permessage_deflate -asyncio.set_event_loop_policy(uvloop.EventLoopPolicy()) - URL_MAP = { "stock": "wss://socket.polygon.io/stocks", "stock_delayed": "wss://delayed.polygon.io/stocks", diff --git a/openbb_platform/providers/tiingo/openbb_tiingo/utils/websocket_client.py b/openbb_platform/providers/tiingo/openbb_tiingo/utils/websocket_client.py index 9f88a667934..5ad5126c67b 100644 --- a/openbb_platform/providers/tiingo/openbb_tiingo/utils/websocket_client.py +++ b/openbb_platform/providers/tiingo/openbb_tiingo/utils/websocket_client.py @@ -7,7 +7,6 @@ import sys from datetime import UTC, datetime import orjson -import uvloop import websockets from openbb_core.provider.utils.errors import UnauthorizedError from openbb_core.provider.utils.websockets.database import Database, DatabaseWriter @@ -22,8 +21,6 @@ from openbb_tiingo.models.websocket_connection import TiingoWebSocketData from pandas import to_datetime from pydantic import ValidationError -asyncio.set_event_loop_policy(uvloop.EventLoopPolicy()) - URL_MAP = { "stock": "wss://api.tiingo.com/iex", "fx": "wss://api.tiingo.com/fx",