Merge pull request #98 from griptape-ai:dev

updated imageQueryTask to not pull api key
This commit is contained in:
Jason Schleifer
2024-07-30 11:33:18 -07:00
committed by GitHub
2 changed files with 4 additions and 3 deletions
+1 -2
View File
@@ -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):
+3 -1
View File
@@ -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,