updated some parameters to proper drivers

This commit is contained in:
Jason Schleifer
2024-07-21 16:00:13 +12:00
parent 9f4f664d5b
commit b0aa83b86e
4 changed files with 4 additions and 3 deletions
@@ -34,6 +34,7 @@ class gtUIAudioTranscriptionTask(gtUIBaseAudioTask):
inputs["optional"].update(
{
"agent": ("AGENT",),
"driver": ("AUDIO_TRANSCRIPTION_DRIVER", {"default": None}),
}
)
return inputs
+1 -1
View File
@@ -27,7 +27,7 @@ class gtUIVectorStoreQueryTask(gtUIBaseVectorStoreTask):
RETURN_TYPES = (
"AGENT",
"DRIVER",
"VECTOR_STORE_DRIVER",
"STRING",
)
RETURN_NAMES = (
+1 -1
View File
@@ -23,7 +23,7 @@ class gtUIVectorStoreUpsertTextTask(gtUIBaseVectorStoreTask):
RETURN_TYPES = (
"AGENT",
"DRIVER",
"VECTOR_STORE_DRIVER",
)
RETURN_NAMES = (
"AGENT",
+1 -1
View File
@@ -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