diff --git a/lib/cache.py b/lib/cache.py index bb1f857..41a3518 100644 --- a/lib/cache.py +++ b/lib/cache.py @@ -48,5 +48,6 @@ def set_cached_result(content_hash: str, result: Any, ttl: int = DEFAULT_TTL) -> result (Any): The result to cache. ttl (int): Time-to-live for the cache in seconds. Default is 86400 seconds (24 hours). """ - client = Cache.get_client() - client.setex(content_hash, ttl, json.dumps(result)) + if content_hash: + client = Cache.get_client() + client.setex(content_hash, ttl, json.dumps(result)) diff --git a/lib/model/model.py b/lib/model/model.py index d7c6ee4..8472193 100644 --- a/lib/model/model.py +++ b/lib/model/model.py @@ -47,7 +47,7 @@ def get_response(self, message: schemas.Message) -> schemas.GenericItem: result = Cache.get_cached_result(message.body.content_hash) if not result: result = self.process(message) - Cache.set_cached_result(message.body.content_hash, message.body.result) + Cache.set_cached_result(message.body.content_hash, result) return result def respond(self, messages: Union[List[schemas.Message], schemas.Message]) -> List[schemas.Message]: