opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-28 20:17:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add MemoryBlock config to LLM's memory config

Browse Source
This commit is contained in:
Stream 2025-08-20 19:53:44 +08:00
parent bbb640c9a2
commit 8341b8b1c1
No known key found for this signature in database
GPG Key ID: 9475891C9507B4F3
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/core/prompt/entities/advanced_prompt_entities.py
View File

@ -45,6 +45,13 @@ class MemoryConfig(BaseModel):
enabled: bool enabled: bool
size: Optional[int] = None 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 role_prefix: Optional[RolePrefix] = None
window: WindowConfig window: WindowConfig
query_prompt_template: Optional[str] = None query_prompt_template: Optional[str] = None
@property
def is_block_mode(self) -> bool:
return self.mode == "block" and bool(self.block_id)