nodes/griptape-ai-ComfyUI-Griptape
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updated google prompt models to have gemini-2.0-flash, gemini-2.0-flash-lite-preview-02-05

Browse Source 
updated google embedding model to text-embedding-004
This commit is contained in:
Jason Schleifer
2025-02-12 05:02:51 +13:00
parent 00b1ed43ab
commit f0803d60c4
2 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/drivers/gtUIGoogleEmbeddingDriver.py
+1 -1
View File
@@ -3,7 +3,7 @@ from griptape.drivers.embedding.google import GoogleEmbeddingDriver
from .gtUIBaseEmbeddingDriver import gtUIBaseEmbeddingDriver
models = [
"models/embedding-001",
"text-embedding-004",
]
task_types = [
"RETRIEVAL_QUERY",
nodes/drivers/gtUIGooglePromptDriver.py
+8 -1
View File
@@ -2,7 +2,14 @@ from griptape.drivers.prompt.google import GooglePromptDriver
from .gtUIBasePromptDriver import gtUIBasePromptDriver
models = ["gemini-1.5-pro", "gemini-1.5-flash", "gemini-1.0-pro", "gemini-pro"]
models = [
"gemini-2.0-flash",
"gemini-2.0-flash-lite-preview-02-05",
"gemini-1.5-pro",
"gemini-1.5-flash",
"gemini-1.0-pro",
"gemini-pro",
]
DEFAULT_API_KEY_ENV_VAR = "GOOGLE_API_KEY"