Enhancement Suite: Anti-loop protection, Kokoro TTS integration, ComfyUI status monitoring, and auto-open image functionality

2026-02-23 00:59:20 +00:00
parent d7efe008b8
commit 90fcba28bc
41 changed files with 1982 additions and 40 deletions
--- a/build/lib/ollama_cli/core/config.py
+++ b/build/lib/ollama_cli/core/config.py
@@ -0,0 +1,57 @@
+import os
+import json
+
+CONFIG_FILE = os.path.expanduser("~/.ollama-cli-config.json")
+
+DEFAULT_CONFIG = {
+    "excluded_models": [],
+    "git": {
+        "remote_url": "",
+        "username": "",
+        "password": ""
+    },
+    "gitea": {
+        "url": "",
+        "token": ""
+    },
+    "auto_save_sessions": True,
+    "mcp_servers": {},
+    "ollama_url": "http://localhost:11434",
+    "comfy_url": "http://localhost:9000",
+    "piper": {
+        "path": "~/.ollama-cli/piper/piper",
+        "model": "~/.ollama-cli/piper/voice.onnx"
+    },
+    "ntfy": {
+        "enabled": False,
+        "topic": "",
+        "server": "https://ntfy.sh"
+    },
+    "default_model": "llama3.2"
+}
+
+def load_config():
+    """Load config file"""
+    if os.path.exists(CONFIG_FILE):
+        try:
+            with open(CONFIG_FILE, 'r') as f:
+                config = json.load(f)
+                # Merge with defaults to ensure all keys exist
+                for k, v in DEFAULT_CONFIG.items():
+                    if k not in config:
+                        config[k] = v
+                return config
+        except Exception:
+            pass
+    return DEFAULT_CONFIG.copy()
+
+def save_config(config):
+    """Save config file"""
+    with open(CONFIG_FILE, 'w') as f:
+        json.dump(config, f, indent=2)
+
+def update_config(key, value):
+    """Update a specific config key"""
+    config = load_config()
+    config[key] = value
+    save_config(config)