bug fix: xcom

docs/getting-started.md

@@ -907,11 +907,37 @@ Parallel(set_xcom_cmd, set_xcom_py) >> Parallel(get_xcom_cmd, get_xcom_py) >> te
 runner.register(test_xcom)
 ```
 
-The example shows that `set-xcom-cmd` and `set-xcom-py` set XCom values `one` and `two` respectively.
+The example shows that `set-xcom-cmd` and `set-xcom-py` set XCom values `one` and `two`, respectively.
 
 On the other hand, `get-xcom-cmd` and `get-xcom-py` fetch the values and print them.
 
-Furthermore, every Zrb Task has their return values saved as `__xcom['execution-id']['task-name']`.
+Furthermore, every Zrb Task has its return values saved as `__xcom['execution-id']['task-name']`. Let's see the following example:
+
+```python
+from zrb import runner, Parallel, CmdTask, python_task
+
+hello_cmd = CmdTask(
+    name='hello-cmd',
+    cmd='echo hello-cmd',
+)
+
+@python_task(
+    name='hello-py'
+)
+def hello_py(*args, **kwargs):
+    return 'hello-py'
+
+hello = CmdTask(
+    name='hello',
+    cmd=[
+        'echo {{task.get_xcom("hello-cmd")}}',
+        'echo {{task.get_xcom("hello-py")}}',
+    ],
+)
+
+Parallel(hello_cmd, hello_py) >> hello
+runner.register(hello)
+```
 
 
 ## Basic Example

src/zrb/task/base_task/base_task.py

@@ -255,7 +255,6 @@ async def _run_and_check_all(
             ]
             results = await asyncio.gather(*coroutines)
             result = results[-1]
-            self.set_xcom(self.get_name(), f'{result}')
             self._print_result(result)
             return result
         except Exception as e:
@@ -379,6 +378,8 @@ async def _cached_run(self, *args: Any, **kwargs: Any) -> Any:
                 self._increase_attempt()
                 await asyncio.sleep(self._retry_interval)
                 await self.on_retry()
+        self.set_xcom(self.get_name(), f'{result}')
+        self.log_debug(f'XCom: {self.__xcom}')
         await self._mark_done()
         return result