Function consume

tests/unittests/streaming/test_streaming.py:1397–1417

(session: testing_utils.Session)

  not_registered_before_call = None

  async def consume(session: testing_utils.Session):
    nonlocal not_registered_before_call
    async for response in runner.runner.run_live(
        session=session,
        live_request_queue=live_request_queue,
    ):
      collected.append(response)
      # On the first non-function-call event, verify the tool is not
      # yet registered (lazy registration).
      active = (
          captured_context.active_streaming_tools if captured_context else None
      )
      if (
          not_registered_before_call is None
          and not response.get_function_calls()
      ):
        not_registered_before_call = (
            active is None or "monitor_video_stream" not in active
        )
      if len(collected) >= 4:
        return

  runner._run_with_loop(asyncio.wait_for(consume(runner.session), timeout=5.0))