capo-sagemaker-runtime-http2 0.1.0__py3-none-any.whl

capo_sagemaker_runtime_http2/types/response_payload_part.py

@@ -0,0 +1,62 @@
+"""Generated from Smithy shape ``com.amazonaws.sagemakerruntimehttp2#ResponsePayloadPart``."""
+
+from typing import TYPE_CHECKING
+
+from typing_extensions import NotRequired, TypedDict
+
+from capo_sagemaker_runtime_http2._protocol.eventstream import HeaderValue, Message
+
+if TYPE_CHECKING:
+    import capo_sagemaker_runtime_http2.types.sensitive_blob
+
+
+class ResponsePayloadPart(TypedDict, closed=True):
+    bytes: NotRequired[
+        "capo_sagemaker_runtime_http2.types.sensitive_blob.SensitiveBlob"
+    ]
+    """<p>The payload bytes.</p>"""
+    data_type: NotRequired["str"]
+    r"""<p>Data type header. Can be one of these possible values: \"UTF8\", \"BINARY\".</p>"""
+    completion_state: NotRequired["str"]
+    r"""<p>Completion state header. Can be one of these possible values: \"PARTIAL\", \"COMPLETE\".</p>"""
+    p: NotRequired["str"]
+    """<p>Padding string for alignment.</p>"""
+
+
+# --- restJson1 ser/de ---
+def serialize_json(value: ResponsePayloadPart) -> dict:
+    out: dict = {}
+    return out
+
+
+def deserialize_json(data: dict) -> ResponsePayloadPart:
+    out: ResponsePayloadPart = {}  # type: ignore[typeddict-item]
+    return out
+
+
+def serialize_event_json(value: ResponsePayloadPart) -> bytes:
+    headers: dict[str, HeaderValue] = {":event-type": "PayloadPart"}
+    payload = b""
+    if "data_type" in value:
+        headers["DataType"] = value["data_type"]
+    if "completion_state" in value:
+        headers["CompletionState"] = value["completion_state"]
+    if "p" in value:
+        headers["P"] = value["p"]
+    payload = value["bytes"]
+    return Message(headers=headers, payload=payload).encode()
+
+
+def deserialize_event_json(message: Message) -> ResponsePayloadPart:
+    headers = message.headers  # noqa: F841
+    payload = message.payload  # noqa: F841
+    out: ResponsePayloadPart = {}  # type: ignore[typeddict-item]
+    if "DataType" in headers:
+        out["data_type"] = headers["DataType"]  # ty: ignore[invalid-assignment]
+    if "CompletionState" in headers:
+        out["completion_state"] = headers["CompletionState"]  # ty: ignore[invalid-assignment]
+    if "P" in headers:
+        out["p"] = headers["P"]  # ty: ignore[invalid-assignment]
+    if payload:
+        out["bytes"] = payload
+    return out

capo_sagemaker_runtime_http2/types/response_stream_event.py

@@ -0,0 +1,100 @@
+"""Generated from Smithy shape ``com.amazonaws.sagemakerruntimehttp2#ResponseStreamEvent``."""
+
+from typing import TYPE_CHECKING, TypeAlias
+
+from typing_extensions import TypedDict
+
+from capo_sagemaker_runtime_http2._iter import AnyIterator
+from capo_sagemaker_runtime_http2._protocol.eventstream import Message
+
+if TYPE_CHECKING:
+    import capo_sagemaker_runtime_http2.errors.internal_stream_failure
+    import capo_sagemaker_runtime_http2.errors.model_stream_error
+    import capo_sagemaker_runtime_http2.types.response_payload_part
+
+
+class _ResponseStreamEvent_PayloadPart(TypedDict, closed=True):
+    PayloadPart: (
+        "capo_sagemaker_runtime_http2.types.response_payload_part.ResponsePayloadPart"
+    )
+
+
+class _ResponseStreamEvent_ModelStreamError(TypedDict, closed=True):
+    ModelStreamError: (
+        "capo_sagemaker_runtime_http2.errors.model_stream_error.ModelStreamError_"
+    )
+
+
+class _ResponseStreamEvent_InternalStreamFailure(TypedDict, closed=True):
+    InternalStreamFailure: "capo_sagemaker_runtime_http2.errors.internal_stream_failure.InternalStreamFailure_"
+
+
+_ResponseStreamEvent: TypeAlias = (
+    _ResponseStreamEvent_PayloadPart
+    | _ResponseStreamEvent_ModelStreamError
+    | _ResponseStreamEvent_InternalStreamFailure
+)
+ResponseStreamEvent: TypeAlias = AnyIterator[_ResponseStreamEvent]
+
+
+def serialize_event_json(value: _ResponseStreamEvent) -> bytes:
+    match value:
+        case {"PayloadPart": payload}:
+            import capo_sagemaker_runtime_http2.types.response_payload_part
+
+            return capo_sagemaker_runtime_http2.types.response_payload_part.serialize_event_json(
+                payload
+            )
+        case {"ModelStreamError": payload}:
+            import capo_sagemaker_runtime_http2.errors.model_stream_error
+
+            return capo_sagemaker_runtime_http2.errors.model_stream_error.serialize_event_json(
+                payload
+            )
+        case {"InternalStreamFailure": payload}:
+            import capo_sagemaker_runtime_http2.errors.internal_stream_failure
+
+            return capo_sagemaker_runtime_http2.errors.internal_stream_failure.serialize_event_json(
+                payload
+            )
+        case _:
+            raise ValueError(f"ResponseStreamEvent: unrecognized variant {value!r}")
+
+
+def deserialize_event_json(message: Message) -> _ResponseStreamEvent:
+    headers = message.headers
+    message_type = headers.get(":message-type", "event")  # noqa: F841
+    if message_type == "error":
+        error_type = headers.get(":error-type")
+        match error_type:
+            case "ModelStreamError":
+                import capo_sagemaker_runtime_http2.errors.model_stream_error
+
+                raise capo_sagemaker_runtime_http2.errors.model_stream_error.ModelStreamError(
+                    capo_sagemaker_runtime_http2.errors.model_stream_error.deserialize_event_json(
+                        message
+                    )
+                )
+            case "InternalStreamFailure":
+                import capo_sagemaker_runtime_http2.errors.internal_stream_failure
+
+                raise capo_sagemaker_runtime_http2.errors.internal_stream_failure.InternalStreamFailure(
+                    capo_sagemaker_runtime_http2.errors.internal_stream_failure.deserialize_event_json(
+                        message
+                    )
+                )
+        raise ValueError(f"ResponseStreamEvent: unrecognized error-type {error_type!r}")
+    event_type = headers.get(":event-type")
+    match event_type:
+        case "PayloadPart":
+            import capo_sagemaker_runtime_http2.types.response_payload_part
+
+            return {
+                "PayloadPart": capo_sagemaker_runtime_http2.types.response_payload_part.deserialize_event_json(
+                    message
+                )
+            }
+        case _:
+            raise ValueError(
+                f"ResponseStreamEvent: unrecognized event-type {event_type!r}"
+            )

capo_sagemaker_runtime_http2/types/sensitive_blob.py

@@ -0,0 +1,15 @@
+"""Generated from Smithy shape ``com.amazonaws.sagemakerruntimehttp2#SensitiveBlob``."""
+
+import base64
+from typing import TypeAlias
+
+SensitiveBlob: TypeAlias = bytes
+
+
+# --- restJson1 ser/de ---
+def serialize_json(value: SensitiveBlob) -> str:
+    return base64.b64encode(value).decode("ascii")
+
+
+def deserialize_json(data: str) -> SensitiveBlob:
+    return base64.b64decode(data)

capo_sagemaker_runtime_http2-0.1.0.dist-info/METADATA

@@ -0,0 +1,108 @@
+Metadata-Version: 2.4
+Name: capo-sagemaker-runtime-http2
+Version: 0.1.0
+Summary: Python SDK for Sagemaker Runtime Http2.
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Topic :: Software Development :: Code Generators
+Classifier: Typing :: Typed
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: zapros>=0.13.0
+Provides-Extra: sso
+Requires-Dist: capo-sso>=0.2.0; extra == "sso"
+Requires-Dist: capo-sso-oidc>=0.2.0; extra == "sso"
+Requires-Dist: capo-sts>=0.3.0; extra == "sso"
+Dynamic: license-file
+
+# Getting Started
+
+## Installation
+
+```
+pip install capo-sagemaker-runtime-http2
+```
+
+## Usage
+
+```python
+from capo_sagemaker_runtime_http2 import AsyncSageMakerRuntimeHTTP2Client
+
+
+async def main():
+    async with AsyncSageMakerRuntimeHTTP2Client() as sage_maker_runtime_http2:
+        # Example: call the invoke_endpoint_with_bidirectional_stream operation
+        response = await sage_maker_runtime_http2.invoke_endpoint_with_bidirectional_stream()
+        print(response["body"])
+```
+
+## Streaming Request
+
+Some operations accept a streaming request body. Pass an async iterator of `bytes` chunks, or the whole body as `bytes`, for the streaming parameter.
+
+```python
+from capo_sagemaker_runtime_http2 import AsyncSageMakerRuntimeHTTP2Client
+
+
+async def main():
+    async with AsyncSageMakerRuntimeHTTP2Client() as sage_maker_runtime_http2:
+        # Example: call invoke_endpoint_with_bidirectional_stream with a streaming request body
+        async def chunks():
+            yield b'Hello, World!'
+
+        response = await sage_maker_runtime_http2.invoke_endpoint_with_bidirectional_stream(body=chunks())
+        print(response)
+
+        # Or pass the whole body as bytes
+        response = await sage_maker_runtime_http2.invoke_endpoint_with_bidirectional_stream(body=b'Hello, World!')
+        print(response)
+```
+
+## Error Handling
+
+The SDK raises exceptions for errors returned by the API. Catch them to handle failures gracefully.
+
+```python
+from capo_sagemaker_runtime_http2 import AsyncSageMakerRuntimeHTTP2Client
+from capo_sagemaker_runtime_http2.error import InputValidationError
+
+
+async def main():
+    async with AsyncSageMakerRuntimeHTTP2Client() as sage_maker_runtime_http2:
+        try:
+            await sage_maker_runtime_http2.invoke_endpoint_with_bidirectional_stream()
+        except InputValidationError as e:
+            print(f"Error: {e}")
+            print(e.data)  # additional error data
+```
+
+## Retrying
+
+The SDK retries failed operations automatically. Retry behaviour follows the Smithy specification: errors are retried based on their `is_retryable` and `is_throttling_error` attributes. Throttling errors use a longer base delay. Network-level failures (connection errors and timeouts) are also retried. Non-retryable errors, such as client errors without the `@retryable` trait, are raised immediately without further attempts.
+
+The number of attempts defaults to 3 and can be changed at the client level via `retry_max_attempts`, or per call via `config_overrides`.
+
+```python
+from capo_sagemaker_runtime_http2 import AsyncSageMakerRuntimeHTTP2Client
+
+
+async def main():
+    async with AsyncSageMakerRuntimeHTTP2Client() as sage_maker_runtime_http2:
+        # Default: 3 attempts for every operation
+        response = await sage_maker_runtime_http2.invoke_endpoint_with_bidirectional_stream()
+
+        # Override per operation
+        response = await sage_maker_runtime_http2.invoke_endpoint_with_bidirectional_stream(config_overrides={"retry_max_attempts": 5})
+
+        # Disable retries for this call
+        response = await sage_maker_runtime_http2.invoke_endpoint_with_bidirectional_stream(config_overrides={"retry_max_attempts": 1})
+```

capo_sagemaker_runtime_http2-0.1.0.dist-info/RECORD

@@ -0,0 +1,46 @@
+capo_sagemaker_runtime_http2/__init__.py,sha256=ZLlw1X7Fy-SXnaQKtOM5F5JhPHV-PaNussB87763N6A,2244
+capo_sagemaker_runtime_http2/_async.py,sha256=h6ySvzTgw6Vz53pgCi0YvxHurRgLsjGXXfskGimI4AY,451
+capo_sagemaker_runtime_http2/_iter.py,sha256=T_s-wDvQwcKN5JpnWCBFyyXXGsbAwQisFSSK5RPKDTA,3165
+capo_sagemaker_runtime_http2/_pagination.py,sha256=EihZfcSTQN5NRCcdJ26KcpYTUGzctwP6Iz9jYBFntWg,698
+capo_sagemaker_runtime_http2/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+capo_sagemaker_runtime_http2/_auth/_identity.py,sha256=ONv_NxU5ShEezSRyGAKGaxAn0R1HLGktZqKrL7xsPZw,326
+capo_sagemaker_runtime_http2/_auth/_providers.py,sha256=w9zspPdImkujmdnyeAQFm6uUhBJiLYhem91xlFN_s4k,33732
+capo_sagemaker_runtime_http2/_auth/_signers.py,sha256=-7Wzzwm0ybSAWryhwg-rMEO7wWb6KSR8bHCS91Ym7Ps,3237
+capo_sagemaker_runtime_http2/_auth/_sigv4.py,sha256=YfqyANVhhQOVahg3VjKTtiBzR8bkSGxY6415MkS9g0A,14885
+capo_sagemaker_runtime_http2/_auth/_zapros_handler.py,sha256=8xdfKJH-rJB2fFNajj7v5JlSn06u9ZUoOmgqUh2jptA,2978
+capo_sagemaker_runtime_http2/_operations/amazon_sage_maker_runtime_http2/invoke_endpoint_with_bidirectional_stream.py,sha256=iL61chXqFVkFk5t-Z2NOqSlehuW0yFCskaNGlBgM7WE,11068
+capo_sagemaker_runtime_http2/_protocol/__init__.py,sha256=MSIVV1jVmxqtMyjeNhffomlwpuVB7-tzRK4rdPDFCG8,61
+capo_sagemaker_runtime_http2/_protocol/errors.py,sha256=LGBCYv4hreZJWCslL1jZMzcsy9_38EYIYVRyeyrPtg8,2914
+capo_sagemaker_runtime_http2/_protocol/eventstream.py,sha256=nlzm-kxm1tAfwyBWYmKChX0jYOht3R4aodi6n92NXqQ,8741
+capo_sagemaker_runtime_http2/_protocol/serialize.py,sha256=9yv7gMXl03SdUzfWKUvXs-jr9uZmWDLXBM5J44azkUs,1348
+capo_sagemaker_runtime_http2/_protocol/xml.py,sha256=D4VZOoKjh1hoV1HKUVoA6Hg7aKKfUNenyRKTSTjCQFs,1062
+capo_sagemaker_runtime_http2/_rule_engine/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+capo_sagemaker_runtime_http2/_rule_engine/_aws_partition.py,sha256=gxziWLrEyQKXSmsQ-skA8DdYWrHXAURuxBDhZPqMNb8,5555
+capo_sagemaker_runtime_http2/_rule_engine/_endpoint_rule_set.py,sha256=MiwbgXP7GG5uJsdZYWeOepliLrJ5T6TJquWkbBBFfMY,24460
+capo_sagemaker_runtime_http2/_rule_engine/_endpoint_runtime.py,sha256=o94904z9c3fuJJD7vDrl_k1b9nWLueMKAXB6-OxL1Fs,13797
+capo_sagemaker_runtime_http2/_services/_aws_config.py,sha256=mF00YpQGVmKl6gn4YdJTnOMzhjZ_--omqp2tcG4Ln8E,5241
+capo_sagemaker_runtime_http2/_services/_pipeline.py,sha256=4wdGnsls95xdROK1EgI9XsDLhc1Z89VdaDUQNvmnOSQ,5622
+capo_sagemaker_runtime_http2/_services/async_sage_maker_runtime_http2.py,sha256=XEvJiu4xHuxNwrwXjak_vpYaeu73uZM6in5oN-k_UBI,11760
+capo_sagemaker_runtime_http2/_services/sage_maker_runtime_http2.py,sha256=y_3_qe1YkqaTP-c5zQvZiGtfkdjETHTzyaTeq7n2xJE,11525
+capo_sagemaker_runtime_http2/errors/__init__.py,sha256=lGR8_hA0iN-XxP4FJXoDtlUB7uutyNhmBPAheZkrBg8,988
+capo_sagemaker_runtime_http2/errors/_base.py,sha256=QnlqMfl_bTTLx6DVZmKve7WAPBx2FzVtnF-Jgt-0rbc,3111
+capo_sagemaker_runtime_http2/errors/input_validation_error.py,sha256=hCc-b9dnabsKTMRTGZ7xW_oDQxvm2j4CW8hhBOew98A,1649
+capo_sagemaker_runtime_http2/errors/internal_server_error.py,sha256=J8spTF3USMZ7ajDwKyjZk5kFUA6uF-A4RL0C7mHxbeU,1624
+capo_sagemaker_runtime_http2/errors/internal_stream_failure.py,sha256=WEceFrEz7VxOrits1H5pDG35ADo9BV8-ajRIWJfPECY,1988
+capo_sagemaker_runtime_http2/errors/model_error.py,sha256=JInB7voZfK5YUeMMXvshRKWZxlwr97JWcRlxCM3ZEV8,2412
+capo_sagemaker_runtime_http2/errors/model_stream_error.py,sha256=WJ3zuHGlR5EHDLjZpEFNUMrOyvFb6waQSFEfFphRsTo,2128
+capo_sagemaker_runtime_http2/errors/service_unavailable_error.py,sha256=imqBPPgoE2gcYgT16kaF-YWCEnmxuvPFm1IcMvY9wL0,1682
+capo_sagemaker_runtime_http2/types/invoke_endpoint_with_bidirectional_stream_input.py,sha256=kkh0Ys88aqJU3Hhq6QHZizx39OrknzcmfTwFM0mJXgs,832
+capo_sagemaker_runtime_http2/types/invoke_endpoint_with_bidirectional_stream_output.py,sha256=BTuBpeokHNRxj4PhkOTU_EV1jvi6lsxTl6EnnLh8N1E,606
+capo_sagemaker_runtime_http2/types/request_payload_part.py,sha256=sI1h5csv4UmAABUL2NzChn0dynOzIUYWk_fxw8bwRFk,2236
+capo_sagemaker_runtime_http2/types/request_stream_event.py,sha256=lnP_drbRIuQftgRnWO6XBz4myQetaScywJW4dwEShcw,1808
+capo_sagemaker_runtime_http2/types/response_payload_part.py,sha256=bJEwoy_MjeQU9HUfXf9oWMgHftlUkYXyKpk3ctZNYNU,2244
+capo_sagemaker_runtime_http2/types/response_stream_event.py,sha256=a-lwWlCvesjE1SOL1ZqolKxaWC27DzQ7Eps5fSPCSMA,3940
+capo_sagemaker_runtime_http2/types/sensitive_blob.py,sha256=oLzNdlWQkMwYBPIDeQHCBbw9uoJQNMZkmDIDr9vY0rQ,382
+capo_sagemaker_runtime_http2/types/_prelude/blob.py,sha256=L57RpNwdTSboMhnddA-0PBonXBu3oPm3RAr4Ff7IjzQ,277
+capo_sagemaker_runtime_http2/types/_prelude/timestamp.py,sha256=wfGPIPU-oBc2-PPqjNnJIElt1jXcEdb78Ux6jSX1mu0,487
+capo_sagemaker_runtime_http2-0.1.0.dist-info/licenses/LICENSE,sha256=sOQjPuig7ktMdOZpG1TdsPSYAVS_9UYVXwDDbD75nVI,1072
+capo_sagemaker_runtime_http2-0.1.0.dist-info/METADATA,sha256=uKynLXcMsBy_coh7VGhNHjUl1igNIkrgEjeHBVarva4,4188
+capo_sagemaker_runtime_http2-0.1.0.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+capo_sagemaker_runtime_http2-0.1.0.dist-info/top_level.txt,sha256=8ZKmp4kccFYj5wemxq_CjG3CbCApXIZ2gDWKgHozhSQ,29
+capo_sagemaker_runtime_http2-0.1.0.dist-info/RECORD,,

capo_sagemaker_runtime_http2-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (84.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

capo_sagemaker_runtime_http2-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Karen Petrosyan
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

capo_sagemaker_runtime_http2-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+capo_sagemaker_runtime_http2