refactor(mcp): modernize imports to use pry_mcp sub-package directly

- api.py, cli.py, mcp_sse.py now import from pry_mcp

- tests import from pry_mcp (snapshot tests still verify mcp_production shim)
This commit is contained in:
Crypto Rug Munch 2026-07-03 15:18:11 +02:00
parent f7f9fa0c88
commit 4d2603cdd5
5 changed files with 5 additions and 5 deletions

2
api.py
View file

@ -25,10 +25,10 @@ from deps import advanced, automator, extractor, queue, ratelimiter, scraper
from errors import ( from errors import (
PryError, PryError,
) )
from mcp_production import make_fallback_server, register_all
from mcp_sse import mcp_post_message, mcp_sse_endpoint from mcp_sse import mcp_post_message, mcp_sse_endpoint
from observability import REQUEST_COUNT, REQUEST_LATENCY, setup_tracing from observability import REQUEST_COUNT, REQUEST_LATENCY, setup_tracing
from pipeline import get_pipeline from pipeline import get_pipeline
from pry_mcp import make_fallback_server, register_all
from routers.advanced import router as advanced_router from routers.advanced import router as advanced_router
from routers.agency import router as agency_router from routers.agency import router as agency_router
from routers.ai import router as ai_router from routers.ai import router as ai_router

2
cli.py
View file

@ -393,7 +393,7 @@ if click is not None:
sys.path.insert(0, ".") sys.path.insert(0, ".")
import asyncio import asyncio
from mcp_production import main from pry_mcp import main
asyncio.run(main()) asyncio.run(main())

View file

@ -29,7 +29,7 @@ from typing import Any
from fastapi import HTTPException, Request, Response from fastapi import HTTPException, Request, Response
from fastapi.responses import StreamingResponse from fastapi.responses import StreamingResponse
from mcp_production import ( from pry_mcp import (
register_mcp_notification_observer, register_mcp_notification_observer,
unregister_mcp_notification_observer, unregister_mcp_notification_observer,
) )

View file

@ -7,7 +7,7 @@
import asyncio import asyncio
from mcp_production import MCP_PROTOCOL_VERSION, PRY_TOOLS, make_fallback_server, register_all from pry_mcp import MCP_PROTOCOL_VERSION, PRY_TOOLS, make_fallback_server, register_all
def test_mcp_protocol_version() -> None: def test_mcp_protocol_version() -> None:

View file

@ -10,7 +10,7 @@ import base64
import json import json
from typing import Any from typing import Any
from mcp_production import ( from pry_mcp import (
MCP_PROTOCOL_VERSION, MCP_PROTOCOL_VERSION,
PRY_PROMPTS, PRY_PROMPTS,
PRY_RESOURCES, PRY_RESOURCES,