feat: add workspace foundation and search tooling

2026-05-04 01:37:34 +08:00 · 2026-03-14 11:29:27 -07:00
parent 9834e62835
commit b8832022f1
15 changed files with 912 additions and 5 deletions
--- a/tests/tools/test_workspace_tool.py
+++ b/tests/tools/test_workspace_tool.py
@@ -0,0 +1,89 @@
+from __future__ import annotations
+
+import json
+from pathlib import Path
+
+
+def _config(tmp_path: Path) -> dict:
+    return {
+        "workspace": {
+            "enabled": True,
+            "path": str(tmp_path / "workspace"),
+            "auto_create": True,
+            "persist_gateway_uploads": "ask",
+        },
+        "knowledgebase": {
+            "enabled": True,
+            "path": str(tmp_path / "knowledgebase"),
+            "roots": [],
+            "retrieval_mode": "off",
+            "auto_index": True,
+            "watch_for_changes": False,
+            "max_injected_chunks": 6,
+            "max_injected_tokens": 3200,
+            "dense_top_k": 40,
+            "sparse_top_k": 40,
+            "fused_top_k": 30,
+            "final_top_k": 8,
+            "min_fused_score": 0.0,
+            "injection_format": "sourced_note",
+            "chunking": {
+                "default_tokens": 512,
+                "overlap_tokens": 80,
+                "code_strategy": "structural",
+                "markdown_strategy": "headings",
+            },
+            "embeddings": {"provider": "local", "model": "embeddinggemma-300m", "dimensions": 768},
+            "reranker": {"enabled": False, "provider": "local", "model": "bge-reranker-v2-m3"},
+            "indexing": {
+                "respect_gitignore": True,
+                "respect_hermesignore": True,
+                "include_hidden": False,
+                "max_file_mb": 10,
+            },
+        },
+    }
+
+
+class TestWorkspaceTool:
+    def test_status_reports_workspace_roots(self, tmp_path, monkeypatch):
+        from tools.workspace_tool import workspace_tool
+
+        monkeypatch.setattr("tools.workspace_tool.load_config", lambda: _config(tmp_path))
+
+        result = json.loads(workspace_tool(action="status"))
+
+        assert result["success"] is True
+        assert result["workspace_root"].endswith("workspace")
+        assert result["knowledgebase_root"].endswith("knowledgebase")
+
+    def test_index_and_search_round_trip(self, tmp_path, monkeypatch):
+        from tools.workspace_tool import workspace_tool
+
+        cfg = _config(tmp_path)
+        workspace = Path(cfg["workspace"]["path"])
+        (workspace / "docs").mkdir(parents=True)
+        (workspace / "docs" / "deploy.md").write_text("deployment checklist\n", encoding="utf-8")
+        monkeypatch.setattr("tools.workspace_tool.load_config", lambda: cfg)
+
+        indexed = json.loads(workspace_tool(action="index"))
+        assert indexed["success"] is True
+        assert indexed["file_count"] == 1
+
+        searched = json.loads(workspace_tool(action="search", query="deployment"))
+        assert searched["success"] is True
+        assert searched["count"] == 1
+        assert searched["matches"][0]["relative_path"] == "docs/deploy.md"
+
+    def test_list_returns_relative_paths(self, tmp_path, monkeypatch):
+        from tools.workspace_tool import workspace_tool
+
+        cfg = _config(tmp_path)
+        workspace = Path(cfg["workspace"]["path"])
+        (workspace / "notes").mkdir(parents=True)
+        (workspace / "notes" / "todo.txt").write_text("ship it\n", encoding="utf-8")
+        monkeypatch.setattr("tools.workspace_tool.load_config", lambda: cfg)
+
+        listed = json.loads(workspace_tool(action="list"))
+        assert listed["success"] is True
+        assert listed["entries"][0]["relative_path"] == "notes/todo.txt"