Fixed http client

Author: VaghinakDev

src/superannotate/lib/infrastructure/services/http_client.py

@@ -229,39 +229,34 @@ class AIOHttpSession(aiohttp.ClientSession):
     RETRY_LIMIT = 3
     BACKOFF_FACTOR = 0.3
 
-    class AIOHttpSession(aiohttp.ClientSession):
-        RETRY_STATUS_CODES = [401, 403, 502, 503, 504]
-        RETRY_LIMIT = 3
-        BACKOFF_FACTOR = 0.3
-
-        @staticmethod
-        def _copy_form_data(data: aiohttp.FormData) -> aiohttp.FormData:
-            form_data = aiohttp.FormData(quote_fields=False)
-            for field in data._fields:  # noqa
-                if isinstance(field[2], io.IOBase):
-                    field[2].seek(0)
-                form_data.add_field(
-                    value=field[2],
-                    content_type=field[1].get("Content-Type", ""),
-                    **field[0],
-                )
-            return form_data
+    @staticmethod
+    def _copy_form_data(data: aiohttp.FormData) -> aiohttp.FormData:
+        form_data = aiohttp.FormData(quote_fields=False)
+        for field in data._fields:  # noqa
+            if isinstance(field[2], io.IOBase):
+                field[2].seek(0)
+            form_data.add_field(
+                value=field[2],
+                content_type=field[1].get("Content-Type", ""),
+                **field[0],
+            )
+        return form_data
 
-        async def request(self, *args, **kwargs) -> aiohttp.ClientResponse:
-            attempts = self.RETRY_LIMIT
-            delay = 0
-            for _ in range(attempts):
-                delay += self.BACKOFF_FACTOR
-                try:
-                    response = await super()._request(*args, **kwargs)
-                    if attempts <= 1 or response.status not in self.RETRY_STATUS_CODES:
-                        return response
-                except (aiohttp.ClientError, RuntimeError) as e:
-                    if attempts <= 1:
-                        raise
-                    if isinstance(e, RuntimeError):
-                        data = kwargs["data"]
-                        if isinstance(data, aiohttp.FormData):
-                            kwargs["data"] = self._copy_form_data(data)
-                attempts -= 1
-                await asyncio.sleep(delay)
+    async def request(self, *args, **kwargs) -> aiohttp.ClientResponse:
+        attempts = self.RETRY_LIMIT
+        delay = 0
+        for _ in range(attempts):
+            delay += self.BACKOFF_FACTOR
+            try:
+                response = await super()._request(*args, **kwargs)
+                if attempts <= 1 or response.status not in self.RETRY_STATUS_CODES:
+                    return response
+            except (aiohttp.ClientError, RuntimeError) as e:
+                if attempts <= 1:
+                    raise
+                if isinstance(e, RuntimeError):
+                    data = kwargs["data"]
+                    if isinstance(data, aiohttp.FormData):
+                        kwargs["data"] = self._copy_form_data(data)
+            attempts -= 1
+            await asyncio.sleep(delay)

tests/integration/projects/test_basic_project.py

@@ -15,11 +15,40 @@ class TestGenAIProjectBasic(BaseTestCase):
     PROJECT_NAME = "TestGenAICreate"
     PROJECT_TYPE = "GenAI"
     PROJECT_DESCRIPTION = "DESCRIPTION"
+    ANNOTATION_PATH = (
+        "data_set/sample_project_vector/example_image_1.jpg___objects.json"
+    )
+
+    @property
+    def annotation_path(self):
+        return os.path.join(Path(__file__).parent.parent.parent, self.ANNOTATION_PATH)
 
     def test_search(self):
         projects = sa.search_projects(self.PROJECT_NAME, return_metadata=True)
         assert projects
 
+        sa.create_annotation_class(
+            self.PROJECT_NAME,
+            "class1",
+            "#FFAAFF",
+            [
+                {
+                    "name": "Human",
+                    "attributes": [{"name": "yes"}, {"name": "no"}],
+                },
+                {
+                    "name": "age",
+                    "attributes": [{"name": "young"}, {"name": "old"}],
+                },
+            ],
+        )
+        sa.attach_items(self.PROJECT_NAME, attachments=[{"url": "", "name": "name"}])
+        annotation = json.load(open(self.annotation_path))
+        annotation["metadata"]["name"] = "name"
+        sa.upload_annotations(self.PROJECT_NAME, annotations=[annotation])
+        data = sa.get_annotations(self.PROJECT_NAME)
+        assert data
+
 
 class TestProjectBasic(BaseTestCase):
     PROJECT_NAME = "TestWorkflowGet"