code update

Author: codingmoh

.github/demo.gif

@@ -235,3 +235,4 @@ pyrightconfig.json
 
 .idea/
 .vscode/
+brew_formula/

.gitignore

@@ -1,8 +1,7 @@
 # Open Codex
 
-<h1 align="center">Open Codex Codex CLI</h1>
+<h1 align="center">Open Codex CLI</h1>
 <p align="center">Lightweight coding agent that runs in your terminal</p>
-
 <p align="center"><code>
 brew install brew tap codingmoh/open-codex
 brew install open-codex

README.md

@@ -1,7 +1,7 @@
 [project]
 name = "open_codex"
-version = "0.1.9"
-description = "CLI with OSS LLM integration"
+version = "0.1.12"
+description = "AI CLI with OSS LLM integration"
 readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
@@ -23,3 +23,6 @@ package-dir = {"" = "src"}
 
 [tool.setuptools.packages.find]
 where = ["src"]
+
+[tool.setuptools.package-data]
+"open_codex" = ["resources/*.txt"]

pyproject.toml

@@ -1,3 +1,4 @@
+from importlib.resources import files
 
 from open_codex.agents.phi_4_mini import AgentPhi4Mini
 from open_codex.interfaces.llm_agent import LLMAgent
@@ -6,8 +7,8 @@ class AgentBuilder:
 
     @staticmethod
     def get_agent() -> LLMAgent:
-        prompt = AgentBuilder.read_file("src/open_codex/resources/prompt.txt")
-        return AgentPhi4Mini(system_prompt=prompt)
+        system_prompt = files("open_codex.resources").joinpath("prompt.txt").read_text(encoding="utf-8")
+        return AgentPhi4Mini(system_prompt=system_prompt)
 
     @staticmethod
     def read_file(file_path: str) -> str:

src/open_codex/agent_builder.py

@@ -3,7 +3,6 @@
 from open_codex.interfaces.llm_agent import LLMAgent
 import contextlib
 import os
-import sys
 
 
 class AgentPhi4Mini(LLMAgent):
@@ -13,7 +12,7 @@ def __init__(self, system_prompt: str):
         # which prints a lot of warnings and errors to stderr
         # when loading the model
         # this is a temporary solution until the library is fixed
-        with AgentPhi4Mini.suppress_stderr():
+        with AgentPhi4Mini.suppress_native_stderr():
             self.llm: Llama = Llama.from_pretrained( # type: ignore
                 repo_id="lmstudio-community/Phi-4-mini-instruct-GGUF",
                 filename="Phi-4-mini-instruct-Q3_K_L.gguf",
@@ -25,7 +24,8 @@ def one_shot_mode(self, user_input: str) -> str:
         chat_history = [{"role": "system", "content": self.system_prompt}]
         chat_history.append({"role": "user", "content": user_input})
         full_prompt = self.format_chat(chat_history)
-        output_raw = self.llm(prompt=full_prompt, max_tokens=100, temperature=0.2, stream=False)
+        with AgentPhi4Mini.suppress_native_stderr():
+            output_raw = self.llm(prompt=full_prompt, max_tokens=100, temperature=0.2, stream=False)
 
         # unfortuntely llama_cpp has a union type for the output
         output = cast(CreateCompletionResponse, output_raw)
@@ -44,11 +44,16 @@ def format_chat(self, messages: List[dict[str, str]]) -> str:
 
     @contextlib.contextmanager
     @staticmethod
-    def suppress_stderr():
-        with open(os.devnull, 'w') as devnull:
-            old_stderr = sys.stderr
-            sys.stderr = devnull
-            try:
-                yield
-            finally:
-                sys.stderr = old_stderr
+    def suppress_native_stderr():
+        """
+        Redirect C‐level stderr (fd 2) into /dev/null, so llama.cpp logs vanish.
+        """
+        devnull_fd = os.open(os.devnull, os.O_WRONLY)
+        saved_stderr_fd = os.dup(2)
+        try:
+            os.dup2(devnull_fd, 2)
+            yield
+        finally:
+            os.dup2(saved_stderr_fd, 2)
+            os.close(devnull_fd)
+            os.close(saved_stderr_fd)

src/open_codex/agents/phi_4_mini.py

@@ -36,48 +36,61 @@ def get_keypress():
             termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
         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}====================={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
+
+    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":
+        pyperclip.copy(command)
+        print(f"{GREEN}Command copied to clipboard! Paste it manually in your terminal.{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):
     try:
         response = agent.one_shot_mode(prompt)
-        print(f"{GREEN}: {response}{RESET}") 
-        command = response.strip()
-
-        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
-
-        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":
-            pyperclip.copy(command)
-            print(f"{GREEN}Command copied to clipboard! Paste it manually in your terminal.{RESET}")
-
-        elif choice == "a":
-            print(f"{BLUE}Aborted.{RESET}")
-        else:
-            print(f"{RED}Unknown choice. Nothing happened.{RESET}")
-
+        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")
     args = parser.parse_args()
-
     prompt = " ".join(args.prompt).strip()
-    if not prompt:
-        print("Please provide a prompt")
+    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)
 
 if __name__ == "__main__":

src/open_codex/main.py

@@ -1,5 +1,10 @@
-You are a helpful AI that converts natural language instructions  into valid shell commands. Always provide a concise shell command. 
-No shebang, no explanations, no extra text. Just the command. Think hard about the command you are going to provide, and make sure it is the best one.
-Stick to the easiest command possible, and if there is a simpler one, provide it.
-If the command is not straightforward, and there is an easier one, that can be run using a 3rd party tool, 
-INFORM the user about it, and provide the command to install it.
+You are a helpful AI that helps user identify the shell command they are looking for. 
+Always provide a concise shell command. 
+
+Important:
+- Think hard about the command you are going to provide, and make sure it is the best one.
+- No shebang
+- No explanations
+- No extra text. Just the command.
+
+User uses MacOS Catalina