Merge pull request #15 from SepineTam/codex/更新配置文件用法和索引方式

Update config handling

Author: SepineTam

README.md

@@ -43,7 +43,9 @@ The standard config json as follows, you can DIY your config via add envs.
 }
 ```
 
-For more detailed usage information, visit the [Usage guide](source/docs/Usages/Usage.md). 
+For more detailed usage information, visit the [Usage guide](source/docs/Usages/Usage.md).
+
+Configuration values are stored in `~/.stata-mcp/config.toml`. See the [Configuration guide](source/docs/Usages/Configuration.md) for details.
 
 And some advanced usage, visit the [Advanced guide](source/docs/Usages/Advanced.md)
 
@@ -84,6 +86,7 @@ uvx /path/to/your/whl/stata_mcp-1.6.0-py3-non-any.whl  # here is the wheel file
 
 ## 📝 Documentation
 - For more detailed usage information, visit the [Usage guide](source/docs/Usages/Usage.md).
+- See how to edit the config file in the [Configuration guide](source/docs/Usages/Configuration.md).
 - Advanced Usage, visit the [Advanced](source/docs/Usages/Advanced.md)
 - Some questions, visit the [Questions](source/docs/Usages/Questions.md)
 - Difference with [Stata-MCP@hanlulong](https://github.com/hanlulong/stata-mcp), visit the [Difference](source/docs/Difference.md)

source/docs/README/cn/README.md

@@ -46,6 +46,8 @@
 
 有关更详细的使用信息，请访问[使用指南](../../Usages/Usage.md)。
 
+配置文件位于 `~/.stata-mcp/config.toml`，详见[配置指南](../../Usages/Configuration.md)。
+
 一些高级的功能，访问[高级指南](../../Usages/Advanced.md)
 
 ### 前提条件
@@ -85,6 +87,7 @@ uvx /path/to/your/whl/stata_mcp-1.6.0-py3-non-any.whl  # 这里的文件名可
 
 ## 📝 文档
 - 有关更详细的使用信息，请访问[使用指南](../../Usages/Usage.md)。
+- 了解如何编辑配置文件，请阅读[配置指南](../../Usages/Configuration.md)。
 - 高级用法，请访问[高级指南](../../Usages/Advanced.md)
 - 一些问题，请访问[问题](../../Usages/Questions.md)
 - 与[Stata-MCP@hanlulong](https://github.com/hanlulong/stata-mcp)的区别，请访问[区别](../../Difference.md)

source/docs/README/fr/README.md

@@ -44,6 +44,7 @@ Le fichier json de configuration standard est le suivant, vous pouvez personnali
 ```
 
 Pour des informations d'utilisation plus détaillées, consultez le [guide d'utilisation](../../Usages/Usage.md).
+Les valeurs de configuration sont stockées dans `~/.stata-mcp/config.toml`. Consultez le [guide de configuration](../../Usages/Configuration.md) pour plus de détails.
 
 Et pour une utilisation avancée, visitez le [Guide avancé](../../Usages/Advanced.md)
 
@@ -84,6 +85,7 @@ uvx /path/to/your/whl/stata_mcp-1.6.0-py3-non-any.whl  # modifiez le nom du fich
 
 ## 📝 Documentation
 - Pour des informations d'utilisation plus détaillées, consultez le [guide d'utilisation](../../Usages/Usage.md).
+- Consultez le [guide de configuration](../../Usages/Configuration.md) pour modifier le fichier de configuration.
 - Utilisation avancée, visitez le [Guide avancé](../../Usages/Advanced.md)
 - Quelques questions, visitez les [Questions](../../Usages/Questions.md)
 - Différence avec [Stata-MCP@hanlulong](https://github.com/hanlulong/stata-mcp), visitez la [Différence](../../Difference.md)

source/docs/README/sp/README.md

@@ -44,6 +44,7 @@ El archivo json de configuración estándar es el siguiente; puedes personalizar
 ```
 
 Para información más detallada sobre el uso, visita la [guía de Uso](../../Usages/Usage.md).
+Los valores de configuración se almacenan en `~/.stata-mcp/config.toml`. Consulta la [guía de Configuración](../../Usages/Configuration.md) para más detalles.
 
 Y para un uso más avanzado, visita la [Guía avanzada](../../Usages/Advanced.md)
 
@@ -84,6 +85,7 @@ uvx /path/to/your/whl/stata_mcp-1.6.0-py3-non-any.whl  # cambia el nombre del ar
 
 ## 📝 Documentación
 - Para información más detallada sobre el uso, visita la [guía de Uso](../../Usages/Usage.md).
+- Consulta la [guía de Configuración](../../Usages/Configuration.md) para editar el archivo de configuración.
 - Uso avanzado, visita la [Guía avanzada](../../Usages/Advanced.md)
 - Algunas preguntas, visita las [Preguntas](../../Usages/Questions.md)
 - Diferencia con [Stata-MCP@hanlulong](https://github.com/hanlulong/stata-mcp), visita las [Diferencias](../../Difference.md)

source/docs/Usages/Configuration.md

@@ -0,0 +1,17 @@
+# Configuration File
+
+Stata-MCP stores settings in `~/.stata-mcp/config.toml`. A sample file is located at `src/stata_mcp/config/example.toml`.
+Copy it to your home folder if the file does not exist and modify as needed.
+
+## Accessing Values
+
+You can read and write values in the configuration using dot notation:
+
+```python
+from stata_mcp.config import Config
+cfg = Config()
+model = cfg.get("llm.ollama.MODEL")
+cfg.set("stata.stata_cli", "/path/to/stata")
+```
+
+Sections mirror the TOML hierarchy, so `llm.ollama.MODEL` refers to the `MODEL` key inside `[llm.ollama]`.

src/stata_mcp/__init__.py

@@ -21,7 +21,6 @@
 dotenv.load_dotenv()
 mcp = FastMCP(name="stata-mcp")
 config_mgr = Config()
-cfg = config_mgr.config
 
 # Initialize optional parameters
 sys_os = platform.system()
@@ -38,7 +37,7 @@
     sys.exit("Unknown System")
 
 # Use configured output path if available
-output_base_path = cfg.get("output_base_path") or os.path.join(
+output_base_path = config_mgr.get("stata-mcp.output_base_path") or os.path.join(
     documents_path, "stata-mcp-folder"
 )
 os.makedirs(output_base_path, exist_ok=True)
@@ -47,7 +46,9 @@
     # stata_cli
     finder = StataFinder()
     default_cli = finder.find_stata()
-    stata_cli = cfg.get("stata_cli") or os.getenv("stata_cli", default_cli)
+    stata_cli = config_mgr.get("stata.stata_cli") or os.getenv(
+        "stata_cli", default_cli
+    )
     if stata_cli is None:
         exit_msg = (
             "Missing Stata.exe, "

src/stata_mcp/config/__init__.py

@@ -4,7 +4,7 @@
 
 
 class Config:
-    """Simple configuration manager for Stata-MCP."""
+    """Configuration manager supporting nested TOML sections."""
 
     CONFIG_FILE_PATH = os.path.expanduser("~/.stata-mcp/config.toml")
 
@@ -25,29 +25,75 @@ def _default_config(self) -> dict:
         else:
             documents_path = os.path.expanduser("~/Documents")
         return {
-            "stata_cli": "",
-            "output_base_path": os.path.join(documents_path, "stata-mcp-folder"),
+            "stata": {"stata_cli": ""},
+            "stata-mcp": {
+                "output_base_path": os.path.join(
+                    documents_path, "stata-mcp-folder"
+                )
+            },
+            "llm": {
+                "LLM_TYPE": "ollama",
+                "ollama": {
+                    "MODEL": "qwen2.5-coder:7b",
+                    "BASE_URL": "http://localhost:11434",
+                },
+                "openai": {
+                    "MODEL": "gpt-3.5-turbo",
+                    "BASE_URL": "https://api.openai.com/v1",
+                    "API_KEY": "<YOUR_OPENAI_API_KEY>",
+                },
+            },
         }
 
+    def _write_dict(self, f, data: dict, prefix: str = "") -> None:
+        for key, value in data.items():
+            if isinstance(value, dict):
+                section = f"{prefix}.{key}" if prefix else key
+                f.write(f"\n[{section}]\n")
+                self._write_dict(f, value, section)
+            else:
+                escaped = str(value).replace('"', '\\"')
+                f.write(f"{key} = \"{escaped}\"\n")
+
     def _save(self) -> None:
         """Write the current config to the TOML file."""
         with open(self.CONFIG_FILE_PATH, "w", encoding="utf-8") as f:
-            for key, value in self.config.items():
-                escaped = str(value).replace('"', '\\"')
-                f.write(f"{key} = \"{escaped}\"\n")
+            self._write_dict(f, self.config)
 
     def load_config(self) -> dict:
         with open(self.CONFIG_FILE_PATH, "rb") as f:
             return tomllib.load(f)
 
+    def _get_nested(self, data: dict, keys: list[str], default=None):
+        for k in keys:
+            if isinstance(data, dict) and k in data:
+                data = data[k]
+            else:
+                return default
+        return data
+
     def get(self, key: str, default: str | None = None):
-        return self.config.get(key, default)
+        keys = key.split(".")
+        return self._get_nested(self.config, keys, default)
+
+    def _set_nested(self, data: dict, keys: list[str], value):
+        for k in keys[:-1]:
+            data = data.setdefault(k, {})
+        data[keys[-1]] = value
 
     def set(self, key: str, value: str) -> None:
-        self.config[key] = value
+        keys = key.split(".")
+        self._set_nested(self.config, keys, value)
         self._save()
 
+    def _delete_nested(self, data: dict, keys: list[str]):
+        for k in keys[:-1]:
+            data = data.get(k)
+            if not isinstance(data, dict):
+                return
+        data.pop(keys[-1], None)
+
     def delete(self, key: str) -> None:
-        if key in self.config:
-            del self.config[key]
-            self._save()
+        keys = key.split(".")
+        self._delete_nested(self.config, keys)
+        self._save()

src/stata_mcp/webui/__init__.py

@@ -16,14 +16,14 @@ def config():
     if request.method == 'POST':
         stata_cli = request.form['stata_cli']
         output_base_path = request.form['output_base_path']
-        config_mgr.set('stata_cli', stata_cli)
-        config_mgr.set('output_base_path', output_base_path)
+        config_mgr.set('stata.stata_cli', stata_cli)
+        config_mgr.set('stata-mcp.output_base_path', output_base_path)
         return redirect(url_for('config', saved='1'))
 
     return render_template(
         'config.html',
-        stata_cli=config_mgr.get('stata_cli', ''),
-        output_base_path=config_mgr.get('output_base_path', ''),
+        stata_cli=config_mgr.get('stata.stata_cli', ''),
+        output_base_path=config_mgr.get('stata-mcp.output_base_path', ''),
         saved=saved
     )