fix(cloudapp): fix cloudapp message_send result class for both Task and Message (#314)

* fix(cloudapp): fix cloudapp message_send result class for both Task and Message

* fix(cloudapp): fix for UniTest

* fix import bug

* back to origin

* fix

* fix import

* fix import unitest

* fix for unitest

Author: floritange

veadk/cloud/cloud_app.py

@@ -422,7 +422,26 @@ async def message_send(
 
                 # we ignore type checking here, because the response
                 # from CloudApp will not be `Task` type
-                return res.root.result  # type: ignore
+                result = res.root.result  # type: ignore
+                try:
+                    from a2a.types import Task
+                except ImportError:
+                    return result
+
+                if isinstance(result, Task):
+                    if result.history:  # type: ignore
+                        return next(
+                            (
+                                msg
+                                for msg in reversed(result.history)  # type: ignore
+                                if msg.role == "agent"
+                            ),
+                            None,
+                        )
+                    else:
+                        return None
+                else:
+                    return result
             except Exception as e:
                 logger.error(f"Failed to send message to cloud app. Error: {e}")
                 return None