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

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

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: hostel-protocol-python
-Version: 0.3.0
+Version: 0.4.0
 Summary: Pydantic models, transport and client for the Hostel protocol
 Requires-Python: >=3.12
-Requires-Dist: hostel-protocol>=0.3.0
+Requires-Dist: hostel-protocol>=0.4.0
 Requires-Dist: protobuf<8,>=7.35.0
 Requires-Dist: pydantic>=2.5.0
 Provides-Extra: client

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

@@ -4,11 +4,11 @@ build-backend = "hatchling.build"
 
 [project]
 name = "hostel-protocol-python"
-version = "0.3.0"
+version = "0.4.0"
 description = "Pydantic models, transport and client for the Hostel protocol"
 requires-python = ">=3.12"
 dependencies = [
-    "hostel-protocol>=0.3.0",
+    "hostel-protocol>=0.4.0",
     "pydantic>=2.5.0",
     "protobuf>=7.35.0,<8",
 ]

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

@@ -30,6 +30,15 @@ _PYDANTIC_TO_PROTO: dict[type[BaseModel], type[Message]] = {
     models.ChatMessage: chat_pb2.ChatMessage,
     models.ChatRequest: chat_pb2.ChatRequest,
     models.ChatResponse: chat_pb2.ChatResponse,
+    # Conversation
+    models.ConversationSummary: chat_pb2.ConversationSummary,
+    models.ConversationMessage: chat_pb2.ConversationMessage,
+    models.ListConversationsRequest: chat_pb2.ListConversationsRequest,
+    models.ListConversationsResponse: chat_pb2.ListConversationsResponse,
+    models.GetConversationRequest: chat_pb2.GetConversationRequest,
+    models.GetConversationResponse: chat_pb2.GetConversationResponse,
+    models.DeleteConversationRequest: chat_pb2.DeleteConversationRequest,
+    models.DeleteConversationResponse: chat_pb2.DeleteConversationResponse,
     # Agent
     models.ListAgentsRequest: agent_pb2.ListAgentsRequest,
     models.ListAgentsResponse: agent_pb2.ListAgentsResponse,
@@ -62,6 +71,14 @@ _PYDANTIC_TO_PROTO: dict[type[BaseModel], type[Message]] = {
 
 _PROTO_TO_PYDANTIC: dict[type[Message], type[BaseModel]] = {v: k for k, v in _PYDANTIC_TO_PROTO.items()}
 
+# Fields in each proto that are plain strings but should be treated as optional
+# (empty string on the wire maps to None in Pydantic)
+_OPTIONAL_STRING_FIELDS: dict[type[Message], set[str]] = {
+    chat_pb2.ChatRequest: {"conversation_id"},
+    chat_pb2.ChatResponse: {"conversation_id"},
+    chat_pb2.ConversationMessage: {"tool_name", "tool_call_id", "parameters_json"},
+}
+
 # Fields in each proto that use google.protobuf.StringValue (nullable strings)
 _STRING_VALUE_FIELDS: dict[type[Message], set[str]] = {
     task_pb2.TaskData: {"webhook_url", "status", "response", "created_at", "updated_at", "executed_at"},
@@ -87,17 +104,28 @@ _VALUE_FIELDS: dict[type[Message], set[str]] = {
 _REPEATED_MSG_FIELDS: dict[type[Message], dict[str, type[Message]]] = {
     agent_pb2.ListAgentsResponse: {"agents": struct_pb2.Struct},
     chat_pb2.ChatRequest: {"messages": chat_pb2.ChatMessage},
+    chat_pb2.ListConversationsResponse: {"conversations": chat_pb2.ConversationSummary},
+    chat_pb2.GetConversationResponse: {"messages": chat_pb2.ConversationMessage},
     component_pb2.ListComponentsResponse: {"components": struct_pb2.Struct},
     task_pb2.ListTasksResponse: {"tasks": task_pb2.TaskData},
 }
 
 # Fields that hold a single sub-message (non-oneof)
 _SUB_MSG_FIELDS: dict[type[Message], dict[str, type[Message]]] = {
+    chat_pb2.GetConversationResponse: {"conversation": chat_pb2.ConversationSummary},
     task_pb2.CreateTaskResponse: {"task": task_pb2.TaskData},
     task_pb2.GetTaskResponse: {"task": task_pb2.TaskData},
     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: {
@@ -114,6 +142,12 @@ _HOSTEL_MESSAGE_PAYLOAD_FIELDS: dict[str, type[Message]] = {
     "system_payload": struct_pb2.Struct,
     "chat_request": chat_pb2.ChatRequest,
     "chat_response_chunk": chat_pb2.ChatResponse,
+    "conversation_list_request": chat_pb2.ListConversationsRequest,
+    "conversation_list_response": chat_pb2.ListConversationsResponse,
+    "conversation_get_request": chat_pb2.GetConversationRequest,
+    "conversation_get_response": chat_pb2.GetConversationResponse,
+    "conversation_delete_request": chat_pb2.DeleteConversationRequest,
+    "conversation_delete_response": chat_pb2.DeleteConversationResponse,
     "task_create": task_pb2.CreateTaskRequest,
     "task_create_response": task_pb2.CreateTaskResponse,
     "task_list": task_pb2.ListTasksRequest,
@@ -191,11 +225,13 @@ def pydantic_to_proto(model: BaseModel) -> Message:
 def _pydantic_to_proto_inner(model: BaseModel, proto_cls: type[Message]) -> Message:
     proto = proto_cls()
     string_value_fields = _STRING_VALUE_FIELDS.get(proto_cls, set())
+    optional_string_fields = _OPTIONAL_STRING_FIELDS.get(proto_cls, set())
     struct_fields = _STRUCT_FIELDS.get(proto_cls, set())
     value_fields = _VALUE_FIELDS.get(proto_cls, set())
     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 +288,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)
@@ -275,11 +316,13 @@ def proto_to_pydantic(proto: Message) -> BaseModel:
 def _proto_to_pydantic_inner(proto: Message, pydantic_cls: type[BaseModel]) -> BaseModel:
     proto_cls = type(proto)
     string_value_fields = _STRING_VALUE_FIELDS.get(proto_cls, set())
+    optional_string_fields = _OPTIONAL_STRING_FIELDS.get(proto_cls, set())
     struct_fields = _STRUCT_FIELDS.get(proto_cls, set())
     value_fields = _VALUE_FIELDS.get(proto_cls, set())
     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,8 +404,16 @@ 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)
+            raw = getattr(proto, field_name)
+            if field_name in optional_string_fields:
+                kwargs[field_name] = raw if raw != "" else None
+            else:
+                kwargs[field_name] = raw
 
     return pydantic_cls(**kwargs)

{hostel_protocol_python-0.3.0 → hostel_protocol_python-0.4.0}/src/hostel/protocol/models.py

@@ -42,6 +42,7 @@ class ChatRequest(BaseModel):
     messages: list[ChatMessage] = []
     planning_mode: Literal["react", "plan_and_execute"] = "react"
     require_plan_approval: bool = True
+    conversation_id: str | None = None
 
 
 class ChatResponse(BaseModel):
@@ -49,6 +50,7 @@ class ChatResponse(BaseModel):
     role: str = ""
     scope: str = ""
     content: Any = None
+    conversation_id: str | None = None
 
 
 # ---------------------------------------------------------------------------
@@ -209,6 +211,62 @@ class DeleteTaskResponse(BaseModel):
     success: bool = False
 
 
+# ---------------------------------------------------------------------------
+# Conversation
+# ---------------------------------------------------------------------------
+
+
+class ConversationSummary(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+    id: str = ""
+    agent_name: str = ""
+    created_at: str = ""
+    updated_at: str = ""
+
+
+class ConversationMessage(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+    id: str = ""
+    conversation_id: str = ""
+    role: Literal["human", "ai", "tool_call", "tool_response"] = "human"
+    content: str = ""
+    tool_name: str | None = None
+    tool_call_id: str | None = None
+    parameters_json: str | None = None
+    sequence: int = 0
+    is_complete: bool = True
+    created_at: str = ""
+
+
+class ListConversationsRequest(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+
+
+class ListConversationsResponse(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+    conversations: list[ConversationSummary] = []
+
+
+class GetConversationRequest(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+    conversation_id: str
+
+
+class GetConversationResponse(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+    conversation: ConversationSummary | None = None
+    messages: list[ConversationMessage] = []
+
+
+class DeleteConversationRequest(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+    conversation_id: str
+
+
+class DeleteConversationResponse(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+
+
 # ---------------------------------------------------------------------------
 # Envelope  (HostelMessage)
 # ---------------------------------------------------------------------------
@@ -241,6 +299,14 @@ class HostelMessage(BaseModel):
     chat_request: ChatRequest | None = None
     chat_response_chunk: ChatResponse | None = None
 
+    # oneof payload – Conversation
+    conversation_list_request: ListConversationsRequest | None = None
+    conversation_list_response: ListConversationsResponse | None = None
+    conversation_get_request: GetConversationRequest | None = None
+    conversation_get_response: GetConversationResponse | None = None
+    conversation_delete_request: DeleteConversationRequest | None = None
+    conversation_delete_response: DeleteConversationResponse | None = None
+
     # oneof payload – Task
     task_create: CreateTaskRequest | None = None
     task_create_response: CreateTaskResponse | None = None