Initial clean commit - unified Lyra stack
This commit is contained in:
61
cortex/context.py
Normal file
61
cortex/context.py
Normal file
@@ -0,0 +1,61 @@
|
||||
import os, requests, datetime
|
||||
from typing import Dict, Any, Tuple
|
||||
|
||||
INTAKE_API_URL = os.getenv("INTAKE_API_URL", "http://intake:7080")
|
||||
DEFAULT_SESSION_ID = os.getenv("DEFAULT_SESSION_ID", "default")
|
||||
LOCAL_TZ_LABEL = os.getenv("LOCAL_TZ_LABEL", "America/New_York")
|
||||
|
||||
def fetch_intake_context(session_id: str | None) -> Dict[str, Any]:
|
||||
sid = session_id or DEFAULT_SESSION_ID
|
||||
try:
|
||||
r = requests.get(f"{INTAKE_API_URL}/summaries", params={"session_id": sid}, timeout=4)
|
||||
r.raise_for_status()
|
||||
data = r.json() or {}
|
||||
except Exception:
|
||||
data = {}
|
||||
# Normalize expected fields
|
||||
return {
|
||||
"summary_text": data.get("summary_text", ""),
|
||||
"last_message_ts": data.get("last_message_ts"), # ISO8601 or None
|
||||
"session_id": sid,
|
||||
"exchange_count": data.get("exchange_count", 0),
|
||||
}
|
||||
|
||||
def build_temporal_snapshot(last_ts_iso: str | None) -> Dict[str, Any]:
|
||||
now = datetime.datetime.now() # system local time
|
||||
now_str = now.strftime("%A, %b %-d, %Y, %H:%M")
|
||||
elapsed_str = "unknown"
|
||||
if last_ts_iso:
|
||||
try:
|
||||
# parse ISO (with/without tz). If it has a timezone offset, fromisoformat handles it.
|
||||
last = datetime.datetime.fromisoformat(last_ts_iso.replace("Z", "+00:00"))
|
||||
delta = now - last.replace(tzinfo=None)
|
||||
mins = int(delta.total_seconds() // 60)
|
||||
if mins < 60:
|
||||
elapsed_str = f"{mins} min"
|
||||
else:
|
||||
hrs = mins // 60
|
||||
rem = mins % 60
|
||||
elapsed_str = f"{hrs} hr {rem} min"
|
||||
except Exception:
|
||||
pass
|
||||
return {
|
||||
"local_time_label": LOCAL_TZ_LABEL,
|
||||
"local_time_now": now_str,
|
||||
"elapsed_since_last": elapsed_str,
|
||||
}
|
||||
|
||||
def get_intake_block(session_id: str | None) -> Tuple[str, Dict[str, Any]]:
|
||||
ctx = fetch_intake_context(session_id)
|
||||
temporal = build_temporal_snapshot(ctx.get("last_message_ts"))
|
||||
# A short, ready-to-inject block for prompts:
|
||||
intake_block = (
|
||||
f"[Intake]\n"
|
||||
f"Session: {ctx['session_id']}\n"
|
||||
f"Exchanges: {ctx['exchange_count']}\n"
|
||||
f"Local time ({temporal['local_time_label']}): {temporal['local_time_now']}\n"
|
||||
f"Elapsed since last: {temporal['elapsed_since_last']}\n"
|
||||
f"Recent summary: {ctx['summary_text'] or '(none)'}\n"
|
||||
)
|
||||
# Also return raw dicts if you want to use fields programmatically
|
||||
return intake_block, {"intake": ctx, "temporal": temporal}
|
||||
Reference in New Issue
Block a user