Fixed missing codeType and other tweaks.

Author: sbyrne-sf

src/datacustomcode/cli.py

@@ -190,16 +190,16 @@ def deploy(
         raise click.Abort()
 
     logger.debug(f"Deploying with CPU size: {cpu_size}")
-
+    base_directory = find_base_directory(path)
+    package_type = get_package_type(base_directory)
     metadata = CodeExtensionMetadata(
         name=name,
         version=version,
         description=description,
         computeType=COMPUTE_TYPES[cpu_size],
+        codeType=package_type,
     )
 
-    base_directory = find_base_directory(path)
-    package_type = get_package_type(base_directory)
     if package_type == "function":
         if not function_invoke_opt:
             click.secho(

src/datacustomcode/deploy.py

@@ -61,6 +61,7 @@ class CodeExtensionMetadata(BaseModel):
     version: str
     description: str
     computeType: str
+    codeType: str
     functionInvokeOptions: Union[list[str], None] = None
 
     def __init__(self, **data):
@@ -161,6 +162,7 @@ def create_deployment(
             "description": metadata.description,
             "version": metadata.version,
             "computeType": metadata.computeType,
+            "codeType": metadata.codeType,
         }
     )
     if metadata.functionInvokeOptions:
@@ -294,11 +296,11 @@ def wait_for_deployment(
 
 
 class BaseConfig(BaseModel):
-    sdkVersion: str
     entryPoint: str
 
 
 class DataTransformConfig(BaseConfig):
+    sdkVersion: str
     dataspace: str
     permissions: Permissions
 

src/datacustomcode/scan.py

@@ -44,7 +44,6 @@
 }
 
 FUNCTION_CONFIG_TEMPLATE = {
-    "sdkVersion": get_version(),
     "entryPoint": "",
 }
 STANDARD_LIBS = set(sys.stdlib_module_names)

tests/test_deploy.py

@@ -496,6 +496,7 @@ def test_create_deployment_success(self, mock_make_api_call):
             version="1.0.0",
             description="Test job",
             computeType="CPU_M",
+            codeType="script",
         )
 
         mock_make_api_call.return_value = {
@@ -519,6 +520,7 @@ def test_create_deployment_conflict(self, mock_make_api_call):
             version="1.0.0",
             description="Test job",
             computeType="CPU_M",
+            codeType="script",
         )
 
         # Mock HTTP error with 409 Conflict
@@ -543,6 +545,7 @@ def test_create_deployment_function_invoke_options(self, mock_make_api_call):
             description="Test job",
             computeType="CPU_M",
             functionInvokeOptions=["option1", "option2"],
+            codeType="function",
         )
 
         mock_make_api_call.return_value = {
@@ -663,6 +666,7 @@ def test_get_deployments(self, mock_make_api_call):
             version="1.0.0",
             description="Test job",
             computeType="CPU_M",
+            codeType="script",
         )
 
         mock_make_api_call.return_value = {"deploymentStatus": "Deployed"}
@@ -690,6 +694,7 @@ def test_wait_for_deployment_success(
             version="1.0.0",
             description="Test job",
             computeType="CPU_M",
+            codeType="script",
         )
         callback = MagicMock()
 
@@ -720,6 +725,7 @@ def test_wait_for_deployment_timeout(
             version="1.0.0",
             description="Test job",
             computeType="CPU_M",
+            codeType="script",
         )
 
         # Mock time to simulate timeout
@@ -800,6 +806,7 @@ def test_create_data_transform(self, mock_make_api_call, mock_get_config):
             version="1.0.0",
             description="Test job",
             computeType="CPU_M",
+            codeType="script",
         )
 
         data_transform_config = DataTransformConfig(
@@ -878,6 +885,7 @@ def test_deploy_full(
             version="1.0.0",
             description="Test job",
             computeType="CPU_M",
+            codeType="script",
         )
         callback = MagicMock()
 
@@ -918,6 +926,7 @@ def test_run_data_transform(self, mock_make_api_call):
             version="1.0.0",
             description="Test job",
             computeType="CPU_M",
+            codeType="script",
         )
 
         mock_make_api_call.return_value = {"status": "Running"}
@@ -962,6 +971,7 @@ def test_deploy_full_happy_path(
             version="1.0.0",
             description="Test job",
             computeType="CPU_M",
+            codeType="script",
         )
         callback = MagicMock()