Method run

tests/integration/utils/test_runner.py:58–75

(self, prompt: str)

    ).id

  def run(self, prompt: str) -> list[Event]:
    current_session = self.session_service.get_session(
        app_name=self.app_name,
        user_id=self.user_id,
        session_id=self.current_session_id,
    )
    assert current_session is not None

    return list(
        self.agent_client.run(
            user_id=current_session.user_id,
            session_id=current_session.id,
            new_message=types.Content(
                role="user",
                parts=[types.Part.from_text(text=prompt)],
            ),
        )
    )

  def get_current_session(self) -> Optional[Session]:
    return self.session_service.get_session(