diff --git a/autogpt/config/ai_config.py b/autogpt/config/ai_config.py index 1e48ab4db..a622c9639 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 d5a1c63b3..5d98e6128 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