codebase-retrieval-context-engine 2.0.2__tar.gz → 2.0.4__tar.gz

{codebase_retrieval_context_engine-2.0.2 → codebase_retrieval_context_engine-2.0.4}/.env.example

@@ -1,104 +1,121 @@
-# ============================================================
-# Corbell — Environment Variables
-# Copy this file to .env and fill in the values you need.
-# ============================================================
-
-# ----------------------------------------------------------
-# LLM Provider API Keys (pick one or more)
-# ----------------------------------------------------------
-ANTHROPIC_API_KEY=
-OPENAI_API_KEY=
-GOOGLE_API_KEY=
-# Multiple Google keys for round-robin: GOOGLE_API_KEY=key1,key2,key3
-VOYAGE_API_KEY=
-# Multiple Voyage keys for round-robin: VOYAGE_API_KEY=pa-key1,pa-key2,pa-key3
-
-# AWS Bedrock
-BEDROCK_API_KEY=
-AWS_ACCESS_KEY_ID=
-AWS_SECRET_ACCESS_KEY=
-AWS_REGION=us-east-1
-
-# Azure OpenAI
-AZURE_OPENAI_API_KEY=
-AZURE_OPENAI_ENDPOINT=
-AZURE_OPENAI_DEPLOYMENT=
-AZURE_OPENAI_API_VERSION=2024-02-01
-
-# GCP Vertex AI
-GOOGLE_APPLICATION_CREDENTIALS=
-GCP_PROJECT=
-GOOGLE_CLOUD_PROJECT=
-GCP_REGION=us-central1
-
-# Generic fallback key (used when provider-specific key is not set)
-CORBELL_LLM_API_KEY=
-
-# ----------------------------------------------------------
-# Model Overrides
-# ----------------------------------------------------------
-
-# Embedding model — cloud only (default: voyage-code-3)
-# Supported: voyage-code-3, voyage-4-lite (VOYAGE_API_KEY), gemini-embedding-001 (GOOGLE_API_KEY)
-CORBELL_EMBEDDING_MODEL=
-
-# Embedding dimension override — overrides the model's default dimension.
-# Voyage default: 1024, Google (gemini-embedding-001) default: 768.
-CORBELL_EMBEDDING_DIM=
-
-# LLM model — generic (overrides llm.model default)
-CORBELL_LLM_MODEL=
-
-# LLM model — provider-specific (takes priority over CORBELL_LLM_MODEL)
-ANTHROPIC_MODEL=
-OPENAI_MODEL=
-GOOGLE_MODEL=
-OLLAMA_MODEL=
-AWS_MODEL=
-AZURE_MODEL=
-GCP_MODEL=
-
-# ----------------------------------------------------------
-# Workspace Path
-# ----------------------------------------------------------
-
-# Full path to the workspace (repository) root directory.
-# Used as fallback when --workspace-full-path is not passed to CLI.
-# Resolution order: --workspace-full-path flag → CORBELL_WORKSPACE → cwd
-CORBELL_WORKSPACE=
-
-# ----------------------------------------------------------
-# Query Configuration (all optional, sensible defaults shown)
-# ----------------------------------------------------------
-
-# Maximum number of code chunks returned per query (default: 50)
-CORBELL_TOP_K=
-
-# Enable LLM reranking of results (default: true)
-CORBELL_RERANK=
-
-# BFS depth for call-graph expansion (default: 2)
-CORBELL_EXPAND_CALL_DEPTH=
-
-# Maximum chunks added via graph expansion (default: 30)
-CORBELL_EXPAND_MAX_CHUNKS=
-
-# ----------------------------------------------------------
-# Indexing Configuration (all optional, sensible defaults shown)
-# ----------------------------------------------------------
-
-# Number of lines per embedding chunk (default: 50)
-CORBELL_CHUNK_SIZE=
-
-# Overlap between consecutive chunks in lines (default: 10)
-CORBELL_CHUNK_OVERLAP=
-
-# Maximum file size to index in bytes (default: 1048576 = 1 MB)
-CORBELL_MAX_FILE_BYTES=
-
-# Comma-separated list of directory names to skip during indexing
-# (in addition to built-in skip list: .git, node_modules, __pycache__, etc.)
-CORBELL_SKIP_DIRS=
-
-# Number of parallel workers for indexing (default: min(cpu_count, 8))
-# CORBELL_INDEX_WORKERS=4
+# ============================================================
+# Corbell — Environment Variables
+# Copy this file to .env and fill in the values you need.
+# ============================================================
+
+# ----------------------------------------------------------
+# LLM Provider API Keys (pick one or more)
+# ----------------------------------------------------------
+ANTHROPIC_API_KEY=
+OPENAI_API_KEY=
+GOOGLE_API_KEY=
+# Multiple Google keys for round-robin: GOOGLE_API_KEY=key1,key2,key3
+VOYAGE_API_KEY=
+# Multiple Voyage keys for round-robin: VOYAGE_API_KEY=pa-key1,pa-key2,pa-key3
+
+# AWS Bedrock
+BEDROCK_API_KEY=
+AWS_ACCESS_KEY_ID=
+AWS_SECRET_ACCESS_KEY=
+AWS_REGION=us-east-1
+
+# Azure OpenAI
+AZURE_OPENAI_API_KEY=
+AZURE_OPENAI_ENDPOINT=
+AZURE_OPENAI_DEPLOYMENT=
+AZURE_OPENAI_API_VERSION=2024-02-01
+
+# GCP Vertex AI
+GOOGLE_APPLICATION_CREDENTIALS=
+GCP_PROJECT=
+GOOGLE_CLOUD_PROJECT=
+GCP_REGION=us-central1
+
+# Generic fallback key (used when provider-specific key is not set)
+CORBELL_LLM_API_KEY=
+
+# ----------------------------------------------------------
+# Model Overrides
+# ----------------------------------------------------------
+
+# Embedding model — cloud only (default: voyage-code-3)
+# Supported: voyage-code-3, voyage-4-lite (VOYAGE_API_KEY), gemini-embedding-001 (GOOGLE_API_KEY)
+CORBELL_EMBEDDING_MODEL=
+
+# Embedding dimension override — overrides the model's default dimension.
+# Voyage default: 1024, Google (gemini-embedding-001) default: 768.
+CORBELL_EMBEDDING_DIM=
+
+# LLM model — generic (overrides llm.model default)
+CORBELL_LLM_MODEL=
+
+# LLM provider — which API to use for reranking
+# Options: anthropic, openai, google, ollama, aws, azure, gcp (default: anthropic)
+CORBELL_LLM_PROVIDER=
+
+# LLM model — provider-specific (takes priority over CORBELL_LLM_MODEL)
+ANTHROPIC_MODEL=
+OPENAI_MODEL=
+GOOGLE_MODEL=
+OLLAMA_MODEL=
+AWS_MODEL=
+AZURE_MODEL=
+GCP_MODEL=
+
+# ----------------------------------------------------------
+# Workspace Path
+# ----------------------------------------------------------
+
+# Full path to the workspace (repository) root directory.
+# Used as fallback when --workspace-full-path is not passed to CLI.
+# Resolution order: --workspace-full-path flag → CORBELL_WORKSPACE → cwd
+CORBELL_WORKSPACE=
+
+# ----------------------------------------------------------
+# Query Configuration (all optional, sensible defaults shown)
+# ----------------------------------------------------------
+
+# Maximum number of code chunks returned per query (default: 50)
+CORBELL_TOP_K=
+
+# Enable LLM reranking of results (default: true)
+CORBELL_RERANK=
+
+# BFS depth for call-graph expansion (default: 2)
+CORBELL_EXPAND_CALL_DEPTH=
+
+# Maximum chunks added via graph expansion (default: 30)
+CORBELL_EXPAND_MAX_CHUNKS=
+
+# ----------------------------------------------------------
+# Indexing Configuration (all optional, sensible defaults shown)
+# ----------------------------------------------------------
+
+# Number of lines per embedding chunk (default: 50)
+CORBELL_CHUNK_SIZE=
+
+# Overlap between consecutive chunks in lines (default: 10)
+CORBELL_CHUNK_OVERLAP=
+
+# Maximum file size to index in bytes (default: 1048576 = 1 MB)
+CORBELL_MAX_FILE_BYTES=
+
+# Comma-separated list of directory names to skip during indexing
+# (in addition to built-in skip list: .git, node_modules, __pycache__, etc.)
+CORBELL_SKIP_DIRS=
+
+# Number of parallel workers for indexing (default: min(cpu_count, 8))
+# CORBELL_INDEX_WORKERS=4
+
+# Number of concurrent embedding API threads per encode call (default: provider-aware)
+# VoyageEmbeddingModel default: 30, GoogleEmbeddingModel default: 8, other: 4
+CORBELL_EMBED_CONCURRENCY=
+
+# Super-batch size: number of chunks encoded and written per streaming slice (default: 6000)
+# Sized to saturate the concurrent thread pool. Lower values reduce peak memory
+# at the cost of throughput; higher values may exceed API rate limits.
+CORBELL_EMBED_BATCH=
+
+# Enable verbose performance logging during index builds (any non-empty value)
+# Also available as --verbose / -v flag on `corbell index build`
+CORBELL_VERBOSE=

codebase_retrieval_context_engine-2.0.4/PKG-INFO

@@ -0,0 +1,95 @@
+Metadata-Version: 2.4
+Name: codebase-retrieval-context-engine
+Version: 2.0.4
+Summary: Code retrieval engine — hybrid embedding + graph search for LLM context injection.
+Project-URL: Homepage, https://github.com/nullmastermind/local-context-engine
+Project-URL: Repository, https://github.com/nullmastermind/local-context-engine
+Project-URL: Issues, https://github.com/nullmastermind/local-context-engine/issues
+Author: nullmastermind
+License: Apache-2.0
+License-File: LICENSE
+Keywords: LLM,MCP,code-search,codebase-retrieval,context,embeddings,retrieval
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries
+Requires-Python: >=3.11
+Requires-Dist: google-genai>=2.7.0
+Requires-Dist: mcp>=1.1.2
+Requires-Dist: numpy>=2.0
+Requires-Dist: pathspec>=0.11
+Requires-Dist: pydantic>=2.0
+Requires-Dist: python-dotenv>=1.0
+Requires-Dist: rich>=13.0
+Requires-Dist: typer>=0.12
+Requires-Dist: voyageai>=0.3
+Provides-Extra: anthropic
+Requires-Dist: anthropic>=0.25; extra == 'anthropic'
+Provides-Extra: aws
+Requires-Dist: boto3>=1.34; extra == 'aws'
+Provides-Extra: azure
+Requires-Dist: openai>=1.0; extra == 'azure'
+Provides-Extra: debug
+Requires-Dist: gradio>=4.0; extra == 'debug'
+Provides-Extra: dev
+Requires-Dist: httpx; extra == 'dev'
+Requires-Dist: mypy; extra == 'dev'
+Requires-Dist: pytest-asyncio; extra == 'dev'
+Requires-Dist: pytest-cov; extra == 'dev'
+Requires-Dist: pytest>=8.0; extra == 'dev'
+Requires-Dist: respx; extra == 'dev'
+Requires-Dist: ruff; extra == 'dev'
+Provides-Extra: gcp
+Requires-Dist: anthropic[vertex]>=0.25; extra == 'gcp'
+Requires-Dist: google-cloud-aiplatform>=1.38; extra == 'gcp'
+Provides-Extra: openai
+Requires-Dist: openai>=1.0; extra == 'openai'
+Provides-Extra: treesitter
+Requires-Dist: tree-sitter-c-sharp>=0.21; extra == 'treesitter'
+Requires-Dist: tree-sitter-go>=0.21; extra == 'treesitter'
+Requires-Dist: tree-sitter-java>=0.21; extra == 'treesitter'
+Requires-Dist: tree-sitter-javascript>=0.21; extra == 'treesitter'
+Requires-Dist: tree-sitter-php>=0.21; extra == 'treesitter'
+Requires-Dist: tree-sitter-python>=0.21; extra == 'treesitter'
+Requires-Dist: tree-sitter-ruby>=0.21; extra == 'treesitter'
+Requires-Dist: tree-sitter-rust>=0.21; extra == 'treesitter'
+Requires-Dist: tree-sitter-typescript>=0.21; extra == 'treesitter'
+Requires-Dist: tree-sitter>=0.21; extra == 'treesitter'
+Description-Content-Type: text/markdown
+
+<div align="center">
+  <h1>codebase-retrieval-context-engine</h1>
+  <p><strong>Code retrieval engine for LLM context via MCP.</strong></p>
+  <p>
+    <a href="LICENSE"><img src="https://img.shields.io/badge/License-Apache_2.0-blue.svg" alt="License"/></a>
+  </p>
+</div>
+
+---
+
+## Add to Claude Code
+
+```bash
+claude mcp add codebase-retrieval -e CORBELL_LLM_PROVIDER=google -e GOOGLE_API_KEY=your-google-api-key -e GOOGLE_MODEL=gemini-3.1-flash-lite -e CORBELL_EMBEDDING_MODEL=voyage-4-lite -e VOYAGE_API_KEY=your-voyage-api-key -- uvx codebase-retrieval-context-engine
+```
+
+That's it. The AI agent passes workspace path and triggers index builds automatically.
+
+---
+
+## Environment variables
+
+| Variable | Description |
+|---|---|
+| `CORBELL_LLM_PROVIDER` | LLM provider for reranking (`google`, `anthropic`, `openai`) |
+| `GOOGLE_API_KEY` | Google AI API key (supports multiple: `key1,key2,key3`) |
+| `GOOGLE_MODEL` | e.g. `gemini-3.1-flash-lite` |
+| `CORBELL_EMBEDDING_MODEL` | `voyage-4-lite`, `voyage-code-3`, or `gemini-embedding-001` |
+| `VOYAGE_API_KEY` | Voyage AI API key (supports multiple: `key1,key2,key3`). Add a card to billing to unlock rate limits. |
+
+---
+
+## License
+
+Apache 2.0

codebase_retrieval_context_engine-2.0.4/README.md

@@ -0,0 +1,35 @@
+<div align="center">
+  <h1>codebase-retrieval-context-engine</h1>
+  <p><strong>Code retrieval engine for LLM context via MCP.</strong></p>
+  <p>
+    <a href="LICENSE"><img src="https://img.shields.io/badge/License-Apache_2.0-blue.svg" alt="License"/></a>
+  </p>
+</div>
+
+---
+
+## Add to Claude Code
+
+```bash
+claude mcp add codebase-retrieval -e CORBELL_LLM_PROVIDER=google -e GOOGLE_API_KEY=your-google-api-key -e GOOGLE_MODEL=gemini-3.1-flash-lite -e CORBELL_EMBEDDING_MODEL=voyage-4-lite -e VOYAGE_API_KEY=your-voyage-api-key -- uvx codebase-retrieval-context-engine
+```
+
+That's it. The AI agent passes workspace path and triggers index builds automatically.
+
+---
+
+## Environment variables
+
+| Variable | Description |
+|---|---|
+| `CORBELL_LLM_PROVIDER` | LLM provider for reranking (`google`, `anthropic`, `openai`) |
+| `GOOGLE_API_KEY` | Google AI API key (supports multiple: `key1,key2,key3`) |
+| `GOOGLE_MODEL` | e.g. `gemini-3.1-flash-lite` |
+| `CORBELL_EMBEDDING_MODEL` | `voyage-4-lite`, `voyage-code-3`, or `gemini-embedding-001` |
+| `VOYAGE_API_KEY` | Voyage AI API key (supports multiple: `key1,key2,key3`). Add a card to billing to unlock rate limits. |
+
+---
+
+## License
+
+Apache 2.0

{codebase_retrieval_context_engine-2.0.2 → codebase_retrieval_context_engine-2.0.4}/corbell/__init__.py

@@ -2,5 +2,5 @@
 Corbell — Code retrieval engine for LLM context injection.
 """
 
-__version__ = "2.0.2"
+__version__ = "2.0.4"
 __all__ = ["__version__"]

codebase_retrieval_context_engine-2.0.4/corbell/cli/commands/debug.py

@@ -0,0 +1,305 @@
+"""CLI: corbell debug — launch a Gradio UI for inspecting query pipeline internals."""
+
+from __future__ import annotations
+
+import os
+from pathlib import Path
+
+import typer
+from rich.console import Console
+
+app = typer.Typer(no_args_is_help=False, help="Query debug UI commands.")
+console = Console()
+
+
+@app.callback(invoke_without_command=True)
+def debug(
+    ctx: typer.Context,
+    workspace: str = typer.Option(
+        "",
+        "--workspace",
+        "-w",
+        help="Path to the workspace root (default: current directory).",
+    ),
+    port: int = typer.Option(7860, "--port", "-p", help="Port for the Gradio server."),
+    share: bool = typer.Option(False, "--share", help="Create a public Gradio share link."),
+) -> None:
+    """Launch the Gradio debug UI for inspecting the query pipeline.
+
+    The UI lets you run a query against a workspace and inspect:
+      - Per-phase timing
+      - Final formatted results
+      - Pre-rerank chunk table (file, lines, score, symbol, type)
+      - LLM rerank prompts and raw response
+    """
+    if ctx.invoked_subcommand is not None:
+        return
+
+    try:
+        import gradio as gr  # type: ignore[import-untyped]
+    except ImportError:
+        console.print(
+            "[red]Gradio is not installed. Install it with:[/red]\n"
+            "  pip install 'codebase-retrieval-context-engine[debug]'"
+        )
+        raise typer.Exit(1)
+
+    default_workspace = workspace or os.environ.get("CORBELL_WORKSPACE") or str(Path.cwd())
+
+    def run_mcp_tool(
+        env_vars_text: str,
+        mcp_workspace: str,
+        mcp_query: str,
+    ):  # type: ignore[no-untyped-def]
+        """Invoke context_engine_codebase_retrieval directly and return results."""
+        if not mcp_query.strip():
+            return "", ""
+
+        # Apply env var overrides for this invocation
+        env_backup: dict[str, str | None] = {}
+        if env_vars_text.strip():
+            for line in env_vars_text.strip().splitlines():
+                line = line.strip()
+                if not line or line.startswith("#"):
+                    continue
+                if "=" not in line:
+                    continue
+                key, _, value = line.partition("=")
+                key, value = key.strip(), value.strip()
+                env_backup[key] = os.environ.get(key)
+                os.environ[key] = value
+
+        try:
+            from corbell.core.mcp.server import context_engine_codebase_retrieval
+
+            result = context_engine_codebase_retrieval(
+                query=mcp_query.strip(),
+                workspace_full_path=mcp_workspace.strip(),
+            )
+
+            if result.startswith("Error:"):
+                return result, ""
+            return "", result
+        except Exception as exc:
+            return f"Error: {exc}", ""
+        finally:
+            for key, original in env_backup.items():
+                if original is None:
+                    os.environ.pop(key, None)
+                else:
+                    os.environ[key] = original
+
+    def run_query(workspace_path: str, query: str):  # type: ignore[no-untyped-def]
+        """Run the debug pipeline and return Gradio component values."""
+        from corbell.core.query.engine import codebase_retrieval_debug
+
+        if not query.strip():
+            return (
+                "",              # error_box
+                "",              # timing_md
+                "",              # final_results
+                [],              # pre_rerank_table
+                "",              # rerank_system
+                "",              # rerank_user
+                "",              # rerank_response
+            )
+
+        ws = workspace_path.strip() or default_workspace
+        result = codebase_retrieval_debug(query=query, workspace_path=ws)
+
+        # --- Error banner ---
+        error_text = result.error or ""
+
+        # --- Timing table ---
+        timing = result.diagnostics.timing if result.diagnostics else {}
+        if timing:
+            rows = "".join(
+                f"| {phase} | {elapsed:.3f}s |\n"
+                for phase, elapsed in timing.items()
+            )
+            timing_md = (
+                "| Phase | Elapsed |\n"
+                "|---|---|\n"
+                + rows
+            )
+        else:
+            timing_md = "_No timing data available._"
+
+        # --- Final results ---
+        final_results = result.final_output or ""
+
+        # --- Pre-rerank table ---
+        pre_rerank_rows = []
+        graph_ids = set()
+        if result.diagnostics and result.diagnostics.graph_chunk_ids:
+            graph_ids = result.diagnostics.graph_chunk_ids
+        for chunk in result.pre_rerank_chunks:
+            chunk_id = getattr(chunk, "chunk_id", "")
+            parts = chunk_id.split("+") if chunk_id else []
+            has_graph = any(p in graph_ids for p in parts) if graph_ids else False
+            has_embedding = any(p not in graph_ids for p in parts) if graph_ids else True
+            if has_graph and has_embedding and len(parts) > 1:
+                source = "embedding+graph"
+            elif has_graph:
+                source = "graph"
+            else:
+                source = "embedding"
+            pre_rerank_rows.append([
+                getattr(chunk, "file_path", ""),
+                f"{getattr(chunk, 'start_line', '')}-{getattr(chunk, 'end_line', '')}",
+                f"{getattr(chunk, 'score', 0.0):.4f}",
+                getattr(chunk, "symbol", "") or "",
+                getattr(chunk, "chunk_type", "") or "",
+                source,
+                getattr(chunk, "content", "") or "",
+            ])
+
+        # --- Rerank prompts ---
+        detail = result.rerank_detail
+        if detail is None or not detail.system_prompt:
+            rerank_system = "_LLM not configured — reranking skipped_"
+            rerank_user = ""
+            rerank_response = ""
+        else:
+            rerank_system = detail.system_prompt
+            rerank_user = detail.user_prompt
+            rerank_response = detail.raw_response or "_No response (LLM call failed)_"
+
+        return (
+            error_text,
+            timing_md,
+            final_results,
+            pre_rerank_rows,
+            rerank_system,
+            rerank_user,
+            rerank_response,
+        )
+
+    with gr.Blocks(title="Corbell Query Debugger") as demo:
+        gr.Markdown("# Corbell Query Debugger")
+        gr.Markdown("Inspect query pipeline internals: timing, pre-rerank chunks, and LLM rerank prompts.")
+
+        with gr.Row():
+            workspace_input = gr.Textbox(
+                label="Workspace Path",
+                value=default_workspace,
+                placeholder="Path to repository root",
+                scale=2,
+            )
+            query_input = gr.Textbox(
+                label="Query",
+                placeholder="e.g. authentication middleware",
+                scale=3,
+            )
+
+        run_btn = gr.Button("Run Query", variant="primary")
+
+        error_box = gr.Textbox(
+            label="Error",
+            visible=True,
+            interactive=False,
+            lines=2,
+        )
+
+        timing_md = gr.Markdown(label="Timing")
+
+        with gr.Tabs():
+            with gr.Tab("Final Results"):
+                final_output = gr.Code(label="Formatted Output", language=None)
+
+            with gr.Tab("Pre-Rerank Chunks"):
+                pre_rerank_table = gr.Dataframe(
+                    headers=["File", "Lines", "Score", "Symbol", "Type", "Source", "Content"],
+                    datatype=["str", "str", "str", "str", "str", "str", "str"],
+                    label="Chunks before reranking",
+                    wrap=False,
+                )
+
+            with gr.Tab("LLM Rerank"):
+                rerank_system_box = gr.Textbox(
+                    label="System Prompt",
+                    lines=6,
+                    interactive=False,
+                )
+                rerank_user_box = gr.Textbox(
+                    label="User Prompt",
+                    lines=12,
+                    interactive=False,
+                )
+                rerank_response_box = gr.Textbox(
+                    label="Raw LLM Response",
+                    lines=4,
+                    interactive=False,
+                )
+
+            with gr.Tab("MCP Debug"):
+                gr.Markdown(
+                    "### MCP Tool Tester\n"
+                    "Configure environment and invoke "
+                    "`context_engine_codebase_retrieval` directly."
+                )
+
+                with gr.Accordion("Environment Configuration", open=False):
+                    mcp_env_vars = gr.Textbox(
+                        label="Environment Variables (one per line, KEY=VALUE)",
+                        placeholder=(
+                            "# Example:\n"
+                            "CORBELL_LLM_PROVIDER=anthropic\n"
+                            "CORBELL_RERANK=true\n"
+                            "ANTHROPIC_API_KEY=sk-..."
+                        ),
+                        lines=6,
+                    )
+
+                gr.Markdown("#### Tool Parameters")
+                with gr.Row():
+                    mcp_workspace_input = gr.Textbox(
+                        label="workspace_full_path",
+                        value=default_workspace,
+                        placeholder="Path to repository root",
+                        scale=3,
+                    )
+                mcp_query_input = gr.Textbox(
+                    label="query",
+                    placeholder="e.g. authentication middleware",
+                )
+
+                mcp_run_btn = gr.Button("Invoke MCP Tool", variant="primary")
+
+                mcp_error_box = gr.Textbox(
+                    label="Error",
+                    visible=True,
+                    interactive=False,
+                    lines=2,
+                )
+                mcp_result_box = gr.Code(
+                    label="Tool Response",
+                    language=None,
+                )
+
+        run_btn.click(
+            fn=run_query,
+            inputs=[workspace_input, query_input],
+            outputs=[
+                error_box,
+                timing_md,
+                final_output,
+                pre_rerank_table,
+                rerank_system_box,
+                rerank_user_box,
+                rerank_response_box,
+            ],
+        )
+
+        mcp_run_btn.click(
+            fn=run_mcp_tool,
+            inputs=[
+                mcp_env_vars,
+                mcp_workspace_input,
+                mcp_query_input,
+            ],
+            outputs=[mcp_error_box, mcp_result_box],
+        )
+
+    console.print(f"[green]Starting Corbell debug UI on port {port}...[/green]")
+    demo.launch(server_port=port, share=share)