Add support for GenericMessageContent to AmazonBedrockPromptDriver

CHANGELOG.md

@@ -16,6 +16,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Support for `BranchTask` in `StructureVisualizer`.
 - `EvalEngine` for evaluating the performance of an LLM's output against a given input.
 - `BaseFileLoader.save()` method for saving an Artifact to a destination.
+- Support for `GenericMessageContent` in `AmazonBedrockPromptDriver`.
 
 ### Changed
 

griptape/drivers/prompt/amazon_bedrock_prompt_driver.py

@@ -22,6 +22,7 @@
     BaseDeltaMessageContent,
     BaseMessageContent,
     DeltaMessage,
+    GenericMessageContent,
     ImageMessageContent,
     Message,
     TextDeltaMessageContent,
@@ -187,6 +188,8 @@ def __to_bedrock_message_content(self, content: BaseMessageContent) -> dict:
                     "status": "error" if isinstance(artifact, ErrorArtifact) else "success",
                 },
             }
+        elif isinstance(content, GenericMessageContent):
+            return content.artifact.value
         else:
             raise ValueError(f"Unsupported content type: {type(content)}")
 

tests/unit/drivers/prompt/test_amazon_bedrock_prompt_driver.py

@@ -1,6 +1,6 @@
 import pytest
 
-from griptape.artifacts import ActionArtifact, ErrorArtifact, ImageArtifact, ListArtifact, TextArtifact
+from griptape.artifacts import ActionArtifact, ErrorArtifact, GenericArtifact, ImageArtifact, ListArtifact, TextArtifact
 from griptape.common import ActionCallDeltaMessageContent, PromptStack, TextDeltaMessageContent, ToolAction
 from griptape.drivers import AmazonBedrockPromptDriver
 from tests.mocks.mock_tool.tool import MockTool
@@ -299,6 +299,7 @@ def prompt_stack(self, request):
                 ]
             )
         )
+        prompt_stack.add_user_message(GenericArtifact("video-file"))
 
         return prompt_stack
 
@@ -354,6 +355,7 @@ def messages(self):
                 ],
                 "role": "user",
             },
+            {"content": ["video-file"], "role": "user"},
         ]
 
     @pytest.mark.parametrize("use_native_tools", [True, False])