feat: implementing basic tinydb in memory storage

memory/controller.py

@@ -0,0 +1,29 @@
+# minimal memory module for saving and loading converstations
+
+# package imports
+from tinydb import TinyDB, Query
+from tinydb.storages import MemoryStorage
+
+#temporary json db setup
+db = TinyDB('db.json',storage=MemoryStorage)
+query = Query()
+
+def conversation_mapper(data):
+    mapping = {
+        {
+            'user': 'Hello, how are you?',
+            'bot': 'I am good, thank you.'
+        },
+    }
+    return mapping
+
+# function to dump json data into the db
+def dump (data):
+    mapping = conversation_mapper(data)
+    doc_id = db.insert(data)
+    return doc_id, mapping
+
+# function to query data from the db
+def get (doc_id:int):
+    data = db.get(doc_id=doc_id)
+    return data

requirements.txt

@@ -6,4 +6,5 @@ accelerate~=0.24.1
 markdown~=3.5.1
 huggingface_hub~=0.19.4
 fastapi~=0.104.1
-uvicorn~=0.24.0
+uvicorn~=0.24.0
+tinydb~=4.8.0