added tool config for dalle

GuardingUs-AI · Aug 30, 2023 · 2d67c9e · 2d67c9e
1 parent 7118e79
commit 2d67c9e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/superagi/tools/image_generation/image_generation_toolkit.py b/superagi/tools/image_generation/image_generation_toolkit.py
@@ -15,6 +15,7 @@ def get_tools(self) -> List[BaseTool]:
 
     def get_env_keys(self) -> List[ToolConfiguration]:
         return [
-            ToolConfiguration(key="STABILITY_API_KEY", key_type=ToolConfigKeyType.STRING, is_required= True, is_secret = True),
-            ToolConfiguration(key="ENGINE_ID", key_type=ToolConfigKeyType.STRING, is_required=False, is_secret=False)
+            ToolConfiguration(key="STABILITY_API_KEY", key_type=ToolConfigKeyType.STRING, is_required= False, is_secret = True),
+            ToolConfiguration(key="ENGINE_ID", key_type=ToolConfigKeyType.STRING, is_required=False, is_secret=False),
+            ToolConfiguration(key= "OPENAI API KEY", key_type=ToolConfigKeyType.STRING, is_required=False, is_secret=True),
         ]