youtube-summarizer/venv311/lib/python3.11/site-packages/fastmcp/client/auth/bearer.py

18 lines
399 B
Python

import httpx
from pydantic import SecretStr
from fastmcp.utilities.logging import get_logger
__all__ = ["BearerAuth"]
logger = get_logger(__name__)
class BearerAuth(httpx.Auth):
def __init__(self, token: str):
self.token = SecretStr(token)
def auth_flow(self, request):
request.headers["Authorization"] = f"Bearer {self.token.get_secret_value()}"
yield request