|
import os, sys |
|
import utils |
|
|
|
FILE_DIR = os.path.dirname(os.path.abspath(__file__)) |
|
REPO_DIR = os.path.dirname(FILE_DIR) |
|
sys.path.append(REPO_DIR) |
|
import threading |
|
from functools import partial |
|
import time |
|
|
|
def get_openai_api_key(): |
|
return os.getenv("OPENAI_API_KEY") |
|
|
|
running_apis = [] |
|
|
|
class AutoAPI: |
|
def __init__(self, openai_key, ai_name, ai_role, top_5_goals): |
|
print(openai_key) |
|
self.openai_key = openai_key |
|
newline = "\n" |
|
with open(os.path.join(REPO_DIR, "ai_settings.yaml"), "w") as f: |
|
f.write( |
|
f"""ai_goals: |
|
{newline.join([f'- {goal[0]}' for goal in top_5_goals if goal[0]])} |
|
ai_name: {ai_name} |
|
ai_role: {ai_role} |
|
""" |
|
) |
|
|
|
|
|
thread = threading.Thread(target=self.client_thread) |
|
thread.start() |
|
self.thread = thread |
|
self.pending_input = None |
|
self.awaiting_input = False |
|
self.messages = [] |
|
self.last_message_read_index = -1 |
|
|
|
def add_message(self, title, content): |
|
|
|
self.messages.append((title, content)) |
|
|
|
def client_thread(self): |
|
os.environ["OPENAI_API_KEY"] = self.openai_key |
|
import autogpt.config.config |
|
from autogpt.logs import logger |
|
from autogpt.cli import main |
|
import autogpt.utils |
|
from autogpt.spinner import Spinner |
|
|
|
|
|
def typewriter_log(self, title="", title_color="", content="", *args, **kwargs): |
|
self.add_message(title, content) |
|
|
|
def warn(self, message, title="", *args, **kwargs): |
|
self.add_message(title, message) |
|
|
|
def error(self, title, message="", *args, **kwargs): |
|
self.add_message(title, message) |
|
|
|
def clean_input(self, prompt=""): |
|
self.add_message(None, prompt) |
|
self.awaiting_input = True |
|
while self.pending_input is None: |
|
time.sleep(1) |
|
pending_input = self.pending_input |
|
self.pending_input = None |
|
print("Sending message:", pending_input) |
|
return pending_input |
|
|
|
def spinner_start(self): |
|
self.add_message(None, "Thinking...") |
|
|
|
logger.typewriter_log = partial(typewriter_log, self) |
|
logger.warn = partial(warn, self) |
|
logger.error = partial(error, self) |
|
autogpt.utils.clean_input = partial(clean_input, self) |
|
Spinner.spin = partial(spinner_start, self) |
|
|
|
main() |
|
|
|
def send_message(self, message="Y"): |
|
self.pending_input = message |
|
self.awaiting_input = False |
|
|
|
def get_chatbot_response(self): |
|
while (not self.awaiting_input) or self.last_message_read_index < len(self.messages) - 1: |
|
if self.last_message_read_index >= len(self.messages) - 1: |
|
time.sleep(1) |
|
else: |
|
self.last_message_read_index += 1 |
|
title, content = self.messages[self.last_message_read_index] |
|
yield (f"**{title.strip()}** " if title else "") + utils.remove_color(content).replace("\n", "<br />") |
|
|
