Spaces:
Sleeping
Sleeping
File size: 2,518 Bytes
d309047 f8ee24f d309047 f8ee24f d309047 f8ee24f d309047 f8ee24f d309047 f8ee24f d309047 f8ee24f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 | import json
import os
from typing import Optional
# Try to use Redis if available, otherwise fall back to in-memory store
_client = None
_in_memory_store: dict = {}
_use_redis = False
def _init_redis():
global _client, _use_redis
redis_url = os.getenv("REDIS_URL", "")
if not redis_url:
print("No REDIS_URL set — using in-memory session store.")
return
try:
import redis
_client = redis.from_url(redis_url, decode_responses=True, socket_connect_timeout=3)
_client.ping()
_use_redis = True
print(f"Connected to Redis at {redis_url[:30]}...")
except Exception as e:
print(f"Redis unavailable ({e}) — falling back to in-memory store.")
_client = None
_use_redis = False
_init_redis()
SESSION_TTL = 60 * 60 * 4 # 4 hours
def create_session(session_id: str) -> dict:
session = {
"id": session_id,
"transcript": [],
"summaries": [],
"action_items": [],
"decisions": [],
"open_questions": [],
"vectors": [],
"active": True,
}
if _use_redis:
_client.setex(f"session:{session_id}", SESSION_TTL, json.dumps(session))
else:
_in_memory_store[session_id] = session
return session
def get_session(session_id: str) -> Optional[dict]:
if _use_redis:
data = _client.get(f"session:{session_id}")
return json.loads(data) if data else None
else:
return _in_memory_store.get(session_id)
def update_session(session_id: str, session: dict):
if _use_redis:
_client.setex(f"session:{session_id}", SESSION_TTL, json.dumps(session))
else:
_in_memory_store[session_id] = session
def append_transcript(session_id: str, entry: dict):
session = get_session(session_id)
if not session:
return
session["transcript"].append(entry)
update_session(session_id, session)
def append_summary(session_id: str, summary: dict):
session = get_session(session_id)
if not session:
return
session["summaries"].append(summary)
update_session(session_id, session)
def append_action_item(session_id: str, item: dict):
session = get_session(session_id)
if not session:
return
session["action_items"].append(item)
update_session(session_id, session)
def close_session(session_id: str):
session = get_session(session_id)
if session:
session["active"] = False
update_session(session_id, session)
|