remove chroma dependency; add local storage that doesn't have retrieval

platogram/library/__init__.py

@@ -34,3 +34,9 @@ def get_keyword_local_bm25(home_dir: Path = Path("./my_library")) -> Library:
     from .keyword_local_bm25 import LocalBM25Library
 
     return LocalBM25Library(home_dir)
+
+
+def get_local_dumb(home_dir: Path = Path("./my_library")) -> Library:
+    from .local_dumb import LocalDumbLibrary
+
+    return LocalDumbLibrary(home_dir)

platogram/library/local_dumb.py

@@ -0,0 +1,45 @@
+import json
+from pathlib import Path
+
+from platogram.types import Content
+from platogram.utils import make_filesystem_safe
+
+
+class LocalDumbLibrary:
+    def __init__(self, home_dir: Path):
+        if not home_dir.exists():
+            home_dir.mkdir(parents=True)
+        self.home_dir = home_dir
+
+    @property
+    def home(self) -> Path:
+        return self.home_dir
+
+    def ls(self) -> list[str]:
+        return [f.stem for f in self.home.glob("*.json")]
+
+    def exists(self, id: str) -> bool:
+        return (self.home / f"{make_filesystem_safe(id)}.json").exists()
+
+    def put(self, id: str, content: Content) -> None:
+        file = self.home / f"{make_filesystem_safe(id)}.json"
+        with open(file, "w") as f:
+            json.dump(content.model_dump(mode="json"), f)
+
+    def get_content(self, id: str) -> Content:
+        file = self.home_dir / f"{make_filesystem_safe(id)}.json"
+        with open(file, "r") as f:
+            content = Content(**json.load(f))
+        return content
+
+    def delete(self, id: str) -> None:
+        file = self.home_dir / f"{make_filesystem_safe(id)}.json"
+        file.unlink()
+
+    def retrieve(
+        self,
+        query: str,
+        n_results: int,
+        filter_keys: list[str],
+    ) -> tuple[list[Content], list[float]]:
+        raise NotImplementedError("Dumb local storage does not support retrieval. Use get_content().")

platogram/library/semantic_local_chroma.py

@@ -2,8 +2,15 @@
 import os
 from pathlib import Path
 
-import chromadb
-from chromadb.utils import embedding_functions
+try:
+    import chromadb
+except ImportError:
+    pass
+
+try:
+    from chromadb.utils import embedding_functions
+except ImportError:
+    pass
 
 from platogram.ops import remove_markers
 from platogram.types import Content