Class JWTPayload

app/api/agentops/api/auth.py:27–68

Dataclass to represent the payload of a JWT token. This is used for type checking and validation of the JWT payload.

@dataclass
class JWTPayload:
    """
    Dataclass to represent the payload of a JWT token.
    This is used for type checking and validation of the JWT payload.
    """

    exp: int
    aud: str
    project_id: str
    project_prem_status: str
    api_key: str
    dev: bool = False

    def asdict(self) -> dict:
        """Convert the payload to a dictionary format."""
        properties = {
            "exp": self.exp,
            "aud": self.aud,
            "project_id": self.project_id,
            "project_prem_status": self.project_prem_status,
            "api_key": self.api_key,
        }

        if self.dev:
            properties["dev"] = self.dev

        return properties

    @classmethod
    def from_project(cls, project: ProjectModel, dev: bool = False) -> "JWTPayload":
        """
        Create a new instance of JWTPayload with the given project_id and role.
        The expiration time is set to 30 days from now.
        """
        return cls(
            exp=_generate_jwt_timestamp(),
            aud="authenticated",
            project_id=str(project.id),
            project_prem_status=project.org.prem_status.value,
            api_key=str(project.api_key),
            dev=dev,
        )


def generate_jwt(project: ProjectModel) -> str: