Update agent_controller.py (#1004)

This commit is contained in:
Robert Brennan 2024-04-10 18:28:30 -05:00 committed by GitHub
parent cd723abfdd
commit 7f5d9c7d92
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -141,7 +141,7 @@ class AgentController:
print_with_color(observation, 'ERROR')
traceback.print_exc()
# TODO Change to more robust error handling
if 'The api_key client option must be set' or 'Incorrect API key provided:' in observation.content:
if 'The api_key client option must be set' in observation.content or 'Incorrect API key provided:' in observation.content:
raise
self.update_state_after_step()