Retry context diagram endpoints (#1990)

cognitedata · Oct 26, 2024 · 9cb61f8 · 9cb61f8
1 parent b5213d4
commit 9cb61f8
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -17,6 +17,10 @@ Changes are grouped as follows
 - `Fixed` for any bug fixes.
 - `Security` in case of vulnerabilities.
 
+## [7.63.11] - 2024-10-26
+### Fixed
+- The `/context/diagram/` endpoints are now retried on 5xx and 429 errors.
+
 ## [7.63.10] - 2024-10-22
 ### Fixed
 - The Not() filter now only accepts a single filter (and no longer silently ignores the rest).

diff --git a/cognite/client/_api_client.py b/cognite/client/_api_client.py
@@ -103,6 +103,7 @@ class APIClient:
                     "workflows/.*",
                     "hostedextractors/.*",
                     "postgresgateway/.*",
+                    "context/diagram/.*",
                 )
             )
         ]

diff --git a/cognite/client/_version.py b/cognite/client/_version.py
@@ -1,4 +1,4 @@
 from __future__ import annotations
 
-__version__ = "7.63.10"
+__version__ = "7.63.11"
 __api_subversion__ = "20230101"
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "cognite-sdk"
 
-version = "7.63.10"
+version = "7.63.11"
 description = "Cognite Python SDK"
 readme = "README.md"
 documentation = "https://cognite-sdk-python.readthedocs-hosted.com"

diff --git a/tests/tests_unit/test_api_client.py b/tests/tests_unit/test_api_client.py
@@ -1361,6 +1361,11 @@ def test_is_retryable_resource_api_endpoints(self, api_client_with_token, method
                 ("POST", "https://api.cognitedata.com/api/v1/projects/bla/raw/dbs/db/tables/t", False),
                 ("POST", "https://api.cognitedata.com/api/v1/projects/bla/raw/dbs/db/tables/t/rows", True),
                 ("POST", "https://api.cognitedata.com/api/v1/projects/bla/raw/dbs/db/tables/t/rows/delete", True),
+                # Engineering diagrams
+                ("POST", "https://api.cognitedata.com/api/v1/projects/bla/context/diagram/convert", True),
+                ("POST", "https://api.cognitedata.com/api/v1/projects/bla/context/diagram/detect", True),
+                ("GET", "https://api.cognitedata.com/api/v1/projects/bla/context/diagram/convert/123", True),
+                ("GET", "https://api.cognitedata.com/api/v1/projects/bla/context/diagram/detect/456", True),
             ]
         ),
     )