deployment changes dev

Author: shushilgirish

scripts/check_udf_signature.py

@@ -0,0 +1,55 @@
+import os
+import sys
+import argparse
+
+def check_udf_signature(udf_path):
+    """Check the signature of a UDF function.py file."""
+    function_file = os.path.join(udf_path, "function.py")
+    
+    if not os.path.exists(function_file):
+        print(f"Error: Function file not found at: {function_file}")
+        return False
+    
+    with open(function_file, 'r') as f:
+        content = f.read()
+    
+    # Look for the main function definition
+    if "def main(session, input_data" in content:
+        print("✅ UDF uses Snowpark style with session and input_data parameters")
+        print("Use this SQL:")
+        print("""
+        CREATE OR REPLACE FUNCTION UDF_NAME(input_data VARIANT)
+        RETURNS VARIANT
+        LANGUAGE PYTHON
+        RUNTIME_VERSION=3.8
+        PACKAGES = ('snowflake-snowpark-python')
+        IMPORTS = ('@STAGE/path/to/zip')
+        HANDLER = 'function.main'
+        """)
+        return True
+        
+    elif "def main(input_data" in content:
+        print("✅ UDF uses basic style with just input_data parameter")
+        print("Use this SQL:")
+        print("""
+        CREATE OR REPLACE FUNCTION UDF_NAME(input_data VARIANT)
+        RETURNS VARIANT
+        LANGUAGE PYTHON
+        RUNTIME_VERSION=3.8
+        PACKAGES = ('snowflake-snowpark-python')
+        IMPORTS = ('@STAGE/path/to/zip')
+        HANDLER = 'function.main'
+        """)
+        return True
+    
+    else:
+        print("❌ Could not identify UDF signature pattern")
+        print("Please check the function.py file manually")
+        return False
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description='Check UDF function signature')
+    parser.add_argument('udf_path', help='Path to UDF directory')
+    args = parser.parse_args()
+    
+    check_udf_signature(args.udf_path)

scripts/snowflake_deployer.py

@@ -190,6 +190,19 @@ def deploy_component(profile_name, component_path, component_name, component_typ
                 for f in files:
                     logger.info(f"{sub_indent}{f}")
 
+            # Check if UDF requires session parameter by examining function.py
+            has_session_param = False
+            try:
+                function_file = os.path.join(code_dir, "function.py")
+                if os.path.exists(function_file):
+                    with open(function_file, 'r') as f:
+                        content = f.read()
+                        if "def main(session, input_data" in content:
+                            logger.info("Detected UDF with session parameter")
+                            has_session_param = True
+            except Exception as e:
+                logger.warning(f"Could not check function signature: {str(e)}")
+            
             # Zip the directory
             zip_directory(code_dir, zip_path)
             logger.info(f"Created zip file: {zip_path}")
@@ -204,6 +217,7 @@ def deploy_component(profile_name, component_path, component_name, component_typ
             import_path = f"@{stage_name}/{component_name.replace(' ', '_')}/{zip_filename}"
 
             if component_type.lower() == "udf":
+                # For Snowpark UDFs that use session parameter
                 sql = f"""
                 CREATE OR REPLACE FUNCTION {component_name.replace(' ', '_')}(input_data VARIANT)
                 RETURNS VARIANT
@@ -305,4 +319,4 @@ def execute_sql_file(profile_name, sql_file):
 
     else:
         parser.print_help()
-        sys.exit(1)
+        sys.exit(1)