Merge pull request #22 from bothub-it/feature/repository_info_update

Update repository_info endpoint
weni-ai · May 19, 2021 · 3fa4d7a · 3fa4d7a
2 parents c606d93 + ded7c5b
commit 3fa4d7a
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 7 deletions.
diff --git a/bothub_backend/bothub.py b/bothub_backend/bothub.py
@@ -81,10 +81,13 @@ def request_backend_evaluate(
 
     @print_execution_time
     def request_backend_info(
-        self, repository_authorization, language=None, repository_version=None
+        self, repository_authorization, repository_version_id=None, repository_version_language_id=None
     ):
         url = f"{self.backend}/v2/repository/nlp/authorization/info/{repository_authorization}/"
-        query_params = {"language": language, "repository_version": repository_version}
+        query_params = {
+            "repository_version": repository_version_id,
+            "repository_version_language": repository_version_language_id
+        }
         headers = {"Authorization": f"Bearer {repository_authorization}"}
         response = requests.get(url, params=query_params, headers=headers).json()
 
@@ -100,7 +103,7 @@ def request_backend_train(
         response = requests.get(url, params=query_params, headers=headers).json()
 
         return response
-      
+
     @print_execution_time
     def request_all_readytotrain_languages(
         self, repository_authorization, repository_version=None

diff --git a/bothub_backend/tests.py b/bothub_backend/tests.py
@@ -178,16 +178,15 @@ def test_request_backend_evaluate(self, request_mock):
     @requests_mock.Mocker()
     def test_request_backend_info(self, request_mock):
         query_params = (
-            f"?language={self.language}&repository_version={self.repository_version}"
+            f"?repository_version={self.repository_version}"
         )
         url = f"{BOTHUB_API_REPOSITORY_NLP_URL}/authorization/info/{self.repository_authorization}/{query_params}"
         json = {"intents": ["comprar", "alugar"]}
         request_mock.get(url=url, json=json)
 
         response = self.bh.request_backend_info(
             repository_authorization=self.repository_authorization,
-            language=self.language,
-            repository_version=self.repository_version,
+            repository_version_id=self.repository_version,
         )
 
         self.assertEqual(response, json)

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 
 setup(
     name='bothub_backend',
-    version='1.0.19',
+    version='1.0.21',
     description='Bothub NLP Backend',
     packages=find_packages(),
     install_requires=[