nodes/griptape-ai-ComfyUI-Griptape
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

removed unused query for models

Browse Source
This commit is contained in:
Jason Schleifer
2024-08-21 04:40:52 +12:00
parent d5adf65a1b
commit 52d310ea31
3 changed files with 0 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/config/gtUIOllamaStructureConfig.py
-3
View File
@@ -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):
nodes/drivers/gtUIOllamaEmbeddingDriver.py
-4
View File
@@ -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):
nodes/drivers/gtUIOllamaPromptDriver.py
-4
View File
@@ -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):