from langchain_core.messages import HumanMessage

from deerflow.agents.memory.thread_prompt import build_thread_memory_prompt, format_thread_memory_for_injection


def test_thread_memory_injection_keeps_profile_and_preferences_under_small_budget(monkeypatch):
    monkeypatch.setattr("deerflow.agents.memory.thread_prompt._count_tokens", lambda text, encoding_name="cl100k_base": len(text))
    memory = {
        "user": {
            "workContext": {"summary": "Building APIs", "updatedAt": "2026-05-08T00:00:00Z"},
            "personalContext": {"summary": "Engineer using Python and React", "updatedAt": "2026-05-08T00:00:00Z"},
            "topOfMind": {"summary": "Improving thread memory", "updatedAt": "2026-05-08T00:00:00Z"},
        },
        "history": {
            "recentMonths": {"summary": "Shipped memory features", "updatedAt": "2026-05-08T00:00:00Z"},
            "earlierContext": {"summary": "Started from TS projects", "updatedAt": "2026-05-08T00:00:00Z"},
            "longTermBackground": {"summary": "Frontend developer", "updatedAt": "2026-05-08T00:00:00Z"},
        },
        "facts": [
            {"content": "Fact one that might be trimmed", "category": "context", "confidence": 0.9},
            {"content": "Fact two that might be trimmed", "category": "context", "confidence": 0.8},
        ],
    }

    result = format_thread_memory_for_injection(memory, max_tokens=140)
    assert "User:" in result
    assert "History:" in result


def test_build_thread_memory_prompt_does_not_raise_format_key_error():
    prompt = build_thread_memory_prompt(
        {"user": {}, "history": {}, "facts": []},
        [HumanMessage(content="My name is Alice.")],
    )
    assert "Current per-thread memory" in prompt
    assert '"user"' in prompt