-
Notifications
You must be signed in to change notification settings - Fork 21
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: Adding Chat print node for pretty printing chat conversation be…
…tween agents (#250) * Adding OpenHands node * [autofix.ci] apply automated fixes * Added LLM Agent Node * [autofix.ci] apply automated fixes * removing openhands * [autofix.ci] apply automated fixes * Correcting mypy error in tick agent * Adding scene context for agents * [autofix.ci] apply automated fixes * Added chat print node * [autofix.ci] apply automated fixes * Fixing the runtime channel * Moving everything to examples since its not specific to sotopia * [autofix.ci] apply automated fixes * Adding some additional final checks * Adding scene context for agents * [autofix.ci] apply automated fixes * Renaming scene context to initial message * Adding the initial message channel * Adding scene context for agents * Added chat print node * [autofix.ci] apply automated fixes * Moving chat node to experiment * Renaming to interview openhands * Adding documentation * [autofix.ci] apply automated fixes * Updating the readme * [autofix.ci] apply automated fixes * feat: Initial Message for LLM agnets in Sotopia Aact (Experimental) (#249) * Adding OpenHands node * [autofix.ci] apply automated fixes * Added LLM Agent Node * [autofix.ci] apply automated fixes * removing openhands * [autofix.ci] apply automated fixes * Correcting mypy error in tick agent * Moving everything to examples since its not specific to sotopia * [autofix.ci] apply automated fixes * Adding some additional final checks * Adding scene context for agents * [autofix.ci] apply automated fixes * Renaming scene context to initial message * Adding the initial message channel * Renaming to interview openhands * Adding documentation * [autofix.ci] apply automated fixes * Updating the readme * [autofix.ci] apply automated fixes * Adding scene context for agents * [autofix.ci] apply automated fixes * Renaming scene context to initial message * Adding the initial message channel * Rebasing --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> * Adding scene context for agents * [autofix.ci] apply automated fixes * Renaming scene context to initial message * Adding scene context for agents * Added chat print node * [autofix.ci] apply automated fixes * Adding scene context for agents * [autofix.ci] apply automated fixes * Fixing the runtime channel * Moving chat node to experiment * Updating the toml file paths --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
- Loading branch information
1 parent
df95426
commit 8f98ad8
Showing
3 changed files
with
259 additions
and
9 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,247 @@ | ||
| import sys | ||
| import json | ||
| import logging | ||
| from typing import Dict, Any, Literal | ||
|
|
||
| from rich.console import Console | ||
| from rich.syntax import Syntax | ||
| from rich.panel import Panel | ||
| from rich.text import Text as RichText | ||
| from rich.align import Align | ||
| from rich.logging import RichHandler | ||
|
|
||
| from aact import NodeFactory | ||
| from aact.nodes import PrintNode | ||
|
|
||
| console = Console() | ||
|
|
||
| if sys.version_info >= (3, 11): | ||
| pass | ||
| else: | ||
| pass | ||
|
|
||
| FORMAT = "%(asctime)s - %(levelname)s - %(name)s - %(message)s" | ||
| logging.basicConfig( | ||
| level=logging.WARNING, | ||
| format=FORMAT, | ||
| datefmt="[%X]", | ||
| handlers=[RichHandler()], | ||
| ) | ||
|
|
||
|
|
||
| @NodeFactory.register("chat_print") | ||
| class ChatPrint(PrintNode): | ||
| def __init__(self, env_agents: list[str], *args: Any, **kwargs: Any): | ||
| super().__init__(*args, **kwargs) | ||
| self.env_agents: list[str] = env_agents | ||
|
|
||
| def convert_to_sentence(self, data: Dict[str, Any], agent_name: str) -> None: | ||
| if "action_type" in data: | ||
| action = data["action_type"] | ||
|
|
||
| # Unpack the two agents from self.env_agents | ||
| agent1, agent2 = self.env_agents | ||
|
|
||
| # Define color styles based on agent_name | ||
| name_color_map = { | ||
| agent1: "green", | ||
| agent2: "blue", | ||
| } | ||
|
|
||
| panel_style = name_color_map.get(agent_name, "white") | ||
|
|
||
| # Determine alignment based on agent name using self.env_agents | ||
| alignment: Literal["left", "center", "right"] = ( | ||
| "left" if agent_name == agent1 else "right" | ||
| ) | ||
|
|
||
| if action == "write": | ||
| try: | ||
| path = data["path"] | ||
| content = data["argument"] | ||
| syntax = self.determine_syntax(path, content) | ||
| combined_panel = Panel( | ||
| syntax, | ||
| title=f"{agent_name} writes to {path}", | ||
| expand=False, | ||
| border_style=panel_style, | ||
| title_align=alignment, | ||
| ) | ||
| aligned_panel = Align(combined_panel, align=alignment) | ||
| console.print(aligned_panel) | ||
| except Exception as e: | ||
| console.print( | ||
| Panel( | ||
| RichText( | ||
| f"Error processing write action: {e}", | ||
| style="bold red", | ||
| justify="center", | ||
| ), | ||
| title="Error", | ||
| expand=False, | ||
| border_style="red", | ||
| title_align="center", | ||
| ) | ||
| ) | ||
|
|
||
| elif action == "speak": | ||
| content = data["argument"] | ||
| panel_content = RichText(content, style="bold", justify="left") | ||
| panel = Panel( | ||
| panel_content, | ||
| title=f"{agent_name} speaks", | ||
| expand=False, | ||
| border_style=panel_style, | ||
| title_align=alignment, | ||
| ) | ||
| aligned_panel = Align(panel, align=alignment) | ||
| console.print(aligned_panel) | ||
|
|
||
| elif action == "non-verbal": | ||
| content = data["argument"] | ||
| panel_content = RichText(content, style="bold", justify="left") | ||
| panel = Panel( | ||
| panel_content, | ||
| title=f"{agent_name} does", | ||
| expand=False, | ||
| border_style=panel_style, | ||
| title_align=alignment, | ||
| ) | ||
| aligned_panel = Align(panel, align=alignment) | ||
| console.print(aligned_panel) | ||
|
|
||
| elif action == "thought": | ||
| content = data["argument"] | ||
| panel_content = RichText(content, style="bold", justify="left") | ||
| panel = Panel( | ||
| panel_content, | ||
| title=f"{agent_name} thinks", | ||
| expand=False, | ||
| border_style=panel_style, | ||
| title_align=alignment, | ||
| ) | ||
| aligned_panel = Align(panel, align=alignment) | ||
| console.print(aligned_panel) | ||
|
|
||
| elif action == "browse": | ||
| url = data["argument"] | ||
| panel_content = RichText(url, style="bold", justify="center") | ||
| panel = Panel( | ||
| panel_content, | ||
| title=f"{agent_name} browses", | ||
| expand=False, | ||
| border_style=panel_style, | ||
| title_align=alignment, | ||
| ) | ||
| aligned_panel = Align(panel, align=alignment) | ||
| console.print(aligned_panel) | ||
|
|
||
| elif action == "browse_action": | ||
| command = data["argument"] | ||
| panel_content = RichText(command, style="bold", justify="center") | ||
| panel = Panel( | ||
| panel_content, | ||
| title=f"{agent_name} executes browser command", | ||
| expand=False, | ||
| border_style=panel_style, | ||
| title_align=alignment, | ||
| ) | ||
| aligned_panel = Align(panel, align=alignment) | ||
| console.print(aligned_panel) | ||
|
|
||
| elif action == "run": | ||
| command = data["argument"] | ||
| panel_content = RichText(command, style="bold", justify="center") | ||
| panel = Panel( | ||
| panel_content, | ||
| title=f"{agent_name} runs command", | ||
| expand=False, | ||
| border_style=panel_style, | ||
| title_align=alignment, | ||
| ) | ||
| aligned_panel = Align(panel, align=alignment) | ||
| console.print(aligned_panel) | ||
|
|
||
| elif action == "read": | ||
| path = data["path"] | ||
| panel_content = RichText( | ||
| f"Reading from {path}", style="bold", justify="center" | ||
| ) | ||
| panel = Panel( | ||
| panel_content, | ||
| title=f"{agent_name} reads", | ||
| expand=False, | ||
| border_style=panel_style, | ||
| title_align=alignment, | ||
| ) | ||
| aligned_panel = Align(panel, align=alignment) | ||
| console.print(aligned_panel) | ||
|
|
||
| elif action == "none": | ||
| return | ||
|
|
||
| else: | ||
| panel_content = RichText( | ||
| f"Action: {action}\n", style="bold", justify="center" | ||
| ) | ||
| for key, value in data.items(): | ||
| panel_content.append(f"{key.capitalize()}: {value}\n") | ||
| panel = Panel( | ||
| panel_content, | ||
| title=f"{agent_name} performs {action}", | ||
| expand=False, | ||
| border_style=panel_style, | ||
| title_align=alignment, | ||
| ) | ||
| aligned_panel = Align(panel, align=alignment) | ||
| console.print(aligned_panel) | ||
| else: | ||
| console.print( | ||
| Panel( | ||
| RichText("Invalid data format", style="bold red", justify="center"), | ||
| title="Error", | ||
| expand=False, | ||
| border_style="red", | ||
| title_align="center", | ||
| ) | ||
| ) | ||
|
|
||
| def determine_syntax(self, path: str, content: str) -> Syntax: | ||
| """Determine the appropriate syntax highlighting based on the file extension.""" | ||
| if path.endswith(".html"): | ||
| return Syntax(content, "html", theme="monokai", line_numbers=True) | ||
| elif path.endswith(".py"): | ||
| return Syntax( | ||
| content, | ||
| "python", | ||
| theme="monokai", | ||
| line_numbers=True, | ||
| ) | ||
| elif path.endswith(".js"): | ||
| return Syntax( | ||
| content, | ||
| "javascript", | ||
| theme="monokai", | ||
| line_numbers=True, | ||
| ) | ||
| elif path.endswith(".css"): | ||
| return Syntax(content, "css", theme="monokai", line_numbers=True) | ||
| else: | ||
| return Syntax(content, "text", theme="monokai", line_numbers=True) | ||
|
|
||
| async def write_to_screen(self) -> None: | ||
| while self.output: | ||
| data_entry = await self.write_queue.get() | ||
|
|
||
| data = json.loads(data_entry.model_dump_json()) | ||
|
|
||
| if "data" in data and "agent_name" in data["data"]: | ||
| agent_name = data["data"]["agent_name"] | ||
| try: | ||
| self.convert_to_sentence(data["data"], agent_name) | ||
| except Exception as e: | ||
| print(f"Error in convert_to_sentence: {e}") | ||
| else: | ||
| print("Invalid data structure:", data) | ||
|
|
||
| await self.output.flush() |