From 397c091fec788dff158faa5748bd2e9d6a78e679 Mon Sep 17 00:00:00 2001
From: Brayo <vukubrian@gmail.com>
Date: Tue, 8 Oct 2024 18:00:35 +0300
Subject: [PATCH] feat: add system info to the about_user prompt

---
 gptme/prompts.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/gptme/prompts.py b/gptme/prompts.py
index 2ec2ec0d..71979e8f 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]:
@@ -208,8 +211,39 @@ def prompt_tools(examples: bool = True) -> Generator[Message, None, None]:
     prompt += "\n\n*End of Tools List.*"
     yield Message("system", prompt.strip() + "\n\n")
 
+def prompt_systeminfo() -> Generator[Message, None, None]:
+    """Generate the system information prompt."""
+    if os.name == "posix":
+        os_info = get_system_distro()
+        os_version = platform.uname().release
+    elif os.name == "nt":
+        os_info = "Windows"
+        os_version = platform.version()
+    elif os.name == "darwin":
+        os_info = "macOS"
+        os_version = platform.mac_ver()[0]
+    else:
+        os_info = "unknown"
+        os_version = "unknown"
+    
+    prompt = f"## System Information\n\n**OS:** {os_info}\n**Version:** {os_version}"
+
+    yield Message( "system", prompt,)
+
+def get_system_distro() -> str:
+    """Get the system distribution name."""
+    regex = r"^NAME=\"([^\"]+)\""
+
+    if os.path.exists("/etc/os-release"):
+        with open("/etc/os-release") as f:
+            for line in f:
+                match = re.match(regex, line)
+                if match:
+                    return match.group(1)
+    return "unknown"
 
 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)