h-message-bus 0.0.24__tar.gz → 0.0.26__tar.gz

{h_message_bus-0.0.24 → h_message_bus-0.0.26}/PKG-INFO

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

{h_message_bus-0.0.24 → h_message_bus-0.0.26}/pyproject.toml

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

{h_message_bus-0.0.24 → h_message_bus-0.0.26}/src/h_message_bus/domain/models/request_message_topic.py

@@ -30,6 +30,8 @@ class RequestMessageTopic(str, Enum):
     # Twitter
     TWITTER_GET_USER = "hai.twitter.get.user"
     TWITTER_GET_USER_RESPONSE = "hai.twitter.get.user.response"
+    TWITTER_GET_USERS = "hai.twitter.get.users"
+    TWITTER_GET_USERS_RESPONSE = "hai.twitter.get.users.response"
 
     TWITTER_USER_SEND_AI_CHAT_SEND = "hai.twitter.user.chat.send"
     TWITTER_USER_SEND_AI_CHAT_SEND_RESPONSE = "hai.twitter.user.chat.send.response"

h_message_bus-0.0.26/src/h_message_bus/domain/models/twitter_user_metadata.py

@@ -0,0 +1,76 @@
+class TwitterUserMetadata:
+    def __init__(self,
+                 id: str,
+                 screen_name: str,
+                 user_name: str = None,
+                 description: str = None,
+                 followers_count: int = 0,
+                 like_count: int = 0,
+                 is_verified: bool = False,
+                 url: str = None,
+                 bio_urls: list = None):
+        """
+        Initialize a TwitterUserMetadata instance.
+
+        Args:
+            id (str): The user's Twitter ID
+            screen_name (str): The user's screen name
+            user_name (str, optional): The user's name
+            description (str, optional): The user's bio description
+            followers_count (int, optional): Number of followers
+            like_count (int, optional): Number of likes
+            is_verified (bool, optional): Whether the user is verified
+            url (str, optional): The user's profile URL
+            bio_urls (list, optional): URLs mentioned in the user's bio
+        """
+        self.id = id
+        self.screen_name = screen_name
+        self.user_name = user_name
+        self.description = description
+        self.followers_count = followers_count
+        self.like_count = like_count
+        self.is_verified = is_verified
+        self.url = url
+        self.bio_urls = bio_urls or []
+
+    def to_dict(self) -> dict:
+        """
+        Convert TwitterUserMetadata instance to a dictionary.
+
+        Returns:
+            dict: A dictionary representation of the TwitterUserMetadata instance.
+        """
+        return {
+            'id': self.id,
+            'screen_name': self.screen_name,
+            'user_name': self.user_name,
+            'description': self.description,
+            'followers_count': self.followers_count,
+            'like_count': self.like_count,
+            'is_verified': self.is_verified,
+            'url': self.url,
+            'bio_urls': self.bio_urls
+        }
+
+    @classmethod
+    def from_dict(cls, data: dict) -> 'TwitterUserMetadata':
+        """
+        Create a TwitterUserMetadata instance from a dictionary.
+
+        Args:
+            data (dict): A dictionary containing TwitterUserMetadata attributes
+
+        Returns:
+            TwitterUserMetadata: A new TwitterUserMetadata instance
+        """
+        return cls(
+            id=data.get('id'),
+            screen_name=data.get('screen_name'),
+            user_name=data.get('user_name'),
+            description=data.get('description'),
+            followers_count=data.get('followers_count', 0),
+            like_count=data.get('like_count', 0),
+            is_verified=data.get('is_verified', False),
+            url=data.get('url'),
+            bio_urls=data.get('bio_urls', [])
+        )

h_message_bus-0.0.26/src/h_message_bus/domain/request_messages/twitter_get_users_request_message.py

@@ -0,0 +1,42 @@
+import json
+from typing import Type, TypeVar, Dict, Any, List
+
+from ..models.request_message_topic import RequestMessageTopic
+from ...domain.models.hai_message import HaiMessage
+
+T = TypeVar('T', bound='HaiMessage')
+
+class TwitterGetUsersRequestMessage(HaiMessage):
+    """Message to request Twitter user information"""
+
+    @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, usernames: List[str]) -> 'TwitterGetUsersRequestMessage':
+        """Create a message requesting Twitter user data"""
+        return cls.create(
+            topic=RequestMessageTopic.TWITTER_GET_USERS,
+            payload={"usernames": json.dumps(usernames)},
+        )
+
+    @property
+    def usernames(self) -> List[str]:
+        """Get the username from the payload"""
+        usernames_payload = self.payload.get("usernames", "")
+        usernames = json.loads(usernames_payload)
+        response_list = []
+        for username in usernames:
+            response_list.append(username)
+        return response_list
+
+    @classmethod
+    def from_hai_message(cls, message: HaiMessage) -> 'TwitterGetUsersRequestMessage':
+        payload = message.payload
+        usernames = payload.get("usernames", "")
+        usernames_list = json.loads(usernames)
+        return cls.create_message(
+        usernames=usernames_list
+        )

h_message_bus-0.0.26/src/h_message_bus/domain/request_messages/twitter_get_users_response_message.py

@@ -0,0 +1,44 @@
+import json
+from typing import Type, TypeVar, Dict, Any, List
+
+from ..models.request_message_topic import RequestMessageTopic
+from ..models.twitter_user_metadata import TwitterUserMetadata
+from ...domain.models.hai_message import HaiMessage
+
+T = TypeVar('T', bound='HaiMessage')
+
+class TwitterGetUsersResponseMessage(HaiMessage):
+
+    @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, users: List[dict[str, str]]) -> 'TwitterGetUsersResponseMessage':
+        user_result = json.dumps(users)
+        return cls.create(
+            topic=RequestMessageTopic.TWITTER_GET_USERS_RESPONSE,
+            payload={
+                "users": user_result,
+            },
+        )
+
+    @property
+    def users(self) -> List[TwitterUserMetadata]:
+        """Get the collection name from the message payload"""
+        users = self.payload.get("users", "")
+        users_dict = json.loads(users)
+        response_list = []
+        for user in users_dict:
+            response_list.append(TwitterUserMetadata.from_dict(user))
+        return response_list
+
+    @classmethod
+    def from_hai_message(cls, message: HaiMessage) -> 'TwitterGetUsersResponseMessage':
+        payload = message.payload
+        metadata = payload.get("users", "")
+        meta_list = json.loads(metadata)
+        return cls.create_message(
+            users=meta_list
+        )

{h_message_bus-0.0.24 → h_message_bus-0.0.26}/src/h_message_bus/domain/request_messages/vector_save_request_message.py

@@ -33,7 +33,7 @@ class VectorSaveRequestMessage(HaiMessage):
         return self.payload.get("collection_name")
 
     @property
-    def collection_metadata(self) -> str:
+    def collection_metadata(self) -> dict[str, str]:
         """Get the collection name from the payload"""
         return self.payload.get("collection_metadata")
 

{h_message_bus-0.0.24 → h_message_bus-0.0.26}/src/h_message_bus.egg-info/PKG-INFO

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

{h_message_bus-0.0.24 → h_message_bus-0.0.26}/src/h_message_bus.egg-info/SOURCES.txt

@@ -21,6 +21,7 @@ src/h_message_bus/domain/models/__init__.py
 src/h_message_bus/domain/models/event_message_topic.py
 src/h_message_bus/domain/models/hai_message.py
 src/h_message_bus/domain/models/request_message_topic.py
+src/h_message_bus/domain/models/twitter_user_metadata.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
@@ -29,6 +30,8 @@ src/h_message_bus/domain/request_messages/tg_user_message_reply_request_message.
 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
+src/h_message_bus/domain/request_messages/twitter_get_users_request_message.py
+src/h_message_bus/domain/request_messages/twitter_get_users_response_message.py
 src/h_message_bus/domain/request_messages/vector_query_collection_request_message.py
 src/h_message_bus/domain/request_messages/vector_query_collection_response_message.py
 src/h_message_bus/domain/request_messages/vector_read_metadata_request_message.py