h-message-bus 0.0.22__tar.gz → 0.0.24__tar.gz

{h_message_bus-0.0.22 → h_message_bus-0.0.24}/PKG-INFO

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

{h_message_bus-0.0.22 → h_message_bus-0.0.24}/pyproject.toml

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

{h_message_bus-0.0.22 → h_message_bus-0.0.24}/src/h_message_bus/adapters/nats_publisher_adapter.py

@@ -1,3 +1,4 @@
+import asyncio
 from typing import Optional
 
 from ..application.message_publisher import MessagePublisher
@@ -23,14 +24,14 @@ class NatsPublisherAdapter(MessagePublisher):
     def __init__(self, nats_client: NatsClientRepository):
         self.nats_client = nats_client
 
-    async def publish(self, message: HaiMessage):
+    async def publish(self, message: HaiMessage, wait_time: float = 0.1):
         """Publish a message to NATS."""
         try:
-
             await self.nats_client.publish(
                 message.topic,
                 message.to_json().encode()
             )
+            await asyncio.sleep(wait_time)
         except Exception as e:
             print(f"Failed to publish message: {e}")
             return

{h_message_bus-0.0.22 → h_message_bus-0.0.24}/src/h_message_bus/adapters/nats_subscriber_adapter.py

@@ -1,4 +1,4 @@
-from typing import Dict
+from typing import Dict, Any
 
 from ..adapters.nats_message_adapter import NatsMessageAdapter
 from ..application.message_processor import MessageProcessor
@@ -30,7 +30,7 @@ class NatsSubscriberAdapter(MessageSubscriber):
         self.subscriptions = []
         self.adapters: Dict[str, NatsMessageAdapter] = {}
 
-    async def subscribe(self, topic: str, processor: MessageProcessor) -> None:
+    async def subscribe(self, topic: str, processor: MessageProcessor) -> Any:
         """Subscribe to a topic with a message handler."""
         # Create an adapter for this use case
         message_adapter = NatsMessageAdapter(processor)
@@ -43,6 +43,7 @@ class NatsSubscriberAdapter(MessageSubscriber):
         )
 
         self.subscriptions.append(subscription)
+        return subscription
 
     async def unsubscribe_all(self) -> None:
         """Unsubscribe from all subscriptions."""

{h_message_bus-0.0.22 → h_message_bus-0.0.24}/src/h_message_bus/domain/request_messages/vector_save_request_message.py

@@ -14,12 +14,13 @@ class VectorSaveRequestMessage(HaiMessage):
         return super().create(topic=topic, payload=payload)
 
     @classmethod
-    def create_message(cls, collection_name: str, document_id: str, content: str, metadata: dict[str,str]) -> 'VectorSaveRequestMessage':
+    def create_message(cls, collection_name: str, collection_metadata: dict[str,str], document_id: str, content: str, metadata: dict[str,str]) -> 'VectorSaveRequestMessage':
         """Create a message requesting Twitter user data"""
         return cls.create(
             topic=RequestMessageTopic.VECTORS_SAVE,
             payload={
                 "collection_name": collection_name,
+                "collection_metadata": collection_metadata,
                 "document_id": document_id,
                 "content": content,
                 "metadata": metadata
@@ -31,6 +32,11 @@ class VectorSaveRequestMessage(HaiMessage):
         """Get the collection name from the payload"""
         return self.payload.get("collection_name")
 
+    @property
+    def collection_metadata(self) -> str:
+        """Get the collection name from the payload"""
+        return self.payload.get("collection_metadata")
+
     @property
     def document_id(self) -> str:
         """Get the document ID from the payload"""
@@ -53,6 +59,7 @@ class VectorSaveRequestMessage(HaiMessage):
 
         return cls.create_message(
             collection_name=payload.get("collection_name", ''),
+            collection_metadata=payload.get("collection_metadata", {}),
             document_id=payload.get("document_id", ''),
             content=payload.get("content", ''),
             metadata=payload.get("metadata", {})

{h_message_bus-0.0.22 → h_message_bus-0.0.24}/src/h_message_bus/infrastructure/nats_client_repository.py

@@ -1,3 +1,4 @@
+import logging
 from typing import Optional, Callable, Any
 
 import nats
@@ -30,11 +31,13 @@ class NatsClientRepository:
         self.client: NatsClient | None = None
 
         self.subscriptions = []
+        self.logger = logging.getLogger(__name__)
 
     async def connect(self) -> None:
         """Connect to NATS server."""
         if self.client and self.client.is_connected:
             return
+        self.logger.info(f"Connecting to NATS server at {self.config.server}")
 
         self.client = await nats.connect(
             servers=self.config.server,
@@ -50,7 +53,11 @@ class NatsClientRepository:
         if not self.client or not self.client.is_connected:
             await self.connect()
 
-        await self.client.publish(subject, payload)
+        try:
+            await self.client.publish(subject, payload)
+        except Exception as e:
+            print(f"Failed to publish message: {e}")
+            return
 
     async def subscribe(self, subject: str, callback: Callable) -> Any:
         """Subscribe to a subject with a callback."""

{h_message_bus-0.0.22 → h_message_bus-0.0.24}/src/h_message_bus.egg-info/PKG-INFO

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