mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
35884ef3a7
FastMCP library changes now require calling decorated functions via the .fn attribute: - Tools: @mcp.tool() functions return FunctionTool, call with tool.fn() - Prompts: @mcp.prompt() functions return FunctionPrompt, call with prompt.fn() - Resources: @mcp.resource() functions return FunctionResource, call with resource.fn() Updated core files: - view_note.py: read_note() → read_note.fn() - read_note.py: search_notes() → search_notes.fn() (2 locations) - tool.py: 6 MCP tool calls updated to use .fn - recent_activity.py: recent_activity() → recent_activity.fn() - project.py: project_info() → project_info.fn() with type ignore Updated 100+ test files systematically to use .fn attribute and fixed mock targets. All 869 tests now pass. Fixes view_note tool error in Claude Desktop. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
67 lines
2.4 KiB
Python
67 lines
2.4 KiB
Python
"""View note tool for Basic Memory MCP server."""
|
|
|
|
from textwrap import dedent
|
|
from typing import Optional
|
|
|
|
from loguru import logger
|
|
|
|
from basic_memory.mcp.server import mcp
|
|
from basic_memory.mcp.tools.read_note import read_note
|
|
|
|
|
|
@mcp.tool(
|
|
description="View a note as a formatted artifact for better readability.",
|
|
)
|
|
async def view_note(
|
|
identifier: str, page: int = 1, page_size: int = 10, project: Optional[str] = None
|
|
) -> str:
|
|
"""View a markdown note as a formatted artifact.
|
|
|
|
This tool reads a note using the same logic as read_note but displays the content
|
|
as a markdown artifact for better viewing experience in Claude Desktop.
|
|
|
|
After calling this tool, create an artifact using the returned content to display
|
|
the note in a readable format. The tool returns the note content that should be
|
|
used to create a markdown artifact.
|
|
|
|
Args:
|
|
identifier: The title or permalink of the note to view
|
|
page: Page number for paginated results (default: 1)
|
|
page_size: Number of items per page (default: 10)
|
|
project: Optional project name to read from. If not provided, uses current active project.
|
|
|
|
Returns:
|
|
The note content as a markdown artifact with a confirmation message.
|
|
"""
|
|
|
|
logger.info(f"Viewing note: {identifier}")
|
|
|
|
# Call the existing read_note logic
|
|
content = await read_note.fn(identifier, page, page_size, project)
|
|
|
|
# Check if this is an error message (note not found)
|
|
if "# Note Not Found:" in content:
|
|
return content # Return error message directly instead of creating artifact
|
|
|
|
# Extract title from content if possible
|
|
title = identifier
|
|
lines = content.split("\n")
|
|
for line in lines:
|
|
if line.startswith("title:"):
|
|
# Extract title from frontmatter
|
|
title = line.split("title:", 1)[1].strip().strip('"').strip("'")
|
|
break
|
|
elif line.startswith("# "):
|
|
# Extract title from first heading
|
|
title = line[2:].strip()
|
|
break
|
|
|
|
# Return the artifact XML as part of the response
|
|
artifact = f'<artifact identifier="note-{hash(identifier) & 0x7FFFFFFF}" type="text/markdown" title="{title}">\n{content}\n</artifact>'
|
|
|
|
return dedent(f"""
|
|
<instructions>
|
|
Create an artifact using the returned artifact content to display the note in a readable format.
|
|
</instructions>
|
|
{artifact}\n\n✅ Note displayed as artifact: **{title}**""")
|