tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

fix: removed the panel in the output so that the user won't have unnecessary pane borders in the copied content (#109)

Browse Source 
Co-authored-by: Bradley Axen <baxen@squareup.com>
This commit is contained in:
Lifei Zhou
2024-10-04 07:01:01 +10:00
committed by GitHub
parent 8f671e7273
commit 9da4881006
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/goose/toolkit/developer.py
+11 -6
View File
@@ -10,12 +10,14 @@ from goose.toolkit.base import Toolkit, tool
from goose.toolkit.utils import get_language, render_template from goose.toolkit.utils import get_language, render_template
from goose.utils.ask import ask_an_ai from goose.utils.ask import ask_an_ai
from goose.utils.check_shell_command import is_dangerous_command from goose.utils.check_shell_command import is_dangerous_command
from rich import box
from rich.markdown import Markdown from rich.markdown import Markdown
from rich.panel import Panel
from rich.prompt import Confirm from rich.prompt import Confirm
from rich.table import Table from rich.table import Table
from rich.text import Text from rich.text import Text
from rich.rule import Rule
RULESTYLE = "bold"
RULEPREFIX = f"[{RULESTYLE}]───[/] "
def keep_unsafe_command_prompt(command: str) -> bool: def keep_unsafe_command_prompt(command: str) -> bool:
@@ -124,7 +126,8 @@ class Developer(Toolkit):
{after} {after}
``` ```
""" """
self.notifier.log(Panel.fit(Markdown(output), title=path)) self.notifier.log(Rule(RULEPREFIX + path, style=RULESTYLE, align="left"))
self.notifier.log(Markdown(output))
return "Succesfully replaced before with after." return "Succesfully replaced before with after."
@tool @tool
@@ -136,7 +139,7 @@ class Developer(Toolkit):
""" """
language = get_language(path) language = get_language(path)
content = Path(path).expanduser().read_text() content = Path(path).expanduser().read_text()
self.notifier.log(Panel.fit(Markdown(f"```\ncat {path}\n```"), box=box.MINIMAL)) self.notifier.log(Markdown(f"```\ncat {path}\n```"))
# Record the last read timestamp # Record the last read timestamp
self.timestamps[path] = os.path.getmtime(path) self.timestamps[path] = os.path.getmtime(path)
return f"```{language}\n{content}\n```" return f"```{language}\n{content}\n```"
@@ -155,7 +158,8 @@ class Developer(Toolkit):
if you need to run more than one at a time if you need to run more than one at a time
""" """
# Log the command being executed in a visually structured format (Markdown). # Log the command being executed in a visually structured format (Markdown).
self.notifier.log(Panel.fit(Markdown(f"```bash\n{command}\n```"), title="shell")) self.notifier.log(Rule(RULEPREFIX + "shell", style=RULESTYLE, align="left"))
self.notifier.log(Markdown(f"```bash\n{command}\n```"))
if is_dangerous_command(command): if is_dangerous_command(command):
# Stop the notifications so we can prompt # Stop the notifications so we can prompt
@@ -262,7 +266,8 @@ class Developer(Toolkit):
# Log the content that will be written to the file # Log the content that will be written to the file
# .log` method is used here to log the command execution in the application's UX # .log` method is used here to log the command execution in the application's UX
# this method is dynamically attached to functions in the Goose framework # this method is dynamically attached to functions in the Goose framework
self.notifier.log(Panel.fit(Markdown(md), title=path)) self.notifier.log(Rule(RULEPREFIX + path, style=RULESTYLE, align="left"))
self.notifier.log(Markdown(md))
_path = Path(path) _path = Path(path)
if path in self.timestamps: if path in self.timestamps: