From 8341b8b1c1c77dab6e2cc4ca2df3efa60755cb90 Mon Sep 17 00:00:00 2001 From: Stream Date: Wed, 20 Aug 2025 19:53:44 +0800 Subject: [PATCH] feat: add MemoryBlock config to LLM's memory config --- api/core/prompt/entities/advanced_prompt_entities.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/api/core/prompt/entities/advanced_prompt_entities.py b/api/core/prompt/entities/advanced_prompt_entities.py index c8e7b414df..ba7fee8ada 100644 --- a/api/core/prompt/entities/advanced_prompt_entities.py +++ b/api/core/prompt/entities/advanced_prompt_entities.py @@ -45,6 +45,13 @@ class MemoryConfig(BaseModel): enabled: bool size: Optional[int] = None + mode: Optional[Literal["linear", "block"]] = "linear" + block_id: Optional[list[str]] = None # available only in block mode + role_prefix: Optional[RolePrefix] = None window: WindowConfig query_prompt_template: Optional[str] = None + + @property + def is_block_mode(self) -> bool: + return self.mode == "block" and bool(self.block_id)