nucliadb 6.4.0.post4167__py3-none-any.whl → 6.4.0.post4171__py3-none-any.whl

nucliadb/ingest/consumer/auditing.py

@@ -22,6 +22,7 @@ import asyncio
 import logging
 import uuid
 from functools import partial
+from typing import Any
 
 from nidx_protos import nodereader_pb2, noderesources_pb2
 
@@ -85,8 +86,8 @@ class IndexAuditHandler:
         await self.pubsub.unsubscribe(self.subscription_id)
         await self.task_handler.finalize()
 
-    async def handle_message(self, raw_data: bytes) -> None:
-        data = self.pubsub.parse(raw_data)
+    async def handle_message(self, msg: Any) -> None:
+        data = self.pubsub.parse(msg)
         notification = writer_pb2.Notification()
         notification.ParseFromString(data)
 
@@ -169,8 +170,8 @@ class ResourceWritesAuditHandler:
     async def finalize(self) -> None:
         await self.pubsub.unsubscribe(self.subscription_id)
 
-    async def handle_message(self, raw_data) -> None:
-        data = self.pubsub.parse(raw_data)
+    async def handle_message(self, msg: Any) -> None:
+        data = self.pubsub.parse(msg)
         notification = writer_pb2.Notification()
         notification.ParseFromString(data)
 

nucliadb/ingest/consumer/materializer.py

@@ -21,6 +21,7 @@
 import logging
 import uuid
 from functools import partial
+from typing import Any
 
 from nucliadb.common import datamanagers
 from nucliadb.common.cluster.utils import get_shard_manager
@@ -77,8 +78,8 @@ class MaterializerHandler:
         await self.pubsub.unsubscribe(self.subscription_id)
         await self.task_handler.finalize()
 
-    async def handle_message(self, raw_data) -> None:
-        data = self.pubsub.parse(raw_data)
+    async def handle_message(self, msg: Any) -> None:
+        data = self.pubsub.parse(msg)
         notification = writer_pb2.Notification()
         notification.ParseFromString(data)
 

nucliadb/ingest/consumer/shard_creator.py

@@ -21,6 +21,7 @@
 import logging
 import uuid
 from functools import partial
+from typing import Any
 
 from nidx_protos import nodereader_pb2, noderesources_pb2
 
@@ -75,8 +76,8 @@ class ShardCreatorHandler:
         await self.pubsub.unsubscribe(self.subscription_id)
         await self.task_handler.finalize()
 
-    async def handle_message(self, raw_data) -> None:
-        data = self.pubsub.parse(raw_data)
+    async def handle_message(self, msg: Any) -> None:
+        data = self.pubsub.parse(msg)
         notification = writer_pb2.Notification()
         notification.ParseFromString(data)
 

nucliadb/reader/reader/notifications.py

@@ -24,6 +24,7 @@ from collections.abc import AsyncGenerator
 from typing import Optional
 
 import async_timeout
+from nats.aio.msg import Msg
 
 from nucliadb.common import datamanagers
 from nucliadb.common.context import ApplicationContext
@@ -91,8 +92,8 @@ async def kb_notifications(kbid: str) -> AsyncGenerator[writer_pb2.Notification,
 
     subscription_key = const.PubSubChannels.RESOURCE_NOTIFY.format(kbid=kbid)
 
-    def subscription_handler(raw_data: bytes):
-        data = pubsub.parse(raw_data)
+    def subscription_handler(msg: Msg):
+        data = pubsub.parse(msg)
         notification = writer_pb2.Notification()
         notification.ParseFromString(data)
         try:

{nucliadb-6.4.0.post4167.dist-info → nucliadb-6.4.0.post4171.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb
-Version: 6.4.0.post4167
+Version: 6.4.0.post4171
 Summary: NucliaDB
 Author-email: Nuclia <nucliadb@nuclia.com>
 License: AGPL
@@ -20,11 +20,11 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3 :: Only
 Requires-Python: <4,>=3.9
 Description-Content-Type: text/markdown
-Requires-Dist: nucliadb-telemetry[all]>=6.4.0.post4167
-Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.4.0.post4167
-Requires-Dist: nucliadb-protos>=6.4.0.post4167
-Requires-Dist: nucliadb-models>=6.4.0.post4167
-Requires-Dist: nidx-protos>=6.4.0.post4167
+Requires-Dist: nucliadb-telemetry[all]>=6.4.0.post4171
+Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.4.0.post4171
+Requires-Dist: nucliadb-protos>=6.4.0.post4171
+Requires-Dist: nucliadb-models>=6.4.0.post4171
+Requires-Dist: nidx-protos>=6.4.0.post4171
 Requires-Dist: nucliadb-admin-assets>=1.0.0.post1224
 Requires-Dist: nuclia-models>=0.24.2
 Requires-Dist: uvicorn[standard]

{nucliadb-6.4.0.post4167.dist-info → nucliadb-6.4.0.post4171.dist-info}/RECORD

@@ -125,13 +125,13 @@ nucliadb/ingest/serialize.py,sha256=-TIjibJTbMqAowzRvyrG3R209vKqBZqXpdrQL9Dq4lo,
 nucliadb/ingest/settings.py,sha256=0B-wQNa8FLqtNcQgRzh-fuIuGptM816XHcbH1NQKfmE,3050
 nucliadb/ingest/utils.py,sha256=l1myURu3r8oA11dx3GpHw-gNTUc1AFX8xdPm9Lgl2rA,2275
 nucliadb/ingest/consumer/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
-nucliadb/ingest/consumer/auditing.py,sha256=_4lwHWQPHRqYn-rAteTEVev25cxFo7566xxBLYu61cA,7198
+nucliadb/ingest/consumer/auditing.py,sha256=xK21DIa_ZAiOJVVbnkmT4jgCRGshNGyPyxsqhE6kROE,7204
 nucliadb/ingest/consumer/consumer.py,sha256=OgS1fr5Yo55u-XbC6zypTH1aJ562Y1vZHnPDlJJpCXQ,13703
-nucliadb/ingest/consumer/materializer.py,sha256=7ofLbwjldJA8TWXDRZRM4U5EviZt3qNSQ8oadmkzS0Y,3840
+nucliadb/ingest/consumer/materializer.py,sha256=tgD_rDI2twQzcz8kKNiW_L4YIth16IGh9mUfD5wiSD4,3858
 nucliadb/ingest/consumer/metrics.py,sha256=ji1l_4cKiHJthQd8YNem1ft4iMbw9KThmVvJmLcv3Xg,1075
 nucliadb/ingest/consumer/pull.py,sha256=EYT0ImngMQgatStG68p2GSrPQBbJxeuq8nFm8DdAbwk,9280
 nucliadb/ingest/consumer/service.py,sha256=BLM_dmKZkFBsYl3sj4MZZp5M3kkxHLuO7sE18PqIatw,6538
-nucliadb/ingest/consumer/shard_creator.py,sha256=SoG5e3MJJcM085UQnmLsY0q3WS-B8qkBclbwrhwe-uc,4341
+nucliadb/ingest/consumer/shard_creator.py,sha256=w0smEu01FU_2cjZnsfBRNqT_Ntho11X17zTMST-vKbc,4359
 nucliadb/ingest/consumer/utils.py,sha256=jpX8D4lKzuPCpArQLZeX_Zczq3pfen_zAf8sPJfOEZU,2642
 nucliadb/ingest/fields/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb/ingest/fields/base.py,sha256=DTXFu_g9vSU2h7S4PlYxd-hp4SDuM_EPf8y51ALgD8w,21654
@@ -195,7 +195,7 @@ nucliadb/reader/api/v1/router.py,sha256=eyNmEGSP9zHkCIG5XlAXl6sukq950B7gFT3X2peM
 nucliadb/reader/api/v1/services.py,sha256=Q2n-QMfAVzSylu6qTUuPa6L3AfZ9jS3azVlfQdSqlA4,13443
 nucliadb/reader/api/v1/vectorsets.py,sha256=insTwaykshz442cMKa2VP74wJwvZrIYi0U7M9EM3aCM,1822
 nucliadb/reader/reader/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
-nucliadb/reader/reader/notifications.py,sha256=HVZNUlfbSuoZ9BsSs8wmzPeYurl0U0O2ooVlR9KSM3U,7792
+nucliadb/reader/reader/notifications.py,sha256=6yUsPvvSRqayJ2peOtE0JY0v4657P_S5SAm32th0Y88,7809
 nucliadb/search/__init__.py,sha256=tnypbqcH4nBHbGpkINudhKgdLKpwXQCvDtPchUlsyY4,1511
 nucliadb/search/app.py,sha256=-WEX1AZRA8R_9aeOo9ovOTwjXW_7VfwWN7N2ccSoqXg,3387
 nucliadb/search/lifecycle.py,sha256=V_Pj5PRP0yyDY8d5LytO4X8p9HhN7UomqRG6Ri0UaFA,2206
@@ -365,8 +365,8 @@ nucliadb/writer/tus/local.py,sha256=7jYa_w9b-N90jWgN2sQKkNcomqn6JMVBOVeDOVYJHto,
 nucliadb/writer/tus/s3.py,sha256=vF0NkFTXiXhXq3bCVXXVV-ED38ECVoUeeYViP8uMqcU,8357
 nucliadb/writer/tus/storage.py,sha256=ToqwjoYnjI4oIcwzkhha_MPxi-k4Jk3Lt55zRwaC1SM,2903
 nucliadb/writer/tus/utils.py,sha256=MSdVbRsRSZVdkaum69_0wku7X3p5wlZf4nr6E0GMKbw,2556
-nucliadb-6.4.0.post4167.dist-info/METADATA,sha256=THJ-YeTQ5PJ46_51WA0fMucJixuM9B8mgGzi_q8j9-E,4226
-nucliadb-6.4.0.post4167.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
-nucliadb-6.4.0.post4167.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
-nucliadb-6.4.0.post4167.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
-nucliadb-6.4.0.post4167.dist-info/RECORD,,
+nucliadb-6.4.0.post4171.dist-info/METADATA,sha256=GIwlyI4vyk76Z2HTHRqHV3nLbhqmPUGI_eJv4Efheas,4226
+nucliadb-6.4.0.post4171.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
+nucliadb-6.4.0.post4171.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
+nucliadb-6.4.0.post4171.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
+nucliadb-6.4.0.post4171.dist-info/RECORD,,