Apply autopep8 formatting to entire codebase

This commit is contained in:
Torantulino
2023-04-02 09:13:15 +01:00
parent a2e5de7469
commit b4685f67e6
11 changed files with 224 additions and 69 deletions

View File

@@ -1,14 +1,16 @@
import openai
next_key = 0
agents = {} # key, (task, full_message_history, model)
agents = {} # key, (task, full_message_history, model)
# Create new GPT agent
def create_agent(task, prompt, model):
global next_key
global agents
messages = [{"role": "user", "content": prompt},]
messages = [{"role": "user", "content": prompt}, ]
# Start GTP3 instance
response = openai.ChatCompletion.create(
@@ -22,12 +24,15 @@ def create_agent(task, prompt, model):
messages.append({"role": "assistant", "content": agent_reply})
key = next_key
next_key += 1 # This is done instead of len(agents) to make keys unique even if agents are deleted
# This is done instead of len(agents) to make keys unique even if agents
# are deleted
next_key += 1
agents[key] = (task, messages, model)
return key, agent_reply
def message_agent(key, message):
global agents
@@ -50,12 +55,14 @@ def message_agent(key, message):
return agent_reply
def list_agents():
global agents
# Return a list of agent keys and their tasks
return [(key, task) for key, (task, _, _) in agents.items()]
def delete_agent(key):
global agents
@@ -64,5 +71,3 @@ def delete_agent(key):
return True
except KeyError:
return False