h-message-bus 0.0.4__py3-none-any.whl → 0.0.6__py3-none-any.whl

h_message_bus/domain/hai_message.py

@@ -2,7 +2,9 @@ import json
 import uuid
 from dataclasses import dataclass
 from datetime import datetime
-from typing import Dict, Any
+from typing import Dict, Any, TypeVar
+
+T = TypeVar('T', bound='HaiMessage')
 
 
 @dataclass
@@ -35,7 +37,7 @@ class HaiMessage:
             self.timestamp = datetime.utcnow().isoformat()
 
     @classmethod
-    def create(cls, topic: str, payload: Dict[Any, Any]) -> 'HaiMessage':
+    def create(cls, topic: str, payload: Dict[Any, Any]) -> T:
         """Factory method to create a new domain message."""
         return cls(
             id=str(uuid.uuid4()),

h_message_bus/domain/topics.py

@@ -1,7 +1,7 @@
 from enum import Enum
 
 
-class Topic(str, Enum):
+class MessageTopic(str, Enum):
     """
     Represents a collection of predefined topics as an enumeration.
 
@@ -14,11 +14,20 @@ class Topic(str, Enum):
 
     """
     # AI to Telegram
-    AI_SEND_TG_CHAT_MESSAGE = "hai.ai.tg.chat.send"
+    AI_SEND_TG_CHAT_SEND = "hai.ai.tg.chat.send"
 
     # AI to vector database
     AI_VECTORS_SAVE = "hai.ai.vectors.save"
+
     AI_VECTORS_QUERY = "hai.ai.vectors.query"
+    AI_VECTORS_QUERY_RESPONSE = "hai.ai.vectors.query.response"
+
+    # AI to twitter
+    AI_TWITTER_GET_USER = "hai.ai.twitter.get.user"
+    AI_TWITTER_GET_USER_RESPONSE = "hai.ai.twitter.get.user.response"
+
 
     # TG to AI
-    TG_SEND_AI_CHAT_MESSAGE = "hai.tg.ai.chat.send"
+    TG_SEND_AI_CHAT_SEND = "hai.tg.ai.chat.send"
+
+

h_message_bus/domain/twitter_get_user_request_message.py

@@ -0,0 +1,23 @@
+from typing import Type, TypeVar, Dict, Any
+
+from .topics import MessageTopic
+from ..domain.hai_message import HaiMessage
+
+T = TypeVar('T', bound='HaiMessage')
+
+class TwitterGetUserRequestMessage(HaiMessage):
+    """Message to request Twitter user information"""
+
+    @classmethod
+    def create(cls: Type[T], topic: str, payload: Dict[Any, Any]) -> T:
+        """Create a message - inherited from HaiMessage"""
+        return super().create(topic=topic, payload=payload)
+
+    @classmethod
+    def create_message(cls, username: str) -> 'TwitterGetUserRequestMessage':
+        """Create a message requesting Twitter user data"""
+        return cls.create(
+            topic=MessageTopic.AI_TWITTER_GET_USER,
+            payload={"username": username},
+        )
+

h_message_bus/domain/twitter_get_user_response_message.py

@@ -0,0 +1,30 @@
+from typing import Type, TypeVar, Dict, Any
+
+from .topics import MessageTopic
+from ..domain.hai_message import HaiMessage
+
+T = TypeVar('T', bound='HaiMessage')
+
+class TwitterGetUserResponseMessage(HaiMessage):
+    """Response of Twitter user information request"""
+
+    @classmethod
+    def create(cls: Type[T], topic: str, payload: Dict[Any, Any]) -> T:
+        """Create a message - inherited from HaiMessage"""
+        return super().create(topic=topic, payload=payload)
+
+    @classmethod
+    def create_message(cls, user_id: str, screen_name: str, description: str, followers_count: str, like_count: str, is_verified: str, url: str, bio_urls: [str]) -> 'TwitterGetUserResponseMessage':
+        """Create a response message from Twitter user information"""
+        return cls.create(
+            topic=MessageTopic.AI_TWITTER_GET_USER_RESPONSE,
+            payload={
+                'id': user_id,
+                'screen_name': screen_name,
+                'description': description,
+                'followers_count': followers_count,
+                'like_count': like_count,
+                'is_verified': is_verified,
+                'url': url,
+                'bio_urls': bio_urls
+            })

h_message_bus/domain/vector_read_request_message.py

@@ -0,0 +1,26 @@
+from typing import Type, TypeVar, Dict, Any
+
+from .topics import MessageTopic
+from ..domain.hai_message import HaiMessage
+
+T = TypeVar('T', bound='HaiMessage')
+
+class VectorReadRequestMessage(HaiMessage):
+    """Message to read data from vector store"""
+
+    @classmethod
+    def create(cls: Type[T], topic: str, payload: Dict[Any, Any]) -> T:
+        """Create a message - inherited from HaiMessage"""
+        return super().create(topic=topic, payload=payload)
+
+    @classmethod
+    def create_message(cls, collection_name: str, query: str, top_n: str) -> 'VectorReadRequestMessage':
+        """Create a message requesting Twitter user data"""
+        return cls.create(
+            topic=MessageTopic.AI_VECTORS_QUERY,
+            payload={
+                "collection_name": collection_name,
+                "query": query,
+                "top_n": top_n
+            },
+        )

h_message_bus/domain/vector_read_response_message.py

@@ -0,0 +1,25 @@
+from typing import Type, TypeVar, Dict, Any
+
+from .topics import MessageTopic
+from ..domain.hai_message import HaiMessage
+
+T = TypeVar('T', bound='HaiMessage')
+
+class VectorReadResponseMessage(HaiMessage):
+    """Response Message from reading vector data"""
+
+    @classmethod
+    def create(cls: Type[T], topic: str, payload: Dict[Any, Any]) -> T:
+        """Create a message - inherited from HaiMessage"""
+        return super().create(topic=topic, payload=payload)
+
+    @classmethod
+    def create_message(cls, results: [str], dimensions: [str]) -> 'VectorReadResponseMessage':
+        """Create a message requesting Twitter user data"""
+        return cls.create(
+            topic=MessageTopic.AI_VECTORS_QUERY_RESPONSE,
+            payload={
+                "dimensions": dimensions,
+                "results": results
+            },
+        )

h_message_bus/domain/vector_save_request_message.py

@@ -0,0 +1,27 @@
+from typing import Type, TypeVar, Dict, Any
+
+from .topics import MessageTopic
+from ..domain.hai_message import HaiMessage
+
+T = TypeVar('T', bound='HaiMessage')
+
+class VectorSaveRequestMessage(HaiMessage):
+    """Message to data in vector store"""
+
+    @classmethod
+    def create(cls: Type[T], topic: str, payload: Dict[Any, Any]) -> T:
+        """Create a message - inherited from HaiMessage"""
+        return super().create(topic=topic, payload=payload)
+
+    @classmethod
+    def create_message(cls, collection_name: str, document_id: str, content: str, metadata: str) -> 'VectorSaveRequestMessage':
+        """Create a message requesting Twitter user data"""
+        return cls.create(
+            topic=MessageTopic.AI_VECTORS_SAVE,
+            payload={
+                "collection_name": collection_name,
+                "document_id": document_id,
+                "content": content,
+                "metadata": metadata
+            },
+        )

{h_message_bus-0.0.4.dist-info → h_message_bus-0.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: h_message_bus
-Version: 0.0.4
+Version: 0.0.6
 Summary: Message bus integration for HAI
 Author-email: shoebill <shoebill.hai@gmail.com>
 Classifier: Programming Language :: Python :: 3
@@ -8,7 +8,6 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
-Requires-Dist: python-dotenv~=1.1.0
 Requires-Dist: nats-py~=2.10.0
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"

{h_message_bus-0.0.4.dist-info → h_message_bus-0.0.6.dist-info}/RECORD

@@ -8,12 +8,17 @@ h_message_bus/application/message_processor.py,sha256=-I3yUtGQCxmw7McyCpJFMGfXFC
 h_message_bus/application/message_publisher.py,sha256=hraRMOjjsAZLQOg09V99TjOuBy9Wu2NUtka7-x63doQ,835
 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/hai_message.py,sha256=mODVLRx_hgrnK9dbw57MiFATAiRwsgQ2TAOTA7r5Fcc,2108
-h_message_bus/domain/topics.py,sha256=a3VfQuItmHMaHaDWf2N9DgU8ASUh3-EHHB12YIJPvNw,769
+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=qDyx0Mfm9jkjfk7pmMFu5Jln_lIsCYveSYbVzeVspuI,785
+h_message_bus/domain/twitter_get_user_response_message.py,sha256=YtQBdKZzEGRi2TSj5WW7qTPVLVjBWn9eAfn1UutK8WQ,1237
+h_message_bus/domain/vector_read_request_message.py,sha256=XqrWYYTJbEZAcimIwdKuz3N1xUaEwrB1WDOl3buQn_Y,908
+h_message_bus/domain/vector_read_response_message.py,sha256=shgNrwQN_wmAxPIPqgcv16svTMYNLtCZy2mJoWEdD7Q,872
+h_message_bus/domain/vector_save_request_message.py,sha256=L_pALwa-afqzyFGJ-ayVyzApeVNd920OVm6cf7eYyVc,978
 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.4.dist-info/METADATA,sha256=Jf2par702nSA2LQeoCenp71b77mdkpgHCZJKnT68GuQ,8870
-h_message_bus-0.0.4.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-h_message_bus-0.0.4.dist-info/top_level.txt,sha256=7feCw9I0yIIqsYsVzUMYvCCmUTOiGX1a5F2VbSkFscc,14
-h_message_bus-0.0.4.dist-info/RECORD,,
+h_message_bus-0.0.6.dist-info/METADATA,sha256=Q2WcM242VKQ90GoNGpC4HvVNlpdCQjD39okmk8CDouc,8833
+h_message_bus-0.0.6.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+h_message_bus-0.0.6.dist-info/top_level.txt,sha256=7feCw9I0yIIqsYsVzUMYvCCmUTOiGX1a5F2VbSkFscc,14
+h_message_bus-0.0.6.dist-info/RECORD,,