INNER CODE UNIT · Python

connect

SHAdd0WTAka/Zen-Ai-Pentest · api/main.py:1117

    async def connect(self, websocket: WebSocket, agent_id: str, api_key: str = None):
        await websocket.accept()
        self.active_connections[agent_id] = websocket
        self.agent_info[agent_id] = {
            "api_key": api_key,
            "connected_at": datetime.now(timezone.utc).isoformat(),
            "last_seen": datetime.now(timezone.utc).isoformat(),
        }
        logger.info(f"Agent {sanitize_for_log(agent_id)} connected via WebSocket")

    def disconnect(self, agent_id: str):
        if agent_id in self.active_connections:
            del self.active_connections[agent_id]
        if agent_id in self.agent_info:
            del self.agent_info[agent_id]
        logger.info(f"Agent {sanitize_for_log(agent_id)} disconnected")

    async def send_to_agent(self, agent_id: str, message: dict) -> bool:

View source record →