avsolatorio commited on
Commit
d059c05
·
1 Parent(s): 92510e4

Make chat more responsive

Browse files

Signed-off-by: Aivin V. Solatorio <avsolatorio@gmail.com>

Files changed (1) hide show
  1. 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 + [{"role": "user", "content": message}]
 
 
 
 
 
 
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"