chore: using a database for keys

handler.py

@@ -52,6 +52,7 @@ class EndpointHandler:
         elif method == "inference":
 
             uid = data.pop("uid", data)
+            assert uid in self.key_database.keys()
 
             # Get inputs
             encrypted_inputs = from_json(data.pop("encrypted_inputs", data))

play_with_endpoint.py

@@ -53,6 +53,7 @@ payload = {
     "method": "save_key",
 }
 
+print("Storing the key in the database")
 uid = query(payload)
 
 # Test the handler