Skip to content

Commit 36ba964

Browse files
committed
fix: change the place of file
1 parent f643eb4 commit 36ba964

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

veadk/agent.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -268,20 +268,22 @@ def model_post_init(self, __context: Any) -> None:
268268
"auto_save_session is enabled, but long_term_memory is not initialized."
269269
)
270270
else:
271-
from veadk.tools.builtin_tools.save_session import (
272-
save_session_to_memory,
271+
from veadk.memory.save_session_callback import (
272+
save_session_to_long_term_memory,
273273
)
274274

275275
if self.after_agent_callback:
276276
if isinstance(self.after_agent_callback, list):
277-
self.after_agent_callback.append(save_session_to_memory)
277+
self.after_agent_callback.append(
278+
save_session_to_long_term_memory
279+
)
278280
else:
279281
self.after_agent_callback = [
280282
self.after_agent_callback,
281-
save_session_to_memory,
283+
save_session_to_long_term_memory,
282284
]
283285
else:
284-
self.after_agent_callback = save_session_to_memory
286+
self.after_agent_callback = save_session_to_long_term_memory
285287

286288
logger.info(f"VeADK version: {VERSION}")
287289

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636
) # Minimum seconds between saves (1 minute)
3737

3838

39-
async def save_session_to_memory(
39+
async def save_session_to_long_term_memory(
4040
callback_context: CallbackContext,
4141
) -> None:
4242
"""Save the current session to long-term memory.

0 commit comments

Comments
 (0)