change name to RawName and have method for Name() that is normalized

Author: burmudar

internal/mcp/mcp_parse.go

@@ -12,10 +12,15 @@ import (
 var mcpToolListJSON []byte
 
 type ToolDef struct {
-	Name         string       `json:"name"`
-	Description  string       `json:"description"`
-	InputSchema  SchemaObject `json:"inputSchema"`
-	OutputSchema SchemaObject `json:"outputSchema"`
+	RawName      string `json:"name"`
+	Description  string `json:"description"`
+	InputSchema  Schema `json:"inputSchema"`
+	OutputSchema Schema `json:"outputSchema"`
+}
+
+func (m *ToolDef) Name() string {
+	name, _ := strings.CutPrefix(m.RawName, "sg_")
+	return strings.ReplaceAll(name, "_", "-")
 }
 
 type RawSchema struct {
@@ -93,6 +98,10 @@ func loadToolDefinitions(data []byte) (map[string]*ToolDef, error) {
 		}
 	}
 
+	// make it so that can find a tool definition by it's original name (RawName) and normalized name (Name())
+	tools[def.RawName] = def
+	tools[def.Name()] = def
+
 	if len(decoder.errors) > 0 {
 		return tools, errors.Append(nil, decoder.errors...)
 	}
@@ -162,9 +171,3 @@ func (d *decoder) decodeProperties(props map[string]json.RawMessage) map[string]
 	}
 	return res
 }
-
-// normalizeToolName takes mcp tool names like 'sg_keyword_search' and normalizes it to 'keyword-search"
-func normalizeToolName(toolName string) string {
-	toolName, _ = strings.CutPrefix(toolName, "sg_")
-	return strings.ReplaceAll(toolName, "_", "-")
-}

internal/mcp/mcp_parse_test.go

@@ -46,14 +46,13 @@ func TestLoadToolDefinitions(t *testing.T) {
 		t.Fatalf("Expected 1 tool, got %d", len(tools))
 	}
 
-	// Temporary: map keys have normalized names
 	tool := tools["test-tool"]
 	if tool == nil {
 		t.Fatal("Tool 'test_tool' not found")
 	}
 
-	if tool.Name != "test_tool" {
-		t.Errorf("Expected name 'test_tool', got '%s'", tool.Name)
+	if tool.RawName != "sg_test_tool" {
+		t.Errorf("Expected name 'sg_test_tool', got '%s'", tool.RawName)
 	}
 
 	inputSchema := tool.InputSchema