npcsh 1.1.3__py3-none-any.whl → 1.1.5__py3-none-any.whl

npcsh/routes.py

@@ -1,83 +1,17 @@
-
-
-from typing import Callable, Dict, Any, List, Optional, Union
+from typing import Callable, Dict, Any, List, Optional
 import functools
 import os
-import subprocess
-import sys
-from pathlib import Path
-
 import traceback
-import shlex
-import time
-from datetime import datetime
-from sqlalchemy import create_engine
-import logging 
-import json 
-from npcpy.data.load import load_file_contents
-
-from npcpy.llm_funcs import (
-    get_llm_response,
-    gen_image,
-    gen_video,
-    breathe,
-)
-from npcpy.npc_compiler import initialize_npc_project
-from npcpy.npc_sysenv import render_markdown
-from npcpy.work.plan import execute_plan_command
-from npcpy.work.trigger import execute_trigger_command
-from npcpy.work.desktop import perform_action
-from npcpy.memory.search import execute_rag_command, execute_search_command, execute_brainblast_command
-from npcpy.memory.command_history import CommandHistory, load_kg_from_db, save_kg_to_db
-from npcpy.serve import start_flask_server
-from npcpy.mix.debate import run_debate
-from npcpy.data.image import capture_screenshot
-from npcpy.npc_compiler import NPC, Team, Jinx,initialize_npc_project
-from npcpy.data.web import search_web
-from npcpy.memory.knowledge_graph import kg_sleep_process, kg_dream_process
-
-
-from npcsh._state import (
-    NPCSH_VISION_MODEL, 
-    NPCSH_VISION_PROVIDER, 
-    set_npcsh_config_value,
-    NPCSH_API_URL,
-    NPCSH_CHAT_MODEL, 
-    NPCSH_CHAT_PROVIDER, 
-    NPCSH_STREAM_OUTPUT,
-    NPCSH_IMAGE_GEN_MODEL, 
-    NPCSH_IMAGE_GEN_PROVIDER,
-    NPCSH_VIDEO_GEN_MODEL,
-    NPCSH_VIDEO_GEN_PROVIDER,
-    NPCSH_EMBEDDING_MODEL,
-    NPCSH_EMBEDDING_PROVIDER,
-    NPCSH_REASONING_MODEL,
-    NPCSH_REASONING_PROVIDER,
-    NPCSH_SEARCH_PROVIDER,
-    CANONICAL_ARGS, 
-    normalize_and_expand_flags, 
-    get_argument_help, 
-    get_relevant_memories
-
-)
-from npcsh.corca import enter_corca_mode
-from npcsh.guac import enter_guac_mode
-from npcsh.plonk import execute_plonk_command, format_plonk_summary
-from npcsh.alicanto import alicanto
-from npcsh.pti import enter_pti_mode
-from npcsh.spool import enter_spool_mode
-from npcsh.wander import enter_wander_mode
-from npcsh.yap import enter_yap_mode
-
-
+from pathlib import Path
 
-NPC_STUDIO_DIR = Path.home() / ".npcsh" / "npc-studio"
+from npcpy.npc_compiler import Jinx, load_jinxs_from_directory
 
 
 class CommandRouter:
     def __init__(self):
         self.routes = {}
         self.help_info = {}
+        self.jinx_routes = {}
 
     def route(self, command: str, help_text: str = "") -> Callable:
         def wrapper(func):
@@ -91,8 +25,80 @@ class CommandRouter:
             return wrapped_func
         return wrapper
 
+    def load_jinx_routes(self, jinxs_dir: str):
+        if not os.path.exists(jinxs_dir):
+            print(f"Jinxs directory not found: {jinxs_dir}")
+            return
+        
+        for root, dirs, files in os.walk(jinxs_dir):
+            for filename in files:
+                if filename.endswith('.jinx'):
+                    jinx_path = os.path.join(root, filename)
+                    try:
+                        jinx = Jinx(jinx_path=jinx_path)
+                        self.register_jinx(jinx)
+                    except Exception as e:
+                        print(f"Error loading jinx {filename}: {e}")
+    
+    def register_jinx(self, jinx: Jinx):
+        command_name = jinx.jinx_name
+        
+        def jinx_handler(command: str, **kwargs):
+            return self._execute_jinx(jinx, command, **kwargs)
+        
+        self.jinx_routes[command_name] = jinx_handler
+        self.help_info[command_name] = jinx.description or "Jinx command"
+    
+    def _execute_jinx(self, jinx: Jinx, command: str, **kwargs):
+        messages = kwargs.get("messages", [])
+        npc = kwargs.get('npc')
+        
+        try:
+            import shlex
+            parts = shlex.split(command)
+            args = parts[1:] if len(parts) > 1 else []
+            
+            input_values = {}
+            if hasattr(jinx, 'inputs') and jinx.inputs:
+                for i, input_spec in enumerate(jinx.inputs):
+                    if isinstance(input_spec, str):
+                        input_name = input_spec
+                    elif isinstance(input_spec, dict):
+                        input_name = list(input_spec.keys())[0]
+                    else:
+                        continue
+                    
+                    if i < len(args):
+                        input_values[input_name] = args[i]
+            
+            jinx_output = jinx.execute(
+                input_values=input_values,
+                jinxs_dict=kwargs.get('jinxs_dict', {}),
+                npc=npc,
+                messages=messages
+            )
+            
+            if isinstance(jinx_output, dict):
+                return {
+                    "output": jinx_output.get('output', str(jinx_output)),
+                    "messages": jinx_output.get('messages', messages)
+                }
+            else:
+                return {"output": str(jinx_output), "messages": messages}
+                
+        except Exception as e:
+            traceback.print_exc()
+            return {
+                "output": f"Error executing jinx '{jinx.jinx_name}': {e}",
+                "messages": messages
+            }
+
     def get_route(self, command: str) -> Optional[Callable]:
-        return self.routes.get(command)
+        if command in self.routes:
+            return self.routes[command]
+        elif command in self.jinx_routes:
+            return self.jinx_routes[command]
+        return None
 
     def execute(self, command_str: str, **kwargs) -> Any:
         command_name = command_str.split()[0].lstrip('/')
@@ -102,7 +108,8 @@ class CommandRouter:
         return None
 
     def get_commands(self) -> List[str]:
-        return list(self.routes.keys())
+        all_commands = list(self.routes.keys()) + list(self.jinx_routes.keys())
+        return sorted(set(all_commands))
 
     def get_help(self, command: str = None) -> Dict[str, str]:
         if command:
@@ -111,1352 +118,5 @@ class CommandRouter:
             return {}
         return self.help_info
 
-router = CommandRouter()
-def get_help_text():
-    commands = router.get_commands()
-    help_info = router.help_info
-
-    commands.sort()
-    output = "# Available Commands\n\n"
-    for cmd in commands:
-        help_text = help_info.get(cmd, "")
-        output += f"/{cmd} - {help_text}\n\n"
-    
-    arg_help_map = get_argument_help()
-    if arg_help_map:
-        output += "## Common Command-Line Flags\n\n"
-        output += "The shortest unambiguous prefix works (e.g., `-t` for `--temperature`).\n\n"
-        
-
-        output += "```\n"
-
-        all_args_to_show = CANONICAL_ARGS[:]
-        all_args_to_show.sort()
-
-
-        NUM_COLUMNS = 4
-        FLAG_WIDTH = 18   
-        ALIAS_WIDTH = 12  
-        COLUMN_SEPARATOR = " | "
-
-        rows_per_column = (len(all_args_to_show) + NUM_COLUMNS - 1) // NUM_COLUMNS
-        columns = [all_args_to_show[i:i + rows_per_column] for i in range(0, len(all_args_to_show), rows_per_column)]
-
-        def get_shortest_alias(arg):
-            if arg in arg_help_map and arg_help_map[arg]:
-                return min(arg_help_map[arg], key=len)
-            return ""
-
-        header_parts = []
-        for _ in range(NUM_COLUMNS):
-            flag_header = "Flag".ljust(FLAG_WIDTH)
-            alias_header = "Shorthand".ljust(ALIAS_WIDTH)
-            header_parts.append(f"{flag_header}{alias_header}")
-        output += COLUMN_SEPARATOR.join(header_parts) + "\n"
-
-        divider_parts = []
-        for _ in range(NUM_COLUMNS):
-
-            divider_part = "-" * (FLAG_WIDTH + ALIAS_WIDTH)
-            divider_parts.append(divider_part)
-        output += COLUMN_SEPARATOR.join(divider_parts) + "\n"
-
-
-        for i in range(rows_per_column):
-            row_parts = []
-            for col_idx in range(NUM_COLUMNS):
-                if col_idx < len(columns) and i < len(columns[col_idx]):
-                    arg = columns[col_idx][i]
-                    alias = get_shortest_alias(arg)
-                    alias_display = f"(-{alias})" if alias else ""
-                    
-                    flag_part = f"--{arg}".ljust(FLAG_WIDTH)
-                    alias_part = alias_display.ljust(ALIAS_WIDTH)
-                    row_parts.append(f"{flag_part}{alias_part}")
-                else:
-
-                    row_parts.append(" " * (FLAG_WIDTH + ALIAS_WIDTH))
-            
-            output += COLUMN_SEPARATOR.join(row_parts) + "\n"
-
-
-        output += "```\n"
-
-    output += """
-\n## Note
-- Bash commands and programs can be executed directly (try bash first, then LLM).
-- Use '/exit' or '/quit' to exit the current NPC mode or the npcsh shell.
-- Jinxs defined for the current NPC or Team can also be used like commands (e.g., /screenshot).
-"""
-    return output
-def safe_get(kwargs, key, default=None):
-    return kwargs.get(key, default)
-
-
-@router.route("build", "Build deployment artifacts for NPC team")
-def build_handler(command: str, **kwargs):
-    parts = shlex.split(command)
-    
-    target = safe_get(kwargs, 'target', 'flask')
-    output_dir = safe_get(kwargs, 'output', './build')
-    team_path = safe_get(kwargs, 'team', './npc_team')
-    
-    if len(parts) > 1:
-        target = parts[1]
-    
-    build_config = {
-        'team_path': os.path.abspath(team_path),
-        'output_dir': os.path.abspath(output_dir),
-        'target': target,
-        'port': safe_get(kwargs, 'port', 5337),
-        'cors_origins': safe_get(kwargs, 'cors', None),
-    }
-    
-    builders = {
-        'flask': build_flask_server,
-        'docker': build_docker_compose,
-        'cli': build_cli_executable,
-        'static': build_static_site,
-    }
-    
-    if target not in builders:
-        return {
-            "output": f"Unknown target: {target}. Available: {list(builders.keys())}", 
-            "messages": kwargs.get('messages', [])
-        }
-    
-    return builders[target](build_config, **kwargs)
-
-@router.route("breathe", "Condense context on a regular cadence")
-def breathe_handler(command: str, **kwargs):
-  
-    result = breathe(**kwargs)
-    if isinstance(result, dict): 
-        return result
-  
-  
-  
-  
-  
-
-@router.route("compile", "Compile NPC profiles")
-def compile_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])
-    npc_team_dir = safe_get(kwargs, 'current_path', './npc_team')
-    parts = command.split()
-    npc_file_path_arg = parts[1] if len(parts) > 1 else None
-    output = ""
-    try:
-        if npc_file_path_arg:
-            npc_full_path = os.path.abspath(npc_file_path_arg)
-            if os.path.exists(npc_full_path):
-                npc = NPC(npc_full_path)
-                output = f"Compiled NPC: {npc_full_path}"
-            else:
-                output = f"Error: NPC file not found: {npc_full_path}"
-        else:
-            npc = NPC(npc_full_path)
-
-            output = f"Compiled all NPCs in directory: {npc_team_dir}"
-    except NameError:
-        output = "Compile functions (compile_npc_file, compile_team_npcs) not available."
-    except Exception as e:
-        traceback.print_exc()
-        output = f"Error compiling: {e}"
-    return {"output": output, "messages": messages, "npc": npc}
-
-
-
-@router.route("corca", "Enter the Corca MCP-powered agentic shell. Usage: /corca [--mcp-server-path path]")
-def corca_handler(command: str, **kwargs):
-    from npcsh._state import initial_state, setup_shell 
-    command_history, team, default_npc = setup_shell()
-    
-    
-    return enter_corca_mode(command=command, 
-                            command_history = command_history, 
-                            shell_state=initial_state)
-    
-@router.route("flush", "Flush the last N messages")
-def flush_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])
-    try:
-        parts = command.split()
-        n = int(parts[1]) if len(parts) > 1 else 1
-    except (ValueError, IndexError):
-        return {"output": "Usage: /flush [number_of_messages_to_flush]", "messages": messages}
-
-    if n <= 0:
-        return {"output": "Error: Number of messages must be positive.", "messages": messages}
-
-    new_messages = list(messages)
-    original_len = len(new_messages)
-    removed_count = 0
-
-    if new_messages and new_messages[0].get("role") == "system":
-        system_message = new_messages[0]
-        working_messages = new_messages[1:]
-        num_to_remove = min(n, len(working_messages))
-        if num_to_remove > 0:
-            final_messages = [system_message] + working_messages[:-num_to_remove]
-            removed_count = num_to_remove
-        else:
-            final_messages = [system_message]
-    else:
-        num_to_remove = min(n, original_len)
-        if num_to_remove > 0:
-            final_messages = new_messages[:-num_to_remove]
-            removed_count = num_to_remove
-        else:
-            final_messages = []
-
-    output = f"Flushed {removed_count} message(s). Context is now {len(final_messages)} messages."
-    return {"output": output, "messages": final_messages}
-
-@router.route("guac", "Enter guac mode")
-def guac_handler(command,  **kwargs):
-    '''
-    Guac ignores input npc and npc_team dirs and manually sets them to be at ~/.npcsh/guac/
-    
-    '''
-    config_dir = safe_get(kwargs, 'config_dir', None)
-    plots_dir = safe_get(kwargs, 'plots_dir', None)
-    refresh_period = safe_get(kwargs, 'refresh_period', 100)
-    lang = safe_get(kwargs, 'lang', None)
-    messages = safe_get(kwargs, "messages", [])
-    db_conn = safe_get(kwargs, 'db_conn', create_engine('sqlite:///'+os.path.expanduser('~/npcsh_history.db')))
-    
-    npc_file = '~/.npcsh/guac/npc_team/guac.npc'
-    npc_team_dir = os.path.expanduser('~/.npcsh/guac/npc_team/')
-    
-    npc = NPC(file=npc_file, db_conn=db_conn)
-
-    team = Team(npc_team_dir, db_conn=db_conn)
-
-    
-    enter_guac_mode(
-                    npc=npc, 
-                    team=team, 
-                    config_dir=config_dir, 
-                    plots_dir=plots_dir,
-                    npc_team_dir=npc_team_dir,
-                    refresh_period=refresh_period, lang=lang)
-    
-    return {"output": 'Exiting Guac Mode', "messages": safe_get(kwargs, "messages", [])}
-
-
-@router.route("help", "Show help for commands, NPCs, or Jinxs. Usage: /help [topic]")
-def help_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])
-    parts = shlex.split(command)
-    if len(parts) < 2:
-        return {"output": get_help_text(), "messages": messages}
-    target = parts[1].lstrip('/') 
-    output = ""
-
-
-
-    if target in router.get_commands():
-        help_text = router.get_help(target).get(target, "No description available.")
-        output = f"## Help for Command: `/{target}`\n\n- **Description**: {help_text}"
-        return {"output": output, "messages": messages}
-
-    team = safe_get(kwargs, 'team')
-    if team and target in team.npcs:
-        npc_obj = team.npcs[target]
-        output = f"## Help for NPC: `{target}`\n\n"
-        output += f"- **Primary Directive**: {npc_obj.primary_directive}\n"
-        output += f"- **Default Model**: `{npc_obj.model}`\n"
-        output += f"- **Default Provider**: `{npc_obj.provider}`\n"
-        if hasattr(npc_obj, 'jinxs_dict') and npc_obj.jinxs_dict:
-            jinx_names = ", ".join([f"`{j}`" for j in npc_obj.jinxs_dict.keys()])
-            output += f"- **Associated Jinxs**: {jinx_names}\n"
-        return {"output": output, "messages": messages}
-
-  
-    npc = safe_get(kwargs, 'npc')
-    jinx_obj = None
-    source = ""
-    if npc and hasattr(npc, 'jinxs_dict') and target in npc.jinxs_dict:
-        jinx_obj = npc.jinxs_dict[target]
-        source = f" (from NPC: `{npc.name}`)"
-    elif team and hasattr(team, 'jinxs_dict') and target in team.jinxs_dict:
-        jinx_obj = team.jinxs_dict[target]
-        source = f" (from Team: `{team.name}`)"
-
-    if jinx_obj:
-        output = f"## Help for Jinx: `/{target}`{source}\n\n"
-        output += f"- **Description**: {jinx_obj.description}\n"
-        if hasattr(jinx_obj, 'inputs') and jinx_obj.inputs:
-            inputs_str = json.dumps(jinx_obj.inputs, indent=2)
-            output += f"- **Inputs**:\n```json\n{inputs_str}\n```\n"
-        return {"output": output, "messages": messages}
-
-
-    return {"output": f"Sorry, no help topic found for `{target}`.", "messages": messages}
-
-
-
-
-@router.route("init", "Initialize NPC project")
-def init_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])
-    try:
-        parts = shlex.split(command)
-        directory = "."
-        templates = None
-        context = None
-      
-        if len(parts) > 1 and not parts[1].startswith("-"):
-            directory = parts[1]
-      
-
-        initialize_npc_project(
-            directory=directory,
-            templates=templates,
-            context=context,
-            model=safe_get(kwargs, 'model'),
-            provider=safe_get(kwargs, 'provider')
-        )
-        output = f"NPC project initialized in {os.path.abspath(directory)}."
-    except NameError:
-        output = "Init function (initialize_npc_project) not available."
-    except Exception as e:
-        traceback.print_exc()
-        output = f"Error initializing project: {e}"
-    return {"output": output, "messages": messages}
-
-def ensure_repo():
-    """Clone or update the npc-studio repo."""
-    if not NPC_STUDIO_DIR.exists():
-        os.makedirs(NPC_STUDIO_DIR.parent, exist_ok=True)
-        subprocess.check_call([
-            "git", "clone",
-            "https://github.com/npc-worldwide/npc-studio.git",
-            str(NPC_STUDIO_DIR)
-        ])
-    else:
-        subprocess.check_call(
-            ["git", "pull"],
-            cwd=NPC_STUDIO_DIR
-        )
-
-def install_dependencies():
-    """Install npm and pip dependencies."""
-  
-    subprocess.check_call(["npm", "install"], cwd=NPC_STUDIO_DIR)
-
-  
-    req_file = NPC_STUDIO_DIR / "requirements.txt"
-    if req_file.exists():
-        subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", str(req_file)])
-def launch_npc_studio(path_to_open: str = None):
-    """
-    Launch the NPC Studio backend + frontend.
-    Returns PIDs for processes.
-    """
-    ensure_repo()
-    install_dependencies()
-
-  
-    backend = subprocess.Popen(
-        [sys.executable, "npc_studio_serve.py"],
-        cwd=NPC_STUDIO_DIR, 
-        shell = False
-    )
-
-  
-    dev_server = subprocess.Popen(
-        ["npm", "run", "dev"],
-        cwd=NPC_STUDIO_DIR,
-        shell=False
-    )
-    
-  
-    frontend = subprocess.Popen(
-        ["npm", "start"],
-        cwd=NPC_STUDIO_DIR,
-        shell=False
-    )
-
-    return backend, dev_server, frontend
-
-@router.route("npc-studio", "Start npc studio")
-def npc_studio_handler(command: str, **kwargs):
-    messages = kwargs.get("messages", [])
-    user_command = " ".join(command.split()[1:])
-
-    try:
-        backend, electron, frontend = launch_npc_studio(user_command or None)
-        return {
-            "output": f"NPC Studio started!\nBackend PID={backend.pid}, Electron PID={electron.pid} Frontend PID={frontend.pid}",
-            "messages": messages
-        }
-    except Exception as e:
-        return {
-            "output": f"Failed to start NPC Studio: {e}",
-            "messages": messages
-        }
-@router.route("ots", "Take screenshot and analyze with vision model")
-def ots_handler(command: str, **kwargs):
-    command_parts = command.split()
-    image_paths = []
-    npc = safe_get(kwargs, 'npc')
-    vision_model = safe_get(kwargs, 
-                            'vmodel',
-                            NPCSH_VISION_MODEL)
-    vision_provider = safe_get(kwargs, 
-                               'vprovider', 
-                               NPCSH_VISION_PROVIDER)
-    messages = safe_get(kwargs, 
-                        'messages', 
-                        [])
-    stream = safe_get(kwargs, 
-                      'stream',
-                        NPCSH_STREAM_OUTPUT)
-
-    try:
-        if len(command_parts) > 1:
-            for img_path_arg in command_parts[1:]:
-                full_path = os.path.abspath(img_path_arg)
-                if os.path.exists(full_path):
-                    image_paths.append(full_path)
-                else:
-                    return {"output": f"Error: Image file not found at {full_path}", "messages": messages}
-        else:
-            screenshot_info = capture_screenshot(full=False)
-            if screenshot_info and "file_path" in screenshot_info:
-                image_paths.append(screenshot_info["file_path"])
-                print(f"Screenshot captured: {screenshot_info.get('filename', os.path.basename(screenshot_info['file_path']))}")
-            else:
-                 return {"output": "Error: Failed to capture screenshot.", "messages": messages}
-
-        if not image_paths:
-            return {"output": "No valid images found or captured.", "messages": messages}
-
-        user_prompt = safe_get(kwargs, 'stdin_input')
-        if user_prompt is None:
-            try:
-                user_prompt = input(
-                    "Enter a prompt for the LLM about these images (or press Enter to skip): "
-                )
-            except EOFError:
-                 user_prompt = "Describe the image(s)."
-
-        if not user_prompt or not user_prompt.strip():
-            user_prompt = "Describe the image(s)."
-
-        response_data = get_llm_response(
-            prompt=user_prompt,
-            model=vision_model,
-            provider=vision_provider,
-            messages=messages,
-            images=image_paths,
-            stream=stream,
-            npc=npc,
-            api_url=safe_get(kwargs, 'api_url'),
-            api_key=safe_get(kwargs, 'api_key')
-        )
-        return {"output": response_data.get('response'), "messages": response_data.get('messages'), "model": vision_model, "provider": vision_provider}
-
-    except Exception as e:
-        traceback.print_exc()
-        return {"output": f"Error during /ots command: {e}", "messages": messages}
-
-
-
-
-@router.route("plan", "Execute a plan command")
-def plan_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])
-    user_command = " ".join(command.split()[1:])
-    if not user_command:
-        return {"output": "Usage: /plan <description_of_plan>", "messages": messages}
-  
-    return execute_plan_command(command=user_command, **kwargs)
-
-  
-  
-  
-  
-
-@router.route("pti", "Enter Pardon-The-Interruption mode for human-in-the-loop reasoning.")
-def pti_handler(command: str, **kwargs):
-    return enter_pti_mode(command=command, **kwargs)
-
-@router.route("plonk", "Use vision model to interact with GUI. Usage: /plonk <task description>")
-def plonk_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])
-    
-  
-  
-    positional_args = safe_get(kwargs, 'positional_args', [])
-    request_str = " ".join(positional_args)
-
-    if not request_str:
-        return {"output": "Usage: /plonk <task_description> [--vmodel model_name] [--vprovider provider_name]", "messages": messages}
-
-    try:
-        plonk_context = safe_get(kwargs, 'plonk_context')
-        
-      
-        summary_data = execute_plonk_command(
-            request=request_str,
-            model=safe_get(kwargs, 'vmodel', NPCSH_VISION_MODEL),
-            provider=safe_get(kwargs, 'vprovider', NPCSH_VISION_PROVIDER),
-            npc=safe_get(kwargs, 'npc'),
-            plonk_context=plonk_context,
-            debug=True 
-        )        
-        
-        if summary_data and isinstance(summary_data, list):
-            output_report = format_plonk_summary(summary_data)
-            return {"output": output_report, "messages": messages}
-        else:
-            return {"output": "Plonk command did not complete within the maximum number of iterations.", "messages": messages}
-
-    except Exception as e:
-        traceback.print_exc()
-        return {"output": f"Error executing plonk command: {e}", "messages": messages}
-
-
-@router.route("brainblast", "Execute an advanced chunked search on command history")
-def brainblast_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])    
-    parts = shlex.split(command)
-    search_query = " ".join(parts[1:]) if len(parts) > 1 else ""
-    
-    if not search_query:
-        return {"output": "Usage: /brainblast <search_terms>", "messages": messages}
-    
-  
-    command_history = kwargs.get('command_history')
-    if not command_history:
-      
-      
-        db_path = safe_get(kwargs, "history_db_path", os.path.expanduser('~/npcsh_history.db'))
-        try:
-            command_history = CommandHistory(db_path)
-            kwargs['command_history'] = command_history
-        except Exception as e:
-            return {"output": f"Error connecting to command history: {e}", "messages": messages}
-    
-    try:
-      
-        if 'messages' in kwargs:
-            del kwargs['messages']
-            
-      
-        return execute_brainblast_command(
-                                    command=search_query,
-                                    **kwargs)   
-    except Exception as e:
-        traceback.print_exc()
-        return {"output": f"Error executing brainblast command: {e}", "messages": messages}
-
-@router.route("rag", "Execute a RAG command using ChromaDB embeddings with optional file input (-f/--file)")
-def rag_handler(command: str, **kwargs):    
-    parts = shlex.split(command)
-    user_command = []
-    file_paths = []
-    
-    i = 1
-    while i < len(parts):
-        if parts[i] == "-f" or parts[i] == "--file":
-          
-            if i + 1 < len(parts):
-                file_paths.append(parts[i + 1])
-                i += 2
-            else:
-                return {"output": "Error: -f/--file flag needs a file path", "messages": messages}
-        else:
-          
-            user_command.append(parts[i])
-            i += 1
-    
-    user_command = " ".join(user_command)
-    
-    vector_db_path = safe_get(kwargs, "vector_db_path", os.path.expanduser('~/npcsh_chroma.db'))
-    embedding_model = safe_get(kwargs, "emodel", NPCSH_EMBEDDING_MODEL)
-    embedding_provider = safe_get(kwargs, "eprovider", NPCSH_EMBEDDING_PROVIDER)
-    
-    if not user_command and not file_paths:
-        return {"output": "Usage: /rag [-f file_path] <query>", "messages": kwargs.get('messages', [])}
-    
-    try:
-      
-        file_contents = []
-        for file_path in file_paths:
-            try:
-                chunks = load_file_contents(file_path)
-                file_name = os.path.basename(file_path)
-                file_contents.extend([f"[{file_name}] {chunk}" for chunk in chunks])
-            except Exception as file_err:
-                file_contents.append(f"Error processing file {file_path}: {str(file_err)}")
-        exe_rag =  execute_rag_command(
-            command=user_command,
-            vector_db_path=vector_db_path,
-            embedding_model=embedding_model,
-            embedding_provider=embedding_provider,
-            file_contents=file_contents if file_paths else None,
-            **kwargs
-        )
-        return {'output':exe_rag.get('response'), 'messages': exe_rag.get('messages', kwargs.get('messages', []))}
-    
-    except Exception as e:
-        traceback.print_exc()
-        return {"output": f"Error executing RAG command: {e}", "messages": kwargs.get('messages', [])}
-@router.route("roll", "generate a video")
-def roll_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])
-    prompt = " ".join(command.split()[1:])
-    num_frames = safe_get(kwargs, 'num_frames', 125)
-    width = safe_get(kwargs, 'width', 256)
-    height = safe_get(kwargs, 'height', 256)
-    output_path = safe_get(kwargs, 'output_path', "output.mp4")    
-    if not prompt:
-        return {"output": "Usage: /roll <your prompt>", "messages": messages}
-    try:
-        result = gen_video(
-            prompt=prompt,
-            model=safe_get(kwargs, 'vgmodel', NPCSH_VIDEO_GEN_MODEL),
-            provider=safe_get(kwargs, 'vgprovider', NPCSH_VIDEO_GEN_PROVIDER),
-            npc=safe_get(kwargs, 'npc'),
-            num_frames = num_frames,
-            width = width,
-            height = height,
-            output_path=output_path,
-            
-            **safe_get(kwargs, 'api_kwargs', {})
-        )
-        return result
-    except Exception as e:
-        traceback.print_exc()
-        return {"output": f"Error generating video: {e}", "messages": messages}
-    
-
-@router.route("sample", "Send a prompt directly to the LLM")
-def sample_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])
-    
-
-    positional_args = safe_get(kwargs, 'positional_args', [])
-    prompt = " ".join(positional_args)
 
-    if not prompt:
-        return {"output": "Usage: /sample <your prompt> [-m --model] model  [-p --provider] provider", 
-                "messages": messages}
-
-    try:
-        result = get_llm_response(
-            prompt=prompt,
-            **kwargs
-        )
-        if result and isinstance(result, dict):
-            return {
-                "output": result.get('response'), 
-                "messages": result.get('messages', messages), 
-                "model": kwargs.get('model'), 
-                "provider":kwargs.get('provider'), 
-                "npc":kwargs.get("npc"),
-            }
-        else:
-          
-            return {"output": str(result), "messages": messages}
-
-    except Exception as e:
-        traceback.print_exc()
-        return {"output": f"Error sampling LLM: {e}", "messages": messages}
-
-
-
-@router.route("search", "Execute web search or memory/KG search")
-def search_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])
-    
-    positional_args = safe_get(kwargs, 'positional_args', [])
-    
-    search_type = None
-    query_parts = []
-    
-    i = 0
-    while i < len(positional_args):
-        arg = positional_args[i]
-        if arg in ['-m', '-mem', '--memory']:
-            search_type = 'memory'
-            i += 1
-        elif arg in ['-kg', '--knowledge-graph']:
-            search_type = 'kg'
-            i += 1
-        else:
-            query_parts.append(arg)
-            i += 1
-    
-    query = " ".join(query_parts)
-    
-    if not query:
-        return {
-            "output": (
-                "Usage:\n"
-                "  /search <query>              - Web search\n"
-                "  /search -m <query>           - Memory search\n"
-                "  /search -kg <query>          - Knowledge graph search"
-            ), 
-            "messages": messages
-        }
-    
-    if search_type == 'memory':
-        return search_memories(query, kwargs, messages)
-    elif search_type == 'kg':
-        return search_knowledge_graph(query, kwargs, messages)
-    else:
-        return search_web_default(query, kwargs, messages)
-
-def search_memories(query: str, kwargs: dict, messages: list):
-    command_history = kwargs.get('command_history')
-    
-    if not command_history:
-        db_path = safe_get(
-            kwargs, 
-            "history_db_path", 
-            os.path.expanduser('~/npcsh_history.db')
-        )
-        try:
-            command_history = CommandHistory(db_path)
-        except Exception as e:
-            return {
-                "output": f"Error connecting to history: {e}", 
-                "messages": messages
-            }
-    
-    state = kwargs.get('state')
-    npc = safe_get(kwargs, 'npc')
-    team = safe_get(kwargs, 'team')
-    
-    npc_name = npc.name if isinstance(npc, NPC) else "__none__"
-    team_name = team.name if team else "__none__"
-    current_path = safe_get(kwargs, 'current_path', os.getcwd())
-    
-    try:
-        memories = get_relevant_memories(
-            command_history=command_history,
-            npc_name=npc_name,
-            team_name=team_name,
-            path=current_path,
-            query=query,
-            max_memories=10,
-            state=state
-        )
-        
-        if not memories:
-            output = f"No memories found for query: '{query}'"
-        else:
-            output = f"Found {len(memories)} memories:\n\n"
-            for i, mem in enumerate(memories, 1):
-                final_mem = (
-                    mem.get('final_memory') or 
-                    mem.get('initial_memory')
-                )
-                timestamp = mem.get('timestamp', 'unknown')
-                output += f"{i}. [{timestamp}] {final_mem}\n"
-        
-        return {"output": output, "messages": messages}
-        
-    except Exception as e:
-        import traceback
-        traceback.print_exc()
-        return {
-            "output": f"Error searching memories: {e}", 
-            "messages": messages
-        }
-
-def search_knowledge_graph(query: str, kwargs: dict, messages: list):
-    command_history = kwargs.get('command_history')
-    
-    if not command_history:
-        db_path = safe_get(
-            kwargs, 
-            "history_db_path", 
-            os.path.expanduser('~/npcsh_history.db')
-        )
-        try:
-            command_history = CommandHistory(db_path)
-        except Exception as e:
-            return {
-                "output": f"Error connecting to history: {e}", 
-                "messages": messages
-            }
-    
-    npc = safe_get(kwargs, 'npc')
-    team = safe_get(kwargs, 'team')
-    
-    npc_name = npc.name if isinstance(npc, NPC) else "__none__"
-    team_name = team.name if team else "__none__"
-    current_path = safe_get(kwargs, 'current_path', os.getcwd())
-    
-    try:
-        engine = command_history.engine
-        kg = load_kg_from_db(
-            engine, 
-            team_name, 
-            npc_name, 
-            current_path
-        )
-        
-        if not kg or not kg.get('facts'):
-            return {
-                "output": (
-                    f"No knowledge graph found for current scope.\n"
-                    f"Scope: Team='{team_name}', "
-                    f"NPC='{npc_name}', Path='{current_path}'"
-                ),
-                "messages": messages
-            }
-        
-        query_lower = query.lower()
-        matching_facts = []
-        matching_concepts = []
-        
-        for fact in kg.get('facts', []):
-            statement = fact.get('statement', '').lower()
-            if query_lower in statement:
-                matching_facts.append(fact)
-        
-        for concept in kg.get('concepts', []):
-            name = concept.get('name', '').lower()
-            desc = concept.get('description', '').lower()
-            if query_lower in name or query_lower in desc:
-                matching_concepts.append(concept)
-        
-        output = f"Knowledge Graph Search Results for '{query}':\n\n"
-        
-        if matching_facts:
-            output += f"## Facts ({len(matching_facts)}):\n"
-            for i, fact in enumerate(matching_facts, 1):
-                output += f"{i}. {fact.get('statement')}\n"
-            output += "\n"
-        
-        if matching_concepts:
-            output += f"## Concepts ({len(matching_concepts)}):\n"
-            for i, concept in enumerate(matching_concepts, 1):
-                name = concept.get('name')
-                desc = concept.get('description', '')
-                output += f"{i}. {name}: {desc}\n"
-        
-        if not matching_facts and not matching_concepts:
-            output += "No matching facts or concepts found."
-        
-        return {"output": output, "messages": messages}
-        
-    except Exception as e:
-        import traceback
-        traceback.print_exc()
-        return {
-            "output": f"Error searching KG: {e}", 
-            "messages": messages
-        }
-
-def search_web_default(query: str, kwargs: dict, messages: list):
-    search_provider = safe_get(kwargs, 'sprovider', NPCSH_SEARCH_PROVIDER)
-    render_markdown(f'- Searching {search_provider} for "{query}"')
-    
-    try:
-        search_results = search_web(query, provider=search_provider)
-        output = (
-            "\n".join([f"- {res}" for res in search_results]) 
-            if search_results 
-            else "No results found."
-        )
-    except Exception as e:
-        import traceback
-        traceback.print_exc()
-        output = f"Error during web search: {e}"
-    
-    return {"output": output, "messages": messages}
-
-
-
-@router.route("serve", "Serve an NPC Team")
-def serve_handler(command: str, **kwargs):
-  
-  
-
-    port   = safe_get(kwargs, "port", 5337)
-  
-    messages = safe_get(kwargs, "messages", [])
-    cors = safe_get(kwargs, "cors", None)
-    if cors:
-        cors_origins = [origin.strip() for origin in cors.split(",")]
-    else:
-        cors_origins = None
-
-        start_flask_server(
-            port=port, 
-            cors_origins=cors_origins,
-        )
-
-
-    return {"output": None, "messages": messages}
-
-@router.route("set", "Set configuration values")
-def set_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])
-    parts = command.split(maxsplit=1)
-    if len(parts) < 2 or '=' not in parts[1]:
-        return {"output": "Usage: /set <key>=<value>", "messages": messages}
-
-    key_value = parts[1]
-    key, value = key_value.split('=', 1)
-    key = key.strip()
-    value = value.strip().strip('"\'')
-
-    try:
-        set_npcsh_config_value(key, value)
-        output = f"Configuration value '{key}' set."
-    except NameError:
-        output = "Set function (set_npcsh_config_value) not available."
-    except Exception as e:
-        traceback.print_exc()
-        output = f"Error setting configuration '{key}': {e}"
-    return {"output": output, "messages": messages}
-
-@router.route("sleep", "Evolve knowledge graph. Use --dream to also run creative synthesis.")
-def sleep_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])
-    npc = safe_get(kwargs, 'npc')
-    team = safe_get(kwargs, 'team')
-    model = safe_get(kwargs, 'model')
-    provider = safe_get(kwargs, 'provider')
-
-    is_dreaming = safe_get(kwargs, 'dream', False)
-    operations_str = safe_get(kwargs, 'ops')
-    
-    operations_config = None
-    if operations_str and isinstance(operations_str, str):
-        operations_config = [op.strip() for op in operations_str.split(',')]
-
-  
-    team_name = team.name if team else "__none__"
-    npc_name = npc.name if isinstance(npc, NPC) else "__none__"
-    current_path = os.getcwd()
-    scope_str = f"Team: '{team_name}', NPC: '{npc_name}', Path: '{current_path}'"
-
-  
-    render_markdown(f"- Checking knowledge graph for scope: {scope_str}")
-
-    try:
-        db_path = os.getenv("NPCSH_DB_PATH", os.path.expanduser("~/npcsh_history.db"))
-        command_history = CommandHistory(db_path)
-        engine = command_history.engine
-    except Exception as e:
-        return {"output": f"Error connecting to history database for KG access: {e}", "messages": messages}
-
-    try:
-        current_kg = load_kg_from_db(engine, team_name, npc_name, current_path)
-
-      
-        if not current_kg or not current_kg.get('facts'):
-            output_msg = f"Knowledge graph for the current scope is empty. Nothing to process.\n"
-            output_msg += f"  - Scope Checked: {scope_str}\n\n"
-            output_msg += "**Hint:** Have a conversation or run some commands first to build up knowledge in this specific context. The KG is unique to each combination of Team, NPC, and directory."
-            return {"output": output_msg, "messages": messages}
-
-      
-        original_facts = len(current_kg.get('facts', []))
-        original_concepts = len(current_kg.get('concepts', []))
-        
-      
-
-      
-        process_type = "Sleep"
-        ops_display = f"with operations: {operations_config}" if operations_config else "with random operations"
-        render_markdown(f"- Initiating sleep process {ops_display}")
-        
-        evolved_kg, _ = kg_sleep_process(
-            existing_kg=current_kg,
-            model=model,
-            provider=provider,
-            npc=npc,
-            operations_config=operations_config
-        )
-
-      
-        if is_dreaming:
-            process_type += " & Dream"
-            render_markdown(f"- Initiating dream process on the evolved KG...")
-            evolved_kg, _ = kg_dream_process(
-                existing_kg=evolved_kg,
-                model=model,
-                provider=provider,
-                npc=npc
-            )
-
-      
-        save_kg_to_db(conn, evolved_kg, team_name, npc_name, current_path)
-
-      
-        new_facts = len(evolved_kg.get('facts', []))
-        new_concepts = len(evolved_kg.get('concepts', []))
-
-        output = f"{process_type} process complete.\n"
-        output += f"- Facts: {original_facts} -> {new_facts} ({new_facts - original_facts:+})\n"
-        output += f"- Concepts: {original_concepts} -> {new_concepts} ({new_concepts - original_concepts:+})"
-        
-        print(evolved_kg.get('facts'))
-        print(evolved_kg.get('concepts'))
-        
-        return {"output": output, "messages": messages}
-
-    except Exception as e:
-        import traceback
-        traceback.print_exc()
-        return {"output": f"Error during KG evolution process: {e}", "messages": messages}
-    finally:
-        if 'command_history' in locals() and command_history:
-            command_history.close()
-
-
-
-
-@router.route("spool", "Enter interactive chat (spool) mode")
-def spool_handler(command: str, **kwargs):
-    try:
-        npc = safe_get(kwargs, 'npc')
-        team = safe_get(kwargs, 'team')
-        
-        if isinstance(npc, str) and team:
-            npc_name = npc
-            if npc_name in team.npcs:
-                npc = team.npcs[npc_name]
-            else:
-                return {"output": f"Error: NPC '{npc_name}' not found in team. Available NPCs: {', '.join(team.npcs.keys())}", "messages": safe_get(kwargs, "messages", [])}
-        kwargs['npc'] = npc
-        return enter_spool_mode(            
-                                **kwargs)
-    except Exception as e:
-        traceback.print_exc()
-        return {"output": f"Error entering spool mode: {e}", "messages": safe_get(kwargs, "messages", [])}
-    
-@router.route("jinxs", "Show available jinxs for the current NPC/Team")
-def jinxs_handler(command: str, **kwargs):
-    npc = safe_get(kwargs, 'npc')
-    team = safe_get(kwargs, 'team')
-    output = "Available Jinxs:\n"
-    jinxs_listed = set()
-
-    def format_jinx(name, jinx_obj):
-        desc = getattr(jinx_obj, 'description', 'No description available.')
-        return f"- /{name}: {desc}\n"
-
-    if npc and isinstance(npc, NPC) and hasattr(npc, 'jinxs_dict') and npc.jinxs_dict:
-        output += f"\n--- Jinxs for NPC: {npc.name} ---\n"
-        for name, jinx in sorted(npc.jinxs_dict.items()):
-            output += format_jinx(name, jinx)
-            jinxs_listed.add(name)
-
-    if team and hasattr(team, 'jinxs_dict') and team.jinxs_dict:
-         team_has_jinxs = False
-         team_output = ""
-         for name, jinx in sorted(team.jinxs_dict.items()):
-             if name not in jinxs_listed:
-                 team_output += format_jinx(name, jinx)
-                 team_has_jinxs = True
-         if team_has_jinxs:
-             output += f"\n--- Jinxs for Team: {getattr(team, 'name', 'Unnamed Team')} ---\n"
-             output += team_output
-
-    if not jinxs_listed and not (team and hasattr(team, 'jinxs_dict') and team.jinxs_dict):
-        output = "No jinxs available for the current context."
-
-    return {"output": output.strip(), "messages": safe_get(kwargs, "messages", [])}
-
-@router.route("trigger", "Execute a trigger command")
-def trigger_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])
-    user_command = " ".join(command.split()[1:])
-    if not user_command:
-        return {"output": "Usage: /trigger <trigger_description>", "messages": messages}
-    try:
-        return execute_trigger_command(command=user_command, **kwargs)
-    except NameError:
-        return {"output": "Trigger function (execute_trigger_command) not available.", "messages": messages}
-    except Exception as e:
-        traceback.print_exc()
-        return {"output": f"Error executing trigger: {e}", "messages": messages}
-@router.route("vixynt", "Generate images from text descriptions")
-def vixynt_handler(command: str, **kwargs):
-    npc = safe_get(kwargs, 'npc')
-    model = safe_get(kwargs, 'igmodel', NPCSH_IMAGE_GEN_MODEL)
-    provider = safe_get(kwargs, 'igprovider', NPCSH_IMAGE_GEN_PROVIDER)
-    height = safe_get(kwargs, 'height', 1024)
-    width = safe_get(kwargs, 'width', 1024)
-    output_file_base = safe_get(kwargs, 'output_file')
-    attachments = safe_get(kwargs, 'attachments')
-    n_images = safe_get(kwargs, 'n_images', 1) 
-    if isinstance(attachments, str):
-        attachments = attachments.split(',')
-    
-    messages = safe_get(kwargs, 'messages', [])
-
-    user_prompt = " ".join(safe_get(kwargs, 'positional_args', []))
-
-    if not user_prompt:
-        return {"output": "Usage: /vixynt <prompt> [--output_file path] [--attachments path] [--n_images num]", "messages": messages}
-    
-    try:
-      
-        images_list = gen_image(
-            prompt=user_prompt,
-            model=model,
-            provider=provider,
-            npc=npc,
-            height=height,
-            width=width,
-            n_images=n_images, 
-            input_images=attachments
-        )
-
-        saved_files = []
-        if not isinstance(images_list, list):
-            images_list = [images_list] if images_list is not None else []
-
-        for i, image in enumerate(images_list):
-            if image is None:
-                continue
-
-            if output_file_base is None:
-                os.makedirs(os.path.expanduser("~/.npcsh/images/"), exist_ok=True)
-                current_output_file = (
-                    os.path.expanduser("~/.npcsh/images/")
-                    + f"image_{datetime.now().strftime('%Y%m%d_%H%M%S')}_{i}.png"
-                )
-            else:
-                base_name, ext = os.path.splitext(os.path.expanduser(output_file_base))
-                current_output_file = f"{base_name}_{i}{ext}"
-            
-            image.save(current_output_file)
-            image.show()
-            saved_files.append(current_output_file)
-
-        if saved_files:
-            if attachments:
-                output = f"Image(s) edited and saved to: {', '.join(saved_files)}"
-            else:
-                output = f"Image(s) generated and saved to: {', '.join(saved_files)}"
-        else:
-            output = f"No images {'edited' if attachments else 'generated'}."
-
-    except Exception as e:
-        traceback.print_exc()
-        output = f"Error {'editing' if attachments else 'generating'} image: {e}"
-
-    return {
-        "output": output,
-        "messages": messages,
-        "model": model,
-        "provider": provider
-    }
-
-
-@router.route("wander", "Enter wander mode (experimental)")
-def wander_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])
-    
-  
-    try:
-        parts = shlex.split(command)
-        problem_parts = []
-        wander_params = {}
-        
-        i = 1
-        while i < len(parts):
-            part = parts[i]
-            
-            if '=' in part:
-              
-                key, initial_value = part.split('=', 1)
-                
-              
-                value_parts = [initial_value]
-                j = i + 1
-                while j < len(parts) and '=' not in parts[j]:
-                    value_parts.append(parts[j])
-                    j += 1
-                
-              
-                wander_params[key] = " ".join(value_parts)
-              
-                i = j
-            else:
-              
-                problem_parts.append(part)
-                i += 1
-        
-        problem = " ".join(problem_parts)
-    except Exception as e:
-        return {"output": f"Error parsing arguments: {e}", "messages": messages}
-        
-    if not problem:
-        return {"output": "Usage: /wander <problem> [key=value...]", "messages": messages}
-
-    try:
-      
-        mode_args = {
-            'problem': problem,
-            'npc': safe_get(kwargs, 'npc'),
-            'model': safe_get(kwargs, 'model'),
-            'provider': safe_get(kwargs, 'provider'),
-          
-            'environment': wander_params.get('environment'),
-            'low_temp': float(wander_params.get('low-temp', 0.5)),
-            'high_temp': float(wander_params.get('high-temp', 1.9)),
-            'interruption_likelihood': float(wander_params.get('interruption-likelihood', 1)),
-            'sample_rate': float(wander_params.get('sample-rate', 0.4)),
-            'n_high_temp_streams': int(wander_params.get('n-high-temp-streams', 5)),
-            'include_events': bool(wander_params.get('include-events', False)),
-            'num_events': int(wander_params.get('num-events', 3))
-        }
-        
-        result = enter_wander_mode(**mode_args)
-        
-        if isinstance(result, list) and result:
-            output = result[-1].get("insight", "Wander mode session complete.")
-        else:
-            output = str(result) if result else "Wander mode session complete."
-            
-        messages.append({"role": "assistant", "content": output})
-        return {"output": output, "messages": messages}
-        
-    except Exception as e:
-        traceback.print_exc()
-        return {"output": f"Error during wander mode: {e}", "messages": messages}
-
-@router.route("yap", "Enter voice chat (yap) mode")
-def yap_handler(command: str, **kwargs):
-    try:
-        return enter_yap_mode(
-            ** kwargs
-            )
-    except Exception as e:
-        traceback.print_exc()
-        return {"output": f"Error entering yap mode: {e}", "messages": safe_get(kwargs, "messages", [])}
-
-@router.route("alicanto", "Conduct deep research with multiple perspectives, identifying gold insights and cliff warnings")
-def alicanto_handler(command: str, **kwargs):
-    messages = safe_get(kwargs, "messages", [])
-    parts = shlex.split(command)
-    skip_research = safe_get(kwargs, "skip_research", True)
-    query = ""
-    num_npcs = safe_get(kwargs, 'num_npcs', 5)
-    depth = safe_get(kwargs, 'depth', 3)
-  
-    i = 1
-    while i < len(parts):
-        if parts[i].startswith('--'):
-            option = parts[i][2:]
-            if option in ['num-npcs', 'npcs']:
-                if i + 1 < len(parts) and parts[i + 1].isdigit():
-                    num_npcs = int(parts[i + 1])
-                    i += 2
-                else:
-                    i += 1
-            elif option in ['depth', 'd']:
-                if i + 1 < len(parts) and parts[i + 1].isdigit():
-                    depth = int(parts[i + 1])
-                    i += 2
-                else:
-                    i += 1
-            elif option in ['exploration', 'e']:
-                if i + 1 < len(parts) and parts[i + 1].replace('.', '', 1).isdigit():
-                    exploration_factor = float(parts[i + 1])
-                    i += 2
-                else:
-                    i += 1
-            elif option in ['creativity', 'c']:
-                if i + 1 < len(parts) and parts[i + 1].replace('.', '', 1).isdigit():
-                    creativity_factor = float(parts[i + 1])
-                    i += 2
-                else:
-                    i += 1
-            elif option in ['format', 'f']:
-                if i + 1 < len(parts):
-                    output_format = parts[i + 1]
-                    i += 2
-                else:
-                    i += 1
-            else:
-              
-                i += 1
-        else:
-          
-            query += parts[i] + " "
-            i += 1
-    
-    query = query.strip()
-    
-  
-    if 'num_npcs' in kwargs:
-        try:
-            num_npcs = int(kwargs['num_npcs'])
-        except ValueError:
-            return {"output": "Error: num_npcs must be an integer", "messages": messages}
-    
-    if 'depth' in kwargs:
-        try:
-            depth = int(kwargs['depth'])
-        except ValueError:
-            return {"output": "Error: depth must be an integer", "messages": messages}
-    
-    if 'exploration' in kwargs:
-        try:
-            exploration_factor = float(kwargs['exploration'])
-        except ValueError:
-            return {"output": "Error: exploration must be a float", "messages": messages}
-            
-    if 'creativity' in kwargs:
-        try:
-            creativity_factor = float(kwargs['creativity'])
-        except ValueError:
-            return {"output": "Error: creativity must be a float", "messages": messages}
-    
-    if not query:
-        return {"output": "Usage: /alicanto <research query> [--num-npcs N] [--depth N] [--exploration 0.3] [--creativity 0.5] [--format report|summary|full]", "messages": messages}
-    
-    try:
-        logging.info(f"Starting Alicanto research on: {query}")
-        model = safe_get(kwargs, 'model')
-        if len(model) == 0 :
-            model = NPCSH_CHAT_MODEL
-        provider = safe_get(kwargs, 'provider')
-        if len(provider) == 0 :
-            provider = NPCSH_CHAT_PROVIDER
-
-
-        print('model: ', model)
-        print('provider: ', provider)
-
-        result = alicanto(
-            query,
-            num_npcs=num_npcs,
-            depth=depth,
-            model=model,
-            provider=provider, 
-            max_steps = safe_get(kwargs, 'max_steps', 20),
-            skip_research = skip_research
-
-        )
-        
-      
-        if isinstance(result, dict):
-            if "integration" in result:
-                output = result["integration"]
-            else:
-                output = "Alicanto research completed. Full results available in returned data."
-        else:
-            output = result
-            
-        return {"output": output, "messages": messages, "alicanto_result": result}
-    except Exception as e:
-        traceback.print_exc()
-        logging.error(f"Error during Alicanto research: {e}")
-        return {"output": f"Error during Alicanto research: {e}", "messages": messages}
+router = CommandRouter()