chore: using a database for keys

handler.py

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

play_with_endpoint.py

@@ -32,6 +32,8 @@ headers = {
 
 def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
+    print(f"{response=}")
+    print(f"{response.json()=}")
     return response.json()
 
 
@@ -53,8 +55,8 @@ payload = {
     "method": "save_key",
 }
 
-print("Storing the key in the database")
 uid = query(payload)
+print(f"Storing the key in the database under {uid=}")
 
 # Test the handler
 nb_good = 0