h-message-bus 0.0.7__tar.gz → 0.0.9__tar.gz

{h_message_bus-0.0.7 → h_message_bus-0.0.9}/PKG-INFO

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

{h_message_bus-0.0.7 → h_message_bus-0.0.9}/pyproject.toml

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

{h_message_bus-0.0.7 → h_message_bus-0.0.9}/src/h_message_bus/domain/twitter_get_user_response_message.py

@@ -69,3 +69,28 @@ class TwitterGetUserResponseMessage(HaiMessage):
         """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.7 → h_message_bus-0.0.9}/src/h_message_bus/domain/vector_read_response_message.py

@@ -33,3 +33,13 @@ class VectorReadResponseMessage(HaiMessage):
     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.7 → h_message_bus-0.0.9}/src/h_message_bus/domain/vector_save_request_message.py

@@ -14,7 +14,7 @@ class VectorSaveRequestMessage(HaiMessage):
         return super().create(topic=topic, payload=payload)
 
     @classmethod
-    def create_message(cls, collection_name: str, document_id: str, content: str, metadata: str) -> 'VectorSaveRequestMessage':
+    def create_message(cls, collection_name: str, document_id: str, content: str, metadata: dict[str,str]) -> 'VectorSaveRequestMessage':
         """Create a message requesting Twitter user data"""
         return cls.create(
             topic=MessageTopic.AI_VECTORS_SAVE,
@@ -42,6 +42,6 @@ class VectorSaveRequestMessage(HaiMessage):
         return self.payload.get("content")
 
     @property
-    def metadata(self) -> str:
+    def metadata(self) -> dict[str, str]:
         """Get the metadata from payload"""
         return self.payload.get("metadata")

{h_message_bus-0.0.7 → h_message_bus-0.0.9}/src/h_message_bus.egg-info/PKG-INFO

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