omnata-plugin-runtime 0.5.9a160__tar.gz → 0.5.9a161__tar.gz

{omnata_plugin_runtime-0.5.9a160 → omnata_plugin_runtime-0.5.9a161}/PKG-INFO

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

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

@@ -2432,8 +2432,9 @@ def find_udtf_classes(path:str = '.') -> List[UDTFDefinition]:
                 udtf_obj = cast(UDTFDefinition,obj)
                 if udtf_obj.language == 'java':
                     # because the decorator wasn't used, we need to check if we need to add the connection_parameters parameter
-                    if udtf_obj.expose_to_consumer and udtf_obj.params[0].name.upper() != 'CONNECTION_PARAMETERS':
-                        udtf_obj.params = [SnowflakeFunctionParameter(
+                    if udtf_obj.expose_to_consumer:
+                        if len(udtf_obj.params) == 0 or udtf_obj.params[0].name.upper() != 'CONNECTION_PARAMETERS':
+                            udtf_obj.params = [SnowflakeFunctionParameter(
                                                 name='CONNECTION_PARAMETERS',
                                                 data_type='OBJECT',
                                                 description='The connection object, obtained from calling PLUGIN.PLUGIN_CONNECTION.')] + udtf_obj.params
@@ -2547,8 +2548,9 @@ def find_udf_functions(path:str = '.') -> List[UDFDefinition]:
                 udf_obj = cast(UDFDefinition,obj)
                 if udf_obj.language == 'java':
                     # because the decorator wasn't used, we need to check if we need to add the connection_parameters parameter
-                    if udf_obj.expose_to_consumer and udf_obj.params[0].name.upper() != 'CONNECTION_PARAMETERS':
-                        udf_obj.params = [SnowflakeFunctionParameter(
+                    if udf_obj.expose_to_consumer:
+                        if len(udf_obj.params) == 0 or udf_obj.params[0].name.upper() != 'CONNECTION_PARAMETERS':
+                            udf_obj.params = [SnowflakeFunctionParameter(
                                                 name='CONNECTION_PARAMETERS',
                                                 data_type='OBJECT',
                                                 description='The connection object, obtained from calling PLUGIN.PLUGIN_CONNECTION.')] + udf_obj.params