h-message-bus 0.0.35__py3-none-any.whl → 0.0.37__py3-none-any.whl

h_message_bus/domain/models/request_message_topic.py

@@ -56,6 +56,9 @@ class RequestMessageTopic(str, Enum):
     TWITTER_QUOTE_RETWEET= "hai.twitter.quote.retweet"
     TWITTER_QUOTE_RETWEET_RESPONSE = "hai.twitter.quote.retweet.response"
 
+    TWITTER_GET_TWEET= "hai.twitter.get.tweet"
+    TWITTER_GET_TWEET_RESPONSE = "hai.twitter.get.tweet.response"
+
     # tools
     WEB_SEARCH = "hai.tools.web.search"
     WEB_SEARCH_RESPONSE = "hai.tools.web.search.response"

h_message_bus/domain/request_messages/twitter_get_tweet_request_message.py

@@ -0,0 +1,40 @@
+from dataclasses import dataclass
+from typing import Type, Dict, Any, TypeVar
+
+from ..models.request_message_topic import RequestMessageTopic
+from ...domain.models.hai_message import HaiMessage
+
+T = TypeVar('T', bound='TwitterGetTweetRequestMessage')
+
+@dataclass
+class TwitterGetTweetRequestMessage(HaiMessage):
+    """Message to request getting a tweet by its ID"""
+
+    @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, tweet_id: str) -> 'TwitterGetTweetRequestMessage':
+        """Create a message requesting to get a tweet by its ID"""
+        return cls.create(
+            topic=RequestMessageTopic.TWITTER_GET_TWEET,  # You'll need to add this to RequestMessageTopic
+            payload={
+                "tweet_id": tweet_id
+            },
+        )
+
+    @property
+    def tweet_id(self) -> str:
+        """Get the tweet ID from the payload"""
+        return self.payload.get("tweet_id", "")
+
+    @classmethod
+    def from_hai_message(cls, message: HaiMessage) -> 'TwitterGetTweetRequestMessage':
+        """Create a TwitterGetTweetRequestMessage from a HaiMessage"""
+        payload = message.payload
+
+        return cls.create_message(
+            tweet_id=payload.get("tweet_id", "")
+        )

h_message_bus/domain/request_messages/twitter_get_tweet_response_message.py

@@ -0,0 +1,54 @@
+from dataclasses import dataclass
+from typing import Type, Dict, Any, TypeVar, Optional
+
+from ..models.request_message_topic import RequestMessageTopic
+from ...domain.models.hai_message import HaiMessage
+
+T = TypeVar('T', bound='TwitterGetTweetResponseMessage')
+
+
+@dataclass
+class TwitterGetTweetResponseMessage(HaiMessage):
+    """Message to respond to a request to get a tweet by its ID"""
+
+    @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,
+                       tweet_id: str,
+                       success: bool,
+                       tweet: Optional[Dict[str, Any]] = None,
+                       error_message: str = "") -> 'TwitterGetTweetResponseMessage':
+        """Create a response message for a request to get a tweet"""
+        return cls.create(
+            topic=RequestMessageTopic.TWITTER_GET_TWEET_RESPONSE,  # You'll need to add this to ResponseMessageTopic
+            payload={
+                "tweet_id": tweet_id,
+                "success": success,
+                "tweet": tweet,
+                "error_message": error_message
+            },
+        )
+
+    @property
+    def tweet_id(self) -> str:
+        """Get the tweet ID that was requested"""
+        return self.payload.get("tweet_id", "")
+
+    @property
+    def success(self) -> bool:
+        """Get whether the get tweet operation was successful"""
+        return self.payload.get("success", False)
+
+    @property
+    def tweet(self) -> Optional[Dict[str, Any]]:
+        """Get the tweet data if the operation was successful"""
+        return self.payload.get("tweet", None)
+
+    @property
+    def error_message(self) -> str:
+        """Get the error message if the get tweet operation failed"""
+        return self.payload.get("error_message", "")

h_message_bus/domain/request_messages/twitter_user_tweets_response_message.py

@@ -73,7 +73,7 @@ class TwitterUserTweetsResponseMessage(HaiMessage):
             tweet_data = {
                 "tweet_id": str(message.tweet_id),
                 "message": message.message,
-                "created_at": message.created_at.isoformat() if message.created_at else None,
+                "created_at": message.created_at,
                 "view_count": message.view_count,
                 "retweet_count": message.retweet_count,
                 "reply_count": message.reply_count,

{h_message_bus-0.0.35.dist-info → h_message_bus-0.0.37.dist-info}/METADATA

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

{h_message_bus-0.0.35.dist-info → h_message_bus-0.0.37.dist-info}/RECORD

@@ -13,7 +13,7 @@ h_message_bus/domain/event_messages/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQ
 h_message_bus/domain/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h_message_bus/domain/models/event_message_topic.py,sha256=fSjRMBwMD2RE9GBUud38XMrgLZcngURERrbuoAhQ0Hk,717
 h_message_bus/domain/models/hai_message.py,sha256=b5CfX7hi5uNq77IVnZzEi9iotc4b_U2MNYwV6JY7JcU,2146
-h_message_bus/domain/models/request_message_topic.py,sha256=uXc3F68WU7O813QQARmObcBrlJ-PODs_259fq7OStuw,3883
+h_message_bus/domain/models/request_message_topic.py,sha256=JuSCMlm1hc6lrw-zu7oR5RuOGAOKsI-4NThO4BUiiwo,4000
 h_message_bus/domain/models/twitter_message_metadata.py,sha256=Cj0dSVQi0Com9nPlJeEM0c-ZAQKGqIq_txvdzYg0myw,1203
 h_message_bus/domain/models/twitter_user_metadata.py,sha256=N7GVyn2txmlMQ19aatNh7lvzO6FNSj4dKVujestha9o,2827
 h_message_bus/domain/models/vector_collection_metadata.py,sha256=fSy3ZQ-zkYISTpc-_M6XWf1JkufVAlzlS2ZT60TzA2s,891
@@ -40,6 +40,8 @@ h_message_bus/domain/request_messages/tg_user_message_reply_request_message.py,s
 h_message_bus/domain/request_messages/tg_user_message_request_message.py,sha256=pHmmGJpMEnwzbTVVvoD6UOMflQOuQKgErZA9MyRwWt8,3121
 h_message_bus/domain/request_messages/twitter_follow_user_request_message.py,sha256=6qQ1zkqL3efHn4Asmkjqlne3YAdKCyDEjWhAUdnvj3I,1446
 h_message_bus/domain/request_messages/twitter_follow_user_response_message.py,sha256=4t1NucWuDgKXI6-cTJ69B3mBDG8aoY5WuSZF5zalXrM,1633
+h_message_bus/domain/request_messages/twitter_get_tweet_request_message.py,sha256=AVL-ativiwbB9ftwXp9MKezonqPYtdSf1stJg0zf2S0,1443
+h_message_bus/domain/request_messages/twitter_get_tweet_response_message.py,sha256=x8pr34SZHeH73TEJN6QiXd-tqW3FNdhN9FzlFW2qEKQ,1991
 h_message_bus/domain/request_messages/twitter_get_user_request_message.py,sha256=OfC93X1M0vgMU26BBrpA7AdlY32IFdVG_vRT8BdhQOc,1209
 h_message_bus/domain/request_messages/twitter_get_user_response_message.py,sha256=dXhK_JN3Sz6hN8EpDcj7gq6htJ8Yu5G-Cal73XXcEng,3299
 h_message_bus/domain/request_messages/twitter_get_users_request_message.py,sha256=y2Bh77py9RPtKfdPrPi0mv3LVIujne107y8vkPydirs,1545
@@ -55,7 +57,7 @@ h_message_bus/domain/request_messages/twitter_reply_response_message.py,sha256=6
 h_message_bus/domain/request_messages/twitter_retweet_request_message.py,sha256=UHnLrgnqgZixb2q5ugFEHcCJxNkuTzrToGCEVoXm4gU,1423
 h_message_bus/domain/request_messages/twitter_retweet_response_message.py,sha256=OGanRbX7AcKJ35H01wvp0XvuL-Chhm7f5J6NSpGq7uQ,1616
 h_message_bus/domain/request_messages/twitter_user_tweets_request_message.py,sha256=9vAhRxbCCDTqQNNKgp5cDIkiGIsnHwRSOwc6PK6TmcA,2397
-h_message_bus/domain/request_messages/twitter_user_tweets_response_message.py,sha256=DAQauYnUEnKX6AGKF6-zFd0X-j4BNfAMwBMLr5JLEwI,3243
+h_message_bus/domain/request_messages/twitter_user_tweets_response_message.py,sha256=2oygt0Ow_xx0NOdPUViiqeAKh9NmYl8kJILOqHQbKgI,3199
 h_message_bus/domain/request_messages/vector_query_collection_request_message.py,sha256=x1J8SLVBlNS4TFzVJY9UG7mo94l9i4azndwzAlyktZw,1800
 h_message_bus/domain/request_messages/vector_query_collection_response_message.py,sha256=Q589JT_QE-oVsTh1GEmK__3Hct7ijfM1UTcPi0YTGn8,1767
 h_message_bus/domain/request_messages/vector_read_metadata_request_message.py,sha256=P9t6pcynS112F9_Jmbvw8wlfXhs5NOnV48SUVq92xt8,873
@@ -66,7 +68,7 @@ h_message_bus/domain/request_messages/web_search_request_message.py,sha256=ZoB4i
 h_message_bus/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h_message_bus/infrastructure/nats_client_repository.py,sha256=bhaWRK94h2EmFIef2DALnoBJKnuTnx6-iU7IXO2S_EU,3642
 h_message_bus/infrastructure/nats_config.py,sha256=Yzqqd1bCfmUv_4FOnA1dvqIpakzV0BUL2_nXQcndWvo,1304
-h_message_bus-0.0.35.dist-info/METADATA,sha256=IsK8LwYq_OgBL1ut9W0-4Bm2dp5uKbkR4UesiNQ7490,8834
-h_message_bus-0.0.35.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-h_message_bus-0.0.35.dist-info/top_level.txt,sha256=BArjhm_lwFR9yJJEIf-LT_X64psuLkXFdbpQRJUreFE,23
-h_message_bus-0.0.35.dist-info/RECORD,,
+h_message_bus-0.0.37.dist-info/METADATA,sha256=yM3f3SNB6PWyNJXHzN09h11c74jTuSfZcOdetKOP1E4,8834
+h_message_bus-0.0.37.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+h_message_bus-0.0.37.dist-info/top_level.txt,sha256=BArjhm_lwFR9yJJEIf-LT_X64psuLkXFdbpQRJUreFE,23
+h_message_bus-0.0.37.dist-info/RECORD,,