From 9dcdb6d6f858d08549b997e0ea5dc3f6d2e8d67a Mon Sep 17 00:00:00 2001
From: Mikel Calvo <admin@mikelcalvo.net>
Date: Fri, 21 Apr 2023 21:44:02 +0200
Subject: [PATCH] Add OS Info into the initial prompt (#2587)

---
 autogpt/config/ai_config.py | 13 +++++++++++++
 requirements.txt            |  1 +
 2 files changed, 14 insertions(+)

diff --git a/autogpt/config/ai_config.py b/autogpt/config/ai_config.py
index 1e48ab4d..a622c963 100644
--- a/autogpt/config/ai_config.py
+++ b/autogpt/config/ai_config.py
@@ -5,9 +5,11 @@ A module that contains the AIConfig class object that contains the configuration
 from __future__ import annotations
 
 import os
+import platform
 from pathlib import Path
 from typing import Optional, Type
 
+import distro
 import yaml
 
 from autogpt.prompts.generator import PromptGenerator
@@ -130,6 +132,17 @@ class AIConfig:
                 continue
             prompt_generator = plugin.post_prompt(prompt_generator)
 
+        if cfg.execute_local_commands:
+            # add OS info to prompt
+            os_name = platform.system()
+            os_info = (
+                platform.platform(terse=True)
+                if os_name != "Linux"
+                else distro.name(pretty=True)
+            )
+
+            prompt_start += f"\nThe OS you are running on is: {os_info}"
+
         # Construct full prompt
         full_prompt = f"You are {prompt_generator.name}, {prompt_generator.role}\n{prompt_start}\n\nGOALS:\n\n"
         for i, goal in enumerate(self.ai_goals):
diff --git a/requirements.txt b/requirements.txt
index d5a1c63b..5d98e612 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,6 @@
 beautifulsoup4>=4.12.2
 colorama==0.4.6
+distro==1.8.0
 openai==0.27.2
 playsound==1.2.2
 python-dotenv==1.0.0