feat: mv data folder to root

backend/tests/test_artifact_endpoints.py

@@ -3,11 +3,12 @@ from pathlib import Path
 import pytest
 from fastapi.testclient import TestClient
 
+from app.core.data_root import get_data_root
 from main import app
 
 
 def _backend_data_root() -> Path:
-    return Path(__file__).resolve().parents[1] / "data"
+    return get_data_root()
 
 
 def test_download_artifact_within_whitelist() -> None:

backend/tests/test_artifacts.py

@@ -1,10 +1,11 @@
 from pathlib import Path
 
 from app.core.artifacts import extract_artifacts
+from app.core.data_root import get_data_root
 
 
 def _backend_data_root() -> Path:
-    return Path(__file__).resolve().parents[1] / "data"
+    return get_data_root()
 
 
 def test_extract_artifacts_from_local_and_tool_paths() -> None:

backend/tests/test_data_root.py

@@ -0,0 +1,28 @@
+from pathlib import Path
+
+from app.core import data_root
+
+
+def test_data_root_prefers_env(monkeypatch, tmp_path: Path) -> None:
+    custom = tmp_path / "custom-data-root"
+    monkeypatch.setenv("DATA_ROOT", str(custom))
+    assert data_root.get_data_root() == custom.resolve()
+
+
+def test_data_root_falls_back_to_legacy(monkeypatch, tmp_path: Path) -> None:
+    monkeypatch.delenv("DATA_ROOT", raising=False)
+    legacy = tmp_path / "legacy-data"
+    default = tmp_path / "default-data"
+    legacy.mkdir(parents=True, exist_ok=True)
+    monkeypatch.setattr(data_root, "LEGACY_DATA_ROOT", legacy)
+    monkeypatch.setattr(data_root, "DEFAULT_DATA_ROOT", default)
+    assert data_root.get_data_root() == legacy
+
+
+def test_ensure_data_layout_creates_children(monkeypatch, tmp_path: Path) -> None:
+    monkeypatch.setenv("DATA_ROOT", str(tmp_path / "dr"))
+    data_root.ensure_data_layout()
+    root = data_root.get_data_root()
+    assert (root / "workspace").exists()
+    assert (root / "uploads").exists()
+    assert (root / "data").exists()

backend/tests/test_nl2sql.py

@@ -0,0 +1,12 @@
+import asyncio
+import json
+from app.agent.nl2sql import process_nl2sql, NL2SQLRequest
+
+async def main():
+    req = NL2SQLRequest(query="列出所有表", source="postgres", generate_chart=False)
+    res = await process_nl2sql(req)
+    print("SQL:", res.sql)
+    print("Error:", res.error)
+    print("Result:", res.result)
+
+asyncio.run(main())