diff --git a/nodes/tasks/gtUIImageQueryTask.py b/nodes/tasks/gtUIImageQueryTask.py index 9ce76eb..e0ed065 100644 --- a/nodes/tasks/gtUIImageQueryTask.py +++ b/nodes/tasks/gtUIImageQueryTask.py @@ -1,5 +1,4 @@ import base64 -import os from griptape.drivers import AmazonBedrockPromptDriver, AnthropicPromptDriver from griptape.loaders import ImageLoader @@ -11,7 +10,7 @@ from ..utilities import ( from .gtUIBaseImageTask import gtUIBaseImageTask default_prompt = "{{ input_string }}" -OPENAI_API_KEY = os.getenv("OPENAI_API_KEY") +# OPENAI_API_KEY = os.getenv("OPENAI_API_KEY") class gtUIImageQueryTask(gtUIBaseImageTask): diff --git a/nodes/tools/gtUIKnowledgeBaseTool.py b/nodes/tools/gtUIKnowledgeBaseTool.py index 95a7006..bf19402 100644 --- a/nodes/tools/gtUIKnowledgeBaseTool.py +++ b/nodes/tools/gtUIKnowledgeBaseTool.py @@ -25,7 +25,7 @@ class gtUIKnowledgeBaseTool(gtUIBaseTool): {"default": "GRIPTAPE_API_KEY"}, ), "base_url": ("STRING", {"default": "https://cloud.griptape.ai"}), - "knowledge_base_id": ("STRING", {"default": ""}), + "knowledge_base_id": ("STRING", {"default": "12345-abcde-1434"}), }, } @@ -62,6 +62,8 @@ class gtUIKnowledgeBaseTool(gtUIBaseTool): name = data.get("name", "Griptape Knowledge Base") description = data.get("description", "Contains helpful information") + print(name) + print(description) tool = GriptapeCloudKnowledgeBaseClient( name=name, description=description,