h-message-bus 0.0.18__py3-none-any.whl → 0.0.19__py3-none-any.whl

h_message_bus/domain/request_messages/tg_message_request_message.py

@@ -0,0 +1,48 @@
+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 TelegramMessageRequestMessage(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) -> 'TelegramMessageRequestMessage':
+
+        payload = {
+            "message": message,
+            "chat_id": chat_id,
+        }
+
+        return cls.create(
+            topic=RequestMessageTopic.AI_TG_CHAT_SEND,
+            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)
+
+    @classmethod
+    def from_hai_message(cls, message: HaiMessage) -> 'TelegramMessageRequestMessage':
+        payload = message.payload
+
+        return cls.create_message(
+            message=payload.get("message", ""),
+            chat_id=payload.get("chat_id")
+        )
+

{h_message_bus-0.0.18.dist-info → h_message_bus-0.0.19.dist-info}/METADATA

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

{h_message_bus-0.0.18.dist-info → h_message_bus-0.0.19.dist-info}/RECORD

@@ -17,7 +17,8 @@ h_message_bus/domain/models/request_message_topic.py,sha256=bg1frk2_X3SHQoTFK-9q
 h_message_bus/domain/models/vector_collection_metadata.py,sha256=fSy3ZQ-zkYISTpc-_M6XWf1JkufVAlzlS2ZT60TzA2s,891
 h_message_bus/domain/models/vector_query_answer.py,sha256=V1JR5Di7kIVbRF_eq2uPOhDnkZpQvo9pxD41V7MNV7Q,551
 h_message_bus/domain/request_messages/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-h_message_bus/domain/request_messages/tg_user_message_reply_message.py,sha256=UQRByX4tDEJSoYfVLzbG0vnaVvzx3-_hLjG1eBq9wpg,1780
+h_message_bus/domain/request_messages/tg_message_request_message.py,sha256=J0q0u5kWWXuEeKx3dptDfO53rXczZYn-52_BPwwjOtg,1460
+h_message_bus/domain/request_messages/tg_user_message_reply_request_message.py,sha256=UQRByX4tDEJSoYfVLzbG0vnaVvzx3-_hLjG1eBq9wpg,1780
 h_message_bus/domain/request_messages/tg_user_message_request_message.py,sha256=xg6GBjMYw6FfPC08UBaLulQMxmoqJj6UNjdtFS1Dqtk,3539
 h_message_bus/domain/request_messages/twitter_get_user_request_message.py,sha256=BX-ytvnEzL9u1Yks5iXIx60Hmv3UP7asOG2uYA0mfgE,1212
 h_message_bus/domain/request_messages/twitter_get_user_response_message.py,sha256=qO1cLtEmZyddv5pXHNgpnpO5t3Oex8cOG0hjGhOfru8,3302
@@ -29,7 +30,7 @@ h_message_bus/domain/request_messages/vector_save_request_message.py,sha256=Jd52
 h_message_bus/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h_message_bus/infrastructure/nats_client_repository.py,sha256=lAapNaL0rqBr-9KaHcexuo5XLmSWRWspkJWXS9dMqPU,3371
 h_message_bus/infrastructure/nats_config.py,sha256=Yzqqd1bCfmUv_4FOnA1dvqIpakzV0BUL2_nXQcndWvo,1304
-h_message_bus-0.0.18.dist-info/METADATA,sha256=p498dQ-mFhEqyoyk-gUUcIylRZ3Lz8Pqs_qfVkaw6xY,8834
-h_message_bus-0.0.18.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-h_message_bus-0.0.18.dist-info/top_level.txt,sha256=BArjhm_lwFR9yJJEIf-LT_X64psuLkXFdbpQRJUreFE,23
-h_message_bus-0.0.18.dist-info/RECORD,,
+h_message_bus-0.0.19.dist-info/METADATA,sha256=PlyIETAeGDKQrTjz9ie2h76YJ8bV7RoIIpP9O9wcfeM,8834
+h_message_bus-0.0.19.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+h_message_bus-0.0.19.dist-info/top_level.txt,sha256=BArjhm_lwFR9yJJEIf-LT_X64psuLkXFdbpQRJUreFE,23
+h_message_bus-0.0.19.dist-info/RECORD,,