fix mathproxyagent bug (#1124)

microsoft · Jul 13, 2023 · ac7c413 · ac7c413
1 parent 3527e8a
commit ac7c413
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/flaml/autogen/agent/math_user_proxy_agent.py b/flaml/autogen/agent/math_user_proxy_agent.py
@@ -314,6 +314,9 @@ def auto_reply(self, message, sender, default_reply=""):
                     output, is_success = self._execute_one_python_code(code)
                 elif lang == "wolfram":
                     output, is_success = self._execute_one_wolfram_query(code)
+                else:
+                    output = "Error: Unknown language."
+                    is_success = False
 
                 reply += output + "\n"
                 if not is_success: