updated to griptape1.0

removed engine
griptape-ai · Dec 11, 2024 · 7880658 · 7880658
1 parent 86c4c8b
commit 7880658
Show file tree

Hide file tree

Showing 7 changed files with 15 additions and 36 deletions.
diff --git a/README.md b/README.md
@@ -7,17 +7,14 @@ This extension provides an [Image Generation Driver](https://docs.griptape.ai/st
 from griptape.black_forest.drivers.black_forest_image_generation_driver import (
     BlackForestImageGenerationDriver,
 )
-from griptape.engines import PromptImageGenerationEngine
 from griptape.structures import Agent
 from griptape.tools import FileManagerTool, PromptImageGenerationTool
 
 agent = Agent(
     tools=[
         PromptImageGenerationTool(
-            engine=PromptImageGenerationEngine(
-                image_generation_driver=BlackForestImageGenerationDriver(
+            image_generation_driver=BlackForestImageGenerationDriver(
                     model="flux-pro-1.1"
-                )
             ),
             off_prompt=True,
         ),

diff --git a/examples/drivers/example_agent.py b/examples/drivers/example_agent.py
@@ -1,17 +1,14 @@
 from griptape.black_forest.drivers.black_forest_image_generation_driver import (
     BlackForestImageGenerationDriver,
 )
-from griptape.engines import PromptImageGenerationEngine
 from griptape.structures import Agent
 from griptape.tools import FileManagerTool, PromptImageGenerationTool
 
 agent = Agent(
     tools=[
         PromptImageGenerationTool(
-            engine=PromptImageGenerationEngine(
-                image_generation_driver=BlackForestImageGenerationDriver(
-                    model="flux-pro-1.1"
-                )
+            image_generation_driver=BlackForestImageGenerationDriver(
+                model="flux-pro-1.1"
             ),
             off_prompt=True,
         ),

diff --git a/examples/drivers/example_agent_canny.py b/examples/drivers/example_agent_canny.py
@@ -1,18 +1,15 @@
 from griptape.black_forest.drivers.black_forest_image_generation_driver import (
     BlackForestImageGenerationDriver,
 )
-from griptape.engines import VariationImageGenerationEngine
 from griptape.structures import Agent
 from griptape.tools import FileManagerTool, VariationImageGenerationTool
 
 agent = Agent(
     tools=[
         VariationImageGenerationTool(
-            engine=VariationImageGenerationEngine(
-                image_generation_driver=BlackForestImageGenerationDriver(
-                    model="flux-pro-1.0-canny",  # flux-pro-1.0-canny is required for Canny.
-                    guidance=100,
-                )
+            image_generation_driver=BlackForestImageGenerationDriver(
+                model="flux-pro-1.0-canny",  # flux-pro-1.0-canny is required for Canny.
+                guidance=100,
             ),
             off_prompt=True,
         ),

diff --git a/examples/drivers/example_agent_depth.py b/examples/drivers/example_agent_depth.py
@@ -1,17 +1,14 @@
 from griptape.black_forest.drivers.black_forest_image_generation_driver import (
     BlackForestImageGenerationDriver,
 )
-from griptape.engines import VariationImageGenerationEngine
 from griptape.structures import Agent
 from griptape.tools import FileManagerTool, VariationImageGenerationTool
 
 agent = Agent(
     tools=[
         VariationImageGenerationTool(
-            engine=VariationImageGenerationEngine(
-                image_generation_driver=BlackForestImageGenerationDriver(
-                    model="flux-pro-1.0-depth",  # flux-pro-1.0-depth is required for Depth.
-                )
+            image_generation_driver=BlackForestImageGenerationDriver(
+                model="flux-pro-1.0-depth",  # flux-pro-1.0-depth is required for Depth.
             ),
             off_prompt=True,
         ),

diff --git a/examples/drivers/example_agent_inpainting.py b/examples/drivers/example_agent_inpainting.py
@@ -1,17 +1,14 @@
 from griptape.black_forest.drivers.black_forest_image_generation_driver import (
     BlackForestImageGenerationDriver,
 )
-from griptape.engines import InpaintingImageGenerationEngine
 from griptape.structures import Agent
 from griptape.tools import FileManagerTool, InpaintingImageGenerationTool
 
 agent = Agent(
     tools=[
         InpaintingImageGenerationTool(
-            engine=InpaintingImageGenerationEngine(
-                image_generation_driver=BlackForestImageGenerationDriver(
-                    model="flux-pro-1.0",  # flux-pro-1.0 is required for inpainting.
-                )
+            image_generation_driver=BlackForestImageGenerationDriver(
+                model="flux-pro-1.0",  # flux-pro-1.0 is required for inpainting.
             ),
             off_prompt=True,
         ),

diff --git a/examples/drivers/example_agent_outpainting.py b/examples/drivers/example_agent_outpainting.py
@@ -1,17 +1,14 @@
 from griptape.black_forest.drivers.black_forest_image_generation_driver import (
     BlackForestImageGenerationDriver,
 )
-from griptape.engines import OutpaintingImageGenerationEngine
 from griptape.structures import Agent
 from griptape.tools import FileManagerTool, OutpaintingImageGenerationTool
 
 agent = Agent(
     tools=[
         OutpaintingImageGenerationTool(
-            engine=OutpaintingImageGenerationEngine(
-                image_generation_driver=BlackForestImageGenerationDriver(
-                    model="flux-pro-1.0",  # flux-pro-1.0 is required for outpainting.
-                )
+            image_generation_driver=BlackForestImageGenerationDriver(
+                model="flux-pro-1.0",  # flux-pro-1.0 is required for outpainting.
             ),
             off_prompt=True,
         ),

diff --git a/examples/drivers/example_agent_variation.py b/examples/drivers/example_agent_variation.py
@@ -1,7 +1,6 @@
 from griptape.black_forest.drivers.black_forest_image_generation_driver import (
     BlackForestImageGenerationDriver,
 )
-from griptape.engines import VariationImageGenerationEngine
 from griptape.structures import Agent
 from griptape.tools import (
     FileManagerTool,
@@ -11,11 +10,9 @@
 agent = Agent(
     tools=[
         VariationImageGenerationTool(
-            engine=VariationImageGenerationEngine(
-                image_generation_driver=BlackForestImageGenerationDriver(
-                    model="flux-pro-1.1-ultra",  # flux-pro-1.1-ultra is a better model for image variation
-                    image_prompt_strength=0.7,
-                )
+            image_generation_driver=BlackForestImageGenerationDriver(
+                model="flux-pro-1.1-ultra",  # flux-pro-1.1-ultra is a better model for image variation
+                image_prompt_strength=0.7,
             ),
             off_prompt=True,
         ),