2323
2424from __future__ import annotations
2525
26+ import asyncio
2627import json
2728import logging
2829
@@ -47,7 +48,7 @@ def serve() -> None:
4748 from mcp .server .stdio import stdio_server
4849 from mcp .types import TextContent , Tool
4950
50- from .pipelines .retrieve import retrieve , _load_all_trees , _node_index
51+ from .pipelines .retrieve import _load_all_trees , _node_index , retrieve
5152
5253 server = Server ("catchme" )
5354
@@ -163,12 +164,14 @@ def _handle_list_days() -> list[TextContent]:
163164 days = []
164165 for t in trees :
165166 node = t .get ("tree" , {})
166- days .append ({
167- "date" : t .get ("date" , node .get ("title" , "?" )),
168- "node_id" : node .get ("node_id" , "" ),
169- "summary" : (node .get ("summary" ) or "" )[:300 ],
170- "session_count" : len (node .get ("children" , [])),
171- })
167+ days .append (
168+ {
169+ "date" : t .get ("date" , node .get ("title" , "?" )),
170+ "node_id" : node .get ("node_id" , "" ),
171+ "summary" : (node .get ("summary" ) or "" )[:300 ],
172+ "session_count" : len (node .get ("children" , [])),
173+ }
174+ )
172175 return [TextContent (type = "text" , text = json .dumps (days , ensure_ascii = False , indent = 2 ))]
173176
174177 # ── get_session ──────────────────────────────────────────────────────────
@@ -197,7 +200,9 @@ def _slim(n: dict, depth: int = 0) -> dict:
197200 out ["children" ] = [_slim (c , depth + 1 ) for c in n .get ("children" , [])]
198201 return out
199202
200- return [TextContent (type = "text" , text = json .dumps (_slim (node ), ensure_ascii = False , indent = 2 ))]
203+ return [
204+ TextContent (type = "text" , text = json .dumps (_slim (node ), ensure_ascii = False , indent = 2 ))
205+ ]
201206
202207 # ── get_tree ─────────────────────────────────────────────────────────────
203208
@@ -214,8 +219,6 @@ def _handle_get_tree(args: dict) -> list[TextContent]:
214219
215220 # ── run ──────────────────────────────────────────────────────────────────
216221
217- import asyncio
218-
219222 async def _run ():
220223 async with stdio_server () as (read_stream , write_stream ):
221224 await server .run (read_stream , write_stream , server .create_initialization_options ())
0 commit comments