added lazy imports and minor clean ups

Author: codingmoh

src/open_codex/agents/phi_4_mini.py

@@ -1,12 +1,14 @@
+import contextlib
+import os
 import time
 from typing import List, cast
+
+from huggingface_hub import hf_hub_download  # type: ignore
 from llama_cpp import CreateCompletionResponse, Llama
 from open_codex.interfaces.llm_agent import LLMAgent
-import contextlib
-import os
-from huggingface_hub import hf_hub_download # type: ignore
 
 class AgentPhi4Mini(LLMAgent):
+    
     def download_model(self, model_filename: str,
                         repo_id: str, 
                         local_dir: str) -> str:
@@ -47,7 +49,10 @@ def __init__(self, system_prompt: str):
         # when loading the model
         # this is a temporary solution until the library is fixed
         with AgentPhi4Mini.suppress_native_stderr():
-            self.llm: Llama = Llama(model_path=model_path)  # type: ignore
+            lib_dir = os.path.join(os.path.dirname(__file__), "llama_cpp", "lib")
+            self.llm: Llama = Llama(
+                lib_path=os.path.join(lib_dir, "libllama.dylib"),
+                model_path=model_path)  
 
         self.system_prompt = system_prompt
 

src/open_codex/main.py

@@ -1,11 +1,6 @@
 import sys
 import argparse
 import subprocess
-import pyperclip
-
-from open_codex.agent_builder import AgentBuilder
-from open_codex.interfaces.llm_agent import LLMAgent
-
 
 GREEN = "\033[92m"
 RED = "\033[91m"
@@ -14,18 +9,15 @@
 
 # Capture single keypress (terminal) from the user
 # and returns it as a string. It works on both Windows and Unix systems.
+
 # Windows
 if sys.platform == "win32":
     import msvcrt
-
     def get_keypress():
         return msvcrt.getch().decode("utf-8")
-
-# Unix (Linux/macOS)
+# Unix
 else:
-    import termios
-    import tty
-
+    import termios, tty
     def get_keypress():
         fd = sys.stdin.fileno()
         old_settings = termios.tcgetattr(fd)
@@ -37,61 +29,60 @@ def get_keypress():
         return key
 
 def print_response(command: str):
-    print(f"{BLUE}Command found:")
-    print(f"{BLUE}=====================")
-    print(f"{GREEN}{command}{RESET}") # Print the command in green
+    print(f"{BLUE}Command found:\n=====================")
+    print(f"{GREEN}{command}{RESET}")
     print(f"{BLUE}====================={RESET}")
-
     print(f"{BLUE}What do you want to do with this command?{RESET}")
     print(f"{BLUE}[c] Copy  [e] Execute  [a] Abort{RESET}")
     print(f"{BLUE}Press key: ", end="", flush=True)
+
     choice = get_keypress().lower()
-    print(f"{RESET}")  # Clear the line after the prompt
+    print(f"{RESET}")
 
     if choice == "e":
         print(f"{BLUE}Executing command: {command}{RESET}")
         result = subprocess.run(command, shell=True, capture_output=True, text=True)
         print(f"{GREEN}Command output: {result.stdout}{RESET}")
         if result.stderr:
             print(f"{RED}Error: {result.stderr}{RESET}")
-
     elif choice == "c":
+        import pyperclip  # ⏱ lazy import
         pyperclip.copy(command)
-        print(f"{GREEN}Command copied to clipboard! Paste it manually in your terminal.{RESET}")
-
+        print(f"{GREEN}Command copied to clipboard!{RESET}")
     elif choice == "a":
         print(f"{BLUE}Aborted.{RESET}")
     else:
         print(f"{RED}Unknown choice. Nothing happened.{RESET}")
 
-def one_shot_mode(agent: LLMAgent, prompt: str):
+def one_shot_mode(prompt: str):
+    from open_codex.agent_builder import AgentBuilder
+    print(f"{BLUE}Using model: phi-4-mini-instruct{RESET}")
     try:
+        agent = AgentBuilder.get_agent()
         response = agent.one_shot_mode(prompt)
         print_response(response)
     except Exception as e:
         print(f"{RED}Error: {e}{RESET}")
 
 def print_help_message():
-    """Print help message with usage examples."""
-    print(f"{BLUE}Open Codex - Natural Language to CLI commands{RESET}")
     print(f"{BLUE}Usage examples:{RESET}")
     print(f"{GREEN}open-codex \"list all files in current directory\"")
     print(f"{GREEN}open-codex \"find all python files modified in the last week\"")
     print(f"{GREEN}open-codex \"create a tarball of the src directory\"")
     print()
 
 def main():
-    parser = argparse.ArgumentParser(description="Open Codex - Natural Language to CLI commands")
-    parser.add_argument("prompt", nargs="*", help="Optional prompt for one-shot mode")
+    parser = argparse.ArgumentParser()
+    parser.add_argument("prompt", nargs="*", help="Natural language prompt")
     args = parser.parse_args()
     prompt = " ".join(args.prompt).strip()
+
     if not prompt or prompt == "--help":
         print_help_message()
-        sys.exit(1)
-    
-    agent = AgentBuilder.get_agent()
-    print(f"{BLUE}Using model: phi-4-mini-instruct{RESET}")
-    one_shot_mode(agent, prompt)
+        sys.exit(0)
+
+    print(f"{BLUE}Prompt: {prompt}{RESET}", flush=True)
+    one_shot_mode(prompt)
 
 if __name__ == "__main__":
     main()