Fix agent memory (#5607)

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

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,