chore: using a database for keys

handler.py

@@ -1,3 +1,4 @@
+import time
 from typing import Dict, List, Any
 import numpy as np
 from concrete.ml.deployment import FHEModelServer
@@ -25,6 +26,8 @@ class EndpointHandler:
 
         self.key_database[43] = "benoit"
 
+        print(f"{time.time()}: enter in the init")
+
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """
          data args:
@@ -34,10 +37,12 @@ class EndpointHandler:
               A :obj:`list` | `dict`: will be serialized and returned
         """
 
+        print(f"{time.time()}: enter in the call")
+
         # Get method
         method = data.pop("method", data)
 
-        if method is "save_key":
+        if method == "save_key":
 
             # Get keys
             evaluation_keys = from_json(data.pop("evaluation_keys", data))

play_with_endpoint.py

@@ -33,6 +33,10 @@ headers = {
 def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     print(f"{response=}")
+
+    if "error" in response:
+        assert False, f"Got an error: {response=}"
+
     print(f"{response.json()=}")
     return response.json()