fix(exit): Using send_any if you send Any type

.kai/local/kai-project-template.yaml

@@ -1,4 +1,4 @@
-version: v0.1.3
+version: v0.1.4
 description: kai-project-template
 config: {}
 workflows:
@@ -31,7 +31,7 @@ workflows:
             limit: 800M
       - name: training
         type: task
-        image: registry.kai.local/kai-project-template_training:v0.0.3
+        image: registry.kai.local/kai-project-template_training:v0.0.4
         replicas: 1
         gpu: false
         objectStore: null

src/demo-workflow/tasks/training/Makefile

@@ -2,7 +2,7 @@
 
 # This variable have the environment to use, by default is local, but you can change when you run the make command
 ENV ?= local
-PROCESS_VERSION ?= v0.0.3
+PROCESS_VERSION ?= v0.0.4
 
 # Read the PRODUCT_NAME variable based on the file inside the corresponding directory
 PRODUCT_NAME := $(shell basename $(wildcard ./../../../../.kai/$(ENV)/*.yaml) .yaml)

src/demo-workflow/tasks/training/main.py

@@ -2,6 +2,7 @@
 from sdk.kai_sdk import KaiSDK
 from runner.runner import Runner
 from google.protobuf.any_pb2 import Any
+from google.protobuf.struct_pb2 import Value
 
 
 async def initializer(sdk: KaiSDK):
@@ -10,7 +11,12 @@ async def initializer(sdk: KaiSDK):
 
 async def handler(sdk: KaiSDK, message: Any):
     sdk.logger.info(f"Received message in training: {message}")
-    await sdk.messaging.send_any(response=message)
+    response = Value()
+    response.struct_value.fields["status_code"].string_value = "200"
+    response.struct_value.fields[
+        "message"
+    ].string_value = "Training completed successfully"
+    await sdk.messaging.send_output(response=response)
 
 
 async def finalizer(sdk: KaiSDK):