omnata-plugin-runtime 0.5.7a146__tar.gz → 0.5.7a148__tar.gz

{omnata_plugin_runtime-0.5.7a146 → omnata_plugin_runtime-0.5.7a148}/PKG-INFO

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

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

@@ -2253,7 +2253,8 @@ class SnowflakeUDTFResultColumn:
     """
     name: str
     data_type: str
-    description: str
+    def __str__(self):
+        return f"{self.name} {self.data_type}"
 
 @dataclass
 class PythonUDTFDefinition:
@@ -2456,7 +2457,9 @@ def find_udtf_classes(path:str = '.') -> List[PythonUDTFDefinition | JavaUDTFDef
                 matching_classes.append(PythonUDTFDefinition(
                     name=obj._omnata_udtf_name,
                     description=obj._omnata_udtf_description,
-                    params=obj._omnata_udtf_params,
+                    params=[SnowflakeFunctionParameter(name='CONNECTION_PARAMETERS',
+                                                       data_type='OBJECT',
+                                                       description='The connection object, obtained from calling PLUGIN.PLUGIN_CONNECTION')]+obj._omnata_udtf_params,
                     result_columns=obj._omnata_udtf_result_columns,
                     expose_to_consumer=obj._omnata_udtf_expose_to_consumer,
                     handler=obj.__module__+'.'+obj.__name__
@@ -2544,7 +2547,9 @@ def find_udf_functions(path:str = '.') -> List[PythonUDFDefinition | JavaUDFDefi
                 matching_classes.append(PythonUDFDefinition(
                     name=obj._omnata_udf_name,
                     description=obj._omnata_udf_description,
-                    params=obj._omnata_udf_params,
+                    params=[SnowflakeFunctionParameter(name='CONNECTION_PARAMETERS',
+                                                       data_type='OBJECT',
+                                                       description='The connection object, obtained from calling PLUGIN.PLUGIN_CONNECTION')] + obj._omnata_udf_params,
                     result_data_type=obj._omnata_udf_result_data_type,
                     expose_to_consumer=obj._omnata_udf_expose_to_consumer,
                     packages=[],