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_KEY=
API_URL= API_URL=
API_MODEL= API_MODEL=
ANNOUNCEMENT_TEXT= ANNOUNCEMENT_TEXT=
API_TIMEOUT=60

View File

@@ -16,6 +16,10 @@ jobs:
- name: Set up QEMU - name: Set up QEMU
uses: docker/setup-qemu-action@v2 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 - name: Login to Docker Hub
uses: docker/login-action@v1 uses: docker/login-action@v1
with: with:
@@ -33,4 +37,4 @@ jobs:
push: true push: true
tags: | tags: |
${{ secrets.DOCKERHUB_USERNAME }}/stock-scanner:latest ${{ 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_KEY=${API_KEY}
- API_URL=${API_URL} - API_URL=${API_URL}
- API_MODEL=${API_MODEL} - API_MODEL=${API_MODEL}
- API_TIMEOUT=${API_TIMEOUT}
volumes: volumes:
- .:/app - .:/app
restart: unless-stopped restart: unless-stopped

View File

@@ -12,9 +12,10 @@ class StockAnalyzer:
# 加载环境变量 # 加载环境变量
load_dotenv() load_dotenv()
# 设置 Gemini API # 设置 API
self.API_URL = os.getenv('API_URL') self.API_URL = os.getenv('API_URL')
self.API_KEY = os.getenv('API_KEY') self.API_KEY = os.getenv('API_KEY')
self.API_TIMEOUT = int(os.getenv('API_TIMEOUT', '60'))
# 配置参数 # 配置参数
self.params = { self.params = {
@@ -255,7 +256,7 @@ class StockAnalyzer:
api_url, api_url,
headers=headers, headers=headers,
json=data, json=data,
timeout=60 timeout=self.API_TIMEOUT
) )
print(api_url) print(api_url)