added mcpClientAuthProvider to QuarkusHttpMcpTransport and QuarkusStreamableHttpMcpTransport

format

Author: KaiSuchomel

mcp/runtime/src/main/java/io/quarkiverse/langchain4j/mcp/runtime/http/QuarkusHttpMcpTransport.java

@@ -63,7 +63,11 @@ public QuarkusHttpMcpTransport(QuarkusHttpMcpTransport.Builder builder) {
             clientBuilder.tlsConfiguration(tlsConfiguration);
         }
 
-        this.mcpClientAuthProvider = McpClientAuthProvider.resolve(builder.mcpClientName).orElse(null);
+        if (builder.mcpClientAuthProvider != null) {
+            this.mcpClientAuthProvider = builder.mcpClientAuthProvider;
+        } else {
+            this.mcpClientAuthProvider = McpClientAuthProvider.resolve(builder.mcpClientName).orElse(null);
+        }
         if (mcpClientAuthProvider != null) {
             clientBuilder.register(new McpClientAuthFilter(mcpClientAuthProvider));
         }
@@ -198,6 +202,7 @@ public static class Builder {
         private boolean logRequests = false;
         private boolean logResponses = false;
         private TlsConfiguration tlsConfiguration;
+        private McpClientAuthProvider mcpClientAuthProvider;
 
         /**
          * The initial URL where to connect to the server and request a SSE
@@ -233,6 +238,11 @@ public QuarkusHttpMcpTransport.Builder tlsConfiguration(TlsConfiguration tlsConf
             return this;
         }
 
+        public QuarkusHttpMcpTransport.Builder mcpClientAuthProvider(McpClientAuthProvider mcpClientAuthProvider) {
+            this.mcpClientAuthProvider = mcpClientAuthProvider;
+            return this;
+        }
+
         public QuarkusHttpMcpTransport build() {
             return new QuarkusHttpMcpTransport(this);
         }

mcp/runtime/src/main/java/io/quarkiverse/langchain4j/mcp/runtime/http/QuarkusStreamableHttpMcpTransport.java

@@ -60,7 +60,11 @@ public QuarkusStreamableHttpMcpTransport(QuarkusStreamableHttpMcpTransport.Build
         this.httpClient = builder.httpClient;
         this.logRequests = builder.logRequests;
         this.logResponses = builder.logResponses;
-        this.mcpClientAuthProvider = McpClientAuthProvider.resolve(builder.mcpClientName).orElse(null);
+        if (builder.mcpClientAuthProvider != null) {
+            this.mcpClientAuthProvider = builder.mcpClientAuthProvider;
+        } else {
+            this.mcpClientAuthProvider = McpClientAuthProvider.resolve(builder.mcpClientName).orElse(null);
+        }
     }
 
     @Override
@@ -246,6 +250,7 @@ public static class Builder {
         private boolean logRequests = false;
         private boolean logResponses = false;
         private HttpClient httpClient;
+        private McpClientAuthProvider mcpClientAuthProvider;
 
         /**
          * The initial URL where to connect to the server and request a SSE
@@ -281,6 +286,11 @@ public Builder httpClient(HttpClient httpClient) {
             return this;
         }
 
+        public Builder mcpClientAuthProvider(McpClientAuthProvider mcpClientAuthProvider) {
+            this.mcpClientAuthProvider = mcpClientAuthProvider;
+            return this;
+        }
+
         public QuarkusStreamableHttpMcpTransport build() {
             return new QuarkusStreamableHttpMcpTransport(this);
         }