hostel-protocol-python 0.3.0__tar.gz → 0.3.2__tar.gz

{hostel_protocol_python-0.3.0 → hostel_protocol_python-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hostel-protocol-python
-Version: 0.3.0
+Version: 0.3.2
 Summary: Pydantic models, transport and client for the Hostel protocol
 Requires-Python: >=3.12
 Requires-Dist: hostel-protocol>=0.3.0

{hostel_protocol_python-0.3.0 → hostel_protocol_python-0.3.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "hostel-protocol-python"
-version = "0.3.0"
+version = "0.3.2"
 description = "Pydantic models, transport and client for the Hostel protocol"
 requires-python = ">=3.12"
 dependencies = [

{hostel_protocol_python-0.3.0 → hostel_protocol_python-0.3.2}/src/hostel/protocol/converter.py

@@ -98,6 +98,14 @@ _SUB_MSG_FIELDS: dict[type[Message], dict[str, type[Message]]] = {
     task_pb2.UpdateTaskResponse: {"task": task_pb2.TaskData},
 }
 
+# Enum fields: proto class -> {field_name: {int_value: str_value}}
+# Used to convert between proto integer enum values and Pydantic string literals.
+_ENUM_FIELDS: dict[type[Message], dict[str, dict[int, str]]] = {
+    chat_pb2.ChatRequest: {
+        "planning_mode": {0: "react", 1: "plan_and_execute"},
+    },
+}
+
 # oneof groups: proto class -> {field_name: proto sub-message class}
 _ONEOF_FIELDS: dict[type[Message], dict[str, type[Message]]] = {
     chat_pb2.ChatMessage: {
@@ -196,6 +204,7 @@ def _pydantic_to_proto_inner(model: BaseModel, proto_cls: type[Message]) -> Mess
     repeated_fields = _REPEATED_MSG_FIELDS.get(proto_cls, {})
     sub_msg_fields = _SUB_MSG_FIELDS.get(proto_cls, {})
     oneof_fields = _ONEOF_FIELDS.get(proto_cls, {})
+    enum_fields = _ENUM_FIELDS.get(proto_cls, {})
 
     # Special handling for HostelMessage payload oneof
     is_envelope = proto_cls is message_pb2.HostelMessage
@@ -252,6 +261,11 @@ def _pydantic_to_proto_inner(model: BaseModel, proto_cls: type[Message]) -> Mess
                 if child_pydantic_cls and isinstance(value, child_pydantic_cls):
                     getattr(proto, field_name).CopyFrom(_pydantic_to_proto_inner(value, child_proto_cls))
 
+        # Enum fields (string literal -> int)
+        elif field_name in enum_fields:
+            str_to_int = {v: k for k, v in enum_fields[field_name].items()}
+            setattr(proto, field_name, str_to_int[value])
+
         # Scalar fields
         else:
             setattr(proto, field_name, value)
@@ -280,6 +294,7 @@ def _proto_to_pydantic_inner(proto: Message, pydantic_cls: type[BaseModel]) -> B
     repeated_fields = _REPEATED_MSG_FIELDS.get(proto_cls, {})
     sub_msg_fields = _SUB_MSG_FIELDS.get(proto_cls, {})
     oneof_fields = _ONEOF_FIELDS.get(proto_cls, {})
+    enum_fields = _ENUM_FIELDS.get(proto_cls, {})
 
     is_envelope = proto_cls is message_pb2.HostelMessage
 
@@ -361,6 +376,10 @@ def _proto_to_pydantic_inner(proto: Message, pydantic_cls: type[BaseModel]) -> B
             else:
                 kwargs[field_name] = None
 
+        # Enum fields (int -> string literal)
+        elif field_name in enum_fields:
+            kwargs[field_name] = enum_fields[field_name][getattr(proto, field_name)]
+
         # Scalar fields
         else:
             kwargs[field_name] = getattr(proto, field_name)