[fix] Support OAS 3.1X

Update spec_info.go , handle version prefix
pb33f · Nov 18, 2024 · 4318337 · 4318337
1 parent c757033
commit 4318337
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/datamodel/spec_info.go b/datamodel/spec_info.go
@@ -117,8 +117,13 @@ func ExtractSpecInfoWithDocumentCheck(spec []byte, bypass bool) (*SpecInfo, erro
 		specInfo.Version = version
 		specInfo.SpecFormat = OAS3
 
-		switch specInfo.Version {
-		case "3.1.0", "3.1":
+		// Extract the prefix version
+		prefixVersion := specInfo.Version
+		if len(specInfo.Version) >= 3 {
+			prefixVersion = specInfo.Version[:3]
+		}
+		switch prefixVersion {
+		case "3.1":
 			specInfo.VersionNumeric = 3.1
 			specInfo.APISchema = OpenAPI31SchemaData
 			specInfo.SpecFormat = OAS31