diff --git a/cryptoai/api/__pycache__/binance_api.cpython-313.pyc b/cryptoai/api/__pycache__/binance_api.cpython-313.pyc index b12ba3a..7d3d072 100644 Binary files a/cryptoai/api/__pycache__/binance_api.cpython-313.pyc and b/cryptoai/api/__pycache__/binance_api.cpython-313.pyc differ diff --git a/cryptoai/api/binance_api.py b/cryptoai/api/binance_api.py index 5d33849..19699a5 100644 --- a/cryptoai/api/binance_api.py +++ b/cryptoai/api/binance_api.py @@ -84,7 +84,7 @@ class BinanceAPI: print(f"获取交易对大户持仓多空比时出错: {e}") return 0 - def get_historical_klines(self, symbol: str, interval: str, start_str: str, end_str: Optional[str] = None) -> pd.DataFrame: + def get_historical_klines(self, symbol: str, interval: str, start_str: str, end_str: Optional[str] = None, limit: Optional[int] = None) -> pd.DataFrame: """ 获取历史K线数据 @@ -102,7 +102,8 @@ class BinanceAPI: symbol=symbol, interval=interval, start_str=start_str, - end_str=end_str + end_str=end_str, + limit=limit ) # 转换为DataFrame diff --git a/cryptoai/routes/crypto.py b/cryptoai/routes/crypto.py index d9594a7..fa5a963 100644 --- a/cryptoai/routes/crypto.py +++ b/cryptoai/routes/crypto.py @@ -32,7 +32,7 @@ class CryptoAnalysisRequest(BaseModel): timeframe: Optional[str] = None @router.get("/kline/{symbol}") -async def get_crypto_kline(symbol: str, timeframe: Optional[str] = None, start_time: Optional[str] = None, end_time: Optional[str] = None): +async def get_crypto_kline(symbol: str, timeframe: Optional[str] = None, start_time: Optional[str] = None, end_time: Optional[str] = None, limit: Optional[int] = 200): binance_api = get_binance_api() @@ -45,7 +45,7 @@ async def get_crypto_kline(symbol: str, timeframe: Optional[str] = None, start_t print(f"symbol: {symbol}, timeframe: {timeframe}, start_time: {start_time}, end_time: {end_time}") - kline = binance_api.get_historical_klines(symbol, timeframe, start_time, end_time) + kline = binance_api.get_historical_klines(symbol, timeframe, start_time, end_time, limit=limit) data_processor = DataProcessor() diff --git a/docker-compose.yml b/docker-compose.yml index 242c96f..1df7cc4 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -29,7 +29,7 @@ services: cryptoai-api: build: . container_name: cryptoai-api - image: cryptoai-api:0.0.24 + image: cryptoai-api:0.0.25 restart: always ports: - "8000:8000"