databutton 0.36.10__tar.gz → 0.36.12__tar.gz

{databutton-0.36.10 → databutton-0.36.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: databutton
-Version: 0.36.10
+Version: 0.36.12
 Summary: The CLI for databutton.com
 License: Databutton
 Author: Databutton
@@ -13,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.11
 Requires-Dist: PyJWT (>=2.2.0,<3.0.0)
 Requires-Dist: PyYAML (>=6.0,<7.0)
 Requires-Dist: certifi (>=2022.12.7,<2023.0.0)
+Requires-Dist: chromadb (>=0.4.22,<0.5.0)
 Requires-Dist: cryptography (>=40.0.1,<41.0.0)
 Requires-Dist: httpx (>=0.20.0,<1)
 Requires-Dist: pandas (>=0.25,<3)
@@ -29,6 +30,7 @@ Description-Content-Type: text/markdown
 
 
 
+
 The CLI for building and deploying databutton projects
 
 ## Getting Started

{databutton-0.36.10 → databutton-0.36.12}/README.md

@@ -6,6 +6,7 @@
 
 
 
+
 The CLI for building and deploying databutton projects
 
 ## Getting Started

{databutton-0.36.10 → databutton-0.36.12}/databutton/__init__.py

@@ -5,7 +5,7 @@ Create data apps
 .. moduleauthor:: Databutton <support@databutton.com>
 """
 
-from . import notify, secrets, storage, user
+from . import notify, secrets, storage, user, rag
 from .cachetools import cache, clear_cache
 from .version import __version__
 
@@ -16,5 +16,6 @@ __all__ = [
     "storage",
     "cache",
     "clear_cache",
+    "rag",
     "__version__",
 ]

databutton-0.36.12/databutton/rag/__init__.py

@@ -0,0 +1,5 @@
+from .chroma import Chroma
+
+chroma = Chroma()
+
+__all__ = ["chroma"]

databutton-0.36.12/databutton/rag/chroma.py

@@ -0,0 +1,33 @@
+import chromadb
+import os
+import shutil
+import databutton as db
+
+
+class Chroma:
+    path = ".tmp/chroma/"
+    file_path = "path/chroma.sqlite3"
+
+    def __init__(self):
+        pass
+
+    def persist(self):
+        shutil.make_archive("chroma", "zip", self.path)
+        with open("chroma.zip", "rb") as file:
+            db.storage.binary.put("chromadb", file.read())
+
+    def client(self):
+        # if local file doesn't exists, try to fetch it
+        if not os.path.exists(self.file_path):
+            try:
+                stream = db.storage.binary.get("chromadb")
+                with open(".tmp/chroma/chroma.zip", "wb") as file:
+                    file.write(stream)
+                shutil.unpack_archive("chroma.zip", self.path)
+            except FileNotFoundError:
+                pass
+        # Instigate the client
+        the_client = chromadb.PersistentClient(self.path)
+        the_client.persist = self.persist
+
+        return the_client

{databutton-0.36.10 → databutton-0.36.12}/databutton/version.py

@@ -5,4 +5,4 @@ This module contains project version information.
 .. moduleauthor:: Databutton <support@databutton.com>
 """
 
-__version__ = "0.36.10"
+__version__ = "0.36.12"

{databutton-0.36.10 → databutton-0.36.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "databutton"
-version = "0.36.10"
+version = "0.36.12"
 description = "The CLI for databutton.com"
 authors = ["Databutton <hi@databutton.io>"]
 license = "Databutton"
@@ -18,6 +18,7 @@ tenacity = "^8.0.0"
 cryptography = "^40.0.1"
 certifi = "^2022.12.7"
 streamlit = "^1.20.0"
+chromadb = "^0.4.22"
 
 [tool.poetry.group.dev.dependencies]
 mypy = "^1.2.0"
@@ -30,7 +31,7 @@ isort = "^5.12.0"
 python-semantic-release = "7.28.1"
 pytest-mock = "^3.7.0"
 pandas-stubs = "^1.5.2.221124"
-ruff = "^0.1.0"
+ruff = "^0.2.0"
 streamlit = "^1.21.0"
 
 [build-system]