Support embed tool calling from mai

Author: radare

go.mod

@@ -4,9 +4,10 @@ go 1.20.0
 
 require github.com/gorilla/mux v1.8.0
 
-require (
-	gopkg.in/yaml.v3 v3.0.1
-	mcplib v0.0.0
-)
+require gopkg.in/yaml.v3 v3.0.1
+
+require wmcplib v0.0.0
 
 replace mcplib => ./src/mcps/lib
+
+replace wmcplib => ./src/wmcp/lib

src/repl/conf.go

@@ -150,6 +150,8 @@ func NewConfigOptions() *ConfigOptions {
 	co.RegisterOption("mcp.config", StringOption, "Path to MCP configuration file", "")
 	co.RegisterOption("mcp.args", StringOption, "Command-line arguments to pass to mai-wmcp", "")
 	co.RegisterOption("mcp.daemon", BooleanOption, "Enable starting the mai-wmcp server", "true")
+	co.RegisterOption("mcp.transport", StringOption, "MCP transport: http (default, spawns mai-wmcp) or embed (in-process via wmcplib)", "http")
+	co.RegisterOption("mcp.yolo", BooleanOption, "Skip tool confirmation prompts when running in embed transport", "false")
 	co.RegisterOption("mcp.debug", BooleanOption, "Enable debug output for MCP communication between agent, model, and servers", "false")
 	co.RegisterOption("mcp.prompt", StringOption, "Custom text to be included in the instructions prompt for react loops", "")
 	co.RegisterOption("mcp.toolformat", StringOption, "Tool list format: xml, markdown, simple, quiet, json (empty=auto)", "")

src/repl/go.mod

@@ -7,8 +7,11 @@ require (
 	golang.org/x/sys v0.35.0
 	golang.org/x/term v0.33.0
 	mai/src/mcps/lib v0.0.0
+	mai/src/wmcp/lib v0.0.0
 )
 
 require github.com/clipperhouse/uax29/v2 v2.2.0 // indirect
 
 replace mai/src/mcps/lib => ../mcps/lib
+
+replace mai/src/wmcp/lib => ../wmcp/lib

src/repl/mcprompt.go

@@ -12,6 +12,9 @@ import (
 
 // GetAvailableMCPrompts runs 'mai-tool prompts list' and returns the output as a string
 func GetAvailableMCPrompts(f Format) (string, error) {
+	if embedActiveRepl != nil {
+		return embedListPromptsSimple(embedActiveRepl)
+	}
 	var cmd *exec.Cmd
 	switch f {
 	case Quiet:
@@ -33,6 +36,9 @@ func GetAvailableMCPrompts(f Format) (string, error) {
 
 // GetMCPromptContent fetches a specific prompt's rendered content
 func GetMCPromptContent(fullName string) (string, error) {
+	if embedActiveRepl != nil {
+		return embedGetPrompt(embedActiveRepl, fullName, nil)
+	}
 	// fullName format: server/prompt or just prompt
 	cmd := exec.Command("mai-tool", "prompts", "get", fullName)
 	var out bytes.Buffer

src/repl/repl_core.go

@@ -256,8 +256,12 @@ func NewREPL(configOptions ConfigOptions, initialCommand string, quitAfterAction
 		}
 	}
 
-	// Only start mai-wmcp daemon if mcp.use is enabled
-	if repl.configOptions.GetBool("mcp.use") && repl.configOptions.GetBool("mcp.daemon") {
+	// Embed transport runs wmcp in-process instead of spawning mai-wmcp.
+	if repl.configOptions.GetBool("mcp.use") && replEmbedActive(repl) {
+		if _, err := embedGetService(repl); err != nil {
+			fmt.Fprintf(os.Stderr, "Error starting embedded wmcp: %v\n", err)
+		}
+	} else if repl.configOptions.GetBool("mcp.use") && repl.configOptions.GetBool("mcp.daemon") {
 		// Spawn mai-wmcp if mcp.config or mcp.args is set
 		var wmcpArgs []string
 		if v := repl.configOptions.Get("mcp.config"); v != "" {
@@ -367,6 +371,8 @@ func (r *REPL) Run() error {
 }
 
 func (r *REPL) cleanup() {
+	// Shut down the embedded wmcp service, if any. Harmless when not active.
+	embedStop()
 	// Kill wmcp process first to ensure it's stopped regardless of what happens next
 	if r.wmcpProcess != nil && r.wmcpProcess.Process != nil {
 		if err := r.wmcpProcess.Process.Kill(); err != nil {

src/repl/tools.go

@@ -66,6 +66,9 @@ func parseToolFormat(formatStr string) Format {
 
 // GetAvailableTools runs the 'mai-tool list' command and returns the output as a string
 func GetAvailableTools(f Format) (string, error) {
+	if embedActiveRepl != nil {
+		return embedListToolsFormatted(embedActiveRepl, f)
+	}
 	var cmd *exec.Cmd
 	switch f {
 	case Quiet:
@@ -183,6 +186,10 @@ func callTool(tool *Tool, debug bool, format string, timeoutSeconds int) (string
 		}
 	}
 
+	if embedActiveRepl != nil {
+		return embedCallToolStringArgs(embedActiveRepl, toolName, safeArgs, timeoutSeconds)
+	}
+
 	var out bytes.Buffer
 	var stderr bytes.Buffer
 	cmdArgs := append([]string{"call", toolName}, safeArgs...)
@@ -256,6 +263,10 @@ func (r *REPL) executeToolNative(toolName string, args ...string) (string, error
 		return "", fmt.Errorf("failed to parse tool arguments JSON: %v", err)
 	}
 
+	if replEmbedActive(r) {
+		return embedCallTool(r, toolName, params, 60)
+	}
+
 	// Build mai-tool command arguments: mai-tool call <tool> [key=value ...]
 	cmdArgs := []string{"call", toolName}
 	for key, value := range params {