Fix: The base64 images are stored in the session history, causing context overflow.

2026-02-25 20:58:59 +08:00
parent 9e806d7159
commit 6aed4265b7
2 changed files with 30 additions and 1 deletions
@@ -427,12 +427,15 @@ class AgentLoop:
    def _save_turn(self, session: Session, messages: list[dict], skip: int) -> None:
        """Save new-turn messages into session, truncating large tool results."""
        from datetime import datetime
+        from nanobot.utils import helpers
        for m in messages[skip:]:
            entry = {k: v for k, v in m.items() if k != "reasoning_content"}
            if entry.get("role") == "tool" and isinstance(entry.get("content"), str):
                content = entry["content"]
                if len(content) > self._TOOL_RESULT_MAX_CHARS:
-                    entry["content"] = content[:self._TOOL_RESULT_MAX_CHARS] + "\n... (truncated)"
+                    entry["content"] = content[: self._TOOL_RESULT_MAX_CHARS] + "\n... (truncated)"
+            if entry.get("role") == "user":
+                entry["content"] = helpers.strip_base64_images(entry["content"])
            entry.setdefault("timestamp", datetime.now().isoformat())
            session.messages.append(entry)
        session.updated_at = datetime.now()