UncountablePythonSDK 0.0.70__py3-none-any.whl → 0.0.71__py3-none-any.whl

{UncountablePythonSDK-0.0.70.dist-info → UncountablePythonSDK-0.0.71.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.70
+Version: 0.0.71
 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.70.dist-info → UncountablePythonSDK-0.0.71.dist-info}/RECORD

@@ -102,7 +102,7 @@ uncountable/integration/executors/executors.py,sha256=CbwatKkHrLhnqYr_nsBjr0KYeO
 uncountable/integration/executors/generic_upload_executor.py,sha256=NlW5WcYePPA7_fwp5uW_2afAiQLKK7rCkKF06wQ948E,10375
 uncountable/integration/executors/script_executor.py,sha256=OmSBOtU48G3mqza9c2lCm84pGGyaDk-ZBJCx3RsdJXc,846
 uncountable/integration/queue_runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-uncountable/integration/queue_runner/job_scheduler.py,sha256=4AdI_89mPNAgO-WSJvqRaN3HbKlzlVRh9aIkwfQvL7o,4679
+uncountable/integration/queue_runner/job_scheduler.py,sha256=n6bM6ZqVOPD0PoJuZV5Y5tuhmw2gI-_p6JbnVlK42uI,5016
 uncountable/integration/queue_runner/queue_runner.py,sha256=0BmYu5zHdothTevGsB-nXg6MBd1UD-WkP3h1WCKMdQg,710
 uncountable/integration/queue_runner/types.py,sha256=8qTq29BTSa5rmW6CBlBntP0pNIiDcwu1wHa78pjroS0,219
 uncountable/integration/queue_runner/worker.py,sha256=GNz8PmPUBMLA-R2VWkC_86WOaeHjHFI2mvBlNRcI3wE,4238
@@ -269,7 +269,7 @@ uncountable/types/api/recipes/set_recipe_inputs.py,sha256=yCGFQoguzsSK_weOK4kcvq
 uncountable/types/api/recipes/set_recipe_metadata.py,sha256=J2F1FDgFLKjlbGw29TW1Xefy38Zt-C7V1WDK70gRiuY,914
 uncountable/types/api/recipes/set_recipe_output_annotations.py,sha256=ayG9vnCroYGQ3mmNBALLlQhf9otrR-q0QiNLnWgmIsw,2930
 uncountable/types/api/recipes/set_recipe_output_file.py,sha256=8s0oxgjhv_hIyV3EoAjIBb970TrJJbKTFaSdNvbgU78,1225
-uncountable/types/api/recipes/set_recipe_outputs.py,sha256=I7NUDO_QrKt1rZB7zHr6WWS2FdXCK8Bqc0BxFkDfjyQ,1729
+uncountable/types/api/recipes/set_recipe_outputs.py,sha256=bLR-FtncB6hzqvzQhCts6ERdHJWgPMne4bQi6-xxu_k,1890
 uncountable/types/api/recipes/set_recipe_tags.py,sha256=kYX13RvHyUOlzL6LbBKDQ8yNvpvNobbnF2WDtgZsKRc,2493
 uncountable/types/api/recipes/unarchive_recipes.py,sha256=KZKzw0fbQdOBUR3YBunFrHFcex4qPMiA8mDgsB6n24k,814
 uncountable/types/api/recipes/unlock_recipes.py,sha256=AvzQeZCLs9i7CuhMs3Xltdi4n0q4Td55yHYyBGQgWag,1103
@@ -277,7 +277,7 @@ uncountable/types/api/triggers/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr
 uncountable/types/api/triggers/run_trigger.py,sha256=_Rpha9nxXI3Xr17CrGDtofg4HZ81x2lt0rMZ6As0qfE,893
 uncountable/types/api/uploader/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
 uncountable/types/api/uploader/invoke_uploader.py,sha256=Rc77y5q-3R9-SNQgm8P35zKaW2D1Hbtm7PDixnOn1G0,1025
-UncountablePythonSDK-0.0.70.dist-info/METADATA,sha256=_GI0ixHy91cpiMAMQfwyE-Sfvowp982PEBAw6-emnM0,2051
-UncountablePythonSDK-0.0.70.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-UncountablePythonSDK-0.0.70.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
-UncountablePythonSDK-0.0.70.dist-info/RECORD,,
+UncountablePythonSDK-0.0.71.dist-info/METADATA,sha256=-WFnnxQlZ9qZ37_W7tekpeHnbqw0uyYzNqqAbd8BL_4,2051
+UncountablePythonSDK-0.0.71.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+UncountablePythonSDK-0.0.71.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
+UncountablePythonSDK-0.0.71.dist-info/RECORD,,

uncountable/integration/queue_runner/job_scheduler.py

@@ -3,6 +3,8 @@ import typing
 from concurrent.futures import ProcessPoolExecutor
 from dataclasses import dataclass
 
+from opentelemetry.trace import get_current_span
+
 from uncountable.integration.db.connect import IntegrationDBService, create_db_engine
 from uncountable.integration.db.session import get_session_maker
 from uncountable.integration.queue_runner.command_server import (
@@ -14,6 +16,7 @@ from uncountable.integration.queue_runner.datastore import DatastoreSqlite
 from uncountable.integration.queue_runner.datastore.interface import Datastore
 from uncountable.integration.queue_runner.worker import Worker
 from uncountable.integration.scan_profiles import load_profiles
+from uncountable.integration.telemetry import Logger
 from uncountable.types import job_definition_t, queued_job_t
 
 from .types import ResultQueue, ResultTask
@@ -72,6 +75,7 @@ def _start_workers(
 
 
 async def start_scheduler(command_queue: CommandQueue) -> None:
+    logger = Logger(get_current_span())
     result_queue: ResultQueue = asyncio.Queue()
     engine = create_db_engine(IntegrationDBService.RUNNER)
     session_maker = get_session_maker(engine)
@@ -87,7 +91,12 @@ async def start_scheduler(command_queue: CommandQueue) -> None:
         queued_jobs = datastore.load_job_queue()
 
         async def enqueue_queued_job(queued_job: queued_job_t.QueuedJob) -> None:
-            worker = job_worker_lookup[queued_job.job_ref_name]
+            try:
+                worker = job_worker_lookup[queued_job.job_ref_name]
+            except KeyError as e:
+                logger.log_exception(e)
+                datastore.remove_job_from_queue(queued_job.queued_job_uuid)
+                return
             await worker.listen_queue.put(queued_job)
 
         for queued_job in queued_jobs:

uncountable/types/api/recipes/set_recipe_outputs.py

@@ -10,6 +10,7 @@ from decimal import Decimal  # noqa: F401
 import dataclasses
 from pkgs.serialization import serial_class
 from ... import base_t
+from ... import field_values_t
 from ... import recipes_t
 from ... import response_t
 
@@ -47,6 +48,7 @@ class RecipeOutputValue:
     value_str: typing.Optional[str] = None
     value_curve: typing.Optional[CurveValues] = None
     formatting: typing.Optional[recipes_t.RecipeAttributeFormatting] = None
+    field_values: typing.Optional[list[typing.Union[field_values_t.ArgumentValueRefName, field_values_t.ArgumentValueId]]] = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec