h-message-bus 0.0.20__py3-none-any.whl → 0.0.22__py3-none-any.whl

h_message_bus/adapters/nats_publisher_adapter.py

@@ -23,12 +23,17 @@ class NatsPublisherAdapter(MessagePublisher):
     def __init__(self, nats_client: NatsClientRepository):
         self.nats_client = nats_client
 
-    async def publish(self, message: HaiMessage) -> None:
+    async def publish(self, message: HaiMessage):
         """Publish a message to NATS."""
-        await self.nats_client.publish(
-            message.topic,
-            message.to_json().encode()
-        )
+        try:
+
+            await self.nats_client.publish(
+                message.topic,
+                message.to_json().encode()
+            )
+        except Exception as e:
+            print(f"Failed to publish message: {e}")
+            return
 
     async def request(self, message: HaiMessage, timeout: float = 2.0) -> Optional[HaiMessage]:
         """Send a request and wait for a response."""

h_message_bus/application/message_publisher.py

@@ -15,7 +15,7 @@ class MessagePublisher(ABC):
     """
 
     @abstractmethod
-    async def publish(self, message: HaiMessage) -> None:
+    async def publish(self, message: HaiMessage):
         """Publish a message."""
         pass
 

h_message_bus/domain/request_messages/web_get_docs_request_message.py

@@ -13,12 +13,13 @@ class WebGetDocsRequestMessage(HaiMessage):
         return super().create(topic=topic, payload=payload)
 
     @classmethod
-    def create_message(cls, twitter_user: Optional[str], root_url: Optional[str]) -> 'WebGetDocsRequestMessage':
+    def create_message(cls, collection_name: str, twitter_user: Optional[str], root_url: Optional[str]) -> 'WebGetDocsRequestMessage':
         return cls.create(
             topic=RequestMessageTopic.WEB_GET_DOCS,
             payload={
                 "twitter_user": twitter_user,
                 "root_url": root_url,
+                "collection_name": collection_name,
             },
         )
 
@@ -30,11 +31,16 @@ class WebGetDocsRequestMessage(HaiMessage):
     def root_url(self) -> str:
         return self.payload.get("root_url", "")
 
+    @property
+    def collection_name(self) -> str:
+        return self.payload.get("collection_name", "")
+
     @classmethod
     def from_hai_message(cls, message: HaiMessage) -> 'WebGetDocsRequestMessage':
         payload = message.payload
 
         return cls.create_message(
             twitter_user=payload.get("twitter_user", ""),
-            root_url=payload.get("root_url", "")
+            root_url=payload.get("root_url", ""),
+            collection_name=payload.get("collection_name", "")
         )

{h_message_bus-0.0.20.dist-info → h_message_bus-0.0.22.dist-info}/METADATA

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

{h_message_bus-0.0.20.dist-info → h_message_bus-0.0.22.dist-info}/RECORD

@@ -2,11 +2,11 @@ __init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h_message_bus/__init__.py,sha256=T_07sHtXLlRuLwK-BRVfFIIOGc48TKbtVLvKtIw2K5U,516
 h_message_bus/adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h_message_bus/adapters/nats_message_adapter.py,sha256=-hzx0UJmKJT4VWfRA_zVT68hddl9dejvUiSdLU1rY-w,1378
-h_message_bus/adapters/nats_publisher_adapter.py,sha256=D2IV4eZyclQUB9ZU8WL0yvtTMTqxyB07wum6INeN3M8,1770
+h_message_bus/adapters/nats_publisher_adapter.py,sha256=kIh7J-eS-MPvfGsnmdGwsWYfLgepBC99v5d04KXyOaE,1900
 h_message_bus/adapters/nats_subscriber_adapter.py,sha256=WX7xvJhVJVa24CRpWGG3SE8CLP7AXcO6ky-qhMTICCY,2072
 h_message_bus/application/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h_message_bus/application/message_processor.py,sha256=1vdVfPn1BUDmswJ-Y5aS-YDbLSizrjX6_CMIBa5SGjc,734
-h_message_bus/application/message_publisher.py,sha256=YrIaykzgBd8tirFs23pr9DIs-VFEh6vHp923u1CLf3I,842
+h_message_bus/application/message_publisher.py,sha256=pYX4JzJLe7L_59gKVYBLqT6iTwFB73zhbzxd7yv2TqI,834
 h_message_bus/application/message_subcriber.py,sha256=2yqUUHVHtZlA6-zzyHzghdWQ7ZmdcFic5c0xi_rgG-M,741
 h_message_bus/domain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h_message_bus/domain/event_messages/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -27,12 +27,12 @@ h_message_bus/domain/request_messages/vector_query_collection_response_message.p
 h_message_bus/domain/request_messages/vector_read_metadata_request_message.py,sha256=P9t6pcynS112F9_Jmbvw8wlfXhs5NOnV48SUVq92xt8,873
 h_message_bus/domain/request_messages/vector_read_metadata_response_message.py,sha256=G2TDV_lbUM7RVrvEaYQAP-3I36J0pqLXqxnqwVBMWhk,1648
 h_message_bus/domain/request_messages/vector_save_request_message.py,sha256=NvRdSUtE-X3sQTfJg7_QLic3PtbpgYUH2lQrNSC1ggg,2115
-h_message_bus/domain/request_messages/web_get_docs_request_message.py,sha256=9Pw-szMBVF7yuJGSZO3tFoiGk9Kf_sz5Uv5JDVbPEVA,1324
+h_message_bus/domain/request_messages/web_get_docs_request_message.py,sha256=MkJySkRBlQ2CacMoPCd3FwXQt1tVVFdrZKxsA6yMsrk,1576
 h_message_bus/domain/request_messages/web_search_request_message.py,sha256=ZoB4idrFEs7HQ6IRxmwKrrfHWe3GlGF4LuOK68K5NEM,1000
 h_message_bus/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h_message_bus/infrastructure/nats_client_repository.py,sha256=lAapNaL0rqBr-9KaHcexuo5XLmSWRWspkJWXS9dMqPU,3371
 h_message_bus/infrastructure/nats_config.py,sha256=Yzqqd1bCfmUv_4FOnA1dvqIpakzV0BUL2_nXQcndWvo,1304
-h_message_bus-0.0.20.dist-info/METADATA,sha256=N09DwBVYf3hV1AjK_A9iS6q7FsAz-CvDDcojyKzEigk,8834
-h_message_bus-0.0.20.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-h_message_bus-0.0.20.dist-info/top_level.txt,sha256=BArjhm_lwFR9yJJEIf-LT_X64psuLkXFdbpQRJUreFE,23
-h_message_bus-0.0.20.dist-info/RECORD,,
+h_message_bus-0.0.22.dist-info/METADATA,sha256=EwONbFsl__2n8Yv_5q2_Qyx7JguhaTqt4yrp2Yghb_M,8834
+h_message_bus-0.0.22.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+h_message_bus-0.0.22.dist-info/top_level.txt,sha256=BArjhm_lwFR9yJJEIf-LT_X64psuLkXFdbpQRJUreFE,23
+h_message_bus-0.0.22.dist-info/RECORD,,