add possibility to start the mcp-server as a streamable http-server

Author: rainu

README.md

@@ -145,6 +145,11 @@ Or you can rename the binary to `ask-mai-mcp-server` and start it like this:
 ask-mai-mcp-server
 ```
 
+Do run the MCP server as a streamable HTTP server, you can use the `-http-address` flag:
+```bash
+ask-mai ask-mai-mcp-server --http-address=":8080"
+```
+
 ## How to build this application
 
 1. Install dependencies [see wails documentation](https://wails.io/docs/gettingstarted/installation)

frontend/wailsjs/go/models.ts

@@ -2221,9 +2221,24 @@ export namespace llm {
 
 export namespace mcp {
 
+	export class Meta {
+	    ProgressToken: any;
+	    AdditionalFields: Record<string, any>;
+	
+	    static createFrom(source: any = {}) {
+	        return new Meta(source);
+	    }
+	
+	    constructor(source: any = {}) {
+	        if ('string' === typeof source) source = JSON.parse(source);
+	        this.ProgressToken = source["ProgressToken"];
+	        this.AdditionalFields = source["AdditionalFields"];
+	    }
+	}
 	export class Annotations {
 	    audience?: string[];
 	    priority?: number;
+	    lastModified?: string;
 
 	    static createFrom(source: any = {}) {
 	        return new Annotations(source);
@@ -2233,11 +2248,13 @@ export namespace mcp {
 	        if ('string' === typeof source) source = JSON.parse(source);
 	        this.audience = source["audience"];
 	        this.priority = source["priority"];
+	        this.lastModified = source["lastModified"];
 	    }
 	}
 	export class AudioContent {
 	    // Go type: Annotations
 	    annotations?: any;
+	    _meta?: Meta;
 	    type: string;
 	    data: string;
 	    mimeType: string;
@@ -2249,6 +2266,7 @@ export namespace mcp {
 	    constructor(source: any = {}) {
 	        if ('string' === typeof source) source = JSON.parse(source);
 	        this.annotations = this.convertValues(source["annotations"], null);
+	        this._meta = this.convertValues(source["_meta"], Meta);
 	        this.type = source["type"];
 	        this.data = source["data"];
 	        this.mimeType = source["mimeType"];
@@ -2273,8 +2291,9 @@ export namespace mcp {
 		}
 	}
 	export class CallToolResult {
-	    _meta?: Record<string, any>;
+	    _meta?: Meta;
 	    content: any[];
+	    structuredContent?: any;
 	    isError?: boolean;
 
 	    static createFrom(source: any = {}) {
@@ -2283,14 +2302,34 @@ export namespace mcp {
 
 	    constructor(source: any = {}) {
 	        if ('string' === typeof source) source = JSON.parse(source);
-	        this._meta = source["_meta"];
+	        this._meta = this.convertValues(source["_meta"], Meta);
 	        this.content = source["content"];
+	        this.structuredContent = source["structuredContent"];
 	        this.isError = source["isError"];
 	    }
+	
+		convertValues(a: any, classs: any, asMap: boolean = false): any {
+		    if (!a) {
+		        return a;
+		    }
+		    if (a.slice && a.map) {
+		        return (a as any[]).map(elem => this.convertValues(elem, classs));
+		    } else if ("object" === typeof a) {
+		        if (asMap) {
+		            for (const key of Object.keys(a)) {
+		                a[key] = new classs(a[key]);
+		            }
+		            return a;
+		        }
+		        return new classs(a);
+		    }
+		    return a;
+		}
 	}
 	export class EmbeddedResource {
 	    // Go type: Annotations
 	    annotations?: any;
+	    _meta?: Meta;
 	    type: string;
 	    resource: any;
 
@@ -2301,6 +2340,7 @@ export namespace mcp {
 	    constructor(source: any = {}) {
 	        if ('string' === typeof source) source = JSON.parse(source);
 	        this.annotations = this.convertValues(source["annotations"], null);
+	        this._meta = this.convertValues(source["_meta"], Meta);
 	        this.type = source["type"];
 	        this.resource = source["resource"];
 	    }
@@ -2323,9 +2363,26 @@ export namespace mcp {
 		    return a;
 		}
 	}
+	export class Icon {
+	    src: string;
+	    mimeType?: string;
+	    sizes?: string[];
+	
+	    static createFrom(source: any = {}) {
+	        return new Icon(source);
+	    }
+	
+	    constructor(source: any = {}) {
+	        if ('string' === typeof source) source = JSON.parse(source);
+	        this.src = source["src"];
+	        this.mimeType = source["mimeType"];
+	        this.sizes = source["sizes"];
+	    }
+	}
 	export class ImageContent {
 	    // Go type: Annotations
 	    annotations?: any;
+	    _meta?: Meta;
 	    type: string;
 	    data: string;
 	    mimeType: string;
@@ -2337,6 +2394,7 @@ export namespace mcp {
 	    constructor(source: any = {}) {
 	        if ('string' === typeof source) source = JSON.parse(source);
 	        this.annotations = this.convertValues(source["annotations"], null);
+	        this._meta = this.convertValues(source["_meta"], Meta);
 	        this.type = source["type"];
 	        this.data = source["data"];
 	        this.mimeType = source["mimeType"];
@@ -2360,6 +2418,7 @@ export namespace mcp {
 		    return a;
 		}
 	}
+	
 	export class Timeout {
 	    Init?: number;
 	    List?: number;
@@ -2423,6 +2482,7 @@ export namespace mcp {
 	export class TextContent {
 	    // Go type: Annotations
 	    annotations?: any;
+	    _meta?: Meta;
 	    type: string;
 	    text: string;
 
@@ -2433,6 +2493,7 @@ export namespace mcp {
 	    constructor(source: any = {}) {
 	        if ('string' === typeof source) source = JSON.parse(source);
 	        this.annotations = this.convertValues(source["annotations"], null);
+	        this._meta = this.convertValues(source["_meta"], Meta);
 	        this.type = source["type"];
 	        this.text = source["text"];
 	    }
@@ -2456,6 +2517,18 @@ export namespace mcp {
 		}
 	}
 
+	export class ToolExecution {
+	    taskSupport?: string;
+	
+	    static createFrom(source: any = {}) {
+	        return new ToolExecution(source);
+	    }
+	
+	    constructor(source: any = {}) {
+	        if ('string' === typeof source) source = JSON.parse(source);
+	        this.taskSupport = source["taskSupport"];
+	    }
+	}
 	export class ToolAnnotation {
 	    title?: string;
 	    readOnlyHint?: boolean;
@@ -2476,38 +2549,72 @@ export namespace mcp {
 	        this.openWorldHint = source["openWorldHint"];
 	    }
 	}
+	export class ToolOutputSchema {
+	    $defs?: Record<string, any>;
+	    type: string;
+	    properties?: Record<string, any>;
+	    required?: string[];
+	    additionalProperties?: any;
+	
+	    static createFrom(source: any = {}) {
+	        return new ToolOutputSchema(source);
+	    }
+	
+	    constructor(source: any = {}) {
+	        if ('string' === typeof source) source = JSON.parse(source);
+	        this.$defs = source["$defs"];
+	        this.type = source["type"];
+	        this.properties = source["properties"];
+	        this.required = source["required"];
+	        this.additionalProperties = source["additionalProperties"];
+	    }
+	}
 	export class ToolInputSchema {
+	    $defs?: Record<string, any>;
 	    type: string;
 	    properties?: Record<string, any>;
 	    required?: string[];
+	    additionalProperties?: any;
 
 	    static createFrom(source: any = {}) {
 	        return new ToolInputSchema(source);
 	    }
 
 	    constructor(source: any = {}) {
 	        if ('string' === typeof source) source = JSON.parse(source);
+	        this.$defs = source["$defs"];
 	        this.type = source["type"];
 	        this.properties = source["properties"];
 	        this.required = source["required"];
+	        this.additionalProperties = source["additionalProperties"];
 	    }
 	}
 	export class Tool {
+	    _meta?: Meta;
 	    name: string;
 	    description?: string;
 	    inputSchema: ToolInputSchema;
+	    outputSchema?: ToolOutputSchema;
 	    annotations: ToolAnnotation;
+	    defer_loading?: boolean;
+	    icons?: Icon[];
+	    execution?: ToolExecution;
 
 	    static createFrom(source: any = {}) {
 	        return new Tool(source);
 	    }
 
 	    constructor(source: any = {}) {
 	        if ('string' === typeof source) source = JSON.parse(source);
+	        this._meta = this.convertValues(source["_meta"], Meta);
 	        this.name = source["name"];
 	        this.description = source["description"];
 	        this.inputSchema = this.convertValues(source["inputSchema"], ToolInputSchema);
+	        this.outputSchema = this.convertValues(source["outputSchema"], ToolOutputSchema);
 	        this.annotations = this.convertValues(source["annotations"], ToolAnnotation);
+	        this.defer_loading = source["defer_loading"];
+	        this.icons = this.convertValues(source["icons"], Icon);
+	        this.execution = this.convertValues(source["execution"], ToolExecution);
 	    }
 
 		convertValues(a: any, classs: any, asMap: boolean = false): any {
@@ -2529,6 +2636,8 @@ export namespace mcp {
 		}
 	}
 
+	
+	
 
 }
 
@@ -2970,6 +3079,7 @@ export namespace model {
 	    Profiles: Record<string, Profile>;
 	    Themes: Themes;
 	    Version: boolean;
+	    HttpAddress: string;
 	    Help: Help;
 
 	    static createFrom(source: any = {}) {
@@ -2985,6 +3095,7 @@ export namespace model {
 	        this.Profiles = this.convertValues(source["Profiles"], Profile, true);
 	        this.Themes = this.convertValues(source["Themes"], Themes);
 	        this.Version = source["Version"];
+	        this.HttpAddress = source["HttpAddress"];
 	        this.Help = this.convertValues(source["Help"], Help);
 	    }
 

internal/app/mcp-server/main.go

@@ -2,11 +2,12 @@ package mcp_server
 
 import (
 	"fmt"
+	"log/slog"
+	"os"
+
 	"github.com/mark3labs/mcp-go/server"
 	"github.com/rainu/ask-mai/internal/config"
 	mcpServer "github.com/rainu/ask-mai/internal/mcp/server"
-	"log/slog"
-	"os"
 )
 
 type Args struct {
@@ -29,7 +30,13 @@ func Main(args Args) int {
 	ap := cfg.GetActiveProfile()
 
 	ms := mcpServer.NewServer(args.VersionLine, ap.LLM.Tool.BuiltIns, ap.LLM.Tool.Custom)
-	err := server.ServeStdio(ms)
+
+	var err error
+	if cfg.HttpAddress != "" {
+		err = server.NewStreamableHTTPServer(ms).Start(cfg.HttpAddress)
+	} else {
+		err = server.ServeStdio(ms)
+	}
 
 	if err != nil {
 		fmt.Fprintln(os.Stderr, err.Error())

internal/config/model/general.go

@@ -14,7 +14,8 @@ type Config struct {
 	Profiles      map[string]*Profile `yaml:"profiles,omitempty" usage:"Configuration profiles. Each profile has the same structure as the main configuration: "`
 	Themes        Themes              `yaml:"themes,omitempty" usage:"Theme settings for the application: "`
 
-	Version bool `yaml:"version,omitempty" short:"v" usage:"Show the version"`
+	Version     bool   `yaml:"version,omitempty" short:"v" usage:"Show the version"`
+	HttpAddress string `yaml:"http-address,omitempty" usage:"Address to listen on for MCP-HTTP server (e.g. \":8080\"). If not set, the server will run in stdio mode."`
 
 	Help Help `yaml:",inline,omitempty"`
 }