From efffa8fc5a039da5d23e52c158e5abc085a2bb80 Mon Sep 17 00:00:00 2001 From: Jason Schleifer Date: Thu, 23 May 2024 20:29:03 -0700 Subject: [PATCH] checked for Amazon Bedrock --- nodes/tasks.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/nodes/tasks.py b/nodes/tasks.py index 1352059..5fae797 100644 --- a/nodes/tasks.py +++ b/nodes/tasks.py @@ -1,7 +1,7 @@ import base64 import os -from griptape.drivers import OpenAiImageGenerationDriver +from griptape.drivers import AmazonBedrockImageQueryDriver, OpenAiImageGenerationDriver from griptape.engines import ( ImageQueryEngine, PromptImageGenerationEngine, @@ -262,13 +262,17 @@ class gtUIImageQueryTask(gtUIBaseImageTask): if not agent: agent = Agent() - engine = ImageQueryEngine( - image_query_driver=agent.config.global_drivers.image_query_driver - ) + image_query_driver = agent.config.global_drivers.image_query_driver + engine = ImageQueryEngine(image_query_driver=image_query_driver) image_artifact = ImageLoader().load(base64.b64decode(final_image)) prompt_text = self.get_prompt_text(STRING, input_string) + # If the driver is AmazonBedrock, the prompt_text cannot be empty + if prompt_text.strip() == "": + if isinstance(image_query_driver, AmazonBedrockImageQueryDriver): + prompt_text = "Describe this image" + task = ImageQueryTask( input=(prompt_text, [image_artifact]), image_query_engine=engine )