Merge pull request #6 from allora-network/fix/add-inference-serviece-healthcheck

Add inference service healthcheck to ensure the updater service starts
This commit is contained in:
clementupshot 2024-05-29 12:55:49 +02:00 committed by GitHub
commit 911555f358
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 29 additions and 7 deletions

1
.gitignore vendored
View File

@ -6,6 +6,7 @@ logs/*
.env
keys
data
inference-data
worker-data
head-data
lib

View File

@ -13,15 +13,29 @@ services:
aliases:
- inference
ipv4_address: 172.22.0.4
healthcheck:
test: ["CMD", "curl", "-f", "http://localhost:8000/inference/ETH"]
interval: 10s
timeout: 5s
retries: 12
volumes:
- ./inference-data:/app/data
updater:
container_name: updater-basic-eth-pred
build: .
environment:
- INFERENCE_API_ADDRESS=http://inference:8000
command: python -u /app/update_app.py
command: >
sh -c "
while true; do
python -u /app/update_app.py;
sleep 24h;
done
"
depends_on:
- inference
inference:
condition: service_healthy
networks:
eth-model-local:
aliases:
@ -102,5 +116,6 @@ networks:
- subnet: 172.22.0.0/24
volumes:
inference-data:
worker-data:
head-data:

View File

@ -4,6 +4,8 @@ import requests
inference_address = os.environ["INFERENCE_API_ADDRESS"]
url = f"{inference_address}/update"
print("UPDATING INFERENCE WORKER DATA")
response = requests.get(url)
if response.status_code == 200:
# Request was successful

View File

@ -5,18 +5,22 @@ from concurrent.futures import ThreadPoolExecutor
# Function to download the URL, called asynchronously by several child processes
def download_url(url, download_path):
target_file_path = os.path.join(download_path, os.path.basename(url))
if os.path.exists(target_file_path):
print(f"File already exists: {url}")
return
response = requests.get(url)
if response.status_code == 404:
print(f"File not exist: {url}")
else:
file_name = os.path.join(download_path, os.path.basename(url))
# create the entire path if it doesn't exist
os.makedirs(os.path.dirname(file_name), exist_ok=True)
os.makedirs(os.path.dirname(target_file_path), exist_ok=True)
with open(file_name, "wb") as f:
with open(target_file_path, "wb") as f:
f.write(response.content)
print(f"Downloaded: {url} to {file_name}")
print(f"Downloaded: {url} to {target_file_path}")
def download_binance_monthly_data(