databutton 0.36.10__py3-none-any.whl → 0.36.12__py3-none-any.whl

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/rag/__init__.py

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

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/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.dist-info → databutton-0.36.12.dist-info}/METADATA

@@ -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.dist-info → databutton-0.36.12.dist-info}/RECORD

@@ -1,4 +1,4 @@
-databutton/__init__.py,sha256=Dfi2yshPIb5991i4FJ0Ly-FRDqPQKVu5cYAU3dyPT3o,354
+databutton/__init__.py,sha256=VTlMJ0fUvTLKj5hsVjaf1wwlKMv72QVg_bUsQwUQpyI,370
 databutton/cachetools.py,sha256=soq0BDkE_YNaybBx-fI4pJ1x8k9KuotZ0f_OuZCCI9k,3554
 databutton/internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 databutton/internal/auth.py,sha256=sQTtdyyWHcoDN7iUHFMqoj1Okk4NFeiWMoNnaFnHg5c,3205
@@ -13,13 +13,15 @@ databutton/jobs/run.py,sha256=Td0BSiIrYKiymDKVf8w5MT3C5uiwzeulYNJUGcxQk2k,632
 databutton/notify/__init__.py,sha256=p6x5D_8q-EXQHS4TuK7_Dli_PdBRPowrQjKOA1_TaVk,386
 databutton/notify/email.py,sha256=Yq7UD7D_g2epeIPchLIQXXcq61rZqqIzSfRxggK-sXg,12349
 databutton/notify/send.py,sha256=cj-TG0jwolEXwJJVxeBxwpfg_it03KoImUn30qAbbQE,749
+databutton/rag/__init__.py,sha256=O_QGqsZGxDLE-HXoC7E3nwB1MuQwNpL0kTm21Y7an_k,68
+databutton/rag/chroma.py,sha256=bBYEHMgV6EL0hWZyaDzAvfkSkhM_s5qicREZhsXJ-7U,956
 databutton/secrets/__init__.py,sha256=sI0okrfRBs2l5tcLTrKc9uUfPVVRMM9D7WED8p226tI,128
 databutton/secrets/secrets.py,sha256=pehXsD09iKTxHaNQJ6awvIHXgRXBJrf4Xhrc11ZUeOk,3427
 databutton/storage/__init__.py,sha256=7ZNd4eQQ7lYYIe1MqCuFbFF1aAVCuj2c_zmJfaPVlDY,252
 databutton/storage/storage.py,sha256=2qsszaYaagKUU08aDr6psexpHQlQP4i3TswdkijUZEY,20355
 databutton/user.py,sha256=QWyWV_G_bnfIJ9js2itOwBk4zqXgBUF9h8lf5mygef8,503
-databutton/version.py,sha256=uTcTwwqqzot55dsnp7e-vadYt9q5cASpLG9u4UOYVz4,176
-databutton-0.36.10.dist-info/LICENSE,sha256=c7h4pcVZapFEmqWoRTRv_S1P_aibrtDobqmlR-QtMUk,45
-databutton-0.36.10.dist-info/METADATA,sha256=4H-h9E241j4p69nbAeq4SP38etA2_PimKryK6kRkW_I,2753
-databutton-0.36.10.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-databutton-0.36.10.dist-info/RECORD,,
+databutton/version.py,sha256=h3tpMa_35SMqh9w2wb4wegJIcuJ4zv-YyQbRGvCbXb8,176
+databutton-0.36.12.dist-info/LICENSE,sha256=c7h4pcVZapFEmqWoRTRv_S1P_aibrtDobqmlR-QtMUk,45
+databutton-0.36.12.dist-info/METADATA,sha256=4oy8Bp7JhBCycCFCGgSkw_qk09KVKFenvp-JkV-QSqU,2796
+databutton-0.36.12.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+databutton-0.36.12.dist-info/RECORD,,