Thêm hàm format_created_at

tools/memory.py

@@ -9,16 +9,9 @@ from typing import Optional
 from pydantic import BaseModel, Field
 
 from .base import register_tool
+from .utils import format_created_at
 from ..redis_client import redis_client
 
-
-def _format_memory_timestamp(ts) -> str:
-    try:
-        dt = datetime.fromtimestamp(int(ts) / 1000, tz=timezone.utc)
-        return dt.strftime("%B %d, %Y, %H:%M:%S UTC")
-    except (ValueError, TypeError):
-        return str(ts)
-
 logger = logging.getLogger(__name__)
 
 @register_tool(
@@ -71,7 +64,11 @@ def tool_get_memories(query: str = None, limit: int = 100) -> dict:
         memories = redis_client.list_memories(query=query, limit=limit)
         for mem in memories:
             if "timestamp" in mem:
-                mem["timestamp"] = _format_memory_timestamp(mem["timestamp"])
+                try:
+                    iso = datetime.fromtimestamp(int(mem["timestamp"]) / 1000, tz=timezone.utc).isoformat()
+                except (ValueError, TypeError):
+                    iso = str(mem["timestamp"])
+                mem["timestamp"] = format_created_at(iso)
         return {
             "status": "success",
             "count": len(memories),

tools/utils.py

@@ -1,6 +1,17 @@
 import re
 import json
 from collections import defaultdict
+from datetime import datetime, timezone
+
+
+def format_created_at(raw: str) -> str:
+    if not raw:
+        return "Unknown time"
+    try:
+        dt = datetime.fromisoformat(raw).astimezone(timezone.utc)
+        return dt.strftime("%B %d, %Y, %H:%M:%S UTC")
+    except (ValueError, TypeError):
+        return raw
 
 def extract_metadata_from_messages(messages: list[dict]) -> dict:
     """
@@ -67,9 +78,9 @@ def format_thread_for_prompt(room_id: str, messages: list[dict]) -> str:
     """
     lines = []
     for msg in messages:
-        # Use senderName as defined in docs/redis-data-structure.md
-        timestamp = msg.get("timestamp", "??:??")
-        user = msg.get("senderName", "Unknown")
+        raw_ts = msg.get("created_at") or msg.get("timestamp", "")
+        timestamp = format_created_at(raw_ts) if raw_ts else "Unknown time"
+        user = msg.get("name") or msg.get("senderName") or msg.get("sender_id", "Unknown")
         content = msg.get("content", "")
         lines.append(f"[{timestamp}] {user}: {content}")