omnata-plugin-runtime 0.5.8a159__py3-none-any.whl → 0.5.9a160__py3-none-any.whl

omnata_plugin_runtime/omnata_plugin.py

@@ -2429,7 +2429,15 @@ def find_udtf_classes(path:str = '.') -> List[UDTFDefinition]:
         for name, obj in inspect.getmembers(module):
             # Check if the class has the specified attribute
             if isinstance(obj, UDTFDefinition) and cast(UDTFDefinition,obj).language == 'java':
-                matching_classes.append(obj)
+                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(
+                                                name='CONNECTION_PARAMETERS',
+                                                data_type='OBJECT',
+                                                description='The connection object, obtained from calling PLUGIN.PLUGIN_CONNECTION.')] + udtf_obj.params
+                    matching_classes.append(udtf_obj)
             
     return matching_classes
 
@@ -2535,7 +2543,15 @@ def find_udf_functions(path:str = '.') -> List[UDFDefinition]:
         # java doesn't use the python decorator, so we need to check for the class directly
         for name, obj in inspect.getmembers(module):
             # Check if the class has the specified attribute
-            if isinstance(obj, UDFDefinition) and cast(UDFDefinition,obj).language == 'java':
-                matching_classes.append(obj)
+            if isinstance(obj, 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(
+                                                name='CONNECTION_PARAMETERS',
+                                                data_type='OBJECT',
+                                                description='The connection object, obtained from calling PLUGIN.PLUGIN_CONNECTION.')] + udf_obj.params
+                    matching_classes.append(udf_obj)
 
     return matching_classes

{omnata_plugin_runtime-0.5.8a159.dist-info → omnata_plugin_runtime-0.5.9a160.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.5.8a159
+Version: 0.5.9a160
 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.8a159.dist-info → omnata_plugin_runtime-0.5.9a160.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=7fZ3CQk4jGGyyUrxuQI4DJV7OTavP1IuMDuxaPdV8RM,127455
+omnata_plugin_runtime/omnata_plugin.py,sha256=gOSN0cqFG5pFesdvS7COQnw9fHp4dFUwM9XIMbng_Xc,128844
 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.8a159.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-omnata_plugin_runtime-0.5.8a159.dist-info/METADATA,sha256=wLZepggw9k2OIpG7qMtOpDBJANXZ-8oJt3pISQAWrZQ,1985
-omnata_plugin_runtime-0.5.8a159.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-omnata_plugin_runtime-0.5.8a159.dist-info/RECORD,,
+omnata_plugin_runtime-0.5.9a160.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+omnata_plugin_runtime-0.5.9a160.dist-info/METADATA,sha256=mFgWPJAcpXT_pZUDsJhqm1LJe5gsJyEA16HPKhKow3I,1985
+omnata_plugin_runtime-0.5.9a160.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+omnata_plugin_runtime-0.5.9a160.dist-info/RECORD,,