Pass to underlying tool

Signed-off-by: David Gageot <david.gageot@docker.com>

Author: dgageot

pkg/tools/mcp/gateway.go

@@ -17,8 +17,6 @@ import (
 )
 
 type GatewayToolset struct {
-	tools.ElicitationTool
-
 	mcpServerName string
 	config        any
 	envProvider   environment.Provider
@@ -152,3 +150,11 @@ func writeTempFile(nameTemplate string, content []byte) (string, error) {
 
 	return f.Name(), nil
 }
+
+func (t *GatewayToolset) SetElicitationHandler(handler tools.ElicitationHandler) {
+	t.cmdToolset.SetElicitationHandler(handler)
+}
+
+func (t *GatewayToolset) SetOAuthSuccessHandler(handler func()) {
+	t.cmdToolset.SetOAuthSuccessHandler(handler)
+}