-
Notifications
You must be signed in to change notification settings - Fork 45.1k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #3152 from collijk/refactor/add-indirection-layer-…
…around-entry-point Add indirection layer to entry point
- Loading branch information
Showing
2 changed files
with
149 additions
and
114 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,146 @@ | ||
| """The application entry point. Can be invoked by a CLI or any other front end application.""" | ||
|
|
||
| import logging | ||
| import sys | ||
| from pathlib import Path | ||
|
|
||
| from colorama import Fore | ||
|
|
||
| from autogpt.agent.agent import Agent | ||
| from autogpt.commands.command import CommandRegistry | ||
| from autogpt.config import Config, check_openai_api_key | ||
| from autogpt.configurator import create_config | ||
| from autogpt.logs import logger | ||
| from autogpt.memory import get_memory | ||
| from autogpt.plugins import scan_plugins | ||
| from autogpt.prompts.prompt import construct_main_ai_config | ||
| from autogpt.utils import get_current_git_branch, get_latest_bulletin | ||
| from autogpt.workspace import Workspace | ||
|
|
||
|
|
||
| def run_auto_gpt( | ||
| continuous: bool, | ||
| continuous_limit: int, | ||
| ai_settings: str, | ||
| skip_reprompt: bool, | ||
| speak: bool, | ||
| debug: bool, | ||
| gpt3only: bool, | ||
| gpt4only: bool, | ||
| memory_type: str, | ||
| browser_name: str, | ||
| allow_downloads: bool, | ||
| skip_news: bool, | ||
| workspace_directory: str, | ||
| ): | ||
| cfg = Config() | ||
| # TODO: fill in llm values here | ||
| check_openai_api_key() | ||
| create_config( | ||
| continuous, | ||
| continuous_limit, | ||
| ai_settings, | ||
| skip_reprompt, | ||
| speak, | ||
| debug, | ||
| gpt3only, | ||
| gpt4only, | ||
| memory_type, | ||
| browser_name, | ||
| allow_downloads, | ||
| skip_news, | ||
| ) | ||
| logger.set_level(logging.DEBUG if cfg.debug_mode else logging.INFO) | ||
| if not cfg.skip_news: | ||
| motd = get_latest_bulletin() | ||
| if motd: | ||
| logger.typewriter_log("NEWS: ", Fore.GREEN, motd) | ||
| git_branch = get_current_git_branch() | ||
| if git_branch and git_branch != "stable": | ||
| logger.typewriter_log( | ||
| "WARNING: ", | ||
| Fore.RED, | ||
| f"You are running on `{git_branch}` branch " | ||
| "- this is not a supported branch.", | ||
| ) | ||
| if sys.version_info < (3, 10): | ||
| logger.typewriter_log( | ||
| "WARNING: ", | ||
| Fore.RED, | ||
| "You are running on an older version of Python. " | ||
| "Some people have observed problems with certain " | ||
| "parts of Auto-GPT with this version. " | ||
| "Please consider upgrading to Python 3.10 or higher.", | ||
| ) | ||
|
|
||
| # TODO: have this directory live outside the repository (e.g. in a user's | ||
| # home directory) and have it come in as a command line argument or part of | ||
| # the env file. | ||
| if workspace_directory is None: | ||
| workspace_directory = Path(__file__).parent / "auto_gpt_workspace" | ||
| else: | ||
| workspace_directory = Path(workspace_directory) | ||
| # TODO: pass in the ai_settings file and the env file and have them cloned into | ||
| # the workspace directory so we can bind them to the agent. | ||
| workspace_directory = Workspace.make_workspace(workspace_directory) | ||
| cfg.workspace_path = str(workspace_directory) | ||
|
|
||
| # HACK: doing this here to collect some globals that depend on the workspace. | ||
| file_logger_path = workspace_directory / "file_logger.txt" | ||
| if not file_logger_path.exists(): | ||
| with file_logger_path.open(mode="w", encoding="utf-8") as f: | ||
| f.write("File Operation Logger ") | ||
|
|
||
| cfg.file_logger_path = str(file_logger_path) | ||
|
|
||
| cfg.set_plugins(scan_plugins(cfg, cfg.debug_mode)) | ||
| # Create a CommandRegistry instance and scan default folder | ||
| command_registry = CommandRegistry() | ||
| command_registry.import_commands("autogpt.commands.analyze_code") | ||
| command_registry.import_commands("autogpt.commands.audio_text") | ||
| command_registry.import_commands("autogpt.commands.execute_code") | ||
| command_registry.import_commands("autogpt.commands.file_operations") | ||
| command_registry.import_commands("autogpt.commands.git_operations") | ||
| command_registry.import_commands("autogpt.commands.google_search") | ||
| command_registry.import_commands("autogpt.commands.image_gen") | ||
| command_registry.import_commands("autogpt.commands.improve_code") | ||
| command_registry.import_commands("autogpt.commands.twitter") | ||
| command_registry.import_commands("autogpt.commands.web_selenium") | ||
| command_registry.import_commands("autogpt.commands.write_tests") | ||
| command_registry.import_commands("autogpt.app") | ||
|
|
||
| ai_name = "" | ||
| ai_config = construct_main_ai_config() | ||
| ai_config.command_registry = command_registry | ||
| # print(prompt) | ||
| # Initialize variables | ||
| full_message_history = [] | ||
| next_action_count = 0 | ||
| # Make a constant: | ||
| triggering_prompt = ( | ||
| "Determine which next command to use, and respond using the" | ||
| " format specified above:" | ||
| ) | ||
| # Initialize memory and make sure it is empty. | ||
| # this is particularly important for indexing and referencing pinecone memory | ||
| memory = get_memory(cfg, init=True) | ||
| logger.typewriter_log( | ||
| "Using memory of type:", Fore.GREEN, f"{memory.__class__.__name__}" | ||
| ) | ||
| logger.typewriter_log("Using Browser:", Fore.GREEN, cfg.selenium_web_browser) | ||
| system_prompt = ai_config.construct_full_prompt() | ||
| if cfg.debug_mode: | ||
| logger.typewriter_log("Prompt:", Fore.GREEN, system_prompt) | ||
|
|
||
| agent = Agent( | ||
| ai_name=ai_name, | ||
| memory=memory, | ||
| full_message_history=full_message_history, | ||
| next_action_count=next_action_count, | ||
| command_registry=command_registry, | ||
| config=ai_config, | ||
| system_prompt=system_prompt, | ||
| triggering_prompt=triggering_prompt, | ||
| workspace_directory=workspace_directory, | ||
| ) | ||
| agent.start_interaction_loop() |