h-message-bus 0.0.6__tar.gz → 0.0.8__tar.gz

{h_message_bus-0.0.6 → h_message_bus-0.0.8}/PKG-INFO

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

{h_message_bus-0.0.6 → h_message_bus-0.0.8}/pyproject.toml

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

{h_message_bus-0.0.6 → h_message_bus-0.0.8}/src/h_message_bus/domain/twitter_get_user_request_message.py

@@ -21,3 +21,7 @@ class TwitterGetUserRequestMessage(HaiMessage):
             payload={"username": username},
         )
 
+    @property
+    def username(self) -> str:
+        """Get the username from the payload"""
+        return self.payload.get("username", "")

h_message_bus-0.0.8/src/h_message_bus/domain/twitter_get_user_response_message.py

@@ -0,0 +1,96 @@
+from typing import Type, TypeVar, Dict, Any, List
+
+from .topics import MessageTopic
+from ..domain.hai_message import HaiMessage
+
+T = TypeVar('T', bound='HaiMessage')
+
+class TwitterGetUserResponseMessage(HaiMessage):
+    """Response of Twitter user information request"""
+
+    @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, user_id: str, screen_name: str, description: str, followers_count: str, like_count: str, is_verified: str, url: str, bio_urls: [str]) -> 'TwitterGetUserResponseMessage':
+        """Create a response message from Twitter user information"""
+        return cls.create(
+            topic=MessageTopic.AI_TWITTER_GET_USER_RESPONSE,
+            payload={
+                'id': user_id,
+                'screen_name': screen_name,
+                'description': description,
+                'followers_count': followers_count,
+                'like_count': like_count,
+                'is_verified': is_verified,
+                'url': url,
+                'bio_urls': bio_urls
+            })
+
+    @property
+    def user_id(self) -> str:
+        """Get user ID"""
+        return self.payload.get('id', '')
+
+    @property
+    def screen_name(self) -> str:
+        """Get screen name"""
+        return self.payload.get('screen_name', '')
+
+    @property
+    def description(self) -> str:
+        """Get user description"""
+        return self.payload.get('description', '')
+
+    @property
+    def followers_count(self) -> str:
+        """Get followers count"""
+        return self.payload.get('followers_count', '')
+
+    @property
+    def like_count(self) -> str:
+        """Get like count"""
+        return self.payload.get('like_count', '')
+
+    @property
+    def is_verified(self) -> str:
+        """Get verification status"""
+        return self.payload.get('is_verified', '')
+
+    @property
+    def url(self) -> str:
+        """Get user URL"""
+        return self.payload.get('url', '')
+
+    @property
+    def bio_urls(self) -> List[str]:
+        """Get URLs from user bio"""
+        return self.payload.get('bio_urls', [])
+
+    @classmethod
+    def from_hai_message(cls, message: HaiMessage) -> 'TwitterGetUserResponseMessage':
+        """
+        Convert a HaiMessage to TwitterGetUserResponseMessage
+
+        Args:
+            message: The source HaiMessage object
+
+        Returns:
+            A new TwitterGetUserResponseMessage with data from the source message
+        """
+        # Extract necessary fields from the message payload
+        payload = message.payload
+
+        return cls.create_message(
+            user_id=payload.get('id', ''),
+            screen_name=payload.get('screen_name', ''),
+            description=payload.get('description', ''),
+            followers_count=payload.get('followers_count', ''),
+            like_count=payload.get('like_count', ''),
+            is_verified=payload.get('is_verified', ''),
+            url=payload.get('url', ''),
+            bio_urls=payload.get('bio_urls', [])
+        )
+

{h_message_bus-0.0.6 → h_message_bus-0.0.8}/src/h_message_bus/domain/vector_read_request_message.py

@@ -23,4 +23,19 @@ class VectorReadRequestMessage(HaiMessage):
                 "query": query,
                 "top_n": top_n
             },
-        )
+        )
+
+    @property
+    def collection_name(self) -> str:
+        """Get the collection name from the message payload"""
+        return self.payload.get("collection_name", "")
+
+    @property
+    def query(self) -> str:
+        """Get the query from the message payload"""
+        return self.payload.get("query", "")
+
+    @property
+    def top_n(self) -> str:
+        """Get the top_n value from the message payload"""
+        return self.payload.get("top_n", "")

{h_message_bus-0.0.6 → h_message_bus-0.0.8}/src/h_message_bus/domain/vector_read_response_message.py

@@ -22,4 +22,24 @@ class VectorReadResponseMessage(HaiMessage):
                 "dimensions": dimensions,
                 "results": results
             },
-        )
+        )
+
+    @property
+    def dimensions(self) -> list[str]:
+        """Returns the dimensions from the message payload"""
+        return self.payload.get("dimensions", [])
+
+    @property
+    def results(self) -> list[str]:
+        """Returns the results from the message payload"""
+        return self.payload.get("results", [])
+
+    @classmethod
+    def from_hai_message(cls, message: HaiMessage) -> 'VectorReadResponseMessage':
+        # Extract necessary fields from the message payload
+        payload = message.payload
+
+        return cls.create_message(
+            dimensions=payload.get('dimensions', []),
+            results=payload.get('results', [])
+        )

{h_message_bus-0.0.6 → h_message_bus-0.0.8}/src/h_message_bus/domain/vector_save_request_message.py

@@ -24,4 +24,24 @@ class VectorSaveRequestMessage(HaiMessage):
                 "content": content,
                 "metadata": metadata
             },
-        )
+        )
+
+    @property
+    def collection_name(self) -> str:
+        """Get the collection name from payload"""
+        return self.payload.get("collection_name")
+
+    @property
+    def document_id(self) -> str:
+        """Get the document ID from payload"""
+        return self.payload.get("document_id")
+
+    @property
+    def content(self) -> str:
+        """Get the content from payload"""
+        return self.payload.get("content")
+
+    @property
+    def metadata(self) -> str:
+        """Get the metadata from payload"""
+        return self.payload.get("metadata")

{h_message_bus-0.0.6 → h_message_bus-0.0.8}/src/h_message_bus.egg-info/PKG-INFO

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

h_message_bus-0.0.6/src/h_message_bus/domain/twitter_get_user_response_message.py

@@ -1,30 +0,0 @@
-from typing import Type, TypeVar, Dict, Any
-
-from .topics import MessageTopic
-from ..domain.hai_message import HaiMessage
-
-T = TypeVar('T', bound='HaiMessage')
-
-class TwitterGetUserResponseMessage(HaiMessage):
-    """Response of Twitter user information request"""
-
-    @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, user_id: str, screen_name: str, description: str, followers_count: str, like_count: str, is_verified: str, url: str, bio_urls: [str]) -> 'TwitterGetUserResponseMessage':
-        """Create a response message from Twitter user information"""
-        return cls.create(
-            topic=MessageTopic.AI_TWITTER_GET_USER_RESPONSE,
-            payload={
-                'id': user_id,
-                'screen_name': screen_name,
-                'description': description,
-                'followers_count': followers_count,
-                'like_count': like_count,
-                'is_verified': is_verified,
-                'url': url,
-                'bio_urls': bio_urls
-            })