feat: Get tools from MCP SSE server

pyproject.toml

@@ -16,6 +16,7 @@ dependencies = [
     "gradio[mcp]>=5.32.1",
     "huggingface-hub>=0.32.3",
     "langchain-aws>=0.2.24",
+    "langchain-mcp-adapters>=0.1.1",
     "langgraph>=0.4.7",
 ]
 
@@ -103,7 +104,17 @@ line-length = 88
 
 [tool.ruff.lint]
 select = ["ALL"]
-ignore = ["D100", "D104", "D107", "D401", "EM102", "ERA001", "TRY003"]
+ignore = [
+    "D100",
+    "D104",
+    "D107",
+    "D401",
+    "EM102",
+    "ERA001",
+    "TC002",
+    "TC003",
+    "TRY003",
+]
 
 [tool.ruff.lint.flake8-quotes]
 inline-quotes = "double"
@@ -121,6 +132,9 @@ lines-after-imports = 2
 [tool.ruff.lint.mccabe]
 max-complexity = 18
 
+[tool.ruff.lint.pylint]
+max-args = 7
+
 [tool.ruff.lint.pydocstyle]
 convention = "google"
 

requirements-dev.txt

@@ -41,6 +41,7 @@ jsonpatch==1.33
 jsonpointer==3.0.0
 langchain-aws==0.2.24
 langchain-core==0.3.63
+langchain-mcp-adapters==0.1.1
 langgraph==0.4.7
 langgraph-checkpoint==2.0.26
 langgraph-prebuilt==0.2.2

requirements.txt

@@ -34,6 +34,7 @@ jsonpatch==1.33
 jsonpointer==3.0.0
 langchain-aws==0.2.24
 langchain-core==0.3.63
+langchain-mcp-adapters==0.1.1
 langgraph==0.4.7
 langgraph-checkpoint==2.0.26
 langgraph-prebuilt==0.2.2

tdagent/grchat.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
 
-from collections.abc import Mapping
+from collections.abc import Mapping, Sequence
 from types import MappingProxyType
+from typing import TYPE_CHECKING
 
 import boto3
 import botocore
@@ -9,9 +10,15 @@ import botocore.exceptions
 import gradio as gr
 from langchain_aws import ChatBedrock
 from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
-from langgraph.graph.graph import CompiledGraph
+from langchain_mcp_adapters.client import MultiServerMCPClient
 from langgraph.prebuilt import create_react_agent
 
+from tdagent.grcomponents import MutableCheckBoxGroup, MutableCheckBoxGroupEntry
+
+
+if TYPE_CHECKING:
+    from langgraph.graph.graph import CompiledGraph
+
 
 #### Constants ####
 
@@ -59,7 +66,7 @@ def create_bedrock_llm(
     try:
         bedrock_client = boto3.client("bedrock-runtime", **boto3_config)
         llm = ChatBedrock(
-            model=bedrock_model_id,
+            model_id=bedrock_model_id,
             client=bedrock_client,
             model_kwargs={"temperature": 0.7},
         )
@@ -78,6 +85,7 @@ async def gr_connect_to_bedrock(
     secret_key: str,
     session_token: str,
     region: str,
+    mcp_servers: Sequence[MutableCheckBoxGroupEntry] | None,
 ) -> str:
     """Initialize Bedrock agent."""
     global llm_agent  # noqa: PLW0603
@@ -96,9 +104,33 @@ async def gr_connect_to_bedrock(
     if llm is None:
         return f"❌ Connection failed: {error}"
 
+    # client = MultiServerMCPClient(
+    #     {
+    #         "toolkit": {
+    #             "url": "https://agents-mcp-hackathon-tdagenttools.hf.space/gradio_api/mcp/sse",
+    #             "transport": "sse",
+    #         },
+    #     }
+    # )
+    # tools = await client.get_tools()
+
+    if mcp_servers:
+        client = MultiServerMCPClient(
+            {
+                server.name.replace(" ", "-"): {
+                    "url": server.value,
+                    "transport": "sse",
+                }
+                for server in mcp_servers
+            },
+        )
+        tools = await client.get_tools()
+    else:
+        tools = []
+
     llm_agent = create_react_agent(
         model=llm,
-        tools=[],
+        tools=tools,
         prompt=SYSTEM_MESSAGE,
     )
 
@@ -134,6 +166,18 @@ async def gr_chat_function(  # noqa: D103
 with gr.Blocks() as gr_app:
     gr.Markdown("# 🔐 Secure Bedrock Chatbot")
 
+    ### MCP Servers ###
+    with gr.Accordion():
+        mcp_list = MutableCheckBoxGroup(
+            values=[
+                MutableCheckBoxGroupEntry(
+                    name="TDAgent tools",
+                    value="https://agents-mcp-hackathon-tdagenttools.hf.space/gradio_api/mcp/sse",
+                ),
+            ],
+            label="MCP Servers",
+        )
+
     # Credentials section (collapsible)
     with gr.Accordion("🔑 Bedrock Configuration", open=True):
         gr.Markdown(
@@ -185,6 +229,7 @@ with gr.Blocks() as gr_app:
                 aws_secret_key_textbox,
                 aws_session_token_textbox,
                 aws_region_dropdown,
+                mcp_list.state,
             ],
             outputs=[status_textbox],
         )

uv.lock

@@ -705,6 +705,19 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/5c/71/a748861e6a69ab6ef50ab8e65120422a1f36245c71a0dd0f02de49c208e1/langchain_core-0.3.63-py3-none-any.whl", hash = "sha256:f91db8221b1bc6808f70b2e72fded1a94d50ee3f1dff1636fb5a5a514c64b7f5", size = 438468 },
 ]
 
+[[package]]
+name = "langchain-mcp-adapters"
+version = "0.1.1"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "langchain-core" },
+    { name = "mcp" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/b0/de/62a6ee2c21f74eed961773e75a4e3170f8abc79fd5fd7a1b4e2ea07f4c04/langchain_mcp_adapters-0.1.1.tar.gz", hash = "sha256:e43ddf06e4ce237ad80f5c91d0efa7fd7c845b274a5a07e7e8a7f9d7239331d7", size = 17359 }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/30/68/13405f252b38a8e1bd7ef345907a4e0eda535c2ca36fe4d6821fc7e9f5de/langchain_mcp_adapters-0.1.1-py3-none-any.whl", hash = "sha256:81594b265d824012040ebd24056fbdb5aabf0b46f780e369ed132421e3411e4d", size = 12100 },
+]
+
 [[package]]
 name = "langgraph"
 version = "0.4.7"
@@ -1905,6 +1918,7 @@ dependencies = [
     { name = "gradio", extra = ["mcp"] },
     { name = "huggingface-hub" },
     { name = "langchain-aws" },
+    { name = "langchain-mcp-adapters" },
     { name = "langgraph" },
 ]
 
@@ -1927,6 +1941,7 @@ requires-dist = [
     { name = "gradio", extras = ["mcp"], specifier = ">=5.32.1" },
     { name = "huggingface-hub", specifier = ">=0.32.3" },
     { name = "langchain-aws", specifier = ">=0.2.24" },
+    { name = "langchain-mcp-adapters", specifier = ">=0.1.1" },
     { name = "langgraph", specifier = ">=0.4.7" },
 ]