Skip to content

Commit 9c52a01

Browse files
authored
Merge pull request #646 from gargnipungarg/main
Fastmcp v2 migration
2 parents c623270 + d4318a3 commit 9c52a01

File tree

5 files changed

+815
-144
lines changed

5 files changed

+815
-144
lines changed

model-deployment/mcp-servers/oci-cache-redis-server/model-deplyment.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def get_datascience_client():
7878
"healthCheckPort": 8000,
7979
"environmentVariables": {
8080
"WEB_CONCURRENCY": "1",
81-
"MODEL_DEPLOY_PREDICT_ENDPOINT": "/mcp/"
81+
"MODEL_DEPLOY_PREDICT_ENDPOINT": "/mcp"
8282
}
8383
}
8484
},

model-deployment/mcp-servers/oci-cache-redis-server/pyproject.toml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,4 +9,6 @@ dependencies = [
99
"redis>=6.0.0",
1010
"dotenv>=0.9.9",
1111
"numpy>=2.2.4",
12+
"fastmcp>=2.12.3",
13+
"uvicorn>=0.34.0",
1214
]
Lines changed: 7 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from mcp.server.fastmcp import FastMCP
1+
from fastmcp import FastMCP
22
from starlette.responses import JSONResponse
33
from starlette.requests import Request
44
from starlette.responses import Response
@@ -13,25 +13,14 @@
1313
"Redis MCP Server",
1414
host=MCP_HOST,
1515
port=MCP_PORT,
16-
dependencies=["redis", "dotenv", "numpy"]
16+
dependencies=["redis", "dotenv", "numpy"],
17+
stateless_http=True,
18+
json_response=True
1719
)
1820

19-
mcp._session_manager = StreamableHTTPSessionManager(
20-
app=mcp._mcp_server,
21-
event_store=None,
22-
json_response=True,
23-
stateless=True,
24-
)
25-
2621
def handle_health(request):
2722
return JSONResponse({"status": "success"})
2823

29-
async def handle_streamable_http(
30-
scope: Scope, receive: Receive, send: Send
31-
) -> None:
32-
await mcp._session_manager.handle_request(scope, receive, send)
33-
34-
mcp._custom_starlette_routes=[
35-
Mount("/mcp", app=handle_streamable_http),
36-
Route('/health', endpoint=handle_health),
37-
]
24+
mcp._additional_http_routes = [
25+
Route('/health', endpoint=handle_health),
26+
]

model-deployment/mcp-servers/oci-cache-redis-server/src/main.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,19 @@
11
import sys
22

33
from common.connection import RedisConnectionManager
4-
from common.server import mcp
54
import tools.server_management
65
import tools.misc
76
import tools.redis_query_engine
87
import tools.hash
9-
import tools.list
10-
import tools.string
11-
import tools.json
12-
import tools.sorted_set
13-
import tools.set
14-
import tools.stream
15-
import tools.pub_sub
16-
from common.config import MCP_TRANSPORT
8+
#import tools.list
9+
#import tools.string
10+
#import tools.json
11+
#import tools.sorted_set
12+
#import tools.set
13+
#import tools.stream
14+
#import tools.pub_sub
15+
from common.server import mcp
16+
from common.config import MCP_TRANSPORT, MCP_HOST, MCP_PORT
1717

1818

1919
class RedisMCPServer:

0 commit comments

Comments
 (0)