Spaces:
Paused
Paused
:zap: [Enhance] Cleanse console outputs
Browse files- conversation_connector.py +4 -8
- message_parser.py +3 -2
conversation_connector.py
CHANGED
@@ -88,33 +88,29 @@ class ConversationConnector:
|
|
88 |
async def stream_chat(self, prompt=""):
|
89 |
await self.init_wss_connection()
|
90 |
await self.send_chathub_request(prompt)
|
91 |
-
|
92 |
message_parser = MessageParser()
|
93 |
while not self.wss.closed:
|
94 |
response_lines_str = await self.wss.receive_str()
|
95 |
-
|
96 |
if isinstance(response_lines_str, str):
|
97 |
response_lines = response_lines_str.split("\x1e")
|
98 |
else:
|
99 |
continue
|
100 |
-
|
101 |
for line in response_lines:
|
102 |
if not line:
|
103 |
continue
|
104 |
-
|
105 |
data = json.loads(line)
|
106 |
-
|
107 |
# Stream: Meaningful Messages
|
108 |
if data.get("type") == 1:
|
109 |
message_parser.parse(data)
|
110 |
# Stream: List of all messages in the whole conversation
|
111 |
elif data.get("type") == 2:
|
112 |
if data.get("item"):
|
113 |
-
item = data.get("item")
|
114 |
-
logger.note("\n[Saving chat messages ...]")
|
|
|
115 |
# Stream: End of Conversation
|
116 |
elif data.get("type") == 3:
|
117 |
-
logger.success("[Finished]")
|
118 |
self.invocation_id += 1
|
119 |
await self.wss.close()
|
120 |
await self.aiohttp_session.close()
|
|
|
88 |
async def stream_chat(self, prompt=""):
|
89 |
await self.init_wss_connection()
|
90 |
await self.send_chathub_request(prompt)
|
|
|
91 |
message_parser = MessageParser()
|
92 |
while not self.wss.closed:
|
93 |
response_lines_str = await self.wss.receive_str()
|
|
|
94 |
if isinstance(response_lines_str, str):
|
95 |
response_lines = response_lines_str.split("\x1e")
|
96 |
else:
|
97 |
continue
|
|
|
98 |
for line in response_lines:
|
99 |
if not line:
|
100 |
continue
|
|
|
101 |
data = json.loads(line)
|
|
|
102 |
# Stream: Meaningful Messages
|
103 |
if data.get("type") == 1:
|
104 |
message_parser.parse(data)
|
105 |
# Stream: List of all messages in the whole conversation
|
106 |
elif data.get("type") == 2:
|
107 |
if data.get("item"):
|
108 |
+
# item = data.get("item")
|
109 |
+
# logger.note("\n[Saving chat messages ...]")
|
110 |
+
pass
|
111 |
# Stream: End of Conversation
|
112 |
elif data.get("type") == 3:
|
113 |
+
logger.success("\n[Finished]")
|
114 |
self.invocation_id += 1
|
115 |
await self.wss.close()
|
116 |
await self.aiohttp_session.close()
|
message_parser.py
CHANGED
@@ -23,7 +23,7 @@ class MessageParser:
|
|
23 |
self.delta_content_pointer = len(content)
|
24 |
# Message: Suggested Questions
|
25 |
if message.get("suggestedResponses"):
|
26 |
-
logger.note("\
|
27 |
for suggestion in message.get("suggestedResponses"):
|
28 |
suggestion_text = suggestion.get("text")
|
29 |
logger.file(f"- {suggestion_text}")
|
@@ -36,7 +36,8 @@ class MessageParser:
|
|
36 |
logger.note("[Analyzing search results ...]")
|
37 |
# Message: Loader status, such as "Generating Answers"
|
38 |
elif message_type in ["InternalLoaderMessage"]:
|
39 |
-
logger.note("[Generating answers ...]\n")
|
|
|
40 |
# Message: Render Cards for Webpages
|
41 |
elif message_type in ["RenderCardRequest"]:
|
42 |
continue
|
|
|
23 |
self.delta_content_pointer = len(content)
|
24 |
# Message: Suggested Questions
|
25 |
if message.get("suggestedResponses"):
|
26 |
+
logger.note("\nSuggested Questions: ")
|
27 |
for suggestion in message.get("suggestedResponses"):
|
28 |
suggestion_text = suggestion.get("text")
|
29 |
logger.file(f"- {suggestion_text}")
|
|
|
36 |
logger.note("[Analyzing search results ...]")
|
37 |
# Message: Loader status, such as "Generating Answers"
|
38 |
elif message_type in ["InternalLoaderMessage"]:
|
39 |
+
# logger.note("[Generating answers ...]\n")
|
40 |
+
pass
|
41 |
# Message: Render Cards for Webpages
|
42 |
elif message_type in ["RenderCardRequest"]:
|
43 |
continue
|