From 52d310ea3166eb7c4fbe5fbbab7a8a74298fedfa Mon Sep 17 00:00:00 2001 From: Jason Schleifer Date: Wed, 21 Aug 2024 04:40:52 +1200 Subject: [PATCH] removed unused query for models --- nodes/config/gtUIOllamaStructureConfig.py | 3 --- nodes/drivers/gtUIOllamaEmbeddingDriver.py | 4 ---- nodes/drivers/gtUIOllamaPromptDriver.py | 4 ---- 3 files changed, 11 deletions(-) diff --git a/nodes/config/gtUIOllamaStructureConfig.py b/nodes/config/gtUIOllamaStructureConfig.py index e17652b..9dd2be9 100644 --- a/nodes/config/gtUIOllamaStructureConfig.py +++ b/nodes/config/gtUIOllamaStructureConfig.py @@ -7,13 +7,10 @@ from griptape.drivers import ( OllamaPromptDriver, ) -from ..utilities import get_models from .gtUIBaseConfig import gtUIBaseConfig ollama_port = "11434" ollama_base_url = "http://127.0.0.1" -ollama_models = get_models("ollama", ollama_base_url, ollama_port) -ollama_models.append("") class gtUIOllamaStructureConfig(gtUIBaseConfig): diff --git a/nodes/drivers/gtUIOllamaEmbeddingDriver.py b/nodes/drivers/gtUIOllamaEmbeddingDriver.py index 6146030..9640a07 100644 --- a/nodes/drivers/gtUIOllamaEmbeddingDriver.py +++ b/nodes/drivers/gtUIOllamaEmbeddingDriver.py @@ -1,13 +1,9 @@ from griptape.drivers import OllamaEmbeddingDriver -from ..utilities import get_models from .gtUIBaseEmbeddingDriver import gtUIBaseEmbeddingDriver default_port = "11434" default_base_url = "http://127.0.0.1" -models = get_models("ollama", default_base_url, default_port) -if len(models) == 0: - models.append("") class gtUIOllamaEmbeddingDriver(gtUIBaseEmbeddingDriver): diff --git a/nodes/drivers/gtUIOllamaPromptDriver.py b/nodes/drivers/gtUIOllamaPromptDriver.py index 1ca493d..352b1d8 100644 --- a/nodes/drivers/gtUIOllamaPromptDriver.py +++ b/nodes/drivers/gtUIOllamaPromptDriver.py @@ -1,13 +1,9 @@ from griptape.drivers import OllamaPromptDriver -from ..utilities import get_models from .gtUIBasePromptDriver import gtUIBasePromptDriver default_port = "11434" default_base_url = "http://127.0.0.1" -models = get_models("ollama", default_base_url, default_port) -if len(models) == 0: - models.append("") class gtUIOllamaPromptDriver(gtUIBasePromptDriver):