Spaces:
Running
Running
File size: 2,104 Bytes
c3674d7 | 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 | """
In-memory fallback stores used when Firestore credentials are not configured.
Data lives only for the lifetime of the process — suitable for demo/conference use.
"""
from typing import Any, Dict, Optional
from datetime import datetime
from .session_store import SessionStore
# Module-level dicts shared across all store instances (process-scoped)
_sessions: Dict[str, Dict[str, Any]] = {}
_conversations: Dict[str, Dict[str, Any]] = {}
class MemorySessionStore(SessionStore):
async def create_session(self, session_id: str, session_data: Dict[str, Any]) -> Dict[str, Any]:
now = datetime.now().isoformat()
record = {**session_data, "id": session_id, "created_at": now, "last_activity": now}
_sessions[session_id] = record
return record
async def get_session(self, session_id: str) -> Optional[Dict[str, Any]]:
return _sessions.get(session_id)
async def update_session(self, session_id: str, updates: Dict[str, Any]) -> bool:
existing = _sessions.get(session_id, {})
existing.update(updates)
existing["last_activity"] = datetime.now().isoformat()
_sessions[session_id] = existing
return True
async def delete_session(self, session_id: str) -> bool:
_sessions.pop(session_id, None)
return True
class MemoryConversationStore:
async def create_conversation(self, session_id: str, data: Dict[str, Any]) -> Dict[str, Any]:
now = datetime.now().isoformat()
record = {**data, "session_id": session_id, "created_at": now, "updated_at": now}
_conversations[session_id] = record
return record
async def get_conversation(self, session_id: str) -> Optional[Dict[str, Any]]:
return _conversations.get(session_id)
async def update_conversation(self, session_id: str, updates: Dict[str, Any]) -> bool:
existing = _conversations.get(session_id, {"session_id": session_id})
existing.update(updates)
existing["updated_at"] = datetime.now().isoformat()
_conversations[session_id] = existing
return True
|