From 4fbfb90737bfd3ce0e9dd275eebd8e32ce8e74bb Mon Sep 17 00:00:00 2001
From: BV003 <2022302111185@whu.edu.cn>
Date: Thu, 9 Oct 2025 10:40:11 +0800
Subject: [PATCH] fix: restore compatibility with latest agno version

---
 mcp_ai_agents/notion_mcp_agent/README.md           |  2 +-
 mcp_ai_agents/notion_mcp_agent/notion_mcp_agent.py | 10 +++++-----
 mcp_ai_agents/notion_mcp_agent/requirements.txt    |  1 +
 3 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/mcp_ai_agents/notion_mcp_agent/README.md b/mcp_ai_agents/notion_mcp_agent/README.md
index 861465f..d43ca7d 100644
--- a/mcp_ai_agents/notion_mcp_agent/README.md
+++ b/mcp_ai_agents/notion_mcp_agent/README.md
@@ -14,7 +14,7 @@ A terminal-based Notion Agent for interacting with your Notion pages using natur
 
 ## Prerequisites
 
-- Python 3.9+
+- Python 3.10+
 - A Notion account with admin permissions
 - A Notion Integration token
 - An OpenAI API key
diff --git a/mcp_ai_agents/notion_mcp_agent/notion_mcp_agent.py b/mcp_ai_agents/notion_mcp_agent/notion_mcp_agent.py
index a2ee2c3..6fc1a5d 100644
--- a/mcp_ai_agents/notion_mcp_agent/notion_mcp_agent.py
+++ b/mcp_ai_agents/notion_mcp_agent/notion_mcp_agent.py
@@ -7,7 +7,7 @@ from textwrap import dedent
 from agno.agent import Agent 
 from agno.models.openai import OpenAIChat
 from agno.tools.mcp import MCPTools 
-from agno.memory.v2 import Memory
+from agno.db.sqlite import SqliteDb
 from mcp import StdioServerParameters
 from dotenv import load_dotenv
 
@@ -68,7 +68,7 @@ async def main():
     # Start the MCP Tools session
     async with MCPTools(server_params=server_params) as mcp_tools:
         print("Connected to Notion MCP server successfully!")
-        
+        db = SqliteDb(db_file="agno.db") # SQLite DB for memory
         # Create the agent
         agent = Agent(
             name="NotionDocsAgent",
@@ -97,10 +97,10 @@ async def main():
                 The user's current page ID is: {page_id}
             """),
             markdown=True,
-            show_tool_calls=True,
             retries=3,
-            memory=Memory(),  # Use Memory v2 for better multi-session support
-            add_history_to_messages=True,  # Include conversation history
+            db=db,
+            enable_user_memories=True, # This enables Memory for the Agent
+            add_history_to_context=True,  # Include conversation history
             num_history_runs=5,  # Keep track of the last 5 interactions
         )
         
diff --git a/mcp_ai_agents/notion_mcp_agent/requirements.txt b/mcp_ai_agents/notion_mcp_agent/requirements.txt
index 791b9d8..960c41d 100644
--- a/mcp_ai_agents/notion_mcp_agent/requirements.txt
+++ b/mcp_ai_agents/notion_mcp_agent/requirements.txt
@@ -2,3 +2,4 @@ agno
 python-dotenv
 mcp
 openai
+sqlalchemy
\ No newline at end of file