nucliadb 6.4.0.post4161__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/ingest/service/__init__.py

@@ -17,12 +17,10 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program. If not, see <http://www.gnu.org/licenses/>.
 #
-from typing import Optional
-
 from grpc import aio
 
 from nucliadb import health
-from nucliadb.ingest import logger
+from nucliadb.ingest import SERVICE_NAME, logger
 from nucliadb.ingest.service.writer import WriterServicer
 from nucliadb.ingest.settings import settings
 from nucliadb_protos import writer_pb2_grpc
@@ -30,12 +28,12 @@ from nucliadb_telemetry.utils import setup_telemetry
 from nucliadb_utils.grpc import get_traced_grpc_server
 
 
-async def start_grpc(service_name: Optional[str] = None):
+async def start_grpc(service_name: str = SERVICE_NAME):
     aio.init_grpc_aio()  # type: ignore
 
-    await setup_telemetry(service_name or "ingest")
+    await setup_telemetry(service_name)
     server = get_traced_grpc_server(
-        service_name or "ingest",
+        service_name,
         max_receive_message=settings.max_receive_message_length,
     )
 

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.post4161.dist-info → nucliadb-6.4.0.post4171.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb
-Version: 6.4.0.post4161
+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.post4161
-Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.4.0.post4161
-Requires-Dist: nucliadb-protos>=6.4.0.post4161
-Requires-Dist: nucliadb-models>=6.4.0.post4161
-Requires-Dist: nidx-protos>=6.4.0.post4161
+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.post4161.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
@@ -158,7 +158,7 @@ nucliadb/ingest/orm/processor/data_augmentation.py,sha256=v-pj4GbBWSuO8dQyahs5UD
 nucliadb/ingest/orm/processor/pgcatalog.py,sha256=ht9_I5WlPc6sSFTY8PsxHlpjN-EsaBaChwqsLlMXwUk,3100
 nucliadb/ingest/orm/processor/processor.py,sha256=jLGLyfj6TIXoLubmNvQvkj5K32MxDDSoktY6M8z5pVk,33149
 nucliadb/ingest/orm/processor/sequence_manager.py,sha256=uqEphtI1Ir_yk9jRl2gPf7BlzzXWovbARY5MNZSBI_8,1704
-nucliadb/ingest/service/__init__.py,sha256=MME_G_ERxzJR6JW_hfE2qcfXpmpH1kdG-S0a-M0qRm8,2043
+nucliadb/ingest/service/__init__.py,sha256=LHQFUkdmNBOWqBG0Md9sMMI7g5TQZ-hLAnhw6ZblrJg,2002
 nucliadb/ingest/service/exceptions.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb/ingest/service/writer.py,sha256=EN-qEDJnL5in7cs1lC-z6b4ui_byFJMJ5huyCtwR_sU,21637
 nucliadb/middleware/__init__.py,sha256=A8NBlBuEkunCFMKpR9gnfNELsVn0Plc55BIQMbWDM8Q,2202
@@ -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.post4161.dist-info/METADATA,sha256=xZYY6m0tHa_k-3X9XuzMbfSy9oofa64mPUaTaCpq9GM,4226
-nucliadb-6.4.0.post4161.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
-nucliadb-6.4.0.post4161.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
-nucliadb-6.4.0.post4161.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
-nucliadb-6.4.0.post4161.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,,

{nucliadb-6.4.0.post4161.dist-info → nucliadb-6.4.0.post4171.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.0.0)
+Generator: setuptools (80.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any