casual-mcp 0.6.0__py3-none-any.whl → 0.7.1__py3-none-any.whl

casual_mcp/cli.py

@@ -1,7 +1,12 @@
 import asyncio
+import gc
+import json
+import warnings
+from pathlib import Path
 from typing import Any
 
 import mcp
+import questionary
 import typer
 import uvicorn
 from fastmcp import Client
@@ -9,6 +14,8 @@ from rich.console import Console
 from rich.table import Table
 
 from casual_mcp.models.mcp_server_config import RemoteServerConfig
+from casual_mcp.models.toolset_config import ExcludeSpec, ToolSpec
+from casual_mcp.tool_filter import extract_server_and_tool
 from casual_mcp.utils import load_config, load_mcp_client
 
 app = typer.Typer()
@@ -68,16 +75,382 @@ def tools() -> None:
     config = load_config("casual_mcp_config.json")
     mcp_client = load_mcp_client(config)
     table = Table("Name", "Description")
-    # async with mcp_client:
-    tools = asyncio.run(get_tools(mcp_client))
-    for tool in tools:
+    tool_list = run_async_with_cleanup(get_tools_and_cleanup(mcp_client))
+    for tool in tool_list:
         table.add_row(tool.name, tool.description)
     console.print(table)
 
 
-async def get_tools(client: Client[Any]) -> list[mcp.Tool]:
-    async with client:
-        return await client.list_tools()
+def run_async_with_cleanup(coro: Any) -> Any:
+    """Run async coroutine with proper subprocess cleanup.
+
+    This wrapper filters/ignores the "Event loop is closed" warning that occurs
+    when subprocess transports don't finish cleanup before the event loop closes.
+    It also forces gc.collect() after execution to help clean up remaining transports.
+    """
+    with warnings.catch_warnings():
+        warnings.filterwarnings("ignore", message="Event loop is closed")
+        try:
+            return asyncio.run(coro)
+        finally:
+            # Force garbage collection to clean up any remaining transports
+            gc.collect()
+
+
+async def get_tools_and_cleanup(client: Client[Any]) -> list[mcp.Tool]:
+    """Get tools and ensure proper cleanup to avoid subprocess warnings."""
+    try:
+        async with client:
+            return await client.list_tools()
+    finally:
+        # Give transports time to close cleanly
+        await asyncio.sleep(0.1)
+
+
+def _build_server_tool_map(tools: list[mcp.Tool], server_names: set[str]) -> dict[str, list[str]]:
+    """Build a mapping of server names to their tool names."""
+    server_tools: dict[str, list[str]] = {s: [] for s in server_names}
+    for tool in tools:
+        server_name, base_name = extract_server_and_tool(tool.name, server_names)
+        if server_name in server_tools:
+            server_tools[server_name].append(base_name)
+    return server_tools
+
+
+def _format_tool_spec(spec: ToolSpec) -> str:
+    """Format a tool spec for display.
+
+    Note: Brackets are escaped for Rich console output.
+    """
+    if spec is True:
+        return "\\[all tools]"
+    elif isinstance(spec, list):
+        return ", ".join(spec)
+    elif isinstance(spec, ExcludeSpec):
+        return f"\\[all except: {', '.join(spec.exclude)}]"
+    return str(spec)
+
+
+@app.command()
+def toolsets() -> None:
+    """Interactively manage toolsets - create, edit, and delete."""
+    config_path = Path("casual_mcp_config.json")
+
+    while True:
+        config = load_config(config_path)
+
+        # Build menu choices
+        choices: list[Any] = []
+
+        if config.tool_sets:
+            for name, ts in config.tool_sets.items():
+                servers = ", ".join(ts.servers.keys()) or "no servers"
+                desc = ts.description[:40] + "..." if len(ts.description) > 40 else ts.description
+                display = f"{name} - {desc} ({servers})"
+                choices.append(questionary.Choice(title=display, value=name))
+
+            choices.append(questionary.Separator())
+
+        choices.append(questionary.Choice(title="➕ Create new toolset", value="__create__"))
+        choices.append(questionary.Choice(title="❌ Exit", value="__exit__"))
+
+        selection = questionary.select(
+            "Toolsets:" if config.tool_sets else "No toolsets configured:",
+            choices=choices,
+        ).ask()
+
+        if selection is None or selection == "__exit__":
+            return
+
+        if selection == "__create__":
+            _create_toolset(config_path)
+            continue
+
+        # Selected an existing toolset - show actions
+        _toolset_actions(config_path, selection)
+
+
+def _create_toolset(config_path: Path) -> None:
+    """Prompt for name and create a new toolset."""
+    config = load_config(config_path)
+
+    name = questionary.text("Toolset name:").ask()
+    if not name:
+        return
+
+    if name in config.tool_sets:
+        console.print(f"[red]Toolset '{name}' already exists[/red]")
+        return
+
+    _interactive_toolset_edit(config_path, config, name, is_new=True)
+
+
+def _toolset_actions(config_path: Path, name: str) -> None:
+    """Show actions for an existing toolset."""
+    config = load_config(config_path)
+    ts = config.tool_sets.get(name)
+
+    if not ts:
+        return
+
+    # Show toolset details
+    console.print(f"\n[bold]{name}[/bold]")
+    console.print(f"Description: {ts.description}")
+    console.print("Servers:")
+    for server, spec in ts.servers.items():
+        console.print(f"  {server}: {_format_tool_spec(spec)}")
+    console.print()
+
+    action = questionary.select(
+        "Action:",
+        choices=[
+            questionary.Choice(title="✏️  Edit", value="edit"),
+            questionary.Choice(title="🗑️  Delete", value="delete"),
+            questionary.Choice(title="← Back", value="back"),
+        ],
+    ).ask()
+
+    if action is None or action == "back":
+        return
+
+    if action == "edit":
+        _interactive_toolset_edit(config_path, config, name, is_new=False)
+    elif action == "delete":
+        _delete_toolset(config_path, name)
+
+
+def _delete_toolset(config_path: Path, name: str) -> None:
+    """Delete a toolset after confirmation."""
+    confirmed = questionary.confirm(f"Delete toolset '{name}'?", default=False).ask()
+    if not confirmed:
+        return
+
+    with config_path.open("r", encoding="utf-8") as f:
+        raw = json.load(f)
+
+    # Check if tool_sets exists and contains the toolset
+    if "tool_sets" not in raw:
+        console.print("[red]No toolsets found in config[/red]")
+        return
+
+    if name not in raw["tool_sets"]:
+        console.print(f"[red]Toolset '{name}' not found in config[/red]")
+        return
+
+    del raw["tool_sets"][name]
+
+    # Remove tool_sets key if empty
+    if not raw["tool_sets"]:
+        del raw["tool_sets"]
+
+    with config_path.open("w", encoding="utf-8") as f:
+        json.dump(raw, f, indent=4)
+
+    console.print(f"[green]Deleted toolset '{name}'[/green]")
+
+
+def _format_server_status(server: str, spec: ToolSpec | None, tool_count: int) -> str:
+    """Format server status for display in the menu."""
+    if spec is None:
+        return f"{server} ({tool_count} tools) - [dim]not included[/dim]"
+    elif spec is True:
+        return f"{server} ({tool_count} tools) - [green]all tools[/green]"
+    elif isinstance(spec, list):
+        return f"{server} ({tool_count} tools) - [cyan]{len(spec)} included[/cyan]"
+    elif isinstance(spec, ExcludeSpec):
+        return f"{server} ({tool_count} tools) - [yellow]{len(spec.exclude)} excluded[/yellow]"
+    return f"{server} ({tool_count} tools)"
+
+
+def _interactive_toolset_edit(
+    config_path: Path,
+    config: Any,
+    name: str,
+    is_new: bool,
+) -> None:
+    """Interactive toolset creation/editing with arrow-key navigation."""
+    from casual_mcp.models.config import Config
+
+    config = Config.model_validate(config.model_dump())
+
+    # Get available tools from servers
+    mcp_client = load_mcp_client(config)
+    tools = run_async_with_cleanup(get_tools_and_cleanup(mcp_client))
+    server_names = set(config.servers.keys())
+    server_tools = _build_server_tool_map(tools, server_names)
+
+    # Get existing config if editing
+    existing = config.tool_sets.get(name)
+    existing_description = existing.description if existing else ""
+    existing_servers: dict[str, Any] = dict(existing.servers) if existing else {}
+
+    # Working copy of server configs
+    new_servers: dict[str, Any] = dict(existing_servers)
+
+    if is_new:
+        console.print(f"[green]Creating new toolset '{name}'[/green]\n")
+    else:
+        console.print(f"[yellow]Editing toolset '{name}'[/yellow]\n")
+
+    # Get description
+    description = questionary.text(
+        "Description:",
+        default=existing_description,
+    ).ask()
+
+    if description is None:
+        raise typer.Abort()
+
+    # Main loop - configure servers one at a time
+    sorted_servers = sorted(server_names)
+
+    while True:
+        console.print("\n[bold]Server Configuration:[/bold]")
+        console.print("[dim]Configure each server, then select 'Save and exit' when done.[/dim]\n")
+
+        # Build menu choices showing current status
+        choices = []
+        for server in sorted_servers:
+            spec = new_servers.get(server)
+            tool_count = len(server_tools[server])
+            status = _format_server_status(server, spec, tool_count)
+            # Strip Rich markup for questionary display
+            plain_status = (
+                status.replace("[dim]", "")
+                .replace("[/dim]", "")
+                .replace("[green]", "")
+                .replace("[/green]", "")
+                .replace("[cyan]", "")
+                .replace("[/cyan]", "")
+                .replace("[yellow]", "")
+                .replace("[/yellow]", "")
+            )
+            choices.append(questionary.Choice(title=plain_status, value=server))
+
+        choices.append(questionary.Separator())
+        choices.append(questionary.Choice(title="💾 Save and exit", value="__save__"))
+        choices.append(questionary.Choice(title="❌ Cancel", value="__cancel__"))
+
+        selection = questionary.select(
+            "Select a server to configure:",
+            choices=choices,
+        ).ask()
+
+        if selection is None or selection == "__cancel__":
+            raise typer.Abort()
+
+        if selection == "__save__":
+            break
+
+        # Configure the selected server
+        server = selection
+        available = server_tools[server]
+        existing_spec = new_servers.get(server)
+
+        # Determine current mode for default selection
+        if existing_spec is None:
+            default_mode = "Don't include"
+        elif existing_spec is True:
+            default_mode = "All tools"
+        elif isinstance(existing_spec, list):
+            default_mode = "Include specific tools"
+        elif isinstance(existing_spec, ExcludeSpec):
+            default_mode = "Exclude specific tools"
+        else:
+            default_mode = "Don't include"
+
+        mode = questionary.select(
+            f"Configure {server} ({len(available)} tools):",
+            choices=[
+                "Don't include",
+                "All tools",
+                "Include specific tools",
+                "Exclude specific tools",
+            ],
+            default=default_mode,
+        ).ask()
+
+        if mode is None:
+            continue  # Go back to server list
+
+        if mode == "Don't include":
+            new_servers.pop(server, None)
+        elif mode == "All tools":
+            new_servers[server] = True
+        elif mode == "Include specific tools":
+            # Determine which tools were previously selected
+            if isinstance(existing_spec, list):
+                pre_selected = set(existing_spec)
+            else:
+                pre_selected = set()
+
+            tool_choices = [
+                questionary.Choice(title=tool, value=tool, checked=tool in pre_selected)
+                for tool in available
+            ]
+
+            console.print("\n[dim]Use space to select, enter to confirm[/dim]")
+            selected_tools = questionary.checkbox(
+                f"Select tools to include from {server}:",
+                choices=tool_choices,
+            ).ask()
+
+            if selected_tools is None:
+                continue  # Go back to server list
+
+            if selected_tools:
+                new_servers[server] = selected_tools
+            else:
+                new_servers.pop(server, None)  # No tools = don't include
+        elif mode == "Exclude specific tools":
+            # Determine which tools were previously excluded
+            if isinstance(existing_spec, ExcludeSpec):
+                pre_excluded = set(existing_spec.exclude)
+            else:
+                pre_excluded = set()
+
+            tool_choices = [
+                questionary.Choice(title=tool, value=tool, checked=tool in pre_excluded)
+                for tool in available
+            ]
+
+            console.print("\n[dim]Use space to select, enter to confirm[/dim]")
+            excluded_tools = questionary.checkbox(
+                f"Select tools to exclude from {server}:",
+                choices=tool_choices,
+            ).ask()
+
+            if excluded_tools is None:
+                continue  # Go back to server list
+
+            if excluded_tools:
+                new_servers[server] = {"exclude": excluded_tools}
+            else:
+                new_servers[server] = True  # No exclusions = all tools
+
+    # Check if any servers are configured
+    if not new_servers:
+        console.print("[yellow]No servers configured - toolset will be empty[/yellow]")
+        confirm = questionary.confirm("Save empty toolset?", default=False).ask()
+        if not confirm:
+            raise typer.Abort()
+
+    # Save to config
+    with config_path.open("r", encoding="utf-8") as f:
+        raw = json.load(f)
+
+    if "tool_sets" not in raw:
+        raw["tool_sets"] = {}
+
+    raw["tool_sets"][name] = {
+        "description": description,
+        "servers": new_servers,
+    }
+
+    with config_path.open("w", encoding="utf-8") as f:
+        json.dump(raw, f, indent=4)
+
+    console.print(f"\n[green]Saved toolset '{name}'[/green]")
 
 
 if __name__ == "__main__":

casual_mcp/main.py

@@ -8,8 +8,10 @@ from pydantic import BaseModel, Field
 
 from casual_mcp import McpToolChat
 from casual_mcp.logging import configure_logging, get_logger
+from casual_mcp.models.toolset_config import ToolSetConfig
 from casual_mcp.provider_factory import ProviderFactory
 from casual_mcp.tool_cache import ToolCache
+from casual_mcp.tool_filter import ToolSetValidationError
 from casual_mcp.utils import load_config, load_mcp_client, render_system_prompt
 
 # Load environment variables
@@ -46,6 +48,7 @@ class GenerateRequest(BaseModel):
     system_prompt: str | None = Field(default=None, title="System Prompt to use")
     prompt: str = Field(title="User Prompt")
     include_stats: bool = Field(default=False, title="Include usage statistics in response")
+    tool_set: str | None = Field(default=None, title="Name of toolset to use")
 
 
 class ChatRequest(BaseModel):
@@ -53,12 +56,43 @@ class ChatRequest(BaseModel):
     system_prompt: str | None = Field(default=None, title="System Prompt to use")
     messages: list[ChatMessage] = Field(title="Previous messages to supply to the LLM")
     include_stats: bool = Field(default=False, title="Include usage statistics in response")
+    tool_set: str | None = Field(default=None, title="Name of toolset to use")
+
+
+def resolve_tool_set(tool_set_name: str | None) -> ToolSetConfig | None:
+    """Resolve a tool set name to its configuration.
+
+    Args:
+        tool_set_name: Name of the tool set, or None
+
+    Returns:
+        ToolSetConfig if found, None if tool_set_name is None
+
+    Raises:
+        HTTPException: If tool set name is provided but not found
+    """
+    if tool_set_name is None:
+        return None
+
+    if tool_set_name not in config.tool_sets:
+        raise HTTPException(
+            status_code=400,
+            detail=f"Toolset '{tool_set_name}' not found. "
+            f"Available: {list(config.tool_sets.keys())}",
+        )
+
+    return config.tool_sets[tool_set_name]
 
 
 @app.post("/chat")
 async def chat(req: ChatRequest) -> dict[str, Any]:
+    tool_set_config = resolve_tool_set(req.tool_set)
     chat_instance = await get_chat(req.model, req.system_prompt)
-    messages = await chat_instance.chat(req.messages)
+
+    try:
+        messages = await chat_instance.chat(req.messages, tool_set=tool_set_config)
+    except ToolSetValidationError as e:
+        raise HTTPException(status_code=400, detail=str(e))
 
     if not messages:
         error_result: dict[str, Any] = {"messages": [], "response": ""}
@@ -77,8 +111,15 @@ async def chat(req: ChatRequest) -> dict[str, Any]:
 
 @app.post("/generate")
 async def generate(req: GenerateRequest) -> dict[str, Any]:
+    tool_set_config = resolve_tool_set(req.tool_set)
     chat_instance = await get_chat(req.model, req.system_prompt)
-    messages = await chat_instance.generate(req.prompt, req.session_id)
+
+    try:
+        messages = await chat_instance.generate(
+            req.prompt, req.session_id, tool_set=tool_set_config
+        )
+    except ToolSetValidationError as e:
+        raise HTTPException(status_code=400, detail=str(e))
 
     if not messages:
         error_result: dict[str, Any] = {"messages": [], "response": ""}
@@ -104,6 +145,18 @@ async def get_generate_session(session_id: str) -> list[ChatMessage]:
     return session
 
 
+@app.get("/toolsets")
+async def list_toolsets() -> dict[str, dict[str, Any]]:
+    """List all available toolsets."""
+    return {
+        name: {
+            "description": ts.description,
+            "servers": list(ts.servers.keys()),
+        }
+        for name, ts in config.tool_sets.items()
+    }
+
+
 async def get_chat(model: str, system: str | None = None) -> McpToolChat:
     # Get Provider from Model Config
     model_config = config.models[model]
@@ -117,4 +170,10 @@ async def get_chat(model: str, system: str | None = None) -> McpToolChat:
         else:
             system = default_system_prompt
 
-    return McpToolChat(mcp_client, provider, system, tool_cache=tool_cache)
+    return McpToolChat(
+        mcp_client,
+        provider,
+        system,
+        tool_cache=tool_cache,
+        server_names=set(config.servers.keys()),
+    )

casual_mcp/mcp_tool_chat.py

@@ -15,12 +15,17 @@ from fastmcp import Client
 from casual_mcp.convert_tools import tools_from_mcp
 from casual_mcp.logging import get_logger
 from casual_mcp.models.chat_stats import ChatStats
+from casual_mcp.models.toolset_config import ToolSetConfig
 from casual_mcp.tool_cache import ToolCache
+from casual_mcp.tool_filter import filter_tools_by_toolset
 from casual_mcp.utils import format_tool_call_result
 
 logger = get_logger("mcp_tool_chat")
 sessions: dict[str, list[ChatMessage]] = {}
 
+# Type alias for metadata dictionary
+MetaDict = dict[str, Any]
+
 
 def get_session_messages(session_id: str) -> list[ChatMessage]:
     global sessions
@@ -45,11 +50,13 @@ class McpToolChat:
         provider: LLMProvider,
         system: str | None = None,
         tool_cache: ToolCache | None = None,
+        server_names: set[str] | None = None,
     ):
         self.provider = provider
         self.mcp_client = mcp_client
         self.system = system
         self.tool_cache = tool_cache or ToolCache(mcp_client)
+        self.server_names = server_names or set()
         self._tool_cache_version = -1
         self._last_stats: ChatStats | None = None
 
@@ -80,7 +87,28 @@ class McpToolChat:
             return tool_name.split("_", 1)[0]
         return "default"
 
-    async def generate(self, prompt: str, session_id: str | None = None) -> list[ChatMessage]:
+    async def generate(
+        self,
+        prompt: str,
+        session_id: str | None = None,
+        tool_set: ToolSetConfig | None = None,
+        meta: MetaDict | None = None,
+    ) -> list[ChatMessage]:
+        """
+        Generate a response to a prompt, optionally using session history.
+
+        Args:
+            prompt: The user prompt to respond to
+            session_id: Optional session ID for conversation persistence
+            tool_set: Optional tool set configuration to filter available tools
+            meta: Optional metadata to pass through to MCP tool calls.
+                  Useful for passing context like character_id without
+                  exposing it to the LLM. Servers can access this via
+                  ctx.request_context.meta.
+
+        Returns:
+            List of response messages including any tool calls and results
+        """
         # Fetch the session if we have a session ID
         messages: list[ChatMessage]
         if session_id:
@@ -97,7 +125,7 @@ class McpToolChat:
             add_messages_to_session(session_id, [user_message])
 
         # Perform Chat
-        response = await self.chat(messages=messages)
+        response = await self.chat(messages=messages, tool_set=tool_set, meta=meta)
 
         # Add responses to session
         if session_id:
@@ -105,9 +133,33 @@ class McpToolChat:
 
         return response
 
-    async def chat(self, messages: list[ChatMessage]) -> list[ChatMessage]:
+    async def chat(
+        self,
+        messages: list[ChatMessage],
+        tool_set: ToolSetConfig | None = None,
+        meta: MetaDict | None = None,
+    ) -> list[ChatMessage]:
+        """
+        Process a conversation with tool calling support.
+
+        Args:
+            messages: The conversation messages to process
+            tool_set: Optional tool set configuration to filter available tools
+            meta: Optional metadata to pass through to MCP tool calls.
+                  Useful for passing context like character_id without
+                  exposing it to the LLM. Servers can access this via
+                  ctx.request_context.meta.
+
+        Returns:
+            List of response messages including any tool calls and results
+        """
         tools = await self.tool_cache.get_tools()
 
+        # Filter tools if a toolset is specified
+        if tool_set is not None:
+            tools = filter_tools_by_toolset(tools, tool_set, self.server_names, validate=True)
+            logger.info(f"Filtered to {len(tools)} tools using toolset")
+
         # Reset stats at the start of each chat
         self._last_stats = ChatStats()
 
@@ -155,13 +207,18 @@ class McpToolChat:
                 )
 
                 try:
-                    result = await self.execute(tool_call)
+                    result = await self.execute(tool_call, meta=meta)
                 except Exception as e:
                     logger.error(
                         f"Failed to execute tool '{tool_call.function.name}' "
                         f"(id={tool_call.id}): {e}"
                     )
-                    continue
+                    # Surface the failure to the LLM so it knows the tool failed
+                    result = ToolResultMessage(
+                        name=tool_call.function.name,
+                        tool_call_id=tool_call.id,
+                        content=f"Error executing tool: {e}",
+                    )
                 if result:
                     messages.append(result)
                     response_messages.append(result)
@@ -173,12 +230,27 @@ class McpToolChat:
 
         return response_messages
 
-    async def execute(self, tool_call: AssistantToolCall) -> ToolResultMessage:
+    async def execute(
+        self,
+        tool_call: AssistantToolCall,
+        meta: MetaDict | None = None,
+    ) -> ToolResultMessage:
+        """
+        Execute a single tool call.
+
+        Args:
+            tool_call: The tool call to execute
+            meta: Optional metadata to pass to the MCP server.
+                  Servers can access this via ctx.request_context.meta.
+
+        Returns:
+            ToolResultMessage with the tool execution result
+        """
         tool_name = tool_call.function.name
         tool_args = json.loads(tool_call.function.arguments)
         try:
             async with self.mcp_client:
-                result = await self.mcp_client.call_tool(tool_name, tool_args)
+                result = await self.mcp_client.call_tool(tool_name, tool_args, meta=meta)
         except Exception as e:
             if isinstance(e, ValueError):
                 logger.warning(e)

casual_mcp/models/__init__.py

@@ -22,6 +22,11 @@ from .model_config import (
     OllamaModelConfig,
     OpenAIModelConfig,
 )
+from .toolset_config import (
+    ExcludeSpec,
+    ToolSetConfig,
+    ToolSpec,
+)
 
 __all__ = [
     "UserMessage",
@@ -39,4 +44,7 @@ __all__ = [
     "McpServerConfig",
     "StdioServerConfig",
     "RemoteServerConfig",
+    "ExcludeSpec",
+    "ToolSetConfig",
+    "ToolSpec",
 ]