Add error handling

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

mcp_remote_client.py

@@ -140,6 +140,13 @@ class MCPClientWrapper:
 
                 yield messages, gr.Textbox(value="")
 
+                if (
+                    messages[-1]["role"] == "assistant"
+                    and messages[-1]["content"]
+                    == "The LLM API is overloaded now, try again later..."
+                ):
+                    break
+
         with open("messages.log.jsonl", "a+") as fl:
             fl.write(json.dumps(dict(time=f"{datetime.now()}", messages=messages)))
 
@@ -174,6 +181,7 @@ class MCPClientWrapper:
                     "content": "The LLM API is overloaded now, try again later...",
                 }
             ]
+            # TODO: Add a retry mechanism
 
         result_messages = []
         partial_messages = []