mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-04-29 03:00:45 -04:00
4b4ce20f2d
* Add logging config for the app and for llm debug * - switch to python, add special llm logger - add logging to sandbox.py - add session.py - add a directory per session - small additions for AgentController * - add sys log, but try to exclude litellm; log llm responses as json * Update opendevin/_logging.py Co-authored-by: Anas DORBANI <95044293+dorbanianas@users.noreply.github.com> * - use standard file naming - quick pass through a few more files * fix ruff * clean up * mypy types * make mypy happy --------- Co-authored-by: Anas DORBANI <95044293+dorbanianas@users.noreply.github.com>
75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
import time
|
|
from typing import Dict, Callable
|
|
from fastapi import WebSocket, WebSocketDisconnect
|
|
from .msg_stack import message_stack
|
|
|
|
from opendevin.logging import opendevin_logger as logger
|
|
|
|
DEL_DELT_SEC = 60 * 60 * 5
|
|
|
|
|
|
class Session:
|
|
sid: str
|
|
websocket: WebSocket | None
|
|
last_active_ts: int = 0
|
|
is_alive: bool = True
|
|
|
|
def __init__(self, sid: str, ws: WebSocket | None):
|
|
self.sid = sid
|
|
self.websocket = ws
|
|
self.last_active_ts = int(time.time())
|
|
|
|
async def loop_recv(self, dispatch: Callable):
|
|
try:
|
|
if self.websocket is None:
|
|
return
|
|
while True:
|
|
try:
|
|
data = await self.websocket.receive_json()
|
|
except ValueError:
|
|
await self.send_error("Invalid JSON")
|
|
continue
|
|
|
|
message_stack.add_message(self.sid, "user", data)
|
|
action = data.get("action", None)
|
|
await dispatch(action, data)
|
|
except WebSocketDisconnect:
|
|
self.is_alive = False
|
|
logger.info("WebSocket disconnected, sid: %s", self.sid)
|
|
except RuntimeError as e:
|
|
# WebSocket is not connected
|
|
if "WebSocket is not connected" in str(e):
|
|
self.is_alive = False
|
|
logger.exception("Error in loop_recv: %s", e)
|
|
|
|
async def send(self, data: Dict[str, object]) -> bool:
|
|
try:
|
|
if self.websocket is None or not self.is_alive:
|
|
return False
|
|
await self.websocket.send_json(data)
|
|
self.last_active_ts = int(time.time())
|
|
return True
|
|
except WebSocketDisconnect:
|
|
self.is_alive = False
|
|
return False
|
|
|
|
async def send_error(self, message: str) -> bool:
|
|
"""Sends an error message to the client."""
|
|
return await self.send({"error": True, "message": message})
|
|
|
|
async def send_message(self, message: str) -> bool:
|
|
"""Sends a message to the client."""
|
|
return await self.send({"message": message})
|
|
|
|
def update_connection(self, ws: WebSocket):
|
|
self.websocket = ws
|
|
self.is_alive = True
|
|
self.last_active_ts = int(time.time())
|
|
|
|
def load_from_data(self, data: Dict) -> bool:
|
|
self.last_active_ts = data.get("last_active_ts", 0)
|
|
if self.last_active_ts < int(time.time()) - DEL_DELT_SEC:
|
|
return False
|
|
self.is_alive = data.get("is_alive", False)
|
|
return True
|