Skip to content

Commit 296b66c

Browse files
switch to using mcp_bridge namespace
1 parent 23c7f37 commit 296b66c

File tree

24 files changed

+187
-55
lines changed

24 files changed

+187
-55
lines changed

mcp_bridge/config/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
from .initial import initial_settings
2-
from .final import Settings
1+
from mcp_bridge.config.initial import initial_settings
2+
from mcp_bridge.config.final import Settings
33
from typing import Any, Callable
44
from loguru import logger
55
from pydantic import ValidationError

mcp_bridge/endpoints.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,14 @@
22

33
from lmos_openai_types import CreateChatCompletionRequest, CreateCompletionRequest
44

5-
from openai_clients import (
5+
from mcp_bridge.openai_clients import (
66
client,
77
completions,
88
chat_completions,
99
streaming_chat_completions,
1010
)
1111

12-
from openapi_tags import Tag
12+
from mcp_bridge.openapi_tags import Tag
1313

1414
router = APIRouter(prefix="/v1", tags=[Tag.openai])
1515

mcp_bridge/health/router.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from fastapi.responses import JSONResponse
33
from .types import HealthCheckResponse
44
from .manager import manager
5-
from openapi_tags import Tag
5+
from mcp_bridge.openapi_tags import Tag
66

77
router = APIRouter(tags=[Tag.health])
88

mcp_bridge/lifespan.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from contextlib import asynccontextmanager
2-
from mcp_clients.McpClientManager import ClientManager
2+
from mcp_bridge.mcp_clients.McpClientManager import ClientManager
33
from loguru import logger
44

55

mcp_bridge/main.py

Lines changed: 31 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -1,27 +1,37 @@
11
from fastapi import FastAPI
2-
from endpoints import router as endpointRouter
3-
from mcpManagement import router as mcpRouter
4-
from health import router as healthRouter
5-
from mcp_server import router as mcp_server_router
6-
from lifespan import lifespan
7-
from openapi_tags import tags_metadata
8-
from __init__ import __version__ as version
2+
from mcp_bridge.endpoints import router as endpointRouter
3+
from mcp_bridge.mcpManagement import router as mcpRouter
4+
from mcp_bridge.health import router as healthRouter
5+
from mcp_bridge.mcp_server import router as mcp_server_router
6+
from mcp_bridge.lifespan import lifespan
7+
from mcp_bridge.openapi_tags import tags_metadata
8+
from mcp_bridge import __version__ as version
99

10-
app = FastAPI(
11-
title="MCP Bridge",
12-
description="A middleware application to add MCP support to openai compatible apis",
13-
version=version,
14-
lifespan=lifespan,
15-
openapi_tags=tags_metadata,
16-
)
10+
def create_app() -> FastAPI:
11+
"""
12+
Create and configure the FastAPI application.
13+
"""
14+
app = FastAPI(
15+
title="MCP Bridge",
16+
description="A middleware application to add MCP support to OpenAI-compatible APIs",
17+
version=version,
18+
lifespan=lifespan,
19+
openapi_tags=tags_metadata,
20+
)
1721

18-
app.include_router(endpointRouter)
19-
app.include_router(mcpRouter)
20-
app.include_router(healthRouter)
21-
app.include_router(mcp_server_router)
22+
app.include_router(endpointRouter)
23+
app.include_router(mcpRouter)
24+
app.include_router(healthRouter)
25+
app.include_router(mcp_server_router)
2226

23-
if __name__ == "__main__":
24-
import uvicorn
25-
from config import config
27+
return app
28+
29+
app = create_app()
2630

31+
def run():
32+
import uvicorn
33+
from mcp_bridge.config import config
2734
uvicorn.run(app, host=config.network.host, port=config.network.port)
35+
36+
if __name__ == "__main__":
37+
run()

mcp_bridge/mcpManagement/prompts.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from typing import Any
22
from fastapi import APIRouter, HTTPException
3-
from mcp_clients.McpClientManager import ClientManager
3+
from mcp_bridge.mcp_clients.McpClientManager import ClientManager
44
from mcp.types import ListPromptsResult, GetPromptResult
55

66
router = APIRouter(prefix="/prompts")

mcp_bridge/mcpManagement/resources.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from fastapi import APIRouter, HTTPException
2-
from mcp_clients.McpClientManager import ClientManager
2+
from mcp_bridge.mcp_clients.McpClientManager import ClientManager
33
from mcp.types import ListResourcesResult
44

55
router = APIRouter(prefix="/resources")

mcp_bridge/mcpManagement/router.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from fastapi import APIRouter
2-
from openapi_tags import Tag
2+
from mcp_bridge.openapi_tags import Tag
33

44
from .tools import router as tools_router
55
from .prompts import router as prompts_router

mcp_bridge/mcpManagement/server.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
from fastapi import APIRouter, HTTPException
22
from mcp.types import ListPromptsResult, ListToolsResult, ListResourcesResult
3-
from models.mcpServerStatus import McpServerStatus
4-
from mcp_clients.McpClientManager import ClientManager
3+
from mcp_bridge.models.mcpServerStatus import McpServerStatus
4+
from mcp_bridge.mcp_clients.McpClientManager import ClientManager
55

66
router = APIRouter(prefix="/servers")
77

mcp_bridge/mcpManagement/tools.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from fastapi import APIRouter, HTTPException
2-
from mcp_clients.McpClientManager import ClientManager
2+
from mcp_bridge.mcp_clients.McpClientManager import ClientManager
33
from mcp.types import ListToolsResult, CallToolResult
44

55
router = APIRouter(prefix="/tools")

0 commit comments

Comments
 (0)