Fix rpc input deserialization (#76)

* a fix to the rpc input

* Fix rpc input deserialization

* output type assert

* fix lint

Author: longquanzheng

iwf/rpc.py

@@ -69,6 +69,7 @@ def wrapper(*args, **kwargs):
             if v.annotation not in valid_param_types_exclude_input:
                 if not has_input:
                     has_input = True
+                    rpc_info.input_type = v.annotation
                 else:
                     raise rpc_definition_err
         if not need_persistence:

iwf/tests/test_rpc.py

@@ -1,6 +1,7 @@
 import inspect
 import time
 import unittest
+from dataclasses import dataclass
 
 from iwf.client import Client
 from iwf.command_request import CommandRequest, InternalChannelCommand
@@ -22,6 +23,12 @@
 idle_channel_name = "test-3"
 
 
+@dataclass
+class Mydata:
+    strdata: str
+    intdata: int
+
+
 class WaitState(WorkflowState[None]):
     def wait_until(
         self,
@@ -102,6 +109,12 @@ def test_rpc_publish_to_idle_channel(self, com: Communication, data: str):
         com.publish_to_internal_channel(idle_channel_name, data)
         return com.get_internal_channel_size(idle_channel_name)
 
+    @rpc()
+    def test_rpc_input_type(self, input: Mydata) -> Mydata:
+        if input.intdata != 100 or input.strdata != "test":
+            raise Exception("input type test failed")
+        return input
+
 
 class TestRPCs(unittest.TestCase):
     @classmethod
@@ -113,6 +126,13 @@ def setUpClass(cls):
     def test_simple_rpc(self):
         wf_id = f"{inspect.currentframe().f_code.co_name}-{time.time_ns()}"
         self.client.start_workflow(RPCWorkflow, wf_id, 10)
+
+        input = Mydata("test", 100)
+        output = self.client.invoke_rpc(
+            wf_id, RPCWorkflow.test_rpc_input_type, input, Mydata
+        )
+        assert output == input
+
         output = self.client.invoke_rpc(wf_id, RPCWorkflow.test_simple_rpc)
         assert output == 123
         wf = RPCWorkflow()