fix(memory): remove max_tokens cap in deep dream distillation

This commit is contained in:
zhayujie
2026-06-14 11:06:25 +08:00
parent 047fb57630
commit 6538843bdf

View File

@@ -462,13 +462,12 @@ class MemoryFlushManager:
daily_content=daily_content or "(no recent daily records)", daily_content=daily_content or "(no recent daily records)",
) )
from agent.protocol.models import LLMRequest from agent.protocol.models import LLMRequest
# Scale max_tokens based on input size to avoid truncating large MEMORY.md # No output cap: the prompt already keeps MEMORY.md concise (~50
input_chars = len(memory_content) + len(daily_content) # items), so a hard max_tokens would only risk truncating a large
dream_max_tokens = max(2000, min(input_chars, 8000)) # rewrite. Let the model use its default output budget.
request = LLMRequest( request = LLMRequest(
messages=[{"role": "user", "content": user_msg}], messages=[{"role": "user", "content": user_msg}],
temperature=0.3, temperature=0.3,
max_tokens=dream_max_tokens,
stream=False, stream=False,
system=_dream_system_prompt(), system=_dream_system_prompt(),
) )