diff --git a/bothub/api/v2/repository/views.py b/bothub/api/v2/repository/views.py index a933ef6da..8a91247ab 100644 --- a/bothub/api/v2/repository/views.py +++ b/bothub/api/v2/repository/views.py @@ -6,6 +6,7 @@ from django.utils.decorators import method_decorator from django.utils.translation import ugettext_lazy as _ from django_filters.rest_framework import DjangoFilterBackend +from django.conf import settings from drf_yasg2 import openapi from drf_yasg2.utils import swagger_auto_schema from rest_framework import mixins, parsers, permissions, status @@ -900,6 +901,12 @@ class RepositoryNLPLogViewSet( filter_class = RepositoryNLPLogFilter filter_backends = [SearchFilter, DjangoFilterBackend] search_fields = ["$text", "^text", "=text"] + limit = settings.REPOSITORY_NLP_LOG_LIMIT + + def filter_queryset(self, queryset): + queryset = super().filter_queryset(queryset) + + return queryset[:self.limit] class RepositoryEntitiesViewSet( diff --git a/bothub/settings.py b/bothub/settings.py index 5c4d124b6..7b38bd17d 100644 --- a/bothub/settings.py +++ b/bothub/settings.py @@ -223,6 +223,7 @@ "DEFAULT_METADATA_CLASS": "bothub.api.v2.metadata.Metadata", } +REPOSITORY_NLP_LOG_LIMIT = env.int("REPOSITORY_NLP_LOG_LIMIT", default=10000) # cors headers