danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Handle keyboard interrupt

Browse source
This commit is contained in:
Yiorgis Gozadinos 2025-06-20 10:18:07 +02:00
parent 18e1e79598
commit 03b2b20ae2
No known key found for this signature in database
1 changed files with 16 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
src/haiku/rag/app.py
View file

@ -96,12 +96,21 @@ class HaikuRAGApp:
"""Start the MCP server."""
async with HaikuRAG(self.db_path) as client:
monitor = FileWatcher(paths=Config.MONITOR_DIRECTORIES, client=client)
asyncio.create_task(monitor.observe())
monitor_task = asyncio.create_task(monitor.observe())
server = create_mcp_server(self.db_path)
if transport == "stdio":
await server.run_stdio_async()
elif transport == "sse":
await server.run_sse_async("sse")
else:
await server.run_http_async("streamable-http")
try:
if transport == "stdio":
await server.run_stdio_async()
elif transport == "sse":
await server.run_sse_async("sse")
else:
await server.run_http_async("streamable-http")
except KeyboardInterrupt:
pass
finally:
monitor_task.cancel()
try:
await monitor_task
except asyncio.CancelledError:
pass