gnosys-strata 1.1.4__py3-none-any.whl

gnosys_strata-1.1.4.dist-info/METADATA

@@ -0,0 +1,140 @@
+Metadata-Version: 2.4
+Name: gnosys-strata
+Version: 1.1.4
+Summary: GNOSYS Fork of Strata MCP - Enhanced MCP Router and Orchestrator
+Project-URL: Homepage, https://github.com/sancovp/gnosys-strata
+Project-URL: Repository, https://github.com/sancovp/gnosys-strata.git
+Project-URL: Issues, https://github.com/sancovp/gnosys-strata/issues
+Author-email: Isaac Wostrel-Rubin <isaac@sanctuary.systems>, "Klavis AI (Original)" <connect@klavis.ai>
+License: Apache-2.0
+Keywords: ai,llm,mcp,model-context-protocol,router,tools
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Communications
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Topic :: System :: Distributed Computing
+Requires-Python: >=3.10
+Requires-Dist: bm25s>=0.2.14
+Requires-Dist: mcp>=1.0.0
+Requires-Dist: platformdirs>=4.4.0
+Requires-Dist: pystemmer>=3.0.0
+Requires-Dist: starlette>=0.37.0
+Requires-Dist: uvicorn>=0.30.0
+Requires-Dist: watchgod>=0.7
+Provides-Extra: dev
+Requires-Dist: pytest-asyncio>=0.21.0; extra == 'dev'
+Requires-Dist: pytest>=7.0.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# gnosys-strata
+
+Enhanced fork of [Klavis Strata](https://github.com/Klavis-AI/klavis) with JIT server management and catalog mode.
+
+## What's Different
+
+The original Strata has serious scalability issues:
+- **Memory hog**: Every MCP stays loaded in memory (2.7% each)
+- **No caching**: Rebuilds tool index on every startup
+- **Sync loading**: Couldn't even load 5 MCPs without OOM
+
+**gnosys-strata** fixes this:
+
+| Feature | Original | gnosys-strata |
+|---------|----------|---------------|
+| Server loading | All at once, sync | JIT, async |
+| Memory usage | O(n) always | O(active) |
+| Tool catalog | Rebuilt every time | Disk-cached |
+| 20+ MCPs | OOM crash | Works fine |
+| MCP Sets | No | Yes |
+
+## New Features
+
+### Catalog Mode
+Tool schemas are cached to disk. Search across ALL configured servers without loading them:
+
+```bash
+# Search tools across offline servers
+strata search "github issues"
+```
+
+### JIT Server Management
+Servers only load when you actually use them:
+
+```python
+# Via tools
+manage_servers(action="connect", server_name="github")  # Load on demand
+manage_servers(action="disconnect", server_name="github")  # Free memory
+manage_servers(action="list")  # See what's active
+```
+
+### MCP Sets
+Group servers into named sets, swap between them:
+
+```python
+# Coming soon
+activate_set("development")  # github, filesystem, git
+activate_set("production")   # monitoring, alerting, deploy
+```
+
+## Installation
+
+```bash
+pip install gnosys-strata
+```
+
+## Configuration
+
+Same as original Strata - config lives at `~/.config/strata/servers.json`:
+
+```json
+{
+  "mcp": {
+    "servers": {
+      "github": {
+        "command": "npx",
+        "args": ["-y", "@modelcontextprotocol/server-github"],
+        "env": {"GITHUB_TOKEN": "..."},
+        "enabled": true
+      }
+    }
+  }
+}
+```
+
+## Usage
+
+```bash
+# Run as MCP server (stdio mode)
+strata
+
+# Run as HTTP server
+strata run --port 8080
+```
+
+## Tools
+
+Core tools (always available):
+- `discover_server_actions` - Find tools across servers
+- `get_action_details` - Get tool schema
+- `execute_action` - Run a tool
+- `search_documentation` - Search tool docs
+
+New in gnosys-strata:
+- `manage_servers` - Connect/disconnect servers on demand
+- `search_mcp_catalog` - Search cached tool index (offline)
+
+## License
+
+Apache-2.0 (same as original)
+
+## Credits
+
+Fork of [Klavis AI Strata](https://github.com/Klavis-AI/klavis). Enhanced by Isaac Wostrel-Rubin.

gnosys_strata-1.1.4.dist-info/RECORD

@@ -0,0 +1,28 @@
+strata/__init__.py,sha256=GI47BYskraBCsfZuNWWtSmyGkSvNX_LoZ40V24bIDbs,99
+strata/__main__.py,sha256=Slavi7tIXimVnD9xbNi4IUbyJ-CCmJhonpXO88K1OoE,106
+strata/cli.py,sha256=AoAVG3Nd_u4-0AvRuInpNfSp2pWkDeixtYshzVpxsFM,12256
+strata/config.py,sha256=TOsCAOo2K8SOKrCa-ONTv0YQvIgFFdBvVEU-z3dJbwE,11509
+strata/logging_config.py,sha256=eH8SamY2QeiLaqkWT6sDkqyFpmqcjKc7CQoRHDgFDlk,5229
+strata/main.py,sha256=OYQLnz-AOmTF86c-Q8eIQfYei90Kwc-3k-S99ww5I6A,108
+strata/mcp_client_manager.py,sha256=L7_VCHnUDA2ekLBWISSjqAGQTtRqluuSLl4JnXIzOW0,11230
+strata/server.py,sha256=wD4_tlMqtc-Hk9Ph4rdPv89XwQQeIDb-u96MXHfs6LM,7887
+strata/tools.py,sha256=NOBXryyUgLN-h3mxyCFvD8l_mNL7pqNXvo6ci3ul1mo,32695
+strata/treeshell_functions.py,sha256=15nQHR_0DP-yoezkT5QeAdq3e7mMuMACVLswbS0pUMw,16532
+strata/mcp_proxy/__init__.py,sha256=3QUmavXYpUe2o352zTh3s3uw_cMCRItCMvxy3TUmUyc,317
+strata/mcp_proxy/auth_provider.py,sha256=kcLYiGnlWe_onsjHTRpaYM0Ien7PDD67i-0bIfWpNac,7461
+strata/mcp_proxy/client.py,sha256=2JyYyHUgD3AJEqv3P5dEj_lp7KnQ8z8AFQ1d6EHA2bY,5670
+strata/mcp_proxy/transport/__init__.py,sha256=XwOimRw542zh3YM9F4CTE9NfPJGESYsM71p8kIE9z0o,203
+strata/mcp_proxy/transport/base.py,sha256=yYeshpjz5aFObCxboy6Y0hJMaNtMp5zSyScnJfgWYIk,3383
+strata/mcp_proxy/transport/http.py,sha256=YYuCSAuA2qBb6WuBifvGtzVxsDAutAIR4SCl2CH9xhM,2864
+strata/mcp_proxy/transport/stdio.py,sha256=OAyESEl858FvuPnAJRfnIgILSpxQoWm1lu1ss0FOkIE,2235
+strata/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+strata/utils/bm25_search.py,sha256=aDhWLbrOGUZm9sU8dwEdxL9WQzluVBreSHKtjPJzdgY,6580
+strata/utils/catalog.py,sha256=6IomxQQg_l7dyHpr_I_1sIXSH0DM6KjH446TjcsglxM,2857
+strata/utils/dict_utils.py,sha256=VYTasKLQX2jWHOXRMb9vvyhP4Jr755AyNAsHBC1aCuA,757
+strata/utils/field_search.py,sha256=SoM6c2ybpp2mxh3KhcNLqRF0BJp6KYmkh-qK-BSOQzA,9245
+strata/utils/shared_search.py,sha256=6Z-Xnu3Ip0VGsZfDf6qkonuk2AaNa1dJ_VNehRUItqU,7828
+strata/utils/tool_integration.py,sha256=shZEqr8qp40898sxkDaRooLgFh7hOoYCLjCjhwWgJIc,8438
+gnosys_strata-1.1.4.dist-info/METADATA,sha256=VsNrXpdVBA1fm56kgL61O2vkRBaRsLVrafInbAXhAUc,4025
+gnosys_strata-1.1.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+gnosys_strata-1.1.4.dist-info/entry_points.txt,sha256=ibQdeqQnnZfMXBRcOWVzShkx_Fn5Apfm03o5j6I4E1Y,44
+gnosys_strata-1.1.4.dist-info/RECORD,,

gnosys_strata-1.1.4.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

gnosys_strata-1.1.4.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+strata = strata.main:main

strata/__init__.py

@@ -0,0 +1,6 @@
+"""Strata MCP Router package."""
+
+from .main import main
+
+__version__ = "0.1.0"
+__all__ = ["main"]

strata/__main__.py

@@ -0,0 +1,6 @@
+"""Entry point for running as a module."""
+
+from .main import main
+
+if __name__ == "__main__":
+    main()

strata/cli.py

@@ -0,0 +1,364 @@
+"""Command-line interface for Strata MCP Router using argparse."""
+
+import argparse
+import os
+import asyncio
+import sys
+from pathlib import Path
+from urllib.parse import urlparse
+
+from strata.mcp_client_manager import MCPClientManager
+
+from .config import MCPServerConfig, MCPServerList
+from .logging_config import setup_logging
+from .server import run_server, run_stdio_server
+from .utils.tool_integration import add_strata_to_tool
+
+
+def add_command(args):
+    """Add a new MCP server configuration."""
+    server_list = MCPServerList(args.config_path)
+
+    # Normalize type (command is alias for stdio)
+    server_type = args.type
+    if server_type == "command":
+        server_type = "stdio"
+
+    # Parse environment variables
+    env_dict = {}
+    if args.env:
+        for env_var in args.env:
+            if "=" not in env_var:
+                print(
+                    f"Error: Invalid environment variable format: {env_var}",
+                    file=sys.stderr,
+                )
+                print(
+                    "Environment variables must be in KEY=VALUE format", file=sys.stderr
+                )
+                return 1
+            key, value = env_var.split("=", 1)
+            env_dict[key.strip()] = value
+
+    # Parse headers for URL type
+    headers_dict = {}
+    if args.header:
+        for header_var in args.header:
+            if ":" not in header_var:
+                print(f"Error: Invalid header format: {header_var}", file=sys.stderr)
+                print("Headers must be in KEY:VALUE format", file=sys.stderr)
+                return 1
+            key, value = header_var.split(":", 1)
+            headers_dict[key.strip()] = value.strip()
+
+    if server_type in ["sse", "http"]:
+        # Parse HTTP URL to extract base URL and any query parameters
+        parsed = urlparse(args.url_or_command)
+
+        # Reconstruct URL without query parameters for the URL field
+        base_url = f"{parsed.scheme}://{parsed.netloc}{parsed.path}"
+        if parsed.path == "" or parsed.path == "/":
+            base_url = f"{parsed.scheme}://{parsed.netloc}"
+
+        auth = args.auth_type[0] if args.auth_type else ""
+        config = MCPServerConfig(
+            name=args.name,
+            type=server_type,
+            url=base_url,
+            headers=headers_dict,
+            env=env_dict,
+            enabled=args.enabled,
+            auth=auth
+        )
+    else:  # stdio/command
+        # For stdio, url_or_command is the command, args.args contains the arguments
+        config = MCPServerConfig(
+            name=args.name,
+            type="stdio",
+            command=args.url_or_command,
+            args=args.args if args.args else [],
+            env=env_dict,
+            enabled=args.enabled,
+        )
+
+    # Add server to configuration
+    if server_list.add_server(config):
+        server_list.save()
+        print(f"✓ Successfully added server '{args.name}' with {server_type} type")
+    else:
+        print(f"Server '{args.name}' already exists with identical configuration")
+
+    return 0
+
+
+def remove_command(args):
+    """Remove an MCP server configuration."""
+    server_list = MCPServerList(args.config_path)
+
+    if args.name in server_list.servers:
+        del server_list.servers[args.name]
+        server_list.save()
+        print(f"✓ Successfully removed server '{args.name}'")
+        return 0
+    else:
+        print(f"Error: Server '{args.name}' not found", file=sys.stderr)
+        return 1
+
+
+def list_command(args):
+    """List all configured MCP servers."""
+    server_list = MCPServerList(args.config_path)
+
+    if not server_list.servers:
+        print("No servers configured")
+        return 0
+
+    print("Configured MCP servers:")
+    for name, config in server_list.servers.items():
+        status = "enabled" if config.enabled else "disabled"
+        type_str = config.type or "stdio"
+
+        if type_str in ["sse", "http"]:
+            location = config.url
+        else:
+            location = f"{config.command} {' '.join(config.args or [])}"
+
+        print(f"  • {name} ({type_str}, {status}): {location}")
+
+    return 0
+
+
+def enable_command(args):
+    """Enable an MCP server."""
+    server_list = MCPServerList(args.config_path)
+
+    if args.name not in server_list.servers:
+        print(f"Error: Server '{args.name}' not found", file=sys.stderr)
+        return 1
+
+    server_list.servers[args.name].enabled = True
+    server_list.save()
+    print(f"✓ Enabled server '{args.name}'")
+    return 0
+
+
+def disable_command(args):
+    """Disable an MCP server."""
+    server_list = MCPServerList(args.config_path)
+
+    if args.name not in server_list.servers:
+        print(f"Error: Server '{args.name}' not found", file=sys.stderr)
+        return 1
+
+    server_list.servers[args.name].enabled = False
+    server_list.save()
+    print(f"✓ Disabled server '{args.name}'")
+    return 0
+
+async def authenticate(server_name):
+    # Only initialize the specific server we want to authenticate
+    async with MCPClientManager(server_names=[server_name]) as client_manager:
+        await client_manager.authenticate_server(server_name)
+
+def authenticate_command(args):
+    """Authenticate with an OAuth2 provider for a given MCP server URL."""
+    server_list = MCPServerList(args.config_path)
+
+    if args.name in server_list.servers:
+        asyncio.run(authenticate(args.name))
+        return 0
+    else:
+        print(f"Error: Server '{args.name}' not found", file=sys.stderr)
+        return 1
+
+def tool_add_command(args):
+    """Add Strata MCP server to Claude, Gemini, VSCode, or Cursor configurations."""
+    return add_strata_to_tool(args.target, args.scope or "user")
+
+
+def run_command(args):
+    """Run the Strata MCP router."""
+    # Initialize server with config path if provided
+    if args.config_path:
+        import os
+
+        os.environ["MCP_CONFIG_PATH"] = str(args.config_path)
+
+    if args.port is not None:
+        # Server mode with HTTP/SSE
+        print(f"Starting Strata MCP Router server on port {args.port}...")
+        return run_server(
+            args.port, json_response=False
+        )
+    else:
+        # Stdio mode
+        print("Starting Strata MCP Router in stdio mode...", file=sys.stderr)
+        return run_stdio_server()
+
+
+def create_parser():
+    """Create the argument parser."""
+    parser = argparse.ArgumentParser(
+        prog="strata", description="Strata MCP Router - Manage and run MCP servers"
+    )
+
+    # Global options
+    # Check environment variables for default config path
+    default_config_path = os.environ.get("STRATA_CONFIG_PATH")
+    parser.add_argument(
+        "--config-path",
+        type=Path,
+        default=Path(default_config_path) if default_config_path else None,
+        help="Path to configuration file (default: $STRATA_CONFIG_PATH or $MCP_CONFIG_PATH or ~/.config/strata/servers.json)",
+    )
+
+    subparsers = parser.add_subparsers(dest="command", help="Available commands")
+
+    # Add command - use REMAINDER to capture all remaining args including those starting with -
+    add_parser = subparsers.add_parser("add", help="Add a new MCP server configuration")
+    add_parser.add_argument(
+        "--type",
+        choices=["sse", "http", "stdio", "command"],
+        required=True,
+        help="Type of the MCP server (sse/http for URL-based, stdio/command for process)",
+    )
+    add_parser.add_argument(
+        "--env", action="append", help="Environment variables in KEY=VALUE format"
+    )
+    add_parser.add_argument(
+        "--header",
+        action="append",
+        help="HTTP headers in KEY:VALUE format (for URL type only)",
+    )
+    add_parser.add_argument(
+        "--auth_type",
+        action="append",
+        help="Authentication type (oauth etc.)",
+    )
+    add_parser.add_argument(
+        "--enabled/--disabled",
+        dest="enabled",
+        default=True,
+        action=argparse.BooleanOptionalAction,
+        help="Whether the server is enabled by default",
+    )
+    add_parser.add_argument("name", help="Name of the server")
+    add_parser.add_argument(
+        "url_or_command", help="URL for HTTP/SSE or command for stdio"
+    )
+    add_parser.add_argument(
+        "args",
+        nargs="*",
+        help="Arguments for stdio command (use -- before args starting with -)",
+    )
+    add_parser.set_defaults(func=add_command)
+
+    # Remove command
+    remove_parser = subparsers.add_parser(
+        "remove", help="Remove an MCP server configuration"
+    )
+    remove_parser.add_argument("name", help="Name of the server to remove")
+    remove_parser.set_defaults(func=remove_command)
+
+    # List command
+    list_parser = subparsers.add_parser("list", help="List all configured MCP servers")
+    list_parser.set_defaults(func=list_command)
+
+    # Enable command
+    enable_parser = subparsers.add_parser("enable", help="Enable an MCP server")
+    enable_parser.add_argument("name", help="Name of the server to enable")
+    enable_parser.set_defaults(func=enable_command)
+
+    # Disable command
+    disable_parser = subparsers.add_parser("disable", help="Disable an MCP server")
+    disable_parser.add_argument("name", help="Name of the server to disable")
+    disable_parser.set_defaults(func=disable_command)
+
+    # Authenticate command
+    auth_parser = subparsers.add_parser("auth", help="Authenticate with an MCP server")
+    auth_parser.add_argument("name", help="Name of the server to authenticate with")
+    auth_parser.set_defaults(func=authenticate_command)
+
+    # Run command
+    run_parser = subparsers.add_parser("run", help="Run the Strata MCP router")
+    run_parser.add_argument(
+        "--port",
+        type=int,
+        help="Port to listen on for HTTP/SSE server mode. If not provided, runs in stdio mode.",
+    )
+    run_parser.add_argument(
+        "--log-level",
+        default="INFO",
+        help="Logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL)",
+    )
+    run_parser.add_argument(
+        "--no-banner",
+        action="store_true",
+        help="Skip displaying the banner on startup",
+    )
+    run_parser.set_defaults(func=run_command)
+
+    # Set run as default command when no subcommand is provided
+    parser.set_defaults(
+        command="run", func=run_command, port=None, log_level="INFO", no_banner=False
+    )
+
+    # Tool command
+    tool_parser = subparsers.add_parser("tool", help="Tool integration commands")
+    tool_subparsers = tool_parser.add_subparsers(
+        dest="tool_command", help="Tool commands"
+    )
+
+    # Tool add command
+    tool_add_parser = tool_subparsers.add_parser(
+        "add", help="Add Strata to Claude, Gemini, VSCode, or Cursor MCP configuration"
+    )
+    tool_add_parser.add_argument(
+        "target",
+        choices=["claude", "gemini", "vscode", "cursor"],
+        help="Target CLI to add Strata to (claude, gemini, vscode, or cursor)",
+    )
+    tool_add_parser.add_argument(
+        "--scope",
+        choices=["user", "project", "local"],
+        default="user",
+        help="Configuration scope (user, project, or local). Default: user. Note: VSCode doesn't support scope.",
+    )
+    tool_add_parser.set_defaults(func=tool_add_command)
+
+    return parser
+
+
+def main():
+    """Main entry point for the MCP Router."""
+    parser = create_parser()
+    args = parser.parse_args()
+
+    # Initialize logging with appropriate settings
+    log_level = getattr(args, "log_level", "INFO")
+    no_banner = getattr(args, "no_banner", False)
+    # banner is only relevant when running the server
+    if args.command != "run":
+        no_banner = True
+    setup_logging(log_level, no_banner)
+
+    # If no subcommand provided, show help
+    if not hasattr(args, "func"):
+        parser.print_help()
+        return 1
+
+    # Handle tool subcommand specially
+    if args.command == "tool" and not hasattr(args, "func"):
+        # No tool subcommand provided, show help message
+        print("Error: No tool subcommand provided", file=sys.stderr)
+        print("\nAvailable tool commands:")
+        print("  add    Add Strata to Claude, Gemini, or VSCode MCP configuration")
+        print("\nUse 'strata tool <command> --help' for more information on a command.")
+        return 1
+
+    # Execute the command
+    return args.func(args)
+
+
+if __name__ == "__main__":
+    sys.exit(main())