minor image edit

2 files changed, 4 insertions, 1 deletions

diff --git a/src/client/views/nodes/chatbot/agentsystem/Agent.ts b/src/client/views/nodes/chatbot/agentsystem/Agent.ts
index d7aee51e6..6ef52c0f5 100644
--- a/src/client/views/nodes/chatbot/agentsystem/Agent.ts
+++ b/src/client/views/nodes/chatbot/agentsystem/Agent.ts
@@ -30,6 +30,7 @@ import { CreateNewTool } from '../tools/CreateNewTool';
 import { SortDocsTool} from '../tools/SortDocsTool';
 import { TagDocsTool } from '../tools/TagDocsTool';
 import { GPTTutorialTool } from '../tools/TutorialTool';
+import { ImageCreationTool } from '../tools/ImageCreationTool';
 import { DocumentView } from '../../DocumentView';
 
 dotenv.config();
@@ -102,7 +103,7 @@ export class Agent {
             websiteInfoScraper: new WebsiteInfoScraperTool(this._docManager),
             searchTool: new SearchTool(this._docManager),
             noTool: new NoTool(),
-            //imageCreationTool: new ImageCreationTool(createImage),
+            imageCreationTool: new ImageCreationTool(createImage),
             documentMetadata: new DocumentMetadataTool(this._docManager),
             createLinks: new CreateLinksTool(this._docManager),
             codebaseSummarySearch: new CodebaseSummarySearchTool(this.vectorstore),
diff --git a/src/client/views/nodes/chatbot/tools/ImageCreationTool.ts b/src/client/views/nodes/chatbot/tools/ImageCreationTool.ts
index 37907fd4f..c5b1e028b 100644
--- a/src/client/views/nodes/chatbot/tools/ImageCreationTool.ts
+++ b/src/client/views/nodes/chatbot/tools/ImageCreationTool.ts
@@ -26,6 +26,8 @@ const imageCreationToolInfo: ToolInfo<ImageCreationToolParamsType> = {
 };
 
 export class ImageCreationTool extends BaseTool<ImageCreationToolParamsType> {
+
+
     private _createImage: (result: Upload.FileInformation & Upload.InspectionResults, options: DocumentOptions) => void;
     constructor(createImage: (result: Upload.FileInformation & Upload.InspectionResults, options: DocumentOptions) => void) {
         super(imageCreationToolInfo);