Function _create_invocation_context

tests/unittests/telemetry/test_spans.py:113–127

(
    agent: LlmAgent, state: Optional[dict[str, object]] = None
)

async def _create_invocation_context(
    agent: LlmAgent, state: Optional[dict[str, object]] = None
) -> InvocationContext:
  session_service = InMemorySessionService()
  session = await session_service.create_session(
      app_name='test_app', user_id='test_user', state=state
  )
  invocation_context = InvocationContext(
      invocation_id='test_id',
      agent=agent,
      session=session,
      session_service=session_service,
      run_config=RunConfig(),
  )
  return invocation_context


@pytest.mark.asyncio