From 5f59278539313b629ef4bf0494ceffeab4ad04b9 Mon Sep 17 00:00:00 2001 From: Devin Gaffney Date: Thu, 20 Jun 2024 06:02:15 -0700 Subject: [PATCH] tweaks after local testing --- lib/cache.py | 5 +++-- lib/model/model.py | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) 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]: