nucliadb 6.3.1.post3570__py3-none-any.whl → 6.3.1.post3571__py3-none-any.whl

nucliadb/export_import/tasks.py

@@ -17,7 +17,6 @@
 # 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 nucliadb.common.context import ApplicationContext
 from nucliadb.export_import.exporter import export_kb_to_blob_storage
 from nucliadb.export_import.importer import import_kb_from_blob_storage
 from nucliadb.export_import.models import NatsTaskMessage
@@ -60,14 +59,13 @@ def get_exports_consumer() -> NatsTaskConsumer[NatsTaskMessage]:
     )
 
 
-async def get_exports_producer(context: ApplicationContext) -> NatsTaskProducer[NatsTaskMessage]:
+def get_exports_producer() -> NatsTaskProducer[NatsTaskMessage]:
     producer = create_producer(
         name="exports_producer",
         stream=ExportsNatsConfig.stream,
         producer_subject=ExportsNatsConfig.consumer.subject,
         msg_type=NatsTaskMessage,
     )
-    await producer.initialize(context)
     return producer
 
 
@@ -82,12 +80,11 @@ def get_imports_consumer() -> NatsTaskConsumer[NatsTaskMessage]:
     )
 
 
-async def get_imports_producer(context: ApplicationContext) -> NatsTaskProducer[NatsTaskMessage]:
+def get_imports_producer() -> NatsTaskProducer[NatsTaskMessage]:
     producer = create_producer(
         name="imports_producer",
         stream=ImportsNatsConfig.stream,
         producer_subject=ImportsNatsConfig.consumer.subject,
         msg_type=NatsTaskMessage,
     )
-    await producer.initialize(context)
     return producer

nucliadb/tasks/producer.py

@@ -17,13 +17,13 @@
 # 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 Generic, Optional, Type
+from typing import Generic, Type
 
-from nucliadb.common.context import ApplicationContext
 from nucliadb.tasks.logger import logger
 from nucliadb.tasks.models import MsgType
-from nucliadb.tasks.utils import NatsStream, create_nats_stream_if_not_exists
+from nucliadb.tasks.utils import NatsStream
 from nucliadb_telemetry import errors
+from nucliadb_utils.utilities import get_nats_manager
 
 
 class NatsTaskProducer(Generic[MsgType]):
@@ -38,29 +38,17 @@ class NatsTaskProducer(Generic[MsgType]):
         self.stream = stream
         self.producer_subject = producer_subject
         self.msg_type = msg_type
-        self.context: Optional[ApplicationContext] = None
-        self.initialized = False
-
-    async def initialize(self, context: ApplicationContext):
-        self.context = context
-        await create_nats_stream_if_not_exists(
-            self.context,
-            self.stream.name,
-            subjects=self.stream.subjects,
-        )
-        self.initialized = True
+        self.nats_manager = get_nats_manager()
 
     async def send(self, msg: MsgType) -> int:
         """
         Publish message to the producer's nats stream.
         Returns the sequence number of the published message.
         """
-        if not self.initialized:
-            raise RuntimeError("NatsTaskProducer not initialized")
         try:
-            pub_ack = await self.context.nats_manager.js.publish(  # type: ignore
+            pub_ack = await self.nats_manager.js.publish(
                 self.producer_subject,
-                msg.model_dump_json().encode("utf-8"),  # type: ignore
+                msg.model_dump_json().encode("utf-8"),
             )
             logger.info(
                 "Message sent to Nats",

nucliadb/writer/api/v1/export_import.py

@@ -195,7 +195,7 @@ async def start_export_task(context: ApplicationContext, kbid: str, export_id: s
     metadata.task.status = Status.SCHEDULED
     await dm.set_metadata("export", metadata)
     try:
-        producer = await get_exports_producer(context)
+        producer = get_exports_producer()
         msg = NatsTaskMessage(kbid=kbid, id=export_id)
         seqid = await producer.send(msg)
         logger.info(f"Export task produced. seqid={seqid} kbid={kbid} export_id={export_id}")
@@ -212,7 +212,7 @@ async def start_import_task(context: ApplicationContext, kbid: str, import_id: s
     metadata.total = import_size or 0
     await dm.set_metadata("import", metadata)
     try:
-        producer = await get_imports_producer(context)
+        producer = get_imports_producer()
         msg = NatsTaskMessage(kbid=kbid, id=import_id)
         seqid = await producer.send(msg)
         logger.info(f"Import task produced. seqid={seqid} kbid={kbid} import_id={import_id}")

{nucliadb-6.3.1.post3570.dist-info → nucliadb-6.3.1.post3571.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: nucliadb
-Version: 6.3.1.post3570
+Version: 6.3.1.post3571
 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.3.1.post3570
-Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.3.1.post3570
-Requires-Dist: nucliadb-protos>=6.3.1.post3570
-Requires-Dist: nucliadb-models>=6.3.1.post3570
-Requires-Dist: nidx-protos>=6.3.1.post3570
+Requires-Dist: nucliadb-telemetry[all]>=6.3.1.post3571
+Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.3.1.post3571
+Requires-Dist: nucliadb-protos>=6.3.1.post3571
+Requires-Dist: nucliadb-models>=6.3.1.post3571
+Requires-Dist: nidx-protos>=6.3.1.post3571
 Requires-Dist: nucliadb-admin-assets>=1.0.0.post1224
 Requires-Dist: nuclia-models>=0.24.2
 Requires-Dist: uvicorn

{nucliadb-6.3.1.post3570.dist-info → nucliadb-6.3.1.post3571.dist-info}/RECORD

@@ -108,7 +108,7 @@ nucliadb/export_import/exceptions.py,sha256=Dw8WqfG4r6MPJc5TPfbjMvCgXXWTcTOecGHR
 nucliadb/export_import/exporter.py,sha256=k2QVx1EjqFlDYiggriWiEJzwtMXzHbldsqWdpGQM3_U,7074
 nucliadb/export_import/importer.py,sha256=v5cq9Nn8c2zrY_K_00mydR52f8mdFxR7tLdtNLQ0qvk,4229
 nucliadb/export_import/models.py,sha256=dbjScNkiMRv4X3Ktudy1JRliD25bfoDTy3JmEZgQSCc,2121
-nucliadb/export_import/tasks.py,sha256=4JX3bygyLCLSuGxMCStYyoclh_CL8rPxrVVWuGqvcmM,3146
+nucliadb/export_import/tasks.py,sha256=DWbdqY97ffoyfipelGXz3Jqz1iam6JCjQSh367Fc3NA,2947
 nucliadb/export_import/utils.py,sha256=iAQAjYuNx0dhM2b5-1A0NEs8tSRsznuT-izysUrTwS0,19986
 nucliadb/ingest/__init__.py,sha256=fsw3C38VP50km3R-nHL775LNGPpJ4JxqXJ2Ib1f5SqE,1011
 nucliadb/ingest/app.py,sha256=rX1KE5vsAzG9hlArBk8WE2SOlvdYylcb-jNkMQNPJdQ,7407
@@ -271,7 +271,7 @@ nucliadb/tasks/__init__.py,sha256=oFJ3A8HD7w11mBu-IixYE_KxA7juMGlYQb7YD_y6WPM,97
 nucliadb/tasks/consumer.py,sha256=xc0Ql3N1Iq52dJ3t4YYGJFj1NCQAly0J5W_brfLa_F8,6894
 nucliadb/tasks/logger.py,sha256=C7keOEO_mjLVp5VbqAZ2QXfqVB2Hot7NgBlUP_SDSMw,924
 nucliadb/tasks/models.py,sha256=qrZKi5DNDQ07waMsp5L4_Fi7WRs57YiO-kmXlrBzEAA,1168
-nucliadb/tasks/producer.py,sha256=JRGnATkALyr_iLHq0OAjzVbfxZ_SOUa6sx-smU5p6SQ,3136
+nucliadb/tasks/producer.py,sha256=UnpJAzhj_GElsCoO5G6T4m6MshsgOaqR2tVzJmEta64,2625
 nucliadb/tasks/retries.py,sha256=Zv-3Hys-SKayG9VQ7_7EIflkegE5j-xPGrf-nwaxsfY,5075
 nucliadb/tasks/utils.py,sha256=tV1AbWdFc3qfIULX44Veqj41FCD1B6XYjG6brULBeiw,1459
 nucliadb/tests/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
@@ -322,7 +322,7 @@ nucliadb/writer/api/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20
 nucliadb/writer/api/constants.py,sha256=qWEDjFUycrEZnSJyLnNK4PQNodU2oVmkO4NycaEZtio,1738
 nucliadb/writer/api/utils.py,sha256=wIQHlU8RQiIGVLI72suvyVIKlCU44Unh0Ae0IiN6Qwo,1313
 nucliadb/writer/api/v1/__init__.py,sha256=akI9A_jloNLb0dU4T5zjfdyvmSAiDeIdjAlzNx74FlU,1128
-nucliadb/writer/api/v1/export_import.py,sha256=FxGY_rofj3AH-HQReoMDOYfEpACeWB5bsmxrT03QUtQ,8227
+nucliadb/writer/api/v1/export_import.py,sha256=elf-EQY5DD3mhw8kWb9tQpDcbrF9sY6VFYqxQOjuVP0,8201
 nucliadb/writer/api/v1/field.py,sha256=OsWOYA0WQ6onE5Rkl20QIEdtrSi7Jgnu62fUt90Ziy8,17503
 nucliadb/writer/api/v1/knowledgebox.py,sha256=MLeIuym4jPrJgfy1NTcN9CpUGwuBiqDHMcx0hY9DR7g,9530
 nucliadb/writer/api/v1/learning_config.py,sha256=CKBjqcbewkfPwGUPLDWzZSpro6XkmCaVppe5Qtpu5Go,3117
@@ -347,8 +347,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.3.1.post3570.dist-info/METADATA,sha256=a5FFqexlq9oEflkrfSYCAW8j3sV62OJnY1LCgPtpbJg,4291
-nucliadb-6.3.1.post3570.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-nucliadb-6.3.1.post3570.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
-nucliadb-6.3.1.post3570.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
-nucliadb-6.3.1.post3570.dist-info/RECORD,,
+nucliadb-6.3.1.post3571.dist-info/METADATA,sha256=s2P-Covs_cwHf5pNszgsGypGofi5r1zgIOV7ccxAh6M,4291
+nucliadb-6.3.1.post3571.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+nucliadb-6.3.1.post3571.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
+nucliadb-6.3.1.post3571.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
+nucliadb-6.3.1.post3571.dist-info/RECORD,,