ADD: 环境变量支持调整api timeout

This commit is contained in:
兰志宏
2025-03-04 15:12:14 +08:00
parent 3195e579d1
commit 7f625a4c58
4 changed files with 11 additions and 4 deletions

3
.env
View File

@@ -1,4 +1,5 @@
API_KEY=
API_URL=
API_MODEL=
ANNOUNCEMENT_TEXT=
ANNOUNCEMENT_TEXT=
API_TIMEOUT=60

View File

@@ -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 }}

View File

@@ -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

View File

@@ -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)