introduce MAX_ROUND environmental variable

.env.example

@@ -10,4 +10,7 @@ TARGET_USER='bob'
 TARGET_PASSWORD='secret'
 
 # which LLM driver to use (can be openai_rest or oobabooga for now)
-LLM_CONNECTION = "openai_rest"
+LLM_CONNECTION = "openai_rest"
+
+# how many rounds should this thing go?
+MAX_ROUNDS = 20

config.py

@@ -16,4 +16,7 @@ def target_user():
     return os.getenv('TARGET_USER')
 
 def llm_connection():
-    return os.getenv("LLM_CONNECTION")
+    return os.getenv("LLM_CONNECTION")
+
+def max_rounds():
+    return int(os.getenv("MAX_ROUNDS"))

wintermute.py

@@ -14,7 +14,6 @@
 # setup dotenv
 load_dotenv()
 
-
 # setup some infrastructure
 cmd_history = ResultHistory()
 console = Console()
@@ -57,7 +56,10 @@ def handle_ssh(input):
     except paramiko.ssh_exception.AuthenticationException:
         return cmd, "Authentication error, credentials are wrong"
 
-while True:
+round : int = 0
+max_rounds : int = config.max_rounds()
+
+while round < max_rounds:
 
     state_size = num_tokens_from_string(state)
 
@@ -82,4 +84,6 @@ def handle_ssh(input):
 
     # update our command history and output it
     cmd_history.append(diff, next_cmd["type"], cmd, result, success, reason)
-    console.print(cmd_history.create_history_table())
+    console.print(cmd_history.create_history_table())
+
+    round += 1