diff options
| -rw-r--r-- | builder/builder.go | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/builder/builder.go b/builder/builder.go index 0e90c54..134d7db 100644 --- a/builder/builder.go +++ b/builder/builder.go @@ -74,12 +74,16 @@ func WithResponse(status int, mediaType string, schema *jsonschema.Schema) PathO if value, ok := responses[strconv.Itoa(status)]; ok { response := value.(map[string]interface{}) - response["content"].(map[string]interface{})[mediaType] = schema + response["content"].(map[string]interface{})[mediaType] = map[string]*jsonschema.Schema{ + "schema": schema, + } } else { responses[strconv.Itoa(status)] = map[string]interface{}{ "description": http.StatusText(status), "content": map[string]interface{}{ - mediaType: schema, + mediaType: map[string]*jsonschema.Schema{ + "schema": schema, + }, }, } } |