From 8515af0528781377f780048262e31fb49e59587d Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Wed, 24 Jul 2024 21:25:13 +0800 Subject: [PATCH] fix useless params --- .../memory/worker/summary/get_observation_with_time_worker.py | 2 +- memoryscope/memory/worker/summary/load_memory_worker.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/memoryscope/memory/worker/summary/get_observation_with_time_worker.py b/memoryscope/memory/worker/summary/get_observation_with_time_worker.py index 18a77380..3b451a56 100644 --- a/memoryscope/memory/worker/summary/get_observation_with_time_worker.py +++ b/memoryscope/memory/worker/summary/get_observation_with_time_worker.py @@ -2,7 +2,7 @@ from typing import List from memoryscope.constants.common_constants import NEW_OBS_WITH_TIME_NODES from memoryscope.constants.language_constants import COLON_WORD -from memoryscope.memory.worker.write.get_observation_worker import GetObservationWorker +from memoryscope.memory.worker.summary.get_observation_worker import GetObservationWorker from memoryscope.scheme.message import Message from memoryscope.utils.datetime_handler import DatetimeHandler from memoryscope.utils.tool_functions import prompt_to_msg diff --git a/memoryscope/memory/worker/summary/load_memory_worker.py b/memoryscope/memory/worker/summary/load_memory_worker.py index 24972900..1f5e4233 100644 --- a/memoryscope/memory/worker/summary/load_memory_worker.py +++ b/memoryscope/memory/worker/summary/load_memory_worker.py @@ -55,7 +55,7 @@ class LoadMemoryWorker(MemoryBaseWorker): self.memory_handler.set_memories(NOT_UPDATED_NODES, nodes) @timer - def retrieve_insight_memory(self, query: str): + def retrieve_insight_memory(self): """ Retrieves top-K insight memories based on the query and stores them in the memory handler. """