mirror of
https://github.com/TheR1D/shell_gpt.git
synced 2026-07-03 14:10:18 +02:00
Farkhod Sadykov
c48926a4a4
* Added --interaction that works with --shell option. * Changed shell integrations to use new --no-interaction option. * Moved shell integrations into dedicated file integration.py. * Changed --install-integration logic to install integrations without downloading sh scripts. * Removed validation for PROMPT argument, empty string by default. * Fixing an issue when sgpt is being called from non-interactive shell environments. * Fixed and optimised Dockerfile. * README.md improvements, and new feature examples. * New funny demo video.
180 lines
6.5 KiB
Python
180 lines
6.5 KiB
Python
import os
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
from sgpt.config import cfg
|
|
from sgpt.role import DefaultRoles, SystemRole
|
|
|
|
from .utils import app, cmd_args, comp_args, comp_chunks, runner
|
|
|
|
|
|
@patch("openai.resources.chat.Completions.create")
|
|
def test_shell(completion):
|
|
role = SystemRole.get(DefaultRoles.SHELL.value)
|
|
completion.return_value = comp_chunks("git commit -m test")
|
|
|
|
args = {"prompt": "make a commit using git", "--shell": True}
|
|
result = runner.invoke(app, cmd_args(**args))
|
|
|
|
completion.assert_called_once_with(**comp_args(role, args["prompt"]))
|
|
assert result.exit_code == 0
|
|
assert "git commit" in result.stdout
|
|
assert "[E]xecute, [D]escribe, [A]bort:" in result.stdout
|
|
|
|
|
|
@patch("openai.resources.chat.Completions.create")
|
|
def test_shell_stdin(completion):
|
|
completion.return_value = comp_chunks("ls -l | sort")
|
|
role = SystemRole.get(DefaultRoles.SHELL.value)
|
|
|
|
args = {"prompt": "Sort by name", "--shell": True}
|
|
stdin = "What is in current folder"
|
|
result = runner.invoke(app, cmd_args(**args), input=stdin)
|
|
|
|
expected_prompt = f"{stdin}\n\n{args['prompt']}"
|
|
completion.assert_called_once_with(**comp_args(role, expected_prompt))
|
|
assert result.exit_code == 0
|
|
assert "ls -l | sort" in result.stdout
|
|
assert "[E]xecute, [D]escribe, [A]bort:" in result.stdout
|
|
|
|
|
|
@patch("openai.resources.chat.Completions.create")
|
|
def test_describe_shell(completion):
|
|
completion.return_value = comp_chunks("lists the contents of a folder")
|
|
role = SystemRole.get(DefaultRoles.DESCRIBE_SHELL.value)
|
|
|
|
args = {"prompt": "ls", "--describe-shell": True}
|
|
result = runner.invoke(app, cmd_args(**args))
|
|
|
|
completion.assert_called_once_with(**comp_args(role, args["prompt"]))
|
|
assert result.exit_code == 0
|
|
assert "lists" in result.stdout
|
|
|
|
|
|
@patch("openai.resources.chat.Completions.create")
|
|
def test_describe_shell_stdin(completion):
|
|
completion.return_value = comp_chunks("lists the contents of a folder")
|
|
role = SystemRole.get(DefaultRoles.DESCRIBE_SHELL.value)
|
|
|
|
args = {"--describe-shell": True}
|
|
stdin = "What is in current folder"
|
|
result = runner.invoke(app, cmd_args(**args), input=stdin)
|
|
|
|
expected_prompt = f"{stdin}"
|
|
completion.assert_called_once_with(**comp_args(role, expected_prompt))
|
|
assert result.exit_code == 0
|
|
assert "lists" in result.stdout
|
|
|
|
|
|
@patch("os.system")
|
|
@patch("openai.resources.chat.Completions.create")
|
|
def test_shell_run_description(completion, system):
|
|
completion.side_effect = [comp_chunks("echo hello"), comp_chunks("prints hello")]
|
|
args = {"prompt": "echo hello", "--shell": True}
|
|
inputs = "__sgpt__eof__\nd\ne\n"
|
|
result = runner.invoke(app, cmd_args(**args), input=inputs)
|
|
shell = os.environ.get("SHELL", "/bin/sh")
|
|
system.assert_called_once_with(f"{shell} -c 'echo hello'")
|
|
assert result.exit_code == 0
|
|
assert "echo hello" in result.stdout
|
|
assert "prints hello" in result.stdout
|
|
|
|
|
|
@patch("openai.resources.chat.Completions.create")
|
|
def test_shell_chat(completion):
|
|
completion.side_effect = [comp_chunks("ls"), comp_chunks("ls | sort")]
|
|
role = SystemRole.get(DefaultRoles.SHELL.value)
|
|
chat_name = "_test"
|
|
chat_path = Path(cfg.get("CHAT_CACHE_PATH")) / chat_name
|
|
chat_path.unlink(missing_ok=True)
|
|
|
|
args = {"prompt": "list folder", "--shell": True, "--chat": chat_name}
|
|
result = runner.invoke(app, cmd_args(**args))
|
|
assert result.exit_code == 0
|
|
assert "ls" in result.stdout
|
|
assert chat_path.exists()
|
|
|
|
args["prompt"] = "sort by name"
|
|
result = runner.invoke(app, cmd_args(**args))
|
|
assert result.exit_code == 0
|
|
assert "ls | sort" in result.stdout
|
|
|
|
expected_messages = [
|
|
{"role": "system", "content": role.role},
|
|
{"role": "user", "content": "list folder"},
|
|
{"role": "assistant", "content": "ls"},
|
|
{"role": "user", "content": "sort by name"},
|
|
{"role": "assistant", "content": "ls | sort"},
|
|
]
|
|
expected_args = comp_args(role, "", messages=expected_messages)
|
|
completion.assert_called_with(**expected_args)
|
|
assert completion.call_count == 2
|
|
|
|
args["--code"] = True
|
|
result = runner.invoke(app, cmd_args(**args))
|
|
assert result.exit_code == 2
|
|
assert "Error" in result.stdout
|
|
chat_path.unlink()
|
|
# TODO: Shell chat can be recalled without --shell option.
|
|
|
|
|
|
@patch("os.system")
|
|
@patch("openai.resources.chat.Completions.create")
|
|
def test_shell_repl(completion, mock_system):
|
|
completion.side_effect = [comp_chunks("ls"), comp_chunks("ls | sort")]
|
|
role = SystemRole.get(DefaultRoles.SHELL.value)
|
|
chat_name = "_test"
|
|
chat_path = Path(cfg.get("CHAT_CACHE_PATH")) / chat_name
|
|
chat_path.unlink(missing_ok=True)
|
|
|
|
args = {"--repl": chat_name, "--shell": True}
|
|
inputs = ["__sgpt__eof__", "list folder", "sort by name", "e", "exit()"]
|
|
result = runner.invoke(app, cmd_args(**args), input="\n".join(inputs))
|
|
shell = os.environ.get("SHELL", "/bin/sh")
|
|
mock_system.called_once_with(f"{shell} -c 'ls | sort'")
|
|
|
|
expected_messages = [
|
|
{"role": "system", "content": role.role},
|
|
{"role": "user", "content": "list folder"},
|
|
{"role": "assistant", "content": "ls"},
|
|
{"role": "user", "content": "sort by name"},
|
|
{"role": "assistant", "content": "ls | sort"},
|
|
]
|
|
expected_args = comp_args(role, "", messages=expected_messages)
|
|
completion.assert_called_with(**expected_args)
|
|
assert completion.call_count == 2
|
|
|
|
assert result.exit_code == 0
|
|
assert ">>> list folder" in result.stdout
|
|
assert "ls" in result.stdout
|
|
assert ">>> sort by name" in result.stdout
|
|
assert "ls | sort" in result.stdout
|
|
|
|
|
|
@patch("openai.resources.chat.Completions.create")
|
|
def test_shell_and_describe_shell(completion):
|
|
args = {"prompt": "ls", "--describe-shell": True, "--shell": True}
|
|
result = runner.invoke(app, cmd_args(**args))
|
|
|
|
completion.assert_not_called()
|
|
assert result.exit_code == 2
|
|
assert "Error" in result.stdout
|
|
|
|
|
|
@patch("openai.resources.chat.Completions.create")
|
|
def test_shell_no_interaction(completion):
|
|
completion.return_value = comp_chunks("git commit -m test")
|
|
role = SystemRole.get(DefaultRoles.SHELL.value)
|
|
|
|
args = {
|
|
"prompt": "make a commit using git",
|
|
"--shell": True,
|
|
"--no-interaction": True,
|
|
}
|
|
result = runner.invoke(app, cmd_args(**args))
|
|
|
|
completion.assert_called_once_with(**comp_args(role, args["prompt"]))
|
|
assert result.exit_code == 0
|
|
assert "git commit" in result.stdout
|
|
assert "[E]xecute" not in result.stdout
|