feat: add websocket server endpoint for theta-agent

This commit is contained in:
2026-08-02 01:39:45 -04:00
parent 36aa114d7c
commit b948cd8625
5 changed files with 71 additions and 0 deletions
+13
View File
@@ -25,6 +25,19 @@ var server = http.createServer(app);
var io = require('socket.io')(server);
app.io = io;
const WebSocket = require('ws');
const wss = new WebSocket.Server({ noServer: true });
server.on('upgrade', (request, socket, head) => {
// We only handle upgrade for /api/agent/ws.
// Socket.IO handles its own upgrades natively because it attaches directly to `server`.
if (request.url.startsWith('/api/agent/ws')) {
wss.handleUpgrade(request, socket, head, (ws) => {
wss.emit('connection', ws, request);
});
}
});
app.wss = wss;
const models = require('../models');
/**