aitbc1
b4b5a57390
- Add .aitbc.yaml configuration file with test values - Simplify .gitignore by removing merge conflicts and redundant entries - Reorganize .gitignore sections for better clarity - Set chain_id and proposer_id to empty strings in config.py (require explicit configuration) - Add production Helm values configuration - Add production nginx configuration - Update environment variable handling in chain settings
116 lines
3.7 KiB
Python
116 lines
3.7 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
Complete miner workflow - poll for jobs and assign proposer
|
||
"""
|
||
|
||
import httpx
|
||
import json
|
||
import time
|
||
from datetime import datetime
|
||
|
||
# Configuration
|
||
COORDINATOR_URL = "http://localhost:8001"
|
||
MINER_API_KEY = "${MINER_API_KEY}"
|
||
MINER_ID = "localhost-gpu-miner"
|
||
|
||
def poll_and_accept_job():
|
||
"""Poll for a job and accept it"""
|
||
print("🔍 Polling for jobs...")
|
||
|
||
with httpx.Client() as client:
|
||
# Poll for a job
|
||
response = client.post(
|
||
f"{COORDINATOR_URL}/v1/miners/poll",
|
||
headers={
|
||
"Content-Type": "application/json",
|
||
"X-Api-Key": MINER_API_KEY
|
||
},
|
||
json={"max_wait_seconds": 5}
|
||
)
|
||
|
||
if response.status_code == 200:
|
||
job = response.json()
|
||
print(f"✅ Received job: {job['job_id']}")
|
||
print(f" Task: {job['payload'].get('task', 'unknown')}")
|
||
|
||
# Simulate processing
|
||
print("⚙️ Processing job...")
|
||
time.sleep(2)
|
||
|
||
# Submit result
|
||
result_data = {
|
||
"result": {
|
||
"status": "completed",
|
||
"output": f"Job {job['job_id']} completed successfully",
|
||
"execution_time_ms": 2000,
|
||
"miner_id": MINER_ID
|
||
},
|
||
"metrics": {
|
||
"compute_time": 2.0,
|
||
"energy_used": 0.1
|
||
}
|
||
}
|
||
|
||
print(f"📤 Submitting result for job {job['job_id']}...")
|
||
result_response = client.post(
|
||
f"{COORDINATOR_URL}/v1/miners/{job['job_id']}/result",
|
||
headers={
|
||
"Content-Type": "application/json",
|
||
"X-Api-Key": MINER_API_KEY
|
||
},
|
||
json=result_data
|
||
)
|
||
|
||
if result_response.status_code == 200:
|
||
print("✅ Result submitted successfully!")
|
||
return job['job_id']
|
||
else:
|
||
print(f"❌ Failed to submit result: {result_response.status_code}")
|
||
print(f" Response: {result_response.text}")
|
||
return None
|
||
|
||
elif response.status_code == 204:
|
||
print("ℹ️ No jobs available")
|
||
return None
|
||
else:
|
||
print(f"❌ Failed to poll: {response.status_code}")
|
||
return None
|
||
|
||
def check_block_proposer(job_id):
|
||
"""Check if the block now has a proposer"""
|
||
print(f"\n🔍 Checking proposer for job {job_id}...")
|
||
|
||
with httpx.Client() as client:
|
||
response = client.get(f"{COORDINATOR_URL}/v1/explorer/blocks")
|
||
|
||
if response.status_code == 200:
|
||
blocks = response.json()
|
||
for block in blocks['items']:
|
||
if block['hash'] == job_id:
|
||
print(f"📦 Block Info:")
|
||
print(f" Height: {block['height']}")
|
||
print(f" Hash: {block['hash']}")
|
||
print(f" Proposer: {block['proposer']}")
|
||
print(f" Time: {block['timestamp']}")
|
||
return block
|
||
return None
|
||
|
||
def main():
|
||
print("⛏️ AITBC Miner Workflow Demo")
|
||
print(f" Miner ID: {MINER_ID}")
|
||
print(f" Coordinator: {COORDINATOR_URL}")
|
||
print()
|
||
|
||
# Poll and accept a job
|
||
job_id = poll_and_accept_job()
|
||
|
||
if job_id:
|
||
# Check if the block has a proposer now
|
||
time.sleep(1) # Give the server a moment to update
|
||
check_block_proposer(job_id)
|
||
else:
|
||
print("\n💡 Tip: Create a job first using example_client_remote.py")
|
||
|
||
if __name__ == "__main__":
|
||
main()
|