diff --git a/api.py b/api.py index ffc08e4..4bcbb42 100644 --- a/api.py +++ b/api.py @@ -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 diff --git a/cli.py b/cli.py index 9f9d976..eda0e24 100755 --- a/cli.py +++ b/cli.py @@ -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()) diff --git a/mcp_sse.py b/mcp_sse.py index 988b702..d91fa56 100644 --- a/mcp_sse.py +++ b/mcp_sse.py @@ -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, ) diff --git a/tests/test_mcp_production.py b/tests/test_mcp_production.py index 83cce7e..d82ec92 100644 --- a/tests/test_mcp_production.py +++ b/tests/test_mcp_production.py @@ -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: diff --git a/tests/test_x402_mcp_spec.py b/tests/test_x402_mcp_spec.py index bb5df27..341e68e 100644 --- a/tests/test_x402_mcp_spec.py +++ b/tests/test_x402_mcp_spec.py @@ -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,