onetool-mcp 1.0.0b1__py3-none-any.whl

onetool_mcp-1.0.0b1.dist-info/licenses/NOTICE.txt

@@ -0,0 +1,64 @@
+OneTool
+Copyright 2024-present, OneTool Authors
+
+This product includes software developed by third parties.
+
+================================================================================
+trafilatura
+================================================================================
+Copyright 2019-present, Adrien Barbaresi and contributors
+Licensed under the Apache License, Version 2.0
+https://github.com/adbar/trafilatura
+
+Used in: tools/web_fetch.py
+
+================================================================================
+brave-search-mcp-server
+================================================================================
+Copyright (c) 2024 Anthropic, PBC
+Copyright (c) 2025 Brave Software, Inc
+Licensed under the MIT License
+https://github.com/brave/brave-search-mcp-server
+
+Used in: tools/brave_search.py (API integration patterns)
+
+================================================================================
+context7
+================================================================================
+Copyright (c) 2021 Upstash, Inc.
+Licensed under the MIT License
+https://github.com/upstash/context7
+
+Used in: tools/context7.py (API integration patterns)
+
+================================================================================
+ChunkHound
+================================================================================
+Copyright (c) 2025 Ofri Wolfus
+Licensed under the MIT License
+https://github.com/chunkhound/chunkhound
+
+Used in: tools/code_search.py (LanceDB schema and indexing patterns)
+
+================================================================================
+mcp-alchemy
+================================================================================
+Copyright (c) 2024 Rune Kaagaard
+Licensed under the Mozilla Public License, v. 2.0
+https://github.com/runekaagaard/mcp-alchemy
+
+Used in: ot_tools/db.py (database introspection and query patterns)
+
+================================================================================
+firecrawl
+================================================================================
+Copyright (c) 2024 Mendable.ai
+Licensed under the AGPL-3.0 License (SDK and server)
+https://github.com/firecrawl/firecrawl
+
+Used in: ot_tools/firecrawl.py (API client SDK)
+Note: We use the SDK to call Firecrawl's cloud API. No server code is distributed.
+
+================================================================================
+
+Full license texts are available in the licenses/ directory.

ot/__init__.py

@@ -0,0 +1,37 @@
+"""OneTool - MCP server with single 'run' tool for LLM code generation.
+
+Features:
+- Single 'run' tool for Python code execution
+- Tool discovery from src/ot_tools/ directory
+- Configurable prompts and instructions
+- Namespaces, aliases, and snippets for shortcuts
+
+Usage:
+    # Start MCP server (stdio transport)
+    onetool
+
+    # With config
+    onetool --config config/onetool.yaml
+
+    # Run benchmarks
+    bench run harness.yaml
+"""
+
+from importlib.metadata import PackageNotFoundError, version
+from typing import Any
+
+try:
+    __version__ = version("onetool-mcp")
+except PackageNotFoundError:
+    __version__ = "0.0.0-dev"  # Running from source or in worker subprocess
+
+__all__ = ["__version__", "main"]
+
+
+def __getattr__(name: str) -> Any:
+    """Lazy import for server module to avoid loading config at import time."""
+    if name == "main":
+        from ot.server import main
+
+        return main
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

ot/__main__.py

@@ -0,0 +1,6 @@
+"""Run OneTool MCP server when executed as a module."""
+
+from ot.server import main
+
+if __name__ == "__main__":
+    main()

ot/_cli.py

@@ -0,0 +1,107 @@
+"""Shared CLI utilities for OneTool CLIs.
+
+Provides common patterns used across onetool and bench CLIs.
+"""
+
+from __future__ import annotations
+
+from pathlib import Path
+from typing import TYPE_CHECKING
+
+import typer
+from rich.console import Console
+
+if TYPE_CHECKING:
+    from collections.abc import Callable
+
+__all__ = ["console", "create_cli", "version_callback"]
+
+
+def _is_debug_tracebacks() -> bool:
+    """Check if verbose tracebacks are enabled.
+
+    Reads debug_tracebacks from ~/.onetool/onetool.yaml.
+    Returns False on any error (fail-safe for broken configs).
+    """
+    try:
+        import yaml
+
+        config_path = Path.home() / ".onetool" / "onetool.yaml"
+        if not config_path.exists():
+            return False
+        with config_path.open() as f:
+            data = yaml.safe_load(f)
+        return bool(data.get("debug_tracebacks", False)) if data else False
+    except Exception:
+        return False
+
+# Shared console instance for consistent output
+console = Console(highlight=False)
+
+
+def version_callback(name: str, version: str) -> Callable[[bool], None]:
+    """Create a version callback for Typer CLI.
+
+    Args:
+        name: CLI name to display (e.g., "ot", "bench")
+        version: Version string to display
+
+    Returns:
+        Callback function for --version option
+
+    Example:
+        @app.callback()
+        def main(
+            version: bool | None = typer.Option(
+                None,
+                "--version", "-v",
+                callback=version_callback("ot", __version__),
+                is_eager=True,
+                help="Show version and exit.",
+            ),
+        ) -> None:
+            ...
+    """
+
+    def callback(value: bool) -> None:
+        if value:
+            console.print(f"{name} version {version}")
+            raise typer.Exit()
+
+    return callback
+
+
+def create_cli(
+    name: str,
+    help_text: str,
+    *,
+    no_args_is_help: bool = False,
+) -> typer.Typer:
+    """Create a Typer CLI app with standard configuration.
+
+    Args:
+        name: CLI name
+        help_text: Help text for the CLI
+        no_args_is_help: Show help when no args provided (default: False)
+
+    Returns:
+        Configured Typer app
+
+    Note:
+        Set debug_tracebacks: true in ~/.onetool/onetool.yaml for Rich
+        formatted tracebacks with local variables and syntax highlighting.
+
+    Example:
+        app = create_cli(
+            "bench",
+            "OneTool benchmark harness.",
+            no_args_is_help=True,
+        )
+    """
+    debug = _is_debug_tracebacks()
+    return typer.Typer(
+        name=name,
+        help=help_text,
+        no_args_is_help=no_args_is_help,
+        pretty_exceptions_enable=debug,
+    )

ot/_tui.py

@@ -0,0 +1,53 @@
+"""Shared TUI primitives for interactive CLI tools.
+
+Used by bench for interactive selection prompts.
+"""
+
+from __future__ import annotations
+
+import questionary
+from questionary import Style
+
+# Consistent style across all prompts
+APP_STYLE = Style(
+    [
+        ("qmark", "fg:#5c9aff bold"),
+        ("question", "fg:#e0e0e0 bold"),
+        ("answer", "fg:#7dd3a8"),
+        ("pointer", "fg:#5c9aff bold"),
+        ("highlighted", "fg:#ffffff bg:#3d5a80"),
+        ("selected", "fg:#7dd3a8"),
+    ]
+)
+
+
+async def safe_ask(question: questionary.Question) -> str | None:
+    """Wrap questionary ask with graceful cancellation."""
+    try:
+        result = await question.ask_async()
+        return str(result) if result is not None else None
+    except KeyboardInterrupt:
+        return None
+
+
+async def ask_select(
+    prompt: str,
+    choices: list[questionary.Choice],
+) -> str | None:
+    """Prompt for selection with shortcuts."""
+    return await safe_ask(
+        questionary.select(
+            prompt,
+            choices=choices,
+            style=APP_STYLE,
+            use_shortcuts=True,
+            use_arrow_keys=True,
+            instruction="",
+        )
+    )
+
+
+async def ask_text(prompt: str, default: str = "") -> str | None:
+    """Prompt for text input. Ctrl+C to cancel, empty = None."""
+    result = await safe_ask(questionary.text(prompt, default=default, style=APP_STYLE))
+    return result if result else None

ot/config/__init__.py

@@ -0,0 +1,46 @@
+"""Centralized configuration for OneTool.
+
+This module provides a single source of truth for all configuration settings
+via YAML configuration for tool discovery and settings.
+
+Usage:
+    from ot.config import get_config, load_config
+
+    config = get_config()
+    print(config.log_level)
+    print(config.tools_dir)
+
+    # For tools to access their configuration:
+    from ot.config import get_tool_config
+
+    class Config(BaseModel):
+        timeout: float = 60.0
+
+    config = get_tool_config("brave", Config)
+"""
+
+from ot.config.loader import (
+    OneToolConfig,
+    SnippetDef,
+    SnippetParam,
+    get_config,
+    is_log_verbose,
+    load_config,
+)
+from ot.config.mcp import McpServerConfig
+from ot.config.secrets import get_secret, get_secrets, load_secrets
+from ot.config.tool_config import get_tool_config
+
+__all__ = [
+    "McpServerConfig",
+    "OneToolConfig",
+    "SnippetDef",
+    "SnippetParam",
+    "get_config",
+    "get_secret",
+    "get_secrets",
+    "get_tool_config",
+    "is_log_verbose",
+    "load_config",
+    "load_secrets",
+]

ot/config/defaults/bench.yaml

@@ -0,0 +1,4 @@
+# Benchmark Harness configuration
+# Run benchmarks to test OneTool performance and tool behavior
+
+log_level: INFO

ot/config/defaults/diagram-templates/api-flow.mmd

@@ -0,0 +1,33 @@
+%% API Flow Template - Sequence Diagram
+%% Replace placeholders with actual service names and operations
+
+sequenceDiagram
+    participant C as Client
+    participant GW as API Gateway
+    participant S as Service
+    participant DB as Database
+
+    Note over C,DB: Request Flow
+
+    C->>GW: POST /api/resource
+    activate GW
+
+    GW->>GW: Validate token
+    alt Invalid token
+        GW-->>C: 401 Unauthorized
+    else Valid token
+        GW->>S: Forward request
+        activate S
+
+        S->>DB: Query data
+        activate DB
+        DB-->>S: Result set
+        deactivate DB
+
+        S-->>GW: Response payload
+        deactivate S
+
+        GW-->>C: 200 OK + data
+    end
+
+    deactivate GW

ot/config/defaults/diagram-templates/c4-context.puml

@@ -0,0 +1,30 @@
+@startuml C4 Context Diagram
+!include <C4/C4_Context>
+
+title System Context Diagram - [System Name]
+
+' Actors
+Person(user, "User", "A user of the system who needs to accomplish tasks")
+Person(admin, "Administrator", "Manages system configuration and users")
+
+' Core system
+System(system, "My System", "Provides the main functionality that users need")
+
+' External systems
+System_Ext(email, "Email System", "Sends notifications and alerts")
+System_Ext(payment, "Payment Provider", "Handles payment processing")
+System_Ext(identity, "Identity Provider", "Manages authentication via SSO")
+
+' Relationships
+Rel(user, system, "Uses", "HTTPS")
+Rel(admin, system, "Administers", "HTTPS")
+
+Rel(system, email, "Sends emails using", "SMTP")
+Rel(system, payment, "Processes payments via", "HTTPS/REST")
+Rel(system, identity, "Authenticates users with", "OIDC")
+
+' Layout hints
+Lay_D(user, system)
+Lay_R(system, email)
+
+@enduml

ot/config/defaults/diagram-templates/class-diagram.mmd

@@ -0,0 +1,87 @@
+%% Class Diagram Template
+%% Replace with your actual data model
+
+classDiagram
+    class User {
+        +String id
+        +String email
+        +String name
+        +DateTime createdAt
+        +DateTime updatedAt
+        +login() bool
+        +logout() void
+        +updateProfile(data) User
+    }
+
+    class Organisation {
+        +String id
+        +String name
+        +String slug
+        +Plan plan
+        +getMembers() List~User~
+        +addMember(user) void
+        +removeMember(user) void
+    }
+
+    class Project {
+        +String id
+        +String name
+        +String description
+        +Status status
+        +DateTime deadline
+        +getTasks() List~Task~
+        +addTask(task) void
+        +archive() void
+    }
+
+    class Task {
+        +String id
+        +String title
+        +String description
+        +Priority priority
+        +Status status
+        +User assignee
+        +assign(user) void
+        +complete() void
+    }
+
+    class Comment {
+        +String id
+        +String content
+        +User author
+        +DateTime createdAt
+        +edit(content) void
+        +delete() void
+    }
+
+    %% Relationships
+    User "1" --> "*" Organisation : belongs to
+    Organisation "1" --> "*" Project : owns
+    Project "1" --> "*" Task : contains
+    User "1" --> "*" Task : assigned to
+    Task "1" --> "*" Comment : has
+    User "1" --> "*" Comment : writes
+
+    %% Enums
+    class Status {
+        <<enumeration>>
+        DRAFT
+        ACTIVE
+        COMPLETED
+        ARCHIVED
+    }
+
+    class Priority {
+        <<enumeration>>
+        LOW
+        MEDIUM
+        HIGH
+        CRITICAL
+    }
+
+    class Plan {
+        <<enumeration>>
+        FREE
+        PRO
+        ENTERPRISE
+    }

ot/config/defaults/diagram-templates/feature-mindmap.mmd

@@ -0,0 +1,70 @@
+%% Feature Mindmap Template
+%% Replace with your actual feature breakdown
+
+mindmap
+  root((Product))
+    Authentication
+      Login
+        Email/Password
+        Social OAuth
+        SSO/SAML
+      Registration
+        Email verification
+        Onboarding flow
+      Security
+        2FA/MFA
+        Session management
+        Password policies
+    User Management
+      Profiles
+        Avatar upload
+        Preferences
+        Notifications
+      Organisations
+        Team creation
+        Role assignment
+        Invitations
+      Permissions
+        Role-based access
+        Resource sharing
+        Audit logging
+    Core Features
+      Dashboard
+        Analytics widgets
+        Quick actions
+        Recent activity
+      Projects
+        CRUD operations
+        Collaboration
+        Version history
+      Tasks
+        Assignment
+        Due dates
+        Comments
+        Attachments
+    Integrations
+      APIs
+        REST endpoints
+        GraphQL
+        Webhooks
+      Third-party
+        Slack
+        GitHub
+        Jira
+      Import/Export
+        CSV
+        JSON
+        PDF reports
+    Infrastructure
+      Performance
+        Caching
+        CDN
+        Database optimisation
+      Reliability
+        Backups
+        Monitoring
+        Alerting
+      Scalability
+        Load balancing
+        Auto-scaling
+        Multi-region

ot/config/defaults/diagram-templates/microservices.d2

@@ -0,0 +1,81 @@
+# Microservices Architecture Template
+# Replace placeholders with actual service names
+
+direction: right
+
+# External actors
+user: "User" {
+  shape: person
+}
+
+# Frontend layer
+frontend: "Frontend" {
+  style.fill: "#e3f2fd"
+
+  web: "Web App"
+  mobile: "Mobile App"
+}
+
+# API layer
+gateway: "API Gateway" {
+  style.fill: "#fff3e0"
+
+  auth: "Auth"
+  routing: "Router"
+  rate_limit: "Rate Limiter"
+}
+
+# Services layer
+services: "Services" {
+  style.fill: "#e8f5e9"
+
+  user_svc: "User Service"
+  order_svc: "Order Service"
+  notification_svc: "Notification Service"
+}
+
+# Data layer
+data: "Data Stores" {
+  style.fill: "#fce4ec"
+
+  postgres: "PostgreSQL" {
+    shape: cylinder
+  }
+  redis: "Redis" {
+    shape: cylinder
+  }
+  queue: "Message Queue" {
+    shape: queue
+  }
+}
+
+# External services
+external: "External" {
+  style.stroke-dash: 3
+
+  email: "Email Provider"
+  payment: "Payment Gateway"
+}
+
+# Connections
+user -> frontend.web
+user -> frontend.mobile
+
+frontend.web -> gateway.routing
+frontend.mobile -> gateway.routing
+
+gateway.routing -> gateway.auth
+gateway.auth -> services.user_svc
+gateway.routing -> services.order_svc
+gateway.routing -> services.notification_svc
+
+services.user_svc -> data.postgres
+services.order_svc -> data.postgres
+services.order_svc -> data.queue
+
+services.notification_svc -> data.queue
+services.notification_svc -> external.email
+
+services.order_svc -> external.payment
+
+data.redis <- gateway.rate_limit: "Rate limits"

ot/config/defaults/diagram-templates/project-gantt.mmd

@@ -0,0 +1,37 @@
+%% Project Gantt Chart Template
+%% Replace with your actual project timeline
+
+gantt
+    title Project Timeline
+    dateFormat YYYY-MM-DD
+    excludes weekends
+
+    section Planning
+    Requirements gathering     :a1, 2024-01-08, 5d
+    Technical design          :a2, after a1, 5d
+    Architecture review       :milestone, m1, after a2, 0d
+
+    section Development
+    Core infrastructure       :b1, after m1, 10d
+    Feature A implementation  :b2, after b1, 8d
+    Feature B implementation  :b3, after b1, 12d
+    Integration               :b4, after b2, 5d
+    Development complete      :milestone, m2, after b4, 0d
+
+    section Testing
+    Unit testing             :c1, after b2, 5d
+    Integration testing      :c2, after m2, 5d
+    Performance testing      :c3, after c2, 3d
+    UAT                      :c4, after c3, 5d
+    Testing complete         :milestone, m3, after c4, 0d
+
+    section Deployment
+    Staging deployment       :d1, after m3, 2d
+    Production preparation   :d2, after d1, 2d
+    Go-live                  :milestone, m4, after d2, 0d
+    Post-launch support      :d3, after m4, 5d
+
+    section Documentation
+    API documentation        :e1, after b1, 15d
+    User guide               :e2, after c2, 10d
+    Release notes            :e3, after m3, 3d