diff --git a/examples/experimental/llm_agent/llm_agent.toml b/examples/experimental/llm_agent/llm_agent.toml index 68f9ff92..8972b7d5 100644 --- a/examples/experimental/llm_agent/llm_agent.toml +++ b/examples/experimental/llm_agent/llm_agent.toml @@ -45,4 +45,4 @@ node_class = "chat_print" "Agent:Runtime" = "agent_action" [nodes.node_args] -env_agents = ["Jack", "Jane"] \ No newline at end of file +env_agents = ["Jack", "Jane"] diff --git a/sotopia/experimental/nodes/chat_print_node.py b/sotopia/experimental/nodes/chat_print_node.py index efe4bcaa..a19c887e 100644 --- a/sotopia/experimental/nodes/chat_print_node.py +++ b/sotopia/experimental/nodes/chat_print_node.py @@ -11,7 +11,7 @@ from rich.logging import RichHandler from aact import NodeFactory -from aact.nodes import PrintNode +from aact.nodes import PrintNode console = Console() @@ -72,7 +72,11 @@ def convert_to_sentence(self, data: Dict[str, Any], agent_name: str) -> None: except Exception as e: console.print( Panel( - RichText(f"Error processing write action: {e}", style="bold red", justify="center"), + RichText( + f"Error processing write action: {e}", + style="bold red", + justify="center", + ), title="Error", expand=False, border_style="red", @@ -240,4 +244,4 @@ async def write_to_screen(self) -> None: else: print("Invalid data structure:", data) - await self.output.flush() \ No newline at end of file + await self.output.flush()