Django Rest Framework - Jwt - Несколько типов пользователей

Я изучаю django rest framework и начал проект Linkedin clone. Я запутался, что использовать / как реализовать У меня есть несколько типов пользователей: Компания, Работник

class User(AbstractUser):
    class Types(models.TextChoices):
        WORKER = "WORKER", "Worker"
        COMPANY = "COMPANY", "Company"

    type = models.CharField("Type", max_length=50,
                            choices=Types.choices, default=Types.WORKER)


class CompanyManager(models.Manager):
    def get_queryset(self, *args, **kwargs):
        return super().get_queryset(*args, **kwargs).filter(type=User.Types.COMPANY)


class WorkerManager(models.Manager):
    def get_queryset(self, *args, **kwargs):
        return super().get_queryset(*args, **kwargs).filter(type=User.Types.WORKER)


class Worker(User):
    objects = WorkerManager()

    class Meta:
        proxy = True

    def save(self, *args, **kwargs):
        if not self.pk:
            self.type = User.Types.WORKER

        return super().save(*args, **kwargs)


class CompanyProfile(models.Model):
    user = models.OneToOneField(
        User, related_name="profile", on_delete=models.CASCADE)
    year = models.IntegerField()


class Company(User):
    objects = CompanyManager()

    @property
    def profile(self):
        return self.profile

    class Meta:
        proxy = True

    def save(self, *args, **kwargs):
        if not self.pk:
            self.type = User.Types.COMPANY

        return super().save(*args, **kwargs)

И я хочу использовать JWT У меня есть опыт работы с djoser Jwt и react и теперь я немного запутался, как сделать это здесь

Пс: Спасибо за ваше время!

Вернуться на верх