aji/hendrik
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixing new session error when running multiple hendrik

Browse Source
This commit is contained in:
Dita Aji Pratama 2026-06-19 13:31:02 +07:00
parent dd4c6c674d
commit d54fe171fc
1 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
services/session_manager_neo.py
View File

@ -30,7 +30,7 @@ class NeoSessionManager:
self._db = TinyDB(db_path)
self._table = self._db.table("sessions")
def create(self, name: str, model_info: dict) -> NeoSession:
def create(self, name: str, model_info: dict, max_retries: int = 5) -> NeoSession:
now = datetime.now(timezone.utc).isoformat()
doc = {
"session_id": str(uuid.uuid4()),
@ -40,8 +40,24 @@ class NeoSessionManager:
"model_info": model_info,
"messages": [],
}
doc_id = self._table.insert(doc)
return self._doc_to_session(self._table.get(doc_id=doc_id))
import time
import random
for attempt in range(max_retries):
try:
doc_id = self._table.insert(doc)
return self._doc_to_session(self._table.get(doc_id=doc_id))
except ValueError:
if attempt < max_retries - 1:
# Exponential backoff dengan jitter
wait_time = (0.1 * (2 ** attempt)) + random.uniform(0, 0.05)
time.sleep(wait_time)
# Reload DB untuk sinkronisasi
self._db = TinyDB(self._db._root._path)
self._table = self._db.table("sessions")
else:
raise
def list(self) -> list[dict]:
results = []