Move to dalle3

main.py

@@ -3,9 +3,18 @@ from textwrap import dedent
 from crewai import Agent, Task, Crew, Process
 from typing import List
 from pydantic import BaseModel, Field
-from langchain.agents import load_tools
 import asyncio
 from concurrent.futures import ThreadPoolExecutor
+from langchain_community.utilities.dalle_image_generator import DallEAPIWrapper
+
+from langchain_core.tools import Tool
+
+dalle3 = Tool(
+    "Dall-E-Image-Generator",
+    DallEAPIWrapper(model="dall-e-3").run,
+    "A wrapper around OpenAI DALL-E API. Useful for when you need to generate images from a text description. Input "
+    "should be an image description.",
+)
 
 
 class Step(BaseModel):
@@ -46,7 +55,7 @@ illustrator = Agent(
     backstory=dedent(""" You are a really capable artist that can draw very photo-realistic illustration 
         for each step of a recipe.
     """),
-    tools=load_tools(["dalle-image-generator"]),
+    tools=[dalle3],
     verbose=True,
     memory=True
 )
@@ -69,7 +78,8 @@ rewrite_task = Task(
 )
 
 illustrate_task = Task(
-    description=dedent("""For each step the recipe create a photo-realistic illustration. Please keep the full URL for images as they were provided by the tool"""),
+    description=dedent(
+        """For each step the recipe create a photo-realistic illustration. Please keep the full URL for images as they were provided by the tool"""),
     context=[recipe_task],
     expected_output="The list of ingredients and a list containing each step of recipe along with the link to the its respective illustration. Please keep the entire url for the images, do not short them.",
     agent=illustrator