omnata-plugin-runtime 0.5.7a135__tar.gz → 0.5.7a136__tar.gz

{omnata_plugin_runtime-0.5.7a135 → omnata_plugin_runtime-0.5.7a136}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.5.7a135
+Version: 0.5.7a136
 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.5.7a135 → omnata_plugin_runtime-0.5.7a136}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-runtime"
-version = "0.5.7-a135"
+version = "0.5.7-a136"
 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.5.7a135 → omnata_plugin_runtime-0.5.7a136}/src/omnata_plugin_runtime/omnata_plugin.py

@@ -2254,7 +2254,8 @@ def consumer_udtf(
         original_process = getattr(cls, 'process')
         sig = signature(original_process)
         params = sig.parameters
-
+        if len(params) < 2:
+            raise ValueError("The function must have the self parameter, plus at least the mandatory 'connection_parameters' parameter.")
         # Ensure the first argument is mandatory and positional
         if list(params.keys())[0] != 'self':
             raise ValueError("The first argument should be 'self' for class methods.")
@@ -2262,8 +2263,8 @@ def consumer_udtf(
         if list(params.keys())[1] != 'connection_parameters':
             raise ValueError("The second argument should be 'connection_parameters'.")
         
-        if params[list(params.keys())[1]].annotation != Dict:
-            raise ValueError("The second argument must be a dict.")
+        if params[list(params.keys())[1]].annotation != ConnectionConfigurationParameters:
+            raise ValueError("The second argument must be a ConnectionConfigurationParameters.")
 
         @wraps(original_process)
         def wrapped_process(self, connection_parameters, *args, **kwargs):
@@ -2321,12 +2322,13 @@ def consumer_udf(
     def decorator(func):
         sig = signature(func)
         params = sig.parameters
-
+        if len(params) == 0:
+            raise ValueError("The function must have at least one parameter.")
         # Ensure the first argument is mandatory and positional
         if list(params.keys())[0] != 'connection_parameters':
             raise ValueError("The first argument should be 'connection_parameters'.")
-        if params[list(params.keys())[0]].annotation != Dict:
-            raise ValueError("The first argument must be a dict.")
+        if params[list(params.keys())[0]].annotation != ConnectionConfigurationParameters:
+            raise ValueError("The first argument must be a ConnectionConfigurationParameters.")
 
         @wraps(func)
         def wrapper(self, connection_parameters, *args, **kwargs):