Function client_label_context

src/google/adk/utils/_client_labels_utils.py:53–68

Runs the operation within the context of the given client label.

(client_label: str)

@contextmanager
def client_label_context(client_label: str) -> Iterator[None]:
  """Runs the operation within the context of the given client label."""
  current_client_label = _LABEL_CONTEXT.get()

  if current_client_label is not None:
    raise ValueError(
        "Client label already exists. You can only add one client label."
    )

  token = _LABEL_CONTEXT.set(client_label)

  try:
    yield
  finally:
    # Restore the previous state of the context variable
    _LABEL_CONTEXT.reset(token)


def get_client_labels() -> List[str]: