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

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

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: hostel-protocol-python
-Version: 0.3.2
+Version: 0.4.1
 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.2 → hostel_protocol_python-0.4.1}/pyproject.toml

@@ -4,11 +4,11 @@ build-backend = "hatchling.build"
 
 [project]
 name = "hostel-protocol-python"
-version = "0.3.2"
+version = "0.4.1"
 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.2 → hostel_protocol_python-0.4.1}/src/hostel/protocol/__init__.py

@@ -10,16 +10,22 @@ from hostel.protocol.models import (
     ChatMessage,
     ChatRequest,
     ChatResponse,
+    ConversationMessage,
+    ConversationSummary,
     CreateComponentRequest,
     CreateComponentResponse,
     CreateTaskRequest,
     CreateTaskResponse,
     DeleteComponentRequest,
     DeleteComponentResponse,
+    DeleteConversationRequest,
+    DeleteConversationResponse,
     DeleteTaskRequest,
     DeleteTaskResponse,
     GetComponentRequest,
     GetComponentResponse,
+    GetConversationRequest,
+    GetConversationResponse,
     GetTaskRequest,
     GetTaskResponse,
     HostelMessage,
@@ -28,6 +34,8 @@ from hostel.protocol.models import (
     ListAgentsResponse,
     ListComponentsRequest,
     ListComponentsResponse,
+    ListConversationsRequest,
+    ListConversationsResponse,
     ListTasksRequest,
     ListTasksResponse,
     TaskData,
@@ -49,6 +57,15 @@ __all__ = [
     "ChatMessage",
     "ChatRequest",
     "ChatResponse",
+    # Conversation
+    "ConversationSummary",
+    "ConversationMessage",
+    "ListConversationsRequest",
+    "ListConversationsResponse",
+    "GetConversationRequest",
+    "GetConversationResponse",
+    "DeleteConversationRequest",
+    "DeleteConversationResponse",
     # Agent
     "ListAgentsRequest",
     "ListAgentsResponse",

{hostel_protocol_python-0.3.2 → hostel_protocol_python-0.4.1}/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,12 +104,15 @@ _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},
@@ -122,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,
@@ -199,6 +225,7 @@ 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, {})
@@ -289,6 +316,7 @@ 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, {})
@@ -382,6 +410,10 @@ def _proto_to_pydantic_inner(proto: Message, pydantic_cls: type[BaseModel]) -> B
 
         # 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.2 → hostel_protocol_python-0.4.1}/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