mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-28 03:14:32 +01:00
06ae4684c8
Co-authored-by: James Collins <collijk@uw.edu> Co-authored-by: Luke Kyohere <lkyohere@mfsafrica.com> Co-authored-by: k-boikov <64261260+k-boikov@users.noreply.github.com> Co-authored-by: Media <12145726+rihp@users.noreply.github.com> Co-authored-by: Nicholas Tindle <nick@ntindle.com>
122 lines
3.5 KiB
Python
122 lines
3.5 KiB
Python
import os
|
|
|
|
import requests
|
|
import yaml
|
|
from colorama import Fore
|
|
from git.repo import Repo
|
|
|
|
from autogpt.logs import logger
|
|
|
|
# Use readline if available (for clean_input)
|
|
try:
|
|
import readline
|
|
except ImportError:
|
|
pass
|
|
|
|
from autogpt.config import Config
|
|
|
|
|
|
def clean_input(prompt: str = "", talk=False):
|
|
try:
|
|
cfg = Config()
|
|
if cfg.chat_messages_enabled:
|
|
for plugin in cfg.plugins:
|
|
if not hasattr(plugin, "can_handle_user_input"):
|
|
continue
|
|
if not plugin.can_handle_user_input(user_input=prompt):
|
|
continue
|
|
plugin_response = plugin.user_input(user_input=prompt)
|
|
if not plugin_response:
|
|
continue
|
|
if plugin_response.lower() in [
|
|
"yes",
|
|
"yeah",
|
|
"y",
|
|
"ok",
|
|
"okay",
|
|
"sure",
|
|
"alright",
|
|
]:
|
|
return cfg.authorise_key
|
|
elif plugin_response.lower() in [
|
|
"no",
|
|
"nope",
|
|
"n",
|
|
"negative",
|
|
]:
|
|
return cfg.exit_key
|
|
return plugin_response
|
|
|
|
# ask for input, default when just pressing Enter is y
|
|
logger.info("Asking user via keyboard...")
|
|
answer = input(prompt)
|
|
return answer
|
|
except KeyboardInterrupt:
|
|
logger.info("You interrupted Auto-GPT")
|
|
logger.info("Quitting...")
|
|
exit(0)
|
|
|
|
|
|
def validate_yaml_file(file: str):
|
|
try:
|
|
with open(file, encoding="utf-8") as fp:
|
|
yaml.load(fp.read(), Loader=yaml.FullLoader)
|
|
except FileNotFoundError:
|
|
return (False, f"The file {Fore.CYAN}`{file}`{Fore.RESET} wasn't found")
|
|
except yaml.YAMLError as e:
|
|
return (
|
|
False,
|
|
f"There was an issue while trying to read with your AI Settings file: {e}",
|
|
)
|
|
|
|
return (True, f"Successfully validated {Fore.CYAN}`{file}`{Fore.RESET}!")
|
|
|
|
|
|
def readable_file_size(size, decimal_places=2):
|
|
"""Converts the given size in bytes to a readable format.
|
|
Args:
|
|
size: Size in bytes
|
|
decimal_places (int): Number of decimal places to display
|
|
"""
|
|
for unit in ["B", "KB", "MB", "GB", "TB"]:
|
|
if size < 1024.0:
|
|
break
|
|
size /= 1024.0
|
|
return f"{size:.{decimal_places}f} {unit}"
|
|
|
|
|
|
def get_bulletin_from_web():
|
|
try:
|
|
response = requests.get(
|
|
"https://raw.githubusercontent.com/Significant-Gravitas/Auto-GPT/master/BULLETIN.md"
|
|
)
|
|
if response.status_code == 200:
|
|
return response.text
|
|
except requests.exceptions.RequestException:
|
|
pass
|
|
|
|
return ""
|
|
|
|
|
|
def get_current_git_branch() -> str:
|
|
try:
|
|
repo = Repo(search_parent_directories=True)
|
|
branch = repo.active_branch
|
|
return branch.name
|
|
except:
|
|
return ""
|
|
|
|
|
|
def get_latest_bulletin() -> str:
|
|
exists = os.path.exists("CURRENT_BULLETIN.md")
|
|
current_bulletin = ""
|
|
if exists:
|
|
current_bulletin = open("CURRENT_BULLETIN.md", "r", encoding="utf-8").read()
|
|
new_bulletin = get_bulletin_from_web()
|
|
is_new_news = new_bulletin != current_bulletin
|
|
|
|
if new_bulletin and is_new_news:
|
|
open("CURRENT_BULLETIN.md", "w", encoding="utf-8").write(new_bulletin)
|
|
return f" {Fore.RED}::UPDATED:: {Fore.CYAN}{new_bulletin}{Fore.RESET}"
|
|
return current_bulletin
|