Merge 71bedda into 0d1e7a8

weni-ai · May 13, 2021 · 1fdb69a · 1fdb69a
2 parents 0d1e7a8 + 71bedda
commit 1fdb69a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git 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
@@ -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