tweaks after local testing

meedan · Jun 20, 2024 · 5f59278 · 5f59278
1 parent bb6080e
commit 5f59278
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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]: