[fix] Language NLP

weni-ai · Nov 14, 2019 · 67f9af7 · 67f9af7
1 parent 06d4900
commit 67f9af7
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/bothub/api/v2/nlp/views.py b/bothub/api/v2/nlp/views.py
@@ -233,9 +233,13 @@ def retrieve(self, request, *args, **kwargs):
         check_auth(request)
         repository_authorization = self.get_object()
         repository = repository_authorization.repository
-        update = repository.last_trained_update(
-            str(request.query_params.get("language"))
-        )
+
+        language = request.query_params.get("language")
+
+        if language == "None" or language is None:
+            language = str(repository.language)
+
+        update = repository.last_trained_update(language)
         try:
             return Response(
                 {