nucliadb-utils 6.9.1.post5229__py3-none-any.whl → 6.9.6.post5473__py3-none-any.whl

nucliadb_utils/const.py

@@ -41,4 +41,4 @@ class Features:
     SKIP_EXTERNAL_INDEX = "nucliadb_skip_external_index"
     LOG_REQUEST_PAYLOADS = "nucliadb_log_request_payloads"
     IGNORE_EXTRACTED_IN_SEARCH = "nucliadb_ignore_extracted_in_search"
-    REBALANCE_ENABLED = "nucliadb_rebalance"
+    ASK_DECOUPLED = "nucliadb_ask_decoupled"

nucliadb_utils/featureflagging.py

@@ -45,7 +45,10 @@ DEFAULT_FLAG_DATA: dict[str, Any] = {
         "rollout": 0,
         "variants": {"environment": ["local"]},
     },
-    const.Features.REBALANCE_ENABLED: {"rollout": 0, "variants": {"environment": ["local"]}},
+    const.Features.ASK_DECOUPLED: {
+        "rollout": 0,
+        "variants": {"environment": ["local"]},
+    },
 }
 
 

nucliadb_utils/storages/azure.py

@@ -20,6 +20,7 @@
 
 from __future__ import annotations
 
+import base64
 import logging
 from datetime import datetime
 from typing import AsyncGenerator, AsyncIterator, Optional, Union
@@ -426,13 +427,20 @@ class AzureObjectStore(ObjectStore):
     @ops_observer.wrap({"type": "multipart_start"})
     async def upload_multipart_start(self, bucket: str, key: str, metadata: ObjectMetadata) -> None:
         container_client = self.service_client.get_container_client(bucket)
-        custom_metadata = {key: str(value) for key, value in metadata.model_dump().items()}
+        custom_metadata = {
+            "base64_filename": base64.b64encode(metadata.filename.encode()).decode(),
+            "content_type": metadata.content_type,
+            "size": str(metadata.size),
+        }
         blob_client = container_client.get_blob_client(key)
+        safe_filename = (
+            metadata.filename.encode("ascii", "replace").decode().replace('"', "").replace("\n", "")
+        )
         await blob_client.create_append_blob(
             metadata=custom_metadata,
             content_settings=ContentSettings(
                 content_type=metadata.content_type,
-                content_disposition=f"attachment; filename={metadata.filename}",
+                content_disposition=f'attachment; filename="{safe_filename}"',
             ),
         )
 
@@ -460,7 +468,12 @@ def parse_object_metadata(properties: BlobProperties, key: str) -> ObjectMetadat
         size = int(custom_metadata_size)
     else:
         size = properties.size
-    filename = custom_metadata.get("filename") or key.split("/")[-1]
+
+    b64_filename = custom_metadata.get("base64_filename")
+    if b64_filename:
+        filename = base64.b64decode(b64_filename.encode()).decode()
+    else:
+        filename = key.split("/")[-1]
     content_type = custom_metadata.get("content_type") or properties.content_settings.content_type or ""
     return ObjectMetadata(
         filename=filename,

nucliadb_utils/storages/s3.py

@@ -179,7 +179,7 @@ class S3StorageField(StorageField):
             Bucket=bucket_name,
             Key=upload_id,
             Metadata={
-                "FILENAME": cf.filename,
+                "base64_filename": base64.b64encode(cf.filename.encode()).decode(),
                 "SIZE": str(cf.size),
                 "CONTENT_TYPE": cf.content_type,
             },

{nucliadb_utils-6.9.1.post5229.dist-info → nucliadb_utils-6.9.6.post5473.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb_utils
-Version: 6.9.1.post5229
+Version: 6.9.6.post5473
 Summary: NucliaDB util library
 Author-email: Nuclia <nucliadb@nuclia.com>
 License-Expression: AGPL-3.0-or-later
@@ -8,13 +8,12 @@ Project-URL: Homepage, https://nuclia.com
 Project-URL: Repository, https://github.com/nuclia/nucliadb
 Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Python: <4,>=3.9
+Requires-Python: <4,>=3.10
 Description-Content-Type: text/markdown
 Requires-Dist: pydantic>=2.6
 Requires-Dist: pydantic-settings>=2.2
@@ -27,8 +26,9 @@ Requires-Dist: nats-py[nkeys]>=2.6.0
 Requires-Dist: PyNaCl
 Requires-Dist: pyjwt>=2.4.0
 Requires-Dist: mrflagly>=0.2.9
-Requires-Dist: nucliadb-protos>=6.9.1.post5229
-Requires-Dist: nucliadb-telemetry>=6.9.1.post5229
+Requires-Dist: nidx-protos>=6.9.6.post5473
+Requires-Dist: nucliadb-protos>=6.9.6.post5473
+Requires-Dist: nucliadb-telemetry>=6.9.6.post5473
 Provides-Extra: cache
 Requires-Dist: redis>=4.3.4; extra == "cache"
 Requires-Dist: orjson>=3.6.7; extra == "cache"

{nucliadb_utils-6.9.1.post5229.dist-info → nucliadb_utils-6.9.6.post5473.dist-info}/RECORD

@@ -1,10 +1,10 @@
 nucliadb_utils/__init__.py,sha256=EvBCH1iTODe-AgXm48aj4kVUt_Std3PeL8QnwimR5wI,895
 nucliadb_utils/asyncio_utils.py,sha256=h8Y-xpcFFRgNzaiIW0eidz7griAQa7ggbNk34-tAt2c,2888
 nucliadb_utils/authentication.py,sha256=5_b323v2ylJaJvM_0coeSQEtnD-p9IGD-6CPA6IXhik,6471
-nucliadb_utils/const.py,sha256=YtWadXGm044MbwLcfEImNC6skh3e4LKZDu6hjlO0qMU,1521
+nucliadb_utils/const.py,sha256=F1kZzTSbJvg9YYGhSi-H6KJsfTkTiX8PER4RBAOkq40,1521
 nucliadb_utils/debug.py,sha256=Q56Nx9Dp7V2ae3CU2H0ztaZcHTJXdlflPLKLeOPZ170,2436
 nucliadb_utils/exceptions.py,sha256=y_3wk77WLVUtdo-5FtbBsdSkCtK_DsJkdWb5BoPn3qo,1094
-nucliadb_utils/featureflagging.py,sha256=ctd9Nqm_nhoedMIV2GC819-cSP5GlkLYXCRE0DbwxYU,2353
+nucliadb_utils/featureflagging.py,sha256=oKlSC3B-Bfx7xdrpalpmqUk3nCkMIm7BjyBMWlHC_RU,2372
 nucliadb_utils/grpc.py,sha256=apu0uePnkGHCAT7GRQ9YZfRYyFj26kJ440i8jitbM3U,3314
 nucliadb_utils/helpers.py,sha256=eed7_E1MKh9eW3CpqOXka3OvLw5C9eJGC_R-1MPYdfY,3336
 nucliadb_utils/nats.py,sha256=U21Cfg36_IHd3ZLXEC4eZ7nZ1Soh_ZNFFwjryNyd2-8,15248
@@ -35,13 +35,13 @@ nucliadb_utils/nuclia_usage/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn
 nucliadb_utils/nuclia_usage/utils/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb_utils/nuclia_usage/utils/kb_usage_report.py,sha256=6lLuxCCPQVn3dOuZNL5ThPjl2yws-1TJ_7duhQSWkPU,3934
 nucliadb_utils/storages/__init__.py,sha256=5Qc8AUWiJv9_JbGCBpAn88AIJhwDlm0OPQpg2ZdRL4U,872
-nucliadb_utils/storages/azure.py,sha256=pu0IyKPCn32oT0wI3oJIG6iUxnPtwNgg1zu00C8wDjo,18057
+nucliadb_utils/storages/azure.py,sha256=t0ZL_698NqsBz-Ihwkc79tusfzGcm0BVTB5pqceNlvA,18456
 nucliadb_utils/storages/exceptions.py,sha256=6YhFLf8k0ABy5AVfxIJUo7w6AK0SJjktiyQTwF3gCdg,2344
 nucliadb_utils/storages/gcs.py,sha256=VyT72My34N4pEMmrQc5wdAMNLiuqpYl8OW3d50cJfSA,28222
 nucliadb_utils/storages/local.py,sha256=2aCHpZymORG_dUc1FDq0VFcgQulu0w2pZiUaj9dphFs,11686
 nucliadb_utils/storages/nuclia.py,sha256=vEv94xAT7QM2g80S25QyrOw2pzvP2BAX-ADgZLtuCVc,2097
 nucliadb_utils/storages/object_store.py,sha256=2PueRP5Q3XOuWgKhj6B9Kp2fyBql5np0T400YRUbqn4,4535
-nucliadb_utils/storages/s3.py,sha256=eFFVRgNTIxTz1Hpmd6ofRz9KQhPJAmiyetW4EmWN8EM,21835
+nucliadb_utils/storages/s3.py,sha256=EUqlNoJW32AI6jpETbDla3teYbxlz8RFTfxSdHgWZdo,21878
 nucliadb_utils/storages/settings.py,sha256=mepN3wbLGL0Pv5yI6D-sNjSAFinEWT7aRi6N3eClNDg,1384
 nucliadb_utils/storages/storage.py,sha256=aOJnx6-WX8U3AAqPL_sWPCghIzlr8e3GKGi8z3-mtqw,22024
 nucliadb_utils/storages/utils.py,sha256=F4Iboa_0_bhDQr-JOKD9sGPld_-hKwJW5ptyZdn9Oag,1505
@@ -53,7 +53,7 @@ nucliadb_utils/tests/gcs.py,sha256=JNqp5ymeNNU9Ci8rNYTh7-VqP4fjybElhyB3ap7EV1c,4
 nucliadb_utils/tests/local.py,sha256=z9E11_ol1mu7N8Y6PkjKl-WMPPMl7JqQbDj3uhVa1A0,1933
 nucliadb_utils/tests/nats.py,sha256=rbTaC6kv-u6SdZ7N-XBEGS40XCRiUmFUsKHIYWJfxTs,2325
 nucliadb_utils/tests/s3.py,sha256=kz9ULxrAYLVslZ59I8dtweZ9DJz5R8Ioy2XYrveZzHw,3829
-nucliadb_utils-6.9.1.post5229.dist-info/METADATA,sha256=QkcqFXWPuu1Cz5lt5ecN5osQjyvtVIXjUxE-Rde2S3c,2180
-nucliadb_utils-6.9.1.post5229.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nucliadb_utils-6.9.1.post5229.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-6.9.1.post5229.dist-info/RECORD,,
+nucliadb_utils-6.9.6.post5473.dist-info/METADATA,sha256=V7vJvzCfdw4mJL-cZGkQsfGEdo45BVnKX1rI1OBg7fw,2174
+nucliadb_utils-6.9.6.post5473.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nucliadb_utils-6.9.6.post5473.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-6.9.6.post5473.dist-info/RECORD,,