raggiecode 0.2.1__py3-none-any.whl

Agent/command.py

@@ -0,0 +1,49 @@
+class CommandRegistry:
+    """Registry for slash commands and other user-facing commands.
+
+    Commands are registered with a prefix (e.g. ``/undo``, ``!``) and a
+    handler function.  When the user's prompt starts with a registered
+    prefix, the matching handler is invoked with the remaining argument
+    string and the agent instance.
+
+    Handler signature::
+
+        def handle(args: str, agent: Agent) -> str | None
+
+    *args* is everything after the prefix (stripped).  Returning ``None``
+    signals that the command was a no-op (e.g. empty ``!`` with no command).
+    """
+
+    def __init__(self):
+        self.commands = {}
+
+    def register(self, prefix, handler):
+        """Register a command handler for a given prefix."""
+        self.commands[prefix] = handler
+
+
+    # returns <bool>, <str>
+    # if bool is true the agent need to use <str> as prompt
+    # no action is needed form the agent and it will ignore the string
+    def try_handle(self, prompt, agent):
+        command = ""
+        args = ""
+        if len(prompt) >= 1 and prompt[0] == "!":
+            command = "!"
+            args = prompt[1:].strip()
+        elif not " " in prompt:
+            command = prompt
+        else:
+            command = prompt[:prompt.index(" ")]
+            args = prompt[prompt.index(" ") + 1:].strip()
+
+        handler = self.commands.get(command)
+
+        if handler == None:
+            return True, prompt # process normal prompts that does not match the rules must be processed by the agent
+        
+        result = handler(args, agent)
+        if result == "":
+            return False, "" #do not serve the agent any prompt. what string you return do not matter here
+        else:
+            return True, result

Agent/config.py

@@ -0,0 +1,46 @@
+import json
+import shutil
+from pathlib import Path
+
+# The user's config directory (~/.config/raggie)
+USER_CONFIG_DIR = Path.home() / ".config" / "raggie"
+
+# The default config directory in the source code
+DEFAULT_CONFIG_DIR = Path(__file__).parent.parent / "config"
+
+def ensure_config_exists(filename):
+    USER_CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+    user_file = USER_CONFIG_DIR / filename
+    
+    # If the user doesn't have the config file yet, copy it from the defaults
+    if not user_file.exists():
+        default_file = DEFAULT_CONFIG_DIR / filename
+        if default_file.exists():
+            shutil.copy2(default_file, user_file)
+        else:
+            # Fallback if somehow the default doesn't exist
+            user_file.write_text("{}")
+
+    return user_file
+
+def load_roles():
+    config_file = ensure_config_exists("roles.json")
+    with open(config_file, "r") as f:
+        return json.load(f)
+
+
+def save_roles(roles: dict):
+    """Save roles dict to the user's roles.json file."""
+    config_file = ensure_config_exists("roles.json")
+    with open(config_file, "w") as f:
+        json.dump(roles, f, indent=4)
+
+def load_tools():
+    config_file = ensure_config_exists("tools.json")
+    with open(config_file, "r") as f:
+        return json.load(f)
+
+def load_keys():
+    config_file = ensure_config_exists("keys.json")
+    with open(config_file, "r") as f:
+        return json.load(f)

Agent/effort_levels.py

@@ -0,0 +1,33 @@
+EFFORT_LEVELS = {
+    1: {"name": "Zen", "max_depth": 1},
+    2: {"name": "Serious", "max_depth": 2},
+    3: {"name": "Extreme", "max_depth": 4},
+    4: {"name": "Feral", "max_depth": 8},
+    5: {"name": "Insane", "max_depth": 16},
+}
+
+DEFAULT_EFFORT = 1
+UNLIMITED_EFFORT = 99
+
+
+def effort_name(effort_num):
+    if effort_num == UNLIMITED_EFFORT:
+        return "Unlimited"
+    entry = EFFORT_LEVELS.get(effort_num)
+    if entry is None:
+        return "Unknown"
+    return entry["name"]
+
+
+def effort_max_depth(effort_num):
+    entry = EFFORT_LEVELS.get(effort_num)
+    if entry is None:
+        return None
+    return entry["max_depth"]
+
+
+def is_depth_allowed(effort_num, depth):
+    max_depth = effort_max_depth(effort_num)
+    if max_depth is None:
+        return True
+    return depth < max_depth