github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(agent): Fix check when loading an existing agent (#7026)

Browse Source 
Now the check also ensures the chosen agent number is within proper range
This commit is contained in:
Krzysztof Czerwinski
2024-03-22 14:55:49 +01:00
committed by GitHub
parent a1ffe15142
commit 262771a69c
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
autogpts/autogpt/autogpt/app/main.py
View File

@@ -185,9 +185,12 @@ async def run_auto_gpt(
"Enter the number or name of the agent to run,"
" or hit enter to create a new one:",
)
if re.match(r"^\d+$", load_existing_agent):
if re.match(r"^\d+$", load_existing_agent.strip()) and 0 < int(
load_existing_agent
) <= len(existing_agents):
load_existing_agent = existing_agents[int(load_existing_agent) - 1]
elif load_existing_agent and load_existing_agent not in existing_agents:
if load_existing_agent not in existing_agents:
logger.info(
f"Unknown agent '{load_existing_agent}', "
f"creating a new one instead.",