fix: Fix the type annotation for content dictionaries in AdkApp

PiperOrigin-RevId: 756391356

Author: yeesian

vertexai/preview/reasoning_engines/templates/adk.py

@@ -17,13 +17,6 @@
 
 
 if TYPE_CHECKING:
- try:
- from google.genai import types
-
- ContentDict = types.Content
- except (ImportError, AttributeError):
- ContentDict = Dict
-
  try:
  from google.adk.events.event import Event
 
@@ -450,15 +443,15 @@ def set_up(self):
  def stream_query(
  self,
  *,
- message: Union[str, "ContentDict"],
+ message: Union[str, Dict[str, Any]],
  user_id: str,
  session_id: Optional[str] = None,
  **kwargs,
  ):
  """Streams responses from the ADK application in response to a message.
 
  Args:
- message (str):
+ message (Union[str, Dict[str, Any]]):
  Required. The message to stream responses for.
  user_id (str):
  Required. The ID of the user.