Add milvus backend unit test

AGENTS.md

@@ -47,13 +47,16 @@ cp .env.example .env  # Configure any environment variables, defined in `./kaize
 pre-commit install
 ```
 
+## Development Tips
+- This project is managed by `uv`, not `python` or `pip`, so any python commands need to go through `uv`. All dependencies are defined in `pyproject.toml`.
+
 ## Testing Instructions
 - Run pytest verbosely with the `-v` flag by default so that you have more context when tests fail.
 - Use `uv run pytest tests/.../<test_name.py>` to run tests individually.
 - We use the pytest markers `e2e` for end-to-end tests, and `unit` for unit tests, and `phoenix` to test integration with Phoenix.
 - When running `uv run pytest` it will skip the tests marked with `phoenix`.
 - To run specific markers: `uv run pytest -m e2e` or `uv run pytest -m unit`
-- To override and run all: `uv run pytest -m "e2e or unit or phoenix"`
+- To run all tests: `uv run pytest -m "e2e or unit or phoenix"`
 
 ## Available Interfaces
 - MCP Server: `get_guidelines()`, `save_trajectory()`

kaizen/backend/base.py

@@ -13,14 +13,15 @@ def __init__(self, config: BaseSettings | None = None):
         pass
 
     @abstractmethod
-    def ready(self):
+    def ready(self) -> bool:
         pass
 
     @abstractmethod
-    def create_namespace(
-        self,
-        namespace_id: str | None = None
-    ) -> Namespace:
+    def details(self) -> dict:
+        pass
+
+    @abstractmethod
+    def create_namespace(self, namespace_id: str | None = None) -> Namespace:
         pass
 
     @abstractmethod
@@ -39,15 +40,12 @@ def update_entities(
         enable_conflict_resolution: bool = True,
     ) -> list[EntityUpdate]:
         pass
+
     def search_entities(
-        self,
-        namespace_id: str,
-        query: str | None = None,
-        filters: dict | None = None,
-        limit: int = 10
+        self, namespace_id: str, query: str | None = None, filters: dict | None = None, limit: int = 10
     ) -> list[RecordedEntity]:
         pass
 
     @abstractmethod
     def delete_entity_by_id(self, namespace_id: str, entity_id: str):
-        pass
+        pass

kaizen/backend/filesystem.py

@@ -50,9 +50,13 @@ def _save_namespace_data(self, namespace_id: str, data: dict):
         with open(file_path, "w") as f:
             json.dump(data, f, indent=2, default=str)
 
-    def ready(self):
+    def ready(self) -> bool:
         """Check if the backend is healthy."""
-        return {"status": "ok", "data_dir": str(self.data_dir)}
+        return True
+
+    def details(self) -> dict:
+        """Return details about the backend."""
+        return {"data_dir": str(self.data_dir)}
 
     def create_namespace(self, namespace_id: str | None = None) -> Namespace:
         """Create a new namespace for entities to exist in."""
@@ -61,9 +65,7 @@ def create_namespace(self, namespace_id: str | None = None) -> Namespace:
 
         with self._lock:
             if file_path.exists():
-                raise NamespaceAlreadyExistsException(
-                    f'Namespace "{namespace_id}" already exists.'
-                )
+                raise NamespaceAlreadyExistsException(f'Namespace "{namespace_id}" already exists.')
 
             now = datetime.datetime.now(datetime.UTC)
             data = {
@@ -97,9 +99,7 @@ def search_namespaces(self, limit: int = 10) -> list[Namespace]:
                     namespaces.append(
                         Namespace(
                             id=data["id"],
-                            created_at=datetime.datetime.fromisoformat(
-                                data["created_at"]
-                            ),
+                            created_at=datetime.datetime.fromisoformat(data["created_at"]),
                             num_entities=len(data["entities"]),
                         )
                     )
@@ -151,9 +151,7 @@ def update_entities(
                 # Find similar existing entities for conflict resolution
                 old_entities = []
                 for entity in entities:
-                    similar = self._search_entities_internal(
-                        data, query=entity.content, filters=None, limit=10
-                    )
+                    similar = self._search_entities_internal(data, query=entity.content, filters=None, limit=10)
                     old_entities.extend(similar)
 
                 updates = resolve_conflicts(old_entities, entities_with_temporary_ids)
@@ -181,9 +179,7 @@ def update_entities(
                                     ent["metadata"] = update.metadata
                                     break
                         case "DELETE":
-                            data["entities"] = [
-                                e for e in data["entities"] if e["id"] != update.id
-                            ]
+                            data["entities"] = [e for e in data["entities"] if e["id"] != update.id]
                         case "NONE":
                             pass
             else:
@@ -286,9 +282,7 @@ def delete_entity_by_id(self, namespace_id: str, entity_id: str):
         with self._lock:
             data = self._load_namespace_data(namespace_id)
             original_count = len(data["entities"])
-            data["entities"] = [
-                e for e in data["entities"] if str(e["id"]) != entity_id
-            ]
+            data["entities"] = [e for e in data["entities"] if str(e["id"]) != entity_id]
             if len(data["entities"]) == original_count:
                 raise KaizenException(f"Entity `{entity_id}` not found")
             self._save_namespace_data(namespace_id, data)