Merge pull request #20 from insightbuilder/notion-mcp-bd

got the tools working in client and inspector

Author: insightbuilder

docs/mcp_notion_brain.md

@@ -25,6 +25,8 @@ that support python.
 
 ```bash
 cd notion_brain_dump
+# pyproject.toml will contain the libraries that are used in the project
+
 uv run mcpclient.py server.py
 
 After the client start, you will be prompted for the query.
@@ -36,10 +38,10 @@ Query:  Provide me the list of the tasks available.
 
 - Add Task
 - List Tasks
-- Update Tasks
+- Change Title
 - Remove Tasks
-- Add Area
-- Add Resource
+- Read Page Content
+- Add Page Content
 
 These tool are using mcp resources that are
 connecting with the notion db.

notion_brain_dump/server.py

@@ -1,16 +1,17 @@
 from mcp.server.fastmcp import FastMCP
 from notion_client import AsyncClient
 import os
-from typing import List
 
-token = "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"
+token = "ntn_token"
 mcp = FastMCP("notion_bd_server")
 brain = AsyncClient(auth=token)
 
+# The server has tools for doing CRUD operations on the Brain Dump database
+
 
 @mcp.tool()
 async def add_task(task: str):
-    """Add a task to the Notion database and returns the new page ID"""
+    """Add a task to the Notion database and return the new page ID"""
     dump_db = "1d784ade96ac80a3b7ecf54f3eae5f49"
     newpage = {
         "TaskTitle": {"title": [{"text": {"content": task}}]},
@@ -58,6 +59,37 @@ async def get_all_database_items():
     return text_conv
 
 
+@mcp.resource("notion://dumpdb/{task_text}")
+async def search_db(task_text: str) -> str:
+    """Get the filtered task from the Notion database and returns the data as text"""
+    dump_db = "1d784ade96ac80a3b7ecf54f3eae5f49"
+    search_results = await brain.databases.query(
+        database_id=dump_db,
+        filter={
+            "property": "TaskTitle",
+            "title": {
+                "equals": task_text,
+            },
+        },
+    )
+    results = search_results.get("results", [])
+    text_conv = ""
+    for result in results:
+        page_id = result["id"]
+        created_time = result["created_time"]
+        last_edited_time = result["last_edited_time"]
+        # getting the properties now
+        task_title = result["properties"]["TaskTitle"]["title"][0]["text"]["content"]
+        task_status = result["properties"]["TaskStatus"]["select"]
+        area = result["properties"]["Area"]["rich_text"]
+        resource = result["properties"]["Resource"]["rich_text"]
+        due_date = result["properties"]["DueDate"]["date"]
+        url = result["url"]
+        text_conv += f"Task: {task_title}\t Status: {task_status}\t Area: {area}\t Resource: {resource}\t Due Date: {due_date}\t URL: {url}\t Page ID: {page_id}\t  Last Edited Time: {last_edited_time}\t "
+    # return results[0]["text"]
+    return text_conv
+
+
 @mcp.tool()
 async def list_task():
     """Lists the tasks in the Notion database as text"""
@@ -85,8 +117,8 @@ async def remove_task(task_text: str):
 
 
 @mcp.tool()
-async def add_resource(task_text: str, resource_text: str):
-    """Search for the task from the Notion database and add resource to it"""
+async def change_title(task_title: str, new_title: str):
+    """Change title of the task"""
 
     dump_db = "1d784ade96ac80a3b7ecf54f3eae5f49"
 
@@ -95,7 +127,7 @@ async def add_resource(task_text: str, resource_text: str):
         filter={
             "property": "TaskTitle",
             "title": {
-                "equals": task_text,
+                "equals": task_title,
             },
         },
     )
@@ -104,104 +136,51 @@ async def add_resource(task_text: str, resource_text: str):
     page_id = results[0]["id"]
     prop = {}
 
-    prop["Resource"] = {
-        "type": "rich_text",
-        "rich_text": [{"text": {"content": resource_text}}],
+    prop["TaskTitle"] = {
+        "type": "title",
+        "title": [{"text": {"content": new_title}}],
     }
 
     updt = await brain.pages.update(page_id=page_id, properties=prop)
-    return f"Updated the resource to {updt['id']}"
+    return f"Updated the title of the page {updt['id']}"
 
 
 @mcp.tool()
-async def add_area(task_text: str, area_text: str):
-    """Search for the task from the Notion database and add area to it"""
-
+async def add_page_content(task_title: str, task_data: str):
+    """Append the task data into the page of given task title"""
+    # First find the task and get its page id
     dump_db = "1d784ade96ac80a3b7ecf54f3eae5f49"
 
     search_results = await brain.databases.query(
         database_id=dump_db,
         filter={
             "property": "TaskTitle",
             "title": {
-                "equals": task_text,
+                "equals": task_title,
             },
         },
     )
     results = search_results.get("results", [])
-
     page_id = results[0]["id"]
-    prop = {}
-
-    if area_text:
-        prop["Area"] = {
-            "type": "rich_text",
-            "rich_text": [{"text": {"content": area_text}}],
-        }
-
-    updt = await brain.pages.update(page_id=page_id, properties=prop)
-    return f"Updated the area to {updt['id']}"
-
-
-@mcp.prompt()
-def task_steps(task_text: str) -> str:
-    """Prompt to get the steps to complete the tasks"""
-    return f"Provide the step by approach required to complete the {task_text}"
-
-
-@mcp.prompt()
-def task_analysis(tasks: str) -> str:
-    """Prompt to analyse the tasks"""
-    return f"Analyse the {tasks} and provide the steps to complete the task"
-
-
-@mcp.resource("notion://dumpdb/{task_text}")
-async def search_db(task_text: str) -> str:
-    """Get the filtered task from the Notion database and returns the data as text"""
-    dump_db = "1d784ade96ac80a3b7ecf54f3eae5f49"
-    search_results = await brain.databases.query(
-        database_id=dump_db,
-        filter={
-            "property": "TaskTitle",
-            "title": {
-                "contains": task_text,
-            },
-        },
-    )
-    results = search_results.get("results", [])
-    text_conv = ""
-    for result in results:
-        page_id = result["id"]
-        created_time = result["created_time"]
-        last_edited_time = result["last_edited_time"]
-        # getting the properties now
-        task_title = result["properties"]["TaskTitle"]["title"][0]["text"]["content"]
-        task_status = result["properties"]["TaskStatus"]["select"]
-        area = result["properties"]["Area"]["rich_text"]
-        resource = result["properties"]["Resource"]["rich_text"]
-        due_date = result["properties"]["DueDate"]["date"]
-        url = result["url"]
-        text_conv += f"Task: {task_title}\t Status: {task_status}\t Area: {area}\t Resource: {resource}\t Due Date: {due_date}\t URL: {url}\t Page ID: {page_id}\t  Last Edited Time: {last_edited_time}\t "
-    # return results[0]["text"]
-    return text_conv
 
+    # Then update the data into the page using append method
+    page_data = {
+        "object": "block",
+        "type": "paragraph",
+        "paragraph": {"rich_text": [{"text": {"content": task_data}}]},
+    }
 
-# @mcp.tool()
-async def analyse_tasks(task_kw: str) -> str:
-    """Searches the database for the task, and gets the steps to complete the task"""
-    task_info = await mcp.read_resource(f"notion://dumpdb/{task_kw}")
-    assembled_prompt = await mcp.get_prompt(
-        "task_analysis",
-        arguments={
-            "tasks": task_info[0].content,
-        },
+    await brain.blocks.children.append(
+        block_id=page_id,
+        children=[page_data],
     )
-    return assembled_prompt.messages[0].content.text
+    # Inform the LLM that the work has been done
+    return f"The content has been added to {page_id}"
 
 
 @mcp.tool()
-async def update_task(task_title: str, task_data: str):
-    """Update the task in the database with given task title"""
+async def read_page_content(task_title: str):
+    """Append the task data into the page of given task title"""
     # First find the task and get its page id
     dump_db = "1d784ade96ac80a3b7ecf54f3eae5f49"
 
@@ -216,23 +195,14 @@ async def update_task(task_title: str, task_data: str):
     )
     results = search_results.get("results", [])
     page_id = results[0]["id"]
-
-    # Then update the data into the page using append method
-    page_data = {
-        "object": "block",
-        "type": "paragraph",
-        "paragraph": {"rich_text": [{"text": {"content": task_data}}]},
-    }
-
-    await brain.blocks.children.append(
-        block_id=page_id,
-        children=[page_data],
-    )
-    # Inform the LLM that the work has been done
-    return f"The page with {page_id} has been updated"
+    # getting the page blocks
+    page_blocks = await brain.blocks.children.list(block_id=page_id)
+    page_text = ""
+    for block in page_blocks["results"]:
+        page_text += block["paragraph"]["rich_text"][0]["text"]["content"]
+    return f"The content of the page is: {page_text}"
 
 
-# "TaskTitle": {"id": "title", "type": "title", "title": [{"type": "text", "text": {"content": "test", "link": null}, "annotations": {"bold": false, "italic": false, "strikethrough": false, "underline": false, "code": false, "color": "default"}, "plain_text": "test", "href": null}] [ "properties": {"TaskStatus": {"id": "IJax", "type": "select", "select": null}, "Area": {"id": "c%3C%60%7C", "type": "rich_text", "rich_text": []}, "DueDate": {"id": "ce%3DU", "type": "date", "date": null}, "Resource": {"id": "kUxZ", "type": "rich_text", "rich_text": []}, }}, "url": "https://www.notion.so/test-1d784ade96ac81a38f7bdd1eccaa13c4", "public_url": null}]',]
 if __name__ == "__main__":
     print("Starting MCP server...")
     mcp.run()

notion_brain_dump/support_mod.py

@@ -0,0 +1,57 @@
+@mcp.tool()
+async def add_resource(task_text: str, resource_text: str):
+    """Search for the task from the Notion database and add resource to it"""
+
+    dump_db = "1d784ade96ac80a3b7ecf54f3eae5f49"
+
+    search_results = await brain.databases.query(
+        database_id=dump_db,
+        filter={
+            "property": "TaskTitle",
+            "title": {
+                "equals": task_text,
+            },
+        },
+    )
+    results = search_results.get("results", [])
+
+    page_id = results[0]["id"]
+    prop = {}
+
+    prop["Resource"] = {
+        "type": "rich_text",
+        "rich_text": [{"text": {"content": resource_text}}],
+    }
+
+    updt = await brain.pages.update(page_id=page_id, properties=prop)
+    return f"Updated the resource to {updt['id']}"
+
+
+@mcp.tool()
+async def add_area(task_text: str, area_text: str):
+    """Search for the task from the Notion database and add area to it"""
+
+    dump_db = "1d784ade96ac80a3b7ecf54f3eae5f49"
+
+    search_results = await brain.databases.query(
+        database_id=dump_db,
+        filter={
+            "property": "TaskTitle",
+            "title": {
+                "equals": task_text,
+            },
+        },
+    )
+    results = search_results.get("results", [])
+
+    page_id = results[0]["id"]
+    prop = {}
+
+    if area_text:
+        prop["Area"] = {
+            "type": "rich_text",
+            "rich_text": [{"text": {"content": area_text}}],
+        }
+
+    updt = await brain.pages.update(page_id=page_id, properties=prop)
+    return f"Updated the area to {updt['id']}"