enhance: add the ability to close an individual MCP server

Signed-off-by: Donnie Adams <[email protected]>

Author: thedadams

pkg/engine/engine.go

@@ -350,6 +350,9 @@ func (e *Engine) Start(ctx Context, input string) (ret *Return, err error) {
 		}
 		select {
 		case <-ctx.userCancel:
+			if ret == nil {
+				ret = new(Return)
+			}
 			if ret.Result == nil {
 				ret.Result = new(string)
 			}

pkg/mcp/loader.go

@@ -125,6 +125,30 @@ func (l *Local) LoadTools(ctx context.Context, server ServerConfig, toolName str
 	return l.sessionToTools(ctx, session, toolName, allowedTools)
 }
 
+func (l *Local) ShutdownServer(server ServerConfig) error {
+	if l == nil {
+		return nil
+	}
+
+	id := hash.Digest(server)
+
+	l.lock.Lock()
+	defer l.lock.Unlock()
+
+	if l.sessionCtx == nil {
+		return nil
+	}
+
+	session := l.sessions[id]
+	if session == nil {
+		return nil
+	}
+
+	delete(l.sessions, id)
+
+	return session.Client.Close()
+}
+
 func (l *Local) Close() error {
 	if l == nil {
 		return nil