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 (
PryError,
)
from mcp_production import make_fallback_server, register_all
from mcp_sse import mcp_post_message, mcp_sse_endpoint
from observability import REQUEST_COUNT, REQUEST_LATENCY, setup_tracing
from pipeline import get_pipeline
from pry_mcp import make_fallback_server, register_all
from routers.advanced import router as advanced_router
from routers.agency import router as agency_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, ".")
import asyncio
from mcp_production import main
from pry_mcp import main
asyncio.run(main())

View file

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

View file

@ -7,7 +7,7 @@
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:

View file

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