1 files changed, 6 insertions, 6 deletions
diff --git a/src/client/views/nodes/DataVizBox/DocCreatorMenu/Backend/TemplateManager.ts b/src/client/views/nodes/DataVizBox/DocCreatorMenu/Backend/TemplateManager.ts
index 6d63078a8..ef7dbc7ab 100644
--- a/src/client/views/nodes/DataVizBox/DocCreatorMenu/Backend/TemplateManager.ts
+++ b/src/client/views/nodes/DataVizBox/DocCreatorMenu/Backend/TemplateManager.ts
@@ -2,10 +2,10 @@ import { action, makeAutoObservable } from 'mobx';
 import { Col } from '../DocCreatorMenu';
 import { FieldSettings } from '../TemplateFieldTypes/TemplateField';
 import { Template } from '../Template';
-import { NumListCast } from '../../../../../../fields/Doc';
+import { Doc, NumListCast } from '../../../../../../fields/Doc';
 import { DataVizBox } from '../../DataVizBox';
 import { TemplateFieldType } from '../TemplateBackend';
-import { TemplateMenuGPTManager } from './TemplateMenuGPTManager';
+import { TemplateMenuAIUtils } from './TemplateMenuAIUtils';
 
 export type Conditional = {
     field: string;
@@ -28,7 +28,7 @@ export class TemplateManager {
     }
 
     initializeTemplates = (templateSettings: FieldSettings[]) => templateSettings.map(settings => {
-        return new Template(settings, this.fieldConditions)});
+        return new Template(settings)});
 
     getValidTemplates = (cols: Col[]) => this.templates.filter(template => template.isValidTemplate(cols));
 
@@ -53,7 +53,7 @@ export class TemplateManager {
         }
     }
 
-    createDocsFromTemplate = action((dv: DataVizBox, template: Template, csvColumns: Col[], GPTManager: TemplateMenuGPTManager, debug: boolean = false) => {
+    createDocsFromTemplate = action((dv: DataVizBox, template: Template, csvColumns: Col[], debug: boolean = false) => {
         const fields = Array.from(Object.keys(dv.records[0]));
 
         const processContent = (content: { [title: string]: string }) => {
@@ -66,7 +66,7 @@ export class TemplateManager {
                     field && field.setContent(content[title], field.viewType);
                 });
 
-            const gptFunc = (type: TemplateFieldType) => (type === TemplateFieldType.VISUAL ? GPTManager.renderGPTImageCall : GPTManager.renderGPTTextCall);
+            const gptFunc = (type: TemplateFieldType) => (type === TemplateFieldType.VISUAL ? TemplateMenuAIUtils.renderGPTImageCall : TemplateMenuAIUtils.renderGPTTextCall);
             const gptPromises = csvColumns
                 .filter(field => field.type !== TemplateFieldType.UNSET && field.AIGenerated)
                 .map(field => {
@@ -76,7 +76,7 @@ export class TemplateManager {
                     }
                 });
 
-            return Promise.all(gptPromises)
+            return templateCopy.getRenderedDoc();
         };
 
         const rowContents = debug