Ryan Malloy
dca4e80857
Add add_block_path() and get_block_paths() MCP tools for incremental OOT module loading with BlockPathsModel responses. On startup, auto-scan /usr/local/share and ~/.local/share for OOT blocks so modules like gr-lora_sdr are available without manual configuration.
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
import os
|
|
|
|
from fastmcp import FastMCP
|
|
|
|
from gnuradio_mcp.middlewares.platform import PlatformMiddleware
|
|
from gnuradio_mcp.providers.mcp import McpPlatformProvider
|
|
from gnuradio_mcp.providers.mcp_runtime import McpRuntimeProvider
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
try:
|
|
from gnuradio import gr
|
|
from gnuradio.grc.core.platform import Platform
|
|
except ImportError:
|
|
raise Exception("Cannot find GNU Radio!") from None
|
|
|
|
platform = Platform(
|
|
version=gr.version(),
|
|
version_parts=(gr.major_version(), gr.api_version(), gr.minor_version()),
|
|
prefs=gr.prefs(),
|
|
)
|
|
platform.build_library()
|
|
|
|
app: FastMCP = FastMCP("GNU Radio MCP", instructions="Create GNU Radio flowgraphs")
|
|
|
|
pmw = PlatformMiddleware(platform)
|
|
|
|
# Auto-discover OOT modules from common install locations
|
|
oot_candidates = [
|
|
"/usr/local/share/gnuradio/grc/blocks",
|
|
os.path.expanduser("~/.local/share/gnuradio/grc/blocks"),
|
|
]
|
|
for path in oot_candidates:
|
|
if os.path.isdir(path):
|
|
try:
|
|
result = pmw.add_block_path(path)
|
|
if result.blocks_added > 0:
|
|
logger.info(f"OOT: +{result.blocks_added} blocks from {path}")
|
|
except Exception:
|
|
pass
|
|
|
|
McpPlatformProvider.from_platform_middleware(app, pmw)
|
|
McpRuntimeProvider.create(app)
|
|
|
|
if __name__ == "__main__":
|
|
app.run()
|