From 6e3c3c3e09c84720fea200a2ecc67c18fb1e6875 Mon Sep 17 00:00:00 2001
From: aaron <>
Date: Sun, 18 May 2025 11:07:21 +0800
Subject: [PATCH] update

---
 cryptoai/routes/adata.py | 8 ++++++++
 docker-compose.yml       | 2 +-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/cryptoai/routes/adata.py b/cryptoai/routes/adata.py
index bbe6f3c..c8edc74 100644
--- a/cryptoai/routes/adata.py
+++ b/cryptoai/routes/adata.py
@@ -126,6 +126,14 @@ async def get_stock_data_all(stock_code: str):
 
 @router.post('/{stock_code}/analysis', summary="获取股票分析数据")
 async def get_stock_analysis(stock_code: str, current_user: Dict[str, Any] = Depends(get_current_user)):
+
+    # 检查stock_code是否存在
+    codes = get_db_manager().search_stock(stock_code)
+    if not codes or len(codes) == 0:
+        raise HTTPException(status_code=400, detail="您输入的股票代码不存在，请检查后重新输入。")
+    
+    stock_code = codes[0]["stock_code"]
+
     url = 'https://mate.aimateplus.com/v1/workflows/run'
     token = 'app-nWuCOa0YfQVtAosTY3Jr5vFV'
     headers = {
diff --git a/docker-compose.yml b/docker-compose.yml
index f9bdc4e..433945d 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.9
+    image: cryptoai-api:0.1.20
     restart: always
     ports:
       - "8000:8000"