Merge pull request #2245 from usingtechnology/bugfix/issue-2205

./run_demo performance -c 1 --mediation --timing --trace-log

demo/runners/support/agent.py

@@ -663,7 +663,18 @@ def handle_output(self, *output, source: str = None, **kwargs):
             color = self.color or "fg:ansiblue"
         else:
             color = None
-        log_msg(*output, color=color, prefix=self.prefix_str, end=end, **kwargs)
+        try:
+            log_msg(*output, color=color, prefix=self.prefix_str, end=end, **kwargs)
+        except AssertionError as e:
+            if self.trace_enabled and self.trace_target == "log":
+                # when tracing to a log file,
+                # we hit an issue with the underlying prompt_toolkit.
+                # it attempts to output what is written by the log and can't find the
+                # correct terminal and throws an error. The trace log record does show
+                # in the terminal, so let's just ignore this error.
+                pass
+            else:
+                raise e
 
     def log(self, *msg, **kwargs):
         self.handle_output(*msg, **kwargs)

demo/runners/support/utils.py

@@ -113,7 +113,12 @@ def output_reader(handle, callback, *args, **kwargs):
     for line in iter(handle.readline, b""):
         if not line:
             break
-        run_in_terminal(functools.partial(callback, line, *args))
+        try:
+            run_in_terminal(functools.partial(callback, line, *args))
+        except AssertionError as e:
+            # see comment in DemoAgent.handle_output
+            # trace log and prompt_toolkit do not get along...
+            pass
 
 
 def log_msg(*msg, color="fg:ansimagenta", **kwargs):