fix: add extra model

vodyani · Aug 31, 2022 · 00132e3 · 00132e3
1 parent 673fe55
commit 00132e3
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/provider.ts b/src/provider.ts
@@ -1,6 +1,8 @@
 import { Injectable, INestApplication } from '@vodyani/core';
 import { SwaggerDocumentOptions, SwaggerModule, OpenAPIObject, DocumentBuilder } from '@nestjs/swagger';
 
+import { ExtraModelStore } from './struct';
+
 @Injectable()
 export class SwaggerProvider {
   public getConfigBuilder() {
@@ -13,7 +15,11 @@ export class SwaggerProvider {
     config: Omit<OpenAPIObject, 'paths'>,
     options?: SwaggerDocumentOptions,
   ) {
-    const document = SwaggerModule.createDocument(application, config, options);
+    const document = SwaggerModule.createDocument(
+      application,
+      config,
+      { extraModels: ExtraModelStore.get(), ...options },
+    );
 
     SwaggerModule.setup(path, application, document);
   }