added quick workaround for Anthropic Config to not use Voyage API key if not needed.

This commit is contained in:
Jason Schleifer
2024-10-23 13:26:48 +13:00
parent 343166c415
commit 7df4ab17c2
6 changed files with 41 additions and 15 deletions
+4
View File
@@ -103,6 +103,10 @@ You can previous and download more examples [here](examples/README.md).
## Recent Changelog
### Oct 23, 2024
* Updated Anthropic Claude Prompt Driver to include `claude-3-5-sonnet-20241022`
* Updated Anthropic Claude Config to offer option to not use Voyage API for Embedding Driver. Just set `ignore_voyage_embedding_driver` to `True`
### Oct 11, 2024
* Updated to Griptape Framework v0.33.1 to resolve install bugs
+8 -1
View File
@@ -1,11 +1,18 @@
export const versions = {
"version": "0.33.0",
"version": "0.33.1",
"releaseDate": "2024-10-11",
"name": "ComfyUI-Griptape",
"description": "Griptape integration for ComfyUI",
"author": "Jason Schleifer",
"repository": "https://github.com/griptape-ai/ComfyUI-Griptape",
"changelog": [
{
"version": "0.33.1",
"date": "2024-10-23",
"changes": [
"Update to Griptape v0.33.1",
]
},
{
"version": "0.33.0",
"date": "2024-10-11",
+11 -4
View File
@@ -4,6 +4,7 @@ from griptape.configs.drivers import AnthropicDriversConfig
# StructureGlobalDriversConfig,
from griptape.drivers import (
AnthropicPromptDriver,
DummyEmbeddingDriver,
LocalVectorStoreDriver,
VoyageAiEmbeddingDriver,
)
@@ -17,6 +18,7 @@ from .gtUIBaseDriversConfig import (
)
anthropicPromptModels = [
"claude-3-5-sonnet-20241022",
"claude-3-5-sonnet-20240620",
"claude-3-opus-20240229",
"claude-3-sonnet-20240229",
@@ -70,11 +72,16 @@ class gtUIAnthropicDriversConfig(gtUIBaseDriversConfig):
drivers_config_params["prompt_driver"] = AnthropicPromptDriver(
**prompt_driver_params
)
drivers_config_params["embedding_driver"] = VoyageAiEmbeddingDriver(
**embedding_driver_params
)
if embedding_driver_params == {}:
drivers_config_params["embedding_driver"] = VoyageAiEmbeddingDriver(
**embedding_driver_params
)
else:
drivers_config_params["embedding_driver"] = DummyEmbeddingDriver()
drivers_config_params["vector_store_driver"] = LocalVectorStoreDriver(
embedding_driver=VoyageAiEmbeddingDriver(**embedding_driver_params)
embedding_driver=drivers_config_params["embedding_driver"]
)
try:
@@ -3,6 +3,7 @@ from griptape.drivers import AnthropicPromptDriver
from .gtUIBasePromptDriver import gtUIBasePromptDriver
models = [
"claude-3-5-sonnet-20241022",
"claude-3-5-sonnet-20240620",
"claude-3-opus-20240229",
"claude-3-sonnet-20240229",
+15 -8
View File
@@ -1,4 +1,4 @@
from griptape.drivers import VoyageAiEmbeddingDriver
from griptape.drivers import DummyEmbeddingDriver, VoyageAiEmbeddingDriver
from .gtUIBaseEmbeddingDriver import gtUIBaseEmbeddingDriver
@@ -35,6 +35,7 @@ class gtUIVoyageAiEmbeddingDriver(gtUIBaseEmbeddingDriver):
"STRING",
{"default": "document"},
),
"ignore_voyage_embedding_driver": ("BOOLEAN", {"default": False}),
}
)
@@ -46,16 +47,22 @@ class gtUIVoyageAiEmbeddingDriver(gtUIBaseEmbeddingDriver):
)
model = kwargs.get("embedding_model", "voyage-large-2")
input_type = kwargs.get("input_type", "document")
params = {
"api_key": api_key,
"model": model,
"input_type": input_type,
}
ignore = kwargs.get("ignore_voyage_embedding_driver", False)
if ignore:
params = {
"api_key": api_key,
"model": model,
"input_type": input_type,
}
else:
params = {}
return params
def create(self, **kwargs):
params = self.build_params(**kwargs)
driver = VoyageAiEmbeddingDriver(**params)
if params == {}:
driver = DummyEmbeddingDriver()
else:
driver = VoyageAiEmbeddingDriver(**params)
return (driver,)
+2 -2
View File
@@ -1,6 +1,6 @@
[tool.poetry]
name = "comfyui-griptape"
version = "1.0.5"
version = "1.0.9"
description = "Griptape LLM(Large Language Model) Nodes for ComfyUI."
authors = ["Jason Schleifer <jason.schleifer@gmail.com>"]
readme = "README.md"
@@ -26,7 +26,7 @@ priority = "explicit"
[project]
name = "comfyui-griptape"
description = "Griptape LLM(Large Language Model) Nodes for ComfyUI."
version = "1.0.8"
version = "1.0.9"
license = {file = "LICENSE"}
dependencies = ["griptape[all]==0.33.1", "python-dotenv"]