From 7790066b3e97c0e0962b4fcbbdd0777c62fdd5ad Mon Sep 17 00:00:00 2001 From: Torantulino Date: Fri, 31 Mar 2023 03:40:56 +0100 Subject: [PATCH] Fixes bug in message_agent: The model 'model' does not exist. --- AutonomousAI/agent_manager.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/AutonomousAI/agent_manager.py b/AutonomousAI/agent_manager.py index a1d789e0..25b5b823 100644 --- a/AutonomousAI/agent_manager.py +++ b/AutonomousAI/agent_manager.py @@ -1,7 +1,7 @@ import openai next_key = 0 -agents = {} # key, (task, full_message_history) +agents = {} # key, (task, full_message_history, model) # Create new GPT agent def create_agent(task, prompt, model): @@ -24,21 +24,21 @@ def create_agent(task, prompt, model): key = next_key next_key += 1 # This is done instead of len(agents) to make keys unique even if agents are deleted - agents[key] = (task, messages) + agents[key] = (task, messages, model) return key, agent_reply def message_agent(key, message): global agents - task, messages = agents[key] + task, messages, model = agents[key] # Add user message to message history before sending to agent messages.append({"role": "user", "content": message}) # Start GTP3 instance response = openai.ChatCompletion.create( - model="model", + model=model, messages=messages, ) @@ -54,7 +54,7 @@ def list_agents(): global agents # Return a list of agent keys and their tasks - return [(key, task) for key, (task, _) in agents.items()] + return [(key, task) for key, (task, _, _) in agents.items()] def delete_agent(key): global agents