diff --git a/int_tests/managers/test_api_keys.py b/int_tests/managers/test_api_keys.py index 3f76d2f..f9a0d5e 100644 --- a/int_tests/managers/test_api_keys.py +++ b/int_tests/managers/test_api_keys.py @@ -27,6 +27,8 @@ def setUpClass(cls): # Create corpus response = cls.client.corpora.create(name=cls.corpus_name, key=cls.corpus_key) + # Small wait for corpus to be fully provisioned + time.sleep(2) cls.key = response.key def _create_api_key(self, name="test-key", api_key_role="serving"): diff --git a/int_tests/managers/test_document.py b/int_tests/managers/test_document.py index 649f95d..acbe2fd 100644 --- a/int_tests/managers/test_document.py +++ b/int_tests/managers/test_document.py @@ -25,6 +25,8 @@ def setUpClass(cls): timestamp = int(time.time()) corpus_name = f"test-document-manager-{timestamp}" response = cls.client.corpora.create(name=corpus_name, key=corpus_name) + # Small wait for corpus to be fully provisioned + time.sleep(2) cls.corpus_key = response.key cls.created_corpora.add(cls.corpus_key) diff --git a/int_tests/managers/test_upload.py b/int_tests/managers/test_upload.py index 7a7eaf6..2c9d8ea 100644 --- a/int_tests/managers/test_upload.py +++ b/int_tests/managers/test_upload.py @@ -86,6 +86,8 @@ def setUpClass(cls): timestamp = int(time.time()) corpus_name = f"test-upload-{timestamp}" response = cls.client.corpora.create(key=corpus_name, name=corpus_name) + # Small wait for corpus to be fully provisioned + time.sleep(2) cls.corpus = response def _get_test_file(self): diff --git a/int_tests/test_chat.py b/int_tests/test_chat.py index ae5eb44..90909da 100644 --- a/int_tests/test_chat.py +++ b/int_tests/test_chat.py @@ -42,6 +42,8 @@ def setUpClass(cls): cls.corpus_name = f"test-chat-corpus-{timestamp}" cls.client.corpora.create(name=cls.corpus_name, key=cls.corpus_name) + # Small wait for corpus to be fully provisioned + time.sleep(2) cls.client.documents.create(cls.corpus_name, request=cls.TEST_DOCUMENT) # Create default search parameters diff --git a/int_tests/test_query.py b/int_tests/test_query.py index 20e8fe1..6da04ca 100644 --- a/int_tests/test_query.py +++ b/int_tests/test_query.py @@ -56,6 +56,8 @@ def setUpClass(cls): for corpus_name, document in zip(cls.corpus_names, cls.test_documents): cls.client.corpora.create(name=corpus_name, key=corpus_name) + # Small wait for corpus to be fully provisioned + time.sleep(2) cls.client.documents.create(corpus_name, request=document) def setUp(self):