Skip to content

Commit bd74d10

Browse files
authored
feat(fxmcpserver): Added autoconfiguration of the SSE test server endpoints (#352)
1 parent dd07f38 commit bd74d10

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

fxmcpserver/fxmcpservertest/server.go

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,22 @@ type MCPSSETestServer struct {
1818
}
1919

2020
func NewMCPSSETestServer(cfg *config.Config, srv *server.MCPServer, hdl sse.MCPSSEServerContextHandler) *MCPSSETestServer {
21-
testSrv := server.NewTestServer(srv, server.WithSSEContextFunc(hdl.Handle()))
21+
sseEndpoint := cfg.GetString("modules.mcp.server.transport.sse.sse_endpoint")
22+
if sseEndpoint == "" {
23+
sseEndpoint = sse.DefaultSSEEndpoint
24+
}
25+
26+
messageEndpoint := cfg.GetString("modules.mcp.server.transport.sse.message_endpoint")
27+
if messageEndpoint == "" {
28+
messageEndpoint = sse.DefaultMessageEndpoint
29+
}
30+
31+
testSrv := server.NewTestServer(
32+
srv,
33+
server.WithSSEContextFunc(hdl.Handle()),
34+
server.WithSSEEndpoint(sseEndpoint),
35+
server.WithMessageEndpoint(messageEndpoint),
36+
)
2237

2338
return &MCPSSETestServer{
2439
config: cfg,

0 commit comments

Comments
 (0)