diff --git a/api/core/memory/entities.py b/api/core/memory/entities.py index ccc96de3e4..516ca55c8b 100644 --- a/api/core/memory/entities.py +++ b/api/core/memory/entities.py @@ -70,15 +70,6 @@ class MemoryBlock(BaseModel): node_id: Optional[str] = None -class MemoryBlockWithVisibility(BaseModel): - id: str - name: str - value: str - version: int - end_user_visible: bool - end_user_editable: bool - - class MemoryValueData(BaseModel): value: str diff --git a/api/services/chatflow_memory_service.py b/api/services/chatflow_memory_service.py index ac2da9be8f..0f5d7ae19e 100644 --- a/api/services/chatflow_memory_service.py +++ b/api/services/chatflow_memory_service.py @@ -11,7 +11,6 @@ from core.llm_generator.llm_generator import LLMGenerator from core.memory.entities import ( MemoryBlock, MemoryBlockSpec, - MemoryBlockWithVisibility, MemoryScheduleMode, MemoryScope, MemoryTerm, @@ -376,33 +375,6 @@ class ChatflowMemoryService: ) return results - @staticmethod - def _with_visibility( - app: App, - raw_results: Sequence[ChatflowMemoryVariable] - ) -> Sequence[MemoryBlockWithVisibility]: - workflow = WorkflowService().get_published_workflow(app) - if not workflow: - return [] - results = [] - for chatflow_memory_variable in raw_results: - spec = next( - (spec for spec in workflow.memory_blocks if spec.id == chatflow_memory_variable.memory_id), - None - ) - if spec: - results.append( - MemoryBlockWithVisibility( - id=chatflow_memory_variable.memory_id, - name=chatflow_memory_variable.name, - value=MemoryValueData.model_validate_json(chatflow_memory_variable.value).value, - end_user_editable=spec.end_user_editable, - end_user_visible=spec.end_user_visible, - version=chatflow_memory_variable.version - ) - ) - return results - @staticmethod def _should_update_memory( memory_block: MemoryBlock,