Skip to content

Commit b88297d

Browse files
sanskritipatoleadwsingh
authored andcommitted
add metrics observer support to McpServerBuilder
1 parent 552a648 commit b88297d

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

mcp/mcp-server/src/main/java/software/amazon/smithy/java/mcp/server/McpServerBuilder.java

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ public final class McpServerBuilder {
2727
String name;
2828
String version;
2929
ToolFilter toolFilter = (server, tool) -> true;
30+
McpMetricsObserver metricsObserver;
3031
McpService mcpService;
3132

3233
McpServerBuilder() {}
@@ -64,7 +65,8 @@ public Server build() {
6465
.services(services)
6566
.proxyList(proxyList)
6667
.name(name != null ? name : "mcp-server")
67-
.toolFilter(toolFilter);
68+
.toolFilter(toolFilter)
69+
.metricsObserver(metricsObserver);
6870

6971
if (version != null) {
7072
builder.version(version);
@@ -94,6 +96,11 @@ public McpServerBuilder toolFilter(ToolFilter filter) {
9496
return this;
9597
}
9698

99+
public McpServerBuilder metricsObserver(McpMetricsObserver observer) {
100+
this.metricsObserver = observer;
101+
return this;
102+
}
103+
97104
private void validate() {
98105
Objects.requireNonNull(is, "MCP server input stream is required");
99106
Objects.requireNonNull(os, "MCP server output stream is required");

0 commit comments

Comments
 (0)