ursa-ai 0.2.10__tar.gz → 0.2.11__tar.gz

{ursa_ai-0.2.10/src/ursa_ai.egg-info → ursa_ai-0.2.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ursa-ai
-Version: 0.2.10
+Version: 0.2.11
 Summary: Agents for science at LANL
 Author-email: Mike Grosskopf <mikegros@lanl.gov>, Nathan Debardeleben <ndebard@lanl.gov>, Rahul Somasundaram <rsomasundaram@lanl.gov>, Isaac Michaud <imichaud@lanl.gov>, Avanish Mishra <avanish@lanl.gov>, Arthur Lui <alui@lanl.gov>, Russell Bent <rbent@lanl.gov>, Earl Lawrence <earl@lanl.gov>
 License-Expression: BSD-3-Clause

{ursa_ai-0.2.10 → ursa_ai-0.2.11}/src/ursa/util/memory_logger.py

@@ -26,20 +26,30 @@ class AgentMemory:
     * Requires `langchain-chroma`, and `chromadb`.
     """
 
+    @classmethod
+    def get_db_path(cls, path: Optional[str | Path]) -> Path:
+        match path:
+            case None:
+                return Path.home() / ".cache" / "ursa" / "rag" / "db"
+            case str():
+                return Path(str)
+            case Path():
+                return path
+            case _:
+                raise TypeError(
+                    f"Type of path is `{type(path)}` "
+                    "but `Optional[str | Path]` was expected."
+                )
+
     def __init__(
         self,
         embedding_model,
         path: Optional[str | Path] = None,
         collection_name: str = "agent_memory",
     ) -> None:
-        self.path = (
-            Path(path)
-            if path
-            else Path(__file__).resolve().parent / "agent_memory_db"
-        )
+        self.path = self.get_db_path(path)
         self.collection_name = collection_name
         self.path.mkdir(parents=True, exist_ok=True)
-
         self.embeddings = embedding_model
 
         # If a DB already exists, load it; otherwise defer creation until `build_index`.
@@ -165,8 +175,7 @@ def delete_database(path: Optional[str | Path] = None):
         Where the on-disk Chroma DB is for deleting.  If *None*, a folder called
         ``agent_memory_db`` is created in the package’s base directory.
     """
-
-    db_path = Path(path) if path else Path("~/.cache/ursa/rag/db/")
+    db_path = AgentMemory.get_db_path(path)
     if os.path.exists(db_path):
         shutil.rmtree(db_path)
         print(f"Database: {db_path} has been deleted.")

{ursa_ai-0.2.10 → ursa_ai-0.2.11/src/ursa_ai.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ursa-ai
-Version: 0.2.10
+Version: 0.2.11
 Summary: Agents for science at LANL
 Author-email: Mike Grosskopf <mikegros@lanl.gov>, Nathan Debardeleben <ndebard@lanl.gov>, Rahul Somasundaram <rsomasundaram@lanl.gov>, Isaac Michaud <imichaud@lanl.gov>, Avanish Mishra <avanish@lanl.gov>, Arthur Lui <alui@lanl.gov>, Russell Bent <rbent@lanl.gov>, Earl Lawrence <earl@lanl.gov>
 License-Expression: BSD-3-Clause