npcsh 1.1.17__py3-none-any.whl → 1.1.18__py3-none-any.whl

npcsh/npc_team/jinxs/modes/corca.jinx

@@ -0,0 +1,430 @@
+jinx_name: corca
+description: MCP-powered agentic shell - LLM with tool use via MCP servers
+inputs:
+  - mcp_server_path: null
+  - initial_command: null
+  - model: null
+  - provider: null
+
+steps:
+  - name: corca_repl
+    engine: python
+    code: |
+      import os
+      import sys
+      import tty
+      import termios
+      import asyncio
+      import json
+      from contextlib import AsyncExitStack
+      from termcolor import colored
+
+      from npcpy.llm_funcs import get_llm_response
+      from npcpy.npc_sysenv import render_markdown, get_system_message
+
+      # MCP imports
+      try:
+          from mcp import ClientSession, StdioServerParameters
+          from mcp.client.stdio import stdio_client
+          MCP_AVAILABLE = True
+      except ImportError:
+          MCP_AVAILABLE = False
+          print(colored("MCP not available. Install with: pip install mcp-client", "yellow"))
+
+      npc = context.get('npc')
+      team = context.get('team')
+      messages = context.get('messages', [])
+      mcp_server_path = context.get('mcp_server_path')
+      initial_command = context.get('initial_command')
+
+      # Resolve npc if it's a string (npc name) rather than NPC object
+      if isinstance(npc, str) and team:
+          npc = team.get(npc) if hasattr(team, 'get') else None
+      elif isinstance(npc, str):
+          npc = None
+
+      model = context.get('model') or (npc.model if npc and hasattr(npc, 'model') else None)
+      provider = context.get('provider') or (npc.provider if npc and hasattr(npc, 'provider') else None)
+
+      # Use shared_context for MCP state
+      shared_ctx = npc.shared_context if npc and hasattr(npc, 'shared_context') else {}
+
+      # ========== TUI Helper Functions ==========
+      def get_terminal_size():
+          try:
+              size = os.get_terminal_size()
+              return size.columns, size.lines
+          except:
+              return 80, 24
+
+      def tools_tui_browser(tools_llm):
+          """Interactive TUI browser for MCP tools"""
+          if not tools_llm:
+              print(colored("No MCP tools connected.", "yellow"))
+              return None
+
+          # Build tool info list
+          tools = []
+          for t in tools_llm:
+              func = t.get('function', {})
+              tools.append({
+                  'name': func.get('name', 'unknown'),
+                  'description': func.get('description', '')[:100],
+                  'params': func.get('parameters', {})
+              })
+
+          width, height = get_terminal_size()
+          selected = 0
+          scroll = 0
+          list_height = height - 5
+          mode = 'list'
+          preview_scroll = 0
+          preview_lines = []
+
+          fd = sys.stdin.fileno()
+          old_settings = termios.tcgetattr(fd)
+
+          try:
+              tty.setcbreak(fd)
+              sys.stdout.write('\033[?25l')
+              sys.stdout.write('\033[2J\033[H')
+
+              while True:
+                  width, height = get_terminal_size()
+                  list_height = height - 5
+
+                  if mode == 'list':
+                      if selected < scroll:
+                          scroll = selected
+                      elif selected >= scroll + list_height:
+                          scroll = selected - list_height + 1
+
+                  sys.stdout.write('\033[H')
+
+                  # Header
+                  if mode == 'list':
+                      header = f" CORCA MCP TOOLS ({len(tools)} available) "
+                  else:
+                      header = f" TOOL: {tools[selected]['name']} "
+                  sys.stdout.write(f'\033[46;30;1m{header.ljust(width)}\033[0m\n')
+
+                  if mode == 'list':
+                      col_header = f' {"NAME":<25} {"DESCRIPTION":<50}'
+                      sys.stdout.write(f'\033[90m{col_header[:width]}\033[0m\n')
+                  else:
+                      sys.stdout.write(f'\033[90m{"─" * width}\033[0m\n')
+
+                  if mode == 'list':
+                      for i in range(list_height):
+                          idx = scroll + i
+                          sys.stdout.write(f'\033[{3+i};1H\033[K')
+                          if idx >= len(tools):
+                              continue
+
+                          t = tools[idx]
+                          name = t['name'][:25]
+                          desc = t['description'][:50]
+
+                          line = f" {name:<25} {desc}"
+                          line = line[:width-1]
+
+                          if idx == selected:
+                              sys.stdout.write(f'\033[47;30;1m>{line}\033[0m')
+                          else:
+                              sys.stdout.write(f' {line}')
+
+                      # Status bar
+                      sys.stdout.write(f'\033[{height-2};1H\033[K\033[90m{"─" * width}\033[0m')
+                      t = tools[selected] if tools else {}
+                      params = t.get('params', {}).get('properties', {})
+                      param_names = list(params.keys())[:5]
+                      sys.stdout.write(f'\033[{height-1};1H\033[K Params: {", ".join(param_names) if param_names else "none"}'.ljust(width)[:width])
+                      sys.stdout.write(f'\033[{height};1H\033[K\033[46;30m j/k:Nav p:Details Enter:Copy q:Quit [{selected+1}/{len(tools)}] \033[0m')
+
+                  else:  # preview mode
+                      for i in range(list_height):
+                          idx = preview_scroll + i
+                          sys.stdout.write(f'\033[{3+i};1H\033[K')
+                          if idx < len(preview_lines):
+                              sys.stdout.write(preview_lines[idx][:width-1])
+
+                      sys.stdout.write(f'\033[{height-2};1H\033[K\033[90m{"─" * width}\033[0m')
+                      sys.stdout.write(f'\033[{height-1};1H\033[K [{preview_scroll+1}/{len(preview_lines)} lines]')
+                      sys.stdout.write(f'\033[{height};1H\033[K\033[46;30m j/k:Scroll b:Back q:Quit \033[0m')
+
+                  sys.stdout.flush()
+
+                  c = sys.stdin.read(1)
+
+                  if c == '\x1b':
+                      c2 = sys.stdin.read(1)
+                      if c2 == '[':
+                          c3 = sys.stdin.read(1)
+                          if c3 == 'A':  # Up
+                              if mode == 'list' and selected > 0:
+                                  selected -= 1
+                              elif mode == 'preview' and preview_scroll > 0:
+                                  preview_scroll -= 1
+                          elif c3 == 'B':  # Down
+                              if mode == 'list' and selected < len(tools) - 1:
+                                  selected += 1
+                              elif mode == 'preview' and preview_scroll < max(0, len(preview_lines) - list_height):
+                                  preview_scroll += 1
+                      else:
+                          if mode == 'preview':
+                              mode = 'list'
+                              sys.stdout.write('\033[2J\033[H')
+                          else:
+                              return None
+                      continue
+
+                  if c == 'q' or c == '\x03':
+                      return None
+                  elif c == 'k':
+                      if mode == 'list' and selected > 0:
+                          selected -= 1
+                      elif mode == 'preview' and preview_scroll > 0:
+                          preview_scroll -= 1
+                  elif c == 'j':
+                      if mode == 'list' and selected < len(tools) - 1:
+                          selected += 1
+                      elif mode == 'preview' and preview_scroll < max(0, len(preview_lines) - list_height):
+                          preview_scroll += 1
+                  elif c == 'p' and mode == 'list' and tools:
+                      # Preview tool details
+                      t = tools[selected]
+                      preview_str = f"Tool: {t['name']}\n"
+                      preview_str += f"{'=' * 40}\n\n"
+                      preview_str += f"Description:\n{t['description']}\n\n"
+                      preview_str += f"Parameters:\n"
+                      params = t.get('params', {})
+                      props = params.get('properties', {})
+                      required = params.get('required', [])
+                      for pname, pinfo in props.items():
+                          req = "*" if pname in required else ""
+                          ptype = pinfo.get('type', 'any')
+                          pdesc = pinfo.get('description', '')[:60]
+                          preview_str += f"  {pname}{req} ({ptype}): {pdesc}\n"
+                      if not props:
+                          preview_str += "  (no parameters)\n"
+                      preview_lines = preview_str.split('\n')
+                      mode = 'preview'
+                      preview_scroll = 0
+                      sys.stdout.write('\033[2J\033[H')
+                  elif c == 'b' and mode == 'preview':
+                      mode = 'list'
+                      sys.stdout.write('\033[2J\033[H')
+                  elif c in ('\r', '\n') and mode == 'list' and tools:
+                      # Return tool name for use
+                      return tools[selected]['name']
+
+          finally:
+              termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
+              sys.stdout.write('\033[?25h')
+              sys.stdout.write('\033[2J\033[H')
+              sys.stdout.flush()
+
+      print("""
+       ██████╗ ██████╗ ██████╗  ██████╗ █████╗
+      ██╔════╝██╔═══██╗██╔══██╗██╔════╝██╔══██╗
+      ██║     ██║   ██║██████╔╝██║     ███████║
+      ██║     ██║   ██║██╔══██╗██║     ██╔══██╗
+      ╚██████╗╚██████╔╝██║  ██║╚██████╗██║  ██║
+       ╚═════╝ ╚═════╝ ╚═╝  ╚═╝ ╚═════╝╚═╝  ╚═╝
+      """)
+
+      npc_name = npc.name if npc else "corca"
+      print(f"Entering corca mode (NPC: {npc_name}). Type '/cq' to exit.")
+
+      # ========== MCP Connection Setup ==========
+      async def connect_mcp(server_path):
+          """Connect to MCP server and return tools"""
+          if not MCP_AVAILABLE:
+              return [], {}
+
+          abs_path = os.path.abspath(os.path.expanduser(server_path))
+          if not os.path.exists(abs_path):
+              print(colored(f"MCP server not found: {abs_path}", "red"))
+              return [], {}
+
+          try:
+              loop = asyncio.get_event_loop()
+          except RuntimeError:
+              loop = asyncio.new_event_loop()
+              asyncio.set_event_loop(loop)
+
+          exit_stack = AsyncExitStack()
+
+          if abs_path.endswith('.py'):
+              cmd_parts = [sys.executable, abs_path]
+          else:
+              cmd_parts = [abs_path]
+
+          server_params = StdioServerParameters(
+              command=cmd_parts[0],
+              args=[abs_path],
+              env=os.environ.copy()
+          )
+
+          stdio_transport = await exit_stack.enter_async_context(stdio_client(server_params))
+          session = await exit_stack.enter_async_context(ClientSession(*stdio_transport))
+          await session.initialize()
+
+          response = await session.list_tools()
+          tools_llm = []
+          tool_map = {}
+
+          if response.tools:
+              for mcp_tool in response.tools:
+                  tool_def = {
+                      "type": "function",
+                      "function": {
+                          "name": mcp_tool.name,
+                          "description": mcp_tool.description or f"MCP tool: {mcp_tool.name}",
+                          "parameters": getattr(mcp_tool, "inputSchema", {"type": "object", "properties": {}})
+                      }
+                  }
+                  tools_llm.append(tool_def)
+
+                  # Create sync wrapper for async tool call
+                  def make_tool_func(tool_name, sess, lp):
+                      async def call_tool(**kwargs):
+                          cleaned = {k: (None if v == 'None' else v) for k, v in kwargs.items()}
+                          result = await asyncio.wait_for(sess.call_tool(tool_name, cleaned), timeout=30.0)
+                          return result
+                      def sync_call(**kwargs):
+                          return lp.run_until_complete(call_tool(**kwargs))
+                      return sync_call
+
+                  tool_map[mcp_tool.name] = make_tool_func(mcp_tool.name, session, loop)
+
+          # Store in shared context
+          shared_ctx['mcp_client'] = session
+          shared_ctx['mcp_tools'] = tools_llm
+          shared_ctx['mcp_tool_map'] = tool_map
+          shared_ctx['_mcp_exit_stack'] = exit_stack
+          shared_ctx['_mcp_loop'] = loop
+
+          print(colored(f"Connected to MCP server. Tools: {', '.join(tool_map.keys())}", "green"))
+          return tools_llm, tool_map
+
+      # Try to connect if server path provided
+      tools_llm = shared_ctx.get('mcp_tools', [])
+      tool_map = shared_ctx.get('mcp_tool_map', {})
+
+      if mcp_server_path and not tools_llm:
+          try:
+              loop = asyncio.get_event_loop()
+          except RuntimeError:
+              loop = asyncio.new_event_loop()
+              asyncio.set_event_loop(loop)
+          tools_llm, tool_map = loop.run_until_complete(connect_mcp(mcp_server_path))
+
+      # Find default MCP server if none provided
+      if not tools_llm:
+          default_paths = [
+              os.path.expanduser("~/.npcsh/npc_team/mcp_server.py"),
+              os.path.join(team.team_path, "mcp_server.py") if team and hasattr(team, 'team_path') else None,
+          ]
+          for path in default_paths:
+              if path and os.path.exists(path):
+                  try:
+                      loop = asyncio.get_event_loop()
+                  except RuntimeError:
+                      loop = asyncio.new_event_loop()
+                      asyncio.set_event_loop(loop)
+                  tools_llm, tool_map = loop.run_until_complete(connect_mcp(path))
+                  if tools_llm:
+                      break
+
+      # Ensure system message
+      if not messages or messages[0].get("role") != "system":
+          sys_msg = get_system_message(npc) if npc else "You are an AI assistant with access to tools."
+          if tools_llm:
+              sys_msg += f"\n\nYou have access to these tools: {', '.join(t['function']['name'] for t in tools_llm)}"
+          messages.insert(0, {"role": "system", "content": sys_msg})
+
+      # Handle initial command if provided (one-shot mode)
+      if initial_command:
+          resp = get_llm_response(
+              initial_command,
+              model=model,
+              provider=provider,
+              messages=messages,
+              tools=tools_llm if tools_llm else None,
+              tool_map=tool_map if tool_map else None,
+              auto_process_tool_calls=True,
+              npc=npc
+          )
+          messages = resp.get('messages', messages)
+          render_markdown(str(resp.get('response', '')))
+          context['output'] = resp.get('response', 'Done.')
+          context['messages'] = messages
+          # Don't enter REPL for one-shot
+          exit()
+
+      # REPL loop
+      while True:
+          try:
+              prompt_str = f"{npc_name}:corca> "
+              user_input = input(prompt_str).strip()
+
+              if not user_input:
+                  continue
+
+              if user_input.lower() == "/cq":
+                  print("Exiting corca mode.")
+                  break
+
+              # Handle /tools to browse available tools with TUI
+              if user_input.lower() == "/tools":
+                  result = tools_tui_browser(tools_llm)
+                  if result:
+                      print(colored(f"Selected tool: {result}", "cyan"))
+                      print(colored("Use it by describing what you want to do.", "gray"))
+                  continue
+
+              # Handle /connect to connect to new MCP server
+              if user_input.startswith("/connect "):
+                  new_path = user_input[9:].strip()
+                  try:
+                      loop = asyncio.get_event_loop()
+                  except RuntimeError:
+                      loop = asyncio.new_event_loop()
+                      asyncio.set_event_loop(loop)
+                  tools_llm, tool_map = loop.run_until_complete(connect_mcp(new_path))
+                  continue
+
+              # Get LLM response with tools
+              resp = get_llm_response(
+                  user_input,
+                  model=model,
+                  provider=provider,
+                  messages=messages,
+                  tools=tools_llm if tools_llm else None,
+                  tool_map=tool_map if tool_map else None,
+                  auto_process_tool_calls=True,
+                  stream=False,  # Tool calls don't work well with streaming
+                  npc=npc
+              )
+
+              messages = resp.get('messages', messages)
+              response_text = resp.get('response', '')
+              render_markdown(str(response_text))
+
+              # Track usage
+              if 'usage' in resp and npc and hasattr(npc, 'shared_context'):
+                  usage = resp['usage']
+                  npc.shared_context['session_input_tokens'] += usage.get('input_tokens', 0)
+                  npc.shared_context['session_output_tokens'] += usage.get('output_tokens', 0)
+                  npc.shared_context['turn_count'] += 1
+
+          except KeyboardInterrupt:
+              print("\nUse '/cq' to exit or continue.")
+              continue
+          except EOFError:
+              print("\nExiting corca mode.")
+              break
+
+      context['output'] = "Exited corca mode."
+      context['messages'] = messages