Improving external SDK for function

Author: ritaagarwala-sf

src/datacustomcode/cli.py

@@ -83,14 +83,15 @@ def _generate_function_test_file(entrypoint_path: str) -> Optional[str]:
     Returns:
         Path to generated test.json, or None if generation failed
     """
-    from datacustomcode.template import generate_test_json
+    from datacustomcode.function_utils import generate_test_json
 
     tests_dir = os.path.join(os.path.dirname(entrypoint_path), "tests")
     os.makedirs(tests_dir, exist_ok=True)
     test_json_path = os.path.join(tests_dir, "test.json")
 
     try:
         generate_test_json(entrypoint_path, test_json_path)
+        logger.debug(f"Generated test JSON at {test_json_path}")
         return test_json_path
     except Exception as e:
         logger.warning(f"Could not generate test.json: {e}")

src/datacustomcode/run.py

@@ -191,11 +191,7 @@ def run_function_with_test(entrypoint: str, test_file: str) -> None:
     print("RESULT:")
     print("=" * 80)
     if hasattr(result, "model_dump"):
-        # Pydantic v2
         print(json.dumps(result.model_dump(), indent=2))
-    elif hasattr(result, "dict"):
-        # Pydantic v1
-        print(json.dumps(result.dict(), indent=2))
     else:
         print(result)
     print("=" * 80)

src/datacustomcode/template.py

@@ -62,15 +62,3 @@ def copy_function_template(target_dir: str, use_in_feature: str) -> None:
             shutil.copy2(source, destination)
 
 
-# Re-export generate_test_json from function_utils for backwards compatibility
-def generate_test_json(entrypoint_path: str, output_path: str) -> None:
-    """Generate a sample test.json file for a function.
-
-    Args:
-        entrypoint_path: Path to the function entrypoint.py
-        output_path: Output path for test.json
-    """
-    from datacustomcode.function_utils import generate_test_json as _generate_test_json
-
-    _generate_test_json(entrypoint_path, output_path)
-    logger.debug(f"Generated test JSON at {output_path}")