From 2ea62f2c9349b4d7b91c1be2175af6a3d34dcc2e Mon Sep 17 00:00:00 2001 From: yunxiang wei <121092793+yiwangsimple@users.noreply.github.com> Date: Thu, 1 Aug 2024 19:44:36 +0800 Subject: [PATCH] Add files via upload --- nodes/ollama_prompt_extractor.py | 14 -------------- 1 file changed, 14 deletions(-) diff --git a/nodes/ollama_prompt_extractor.py b/nodes/ollama_prompt_extractor.py index 34668c4..93943c4 100644 --- a/nodes/ollama_prompt_extractor.py +++ b/nodes/ollama_prompt_extractor.py @@ -22,22 +22,8 @@ class OllamaPromptExtractor: @classmethod def initialize(cls): - cls.load_config() cls.available_models = get_available_models(cls.base_url) - @classmethod - def load_config(cls): - config_path = os.path.join(os.path.dirname(__file__), 'config.json') - try: - with open(config_path, 'r') as config_file: - config = json.load(config_file) - cls.base_url = config.get('OLLAMA_API_URL', cls.base_url) - print(f"Loaded Ollama API URL: {cls.base_url}") - except FileNotFoundError: - print(f"Warning: config.json not found at {config_path}. Using default URL: {cls.base_url}") - except json.JSONDecodeError: - print(f"Error: Invalid JSON in config.json at {config_path}. Using default URL: {cls.base_url}") - @classmethod def INPUT_TYPES(cls): return {