UncountablePythonSDK 0.0.70__tar.gz → 0.0.71__tar.gz

{uncountablepythonsdk-0.0.70/UncountablePythonSDK.egg-info → uncountablepythonsdk-0.0.71}/PKG-INFO

@@ -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 → uncountablepythonsdk-0.0.71/UncountablePythonSDK.egg-info}/PKG-INFO

@@ -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 → uncountablepythonsdk-0.0.71}/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:

{uncountablepythonsdk-0.0.70 → uncountablepythonsdk-0.0.71}/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