h-message-bus 0.0.5__py3-none-any.whl → 0.0.7__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,27 @@
+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},
+        )
+
+    @property
+    def username(self) -> str:
+        """Get the username from the payload"""
+        return self.payload.get("username", "")

h_message_bus/domain/twitter_get_user_response_message.py

@@ -0,0 +1,71 @@
+from typing import Type, TypeVar, Dict, Any, List
+
+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
+            })
+
+    @property
+    def user_id(self) -> str:
+        """Get user ID"""
+        return self.payload.get('id', '')
+
+    @property
+    def screen_name(self) -> str:
+        """Get screen name"""
+        return self.payload.get('screen_name', '')
+
+    @property
+    def description(self) -> str:
+        """Get user description"""
+        return self.payload.get('description', '')
+
+    @property
+    def followers_count(self) -> str:
+        """Get followers count"""
+        return self.payload.get('followers_count', '')
+
+    @property
+    def like_count(self) -> str:
+        """Get like count"""
+        return self.payload.get('like_count', '')
+
+    @property
+    def is_verified(self) -> str:
+        """Get verification status"""
+        return self.payload.get('is_verified', '')
+
+    @property
+    def url(self) -> str:
+        """Get user URL"""
+        return self.payload.get('url', '')
+
+    @property
+    def bio_urls(self) -> List[str]:
+        """Get URLs from user bio"""
+        return self.payload.get('bio_urls', [])
+

h_message_bus/domain/vector_read_request_message.py

@@ -0,0 +1,41 @@
+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
+            },
+        )
+
+    @property
+    def collection_name(self) -> str:
+        """Get the collection name from the message payload"""
+        return self.payload.get("collection_name", "")
+
+    @property
+    def query(self) -> str:
+        """Get the query from the message payload"""
+        return self.payload.get("query", "")
+
+    @property
+    def top_n(self) -> str:
+        """Get the top_n value from the message payload"""
+        return self.payload.get("top_n", "")

h_message_bus/domain/vector_read_response_message.py

@@ -0,0 +1,35 @@
+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
+            },
+        )
+
+    @property
+    def dimensions(self) -> list[str]:
+        """Returns the dimensions from the message payload"""
+        return self.payload.get("dimensions", [])
+
+    @property
+    def results(self) -> list[str]:
+        """Returns the results from the message payload"""
+        return self.payload.get("results", [])

h_message_bus/domain/vector_save_request_message.py

@@ -0,0 +1,47 @@
+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
+            },
+        )
+
+    @property
+    def collection_name(self) -> str:
+        """Get the collection name from payload"""
+        return self.payload.get("collection_name")
+
+    @property
+    def document_id(self) -> str:
+        """Get the document ID from payload"""
+        return self.payload.get("document_id")
+
+    @property
+    def content(self) -> str:
+        """Get the content from payload"""
+        return self.payload.get("content")
+
+    @property
+    def metadata(self) -> str:
+        """Get the metadata from payload"""
+        return self.payload.get("metadata")

{h_message_bus-0.0.5.dist-info → h_message_bus-0.0.7.dist-info}/METADATA

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

{h_message_bus-0.0.5.dist-info → h_message_bus-0.0.7.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=n6_v7s57AqrIErINcwVDnvClPaFolKZ99o4tbG3B4gI,930
+h_message_bus/domain/twitter_get_user_response_message.py,sha256=LnLHZU-_7WUtE81iSxeyErCVGVfbS89biIFpdbND4Lw,2318
+h_message_bus/domain/vector_read_request_message.py,sha256=8LoJhsEdwIIceS8bL05gDcFxMUySK5CzRTgj_TmEeqg,1384
+h_message_bus/domain/vector_read_response_message.py,sha256=sPj-jVifXW8hPkM4lIdcJauQY5RJ4H88iD6DVNxwnQ0,1207
+h_message_bus/domain/vector_save_request_message.py,sha256=jDSzK93M9bOhqtsJbmzLyVsQNOqqXpQIWySC5jyBaTo,1563
 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.5.dist-info/METADATA,sha256=WzARheZw0LIDmuCiOqrZFtBx9l6e4tzVEalzVdj2CY8,8833
-h_message_bus-0.0.5.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-h_message_bus-0.0.5.dist-info/top_level.txt,sha256=7feCw9I0yIIqsYsVzUMYvCCmUTOiGX1a5F2VbSkFscc,14
-h_message_bus-0.0.5.dist-info/RECORD,,
+h_message_bus-0.0.7.dist-info/METADATA,sha256=5E8RWWFnI1m-dmXFhRyuHhgZ9v24QHvheEE8CSg5tf4,8833
+h_message_bus-0.0.7.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+h_message_bus-0.0.7.dist-info/top_level.txt,sha256=7feCw9I0yIIqsYsVzUMYvCCmUTOiGX1a5F2VbSkFscc,14
+h_message_bus-0.0.7.dist-info/RECORD,,