mirror of
https://github.com/justUmen/Bjornulf_custom_nodes.git
synced 2026-03-21 12:42:11 -03:00
21 lines
643 B
Python
21 lines
643 B
Python
class OllamaConfig:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"ollama_url": ("STRING", {"default": "http://0.0.0.0:11434"}),
|
|
"model_name": ("STRING", {"default": "undefined"}) # Empty list with no default
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("OLLAMA_CONFIG",)
|
|
RETURN_NAMES = ("OLLAMA_CONFIG",)
|
|
FUNCTION = "select_model"
|
|
CATEGORY = "ollama"
|
|
|
|
def select_model(self, ollama_url, model_name):
|
|
return ({"model": model_name, "url": ollama_url},)
|
|
|
|
@classmethod
|
|
def IS_CHANGED(cls, ollama_url, model_name) -> float:
|
|
return 0.0 |