Tạo hàm thêm các tin nhắn đã có vào buffer trên Qdrant

conversation_memory.py

@@ -153,6 +153,51 @@ def add_turn(conversation_id: str, user_msg: str, ai_msg: str) -> None:
     save(conversation_id, summary, buffer)
 
 
+def seed_room(conversation_id: str, messages: list[dict]) -> None:
+    """
+    Seed Qdrant buffer từ danh sách tin nhắn Redis thô.
+    Mỗi message được chuyển thành role='user', content='[ts] name (id): content'.
+    Nếu vượt MAX_BUFFER thì tự động summarize trước khi lưu.
+    """
+    _NAME_FIELDS = ["sender_username", "username", "u_username", "name", "u_name",
+                    "senderName", "displayName", "display_name", "fullName", "sender_id"]
+    _ID_FIELDS   = ["sender_id", "u_id", "userId", "user_id", "sender_username", "username"]
+
+    def _get_field(m: dict, fields: list[str]) -> str:
+        for f in fields:
+            v = m.get(f)
+            if v and str(v).strip():
+                return str(v).strip()
+        return ""
+
+    buffer: list[dict] = []
+    for msg in messages:
+        name    = _get_field(msg, _NAME_FIELDS) or "unknown"
+        sid     = _get_field(msg, _ID_FIELDS) or name
+        content = msg.get("content") or msg.get("text") or msg.get("msg", "")
+        ts      = msg.get("created_at") or msg.get("timestamp", "")
+        label   = f"{name} ({sid})" if sid != name else name
+        buffer.append({"role": "user", "content": f"[{ts}] {label}: {content}"})
+
+    summary = ""
+    while len(buffer) > MAX_BUFFER:
+        to_summarize = buffer[:SUMMARIZE_COUNT]
+        buffer       = buffer[SUMMARIZE_COUNT:]
+        if len(buffer) > KEEP_RECENT:
+            buffer = buffer[-KEEP_RECENT:]
+        logger.info(
+            "[Memory] seed_room: tóm tắt %d tin → giữ %d tin còn lại",
+            len(to_summarize), len(buffer),
+        )
+        summary = _summarize(summary, to_summarize)
+
+    save(conversation_id, summary, buffer)
+    logger.info(
+        "[Memory] seed_room '%s': lưu buffer=%d tin, summary=%s",
+        conversation_id, len(buffer), "có" if summary else "không",
+    )
+
+
 def get_context(conversation_id: str) -> str:
     """Trả về chuỗi context (summary + buffer) để đưa vào prompt."""
     summary, buffer = load(conversation_id)