contree-client 0.1.3__tar.gz → 0.2.1.dev0__tar.gz

{contree_client-0.1.3 → contree_client-0.2.1.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: contree-client
-Version: 0.1.3
+Version: 0.2.1.dev0
 Summary: Python client for the Contree API, generated from the OpenAPI specification
 Author: Dmitry Orlov
 Author-email: Dmitry Orlov <mosquito@nebius.com>

{contree_client-0.1.3 → contree_client-0.2.1.dev0}/contree_client/__init__.py

@@ -38,6 +38,7 @@ from .exceptions import (
 from .models import (
     ACTIVE_STATUSES,
     TERMINAL_STATUSES,
+    ClosableStreamRepr,
     ContreeModel,
     DirectoryList,
     Error,
@@ -53,11 +54,15 @@ from .models import (
     EventDataStream,
     EventDataTruncated,
     EventResources,
+    ExecSpec,
     File,
     FileItem,
     FileResponse,
     FilesListResponse,
     FileSpec,
+    GrepMatch,
+    GrepResult,
+    GrepSubmatch,
     Image,
     ImageImportMetadata,
     ImageImportMetadataRegistry,
@@ -66,6 +71,7 @@ from .models import (
     ImageImportRegistryCredentials,
     ImageImportRequest,
     ImageListResponse,
+    InstanceNetworking,
     InstanceResourcesLimits,
     InstanceResult,
     InstanceResultResources,
@@ -106,6 +112,7 @@ __all__ = [
     "DEFAULT_BASE_URL",
     "TERMINAL_STATUSES",
     "BadRequestError",
+    "ClosableStreamRepr",
     "ConflictError",
     "ContreeAPIError",
     "ContreeError",
@@ -124,6 +131,7 @@ __all__ = [
     "EventDataStream",
     "EventDataTruncated",
     "EventResources",
+    "ExecSpec",
     "File",
     "FileItem",
     "FileResponse",
@@ -131,6 +139,9 @@ __all__ = [
     "FilesListResponse",
     "ForbiddenError",
     "GoneError",
+    "GrepMatch",
+    "GrepResult",
+    "GrepSubmatch",
     "Image",
     "ImageImportMetadata",
     "ImageImportMetadataRegistry",
@@ -139,6 +150,7 @@ __all__ = [
     "ImageImportRegistryCredentials",
     "ImageImportRequest",
     "ImageListResponse",
+    "InstanceNetworking",
     "InstanceResourcesLimits",
     "InstanceResult",
     "InstanceResultResources",

{contree_client-0.1.3 → contree_client-0.2.1.dev0}/contree_client/aiohttp.py

@@ -35,6 +35,7 @@ class ContreeAsyncClient(base.ContreeAsyncClient):
     retryable_errors = (
         aiohttp.ClientConnectionError,
         aiohttp.ServerConnectionError,
+        aiohttp.ClientPayloadError,
     )
     nonretryable_errors = (TimeoutError, asyncio.TimeoutError)
 
@@ -126,7 +127,12 @@ class ContreeAsyncClient(base.ContreeAsyncClient):
         ) as response:
             self.log.debug("%s %s -> %d (stream)", spec.method, url, response.status)
             if not 200 <= response.status < 300:
-                body = await response.read()
+                try:
+                    body = await response.read()
+                except self.retryable_errors:
+                    # The status is authoritative even when its optional
+                    # diagnostic body is interrupted.
+                    body = b""
                 # auto_decompress=False applies to the payload only:
                 # the error body must still be decoded, or the parsed
                 # server message is lost