[API] Fix compatibility with editor v1 if uncompress_document=True

Aedial · Jun 2, 2024 · dda89ca · dda89ca
1 parent 152f32c
commit dda89ca
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/novelai_api/utils.py b/novelai_api/utils.py
@@ -234,9 +234,10 @@ def decrypt_user_data(
                 item["decrypted"] = True
                 item["compressed"] = is_compressed
 
-                if uncompress_document and "document" in data:
+                document = data.get("document")
+                if uncompress_document and isinstance(document, str):
                     unpacker.restore_state(unpacker_state)
-                    data["document"] = unpacker.unpack(b64decode(data["document"]))
+                    data["document"] = unpacker.unpack(b64decode(document))
 
                 continue