From 7f625a4c5872986439512e15031a50d026b43355 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=85=B0=E5=BF=97=E5=AE=8F?= <lanzhihong@jianke.com>
Date: Tue, 4 Mar 2025 15:12:14 +0800
Subject: [PATCH] =?UTF-8?q?ADD=EF=BC=9A=20=E7=8E=AF=E5=A2=83=E5=8F=98?=
 =?UTF-8?q?=E9=87=8F=E6=94=AF=E6=8C=81=E8=B0=83=E6=95=B4api=20timeout?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .env                               | 3 ++-
 .github/workflows/docker-image.yml | 6 +++++-
 docker-compose.yml                 | 1 +
 stock_analyzer.py                  | 5 +++--
 4 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/.env b/.env
index 42413d2..c062d70 100644
--- a/.env
+++ b/.env
@@ -1,4 +1,5 @@
 API_KEY=
 API_URL=
 API_MODEL=
-ANNOUNCEMENT_TEXT=
\ No newline at end of file
+ANNOUNCEMENT_TEXT=
+API_TIMEOUT=60
diff --git a/.github/workflows/docker-image.yml b/.github/workflows/docker-image.yml
index b6198e3..6935921 100644
--- a/.github/workflows/docker-image.yml
+++ b/.github/workflows/docker-image.yml
@@ -16,6 +16,10 @@ jobs:
     - name: Set up QEMU
       uses: docker/setup-qemu-action@v2
     
+    - name: Get current time
+      id: time
+      run: echo "TIME=$(date +'%Y%m%d%H%M')" >> $GITHUB_ENV
+    
     - name: Login to Docker Hub
       uses: docker/login-action@v1
       with:
@@ -33,4 +37,4 @@ jobs:
         push: true
         tags: |
           ${{ secrets.DOCKERHUB_USERNAME }}/stock-scanner:latest
-          ${{ secrets.DOCKERHUB_USERNAME }}/stock-scanner:${{ github.sha }}
+          ${{ secrets.DOCKERHUB_USERNAME }}/stock-scanner:${{ env.TIME }}
diff --git a/docker-compose.yml b/docker-compose.yml
index a89085b..4d0da35 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -9,6 +9,7 @@ services:
       - API_KEY=${API_KEY}
       - API_URL=${API_URL}
       - API_MODEL=${API_MODEL}
+      - API_TIMEOUT=${API_TIMEOUT}
     volumes:
       - .:/app
     restart: unless-stopped
diff --git a/stock_analyzer.py b/stock_analyzer.py
index cfb3c23..066185b 100644
--- a/stock_analyzer.py
+++ b/stock_analyzer.py
@@ -12,9 +12,10 @@ class StockAnalyzer:
         # 加载环境变量
         load_dotenv()
         
-        # 设置 Gemini API
+        # 设置 API
         self.API_URL = os.getenv('API_URL')
         self.API_KEY = os.getenv('API_KEY')
+        self.API_TIMEOUT = int(os.getenv('API_TIMEOUT', '60'))  
         
         # 配置参数
         self.params = {
@@ -255,7 +256,7 @@ class StockAnalyzer:
                 api_url,
                 headers=headers,
                 json=data,
-                timeout=60
+                timeout=self.API_TIMEOUT
             )
             
             print(api_url)