f74a208cbc
chore(gitignore): add development log file patterns - Add patterns for .log, .out, and .err files - Add wget-log and download.log patterns - Organize under new "Development Logs" section ```
145 lines
3.9 KiB
Python
145 lines
3.9 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
AITBC Agent Registry Service
|
|
Central agent discovery and registration system
|
|
"""
|
|
|
|
from fastapi import FastAPI, HTTPException, Depends
|
|
from pydantic import BaseModel
|
|
from typing import List, Optional, Dict, Any
|
|
import json
|
|
import time
|
|
import uuid
|
|
from datetime import datetime, timedelta
|
|
import sqlite3
|
|
from contextlib import contextmanager
|
|
|
|
app = FastAPI(title="AITBC Agent Registry API", version="1.0.0")
|
|
|
|
# Database setup
|
|
def get_db():
|
|
conn = sqlite3.connect('agent_registry.db')
|
|
conn.row_factory = sqlite3.Row
|
|
return conn
|
|
|
|
@contextmanager
|
|
def get_db_connection():
|
|
conn = get_db()
|
|
try:
|
|
yield conn
|
|
finally:
|
|
conn.close()
|
|
|
|
# Initialize database
|
|
def init_db():
|
|
with get_db_connection() as conn:
|
|
conn.execute('''
|
|
CREATE TABLE IF NOT EXISTS agents (
|
|
id TEXT PRIMARY KEY,
|
|
name TEXT NOT NULL,
|
|
type TEXT NOT NULL,
|
|
capabilities TEXT NOT NULL,
|
|
chain_id TEXT NOT NULL,
|
|
endpoint TEXT NOT NULL,
|
|
status TEXT DEFAULT 'active',
|
|
last_heartbeat TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
|
metadata TEXT,
|
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
)
|
|
''')
|
|
|
|
# Models
|
|
class Agent(BaseModel):
|
|
id: str
|
|
name: str
|
|
type: str
|
|
capabilities: List[str]
|
|
chain_id: str
|
|
endpoint: str
|
|
metadata: Optional[Dict[str, Any]] = {}
|
|
|
|
class AgentRegistration(BaseModel):
|
|
name: str
|
|
type: str
|
|
capabilities: List[str]
|
|
chain_id: str
|
|
endpoint: str
|
|
metadata: Optional[Dict[str, Any]] = {}
|
|
|
|
# API Endpoints
|
|
@app.on_event("startup")
|
|
async def startup_event():
|
|
init_db()
|
|
|
|
@app.post("/api/agents/register", response_model=Agent)
|
|
async def register_agent(agent: AgentRegistration):
|
|
"""Register a new agent"""
|
|
agent_id = str(uuid.uuid4())
|
|
|
|
with get_db_connection() as conn:
|
|
conn.execute('''
|
|
INSERT INTO agents (id, name, type, capabilities, chain_id, endpoint, metadata)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?)
|
|
''', (
|
|
agent_id, agent.name, agent.type,
|
|
json.dumps(agent.capabilities), agent.chain_id,
|
|
agent.endpoint, json.dumps(agent.metadata)
|
|
))
|
|
|
|
return Agent(
|
|
id=agent_id,
|
|
name=agent.name,
|
|
type=agent.type,
|
|
capabilities=agent.capabilities,
|
|
chain_id=agent.chain_id,
|
|
endpoint=agent.endpoint,
|
|
metadata=agent.metadata
|
|
)
|
|
|
|
@app.get("/api/agents", response_model=List[Agent])
|
|
async def list_agents(
|
|
agent_type: Optional[str] = None,
|
|
chain_id: Optional[str] = None,
|
|
capability: Optional[str] = None
|
|
):
|
|
"""List registered agents with optional filters"""
|
|
with get_db_connection() as conn:
|
|
query = "SELECT * FROM agents WHERE status = 'active'"
|
|
params = []
|
|
|
|
if agent_type:
|
|
query += " AND type = ?"
|
|
params.append(agent_type)
|
|
|
|
if chain_id:
|
|
query += " AND chain_id = ?"
|
|
params.append(chain_id)
|
|
|
|
if capability:
|
|
query += " AND capabilities LIKE ?"
|
|
params.append(f'%{capability}%')
|
|
|
|
agents = conn.execute(query, params).fetchall()
|
|
|
|
return [
|
|
Agent(
|
|
id=agent["id"],
|
|
name=agent["name"],
|
|
type=agent["type"],
|
|
capabilities=json.loads(agent["capabilities"]),
|
|
chain_id=agent["chain_id"],
|
|
endpoint=agent["endpoint"],
|
|
metadata=json.loads(agent["metadata"] or "{}")
|
|
)
|
|
for agent in agents
|
|
]
|
|
|
|
@app.get("/api/health")
|
|
async def health_check():
|
|
"""Health check endpoint"""
|
|
return {"status": "ok", "timestamp": datetime.utcnow()}
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
uvicorn.run(app, host="0.0.0.0", port=8003)
|