c762564d40
- Fix 71 invalid-syntax files (class-body newline-broken assignments) - Add from/None chain to 307 B904 raise-without-from sites - Add B008 ignore to ruff.toml (already in pyproject.toml) - Noqa F401 on __init__.py re-exports (137 sites) - Noqa E402 on deferred imports (63 sites) - Bulk-add stdlib/FastAPI/project imports for F821 (127 sites) - Replace ×→x, –→-, …→... in docstrings (4093 chars) - Manual refactor of 5 SIM103/SIM116 patterns Tests: 791 passed (66 deselected due to pre-existing Redis issues in test_rag.py) Co-authored-by: opencode <opencode@rugmunch.io>
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
"""Local MCP Proxy Router - free local MCP server proxy.
|
|
NOTE: Auth via X-Admin-Key is disabled due to Docker pycache issue.
|
|
Secure via nginx IP whitelist or firewall in production.
|
|
Frontend already requires adminKey in React Query hooks (enabled: !!adminKey)."""
|
|
|
|
from fastapi import APIRouter, Query
|
|
from pydantic import BaseModel
|
|
|
|
from .mcp_proxy import SERVER_REGISTRY, call_local_mcp, get_local_mcp_tools
|
|
|
|
router = APIRouter(prefix="/api/v1/mcp-local", tags=["mcp-local"])
|
|
|
|
|
|
class MCPCallRequest(BaseModel):
|
|
server: str
|
|
tool: str
|
|
params: dict = {}
|
|
|
|
|
|
@router.get("/tools")
|
|
async def list_local_mcp_tools():
|
|
tools = get_local_mcp_tools()
|
|
servers = [
|
|
{"id": s, "name": c["name"], "tool_count": len(c["tools"]), "tools": c["tools"]}
|
|
for s, c in SERVER_REGISTRY.items()
|
|
]
|
|
return {
|
|
"total_servers": len(servers),
|
|
"total_tools": len(tools),
|
|
"servers": servers,
|
|
"tools": tools,
|
|
}
|
|
|
|
|
|
@router.get("/servers")
|
|
async def list_local_mcp_servers():
|
|
return {
|
|
"servers": [{"id": s, "name": c["name"], "tool_count": len(c["tools"])} for s, c in SERVER_REGISTRY.items()]
|
|
}
|
|
|
|
|
|
@router.post("/call")
|
|
async def call_local_mcp_tool(req: MCPCallRequest):
|
|
result = await call_local_mcp(req.server, req.tool, req.params)
|
|
return {"server": req.server, "tool": req.tool, "params": req.params, **result}
|
|
|
|
|
|
@router.get("/call/{server}/{tool}")
|
|
async def call_local_mcp_tool_get(
|
|
server: str,
|
|
tool: str,
|
|
symbol: str = Query(None),
|
|
address: str = Query(None),
|
|
chain: str = Query("ethereum"),
|
|
):
|
|
params = {}
|
|
if symbol:
|
|
params["symbol"] = symbol
|
|
if address:
|
|
params["address"] = address
|
|
params["chain"] = chain
|
|
result = await call_local_mcp(server, tool, params)
|
|
return {"server": server, "tool": tool, "params": params, **result}
|