Skip to content

Commit e264cf0

Browse files
authored
Fix printing MCP tool name when doing registry list (#155)
Signed-off-by: Juan Antonio Osorio <ozz@stacklok.com>
1 parent 4ac7427 commit e264cf0

File tree

3 files changed

+8
-7
lines changed

3 files changed

+8
-7
lines changed

cmd/thv/app/registry.go

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -126,13 +126,8 @@ func printTextServers(servers []*registry.Server) {
126126

127127
// Print server information
128128
for _, server := range servers {
129-
// Extract server name from image
130-
name := strings.Split(server.Image, ":")[0]
131-
name = strings.TrimPrefix(name, "mcp/")
132-
133-
// Print server information
134129
fmt.Fprintf(w, "%s\t%s\t%s\t%d\t%d\n",
135-
name,
130+
server.Name,
136131
truncateString(server.Description, 60),
137132
server.Transport,
138133
server.Metadata.Stars,
@@ -149,7 +144,7 @@ func printTextServers(servers []*registry.Server) {
149144
// printTextServerInfo prints detailed information about a server in text format
150145
// nolint:gocyclo
151146
func printTextServerInfo(name string, server *registry.Server) {
152-
fmt.Printf("Name: %s\n", name)
147+
fmt.Printf("Name: %s\n", server.Name)
153148
fmt.Printf("Image: %s\n", server.Image)
154149
fmt.Printf("Description: %s\n", server.Description)
155150
fmt.Printf("Transport: %s\n", server.Transport)

pkg/registry/registry.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,11 @@ func GetRegistry() (*Registry, error) {
3333
registryErr = fmt.Errorf("failed to parse registry data: %w", err)
3434
return
3535
}
36+
37+
// Set name field on each server based on map key
38+
for name, server := range registry.Servers {
39+
server.Name = name
40+
}
3641
})
3742

3843
return registry, registryErr

pkg/registry/types.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ type Registry struct {
1616

1717
// Server represents an MCP server in the registry
1818
type Server struct {
19+
Name string `json:"name,omitempty"`
1920
Image string `json:"image"`
2021
Description string `json:"description"`
2122
Transport string `json:"transport"`

0 commit comments

Comments
 (0)