h-message-bus 0.0.13__py3-none-any.whl → 0.0.14__py3-none-any.whl

h_message_bus/domain/models/vector_query_answer.py

@@ -0,0 +1,22 @@
+from dataclasses import dataclass
+
+
+@dataclass
+class VectorQueryAnswer:
+    text: str
+    dimension: str
+
+    @classmethod
+    def from_dict(cls, data: dict) -> 'VectorQueryAnswer':
+        """Create an instance from a dictionary."""
+        return cls(
+            text=data.get('text', ''),
+            dimension=data.get('dimension', ''),
+        )
+
+    def to_dict(self) -> dict:
+        """Convert instance to a dictionary."""
+        return {
+            'text': self.text,
+            'dimension': self.dimension,
+        }

h_message_bus/domain/request_messages/vector_query_collection_response_message.py

@@ -1,6 +1,8 @@
-from typing import Type, TypeVar, Dict, Any
+import json
+from typing import Type, TypeVar, Dict, Any, List
 
 from ..models.request_message_topic import RequestMessageTopic
+from ..models.vector_query_answer import VectorQueryAnswer
 from ...domain.models.hai_message import HaiMessage
 
 T = TypeVar('T', bound='HaiMessage')
@@ -14,32 +16,32 @@ class VectorQueryCollectionResponseMessage(HaiMessage):
         return super().create(topic=topic, payload=payload)
 
     @classmethod
-    def create_message(cls, results: [str], dimensions: [str]) -> 'VectorQueryCollectionResponseMessage':
+    def create_message(cls, result: List[dict[str, str]]) -> 'VectorQueryCollectionResponseMessage':
         """Create a message requesting Twitter user data"""
+        result_list = json.dumps(result)
         return cls.create(
             topic=RequestMessageTopic.AI_VECTORS_QUERY_RESPONSE,
             payload={
-                "dimensions": dimensions,
-                "results": results
+                "result": result_list,
             },
         )
 
     @property
-    def dimensions(self) -> list[str]:
-        """Returns the dimensions from the message payload"""
-        return self.payload.get("dimensions", [])
-
-    @property
-    def results(self) -> list[str]:
+    def results(self) -> list[VectorQueryAnswer]:
         """Returns the results from the message payload"""
-        return self.payload.get("results", [])
+        results = self.payload.get("result", "")
+        meta_dict = json.loads(results)
+        response_list = []
+        for meta in meta_dict:
+            response_list.append(VectorQueryAnswer.from_dict(meta))
+        return response_list
 
     @classmethod
     def from_hai_message(cls, message: HaiMessage) -> 'VectorQueryCollectionResponseMessage':
         # Extract necessary fields from the message payload
         payload = message.payload
-
+        result = payload.get("result", "")
+        result_list = json.loads(result)
         return cls.create_message(
-            dimensions=payload.get('dimensions', []),
-            results=payload.get('results', [])
+           result=result_list
         )

{h_message_bus-0.0.13.dist-info → h_message_bus-0.0.14.dist-info}/METADATA

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

{h_message_bus-0.0.13.dist-info → h_message_bus-0.0.14.dist-info}/RECORD

@@ -15,18 +15,19 @@ h_message_bus/domain/models/event_message_topic.py,sha256=fSjRMBwMD2RE9GBUud38XM
 h_message_bus/domain/models/hai_message.py,sha256=b5CfX7hi5uNq77IVnZzEi9iotc4b_U2MNYwV6JY7JcU,2146
 h_message_bus/domain/models/request_message_topic.py,sha256=5wCJzOFQjHG5bwlNbQzmMAFpVLZYPdPcKTXQMsAjC5M,1132
 h_message_bus/domain/models/vector_collection_metadata.py,sha256=fSy3ZQ-zkYISTpc-_M6XWf1JkufVAlzlS2ZT60TzA2s,891
+h_message_bus/domain/models/vector_query_answer.py,sha256=V1JR5Di7kIVbRF_eq2uPOhDnkZpQvo9pxD41V7MNV7Q,551
 h_message_bus/domain/request_messages/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h_message_bus/domain/request_messages/twitter_get_user_request_message.py,sha256=BX-ytvnEzL9u1Yks5iXIx60Hmv3UP7asOG2uYA0mfgE,1212
 h_message_bus/domain/request_messages/twitter_get_user_response_message.py,sha256=qO1cLtEmZyddv5pXHNgpnpO5t3Oex8cOG0hjGhOfru8,3302
 h_message_bus/domain/request_messages/vector_query_collection_request_message.py,sha256=6zne_IFs4aHZ_hJqbXGAaNU8PMilSVKSAg5G_jzane0,1803
-h_message_bus/domain/request_messages/vector_query_collection_response_message.py,sha256=93Lk-rPQNJJB1kq4F3FfbghWgkPhI9YX4AwSaxgmzcM,1637
+h_message_bus/domain/request_messages/vector_query_collection_response_message.py,sha256=8TtLrgxnQj3w7OqEzegykCfk2aobGVJySxiWRgbIfeQ,1771
 h_message_bus/domain/request_messages/vector_read_metadata_request_message.py,sha256=l321y6ffLOvGESRDalBXtzcOG4qq_iy3a_j4uZ_R5KY,897
 h_message_bus/domain/request_messages/vector_read_metadata_response_message.py,sha256=ecsSiE-0Uar3tqDtwvPGyu352nuW9FDIqwAjQvnPHDI,1651
 h_message_bus/domain/request_messages/vector_save_request_message.py,sha256=Jd52Kc0OLsPLphfZtTesK6J1KqctC7Vm_0lmVxd99Ss,2098
 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.13.dist-info/METADATA,sha256=IPNmbsRnLhcY5N5AvWZWgs8aKC_kuLnJrKh94tkKdxQ,8834
-h_message_bus-0.0.13.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-h_message_bus-0.0.13.dist-info/top_level.txt,sha256=BArjhm_lwFR9yJJEIf-LT_X64psuLkXFdbpQRJUreFE,23
-h_message_bus-0.0.13.dist-info/RECORD,,
+h_message_bus-0.0.14.dist-info/METADATA,sha256=JZp3eFWrSFtJ7GVTvRQPKjEFw6WV5b5cWgkIuucwjsE,8834
+h_message_bus-0.0.14.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+h_message_bus-0.0.14.dist-info/top_level.txt,sha256=BArjhm_lwFR9yJJEIf-LT_X64psuLkXFdbpQRJUreFE,23
+h_message_bus-0.0.14.dist-info/RECORD,,