h-message-bus 0.0.7__py3-none-any.whl → 0.0.9__py3-none-any.whl

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/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/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.dist-info → h_message_bus-0.0.9.dist-info}/METADATA

@@ -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.dist-info → h_message_bus-0.0.9.dist-info}/RECORD

@@ -11,14 +11,14 @@ h_message_bus/domain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 h_message_bus/domain/hai_message.py,sha256=b5CfX7hi5uNq77IVnZzEi9iotc4b_U2MNYwV6JY7JcU,2146
 h_message_bus/domain/topics.py,sha256=OW-HbTmvogqYaXb0ugZMSvNj2Qhllwfh7BRVS6m4-CU,990
 h_message_bus/domain/twitter_get_user_request_message.py,sha256=n6_v7s57AqrIErINcwVDnvClPaFolKZ99o4tbG3B4gI,930
-h_message_bus/domain/twitter_get_user_response_message.py,sha256=LnLHZU-_7WUtE81iSxeyErCVGVfbS89biIFpdbND4Lw,2318
+h_message_bus/domain/twitter_get_user_response_message.py,sha256=yGa9MVcN6-VUcgYLXIv-u9Oll5S6AeJLI5xqgKTE32Y,3257
 h_message_bus/domain/vector_read_request_message.py,sha256=8LoJhsEdwIIceS8bL05gDcFxMUySK5CzRTgj_TmEeqg,1384
-h_message_bus/domain/vector_read_response_message.py,sha256=sPj-jVifXW8hPkM4lIdcJauQY5RJ4H88iD6DVNxwnQ0,1207
-h_message_bus/domain/vector_save_request_message.py,sha256=jDSzK93M9bOhqtsJbmzLyVsQNOqqXpQIWySC5jyBaTo,1563
+h_message_bus/domain/vector_read_response_message.py,sha256=cXMiHnDfeEW4tLoD-T-8DJcX9jRa8PK-gURW4FgCO54,1559
+h_message_bus/domain/vector_save_request_message.py,sha256=d4vDI_pWapziZwDEPWzGAI_gdRB0acwVacdG4AUp7XI,1584
 h_message_bus/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h_message_bus/infrastructure/nats_client_repository.py,sha256=JURLfrTk6188HLaHUHZItBu8utQb68CznP5oQDrTUKU,3343
 h_message_bus/infrastructure/nats_config.py,sha256=Yzqqd1bCfmUv_4FOnA1dvqIpakzV0BUL2_nXQcndWvo,1304
-h_message_bus-0.0.7.dist-info/METADATA,sha256=5E8RWWFnI1m-dmXFhRyuHhgZ9v24QHvheEE8CSg5tf4,8833
-h_message_bus-0.0.7.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-h_message_bus-0.0.7.dist-info/top_level.txt,sha256=7feCw9I0yIIqsYsVzUMYvCCmUTOiGX1a5F2VbSkFscc,14
-h_message_bus-0.0.7.dist-info/RECORD,,
+h_message_bus-0.0.9.dist-info/METADATA,sha256=mUWhFhqWF6skyreUTF1NRJbMS6pzb-cD3zqfRRqNtMU,8833
+h_message_bus-0.0.9.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+h_message_bus-0.0.9.dist-info/top_level.txt,sha256=7feCw9I0yIIqsYsVzUMYvCCmUTOiGX1a5F2VbSkFscc,14
+h_message_bus-0.0.9.dist-info/RECORD,,