From 558003704e0427a08a16516ab93c6e4260a39a10 Mon Sep 17 00:00:00 2001
From: Andres Caicedo <73312784+AndresCdo@users.noreply.github.com>
Date: Fri, 21 Apr 2023 04:00:44 +0200
Subject: [PATCH] Add missing `size` param to generate_image_with_dalle (#2691)

---
 autogpt/commands/image_gen.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/autogpt/commands/image_gen.py b/autogpt/commands/image_gen.py
index 60cdaec0..962c41fb 100644
--- a/autogpt/commands/image_gen.py
+++ b/autogpt/commands/image_gen.py
@@ -77,12 +77,13 @@ def generate_image_with_hf(prompt: str, filename: str) -> str:
     return f"Saved to disk:{filename}"
 
 
-def generate_image_with_dalle(prompt: str, filename: str) -> str:
+def generate_image_with_dalle(prompt: str, filename: str, size: int) -> str:
     """Generate an image with DALL-E.
 
     Args:
         prompt (str): The prompt to use
         filename (str): The filename to save the image to
+        size (int): The size of the image
 
     Returns:
         str: The filename of the image