From 8712e0569fe35e8cefd46702772d2cfee515e918 Mon Sep 17 00:00:00 2001
From: Danglewood <85772166+deeleeramone@users.noreply.github.com>
Date: Mon, 11 Nov 2024 16:59:57 -0800
Subject: [PATCH] raise exc as OpenBBError in client.connect

---
 .../extensions/websockets/openbb_websockets/client.py    | 3 ++-
 .../openbb_polygon/models/websocket_connection.py        | 9 +++++++--
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/openbb_platform/extensions/websockets/openbb_websockets/client.py b/openbb_platform/extensions/websockets/openbb_websockets/client.py
index 0d416d3d6f6..72a720b3ce5 100644
--- a/openbb_platform/extensions/websockets/openbb_websockets/client.py
+++ b/openbb_platform/extensions/websockets/openbb_websockets/client.py
@@ -328,6 +328,7 @@ class WebSocketClient:
         import subprocess
         import threading
         import time
+        from openbb_core.app.model.abstract.error import OpenBBError
 
         if self.is_running:
             self.logger.info("Provider connection already running.")
@@ -388,7 +389,7 @@ class WebSocketClient:
         if self._exception is not None:
             exc = getattr(self, "_exception", None)
             self._exception = None
-            raise exc
+            raise OpenBBError(exc)
 
         if not self.is_running:
             self.logger.error("The provider server failed to start.")
diff --git a/openbb_platform/providers/polygon/openbb_polygon/models/websocket_connection.py b/openbb_platform/providers/polygon/openbb_polygon/models/websocket_connection.py
index 0cea4380e14..c333ddfaf16 100644
--- a/openbb_platform/providers/polygon/openbb_polygon/models/websocket_connection.py
+++ b/openbb_platform/providers/polygon/openbb_polygon/models/websocket_connection.py
@@ -650,7 +650,9 @@ class PolygonStockTradeWebSocketData(WebSocketData):
         "trf_id": "trfi",
         "tape": "z",
         "price": "p",
+        "size": "s",
         "conditions": "c",
+        "trf_timestamp": "trft",
     }
 
     type: str = Field(
@@ -658,7 +660,7 @@ class PolygonStockTradeWebSocketData(WebSocketData):
     )
     date: datetime = Field(
         description=DATA_DESCRIPTIONS.get("date", "")
-        + "The end of the aggregate window.",
+        + "The SIP timestamp of the trade.",
     )
     symbol: str = Field(
         description=DATA_DESCRIPTIONS.get("symbol", ""),
@@ -667,6 +669,9 @@ class PolygonStockTradeWebSocketData(WebSocketData):
         description="The price of the trade.",
         json_schema_extra={"x-unit_measurement": "currency"},
     )
+    size: float = Field(
+        description="The size of the trade.",
+    )
     exchange: str = Field(
         description="The exchange where the trade originated.",
     )
@@ -677,7 +682,7 @@ class PolygonStockTradeWebSocketData(WebSocketData):
         default=None,
         description="The conditions of the trade.",
     )
-    trf_id: Optional[str] = Field(
+    trf_id: Optional[int] = Field(
         default=None,
         description="The ID for the Trade Reporting Facility where the trade took place.",
     )