Add image example

griptape-ai · Nov 13, 2024 · 2767e54 · 2767e54
1 parent 8925ac9
commit 2767e54
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/docs/griptape-framework/drivers/prompt-drivers.md b/docs/griptape-framework/drivers/prompt-drivers.md
@@ -19,6 +19,12 @@ Or use them independently:
 --8<-- "docs/griptape-framework/drivers/src/prompt_drivers_2.py"
 ```
 
+You can pass images to the Driver if the model supports it:
+
+```python
+--8<-- "docs/griptape-framework/drivers/src/prompt_driver_images.py"
+```
+
 ## Prompt Drivers
 
 Griptape offers the following Prompt Drivers for interacting with LLMs.

diff --git a/docs/griptape-framework/drivers/src/prompt_drivers_images.py b/docs/griptape-framework/drivers/src/prompt_drivers_images.py
@@ -0,0 +1,10 @@
+from griptape.artifacts import ListArtifact, TextArtifact
+from griptape.drivers import OpenAiChatPromptDriver
+from griptape.loaders import ImageLoader
+
+driver = OpenAiChatPromptDriver(model="gpt-4o", max_tokens=256)
+
+image_artifact = ImageLoader().load("./tests/resources/mountain.jpg")
+text_artifact = TextArtifact("Describe the weather in the image")
+
+driver.run(ListArtifact([text_artifact, image_artifact]))