h-message-bus 0.0.16__tar.gz → 0.0.18__tar.gz

{h_message_bus-0.0.16 → h_message_bus-0.0.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: h_message_bus
-Version: 0.0.16
+Version: 0.0.18
 Summary: Message bus integration for HAI
 Author-email: shoebill <shoebill.hai@gmail.com>
 Classifier: Programming Language :: Python :: 3

{h_message_bus-0.0.16 → h_message_bus-0.0.18}/pyproject.toml

@@ -4,7 +4,7 @@
 
    [project]
    name = "h_message_bus"
-   version = "0.0.16"
+   version = "0.0.18"
    authors = [
        {name = "shoebill", email = "shoebill.hai@gmail.com"},
    ]

{h_message_bus-0.0.16 → h_message_bus-0.0.18}/src/h_message_bus/domain/models/request_message_topic.py

@@ -14,9 +14,9 @@ class RequestMessageTopic(str, Enum):
 
     """
     # Telegram
-    AI_SEND_TG_CHAT_SEND = "hai.ai.tg.chat.send"
-    TG_USER_SEND_AI_CHAT_SEND = "hai.tg.user.ai.chat.send"
-    TG_USER_SEND_AI_CHAT_SEND_RESPONSE = "hai.tg.user.ai.chat.send.response"
+    AI_TG_CHAT_SEND = "hai.ai.tg.chat.send"
+    TG_AI_USER_CHAT_SEND = "hai.tg.ai.user.chat.send"
+    AI_TG_CHAT_REPLY = "hai.ai.tg.chat.reply"
 
     # vector database
     AI_VECTORS_SAVE = "hai.ai.vectors.collection.save"

h_message_bus-0.0.18/src/h_message_bus/domain/request_messages/tg_user_message_reply_message.py

@@ -0,0 +1,56 @@
+from typing import Type, TypeVar, Dict, Any, Optional
+
+from ..models.request_message_topic import RequestMessageTopic
+from ...domain.models.hai_message import HaiMessage
+
+T = TypeVar('T', bound='HaiMessage')
+
+
+class TelegramUserMessageReplyRequestMessage(HaiMessage):
+    """Message containing user message data"""
+
+    @classmethod
+    def create(cls: Type[T], topic: str, payload: Dict[Any, Any]) -> T:
+        """Create a message - inherited from HaiMessage"""
+        return super().create(topic=topic, payload=payload)
+
+    @classmethod
+    def create_message(cls, message: str, chat_id: int,
+                       message_id: int) -> 'TelegramUserMessageReplyRequestMessage':
+
+        payload = {
+            "message": message,
+            "chat_id": chat_id,
+            "message_id": message_id,
+        }
+
+        return cls.create(
+            topic=RequestMessageTopic.AI_TG_CHAT_REPLY,
+            payload=payload
+        )
+
+    @property
+    def message(self) -> str:
+        """Get the message content from the payload"""
+        return self.payload.get("message", "")
+
+    @property
+    def chat_id(self) -> int:
+        """Get the Telegram chat ID from the payload"""
+        return self.payload.get("chat_id", 0)
+
+    @property
+    def message_id(self) -> int:
+        """Get the Telegram message ID from the payload"""
+        return self.payload.get("message_id", 0)
+
+    @classmethod
+    def from_hai_message(cls, message: HaiMessage) -> 'TelegramUserMessageReplyRequestMessage':
+        payload = message.payload
+
+        return cls.create_message(
+            message=payload.get("message", ""),
+            chat_id=payload.get("chat_id"),
+            message_id=payload.get("message_id")
+        )
+

{h_message_bus-0.0.16 → h_message_bus-0.0.18}/src/h_message_bus/domain/request_messages/tg_user_message_request_message.py

@@ -50,7 +50,7 @@ class TelegramUserMessageRequestMessage(HaiMessage):
             payload["message_id"] = message_id
 
         return cls.create(
-            topic=RequestMessageTopic.TG_USER_SEND_AI_CHAT_SEND,
+            topic=RequestMessageTopic.TG_AI_USER_CHAT_SEND,
             payload=payload
         )
 
@@ -91,7 +91,6 @@ class TelegramUserMessageRequestMessage(HaiMessage):
 
     @classmethod
     def from_hai_message(cls, message: HaiMessage) -> 'TelegramUserMessageRequestMessage':
-        """Create a TelegramUserMessageRequestMessage from a generic HaiMessage"""
         payload = message.payload
 
         return cls.create_message(

{h_message_bus-0.0.16 → h_message_bus-0.0.18}/src/h_message_bus/domain/request_messages/vector_query_collection_response_message.py

@@ -1,5 +1,5 @@
 import json
-from typing import Type, TypeVar, Dict, Any, List
+from typing import Type, TypeVar, Dict, Any
 
 from ..models.request_message_topic import RequestMessageTopic
 from ..models.vector_query_answer import VectorQueryAnswer

{h_message_bus-0.0.16 → h_message_bus-0.0.18}/src/h_message_bus.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: h_message_bus
-Version: 0.0.16
+Version: 0.0.18
 Summary: Message bus integration for HAI
 Author-email: shoebill <shoebill.hai@gmail.com>
 Classifier: Programming Language :: Python :: 3

{h_message_bus-0.0.16 → h_message_bus-0.0.18}/src/h_message_bus.egg-info/SOURCES.txt

@@ -24,6 +24,7 @@ src/h_message_bus/domain/models/request_message_topic.py
 src/h_message_bus/domain/models/vector_collection_metadata.py
 src/h_message_bus/domain/models/vector_query_answer.py
 src/h_message_bus/domain/request_messages/__init__.py
+src/h_message_bus/domain/request_messages/tg_user_message_reply_message.py
 src/h_message_bus/domain/request_messages/tg_user_message_request_message.py
 src/h_message_bus/domain/request_messages/twitter_get_user_request_message.py
 src/h_message_bus/domain/request_messages/twitter_get_user_response_message.py