mini-code-cli 0.0.8.0__py3-none-any.whl → 0.0.8.2__py3-none-any.whl

mini_code_cli/agent_loop.py

@@ -12,13 +12,16 @@ import os
 import json
 import signal
 
+import time
+import uuid
+
+import argparse
+import sys
 
 from . import prompts
 from . import tools
 from . import utils
 
-import argparse
-import sys
 
 # ANSI escape codes for colors and bold
 BOLD = '\033[1m'
@@ -30,11 +33,21 @@ MAGENTA = '\033[95m'
 CYAN = '\033[96m'
 RESET = '\033[0m'
 
-def save_history(messages):
+def save_history(messages, args, session_name=""):
     # Save messages to history.json
-    with open("history.json", "w") as f:
+    if args.cache_dir:
+        cache_dir = args.cache_dir
+    else:
+        cache_dir = os.path.expanduser("~/.cache/mini_code/sessions/")
+    os.makedirs(cache_dir, exist_ok=True)
+    history_path = os.path.join(cache_dir, f"history_{session_name}.json")
+    with open(history_path, 'w') as f:
         json.dump(messages, f, indent=2)
 
+def generate_unique_id():
+    """Generate a unique ID based on current time and a random component."""
+    timestamp = int(time.time() * 1000000)  # microseconds
+    return f"{timestamp}-{uuid.uuid4().hex[:8]}"
 
 def parse_args():
     parser = argparse.ArgumentParser(description="Agent loop for interacting with a language model")
@@ -45,10 +58,14 @@ def parse_args():
     parser.add_argument("--model", type=str, default="default", help="Model name")
     parser.add_argument("--api-key", type=str, default=None, help="API key for authentication. If not set, it tries to find it in env variable: MINI_CODE_API_KEY.")
 
+    # Other settings:
+    parser.add_argument("--cache-dir", type=str, help="The cache dir for the session histories.")
+
     # Agent behaviour related
     parser.add_argument("--system-prompt", type=str, help="Replace system prompt, with a custom system prompt.")
     parser.add_argument("--auto-mode", action="store_true", help="Whether to run the agent in `auto-mode'. Or default: `manual-mode'.")
     parser.add_argument("--agent-md", type=str, help="If the agent should use an agent-md file... (it will added after system message.)")
+    parser.add_argument("--prompt", type=str, help="An initial prompt from the user...")
 
     # Agent permissions related
     parser.add_argument("--ask-permission", action="store_true", help="Ask for permission before any tool call.")
@@ -60,7 +77,7 @@ def parse_args():
     return args
 
 
-def print_welcome(allowed_dir, server_url, args):
+def print_welcome(allowed_dir, server_url, args, unique_id, cache_dir):
     # Calculate the length of the longest line inside the box
     version = __import__('mini_code_cli').__version__
     
@@ -70,6 +87,8 @@ def print_welcome(allowed_dir, server_url, args):
 
     lines = [
         f"  MiniCodeCLI (v{version})",
+        f"  The session-id is: {unique_id}" ,
+        f"  It will be saved under: `{cache_dir}`",
          "  Type 'quit' or 'exit' to exit the loop.",
         f"  Allowed dir: {allowed_dir}",
         f"  Server URL: {server_url}",
@@ -90,8 +109,8 @@ def print_welcome(allowed_dir, server_url, args):
     bottom = "╚" + "═" * inner_width + "╝"
     print(bottom)
 
-def print_help(allowed_dir, server_url, args):
-    print_welcome(allowed_dir, server_url, args)
+def print_help(allowed_dir, server_url, args, unique_id, cache_dir):
+    print_welcome(allowed_dir, server_url, args, unique_id, cache_dir)
     
     help_message = ""
     help_message += "Manual-mode means the agent will ask for user input after each LLM prediction." if not args.auto_mode else "Auto-mode means the agent will keep calling the LLM until no further LLM calls are possible."
@@ -119,9 +138,14 @@ def call_function_with_timeout(func, tool_params, name):
     
     return result
 
+
+
 def main():
     args = parse_args()
 
+    # Get session unique ID:
+    unique_id = generate_unique_id()
+    cache_dir = args.cache_dir if args.cache_dir else "~/.cache/mini_code/sessions/"
 
     # LLM Related
     server_url = f"{args.url}"
@@ -172,7 +196,7 @@ def main():
         # Append the content as a user message after the system prompt
         messages.append({"role": "user", "content": agent_md_content})
 
-    print_welcome(allowed_dir, server_url, args)
+    print_welcome(allowed_dir, server_url, args, unique_id, cache_dir)
 
     # Agent Logic Flags
     llm_response_flag = False
@@ -180,130 +204,157 @@ def main():
     llm_repeat_flag = False
     tool_count = 0
     
+    user_input = args.prompt
+
     while True:
-        if llm_tool_response_flag and args.auto_mode: #skip user input and try to call the model again until there no more tool calls.
-            print(f"{BOLD}{YELLOW}System:{RESET} Skipping user input. Continuing with LLM calls until no more tool calls. [Tool count={tool_count}]")
-            llm_repeat_flag = True
-        else:
-        # get user input
-            if llm_repeat_flag:
-                print(f"{BOLD}{YELLOW}System:{RESET} Called {tool_count} tools in total. Exiting tool loop.")
-
-            tool_count = 0
-            try:
-                print("-"*30)
-                user_input = input(f"{BOLD}{CYAN}You:{RESET} ")
-                if user_input.lower() in ["quit", "exit", "/quit", "/exit"]:
-                    print("\nThank you. Goodbye! Saving history...")
-                    save_history(messages)
-                    break
-            except KeyboardInterrupt:
-                print("\nThank you. Goodbye! Saving history...")
-                save_history(messages)
-                break
-            
-            if user_input == "/help":
-                print_help(allowed_dir, server_url, args)
-                continue
-
-            messages.append({"role": "user", "content": user_input})
-
-        llm_response_flag = False
-        llm_tool_response_flag = False
-
-        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,
-            api_key=api_key,
-        )
-
-        if response_text:
-            llm_response_flag = True
-            print(f"{BOLD}{GREEN}Assistant:{RESET} {response_text}")
-
-        if tool_calls:
-            llm_tool_response_flag = True
-            
-            messages.append({
-                    "role": "assistant",
-                    "content": response_text,
-                    "tool_calls": 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:
-                tool_count += 1
-
-                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}")
+        try:
+
+            if llm_tool_response_flag and args.auto_mode: #skip user input and try to call the model again until there no more tool calls.
+                print(f"{BOLD}{YELLOW}System:{RESET} Skipping user input. Continuing with LLM calls until no more tool calls. [Tool count={tool_count}]")
+                llm_repeat_flag = True
+            else:
+            # get user input
+                if llm_repeat_flag:
+                    print(f"{BOLD}{YELLOW}System:{RESET} Called {tool_count} tools in total. Exiting tool loop.")
+
+                tool_count = 0
+                if not user_input:
+                    try:
+                        print("-"*30)
+                        user_input = input(f"{BOLD}{CYAN}You:{RESET} ")
+                        if user_input.lower() in ["quit", "exit", "/quit", "/exit"]:
+                            print("\nThank you. Goodbye!")
+                            print(f"Saving history to: {cache_dir}/{unique_id}")
+                            save_history(messages, args, unique_id)
+                            break
+                    except KeyboardInterrupt:
+                        print("\nThank you. Goodbye!")
+                        print(f"Saving history to: {cache_dir}/{unique_id}")
+                        save_history(messages, args, unique_id)
+                        break
+                else:
+                    print("-"*30)
+                    print(f"{BOLD}{CYAN}You (preset): {RESET}{user_input}")
+                
+                if user_input == "/help":
+                    print_help(allowed_dir, server_url, args, unique_id, cache_dir)
+                    user_input = ""
                     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":
+                messages.append({"role": "user", "content": user_input})
+                user_input = ""
+
+            llm_response_flag = False
+            llm_tool_response_flag = False
+
+            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,
+                api_key=api_key,
+            )
+
+            if response_text:
+                llm_response_flag = True
+                print(f"{BOLD}{GREEN}Assistant:{RESET} {response_text}")
+
+            if tool_calls:
+                llm_tool_response_flag = True
+                
+                messages.append({
+                        "role": "assistant",
+                        "content": response_text,
+                        "tool_calls": 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:
+                    tool_count += 1
+
+                    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 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"]
-                    # asking permissions
-                    if args.ask_permission:
-                        print(f"{BOLD}{YELLOW}Tool permission:{RESET} Are you sure you want to call the tool?")
-                        print(f"name: {name}, parameters: {params_str}")
-                        try:
-                            permission_input = input("Type: 'y' or 'yes'\n")
-                        except KeyboardInterrupt:
-                            print("\nThank you. Goodbye! Saving history...")
-                            save_history(messages)
-                            sys.exit(0)
-
-                        if not permission_input in ['y', 'yes']:
-                            print(f"{BOLD}{YELLOW}Tool permission denied:{RESET}.")
-                            messages.append({
-                                "role": "user",
-                                "content": f"User denied permission for tool {name} and params {params_str}."
-                            })
-                            continue
-
-                    print(f"{BOLD}{YELLOW}Tool Call:{RESET} LLM is trying to call: {name} with {tool_params}.")
-                    result = call_function_with_timeout(func, tool_params, name)
-                    # 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}, params {params_str}")
-        else:
-            messages.append({
-                    "role": "assistant",
-                    "content": response_text,
-                    # "tool_calls": tool_calls
-            })
-        
-        if not llm_response_flag and not llm_tool_response_flag:
-            print(f"{BOLD}{RED}Warning:{RESET} No response was given by the LLM.")
+                            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"]
+                        # asking permissions
+                        if args.ask_permission:
+                            print(f"{BOLD}{YELLOW}Tool permission:{RESET} Are you sure you want to call the tool?")
+                            print(f"name: {name}, parameters: {params_str}")
+                            try:
+                                permission_input = input("Type: 'y' or 'yes'\n")
+                            except KeyboardInterrupt:
+                                print("\nThank you. Goodbye! Saving history...")
+                                save_history(messages, args, unique_id)
+                                sys.exit(0)
+
+                            if not permission_input in ['y', 'yes']:
+                                print(f"{BOLD}{YELLOW}Tool permission denied:{RESET}.")
+                                messages.append({
+                                    "role": "user",
+                                    "content": f"User denied permission for tool {name} and params {params_str}."
+                                })
+                                continue
+
+                        print(f"{BOLD}{YELLOW}Tool Call:{RESET} LLM is trying to call: {name} with {tool_params}.")
+                        result = call_function_with_timeout(func, tool_params, name)
+                        # 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}Error:{RESET} Unknown tool: {name}, params {params_str}")
+                        messages.append({
+                            "role": "tool",
+                            "tool_call_id": call_id,
+                            "content": f"Error: Unknown tool: {name}, params {params_str}"
+                        })
+            else:
+                messages.append({
+                        "role": "assistant",
+                        "content": response_text,
+                        # "tool_calls": tool_calls
+                })
+            
+            if not llm_response_flag and not llm_tool_response_flag:
+                print(f"{BOLD}{RED}Warning:{RESET} No response was given by the LLM.")
+
+            save_history(messages, args, unique_id)
 
-        save_history(messages)
+        except KeyboardInterrupt:
+            print(f"{BOLD}{RED}INTERRUPTING:{RESET} Mini Code Exiting...")
+
+        except Exception as e:
+            print(f"{BOLD}{RED}FATAL ERROR:{RESET} Mini Code Crashed...:\n{e}")
+        
+        finally:
+            print(f"Saving history to: {cache_dir}/{unique_id}")
+            save_history(messages, args, unique_id)
 
 if __name__ == "__main__":
     main()

mini_code_cli/prompts.py

@@ -6,7 +6,7 @@ def system_prompt(tools_dict=None, allowed_dir=None):
 
     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 assistant with access to the following tools:\n"
+        base_prompt = "You are a helpful AI assistant with access to the following tools (only):\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. "

{mini_code_cli-0.0.8.0.dist-info → mini_code_cli-0.0.8.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mini-code-cli
-Version: 0.0.8.0
+Version: 0.0.8.2
 Summary: Minimal Coding CLI
 Author-email: Nikolai Rozanov <nikolai.rozanov@gmail.com>
 License: The MIT License (MIT)
@@ -63,6 +63,30 @@ Write an agent.md file for me that helps an LLM write efficient triton kernels,
 ```bash
 mini-code --help
 ```
+```
+Agent loop for interacting with a language model
+
+options:
+  -h, --help            show this help message and exit
+  --url URL             API host
+  --max-tokens MAX_TOKENS
+                        Maximum tokens for LLM response
+  --temperature TEMPERATURE
+                        Temperature for LLM
+  --model MODEL         Model name
+  --api-key API_KEY     API key for authentication. If not set, it tries to find it in env variable: MINI_CODE_API_KEY.
+  --cache-dir CACHE_DIR
+                        The cache dir for the session histories.
+  --system-prompt SYSTEM_PROMPT
+                        Replace system prompt, with a custom system prompt.
+  --auto-mode           Whether to run the agent in `auto-mode'. Or default: `manual-mode'.
+  --agent-md AGENT_MD   If the agent should use an agent-md file... (it will added after system message.)
+  --prompt PROMPT       An initial prompt from the user...
+  --ask-permission      Ask for permission before any tool call.
+  --allowed-dir ALLOWED_DIR
+                        Allowed directory for file operations
+  --enable-shell        Allow shell execution. Default: False
+  ```
 
 
 ## (C) Nikolai Rozanov, 2026 - Present

mini_code_cli-0.0.8.2.dist-info/RECORD

@@ -0,0 +1,11 @@
+mini_code_cli/__init__.py,sha256=iNgHmKSHOuH99obGjnUMoS2JP4Gw4AOrIfu68EYaZQM,77
+mini_code_cli/agent_loop.py,sha256=V74UYHU9y4ZYV_OtYkDpFWJnuEgmTRP36Uxc4AP89y0,14783
+mini_code_cli/prompts.py,sha256=hIX-A_F7RBuTAAwYaLaDA6JMC1lhv9Fkx1WkS5v83TU,1992
+mini_code_cli/tools.py,sha256=R8DWRu0zij2Xwioldd6tDeMtVfSaZhqcWAsjvskOxvw,5627
+mini_code_cli/utils.py,sha256=HlmuLH0V4AxZTg0_ZGzlxffqRqs7T3W6FQXaBhioGdk,3647
+mini_code_cli-0.0.8.2.dist-info/licenses/LICENSE,sha256=WCkMINUFi--UtEg_alEE2FBKPtNKkzi7dGzaxzchifc,1088
+mini_code_cli-0.0.8.2.dist-info/METADATA,sha256=GpP0D_izk9dRFw7i3qUzV_2AatFQPsBceunIU1qmFbc,4284
+mini_code_cli-0.0.8.2.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+mini_code_cli-0.0.8.2.dist-info/entry_points.txt,sha256=MwDp5UbXpI7wodSmOSYL8_nKutnBiyp96ZDDtaX7x14,60
+mini_code_cli-0.0.8.2.dist-info/top_level.txt,sha256=3Hb6PpNkJYXvzDKXuuFJRvzCdsTRh0TuRWz-ScJ6qV0,14
+mini_code_cli-0.0.8.2.dist-info/RECORD,,

mini_code_cli-0.0.8.0.dist-info/RECORD

@@ -1,11 +0,0 @@
-mini_code_cli/__init__.py,sha256=iNgHmKSHOuH99obGjnUMoS2JP4Gw4AOrIfu68EYaZQM,77
-mini_code_cli/agent_loop.py,sha256=m2X-53-0oM7ZJTZRzXHOurofcqdUXkr5W0xiRKXkZkg,12130
-mini_code_cli/prompts.py,sha256=1dY5f7pSvfDnjhA-uDmi0Rw66_JFQGfooF699pATO0E,1985
-mini_code_cli/tools.py,sha256=R8DWRu0zij2Xwioldd6tDeMtVfSaZhqcWAsjvskOxvw,5627
-mini_code_cli/utils.py,sha256=HlmuLH0V4AxZTg0_ZGzlxffqRqs7T3W6FQXaBhioGdk,3647
-mini_code_cli-0.0.8.0.dist-info/licenses/LICENSE,sha256=WCkMINUFi--UtEg_alEE2FBKPtNKkzi7dGzaxzchifc,1088
-mini_code_cli-0.0.8.0.dist-info/METADATA,sha256=njKXtDgl-aLKarbmgJ8TxHWFh6Olj3vY8dBCCmxUy4w,3146
-mini_code_cli-0.0.8.0.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
-mini_code_cli-0.0.8.0.dist-info/entry_points.txt,sha256=MwDp5UbXpI7wodSmOSYL8_nKutnBiyp96ZDDtaX7x14,60
-mini_code_cli-0.0.8.0.dist-info/top_level.txt,sha256=3Hb6PpNkJYXvzDKXuuFJRvzCdsTRh0TuRWz-ScJ6qV0,14
-mini_code_cli-0.0.8.0.dist-info/RECORD,,