discovery-server.py

from fastapi import FastAPI, Request
from datetime import datetime, timedelta
import uvicorn

app = FastAPI(title="Agent Discovery Service")
registry = {}

@app.post("/register")
async def register(request: Request):
    data = await request.json()
    name = data["name"]
    registry[name] = {
        "url": data["url"],
        "last_heartbeat": datetime.utcnow()
    }
    return {"status": "registered", "agent": name}

@app.get("/services")
async def list_services():
    now = datetime.utcnow()
    alive = {
        n: {"url": d["url"], "last_heartbeat": d["last_heartbeat"].isoformat()}
        for n, d in registry.items()
        if now - d["last_heartbeat"] < timedelta(seconds=30)
    }
    return {"active_agents": alive}

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8500)