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)