omnata-plugin-runtime 0.10.21a275__py3-none-any.whl → 0.10.22__py3-none-any.whl

omnata_plugin_runtime/omnata_plugin.py

@@ -50,6 +50,8 @@ from tenacity import Retrying, stop_after_attempt, wait_fixed, retry_if_exceptio
 
 from .logging import OmnataPluginLogHandler, logger, tracer
 from opentelemetry import context
+import math
+import numpy as np
 
 from .api import (
     PluginMessage,
@@ -653,6 +655,21 @@ class HttpRateLimiting:
     def __exit__(self, exc_type, exc_value, traceback):
         http.client.HTTPConnection.putrequest = self.original_putrequest  # type: ignore
 
+def wrap_result_value(x):
+    # Check for NaN (float or numpy.nan)
+    if x is None:
+        return None
+    if isinstance(x, float) and math.isnan(x):
+        return None
+    if isinstance(x, str):
+        return json.dumps({"value": x})
+    try:
+        # Try to detect pandas NaN (which is float('nan'))
+        if isinstance(x, np.floating) and np.isnan(x):
+            return None
+    except ImportError:
+        return None
+    return json.dumps(x)
 
 class OutboundSyncRequest(SyncRequest):
     """
@@ -810,9 +827,7 @@ class OutboundSyncRequest(SyncRequest):
             if "LOADBATCH_ID" not in results_df:
                 results_df["LOADBATCH_ID"] = self._get_next_loadbatch_id()
             # we dump the result data to a json string to make uploading to Snowflake less error prone, but only if it's not None
-            results_df["RESULT"] = results_df["RESULT"].apply(
-                lambda x: json.dumps(x) if x is not None else None
-            )
+            results_df["RESULT"] = results_df["RESULT"].apply(wrap_result_value)
         # trim out the columns we don't need to return
         return results_df[
             results_df.columns.intersection(
@@ -1911,7 +1926,7 @@ class FixedSizeGenerator:
 
     def __next__(self):
         with self.thread_lock:
-            logger.debug(f"initial leftovers: {self.leftovers}")
+            logger.debug(f"FixedSizeGenerator initial leftovers: {len(self.leftovers) if self.leftovers is not None else 0}")
             records_df = self.leftovers
             self.leftovers = None
             try:

{omnata_plugin_runtime-0.10.21a275.dist-info → omnata_plugin_runtime-0.10.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: omnata-plugin-runtime
-Version: 0.10.21a275
+Version: 0.10.22
 Summary: Classes and common runtime components for building and running Omnata Plugins
 Author: James Weakley
 Author-email: james.weakley@omnata.com

{omnata_plugin_runtime-0.10.21a275.dist-info → omnata_plugin_runtime-0.10.22.dist-info}/RECORD

@@ -4,10 +4,10 @@ omnata_plugin_runtime/configuration.py,sha256=ud3O9R03BbUjGUJDVpGbnADqKw5oLVy65E
 omnata_plugin_runtime/forms.py,sha256=Lrbr3otsFDrvHWJw7v-slsW4PvEHJ6BG1Yl8oaJfiDo,20529
 omnata_plugin_runtime/json_schema.py,sha256=rqpJ5wjZiaE6shFB6Rv5CRHIl8xGxntibNjlh4A6Zr0,45902
 omnata_plugin_runtime/logging.py,sha256=WBuZt8lF9E5oFWM4KYQbE8dDJ_HctJ1pN3BHwU6rcd0,4461
-omnata_plugin_runtime/omnata_plugin.py,sha256=hssSfMkdwnZjkeMu7rsOtsHgF8-UzA69QdzMq6KqE6k,133542
+omnata_plugin_runtime/omnata_plugin.py,sha256=1tEm-RPx6i9n7v6xYe7qYKsY6WNYRf7Du9uqYlpsXck,134026
 omnata_plugin_runtime/plugin_entrypoints.py,sha256=iqGl8_nEEnPGKg3Aem4YLSQ6d5xS3ju5gq8MJbx6sCA,31968
 omnata_plugin_runtime/rate_limiting.py,sha256=qpr5esU4Ks8hMzuMpSR3gLFdor2ZUXYWCjmsQH_K6lQ,25882
-omnata_plugin_runtime-0.10.21a275.dist-info/LICENSE,sha256=rGaMQG3R3F5-JGDp_-rlMKpDIkg5n0SI4kctTk8eZSI,56
-omnata_plugin_runtime-0.10.21a275.dist-info/METADATA,sha256=TKcGQiX-wmSMadoPZf94lKvH4CtEw8kkcfUeoDCeCPY,2212
-omnata_plugin_runtime-0.10.21a275.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-omnata_plugin_runtime-0.10.21a275.dist-info/RECORD,,
+omnata_plugin_runtime-0.10.22.dist-info/LICENSE,sha256=rGaMQG3R3F5-JGDp_-rlMKpDIkg5n0SI4kctTk8eZSI,56
+omnata_plugin_runtime-0.10.22.dist-info/METADATA,sha256=eh4KD3913zPx5PFGNBFjlqHEFOCLytUnTd1rg7l87fY,2208
+omnata_plugin_runtime-0.10.22.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+omnata_plugin_runtime-0.10.22.dist-info/RECORD,,