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:
parent
f7f9fa0c88
commit
4d2603cdd5
5 changed files with 5 additions and 5 deletions
2
api.py
2
api.py
|
|
@ -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
2
cli.py
|
|
@ -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())
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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,
|
||||||
)
|
)
|
||||||
|
|
|
||||||
|
|
@ -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:
|
||||||
|
|
|
||||||
|
|
@ -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,
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue