From b0aa83b86e481f374b8d7f45c8bbfc3ee295b0e5 Mon Sep 17 00:00:00 2001
From: Jason Schleifer <jason.schleifer@gmail.com>
Date: Sun, 21 Jul 2024 16:00:13 +1200
Subject: [PATCH] updated some parameters to proper drivers

---
 nodes/tasks/gtUIAudioTranscriptionTask.py    | 1 +
 nodes/tasks/gtUIVectorStoreQueryTask.py      | 2 +-
 nodes/tasks/gtUIVectorStoreUpsertTextTask.py | 2 +-
 nodes/tools/gtUIWebSearch.py                 | 2 +-
 4 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/nodes/tasks/gtUIAudioTranscriptionTask.py b/nodes/tasks/gtUIAudioTranscriptionTask.py
index 45f26ac..bec494c 100644
--- a/nodes/tasks/gtUIAudioTranscriptionTask.py
+++ b/nodes/tasks/gtUIAudioTranscriptionTask.py
@@ -34,6 +34,7 @@ class gtUIAudioTranscriptionTask(gtUIBaseAudioTask):
         inputs["optional"].update(
             {
                 "agent": ("AGENT",),
+                "driver": ("AUDIO_TRANSCRIPTION_DRIVER", {"default": None}),
             }
         )
         return inputs
diff --git a/nodes/tasks/gtUIVectorStoreQueryTask.py b/nodes/tasks/gtUIVectorStoreQueryTask.py
index f5ff32b..cec949d 100644
--- a/nodes/tasks/gtUIVectorStoreQueryTask.py
+++ b/nodes/tasks/gtUIVectorStoreQueryTask.py
@@ -27,7 +27,7 @@ class gtUIVectorStoreQueryTask(gtUIBaseVectorStoreTask):
 
     RETURN_TYPES = (
         "AGENT",
-        "DRIVER",
+        "VECTOR_STORE_DRIVER",
         "STRING",
     )
     RETURN_NAMES = (
diff --git a/nodes/tasks/gtUIVectorStoreUpsertTextTask.py b/nodes/tasks/gtUIVectorStoreUpsertTextTask.py
index 1f0fc37..8c81c01 100644
--- a/nodes/tasks/gtUIVectorStoreUpsertTextTask.py
+++ b/nodes/tasks/gtUIVectorStoreUpsertTextTask.py
@@ -23,7 +23,7 @@ class gtUIVectorStoreUpsertTextTask(gtUIBaseVectorStoreTask):
 
     RETURN_TYPES = (
         "AGENT",
-        "DRIVER",
+        "VECTOR_STORE_DRIVER",
     )
     RETURN_NAMES = (
         "AGENT",
diff --git a/nodes/tools/gtUIWebSearch.py b/nodes/tools/gtUIWebSearch.py
index 40ae3b2..e5a309a 100644
--- a/nodes/tools/gtUIWebSearch.py
+++ b/nodes/tools/gtUIWebSearch.py
@@ -18,7 +18,7 @@ class gtUIWebSearch(gtUIBaseTool):
         inputs = super().INPUT_TYPES()
         inputs["optional"].update(
             {
-                "driver": ("WEB_SCRAPER_DRIVER", {"default": None}),
+                "driver": ("WEB_SEARCH_DRIVER", {"default": None}),
             }
         )
         return inputs