50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
"""
|
|
Agent Control Center — Control Plane API
|
|
Часть 1: управление агентами, задачами, памятью
|
|
Часть 2: Web UI
|
|
"""
|
|
from __future__ import annotations
|
|
from pathlib import Path
|
|
from fastapi import FastAPI, HTTPException, WebSocket, WebSocketDisconnect
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.staticfiles import StaticFiles
|
|
from api import runs, agents, memory, skills, connectors, tasks, accounts
|
|
import asyncio, json
|
|
|
|
app = FastAPI(
|
|
title="Agent Control Center",
|
|
version="0.1.0",
|
|
description="Control Plane для управления Hermes-агентами",
|
|
)
|
|
app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_methods=["*"], allow_headers=["*"])
|
|
app.include_router(runs.router)
|
|
app.include_router(agents.router)
|
|
app.include_router(memory.router)
|
|
app.include_router(skills.router)
|
|
app.include_router(connectors.router)
|
|
app.include_router(tasks.router)
|
|
app.include_router(accounts.router)
|
|
|
|
# WebSocket для real-time обновлений
|
|
@app.websocket("/ws")
|
|
async def websocket_endpoint(ws: WebSocket):
|
|
await ws.accept()
|
|
try:
|
|
while True:
|
|
data = await ws.receive_text()
|
|
msg = json.loads(data)
|
|
# Echo + broadcast agent status
|
|
await ws.send_json({"type": "echo", "data": msg})
|
|
except WebSocketDisconnect:
|
|
pass
|
|
except Exception:
|
|
pass
|
|
|
|
# Web UI (доступен на /ui)
|
|
static = Path(__file__).parent / "static"
|
|
static.mkdir(exist_ok=True)
|
|
app.mount("/ui", StaticFiles(directory=str(static), html=True), name="static")
|
|
|
|
@app.get("/health")
|
|
def health():
|
|
return {"status": "ok", "version": app.version}
|