FIRE-831 | Rogue Server | MCP transport

.github/actions/run-tests/action.yml

@@ -11,7 +11,7 @@ runs:
   steps:
     - name: "Run tests"
       shell: bash
-      run: source .venv/bin/activate && uv run --dev pytest --junitxml=test-results.xml --cov=rogue --cov-report=xml
+      run: source .venv/bin/activate && pytest --junitxml=test-results.xml --cov=rogue --cov-report=xml
 
     - name: "Upload Test Results"
       uses: actions/upload-artifact@v4

.github/workflows/rogue.yml

@@ -25,6 +25,8 @@ jobs:
 
       - name: Install uv
         uses: astral-sh/setup-uv@v5
+        with:
+          prune-cache: false
 
       - name: Setup Python
         uses: actions/setup-python@v5
@@ -49,7 +51,7 @@ jobs:
           echo "🚀 Starting AI agent..."
 
           # Not using uv because it will reinstall the sdk from pypi
-          source .venv/bin/activate && uv run python -m examples.tshirt_store_agent --host 0.0.0.0 --port 10001 &
+          source .venv/bin/activate && python -m examples.tshirt_store_agent --host 0.0.0.0 --port 10001 &
           AGENT_PID=$!
           echo "Agent started with PID: $AGENT_PID"
           trap 'echo "🛑 Stopping agent..."; kill $AGENT_PID' EXIT

.github/workflows/test.yml

@@ -33,11 +33,13 @@ jobs:
       - name: Create venv
         run: uv venv
 
-      - name: Install rogue sdk
-        run: source .venv/bin/activate && uv pip install -e sdks/python --force-reinstall
-
-      - name: Install rogue server
-        run: source .venv/bin/activate && uv sync --dev
+      - name: Install rogue server with local sdk
+        run: |
+          source .venv/bin/activate
+          uv sync --dev
+          uv pip install -e .
+          uv pip uninstall rogue-ai-sdk
+          uv pip install -e sdks/python --force-reinstall
 
       - name: Run tests
         uses: ./.github/actions/run-tests

.vscode/launch.json

@@ -58,16 +58,16 @@
             "type": "debugpy",
             "request": "launch",
             "python": "./.venv/bin/python",
-            "program": "./examples/tshirt_store_agent",
+            "module": "examples.tshirt_store_agent",
             "envFile": "${workspaceFolder}/examples/tshirt_store_agent/.env"
         },
         {
             "name": "shirtify mcp",
             "type": "debugpy",
             "request": "launch",
             "python": "./.venv/bin/python",
-            "program": "./examples/mcp/tshirt_store_mcp",
-            "envFile": "${workspaceFolder}/examples/mcp/tshirt_store_mcp/.env"
+            "module": "examples.mcp.tshirt_store_langgraph_mcp",
+            "envFile": "${workspaceFolder}/examples/mcp/tshirt_store_langgraph_mcp/.env"
         }
     ]
 }

examples/mcp/tshirt_store_langgraph_mcp/__main__.py

@@ -1,18 +1,31 @@
+from typing import Literal
+
+import click
 from dotenv import load_dotenv
-from mcp_agent_wrapper import mcp
+
+from .mcp_agent_wrapper import get_mcp_server
 
 load_dotenv()
 
 
-def main() -> None:
+@click.command()
+@click.option("--host", "host", default="127.0.0.1", help="Host to run the server on")
+@click.option("--port", "port", default=10001, help="Port to run the server on")
+@click.option(
+    "--transport",
+    "transport",
+    default="streamable-http",
+    choices=["streamable-http", "sse"],
+    help="Transport to use for the mcp server",
+)
+def main(host: str, port: int, transport: Literal["streamable-http", "sse"]) -> None:
     print("Starting MCP server...")
+    mcp = get_mcp_server(host=host, port=port)
 
-    # Can also be "sse".
     # When using "sse", the url will be http://localhost:10001/sse
     # When using "streamable-http", the url will be http://localhost:10001/mcp
     # stdio isn't supported in this example, since rogue won't be able to connect to it.
-    mcp.run(transport="streamable-http")
-    # mcp.run(transport="sse")
+    mcp.run(transport=transport)
 
 
 if __name__ == "__main__":

examples/mcp/tshirt_store_langgraph_mcp/mcp_agent_wrapper.py

@@ -6,38 +6,45 @@
 you only need to implement the send_message tool.
 """
 
+from functools import lru_cache
+
 from loguru import logger
 from mcp.server.fastmcp import Context, FastMCP
-from shirtify_agent import ShirtifyAgent
 from starlette.requests import Request
 
-agent = ShirtifyAgent()
-mcp = FastMCP(
-    "shirtify_agent_mcp",
-    port=10001,
-    host="127.0.0.1",
-)
+from .shirtify_agent import ShirtifyAgent
+
+
+@lru_cache(maxsize=1)
+def get_mcp_server(host: str = "127.0.0.1", port: int = 10001) -> FastMCP:
+    agent = ShirtifyAgent()
+    mcp = FastMCP(
+        "shirtify_agent_mcp",
+        host=host,
+        port=port,
+    )
 
+    @mcp.tool()
+    def send_message(message: str, context: Context) -> str:
+        session_id: str | None = None
+        try:
+            request: Request = context.request_context.request  # type: ignore
 
-@mcp.tool()
-def send_message(message: str, context: Context) -> str:
-    session_id: str | None = None
-    try:
-        request: Request = context.request_context.request  # type: ignore
+            # The session id should be in the headers for streamable-http transport
+            session_id = request.headers.get("mcp-session-id")
 
-        # The session id should be in the headers for streamable-http transport
-        session_id = request.headers.get("mcp-session-id")
+            # The session id might also be in query param when using sse transport
+            if session_id is None:
+                session_id = request.query_params.get("session_id")
+        except Exception:
+            session_id = None
+            logger.exception("Error while extracting session id")
 
-        # The session id might also be in query param when using sse transport
         if session_id is None:
-            session_id = request.query_params.get("session_id")
-    except Exception:
-        session_id = None
-        logger.exception("Error while extracting session id")
+            logger.error("Couldn't extract session id")
 
-    if session_id is None:
-        logger.error("Couldn't extract session id")
+        # Invoking our agent
+        response = agent.invoke(message, session_id)
+        return response.get("content", "")
 
-    # Invoking our agent
-    response = agent.invoke(message, session_id)
-    return response.get("content", "")
+    return mcp

lefthook.yaml

@@ -30,7 +30,7 @@ pre-commit:
 
     - name: isort
       glob: "*.py"
-      run: isort {staged_files}
+      run: isort --profile black {staged_files}
 
     - name: black
       glob: "*.py"

pyproject.toml

@@ -10,6 +10,7 @@ dependencies = [
     "click>=8.0.0",
     "datasets==3.6.0",
     "fastapi>=0.115.0",
+    "fastmcp>=2.12.5",
     "google-adk==1.5.0",
     "gradio==5.35.0",
     "langchain-openai>=0.3.35",

rogue/__main__.py

@@ -44,8 +44,9 @@ def common_parser() -> ArgumentParser:
     parent_parser.add_argument(
         "--example",
         type=str,
-        choices=["tshirt_store"],
-        help="Run with an example agent (e.g., tshirt_store)",
+        choices=["tshirt_store", "tshirt_store_langgraph_mcp"],
+        help="Run with an example agent "
+        "(e.g., tshirt_store, tshirt_store_langgraph_mcp)",
     )
     parent_parser.add_argument(
         "--example-host",
@@ -126,6 +127,19 @@ def start_example_agent(
             "--port",
             str(port),
         ]
+    elif example_name == "tshirt_store_langgraph_mcp":
+        # Use subprocess to run the example agent
+        cmd = [
+            sys.executable,
+            "-m",
+            "examples.mcp.tshirt_store_langgraph_mcp",
+            "--host",
+            host,
+            "--port",
+            str(port),
+            "--transport",
+            "streamable-http",
+        ]
     else:
         logger.error(f"Unknown example: {example_name}")
         return None

rogue/evaluator_agent/__init__.py

@@ -1 +1,21 @@
-from . import evaluator_agent, policy_evaluation, run_evaluator_agent
+from . import (
+    a2a,
+    base_evaluator_agent,
+    evaluator_agent_factory,
+    mcp,
+    policy_evaluation,
+    run_evaluator_agent,
+)
+from .a2a import A2AEvaluatorAgent
+from .mcp import MCPEvaluatorAgent
+
+__all__ = [
+    "base_evaluator_agent",
+    "evaluator_agent_factory",
+    "policy_evaluation",
+    "run_evaluator_agent",
+    "a2a",
+    "mcp",
+    "A2AEvaluatorAgent",
+    "MCPEvaluatorAgent",
+]

rogue/evaluator_agent/a2a/__init__.py

@@ -0,0 +1,6 @@
+from . import a2a_evaluator_agent
+from .a2a_evaluator_agent import A2AEvaluatorAgent
+
+__all__ = [
+    "A2AEvaluatorAgent",
+]