From 2f99124941541857f14f4b91cee543eee2e6f6a7 Mon Sep 17 00:00:00 2001
From: JianFeeeee <109188060+JianFeeeee@users.noreply.github.com>
Date: Fri, 10 Apr 2026 19:51:20 +0800
Subject: [PATCH] feat: Add config persistence

- Create ConfigManager for save/load
- Save config to config.json
- Load config on startup
- Persist API key, model, base_url
- Fix config not saved issue
---
 graph_memory_tui/app.py                     | 16 ++++++-
 graph_memory_tui/services/config_manager.py | 46 +++++++++++++++++++++
 2 files changed, 60 insertions(+), 2 deletions(-)
 create mode 100644 graph_memory_tui/services/config_manager.py

diff --git a/graph_memory_tui/app.py b/graph_memory_tui/app.py
index 6484f69..357c1c8 100644
--- a/graph_memory_tui/app.py
+++ b/graph_memory_tui/app.py
@@ -17,6 +17,7 @@ from .widgets.message_history import MessageHistory
 from .models.message import Message, ToolCall, ToolResult
 from .models.config import AppConfig
 from .models.log_entry import LogEntry
+from .services.config_manager import ConfigManager
 from .core.imports import (
     Neo4jGraph,
     GraphMemoryClient,
@@ -48,7 +49,15 @@ class GraphMemoryApp(App[None]):
 
     def __init__(self, config: AppConfig | None = None, **kwargs):
         super().__init__(**kwargs)
-        self._config = config or AppConfig.from_env()
+        # 配置管理器
+        self._config_manager = ConfigManager()
+        # 优先使用传入的配置，其次加载持久化配置，最后使用环境变量
+        if config:
+            self._config = config
+        elif self._config_manager.exists():
+            self._config = self._config_manager.load()
+        else:
+            self._config = AppConfig.from_env()
         
         # 核心组件
         self._graph: Neo4jGraph | None = None
@@ -350,10 +359,13 @@ API Key 未配置！
         # 更新配置
         self._config = event.config
         
+        # 持久化保存配置
+        self._config_manager.save(self._config)
+        
         # 重新初始化客户端
         self._init_client()
         
         if self._client:
-            self.notify("✅ 配置已更新并应用", title="配置")
+            self.notify("✅ 配置已保存并应用", title="配置")
         else:
             self.notify("⚠️ 配置已保存，但API Key无效", title="警告")
diff --git a/graph_memory_tui/services/config_manager.py b/graph_memory_tui/services/config_manager.py
new file mode 100644
index 0000000..79e2be6
--- /dev/null
+++ b/graph_memory_tui/services/config_manager.py
@@ -0,0 +1,46 @@
+"""
+配置管理 - 支持持久化
+"""
+
+import json
+from pathlib import Path
+from .config import AppConfig
+
+
+class ConfigManager:
+    """配置管理器 - 支持持久化"""
+    
+    def __init__(self, config_file: str = "config.json"):
+        self.config_file = Path(config_file)
+    
+    def save(self, config: AppConfig) -> None:
+        """保存配置到文件"""
+        data = {
+            "api_key": config.api_key,
+            "model": config.model,
+            "base_url": config.base_url
+        }
+        
+        with open(self.config_file, 'w', encoding='utf-8') as f:
+            json.dump(data, f, indent=2)
+    
+    def load(self) -> AppConfig:
+        """从文件加载配置"""
+        if not self.config_file.exists():
+            return AppConfig()
+        
+        try:
+            with open(self.config_file, 'r', encoding='utf-8') as f:
+                data = json.load(f)
+            
+            return AppConfig(
+                api_key=data.get("api_key", ""),
+                model=data.get("model", "deepseek-chat"),
+                base_url=data.get("base_url", "https://api.deepseek.com")
+            )
+        except Exception:
+            return AppConfig()
+    
+    def exists(self) -> bool:
+        """检查配置文件是否存在"""
+        return self.config_file.exists()