chore: 清理未使用变量

services/ai_analyzer.py

@@ -1,13 +1,11 @@
 import pandas as pd
-import numpy as np
 import os
 import json
-import asyncio
 import httpx
 import re
-from typing import Dict, List, Optional, Any, Generator, AsyncGenerator
+from typing import AsyncGenerator
 from dotenv import load_dotenv
-from logger import get_logger
+from utils.logger import get_logger
 from utils.api_utils import APIUtils
 from datetime import datetime
 

services/fund_service_async.py

@@ -1,7 +1,7 @@
 import asyncio
 import pandas as pd
 from typing import List, Dict, Any, Optional
-from logger import get_logger
+from utils.logger import get_logger
 from datetime import datetime, timedelta
 
 # 获取日志器

services/stock_analyzer_service.py

@@ -1,10 +1,7 @@
-import pandas as pd
-import numpy as np
-import asyncio
 import json
 from datetime import datetime
-from typing import Dict, List, Optional, Tuple, Any, AsyncGenerator
-from logger import get_logger
+from typing import List, AsyncGenerator
+from utils.logger import get_logger
 from services.stock_data_provider import StockDataProvider
 from services.technical_indicator import TechnicalIndicator
 from services.stock_scorer import StockScorer

services/stock_data_provider.py

@@ -1,11 +1,8 @@
 import pandas as pd
-import numpy as np
 from datetime import datetime, timedelta
 import asyncio
-import os
 from typing import Dict, List, Optional, Tuple, Any
-from logger import get_logger
-import re
+from utils.logger import get_logger
 
 # 获取日志器
 logger = get_logger()

services/stock_scorer.py

@@ -1,7 +1,6 @@
 import pandas as pd
-import numpy as np
-from typing import Dict, Optional, Any, List, Tuple
-from logger import get_logger
+from typing import Dict, List, Tuple
+from utils.logger import get_logger
 
 # 获取日志器
 logger = get_logger()
@@ -14,7 +13,7 @@ class StockScorer:
     
     def __init__(self):
         """初始化股票评分服务"""
-        logger.debug("初始化StockScorer")
+        logger.debug("初始化StockScorer股票评分服务")
     
     def calculate_score(self, df: pd.DataFrame) -> int:
         """

services/technical_indicator.py

@@ -1,7 +1,6 @@
 import pandas as pd
-import numpy as np
 from typing import Dict, Optional, Any
-from logger import get_logger
+from utils.logger import get_logger
 
 # 获取日志器
 logger = get_logger()
@@ -29,7 +28,7 @@ class TechnicalIndicator:
             'atr_period': 14
         }
         
-        logger.debug(f"初始化TechnicalIndicator，参数: {self.params}")
+        logger.debug(f"初始化TechnicalIndicator技术指标计算服务，参数: {self.params}")
     
     def calculate_ema(self, series: pd.Series, period: int) -> pd.Series:
         """

services/us_stock_service_async.py

@@ -1,19 +1,19 @@
 import asyncio
 import pandas as pd
 from typing import List, Dict, Any, Optional
-from logger import get_logger
+from utils.logger import get_logger
 
 # 获取日志器
 logger = get_logger()
 
 class USStockServiceAsync:
     """
-    异步美股服务
-    提供美股数据的异步搜索和获取功能
+    美股服务
+    提供美股数据的搜索和获取功能
     """
     
     def __init__(self):
-        """初始化异步美股服务"""
+        """初始化美股服务"""
         logger.debug("初始化USStockServiceAsync")
         
         # 可选：添加缓存以减少频繁请求

tests/test_stream.py

@@ -1,13 +1,13 @@
 import os
 import requests
 import json
-from logger import get_logger, get_stream_logger
+from utils.logger import get_logger
 from dotenv import load_dotenv
 from utils.api_utils import APIUtils
 
 # 获取日志器
 logger = get_logger()
-stream_logger = get_stream_logger()
+
 
 def _truncate_json_for_logging(json_obj, max_length=500):
     """截断JSON对象用于日志记录，避免日志过大
@@ -38,7 +38,7 @@ def test_api_stream():
     # 获取API配置
     api_url = os.getenv('API_URL')
     api_key = os.getenv('API_KEY')
-    api_model = os.getenv('API_MODEL', 'gpt-3.5-turbo')
+    api_model = os.getenv('API_MODEL', 'gemini-2.0-flash')
     
     logger.info(f"开始测试API流式响应，API URL: {api_url}, MODEL: {api_model}")
     

utils/logger.py

@@ -2,7 +2,7 @@ from loguru import logger
 import sys
 import os
 from datetime import datetime
-import shutil
+
 
 # 创建日志目录
 log_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "logs")

web_server.py

@@ -1,8 +1,8 @@
 from fastapi import FastAPI, Request, Response, Depends, HTTPException, BackgroundTasks
-from fastapi.responses import JSONResponse, StreamingResponse, FileResponse, RedirectResponse
+from fastapi.responses import JSONResponse, StreamingResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm
+from fastapi.security import OAuth2PasswordBearer
 from pydantic import BaseModel, Field
 from typing import List, Optional, Dict, Any, Generator
 from services.stock_analyzer_service import StockAnalyzerService
@@ -10,9 +10,9 @@ from services.us_stock_service_async import USStockServiceAsync
 from services.fund_service_async import FundServiceAsync
 import os
 import httpx
-from logger import get_logger
+from utils.logger import get_logger
 from utils.api_utils import APIUtils
-from dotenv import load_dotenv, dotenv_values
+from dotenv import load_dotenv
 import uvicorn
 import json
 import secrets