omnata-plugin-runtime 0.10.22a276__tar.gz → 0.10.23__tar.gz

{omnata_plugin_runtime-0.10.22a276 → omnata_plugin_runtime-0.10.23}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: omnata-plugin-runtime
-Version: 0.10.22a276
+Version: 0.10.23
 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.22a276 → omnata_plugin_runtime-0.10.23}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-runtime"
-version = "0.10.22-a276"
+version = "0.10.23"
 description = "Classes and common runtime components for building and running Omnata Plugins"
 authors = ["James Weakley <james.weakley@omnata.com>"]
 readme = "README.md"

{omnata_plugin_runtime-0.10.22a276 → omnata_plugin_runtime-0.10.23}/src/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(
@@ -1531,7 +1546,7 @@ class InboundSyncRequest(SyncRequest):
         Updates the SYNC table to have the latest stream states.
         TODO: This should be done in concert with the results, revisit
         """
-        if self._last_states_update is None or self._latest_states != self._last_states_update:
+        if self._last_states_update is None or json.dumps(self._latest_states) != json.dumps(self._last_states_update):
             self._plugin_message(PluginMessageStreamState(stream_state=self._latest_states))
             self._last_states_update = self._latest_states