Finish integrating command registry

autogpt/commands/audio_text.py

@@ -1,23 +1,51 @@
-import requests
+"""Commands for converting audio to text."""
 import json
 
+import requests
+
+from autogpt.commands.command import command
 from autogpt.config import Config
 from autogpt.workspace import path_in_workspace
 
-cfg = Config()
+CFG = Config()
 
 
-def read_audio_from_file(audio_path):
+@command(
+    "read_audio_from_file",
+    "Convert Audio to text",
+    '"file": "<file>"',
+    CFG.huggingface_audio_to_text_model,
+    "Configure huggingface_audio_to_text_model.",
+)
+def read_audio_from_file(audio_path: str) -> str:
+    """
+    Convert audio to text.
+
+    Args:
+        audio_path (str): The path to the audio file
+
+    Returns:
+        str: The text from the audio
+    """
     audio_path = path_in_workspace(audio_path)
     with open(audio_path, "rb") as audio_file:
         audio = audio_file.read()
     return read_audio(audio)
 
 
-def read_audio(audio):
-    model = cfg.huggingface_audio_to_text_model
+def read_audio(audio: bytes) -> str:
+    """
+    Convert audio to text.
+
+    Args:
+        audio (bytes): The audio to convert
+
+    Returns:
+        str: The text from the audio
+    """
+    model = CFG.huggingface_audio_to_text_model
     api_url = f"https://api-inference.huggingface.co/models/{model}"
-    api_token = cfg.huggingface_api_token
+    api_token = CFG.huggingface_api_token
     headers = {"Authorization": f"Bearer {api_token}"}
 
     if api_token is None:
@@ -32,4 +60,4 @@ def read_audio(audio):
     )
 
     text = json.loads(response.content.decode("utf-8"))["text"]
-    return "The audio says: " + text
+    return f"The audio says: {text}"

autogpt/commands/command.py

@@ -1,8 +1,8 @@
-import os
-import sys
 import importlib
 import inspect
-from typing import Callable, Any, List, Optional
+import os
+import sys
+from typing import Any, Callable, List, Optional
 
 # Unique identifier for auto-gpt commands
 AUTO_GPT_COMMAND_IDENTIFIER = "auto_gpt_command"

autogpt/commands/evaluate_code.py

@@ -5,7 +5,7 @@ from autogpt.commands.command import command
 from autogpt.llm_utils import call_ai_function
 
 
-@command("evaluate_code", "Evaluate Code", '"code": "<full _code_string>"')
+@command("evaluate_code", "Evaluate Code", '"code": "<full_code_string>"')
 def evaluate_code(code: str) -> list[str]:
     """
     A function that takes in a string and returns a response from create chat

autogpt/commands/execute_code.py

@@ -4,9 +4,10 @@ import subprocess
 
 import docker
 from docker.errors import ImageNotFound
-from autogpt.config import Config
+
 from autogpt.commands.command import command
-from autogpt.workspace import path_in_workspace, WORKSPACE_PATH
+from autogpt.config import Config
+from autogpt.workspace import WORKSPACE_PATH, path_in_workspace
 
 CFG = Config()
 

autogpt/commands/file_operations.py

@@ -5,8 +5,9 @@ import os
 import os.path
 from pathlib import Path
 from typing import Generator
+
 from autogpt.commands.command import command
-from autogpt.workspace import path_in_workspace, WORKSPACE_PATH
+from autogpt.workspace import WORKSPACE_PATH, path_in_workspace
 
 LOG_FILE = "file_logger.txt"
 LOG_FILE_PATH = WORKSPACE_PATH / LOG_FILE

autogpt/commands/git_operations.py

@@ -1,10 +1,19 @@
 """Git operations for autogpt"""
-import git
+from git.repo import Repo
+
+from autogpt.commands.command import command
 from autogpt.config import Config
 
 CFG = Config()
 
 
+@command(
+    "clone_repository",
+    "Clone Repositoryy",
+    '"repository_url": "<url>", "clone_path": "<directory>"',
+    CFG.github_username and CFG.github_api_key,
+    "Configure github_username and github_api_key.",
+)
 def clone_repository(repo_url: str, clone_path: str) -> str:
     """Clone a github repository locally
 
@@ -17,7 +26,7 @@ def clone_repository(repo_url: str, clone_path: str) -> str:
     split_url = repo_url.split("//")
     auth_repo_url = f"//{CFG.github_username}:{CFG.github_api_key}@".join(split_url)
     try:
-        git.Repo.clone_from(auth_repo_url, clone_path)
+        Repo.clone_from(auth_repo_url, clone_path)
         return f"""Cloned {repo_url} to {clone_path}"""
     except Exception as e:
         return f"Error: {str(e)}"

autogpt/commands/google_search.py

@@ -5,11 +5,13 @@ import json
 
 from duckduckgo_search import ddg
 
+from autogpt.commands.command import command
 from autogpt.config import Config
 
 CFG = Config()
 
 
+@command("google", "Google Search", '"input": "<search>"', not CFG.google_api_key)
 def google_search(query: str, num_results: int = 8) -> str:
     """Return the results of a google search
 
@@ -31,9 +33,17 @@ def google_search(query: str, num_results: int = 8) -> str:
     for j in results:
         search_results.append(j)
 
-    return json.dumps(search_results, ensure_ascii=False, indent=4)
+    results = json.dumps(search_results, ensure_ascii=False, indent=4)
+    return safe_google_results(results)
 
 
+@command(
+    "google",
+    "Google Search",
+    '"input": "<search>"',
+    bool(CFG.google_api_key),
+    "Configure google_api_key.",
+)
 def google_official_search(query: str, num_results: int = 8) -> str | list[str]:
     """Return the results of a google search using the official Google API
 
@@ -82,6 +92,26 @@ def google_official_search(query: str, num_results: int = 8) -> str | list[str]:
             return "Error: The provided Google API key is invalid or missing."
         else:
             return f"Error: {e}"
+    # google_result can be a list or a string depending on the search results
 
     # Return the list of search result URLs
-    return search_results_links
+    return safe_google_results(search_results_links)
+
+
+def safe_google_results(results: str | list) -> str:
+    """
+        Return the results of a google search in a safe format.
+
+    Args:
+        results (str | list): The search results.
+
+    Returns:
+        str: The results of the search.
+    """
+    if isinstance(results, list):
+        safe_message = json.dumps(
+            [result.enocde("utf-8", "ignore") for result in results]
+        )
+    else:
+        safe_message = results.encode("utf-8", "ignore").decode("utf-8")
+    return safe_message

autogpt/commands/image_gen.py

@@ -1,12 +1,12 @@
 """ Image Generation Module for AutoGPT."""
 import io
-import os.path
 import uuid
 from base64 import b64decode
 
 import openai
 import requests
 from PIL import Image
+
 from autogpt.commands.command import command
 from autogpt.config import Config
 from autogpt.workspace import path_in_workspace

autogpt/commands/improve_code.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 import json
 
-from autogpt.commands import command
+from autogpt.commands.command import command
 from autogpt.llm_utils import call_ai_function
 
 

autogpt/commands/twitter.py

@@ -1,11 +1,30 @@
-import tweepy
+"""A module that contains a command to send a tweet."""
 import os
+
+import tweepy
 from dotenv import load_dotenv
 
+from autogpt.commands.command import command
+
 load_dotenv()
 
 
-def send_tweet(tweet_text):
+@command(
+    "send_tweet",
+    "Send Tweet",
+    '"text": "<text>"',
+)
+def send_tweet(tweet_text: str) -> str:
+    """
+      A function that takes in a string and returns a response from create chat
+        completion api call.
+
+    Args:
+      tweet_text (str): Text to be tweeted.
+
+      Returns:
+          A result from sending the tweet.
+    """
     consumer_key = os.environ.get("TW_CONSUMER_KEY")
     consumer_secret = os.environ.get("TW_CONSUMER_SECRET")
     access_token = os.environ.get("TW_ACCESS_TOKEN")
@@ -20,6 +39,6 @@ def send_tweet(tweet_text):
     # Send tweet
     try:
         api.update_status(tweet_text)
-        print("Tweet sent successfully!")
+        return "Tweet sent successfully!"
     except tweepy.TweepyException as e:
-        print("Error sending tweet: {}".format(e.reason))
+        return f"Error sending tweet: {e.reason}"

autogpt/commands/web_playwright.py

@@ -8,6 +8,7 @@ except ImportError:
         "Playwright not installed. Please install it with 'pip install playwright' to use."
     )
 from bs4 import BeautifulSoup
+
 from autogpt.processing.html import extract_hyperlinks, format_hyperlinks
 
 

autogpt/commands/web_requests.py

@@ -4,9 +4,9 @@ from __future__ import annotations
 from urllib.parse import urljoin, urlparse
 
 import requests
-from requests.compat import urljoin
-from requests import Response
 from bs4 import BeautifulSoup
+from requests import Response
+from requests.compat import urljoin
 
 from autogpt.config import Config
 from autogpt.memory import get_memory

autogpt/commands/web_selenium.py

@@ -1,22 +1,25 @@
 """Selenium web scraping module."""
 from __future__ import annotations
 
-from selenium import webdriver
-from autogpt.processing.html import extract_hyperlinks, format_hyperlinks
-import autogpt.processing.text as summary
-from bs4 import BeautifulSoup
-from selenium.webdriver.remote.webdriver import WebDriver
-from selenium.webdriver.common.by import By
-from selenium.webdriver.support.wait import WebDriverWait
-from selenium.webdriver.support import expected_conditions as EC
-from webdriver_manager.chrome import ChromeDriverManager
-from webdriver_manager.firefox import GeckoDriverManager
-from selenium.webdriver.chrome.options import Options as ChromeOptions
-from selenium.webdriver.firefox.options import Options as FirefoxOptions
-from selenium.webdriver.safari.options import Options as SafariOptions
 import logging
 from pathlib import Path
+
+from bs4 import BeautifulSoup
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options as ChromeOptions
+from selenium.webdriver.common.by import By
+from selenium.webdriver.firefox.options import Options as FirefoxOptions
+from selenium.webdriver.remote.webdriver import WebDriver
+from selenium.webdriver.safari.options import Options as SafariOptions
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.support.wait import WebDriverWait
+from webdriver_manager.chrome import ChromeDriverManager
+from webdriver_manager.firefox import GeckoDriverManager
+
+from autogpt.commands.command import command
+import autogpt.processing.text as summary
 from autogpt.config import Config
+from autogpt.processing.html import extract_hyperlinks, format_hyperlinks
 
 FILE_DIR = Path(__file__).parent.parent
 CFG = Config()

autogpt/commands/write_tests.py

@@ -2,7 +2,8 @@
 from __future__ import annotations
 
 import json
-from autogpt.commands import command
+
+from autogpt.commands.command import command
 from autogpt.llm_utils import call_ai_function