Spaces:
Runtime error
Runtime error
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"
|