mini-code-cli 0.0.5__tar.gz → 0.0.7__tar.gz

{mini_code_cli-0.0.5 → mini_code_cli-0.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mini-code-cli
-Version: 0.0.5
+Version: 0.0.7
 Summary: Minimal Coding CLI
 Author-email: Nikolai Rozanov <nikolai.rozanov@gmail.com>
 License: The MIT License (MIT)

mini_code_cli-0.0.7/mini_code_cli/agent_loop.py

@@ -0,0 +1,158 @@
+import os
+import json
+
+from . import prompts
+from . import tools
+from . import utils
+
+import argparse
+import sys
+
+# ANSI escape codes for colors and bold
+BOLD = '\033[1m'
+RED = '\033[91m'
+GREEN = '\033[92m'
+YELLOW = '\033[93m'
+BLUE = '\033[94m'
+MAGENTA = '\033[95m'
+CYAN = '\033[96m'
+RESET = '\033[0m'
+
+def save_history(messages):
+    # Save messages to history.json
+    with open("history.json", "w") as f:
+        json.dump(messages, f, indent=2)
+
+
+def parse_args():
+    parser = argparse.ArgumentParser(description="Agent loop for interacting with a language model")
+    parser.add_argument("--url", type=str, default="http://0.0.0.0:30000", help="API host")
+    parser.add_argument("--system_prompt", type=str, help="API host")
+    parser.add_argument("--api-key", type=str, default=None, help="API key for authentication")
+    parser.add_argument("--allowed-dir", type=str, default=".", help="Allowed directory for file operations")
+    parser.add_argument("--max-tokens", type=int, help="Maximum tokens for LLM response")
+    parser.add_argument("--temperature", type=float, default=0.0, help="Temperature for LLM")
+    parser.add_argument("--model", type=str, default="default", help="Model name")
+    args = parser.parse_args()
+    return args
+
+def print_welcome(allowed_dir, server_url):
+    # Calculate the length of the longest line inside the box
+    lines = [
+        "       Coding Agent Loop Started",
+        "  Type 'quit' or 'exit' to exit the loop.",
+        f"  Allowed dir: {allowed_dir}",
+        f"  Server URL: {server_url}"
+    ]
+    max_len = max(len(line) for line in lines)
+    margin = 2  # Space on each side inside the box
+    inner_width = max_len + 2 * margin
+    # Build the box
+    top_bottom = "╔" + "═" * inner_width + "╗"
+    print(top_bottom)
+    for line in lines:
+        # Pad the line to inner_width characters
+        padded = line.ljust(inner_width)
+        print(f"║{padded}║")
+    bottom = "╚" + "═" * inner_width + "╝"
+    print(bottom)
+
+def main():
+    args = parse_args()
+
+    allowed_dir = os.path.abspath(args.allowed_dir)
+    server_url = f"{args.url}"
+
+    tools_dict = tools.util_get_tools_dict()
+    llm_tools_dict = tools.util_get_tools()
+
+    if args.system_prompt:
+        system_prompt = args.system_prompt
+    else:
+        system_prompt = prompts.system_prompt(tools_dict=tools_dict, allowed_dir=allowed_dir)
+
+    messages = [{"role": "system", "content": system_prompt}]
+
+    print_welcome(allowed_dir, server_url)
+
+    while True:
+        llm_response_flag = False
+        llm_tool_response_flag = False
+        try:
+            print("-"*30)
+            user_input = input(f"{BOLD}{CYAN}You:{RESET} ")
+            if user_input.lower() in ["quit", "exit"]:
+                print("\nThank you. Goodbye! Saving history...")
+                save_history(messages)
+                break
+        except KeyboardInterrupt:
+            print("\nThank you. Goodbye! Saving history...")
+            save_history(messages)
+            break
+
+        messages.append({"role": "user", "content": user_input})
+
+        response_text, tool_calls = utils.call_openai_server(
+            messages, 
+            max_tokens=args.max_tokens, 
+            temperature=args.temperature, 
+            model=args.model, 
+            server_url=server_url, 
+            tools=llm_tools_dict
+        )
+
+        if response_text:
+            print(f"{BOLD}{GREEN}Assistant:{RESET} {response_text}")
+            messages.append({"role": "assistant", "content": response_text})
+        else:
+            llm_response_flag = True
+
+        if tool_calls:
+            # Tool call parsing and execution
+            print(f"{BOLD}{YELLOW}Tool Call:{RESET} LLM is trying to call {len(tool_calls)} tools.")
+
+            for call in tool_calls:
+                call_id = call.get("id", "")
+                name = call.get("function", {}).get("name", "")
+                params_str = call.get("function", {}).get("arguments", "{}")
+                try:
+                    tool_params = json.loads(params_str)
+                except json.JSONDecodeError:
+                    print(f"{BOLD}{RED}Warning:{RESET} Error parsing arguments for tool {name}")
+                    continue
+
+                if name in tools_dict:
+                    # Check allowed directory for file operations
+                    if name in ["read_file", "write_file", "grep", "glob"]:
+                        filepath = tool_params.get("filepath", "")
+                        if name == "write_file":
+                            filepath = tool_params.get("filepath", "")
+                        if filepath and not os.path.abspath(filepath).startswith(allowed_dir):
+                            print(f"{BOLD}{RED}Warning:{RESET} File operation not allowed outside allowed directory: {allowed_dir}")
+                            continue
+
+                    func = tools_dict[name]["function"]
+                    result = func(**tool_params)
+                    # Limiting output to 300 characters for readability
+                    result_str = str(result)
+                    if len(result_str) > 300:
+                        result_str = result_str[:300] + "... (truncated to 300 chars)"
+                    print(f"{BOLD}{MAGENTA}Tool '{name}' returned:{RESET} {result_str}")
+
+                    messages.append({
+                        "role": "tool",
+                        "tool_call_id": call_id,
+                        "content": str(result)
+                    })
+                else:
+                    print(f"{BOLD}{RED}Warning:{RESET} Unknown tool: {name}")
+        else:
+            llm_tool_response_flag = True
+        
+        if llm_response_flag and llm_tool_response_flag:
+            print(f"{BOLD}{RED}Warning:{RESET} No response was given by the LLM.")
+
+        save_history(messages)
+
+if __name__ == "__main__":
+    main()

{mini_code_cli-0.0.5 → mini_code_cli-0.0.7}/mini_code_cli/prompts.py

@@ -3,22 +3,30 @@ def system_prompt(tools_dict=None, allowed_dir=None):
     """
     Takes a list of tool names and returns a system prompt string.
     """
-    allowed_dir_prompt = ""
-    if allowed_dir:
-        allowed_dir_prompt = f"\nThe path you are allowed to operate in is called: {allowed_dir}"
+
     if tools_dict:
         tool_descriptions = "\n".join([f"{name} : {data['description']}" for name, data in tools_dict.items()])
-        base_prompt = "You are a helpful AI coding assistant with access to the following tools:\n"
-        end_prompt = "\n\nWhen answering, first use available tools if needed, then provide a compact summary of the results."
+        base_prompt = "You are a helpful AI assistant with access to the following tools:\n"
+        end_prompt = (
+        "Choose the most appropriate tool for the task at hand. "
+        "If a tool fails or you hit a limit, analyze the error and try a different approach. "
+        "Be concise and efficient in your tool usage."
+        )
     else:
         tool_descriptions = ""
-        base_prompt = "You are a helpful AI coding assistant without any tools."
+        base_prompt = "You are a helpful AI assistant without any tools."
         end_prompt = ""
+
+        
+    allowed_dir_prompt = ""
+    if allowed_dir:
+        allowed_dir_prompt = f"\nThe path you are allowed to operate in is called: {allowed_dir}"
+    
     
     prompt = f"""
 {base_prompt}
 {tool_descriptions}{end_prompt}{allowed_dir_prompt}
-Keep responses concise and focused.
+Important: Only follow the user instruction. Do not create additional .md files (unless asked to do so). Do not create unit tests (unless asked to do so), etc.
 """
     return prompt.strip()
 
@@ -42,7 +50,7 @@ Conversation:
     return prompt.strip()
 
 if __name__=="__main__":
-    import .tools
+    from . import tools
     tools_dict = tools.util_get_tools_dict()
     print(system_prompt(tools_dict))
 

{mini_code_cli-0.0.5 → mini_code_cli-0.0.7}/mini_code_cli/tools.py

@@ -31,7 +31,7 @@ def grep(pattern, filepath):
     except Exception as e:
         return f"Error: {e}"
 
-def glob(pattern):
+def glob(pattern="./"):
     """Return list of file paths matching the given glob pattern. Param: pattern"""
     try:
         return glob_module.glob(pattern)

{mini_code_cli-0.0.5 → mini_code_cli-0.0.7}/mini_code_cli/utils.py

@@ -19,9 +19,9 @@ def call_openai_server(messages, max_tokens=2048, temperature=0.0, top_p=0.95, m
     payload = {
         "model": model,
         "messages": messages,
-        "max_tokens": max_tokens,
         "temperature": temperature,
-        "top_p": top_p
+        # "max_tokens": max_tokens,
+        # "top_p": top_p
     }
 
     if tools:
@@ -72,7 +72,7 @@ def parse_args():
 
 
 if __name__ == '__main__':
-    import .tools
+    from . import tools
     tools_dict = tools.util_get_tools()
 
     args = parse_args()

{mini_code_cli-0.0.5 → mini_code_cli-0.0.7}/mini_code_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mini-code-cli
-Version: 0.0.5
+Version: 0.0.7
 Summary: Minimal Coding CLI
 Author-email: Nikolai Rozanov <nikolai.rozanov@gmail.com>
 License: The MIT License (MIT)

{mini_code_cli-0.0.5 → mini_code_cli-0.0.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mini-code-cli"
-version = "0.0.5"
+version = "0.0.7"
 description = "Minimal Coding CLI"
 readme = "README.md"
 requires-python = ">=3.10"

mini_code_cli-0.0.5/mini_code_cli/agent_loop.py

@@ -1,138 +0,0 @@
-import os
-import json
-
-import .prompts
-import .tools
-import .utils
-
-import argparse
-import sys
-
-# ANSI escape codes for colors and bold
-BOLD = '\033[1m'
-RED = '\033[91m'
-GREEN = '\033[92m'
-YELLOW = '\033[93m'
-BLUE = '\033[94m'
-MAGENTA = '\033[95m'
-CYAN = '\033[96m'
-RESET = '\033[0m'
-
-def save_history(messages):
-    # Save messages to history.json
-    with open("history.json", "w") as f:
-        json.dump(messages, f, indent=2)
-
-def parse_args():
-    parser = argparse.ArgumentParser(description="Agent loop for interacting with a language model")
-    parser.add_argument("--url", type=str, default="http://0.0.0.0:30000", help="API host")
-    parser.add_argument("--api-key", type=str, default=None, help="API key for authentication")
-    parser.add_argument("--allowed-dir", type=str, default=".", help="Allowed directory for file operations")
-    parser.add_argument("--max-tokens", type=int, default=2048, help="Maximum tokens for LLM response")
-    parser.add_argument("--temperature", type=float, default=0.0, help="Temperature for LLM")
-    parser.add_argument("--model", type=str, default="default", help="Model name")
-    args = parser.parse_args()
-    return args
-
-def main():
-    args = parse_args()
-
-    allowed_dir = os.path.abspath(args.allowed_dir)
-    server_url = f"{args.url}"
-
-    tools_dict = tools.util_get_tools_dict()
-    llm_tools_dict = tools.util_get_tools()
-
-    system_prompt = prompts.system_prompt(tools_dict=tools_dict, allowed_dir=allowed_dir)
-
-    messages = [{"role": "system", "content": system_prompt}]
-
-    print(f"""
-    ╔═══════════════════════════════════════════════════════════════════╗
-    ║       Coding Agent Loop Started\t\t\t\t\t║
-    ║  Type 'quit' or 'exit' to exit the loop.\t\t\t\t║
-    ║  Allowed dir: {allowed_dir}\t║    
-    ╚═══════════════════════════════════════════════════════════════════╝
-    """)
-
-    while True:
-        llm_response_flag = False
-        llm_tool_response_flag = False
-        try:
-            print("-"*30)
-            user_input = input(f"{BOLD}{CYAN}You:{RESET} ")
-            if user_input.lower() in ["quit", "exit"]:
-                print("\nThank you. Goodbye! Saving history...")
-                save_history(messages)
-                break
-        except KeyboardInterrupt:
-            print("\nThank you. Goodbye! Saving history...")
-            save_history(messages)
-            break
-
-        messages.append({"role": "user", "content": user_input})
-
-        response_text, tool_calls = utils.call_openai_server(
-            messages, 
-            max_tokens=args.max_tokens, 
-            temperature=args.temperature, 
-            model=args.model, 
-            server_url=server_url, 
-            tools=llm_tools_dict
-        )
-
-        if response_text:
-            print(f"{BOLD}{GREEN}Assistant:{RESET} {response_text}")
-            messages.append({"role": "assistant", "content": response_text})
-        else:
-            llm_response_flag = True
-
-        if tool_calls:
-            # Tool call parsing and execution
-            print(f"{BOLD}{YELLOW}Tool Call:{RESET} LLM is trying to call {len(tool_calls)} tools.")
-            if tool_calls:
-                for call in tool_calls:
-                    call_id = call.get("id", "")
-                    name = call.get("function", {}).get("name", "")
-                    params_str = call.get("function", {}).get("arguments", "{}")
-                    try:
-                        tool_params = json.loads(params_str)
-                    except json.JSONDecodeError:
-                        print(f"{BOLD}{RED}Warning:{RESET} Error parsing arguments for tool {name}")
-                        continue
-
-                    if name in tools_dict:
-                        # Check allowed directory for file operations
-                        if name in ["read_file", "write_file", "grep", "glob"]:
-                            filepath = tool_params.get("filepath", "")
-                            if name == "write_file":
-                                filepath = tool_params.get("filepath", "")
-                            if filepath and not os.path.abspath(filepath).startswith(allowed_dir):
-                                print(f"{BOLD}{RED}Warning:{RESET} File operation not allowed outside allowed directory: {allowed_dir}")
-                                continue
-
-                        func = tools_dict[name]["function"]
-                        result = func(**tool_params)
-                        # Limiting output to 300 characters for readability
-                        result_str = str(result)
-                        if len(result_str) > 300:
-                            result_str = result_str[:300] + "... (truncated to 300 chars)"
-                        print(f"{BOLD}{MAGENTA}Tool '{name}' returned:{RESET} {result_str}")
-
-                        messages.append({
-                            "role": "tool",
-                            "tool_call_id": call_id,
-                            "content": str(result)
-                        })
-                    else:
-                        print(f"{BOLD}{RED}Warning:{RESET} Unknown tool: {name}")
-        else:
-            llm_tool_response_flag = True
-        
-        if llm_response_flag and llm_tool_response_flag:
-            print(f"{BOLD}{RED}Warning:{RESET} No response was given by the LLM.")
-
-        save_history(messages)
-
-if __name__ == "__main__":
-    main()