Incoming

Author: KillianLucas

computer_use/loop.py

@@ -61,8 +61,6 @@
 if "--model" in sys.argv and sys.argv[sys.argv.index("--model") + 1]:
     model_choice = sys.argv[sys.argv.index("--model") + 1]
 
-print(model_choice)
-
 md = MarkdownStreamer()
 
 COMPUTER_USE_BETA_FLAG = "computer-use-2024-10-22"
@@ -243,8 +241,8 @@ async def sampling_loop(
         current_block = None
 
         for chunk in raw_response:
-            chunk = chunk.choices[0]
-            # time.sleep(5)
+            # chunk = chunk.choices[0]
+            # # time.sleep(5)
             if isinstance(chunk, BetaRawContentBlockStartEvent):
                 current_block = chunk.content_block
             elif isinstance(chunk, BetaRawContentBlockDeltaEvent):
@@ -344,10 +342,12 @@ async def sampling_loop(
                 print(f"\n\033[38;5;240mRun actions above\033[0m?")
                 user_approval = input("\n(y/n): ").lower().strip()
             elif len(tool_use_blocks) == 1:
-                print(
-                    f"\n\033[38;5;240mRun tool \033[0m\033[1m{tool_use_blocks[0].name}\033[0m?"
-                )
+                print(f"\n\033[38;5;240mRun tool?\033[0m")
+                # print(
+                #     f"\n\033[38;5;240mRun tool \033[0m\033[1m{tool_use_blocks[0].name}\033[0m?"
+                # )
                 user_approval = input("\n(y/n): ").lower().strip()
+                print()
 
         tool_result_content: list[BetaToolResultBlockParam] = []
         for content_block in cast(list[BetaContentBlock], response.content):
@@ -618,15 +618,6 @@ def original_welcome():
         print_markdown("---")
         time.sleep(0.5)
 
-        # Check for API key in environment variable
-        api_key = os.environ.get("ANTHROPIC_API_KEY")
-        if not api_key:
-            api_key = input(
-                "\nAn Anthropic API is required for OS mode.\n\nEnter your Anthropic API key: "
-            )
-            print_markdown("\n---")
-            time.sleep(0.5)
-
         import random
 
         tips = [
@@ -655,6 +646,15 @@ def new_welcome():
         print_markdown("---")
         time.sleep(0.5)
 
+    # Check for API key in environment variable
+    api_key = os.environ.get("ANTHROPIC_API_KEY")
+    if not api_key:
+        api_key = input(
+            "\nAn Anthropic API is required for OS mode.\n\nEnter your Anthropic API key: "
+        )
+        print_markdown("\n---")
+        time.sleep(0.5)
+
     # new_welcome()
 
     # Start the mouse position checking thread

computer_use/ui/edit.py

@@ -146,6 +146,7 @@ def _render_line(self, line):
         #     line_prefix = f"{SchemaRenderer.GRAY_COLOR}    │ {SchemaRenderer.RESET_COLOR}"
         highlighted = highlight(chunks[0] + "\n", lexer, formatter).rstrip()
         sys.stdout.write(f"{line_prefix}{highlighted}\n")
+        # sys.stdout.write(f"{line_prefix}" + " ".join(highlighted) + "\n") # For debugging
 
         # Print remaining chunks with padding and pipe
         continuation_prefix = (
@@ -450,7 +451,7 @@ class OldStrRenderer(ContentRenderer):
     def __init__(self, style):
         super().__init__(style)
         SchemaRenderer.print_separator("┼")
-        self.RED_COLOR = "\033[38;5;197m"  # Monokai red color (more purple)
+        self.RED_COLOR = "\033[39m\033[38;5;204m"  # Monokai red
         self.RESET_COLOR = "\033[0m"
         self.rendered_content = ""
         self.line_number = 1