From 338eaa3dac78983b63f95322dddadded3d032bab Mon Sep 17 00:00:00 2001
From: quobix <dave@quobix.com>
Date: Thu, 16 May 2024 17:29:54 -0400
Subject: [PATCH] Fixed spec format setting for 3.1 in `SpecInfo`

---
 datamodel/spec_info.go | 1 +
 document.go            | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/datamodel/spec_info.go b/datamodel/spec_info.go
index a4dd60b..904cbd5 100644
--- a/datamodel/spec_info.go
+++ b/datamodel/spec_info.go
@@ -114,6 +114,7 @@ func ExtractSpecInfoWithDocumentCheck(spec []byte, bypass bool) (*SpecInfo, erro
 			case "3.1.0", "3.1":
 				specInfo.VersionNumeric = 3.1
 				specInfo.APISchema = OpenAPI31SchemaData
+				specInfo.SpecFormat = OAS31
 			default:
 				specInfo.VersionNumeric = 3.0
 				specInfo.APISchema = OpenAPI3SchemaData
diff --git a/document.go b/document.go
index 98aa9f4..5164247 100644
--- a/document.go
+++ b/document.go
@@ -296,7 +296,7 @@ func (d *document) BuildV3Model() (*DocumentModel[v3high.Document], []error) {
 		errs = append(errs, fmt.Errorf("unable to build document, no specification has been loaded"))
 		return nil, errs
 	}
-	if d.info.SpecFormat != datamodel.OAS3 {
+	if d.info.SpecFormat != datamodel.OAS3 && d.info.SpecFormat != datamodel.OAS31 {
 		errs = append(errs, fmt.Errorf("unable to build openapi document, "+
 			"supplied spec is a different version (%v). Try 'BuildV2Model()'", d.info.SpecFormat))
 		return nil, errs