diff --git a/mem0/vector_stores/valkey.py b/mem0/vector_stores/valkey.py index c4539dcd21..82a2e1adf4 100644 --- a/mem0/vector_stores/valkey.py +++ b/mem0/vector_stores/valkey.py @@ -527,7 +527,9 @@ def _format_timestamp(self, timestamp, timezone=None): str: The formatted timestamp. """ # Use UTC as default timezone if not specified - tz = pytz.timezone(timezone or "UTC") + if timezone is None or timezone == "UTC": + return datetime.utcfromtimestamp(timestamp).replace(tzinfo=pytz.UTC).isoformat(timespec="microseconds") + tz = pytz.timezone(timezone) return datetime.fromtimestamp(timestamp, tz=tz).isoformat(timespec="microseconds") def _process_document_fields(self, result, vector_id):