How to pass the requestContext during the agent invocation?

[raha0574]

When invoking the agent I want to pass the request context(some additional attributes) to agent so that I can access it from ToolContext and Agent prompts. When creating the session I can do this by passing an object to create_session method's state parameter like below.

 session = await session_service.create_session(
          app_name=APP_NAME,
          user_id=USER_ID,
          session_id=SESSION_ID,
          state= {"request_context": request_context}
      )

  runner = Runner(
        agent=stateful_agent,
        app_name=APP_NAME,
        session_service=session_service,
    )

  content = types.Content(role='user', parts=[types.Part(text=query)])

  final_response_text = "Agent did not produce a final response." # Default


  async for event in runner.run_async(user_id=user_id, session_id=session.id, new_message=content, ):

      if event.is_final_response():
          if event.content and event.content.parts:
             final_response_text = event.content.parts[0].text
          elif event.actions and event.actions.escalate:
             final_response_text = f"Agent escalated: {event.error_message or 'No specific message.'}"
          break

When resuming a session, I am modifying state object directly which was obtained from SessionService, before invoking the Agent like below

 session = await session_service.get_session(app_name=APP_NAME, user_id=USER_ID, session_id=session_id)
 session.state['request_context'] = request_context

But the documentation does not recommend directly updating the state object. https://google.github.io/adk-docs/sessions/state/#a-warning-about-direct-state-modification

So is there any other solution to set context to Agent than directly modifying state object when continuing a session?