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):