diff --git a/gptme/prompts.py b/gptme/prompts.py index 2ec2ec0d..89f1d49e 100644 --- a/gptme/prompts.py +++ b/gptme/prompts.py @@ -6,7 +6,9 @@ """ import logging +import re import os +import platform import subprocess from collections.abc import Generator, Iterable from typing import Literal @@ -56,6 +58,7 @@ def prompt_full(interactive: bool) -> Generator[Message, None, None]: if interactive: yield from prompt_user() yield from prompt_project() + yield from prompt_systeminfo() def prompt_short(interactive: bool) -> Generator[Message, None, None]: @@ -81,7 +84,7 @@ def prompt_gptme(interactive: bool) -> Generator[Message, None, None]: base_prompt = f""" You are gptme v{__version__}, a general-purpose AI assistant powered by LLMs. -You are designed to help users with programming tasks, such as writing code, debugging, and learning new concepts. +You are designed to help users with programming tasks, such as writing code, debugging and learning new concepts. You can run code, execute terminal commands, and access the filesystem on the local machine. You will help the user with writing code, either from scratch or in existing projects. You will think step by step when solving a problem, in tags. @@ -209,7 +212,43 @@ def prompt_tools(examples: bool = True) -> Generator[Message, None, None]: yield Message("system", prompt.strip() + "\n\n") +def prompt_systeminfo() -> Generator[Message, None, None]: + """Generate the system information prompt.""" + if platform.system() == "Linux": + os_info = get_system_distro() + os_version = platform.uname().release + elif platform.system() == "Windows": + os_info = "Windows" + os_version = platform.version() + elif platform.system() == "Darwin": + os_info = "macOS" + os_version = platform.mac_ver()[0] + else: + os_info = "unknown" + os_version = "" + + prompt = f"## System Information\n\n**OS:** {os_info} {os_version}".strip() + + yield Message( + "system", + prompt, + ) + + +def get_system_distro() -> str: + """Get the system distribution name.""" + regex = re.compile(r"^NAME=\"?([^\"]+)\"?") + if os.path.exists("/etc/os-release"): + with open("/etc/os-release") as f: + for line in f: + matches = re.search(regex, line) + if matches: + return matches.string[matches.start(1) : matches.end(1)] + return "Linux" + + document_prompt_function(interactive=True)(prompt_gptme) document_prompt_function()(prompt_user) document_prompt_function()(prompt_project) document_prompt_function()(prompt_tools) +document_prompt_function()(prompt_systeminfo) diff --git a/gptme/tools/shell.py b/gptme/tools/shell.py index e2110e5e..d02a054d 100644 --- a/gptme/tools/shell.py +++ b/gptme/tools/shell.py @@ -50,7 +50,6 @@ def get_installed_programs() -> set[str]: When you send a message containing bash code, it will be executed in a stateful bash shell. The shell will respond with the output of the execution. Do not use EOF/HereDoc syntax to send multiline commands, as the assistant will not be able to handle it. -{'The platform is macOS.' if is_macos else ''} These programs are available, among others: {shell_programs_str}