UncountablePythonSDK 0.0.82__tar.gz → 0.0.84__tar.gz

{uncountablepythonsdk-0.0.82/UncountablePythonSDK.egg-info → uncountablepythonsdk-0.0.84}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.82
+Version: 0.0.84
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{uncountablepythonsdk-0.0.82 → uncountablepythonsdk-0.0.84/UncountablePythonSDK.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.82
+Version: 0.0.84
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{uncountablepythonsdk-0.0.82 → uncountablepythonsdk-0.0.84}/uncountable/core/async_batch.py

@@ -4,7 +4,7 @@ from uncountable.types.async_batch import AsyncBatchRequest
 from uncountable.types.async_batch_processor import AsyncBatchProcessorBase
 
 
-class AsyncBatchSubmissionError(BaseException):
+class AsyncBatchSubmissionError(Exception):
     pass
 
 

{uncountablepythonsdk-0.0.82 → uncountablepythonsdk-0.0.84}/uncountable/integration/queue_runner/worker.py

@@ -2,6 +2,8 @@ import asyncio
 from concurrent.futures import ProcessPoolExecutor
 from dataclasses import dataclass
 
+from opentelemetry.trace import get_current_span
+
 from uncountable.core.async_batch import AsyncBatchProcessor
 from uncountable.integration.construct_client import construct_uncountable_client
 from uncountable.integration.executors.executors import execute_job
@@ -9,7 +11,7 @@ from uncountable.integration.job import JobArguments
 from uncountable.integration.queue_runner.datastore.interface import Datastore
 from uncountable.integration.queue_runner.types import ListenQueue, ResultQueue
 from uncountable.integration.scan_profiles import load_profiles
-from uncountable.integration.telemetry import JobLogger, get_otel_tracer
+from uncountable.integration.telemetry import JobLogger, Logger, get_otel_tracer
 from uncountable.types import base_t, job_definition_t, queued_job_t
 
 
@@ -28,19 +30,24 @@ class Worker:
         self.datastore = datastore
 
     async def run_worker_loop(self) -> None:
+        logger = Logger(get_current_span())
         while True:
-            queued_job = await self.listen_queue.get()
-            self.datastore.increment_num_attempts(queued_job.queued_job_uuid)
-            loop = asyncio.get_event_loop()
-            result = await loop.run_in_executor(
-                self.process_pool, run_queued_job, queued_job
-            )
-            assert isinstance(result, job_definition_t.JobResult)
-            await self.result_queue.put(
-                queued_job_t.QueuedJobResult(
-                    job_result=result, queued_job_uuid=queued_job.queued_job_uuid
+            try:
+                queued_job = await self.listen_queue.get()
+                self.datastore.increment_num_attempts(queued_job.queued_job_uuid)
+                loop = asyncio.get_event_loop()
+                result = await loop.run_in_executor(
+                    self.process_pool, run_queued_job, queued_job
                 )
-            )
+                assert isinstance(result, job_definition_t.JobResult)
+                await self.result_queue.put(
+                    queued_job_t.QueuedJobResult(
+                        job_result=result, queued_job_uuid=queued_job.queued_job_uuid
+                    )
+                )
+            except BaseException as e:
+                logger.log_exception(e)
+                raise e
 
 
 @dataclass(kw_only=True)