From 8bd489d27d9144c57de9954cde9ff749a6100dc5 Mon Sep 17 00:00:00 2001
From: Jared Evans <jarede@gmail.com>
Date: Sat, 22 Aug 2026 15:00:38 +0100
Subject: [PATCH 1/4] keep hold of the key name

---
 src/schemaHandler.js | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/schemaHandler.js b/src/schemaHandler.js
index 428553a..5f3a9b8 100644
--- a/src/schemaHandler.js
+++ b/src/schemaHandler.js
@@ -75,7 +75,9 @@ class SchemaHandler {
     const standardisedGatewayModels =
       Object.keys(this.apiGatewayModels).flatMap((key) => {
         const gatewayModel = this.apiGatewayModels[key];
-        return standardModel(gatewayModel);
+        const model = standardModel(gatewayModel);
+        model.apiGatewayKey = key;
+        return model;
       }) || [];
 
     this.models = standardisedModels.concat(
@@ -86,7 +88,8 @@ class SchemaHandler {
 
   async addModelsToOpenAPI() {
     for (const model of this.models) {
-      const modelName = model.name;
+      const modelName = model?.apiGatewayKey || model.name;
+
       const schemas = []
       if (model.schema) {
         // const modelSchema = model.schema;

From a7793bed648b7ad4adcfb550eccb407fe273a94e Mon Sep 17 00:00:00 2001
From: Jared Evans <jarede@gmail.com>
Date: Sat, 22 Aug 2026 15:01:07 +0100
Subject: [PATCH 2/4] check for apigateway models

---
 src/definitionGenerator.js | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/definitionGenerator.js b/src/definitionGenerator.js
index 57a5d81..95e67cc 100644
--- a/src/definitionGenerator.js
+++ b/src/definitionGenerator.js
@@ -696,6 +696,10 @@ class DefinitionGenerator {
     return obj;
   }
 
+  schemaExistsInSchemaModels(model, currentContentType) {
+    return currentContentType === model.name || currentContentType === model?.apiGatewayKey
+  }
+
   async createMediaTypeObject(models, type) {
     const mediaTypeObj = {};
     for (const mediaTypeDocumentation of this.schemaHandler.models) {
@@ -706,9 +710,11 @@ class DefinitionGenerator {
       }
 
       for (const modelContentType in models) {
-        let contentKey
+        let contentKey;
+        let modelName = mediaTypeDocumentation.name;
 
-        if (models[modelContentType] === mediaTypeDocumentation.name) {
+        if (this.schemaExistsInSchemaModels(mediaTypeDocumentation, models[modelContentType])) {
+          if (models[modelContentType] === mediaTypeDocumentation?.apiGatewayKey) { modelName = mediaTypeDocumentation.apiGatewayKey; }
           contentKey = modelContentType;
         }
 
@@ -742,7 +748,7 @@ class DefinitionGenerator {
           }
 
           const schemaRef = await this.schemaHandler
-            .createSchema(mediaTypeDocumentation.name)
+            .createSchema(modelName)
             .catch((err) => {
               throw err;
             });

From a74a335a2e7352d81f15168892f78c22c5c5b295 Mon Sep 17 00:00:00 2001
From: Jared Evans <jarede@gmail.com>
Date: Sat, 22 Aug 2026 15:05:44 +0100
Subject: [PATCH 3/4] 0.1.6

---
 package-lock.json | 4 ++--
 package.json      | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/package-lock.json b/package-lock.json
index fadbe0f..4d826b9 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "serverless-openapi-documenter",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "lockfileVersion": 2,
   "requires": true,
   "packages": {
     "": {
       "name": "serverless-openapi-documenter",
-      "version": "0.1.5",
+      "version": "0.1.6",
       "license": "MIT",
       "dependencies": {
         "@apidevtools/json-schema-ref-parser": "^9.1.0",
diff --git a/package.json b/package.json
index 8944d4e..2501297 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "serverless-openapi-documenter",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "description": "Generate OpenAPI v3 documentation and Postman Collections from your Serverless Config",
   "main": "index.js",
   "keywords": [

From 2331081ca8e218af08acdebc5b1167ce4d30aac5 Mon Sep 17 00:00:00 2001
From: Jared Evans <jarede@gmail.com>
Date: Sat, 22 Aug 2026 15:06:12 +0100
Subject: [PATCH 4/4] fixes the repo link

---
 package.json | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/package.json b/package.json
index 2501297..87507f2 100644
--- a/package.json
+++ b/package.json
@@ -46,7 +46,7 @@
   },
   "repository": {
     "type": "git",
-    "url": "https://github.com/JaredCE/serverless-openapi-documenter.git"
+    "url": "git+https://github.com/JaredCE/serverless-openapi-documenter.git"
   },
   "bugs": {
     "url": "https://github.com/JaredCE/serverless-openapi-documenter/issues"