Add checking

Author: holtskinner

examples/google_adk/calendar_agent/adk_agent_executor.py

@@ -216,11 +216,29 @@ async def execute(
         event_queue: EventQueue,
     ):
         # Run the agent until either complete or the task is suspended.
+        assert context.task_id is not None, (
+            'Task ID must be present for execution'
+        )
+        assert context.context_id is not None, (
+            'Context ID must be present for execution'
+        )
+
         updater = TaskUpdater(event_queue, context.task_id, context.context_id)
         # Immediately notify that the task is submitted.
         if not context.current_task:
             updater.submit()
         updater.start_work()
+
+        if context.message is None:
+            logger.warning('Execute called with no message in context')
+            updater.update_status(
+                TaskState.failed,
+                message=new_agent_text_message(
+                    'No message provided to the agent.'
+                ),
+            )
+            return
+
         await self._process_request(
             types.UserContent(
                 parts=convert_a2a_parts_to_genai(context.message.parts),