Update models/market_data.py

models/market_data.py

@@ -4,7 +4,7 @@ from pydantic import BaseModel, Field
 
 
 class KlineData(BaseModel):
-    timestamp: int
+    timestamp: int = Field(..., description="Timestamp in milliseconds")
     open: float
     high: float
     low: float
@@ -34,13 +34,10 @@ class MarketData(BaseModel):
 
     @classmethod
     def from_kucoin_response(
-        cls, response: Dict[str, Any], symbol: str, granularity: int
+        cls, response_data: Dict[str, Any], symbol: str, granularity: int
     ):
-        return cls(
-            symbol=symbol,
-            granularity=granularity,
-            klines=[KlineData.from_kucoin_data(k) for k in response.get("data", [])],
-        )
+        klines = [KlineData.from_kucoin_data(k) for k in response_data.get("data", [])]
+        return cls(symbol=symbol, granularity=granularity, klines=klines)
 
 
 class SymbolInfo(BaseModel):
@@ -53,9 +50,9 @@ class SymbolInfo(BaseModel):
     @classmethod
     def from_kucoin_data(cls, data: Dict[str, Any]) -> "SymbolInfo":
         return cls(
-            symbol=data["symbol"],
-            name=data["name"],
-            baseCurrency=data["baseCurrency"],
-            quoteCurrency=data["quoteCurrency"],
-            maxLeverage=int(data["maxLeverage"]),
+            symbol=data.get("symbol", ""),
+            name=data.get("name", ""),
+            baseCurrency=data.get("baseCurrency", ""),
+            quoteCurrency=data.get("quoteCurrency", ""),
+            maxLeverage=int(data.get("maxLeverage", 0)),
         )