vibe-client 0.1.1__tar.gz → 0.1.2__tar.gz

{vibe_client-0.1.1 → vibe_client-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vibe_client
-Version: 0.1.1
+Version: 0.1.2
 Summary: vibe client
 License: Apache-2.0
 Keywords: smithy,vibe

{vibe_client-0.1.1 → vibe_client-0.1.2}/pyproject.toml

@@ -3,7 +3,7 @@
 
 [project]
 name = "vibe_client"
-version = "0.1.1"
+version = "0.1.2"
 description = "vibe client"
 readme = "README.md"
 requires-python = ">=3.12"

{vibe_client-0.1.1 → vibe_client-0.1.2}/src/vibe_client/__init__.py

@@ -1,3 +1,3 @@
 # Code generated by smithy-python-codegen DO NOT EDIT.
 
-__version__: str = "0.1.0"
+__version__: str = "0.1.2"

{vibe_client-0.1.1 → vibe_client-0.1.2}/src/vibe_client/_private/schemas.py

@@ -128,6 +128,27 @@ INTEGER_LIST = Schema.collection(
     }
 )
 
+FORBIDDEN_EXCEPTION = Schema.collection(
+    id=ShapeID("vibe.astar.public.api.shared#ForbiddenException"),
+
+    traits=[
+        Trait.new(id=ShapeID("smithy.api#error"), value="client"),
+        Trait.new(id=ShapeID("smithy.api#httpError"), value=403),
+
+    ],
+    members={
+        "message": {
+            "target": STRING,
+            "index": 0,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+    }
+)
+
 UNAUTHORIZED_EXCEPTION = Schema.collection(
     id=ShapeID("vibe.astar.public.api.shared#UnauthorizedException"),
 

{vibe_client-0.1.1 → vibe_client-0.1.2}/src/vibe_client/deserialize.py

@@ -12,6 +12,7 @@ from smithy_json import JSONCodec
 from .config import Config
 from .models import (
     ApiError,
+    ForbiddenException,
     QueryAgentOutput,
     UnauthorizedException,
     UnknownApiError,
@@ -48,6 +49,9 @@ async def _deserialize_error_query_agent(http_response: HTTPResponse, config: Co
         case "vibevalidationexception":
             return await _deserialize_error_vibe_validation_exception(http_response, config, parsed_body, message)
 
+        case "forbiddenexception":
+            return await _deserialize_error_forbidden_exception(http_response, config, parsed_body, message)
+
         case _:
             return UnknownApiError(f"{code}: {message}")
 
@@ -72,6 +76,27 @@ async def _deserialize_error_validation_exception(
 
     return ValidationException(**kwargs)
 
+async def _deserialize_error_forbidden_exception(
+    http_response: HTTPResponse,
+    config: Config,
+    parsed_body: dict[str, DocumentValue] | None,
+    default_message: str,
+) -> ForbiddenException:
+    kwargs: dict[str, Any] = {"message": default_message}
+
+    if parsed_body is None:
+        body = await http_response.consume_body_async()
+    else:
+        body = json.dumps(parsed_body).encode('utf-8')
+
+    if body:
+        codec = JSONCodec(default_timestamp_format=TimestampFormat.EPOCH_SECONDS)
+        deserializer = codec.create_deserializer(body)
+        body_kwargs = ForbiddenException.deserialize_kwargs(deserializer)
+        kwargs.update(body_kwargs)
+
+    return ForbiddenException(**kwargs)
+
 async def _deserialize_error_unauthorized_exception(
     http_response: HTTPResponse,
     config: Config,

{vibe_client-0.1.1 → vibe_client-0.1.2}/src/vibe_client/models.py

@@ -14,6 +14,7 @@ from smithy_core.shapes import ShapeID
 
 from ._private.schemas import (
     ACTION_VALUE as _SCHEMA_ACTION_VALUE,
+    FORBIDDEN_EXCEPTION as _SCHEMA_FORBIDDEN_EXCEPTION,
     OBSERVATION_VALUE as _SCHEMA_OBSERVATION_VALUE,
     QUERY_AGENT as _SCHEMA_QUERY_AGENT,
     QUERY_AGENT_INPUT as _SCHEMA_QUERY_AGENT_INPUT,
@@ -236,6 +237,39 @@ def _deserialize_integer_list(deserializer: ShapeDeserializer, schema: Schema) -
     deserializer.read_list(schema, _read_value)
     return result
 
+@dataclass(kw_only=True)
+class ForbiddenException(ApiError):
+
+    code: ClassVar[str] = "ForbiddenException"
+    fault: ClassVar[Literal["client", "server"]] = "client"
+
+    message: str
+
+    def serialize(self, serializer: ShapeSerializer):
+        serializer.write_struct(_SCHEMA_FORBIDDEN_EXCEPTION, self)
+
+    def serialize_members(self, serializer: ShapeSerializer):
+        serializer.write_string(_SCHEMA_FORBIDDEN_EXCEPTION.members["message"], self.message)
+
+    @classmethod
+    def deserialize(cls, deserializer: ShapeDeserializer) -> Self:
+        return cls(**cls.deserialize_kwargs(deserializer))
+
+    @classmethod
+    def deserialize_kwargs(cls, deserializer: ShapeDeserializer) -> dict[str, Any]:
+        kwargs: dict[str, Any] = {}
+
+        def _consumer(schema: Schema, de: ShapeDeserializer) -> None:
+            match schema.expect_member_index():
+                case 0:
+                    kwargs["message"] = de.read_string(_SCHEMA_FORBIDDEN_EXCEPTION.members["message"])
+
+                case _:
+                    logger.debug("Unexpected member schema: %s", schema)
+
+        deserializer.read_struct(_SCHEMA_FORBIDDEN_EXCEPTION, consumer=_consumer)
+        return kwargs
+
 @dataclass(kw_only=True)
 class UnauthorizedException(ApiError):
 
@@ -703,6 +737,7 @@ QUERY_AGENT = APIOperation(
             ShapeID("vibe.astar.public.api.shared#UnauthorizedException"): UnauthorizedException,
 ShapeID("smithy.framework#ValidationException"): ValidationException,
 ShapeID("vibe.astar.public.api.shared#VibeValidationException"): VibeValidationException,
+ShapeID("vibe.astar.public.api.shared#ForbiddenException"): ForbiddenException,
         }),
         effective_auth_schemes = [
             ShapeID("smithy.api#noAuth")