h-message-bus 0.0.8__py3-none-any.whl → 0.0.10__py3-none-any.whl

h_message_bus/domain/twitter_get_user_request_message.py

@@ -25,3 +25,11 @@ class TwitterGetUserRequestMessage(HaiMessage):
     def username(self) -> str:
         """Get the username from the payload"""
         return self.payload.get("username", "")
+
+    @classmethod
+    def from_hai_message(cls, message: HaiMessage) -> 'TwitterGetUserRequestMessage':
+        payload = message.payload
+
+        return cls.create_message(
+        username=payload.get("username", "")
+        )

h_message_bus/domain/vector_read_request_message.py

@@ -39,3 +39,13 @@ class VectorReadRequestMessage(HaiMessage):
     def top_n(self) -> str:
         """Get the top_n value from the message payload"""
         return self.payload.get("top_n", "")
+
+    @classmethod
+    def from_hai_message(cls, message: HaiMessage) -> 'VectorReadRequestMessage':
+        payload = message.payload
+
+        return cls.create_message(
+            collection_name=payload.get("collection_name", ""),
+            query=payload.get("query", ""),
+            top_n=payload.get("top_n", "")
+        )

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,18 @@ 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")
+
+    @classmethod
+    def from_hai_message(cls, message: HaiMessage) -> 'VectorSaveRequestMessage':
+        # Extract necessary fields from the message payload
+        payload = message.payload
+
+        return cls.create_message(
+            collection_name=payload.get("collection_name", ''),
+            document_id=payload.get("document_id", ''),
+            content=payload.get("content", ''),
+            metadata=payload.get("metadata", {})
+        )

{h_message_bus-0.0.8.dist-info → h_message_bus-0.0.10.dist-info}/METADATA

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

{h_message_bus-0.0.8.dist-info → h_message_bus-0.0.10.dist-info}/RECORD

@@ -10,15 +10,15 @@ h_message_bus/application/message_subcriber.py,sha256=2yqUUHVHtZlA6-zzyHzghdWQ7Z
 h_message_bus/domain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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_request_message.py,sha256=STDbCHY1nHPjEn7Wk0L4YywHkfEBT-cuPFLduwr7XQI,1167
 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_request_message.py,sha256=MH0QxpXsgWaMFAKOMtxlvR6EtikvIqkv7Pk_IOv_s2c,1725
 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=jDSzK93M9bOhqtsJbmzLyVsQNOqqXpQIWySC5jyBaTo,1563
+h_message_bus/domain/vector_save_request_message.py,sha256=si00UNCyHVSnOk9qPbFfaaq7Fwt771sj7VPH0ktvYTA,2053
 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.8.dist-info/METADATA,sha256=dZdXWrigJ5YMiEzsUzbBjHUyuLswarNn5HBMoHnjNhY,8833
-h_message_bus-0.0.8.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-h_message_bus-0.0.8.dist-info/top_level.txt,sha256=7feCw9I0yIIqsYsVzUMYvCCmUTOiGX1a5F2VbSkFscc,14
-h_message_bus-0.0.8.dist-info/RECORD,,
+h_message_bus-0.0.10.dist-info/METADATA,sha256=iM93u3vZx91y6R9GpzNsHtt--lSzuH5zup56Qj6RsK0,8834
+h_message_bus-0.0.10.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+h_message_bus-0.0.10.dist-info/top_level.txt,sha256=7feCw9I0yIIqsYsVzUMYvCCmUTOiGX1a5F2VbSkFscc,14
+h_message_bus-0.0.10.dist-info/RECORD,,