fix for python3.9

pilot/helpers/cli.py

@@ -5,7 +5,7 @@ import threading
 import queue
 import time
 import platform
-from typing import Dict
+from typing import Dict, Union
 
 from logger.logger import logger
 from utils.style import yellow, green, red, yellow_bold, white_bold
@@ -322,7 +322,7 @@ def execute_command_and_check_cli_response(command, timeout, convo):
 
 
 def run_command_until_success(convo, command,
-                              timeout: int | None,
+                              timeout: Union[int, None],
                               process_name: str | None = None,
                               additional_message=None,
                               force=False,

pilot/test_main_e2e.py

@@ -26,9 +26,9 @@ def test_init():
 @pytest.mark.uses_tokens
 @pytest.mark.skip(reason="Uses lots of tokens")
 @pytest.mark.parametrize("endpoint, model", [
-    # ("OPENAI", "gpt-4"),
-    # ("OPENROUTER", "openai/gpt-3.5-turbo"),
-    # ("OPENROUTER", "meta-llama/codellama-34b-instruct"),
+    ("OPENAI", "gpt-4"),
+    ("OPENROUTER", "openai/gpt-3.5-turbo"),
+    ("OPENROUTER", "meta-llama/codellama-34b-instruct"),
     ("OPENROUTER", "google/palm-2-chat-bison"),
     ("OPENROUTER", "google/palm-2-codechat-bison"),
     # TODO: See https://github.com/1rgs/jsonformer-claude/blob/main/jsonformer_claude/main.py