fix: added serializer_class to views needed

core/apps/companies/views/companies.py

@@ -28,6 +28,7 @@ from core.apps.companies.serializers import (
 from core.apps.contracts.serializers import (
     RetrieveContractSerializer,
     BaseContractSerializer,
+    RetrieveContractQuerySerializer,
 )
 from core.apps.contracts.models import ContractModel
 
@@ -79,7 +80,8 @@ class CompanyContractApiView(BaseApiViewMixin, GenericAPIView):  # type: ignore
     #! TODO: status should be added.
     @extend_schema(
         summary="Company Contracts",
-        description="Get List Company Contracts"
+        description="Get List Company Contracts",
+        parameters=[RetrieveContractQuerySerializer]
     )
     def get(
         self,
@@ -88,9 +90,13 @@ class CompanyContractApiView(BaseApiViewMixin, GenericAPIView):  # type: ignore
         **kwargs: object,
     ) -> Response:
         company = self.get_object()
-        contracts = ContractModel.objects.filter(
-            owners__legal_entity__phone=company.phone,
-        ).distinct()
+        contracts = (
+            ContractModel.objects.filter(
+                owners__legal_entity__phone=company.phone,
+            )
+            .select_related("owners")
+            .distinct()
+        )
 
         folders_param = request.data.get("folders")
         if folders_param: