Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 9 additions & 0 deletions langchain_mcp_adapters/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
from langchain_core.messages import AIMessage, HumanMessage
from langchain_core.tools import BaseTool
from mcp import ClientSession
from mcp.types import ResourceTemplate

from langchain_mcp_adapters.callbacks import CallbackContext, Callbacks
from langchain_mcp_adapters.interceptors import ToolCallInterceptor
Expand Down Expand Up @@ -222,6 +223,14 @@ async def get_resources(
async with self.session(server_name) as session:
return await load_mcp_resources(session, uris=uris)

async def get_resource_templates(
self, server_name: str, *, cursor: str | None = None
) -> list[ResourceTemplate]:
"""Send a resources/templates/list request."""
async with self.session(server_name) as session:
resource_templates = await session.list_resource_templates(cursor=cursor)
return resource_templates.resourceTemplates

async def __aenter__(self) -> "MultiServerMCPClient":
"""Async context manager entry point.

Expand Down
11 changes: 11 additions & 0 deletions tests/servers/math_server.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,5 +28,16 @@ def configure_assistant(skills: str) -> list[dict]:
]


@mcp.resource("math://randomformulas/{name}")
def get_random_formulas(name: str) -> str:
"""Get random formulas."""
formulas = {
"energy": "E = mc^2",
"area": "A = πr^2",
"volume": "V = lwh",
}
return formulas.get(name, "Unknown formula")


if __name__ == "__main__":
mcp.run(transport="stdio")
36 changes: 36 additions & 0 deletions tests/test_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@

from langchain_core.messages import AIMessage
from langchain_core.tools import BaseTool
from mcp.types import ResourceTemplate

from langchain_mcp_adapters.client import MultiServerMCPClient
from langchain_mcp_adapters.tools import load_mcp_tools
Expand Down Expand Up @@ -166,3 +167,38 @@ async def test_get_prompt():
assert isinstance(messages[0], AIMessage)
assert "You are a helpful assistant" in messages[0].content
assert "math, addition, multiplication" in messages[0].content


async def test_get_resource_templates():
"""Test retrieving resource templates from MCP servers."""
# Get the absolute path to the server scripts
current_dir = Path(__file__).parent
math_server_path = os.path.join(current_dir, "servers/math_server.py")

client = MultiServerMCPClient(
{
"math": {
"command": "python3",
"args": [math_server_path],
"transport": "stdio",
}
},
)
# Test getting resource templates from the math server
resource_templates = await client.get_resource_templates(
"math",
)

# Check that we got multiple resources back
assert len(resource_templates) == 1
assert all(
isinstance(template, ResourceTemplate) for template in resource_templates
)

# Check the first resource template
random_formulas_template = resource_templates[0]
assert random_formulas_template.name == "get_random_formulas"

assert random_formulas_template.title is None
assert random_formulas_template.description == "Get random formulas."
assert random_formulas_template.uriTemplate == "math://randomformulas/{name}"