Function _create_local_access_token

skyvern/cli/mcp.py:31–47

(subject: str, expires_delta: timedelta)

def _create_local_access_token(subject: str, expires_delta: timedelta) -> str:
    from skyvern.config import settings  # noqa: PLC0415

    if settings.SIGNATURE_ALGORITHM != "HS256":
        raise RuntimeError("Local quickstart token bootstrap only supports HS256 SIGNATURE_ALGORITHM")

    expire = datetime.utcnow() + expires_delta
    header = {"alg": "HS256", "typ": "JWT"}
    payload = {"exp": int(expire.timestamp()), "sub": str(subject)}
    signing_input = ".".join(
        [
            _base64url_encode(json.dumps(header, separators=(",", ":")).encode("utf-8")),
            _base64url_encode(json.dumps(payload, separators=(",", ":")).encode("utf-8")),
        ]
    )
    signature = hmac.new(settings.SECRET_KEY.encode("utf-8"), signing_input.encode("ascii"), hashlib.sha256).digest()
    return f"{signing_input}.{_base64url_encode(signature)}"


async def get_or_create_local_organization(database: Any | None = None) -> Organization: