added quick workaround for Anthropic Config to not use Voyage API key if not needed.
This commit is contained in:
@@ -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
@@ -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",
|
||||
|
||||
@@ -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",
|
||||
|
||||
@@ -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
@@ -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"]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user