mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-17 22:14:28 +01:00
1073954fb7
* Pi's message. * Fix most everything. * Blacked * Add Typing, Docstrings everywhere, organize the code a bit. * Black * fix import * Update message, dedupe. * Increase backoff time. * bump up retries
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
"""Main script for the autogpt package."""
|
|
import logging
|
|
from autogpt.agent.agent import Agent
|
|
from autogpt.args import parse_arguments
|
|
|
|
from autogpt.config import Config, check_openai_api_key
|
|
from autogpt.logs import logger
|
|
from autogpt.memory import get_memory
|
|
|
|
from autogpt.prompt import construct_prompt
|
|
|
|
# Load environment variables from .env file
|
|
|
|
|
|
def main() -> None:
|
|
"""Main function for the script"""
|
|
cfg = Config()
|
|
# TODO: fill in llm values here
|
|
check_openai_api_key()
|
|
parse_arguments()
|
|
logger.set_level(logging.DEBUG if cfg.debug_mode else logging.INFO)
|
|
ai_name = ""
|
|
prompt = construct_prompt()
|
|
# print(prompt)
|
|
# Initialize variables
|
|
full_message_history = []
|
|
next_action_count = 0
|
|
# Make a constant:
|
|
user_input = (
|
|
"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)
|
|
print(f"Using memory of type: {memory.__class__.__name__}")
|
|
agent = Agent(
|
|
ai_name=ai_name,
|
|
memory=memory,
|
|
full_message_history=full_message_history,
|
|
next_action_count=next_action_count,
|
|
prompt=prompt,
|
|
user_input=user_input,
|
|
)
|
|
agent.start_interaction_loop()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|