omnata-plugin-runtime 0.5.7a134__py3-none-any.whl → 0.5.7a136__py3-none-any.whl

omnata_plugin_runtime/omnata_plugin.py

@@ -2232,16 +2232,16 @@ def get_nested_value(nested_dict:Dict, keys:List[str]):
 
 @dataclass
 class SnowflakeFunctionParameter:
-    arg_name: str
+    name: str
     description: str
-    arg_data_type: str
+    data_type: str
     has_default_value: bool = False
     default_value_clause: Optional[str] = None
 
 @dataclass
 class SnowflakeUDTFResultColumn:
-    column_name: str
-    column_data_type: str
+    name: str
+    data_type: str
     description: str
 
 def consumer_udtf(
@@ -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):

{omnata_plugin_runtime-0.5.7a134.dist-info → omnata_plugin_runtime-0.5.7a136.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.5.7a134
+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.7a134.dist-info → omnata_plugin_runtime-0.5.7a136.dist-info}/RECORD

@@ -3,10 +3,10 @@ omnata_plugin_runtime/api.py,sha256=FxzTqri4no8ClkOm7vZADG8aD47jcGBCTTQDEORmOJM,
 omnata_plugin_runtime/configuration.py,sha256=TI6GaVFhewVawBCaYN34GujY57qEP6q2nik4YpSEk5s,38100
 omnata_plugin_runtime/forms.py,sha256=GzSPEwcijsoPCXEO1mHiE8ylvX_KSE5TkhwqkymA2Ss,19755
 omnata_plugin_runtime/logging.py,sha256=bn7eKoNWvtuyTk7RTwBS9UARMtqkiICtgMtzq3KA2V0,3272
-omnata_plugin_runtime/omnata_plugin.py,sha256=WgdUO5PvwwhekdB-hSLr_4hqjK--BXmm6G3opgeDMA8,118075
+omnata_plugin_runtime/omnata_plugin.py,sha256=kW5ZaD763D2yaa9gKyr8e2EoneB3LN01_WPA-lj9cYk,118441
 omnata_plugin_runtime/plugin_entrypoints.py,sha256=PFSLsYEVnWHVvSoOYTtTK2JY6pp6_8_eYP53WqLRiPE,27975
 omnata_plugin_runtime/rate_limiting.py,sha256=DVQ_bc-mVLBkrU1PTns1MWXhHiLpSB5HkWCcdePtJ2A,25611
-omnata_plugin_runtime-0.5.7a134.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-omnata_plugin_runtime-0.5.7a134.dist-info/METADATA,sha256=JGVpgMpr1s7vVOCc9qdKTNKKLxoFnOjx27LtePTLDmM,1985
-omnata_plugin_runtime-0.5.7a134.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-omnata_plugin_runtime-0.5.7a134.dist-info/RECORD,,
+omnata_plugin_runtime-0.5.7a136.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+omnata_plugin_runtime-0.5.7a136.dist-info/METADATA,sha256=e9bGEeVLZ4dOfQsh5NPkpFf2kH3H_JEfAtbYl0wiQE0,1985
+omnata_plugin_runtime-0.5.7a136.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+omnata_plugin_runtime-0.5.7a136.dist-info/RECORD,,