Class RequestContext

src/memos/context/context.py:25–88

Request-scoped context object that holds trace_id and other request data. This provides a Flask g-like object for FastAPI applications.

class RequestContext:
    """
    Request-scoped context object that holds trace_id and other request data.

    This provides a Flask g-like object for FastAPI applications.
    """

    def __init__(
        self,
        trace_id: str | None = None,
        api_path: str | None = None,
        env: str | None = None,
        user_type: str | None = None,
        user_name: str | None = None,
        source: str | None = None,
    ):
        self.trace_id = trace_id or "trace-id"
        self.api_path = api_path
        self.env = env
        self.user_type = user_type
        self.user_name = user_name
        self.source = source
        self._data: dict[str, Any] = {}

    def set(self, key: str, value: Any) -> None:
        """Set a value in the context."""
        self._data[key] = value

    def get(self, key: str, default: Any | None = None) -> Any:
        """Get a value from the context."""
        return self._data.get(key, default)

    def __setattr__(self, name: str, value: Any) -> None:
        if name.startswith("_") or name in (
            "trace_id",
            "api_path",
            "env",
            "user_type",
            "user_name",
            "source",
        ):
            super().__setattr__(name, value)
        else:
            if not hasattr(self, "_data"):
                super().__setattr__(name, value)
            else:
                self._data[name] = value

    def __getattr__(self, name: str) -> Any:
        if hasattr(self, "_data") and name in self._data:
            return self._data[name]
        raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{name}'")

    def to_dict(self) -> dict[str, Any]:
        """Convert context to dictionary."""
        return {
            "trace_id": self.trace_id,
            "api_path": self.api_path,