omnata-plugin-runtime 0.5.7a137__tar.gz → 0.5.7a139__tar.gz

{omnata_plugin_runtime-0.5.7a137 → omnata_plugin_runtime-0.5.7a139}/PKG-INFO

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

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

@@ -2235,7 +2235,6 @@ class SnowflakeFunctionParameter:
     name: str
     description: str
     data_type: str
-    has_default_value: bool = False
     default_value_clause: Optional[str] = None
 
 @dataclass
@@ -2245,6 +2244,7 @@ class SnowflakeUDTFResultColumn:
     description: str
 
 def consumer_udtf(
+        name:str,
         description: str,
         params: List[SnowflakeFunctionParameter],
         result_columns: List[SnowflakeUDTFResultColumn]):
@@ -2252,23 +2252,23 @@ def consumer_udtf(
         # Get the original 'process' method from the class
         original_process = getattr(cls, 'process')
         sig = signature(original_process)
-        params = sig.parameters
-        if len(params) < 2:
+        function_params = sig.parameters
+        if len(function_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
-        first_param_name = list(params.keys())[0]
+        first_param_name = list(function_params.keys())[0]
 
         if first_param_name != 'self':
             raise ValueError(f"The first argument should be 'self', instead it was '{first_param_name}'.")
 
-        second_param_name = list(params.keys())[1]
+        second_param_name = list(function_params.keys())[1]
 
         if second_param_name != 'connection_parameters':
             raise ValueError(f"The second argument should be 'connection_parameters', instead it was {second_param_name}.")
 
 
-        if params[second_param_name].annotation != ConnectionConfigurationParameters:
-            raise ValueError(f"The second argument must be a ConnectionConfigurationParameters, instead it was a {params[second_param_name].annotation}.")
+        if function_params[second_param_name].annotation != ConnectionConfigurationParameters:
+            raise ValueError(f"The second argument must be a ConnectionConfigurationParameters, instead it was a {function_params[second_param_name].annotation}.")
 
         @wraps(original_process)
         def wrapped_process(self, connection_parameters, *args, **kwargs):
@@ -2295,6 +2295,7 @@ def consumer_udtf(
         # Replace the original 'process' method with the wrapped version
         setattr(cls, 'process', wrapped_process)
         cls._is_omnata_consumer_udtf = True
+        cls._omnata_consumer_udtf_name = name
         cls._omnata_consumer_udtf_description = description
         cls._omnata_consumer_udtf_params = params
         cls._omnata_consumer_udtf_result_columns = result_columns
@@ -2323,20 +2324,21 @@ def find_consumer_udtf_classes(path:str = '.'):
 
 
 def consumer_udf(
+        name: str,
         description: str,
         params: List[SnowflakeFunctionParameter],
         result_data_type: str):
     def decorator(func):
         sig = signature(func)
-        params = sig.parameters
-        if len(params) == 0:
+        function_params = sig.parameters
+        if len(function_params) == 0:
             raise ValueError("The function must have at least one parameter.")
         # Ensure the first argument is mandatory and positional
-        first_param_name = list(params.keys())[0]
+        first_param_name = list(function_params.keys())[0]
         if first_param_name != 'connection_parameters':
             raise ValueError(f"The first argument should be 'connection_parameters', instead it was '{first_param_name}'.")
-        if params[first_param_name].annotation != ConnectionConfigurationParameters:
-            raise ValueError(f"The first argument must be a ConnectionConfigurationParameters, instead it was a {params[first_param_name].annotation}.")
+        if function_params[first_param_name].annotation != ConnectionConfigurationParameters:
+            raise ValueError(f"The first argument must be a ConnectionConfigurationParameters, instead it was a {function_params[first_param_name].annotation}.")
 
         @wraps(func)
         def wrapper(self, connection_parameters, *args, **kwargs):
@@ -2363,6 +2365,7 @@ def consumer_udf(
             return func(parameters, *args, **kwargs)
         
         wrapper._is_omnata_consumer_udf = True
+        wrapper._omnata_consumer_udf_name = name
         wrapper._omnata_consumer_udf_description = description
         wrapper._omnata_consumer_udf_params = params
         wrapper._omnata_consumer_udf_result_data_type = result_data_type