Skip to content

lablink_client_service.http_utils

Shared HTTP utilities for LabLink client services.

Build Bearer-auth headers for the given client_secret token.

Source code in packages/client/src/lablink_client_service/http_utils.py 
22
23
24
def get_auth_headers(token: str) -> dict:
    """Build Bearer-auth headers for the given client_secret token."""
    return {"Authorization": f"Bearer {token}"}

Read common client environment variables with config fallback.

Returns:

Type Description
tuple

(base_url, client_secret, vm_name) tuple.

Raises:

Type Description
RuntimeError

If CLIENT_SECRET is not set.
Source code in packages/client/src/lablink_client_service/http_utils.py 
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
def get_client_env(cfg) -> tuple:
    """Read common client environment variables with config fallback.

    Returns:
        (base_url, client_secret, vm_name) tuple.

    Raises:
        RuntimeError: If CLIENT_SECRET is not set.
    """
    allocator_url = os.getenv("ALLOCATOR_URL")
    if allocator_url:
        base_url = sanitize_url(allocator_url)
    else:
        base_url = f"http://{cfg.allocator.host}:{cfg.allocator.port}"

    client_secret = os.environ.get("CLIENT_SECRET")
    if not client_secret:
        raise RuntimeError(
            "CLIENT_SECRET environment variable is required. "
            "Each client must be registered via /api/v1/clients/register "
            "(handled automatically by user_data.sh on AWS and "
            "`lablink register` on manual deployments)."
        )
    vm_name = os.getenv("VM_NAME")

    return base_url, client_secret, vm_name

Sanitize a URL by removing trailing slashes and stray dots.

Handles cases like
  • http://.lablink.sleap.ai -> http://lablink.sleap.ai
  • https://.lablink.sleap.ai -> https://lablink.sleap.ai
  • .lablink.sleap.ai -> lablink.sleap.ai
  • http://example.com/ -> http://example.com
Source code in packages/client/src/lablink_client_service/http_utils.py 
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
def sanitize_url(url: str) -> str:
    """Sanitize a URL by removing trailing slashes and stray dots.

    Handles cases like:
      - http://.lablink.sleap.ai -> http://lablink.sleap.ai
      - https://.lablink.sleap.ai -> https://lablink.sleap.ai
      - .lablink.sleap.ai -> lablink.sleap.ai
      - http://example.com/ -> http://example.com
    """
    url = url.rstrip("/")
    url = url.replace("://.", "://")
    if url.startswith("."):
        url = url[1:]
    return url