Commit
·
d059c05
1
Parent(s):
92510e4
Make chat more responsive
Browse filesSigned-off-by: Aivin V. Solatorio <avsolatorio@gmail.com>
- mcp_remote_client.py +11 -2
mcp_remote_client.py
CHANGED
@@ -153,14 +153,23 @@ class MCPClientWrapper:
|
|
153 |
]
|
154 |
yield messages, gr.Textbox(value="")
|
155 |
else:
|
156 |
-
messages = history + [
|
|
|
|
|
|
|
|
|
|
|
|
|
157 |
|
158 |
yield messages, gr.Textbox(value="")
|
|
|
|
|
|
|
159 |
|
160 |
async for partial in self._process_query(message, history):
|
161 |
messages.extend(partial)
|
162 |
-
|
163 |
yield messages, gr.Textbox(value="")
|
|
|
164 |
|
165 |
if (
|
166 |
messages[-1]["role"] == "assistant"
|
|
|
153 |
]
|
154 |
yield messages, gr.Textbox(value="")
|
155 |
else:
|
156 |
+
messages = history + [
|
157 |
+
{"role": "user", "content": message},
|
158 |
+
{
|
159 |
+
"role": "assistant",
|
160 |
+
"content": "Ok, let me think about your query 🤔...",
|
161 |
+
},
|
162 |
+
]
|
163 |
|
164 |
yield messages, gr.Textbox(value="")
|
165 |
+
# simulate thinking with asyncio.sleep
|
166 |
+
await asyncio.sleep(0.1)
|
167 |
+
messages.pop(-1)
|
168 |
|
169 |
async for partial in self._process_query(message, history):
|
170 |
messages.extend(partial)
|
|
|
171 |
yield messages, gr.Textbox(value="")
|
172 |
+
await asyncio.sleep(0.05)
|
173 |
|
174 |
if (
|
175 |
messages[-1]["role"] == "assistant"
|