FIRE-813 | Gradio UI | MCP Support #110

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged

yuval-qf merged 54 commits into main from feature/FIRE-813-mcp-gradio

Oct 23, 2025

.rogue/user_config.json

-Original file line number
+Diff line change
@@ -1,4 +1,6 @@
     {
+      "protocol": "mcp",
+      "transport": "streamable_http",
       "evaluated_agent_url": "http://localhost:10001/",
       "evaluated_agent_auth_type": "no_auth",
       "evaluated_agent_credentials": "",
@@ Expand Down @@

rogue/ui/app.py

-Original file line number
+Diff line change
@@ -1,7 +1,7 @@
     import json
     from pathlib import Path
-    from rogue_sdk.types import AuthType
+    from rogue_sdk.types import AuthType, Protocol, Transport
     from ..common.workdir_utils import load_config
     from .components.config_screen import create_config_screen
@@ Expand Down Expand Up / @@ -35,6 +35,8 @@ def get_app(workdir: Path, rogue_server_url: str): @@
                 with gr.TabItem("1. Config", id="config"):
                     (
                         agent_url,
+                        protocol,
+                        transport,
                         interview_mode,
                         auth_type,
                         auth_credentials,
@@ Expand Down Expand Up / @@ -115,11 +117,27 @@ def load_and_update_ui(): @@
                     "evaluated_agent_auth_type",
                     AuthType.NO_AUTH.value,
                 )
+                # Safe enum parsing with sensible fallbacks
+                raw_protocol = config.get("protocol", Protocol.A2A.value)
+                raw_transport = config.get("transport", Transport.HTTP.value)
+                try:
+                    protocol_val = Protocol(raw_protocol)
+                except (ValueError, KeyError):
+                    protocol_val = Protocol.A2A
+                try:
+                    transport_val = Transport(raw_transport)
+                except (ValueError, KeyError):
+                    transport_val = Transport.HTTP
+                if not transport_val.is_valid_for_protocol(protocol_val):
+                    transport_val = protocol_val.get_default_transport()
                 return {
                     shared_state: state,
                     agent_url: gr.update(
                         value=config.get("evaluated_agent_url", "http://localhost:10001"),
                     ),
+                    protocol: gr.update(value=protocol_val.value),
+                    transport: gr.update(value=transport_val.value),
                     interview_mode: gr.update(value=config.get("interview_mode", True)),
                     deep_test_mode: gr.update(value=config.get("deep_test_mode", False)),
                     parallel_runs: gr.update(value=config.get("parallel_runs", 1)),
@@ Expand All / @@ -144,6 +162,8 @@ def load_and_update_ui(): @@
                 outputs=[
                     shared_state,
                     agent_url,
+                    protocol,
+                    transport,
                     interview_mode,
                     auth_type,
                     auth_credentials,
@@ Expand Down @@

rogue/ui/components/config_screen.py

-Original file line number
+Diff line change
@@ Expand Up / @@ -2,7 +2,7 @@ @@
     from loguru import logger
     from pydantic import ValidationError
-    from rogue_sdk.types import AgentConfig, AuthType
+    from rogue_sdk.types import PROTOCOL_TO_TRANSPORTS, AgentConfig, AuthType, Protocol
     from ...common.workdir_utils import dump_config
@@ Expand Down Expand Up / @@ -33,6 +33,37 @@ def create_config_screen( @@
             )
             agent_url_error = gr.Markdown(visible=False, elem_classes=["error-label"])
+            gr.Markdown("**Protocol**")
+            protocol = gr.Dropdown(
+                label="Protocol",
+                choices=[p.value for p in Protocol],
+                value=config_data.get(
+                    "protocol",
+                    Protocol.A2A.value,
+                ),
+            )
+            # Get initial transport choices based on current protocol
+            initial_protocol = Protocol(
+                config_data.get("protocol", Protocol.A2A.value),
+            )
+            initial_transport_choices = [
+                t.value for t in PROTOCOL_TO_TRANSPORTS[initial_protocol]
+            ]
+            initial_transport_value = config_data.get(
+                "transport",
+                initial_protocol.get_default_transport().value,
+            )
+            if initial_transport_value not in initial_transport_choices:
+                initial_transport_value = initial_protocol.get_default_transport().value
+            gr.Markdown("**Transport**")
+            transport = gr.Dropdown(
+                label="Transport",
+                choices=initial_transport_choices,
+                value=initial_transport_value,
+            )
             gr.Markdown("**Interview Mode**")
             interview_mode = gr.Checkbox(
                 label="Enable AI-powered business context interview",
@@ Expand Down Expand Up / @@ -153,6 +184,8 @@ def update_state(state, key, value): @@
         for component, key in [
             (agent_url, "agent_url"),
+            (protocol, "protocol"),
+            (transport, "transport"),
             (interview_mode, "interview_mode"),
             (auth_type, "auth_type"),
             (auth_credentials, "auth_credentials"),
@@ Expand All / @@ -179,9 +212,24 @@ def toggle_auth_credentials(auth_t): @@
             outputs=[auth_credentials],
         )
+        def update_transport_choices(protocol_val):
+            """Update transport choices based on selected protocol."""
+            selected_protocol = Protocol(protocol_val)
+            transport_choices = [t.value for t in PROTOCOL_TO_TRANSPORTS[selected_protocol]]
+            default_transport = selected_protocol.get_default_transport().value
+            return gr.update(choices=transport_choices, value=default_transport)
+        protocol.change(
+            fn=update_transport_choices,
+            inputs=[protocol],
+            outputs=[transport],
+        )
         def save_config(
             state,
             url,
+            protocol_val,
+            transport_val,
             interview_mode_val,
             deep_test_mode_val,
             parallel_runs_val,
@@ Expand All / @@ -207,6 +255,8 @@ def save_config( @@
                     judge_llm=llm,
                     judge_llm_api_key=llm_key,
                     deep_test_mode=deep_test_mode_val,
+                    protocol=protocol_val,
+                    transport=transport_val,
                     interview_mode=interview_mode_val,
                     parallel_runs=parallel_runs_val,
                     business_context="",
@@ Expand Down Expand Up / @@ -252,6 +302,8 @@ def save_config( @@
             inputs=[
                 shared_state,
                 agent_url,
+                protocol,
+                transport,
                 interview_mode,
                 deep_test_mode,
                 parallel_runs,
@@ Expand All / @@ -272,6 +324,8 @@ def save_config( @@
         return (
             agent_url,
+            protocol,
+            transport,
             interview_mode,
             auth_type,
             auth_credentials,
@@ Expand Down @@

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

FIRE-813 | Gradio UI | MCP Support #110

Uh oh!

Diff view

Diff view

There are no files selected for viewing

Uh oh!

Uh oh!