wdh-home
05ee541420
Some checks failed
CI / init (push) Has been cancelled
CI / Frontend node 18.16.0 (push) Has been cancelled
CI / Backend go (1.22) (push) Has been cancelled
CI / devops-test (1.22, 18.16.0) (push) Has been cancelled
CI / release-pr (push) Has been cancelled
CI / release-please (push) Has been cancelled
CI / devops-prod (1.22, 18.x) (push) Has been cancelled
CI / docker (push) Has been cancelled
45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
mcpClient "github.com/mark3labs/mcp-go/client"
|
|
"github.com/mark3labs/mcp-go/client/transport"
|
|
"github.com/mark3labs/mcp-go/mcp"
|
|
)
|
|
|
|
func NewClient(baseURL, name, version, serverName string, headers ...map[string]string) (*mcpClient.Client, error) {
|
|
options := make([]transport.StreamableHTTPCOption, 0, 1)
|
|
if len(headers) > 0 && len(headers[0]) > 0 {
|
|
options = append(options, transport.WithHTTPHeaders(headers[0]))
|
|
}
|
|
|
|
client, err := mcpClient.NewStreamableHttpClient(baseURL, options...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ctx := context.Background()
|
|
if err := client.Start(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
initRequest := mcp.InitializeRequest{}
|
|
initRequest.Params.ProtocolVersion = mcp.LATEST_PROTOCOL_VERSION
|
|
initRequest.Params.ClientInfo = mcp.Implementation{
|
|
Name: name,
|
|
Version: version,
|
|
}
|
|
|
|
result, err := client.Initialize(ctx, initRequest)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if result.ServerInfo.Name != serverName {
|
|
return nil, errors.New("server name mismatch")
|
|
}
|
|
|
|
return client, nil
|
|
}
|