From cde2595ccc705630612e609c181ce6a9867108e2 Mon Sep 17 00:00:00 2001
From: jjzhuo <134809928+jjzhuo@users.noreply.github.com>
Date: Sat, 17 Jun 2023 07:35:43 -0700
Subject: [PATCH] Fix an exception for gpt response with lang=sh (#105)

---
 gpt_engineer/steps.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gpt_engineer/steps.py b/gpt_engineer/steps.py
index 11cfc0f..60fea23 100644
--- a/gpt_engineer/steps.py
+++ b/gpt_engineer/steps.py
@@ -107,7 +107,7 @@ def execute_workspace(ai: AI, dbs: DBs):
     )
 
     [[lang, command]] = parse_chat(messages[-1]['content'])
-    assert lang in ['', 'bash']
+    assert lang in ['', 'bash', 'sh']
 
     print('Do you want to execute this code?')
     print(command)