:gem: [Feature] IdleOutputer: Start to customize message output

conversations/conversation_connector.py

@@ -8,7 +8,7 @@ from networks import (
     ChathubRequestPayloadConstructor,
     ConversationRequestHeadersConstructor,
 )
-from networks import MessageParser
+from networks import MessageParser, IdleOutputer
 from utils.logger import logger
 
 http_proxy = "http://localhost:11111"  # Replace with yours
@@ -80,7 +80,7 @@ class ConversationConnector:
     async def stream_chat(self, prompt=""):
         await self.connect()
         await self.send_chathub_request(prompt)
-        message_parser = MessageParser()
+        message_parser = MessageParser(outputer=IdleOutputer())
         while not self.wss.closed:
             response_lines_str = await self.wss.receive_str()
             if isinstance(response_lines_str, str):
@@ -110,6 +110,6 @@ class ConversationConnector:
                 # Stream: Heartbeat Signal
                 elif data.get("type") == 6:
                     continue
-                # Stream: Not Monitored
+                # Stream: Not Implemented
                 else:
                     continue

networks/__init__.py

@@ -3,4 +3,5 @@ from .cookies_constructor import CookiesConstructor
 from .conversation_request_headers_constructor import (
     ConversationRequestHeadersConstructor,
 )
+from .message_outputer import IdleOutputer
 from .message_parser import MessageParser

networks/message_outputer.py

@@ -0,0 +1,3 @@
+class IdleOutputer:
+    def output(self, message_content=None, message_type=None):
+        pass

networks/message_parser.py

@@ -1,9 +1,11 @@
 from utils.logger import logger
+from networks import IdleOutputer
 
 
 class MessageParser:
-    def __init__(self) -> None:
+    def __init__(self, outputer=IdleOutputer()):
         self.delta_content_pointer = 0
+        self.outputer = outputer
 
     def parse(self, data):
         arguments = data["arguments"][0]
@@ -18,6 +20,7 @@ class MessageParser:
                     content = message["adaptiveCards"][0]["body"][0]["text"]
                     delta_content = content[self.delta_content_pointer :]
                     logger.line(delta_content, end="")
+                    self.outputer.output(delta_content, message_type="Completions")
                     self.delta_content_pointer = len(content)
                     # Message: Suggested Questions
                     if message.get("suggestedResponses"):
@@ -25,10 +28,17 @@ class MessageParser:
                         for suggestion in message.get("suggestedResponses"):
                             suggestion_text = suggestion.get("text")
                             logger.file(f"- {suggestion_text}")
+                        self.outputer.output(
+                            message.get("suggestedResponses"),
+                            message_type="Suggestions",
+                        )
                 # Message: Search Query
                 elif message_type in ["InternalSearchQuery"]:
                     message_hidden_text = message["hiddenText"]
                     logger.note(f"\n[Searching: [{message_hidden_text}]]")
+                    self.outputer.output(
+                        message_hidden_text, message_type="InternalSearchQuery"
+                    )
                 # Message: Internal Search Results
                 elif message_type in ["InternalSearchResult"]:
                     logger.note("[Analyzing search results ...]")
@@ -36,9 +46,17 @@ class MessageParser:
                 elif message_type in ["InternalLoaderMessage"]:
                     # logger.note("[Generating answers ...]\n")
                     pass
+                # Message: Internal thoughts, such as "I will generate my response to the user message"
+                elif message_type in ["Internal"]:
+                    pass
+                # Message: Internal Action Marker, no value
+                elif message_type in ["InternalActionMarker"]:
+                    continue
                 # Message: Render Cards for Webpages
                 elif message_type in ["RenderCardRequest"]:
                     continue
+                elif message_type in ["ChatName"]:
+                    continue
                 # Message: Not Implemented
                 else:
                     raise NotImplementedError(