Remove voyageai from anthropic drivers config (#1277)

griptape-ai · Oct 23, 2024 · f6ab3ed · f6ab3ed
1 parent 892f639
commit f6ab3ed
Show file tree

Hide file tree

Showing 6 changed files with 40 additions and 20 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -31,6 +31,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
   - If `EventListener.handler` returns `None`, the event will not be published to the `event_listener_driver`.
   - If `EventListener.handler` is None, the event will be published to the `event_listener_driver` as-is.
 - **BREAKING**: Moved `griptape.common.observable.observable` to `griptape.common.decorators.observable`.
+- **BREAKING**: `AnthropicDriversConfig` no longer bundles `VoyageAiEmbeddingDriver`.
 - Updated `EventListener.handler` return type to `Optional[BaseEvent | dict]`.
 - `BaseTask.parent_outputs` type has changed from `dict[str, str | None]` to `dict[str, BaseArtifact]`.
 - `Workflow.context["parent_outputs"]` type has changed from `dict[str, str | None]` to `dict[str, BaseArtifact]`.

diff --git a/MIGRATION.md b/MIGRATION.md
@@ -2,6 +2,40 @@
 
 This document provides instructions for migrating your codebase to accommodate breaking changes introduced in new versions of Griptape.
 
+## 0.34.X to 0.35.X
+
+### `AnthropicDriversConfig` Embedding Driver
+
+`AnthropicDriversConfig` no longer bundles `VoyageAiEmbeddingDriver`. If you rely on embeddings when using Anthropic, you must specify an Embedding Driver yourself.
+
+#### Before
+
+```python
+from griptape.configs import Defaults
+from griptape.configs.drivers import AnthropicDriversConfig
+from griptape.structures import Agent
+
+Defaults.drivers_config = AnthropicDriversConfig()
+
+agent = Agent()
+```
+
+#### After
+
+```python
+from griptape.configs import Defaults
+from griptape.configs.drivers import AnthropicDriversConfig
+from griptape.drivers import VoyageAiEmbeddingDriver, LocalVectorStoreDriver
+
+Defaults.drivers_config = AnthropicDriversConfig(
+    embedding_driver=VoyageAiEmbeddingDriver(),
+    vector_store_driver=LocalVectorStoreDriver(
+        embedding_driver=VoyageAiEmbeddingDriver()
+    )
+)
+```
+
+
 ## 0.33.X to 0.34.X
 
 ### Removed `CompletionChunkEvent`

diff --git a/docs/griptape-framework/structures/configs.md b/docs/griptape-framework/structures/configs.md
@@ -64,7 +64,6 @@ The [Anthropic Driver config](../../reference/griptape/configs/drivers/anthropic
 
 !!! info
     Anthropic does not provide an embeddings API which means you will need to use another service for embeddings.
-    The `AnthropicDriversConfig` defaults to using `VoyageAiEmbeddingDriver` which integrates with [VoyageAI](https://www.voyageai.com/), the service used in Anthropic's [embeddings documentation](https://docs.anthropic.com/claude/docs/embeddings).
     To override the default embedding driver, see: [Override Default Structure Embedding Driver](../drivers/embedding-drivers.md#override-default-structure-embedding-driver).
 
 ```python

diff --git a/docs/griptape-tools/official-tools/prompt-summary-tool.md b/docs/griptape-tools/official-tools/prompt-summary-tool.md
@@ -4,7 +4,7 @@ The [PromptSummaryTool](../../reference/griptape/tools/prompt_summary/tool.md) e
 --8<-- "docs/griptape-tools/official-tools/src/prompt_summary_tool_1.py"
 ```
 
-````
+```
 [08/12/24 15:54:46] INFO     ToolkitTask 8be73eb542c44418ba880399044c017a
                              Input: How can I build Neovim from source for MacOS according to this https://github.com/neovim/neovim/blob/master/BUILD.md
 [08/12/24 15:54:47] INFO     Subtask cd362a149e1d400997be93c1342d1663
@@ -103,4 +103,4 @@ The [PromptSummaryTool](../../reference/griptape/tools/prompt_summary/tool.md) e
 
                              By following these steps, you should be able to build and install Neovim from source on macOS. For more detailed instructions and
                              troubleshooting tips, refer to the [BUILD.md](https://github.com/neovim/neovim/blob/master/BUILD.md) file in the Neovim repository.
-````
+```
diff --git a/griptape/configs/drivers/anthropic_drivers_config.py b/griptape/configs/drivers/anthropic_drivers_config.py
@@ -4,8 +4,6 @@
 from griptape.drivers import (
     AnthropicImageQueryDriver,
     AnthropicPromptDriver,
-    LocalVectorStoreDriver,
-    VoyageAiEmbeddingDriver,
 )
 from griptape.utils.decorators import lazy_property
 
@@ -16,14 +14,6 @@ class AnthropicDriversConfig(DriversConfig):
     def prompt_driver(self) -> AnthropicPromptDriver:
         return AnthropicPromptDriver(model="claude-3-5-sonnet-20240620")
 
-    @lazy_property()
-    def embedding_driver(self) -> VoyageAiEmbeddingDriver:
-        return VoyageAiEmbeddingDriver(model="voyage-large-2")
-
-    @lazy_property()
-    def vector_store_driver(self) -> LocalVectorStoreDriver:
-        return LocalVectorStoreDriver(embedding_driver=VoyageAiEmbeddingDriver(model="voyage-large-2"))
-
     @lazy_property()
     def image_query_driver(self) -> AnthropicImageQueryDriver:
         return AnthropicImageQueryDriver(model="claude-3-5-sonnet-20240620")
diff --git a/tests/unit/configs/drivers/test_anthropic_drivers_config.py b/tests/unit/configs/drivers/test_anthropic_drivers_config.py
@@ -33,16 +33,12 @@ def test_to_dict(self, config):
                 "max_tokens": 256,
             },
             "embedding_driver": {
-                "type": "VoyageAiEmbeddingDriver",
-                "model": "voyage-large-2",
-                "input_type": "document",
+                "type": "DummyEmbeddingDriver",
             },
             "vector_store_driver": {
-                "type": "LocalVectorStoreDriver",
+                "type": "DummyVectorStoreDriver",
                 "embedding_driver": {
-                    "type": "VoyageAiEmbeddingDriver",
-                    "model": "voyage-large-2",
-                    "input_type": "document",
+                    "type": "DummyEmbeddingDriver",
                 },
             },
             "conversation_memory_driver": {