mini-code-cli 0.0.6__py3-none-any.whl → 0.0.7__py3-none-any.whl

mini_code_cli/agent_loop.py

@@ -23,17 +23,40 @@ def save_history(messages):
     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, default=2048, help="Maximum tokens for LLM response")
+    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()
 
@@ -43,17 +66,14 @@ def main():
     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)
+    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(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║    
-    ╚═══════════════════════════════════════════════════════════════════╝
-    """)
+    print_welcome(allowed_dir, server_url)
 
     while True:
         llm_response_flag = False
@@ -90,42 +110,42 @@ def main():
         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"]:
+
+            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 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}")
+                        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
         

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()
 

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/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:

{mini_code_cli-0.0.6.dist-info → mini_code_cli-0.0.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mini-code-cli
-Version: 0.0.6
+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.dist-info/RECORD

@@ -0,0 +1,11 @@
+mini_code_cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mini_code_cli/agent_loop.py,sha256=IC02Hs2oBx4C1apFXmJ2E3ESmJLP8v2Z6_0ApLDdZo4,5877
+mini_code_cli/prompts.py,sha256=1dY5f7pSvfDnjhA-uDmi0Rw66_JFQGfooF699pATO0E,1985
+mini_code_cli/tools.py,sha256=RO51ZJ61NV_R8KqsW9me4-0HPhjgS3DMoKrASLBvYJI,4931
+mini_code_cli/utils.py,sha256=HlmuLH0V4AxZTg0_ZGzlxffqRqs7T3W6FQXaBhioGdk,3647
+mini_code_cli-0.0.7.dist-info/licenses/LICENSE,sha256=WCkMINUFi--UtEg_alEE2FBKPtNKkzi7dGzaxzchifc,1088
+mini_code_cli-0.0.7.dist-info/METADATA,sha256=6_sPZqA8K3sMvMSufEi6__6favkr6O63yBdq17ZC5S0,1943
+mini_code_cli-0.0.7.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+mini_code_cli-0.0.7.dist-info/entry_points.txt,sha256=MwDp5UbXpI7wodSmOSYL8_nKutnBiyp96ZDDtaX7x14,60
+mini_code_cli-0.0.7.dist-info/top_level.txt,sha256=3Hb6PpNkJYXvzDKXuuFJRvzCdsTRh0TuRWz-ScJ6qV0,14
+mini_code_cli-0.0.7.dist-info/RECORD,,

mini_code_cli-0.0.6.dist-info/RECORD

@@ -1,11 +0,0 @@
-mini_code_cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mini_code_cli/agent_loop.py,sha256=qrrPAdVZY02xAXNYZGb5s_JhdKPgQdXmunC0Is6nidY,5704
-mini_code_cli/prompts.py,sha256=A8nLnYuCWFuge-H5g7hYdRwXJPnRCg4uH__yYq4PzIU,1737
-mini_code_cli/tools.py,sha256=i46l8IvsrXJgqt7r8rKAMTGKNGAGffxKTFlwmAuWoD0,4926
-mini_code_cli/utils.py,sha256=qQk0RurNeWYoyRj-QmQ81Pq1pytJO87SUzm-XzDz9s4,3643
-mini_code_cli-0.0.6.dist-info/licenses/LICENSE,sha256=WCkMINUFi--UtEg_alEE2FBKPtNKkzi7dGzaxzchifc,1088
-mini_code_cli-0.0.6.dist-info/METADATA,sha256=43JAsEkm4zknm4jKtMDyjbTLTIG8d3XPGCSb2KUWRy8,1943
-mini_code_cli-0.0.6.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
-mini_code_cli-0.0.6.dist-info/entry_points.txt,sha256=MwDp5UbXpI7wodSmOSYL8_nKutnBiyp96ZDDtaX7x14,60
-mini_code_cli-0.0.6.dist-info/top_level.txt,sha256=3Hb6PpNkJYXvzDKXuuFJRvzCdsTRh0TuRWz-ScJ6qV0,14
-mini_code_cli-0.0.6.dist-info/RECORD,,