From 41f53d39a0bb20261d55798e7c3e089b8069d7ce Mon Sep 17 00:00:00 2001
From: Aqr-K <95741669+Aqr-K@users.noreply.github.com>
Date: Mon, 23 Mar 2026 04:30:46 +0800
Subject: [PATCH] Fix agent memory (#5607)

---
 app/agent/__init__.py          | 2 +-
 app/agent/middleware/memory.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/app/agent/__init__.py b/app/agent/__init__.py
index 55f2fea9..e6c78ae7 100644
--- a/app/agent/__init__.py
+++ b/app/agent/__init__.py
@@ -13,7 +13,7 @@ from langchain_core.messages import (
 )
 from langgraph.checkpoint.memory import InMemorySaver
 
-from app.agent.callback import StreamingHandler, StreamingHandler
+from app.agent.callback import StreamingHandler
 from app.agent.memory import memory_manager
 from app.agent.middleware.memory import MemoryMiddleware
 from app.agent.middleware.patch_tool_calls import PatchToolCallsMiddleware
diff --git a/app/agent/middleware/memory.py b/app/agent/middleware/memory.py
index 5616314c..d661006d 100644
--- a/app/agent/middleware/memory.py
+++ b/app/agent/middleware/memory.py
@@ -1,7 +1,7 @@
 from collections.abc import Awaitable, Callable
 from typing import Annotated, NotRequired, TypedDict, Dict
 
-from aiopath import AsyncPath
+from aiopathlib import AsyncPath
 from langchain.agents.middleware.types import (
     AgentMiddleware,
     AgentState,