vibe-client 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

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/_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/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/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")

{vibe_client-0.1.0.dist-info → vibe_client-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
-Name: vibe-client
-Version: 0.1.0
+Name: vibe_client
+Version: 0.1.2
 Summary: vibe client
 License: Apache-2.0
 Keywords: smithy,vibe
@@ -33,7 +33,7 @@ A Python client for the Vibe API, enabling seamless interaction with Vibe agents
 ## Installation
 
 ```bash
-pip install vibe
+pip install vibe_client
 ```
 
 ## Features
@@ -47,8 +47,8 @@ pip install vibe
 
 ```python
 import asyncio
-from vibe.models import QueryAgentInput, ObservationValueBox
-from vibe.client import VibeClient
+from vibe_client.models import QueryAgentInput, ObservationValueBox
+from vibe_client.client import VibeClient
 
 async def main():
     # Initialize the client with your API key
@@ -96,7 +96,7 @@ The built documentation will be available in `docs/build/html/index.html`.
 
 ```
 vibe/
-├── src/vibe/            # Main package source code
+├── src/vibe_client/            # Main package source code
 │   ├── __init__.py      # Package initialization
 │   ├── client.py        # API client implementation
 │   ├── config.py        # Configuration utilities

vibe_client-0.1.2.dist-info/RECORD

@@ -0,0 +1,11 @@
+vibe_client/__init__.py,sha256=6skReHbZtASuUiEyPrSGv5sl9dFKQqPMen7Bc4VJSD4,83
+vibe_client/client.py,sha256=93hup4Jfqba5_NQNVCU4f5LxKq1Y5nOs0dGuP0T6Qnw,20538
+vibe_client/config.py,sha256=09qiTYTuJnv0krlJl5srgWncPWnGaMA9w0dDKswnNbI,2669
+vibe_client/deserialize.py,sha256=lm85B43gICgNAqAgYMNUbXyh8IeyqdDbwb7s2L8Xbas,4948
+vibe_client/models.py,sha256=SglS-Zu2hHRZ1OPJW6Zvf9_dMQit62-oW4v9Z6sctMM,27404
+vibe_client/serialize.py,sha256=O2ki-TARbjvwqztWYLCRZGMDzgzD15kI7uiOevjwQYY,1593
+vibe_client/_private/__init__.py,sha256=DxsJq42a0KfQv9zzLrqT0JVpoXrR-IDIYFGuJe4g1Gc,55
+vibe_client/_private/schemas.py,sha256=stUxYIBaNXkR3FZ8U-2HZosRebDjJiWH8xcq-xDpi0w,8260
+vibe_client-0.1.2.dist-info/METADATA,sha256=oPU2hfnwsCxusJg26N0n1AVg5OyI4mqplVicYe-EmCA,3597
+vibe_client-0.1.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+vibe_client-0.1.2.dist-info/RECORD,,

vibe_client-0.1.0.dist-info/RECORD

@@ -1,11 +0,0 @@
-vibe_client/__init__.py,sha256=XekyD00QFziYRdN_E7vlH9X7nnEV6NjGWRiQ0v0NcB8,83
-vibe_client/client.py,sha256=93hup4Jfqba5_NQNVCU4f5LxKq1Y5nOs0dGuP0T6Qnw,20538
-vibe_client/config.py,sha256=09qiTYTuJnv0krlJl5srgWncPWnGaMA9w0dDKswnNbI,2669
-vibe_client/deserialize.py,sha256=EXgPy-O6acFNL3Vs5qR1sJqmbTWE_J4CWE6DuwLWvTE,4065
-vibe_client/models.py,sha256=AAaUMCA0rYtteHI_HunrL-9FaKV_yNvVuYMON3fevDk,26070
-vibe_client/serialize.py,sha256=O2ki-TARbjvwqztWYLCRZGMDzgzD15kI7uiOevjwQYY,1593
-vibe_client/_private/__init__.py,sha256=DxsJq42a0KfQv9zzLrqT0JVpoXrR-IDIYFGuJe4g1Gc,55
-vibe_client/_private/schemas.py,sha256=EghHfaLbHS47726W3AgRHNPyYFaGWY0ELPw9SImbhfo,7785
-vibe_client-0.1.0.dist-info/METADATA,sha256=EZenhgcYREALvj_NXPCPDQsGTvadvF1-gaPzfayw53c,3569
-vibe_client-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-vibe_client-0.1.0.dist-info/RECORD,,