Casts agent key to int in message_agent

pull/17/head
Torantulino 2023-03-31 04:13:00 +01:00
parent 7790066b3e
commit 4bd311c527
1 changed files with 1 additions and 1 deletions

View File

@ -31,7 +31,7 @@ def create_agent(task, prompt, model):
def message_agent(key, message):
global agents
task, messages, model = agents[key]
task, messages, model = agents[int(key)]
# Add user message to message history before sending to agent
messages.append({"role": "user", "content": message})