refac
This commit is contained in:
1 parent
e5b24a22d0
commit
d22bb6703f
1 file changed
+3
-10
@@ -286,13 +286,6 @@ def model_allows_memory(model: dict | None) -> bool:
|
|||||||
return ((model or {}).get('info', {}).get('meta', {}).get('capabilities') or {}).get('memory', True)
|
return ((model or {}).get('info', {}).get('meta', {}).get('capabilities') or {}).get('memory', True)
|
||||||
|
|
||||||
|
|
||||||
def _format_memory_context_section(title: str, memories: list[str]) -> str | None:
|
|
||||||
if not memories:
|
|
||||||
return None
|
|
||||||
ordered = sorted(memories, key=lambda memory: (memory.casefold(), memory))
|
|
||||||
return f'[{title}]\n' + '\n'.join(f'- {memory}' for memory in ordered)
|
|
||||||
|
|
||||||
|
|
||||||
async def add_memory_context(request, form_data: dict, user, model: dict | None = None):
|
async def add_memory_context(request, form_data: dict, user, model: dict | None = None):
|
||||||
if not model_allows_memory(model):
|
if not model_allows_memory(model):
|
||||||
return form_data
|
return form_data
|
||||||
@@ -372,9 +365,9 @@ async def add_memory_context(request, form_data: dict, user, model: dict | None
|
|||||||
('Memory Neighborhood', 'neighborhood'),
|
('Memory Neighborhood', 'neighborhood'),
|
||||||
('Relevant Context', 'context'),
|
('Relevant Context', 'context'),
|
||||||
):
|
):
|
||||||
part = _format_memory_context_section(title, sections[key])
|
if sections[key]:
|
||||||
if part:
|
ordered = sorted(sections[key], key=lambda memory: (memory.casefold(), memory))
|
||||||
parts.append(part)
|
parts.append(f'[{title}]\n' + '\n'.join(f'- {memory}' for memory in ordered))
|
||||||
if not parts:
|
if not parts:
|
||||||
return form_data
|
return form_data
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user