feat: extend telemetry smoke workflow

Signed-off-by: phernandez <paul@basicmachines.co>

Author: phernandez

docs/logfire-instrumentation-strategy.md

@@ -421,6 +421,9 @@ What this does:
 - runs a small CLI workflow:
   - `project add`
   - `tool write-note`
+  - `tool read-note`
+  - `tool edit-note`
+  - `tool build-context`
   - `tool search-notes`
   - `doctor`
 
@@ -445,6 +448,9 @@ You should see a small set of comparable root spans rather than a framework-gene
 - `cli.command.project`
 - `cli.command.tool`
 - `mcp.tool.write_note`
+- `mcp.tool.read_note`
+- `mcp.tool.edit_note`
+- `mcp.tool.build_context`
 - `mcp.tool.search_notes`
 - `sync.project.run`
 

justfile

@@ -238,14 +238,17 @@ telemetry-smoke:
     echo "  project_path=$TMP_PROJECT"
     ./.venv/bin/python -m basic_memory.cli.main project add telemetry-smoke "$TMP_PROJECT" --default --local
     ./.venv/bin/python -m basic_memory.cli.main tool write-note --title "Telemetry Smoke" --folder notes --content "hello from smoke" --project telemetry-smoke --local
+    ./.venv/bin/python -m basic_memory.cli.main tool read-note notes/telemetry-smoke --project telemetry-smoke --local
+    ./.venv/bin/python -m basic_memory.cli.main tool edit-note notes/telemetry-smoke --operation append --content $'\n\nsmoke edit line' --project telemetry-smoke --local
+    ./.venv/bin/python -m basic_memory.cli.main tool build-context notes/telemetry-smoke --project telemetry-smoke --local --page-size 5 --max-related 5
     ./.venv/bin/python -m basic_memory.cli.main tool search-notes telemetry --project telemetry-smoke --local
     ./.venv/bin/python -m basic_memory.cli.main doctor --local
     echo ""
     echo "Telemetry smoke complete."
     echo "Search Logfire for:"
     echo "  service_name: basic-memory-cli"
     echo "  environment: $BASIC_MEMORY_LOGFIRE_ENVIRONMENT"
-    echo "  span names: mcp.tool.write_note, mcp.tool.search_notes, sync.project.run"
+    echo "  span names: mcp.tool.write_note, mcp.tool.read_note, mcp.tool.edit_note, mcp.tool.build_context, mcp.tool.search_notes, sync.project.run"
 
 
 # Update all dependencies to latest versions

src/basic_memory/mcp/tools/build_context.py

@@ -6,6 +6,7 @@
 from fastmcp import Context
 
 from basic_memory.config import ConfigManager
+from basic_memory import telemetry
 from basic_memory.mcp.project_context import (
     detect_project_from_url_prefix,
     get_project_client,
@@ -190,8 +191,6 @@ async def build_context(
         if detected:
             project = detected
 
-    logger.info(f"Building context from {url} in project {project}")
-
     # Convert string depth to integer if needed
     if isinstance(depth, str):
         try:
@@ -203,25 +202,52 @@ async def build_context(
 
     # URL is already validated and normalized by MemoryUrl type annotation
 
-    async with get_project_client(project, workspace, context) as (client, active_project):
-        # Resolve memory:// identifier with project-prefix awareness
-        _, resolved_path, _ = await resolve_project_and_path(client, url, project, context)
-
-        # Import here to avoid circular import
-        from basic_memory.mcp.clients import MemoryClient
-
-        # Use typed MemoryClient for API calls
-        memory_client = MemoryClient(client, active_project.external_id)
-        graph = await memory_client.build_context(
-            resolved_path,
-            depth=depth or 1,
-            timeframe=timeframe,
-            page=page,
-            page_size=page_size,
-            max_related=max_related,
-        )
-
-        if output_format == "text":
-            return _format_context_markdown(graph, active_project.name)
-
-        return graph.model_dump()
+    with telemetry.operation(
+        "mcp.tool.build_context",
+        entrypoint="mcp",
+        tool_name="build_context",
+        requested_project=project,
+        workspace_id=workspace,
+        depth=depth or 1,
+        timeframe=timeframe,
+        page=page,
+        page_size=page_size,
+        max_related=max_related,
+        output_format=output_format,
+        is_memory_url=str(url).startswith("memory://"),
+    ):
+        async with get_project_client(project, workspace, context) as (client, active_project):
+            with telemetry.contextualize(
+                project_name=active_project.name,
+                workspace_id=workspace,
+                tool_name="build_context",
+            ):
+                logger.info(
+                    "Building context",
+                    url=str(url),
+                    project=active_project.name,
+                    depth=depth,
+                    timeframe=timeframe,
+                )
+
+                # Resolve memory:// identifier with project-prefix awareness
+                _, resolved_path, _ = await resolve_project_and_path(client, url, project, context)
+
+                # Import here to avoid circular import
+                from basic_memory.mcp.clients import MemoryClient
+
+                # Use typed MemoryClient for API calls
+                memory_client = MemoryClient(client, active_project.external_id)
+                graph = await memory_client.build_context(
+                    resolved_path,
+                    depth=depth or 1,
+                    timeframe=timeframe,
+                    page=page,
+                    page_size=page_size,
+                    max_related=max_related,
+                )
+
+                if output_format == "text":
+                    return _format_context_markdown(graph, active_project.name)
+
+                return graph.model_dump()