From e3a4cd281a9cc77830e91b0634b10604d5570713 Mon Sep 17 00:00:00 2001
From: aaron <>
Date: Sat, 17 May 2025 21:17:45 +0800
Subject: [PATCH] 1

---
 cryptoai/routes/adata.py | 8 +++++---
 docker-compose.yml       | 2 +-
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/cryptoai/routes/adata.py b/cryptoai/routes/adata.py
index b196709..bbe6f3c 100644
--- a/cryptoai/routes/adata.py
+++ b/cryptoai/routes/adata.py
@@ -57,14 +57,16 @@ async def get_stock_base(stock_code: str):
     return result
 
 @router.get("/stock/data", summary="获取股票数据")
-async def get_stock_data(stock_code: str):
+async def get_stock_data(stock_code: str, start_date: Optional[str] = None, end_date: Optional[str] = None):
     api = AStockAPI()
 
     result = {}
 
     try:
-        start_date = "2025-01-01"   
-        end_date = datetime.now().strftime("%Y-%m-%d")
+        if start_date is None:
+            start_date = "2025-01-01"   
+        if end_date is None:
+            end_date = datetime.now().strftime("%Y-%m-%d")
 
         # 获取市场数据
         market_data = api.get_market_data(stock_code, start_date, end_date)
diff --git a/docker-compose.yml b/docker-compose.yml
index feedc04..9c07b24 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.1.4
+    image: cryptoai-api:0.1.5
     restart: always
     ports:
       - "8000:8000"