aitbc1
bfe6f94b75
Some checks failed
AITBC CI/CD Pipeline / lint-and-test (3.11) (push) Has been cancelled
AITBC CI/CD Pipeline / lint-and-test (3.12) (push) Has been cancelled
AITBC CI/CD Pipeline / lint-and-test (3.13) (push) Has been cancelled
AITBC CI/CD Pipeline / test-cli (push) Has been cancelled
AITBC CI/CD Pipeline / test-services (push) Has been cancelled
AITBC CI/CD Pipeline / test-production-services (push) Has been cancelled
AITBC CI/CD Pipeline / security-scan (push) Has been cancelled
AITBC CI/CD Pipeline / build (push) Has been cancelled
AITBC CI/CD Pipeline / deploy-staging (push) Has been cancelled
AITBC CI/CD Pipeline / deploy-production (push) Has been cancelled
AITBC CI/CD Pipeline / performance-test (push) Has been cancelled
AITBC CI/CD Pipeline / docs (push) Has been cancelled
AITBC CI/CD Pipeline / release (push) Has been cancelled
AITBC CI/CD Pipeline / notify (push) Has been cancelled
Security Scanning / Bandit Security Scan (apps/coordinator-api/src) (push) Has been cancelled
Security Scanning / Bandit Security Scan (cli/aitbc_cli) (push) Has been cancelled
Security Scanning / Bandit Security Scan (packages/py/aitbc-core/src) (push) Has been cancelled
Security Scanning / Bandit Security Scan (packages/py/aitbc-crypto/src) (push) Has been cancelled
Security Scanning / Bandit Security Scan (packages/py/aitbc-sdk/src) (push) Has been cancelled
Security Scanning / Bandit Security Scan (tests) (push) Has been cancelled
Security Scanning / CodeQL Security Analysis (javascript) (push) Has been cancelled
Security Scanning / CodeQL Security Analysis (python) (push) Has been cancelled
Security Scanning / Dependency Security Scan (push) Has been cancelled
Security Scanning / Container Security Scan (push) Has been cancelled
Security Scanning / OSSF Scorecard (push) Has been cancelled
Security Scanning / Security Summary Report (push) Has been cancelled
AITBC CLI Level 1 Commands Test / test-cli-level1 (3.11) (push) Has been cancelled
AITBC CLI Level 1 Commands Test / test-cli-level1 (3.12) (push) Has been cancelled
AITBC CLI Level 1 Commands Test / test-cli-level1 (3.13) (push) Has been cancelled
AITBC CLI Level 1 Commands Test / test-summary (push) Has been cancelled
- Remove debugging service documentation (DEBUgging_SERVICES.md) - Remove development logs policy and quick reference guides - Remove E2E test creation summary - Remove gift certificate example file - Remove GitHub pull summary documentation
73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Simple GPU Registry Server for demonstration
|
|
"""
|
|
|
|
from fastapi import FastAPI, HTTPException
|
|
from pydantic import BaseModel
|
|
from typing import Dict, Any, Optional
|
|
import uvicorn
|
|
from datetime import datetime
|
|
|
|
app = FastAPI(title="GPU Registry Demo")
|
|
|
|
# In-memory storage
|
|
registered_gpus: Dict[str, Dict] = {}
|
|
|
|
class GPURegistration(BaseModel):
|
|
capabilities: Dict[str, Any]
|
|
concurrency: int = 1
|
|
region: Optional[str] = None
|
|
|
|
class Heartbeat(BaseModel):
|
|
inflight: int = 0
|
|
status: str = "ONLINE"
|
|
metadata: Dict[str, Any] = {}
|
|
|
|
@app.get("/")
|
|
async def root():
|
|
return {"message": "GPU Registry Demo", "registered_gpus": len(registered_gpus)}
|
|
|
|
@app.get("/health")
|
|
async def health():
|
|
return {"status": "ok"}
|
|
|
|
@app.post("/miners/register")
|
|
async def register_gpu(miner_id: str, gpu_data: GPURegistration):
|
|
"""Register a GPU miner"""
|
|
registered_gpus[miner_id] = {
|
|
"id": miner_id,
|
|
"registered_at": datetime.utcnow().isoformat(),
|
|
"last_heartbeat": datetime.utcnow().isoformat(),
|
|
**gpu_data.dict()
|
|
}
|
|
return {"status": "ok", "message": f"GPU {miner_id} registered successfully"}
|
|
|
|
@app.post("/miners/heartbeat")
|
|
async def heartbeat(miner_id: str, heartbeat_data: Heartbeat):
|
|
"""Receive heartbeat from GPU miner"""
|
|
if miner_id not in registered_gpus:
|
|
raise HTTPException(status_code=404, detail="GPU not registered")
|
|
|
|
registered_gpus[miner_id]["last_heartbeat"] = datetime.utcnow().isoformat()
|
|
registered_gpus[miner_id]["status"] = heartbeat_data.status
|
|
registered_gpus[miner_id]["metadata"] = heartbeat_data.metadata
|
|
|
|
return {"status": "ok"}
|
|
|
|
@app.get("/miners/list")
|
|
async def list_gpus():
|
|
"""List all registered GPUs"""
|
|
return {"gpus": list(registered_gpus.values())}
|
|
|
|
@app.get("/miners/{miner_id}")
|
|
async def get_gpu(miner_id: str):
|
|
"""Get details of a specific GPU"""
|
|
if miner_id not in registered_gpus:
|
|
raise HTTPException(status_code=404, detail="GPU not registered")
|
|
return registered_gpus[miner_id]
|
|
|
|
if __name__ == "__main__":
|
|
print("Starting GPU Registry Demo on http://localhost:8091")
|
|
uvicorn.run(app, host="0.0.0.0", port=8091)
|