fix: cr

Author: Crescent617

mcp/utils.go

@@ -492,7 +492,7 @@ func ExtractString(data map[string]any, key string) string {
 	return ""
 }
 
-func ParseAnnotaions(data map[string]any) *Annotations {
+func ParseAnnotations(data map[string]any) *Annotations {
 	if data == nil {
 		return nil
 	}
@@ -504,12 +504,17 @@ func ParseAnnotaions(data map[string]any) *Annotations {
 	}
 
 	if value, ok := data["audience"]; ok {
-		if audience, ok := value.([]any); ok {
+		switch audience := value.(type) {
+		case []any:
 			for _, a := range audience {
 				if roleStr, ok := a.(string); ok {
 					annotations.Audience = append(annotations.Audience, Role(roleStr))
 				}
 			}
+		case []string:
+			for _, a := range audience {
+				annotations.Audience = append(annotations.Audience, Role(a))
+			}
 		}
 	}
 	return annotations
@@ -530,7 +535,7 @@ func ParseContent(contentMap map[string]any) (Content, error) {
 
 	var annotations *Annotations
 	if annotationsMap := ExtractMap(contentMap, "annotations"); annotationsMap != nil {
-		annotations = ParseAnnotaions(annotationsMap)
+		annotations = ParseAnnotations(annotationsMap)
 	}
 
 	switch contentType {

mcp/utils_test.go

@@ -13,13 +13,13 @@ func TestParseAnnotations(t *testing.T) {
 		expected *Annotations
 	}{
 		{
-			name: "nil data",
-			data: nil,
+			name:     "nil data",
+			data:     nil,
 			expected: nil,
 		},
 		{
-			name: "empty data",
-			data: map[string]any{},
+			name:     "empty data",
+			data:     map[string]any{},
 			expected: &Annotations{},
 		},
 		{
@@ -74,11 +74,20 @@ func TestParseAnnotations(t *testing.T) {
 				Audience: []Role{"user", "assistant"},
 			},
 		},
+		{
+			name: "audience as []string",
+			data: map[string]any{
+				"audience": []string{"assistant", "user"},
+			},
+			expected: &Annotations{
+				Audience: []Role{"assistant", "user"},
+			},
+		},
 	}
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			result := ParseAnnotaions(tt.data)
+			result := ParseAnnotations(tt.data)
 			assert.Equal(t, tt.expected, result)
 		})
 	}
@@ -296,13 +305,13 @@ func TestParseContent(t *testing.T) {
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			result, err := ParseContent(tt.contentMap)
-			
+
 			if tt.expectError {
 				assert.Error(t, err)
 				assert.Nil(t, result)
 			} else {
 				assert.NoError(t, err)
-				
+
 				// Compare the actual content values
 				switch exp := tt.expected.(type) {
 				case TextContent:
@@ -346,4 +355,5 @@ func TestParseContent(t *testing.T) {
 			}
 		})
 	}
-}
+}
+