mirror of
https://github.com/EvolutionAPI/adk-python.git
synced 2025-12-18 11:22:22 -06:00
feat: google/adk-python#479 support for streamable http MCP servers for MCPToolset
Copybara import of the project: -- c5b9d49d7b6d858ff0a93bd690e6d653b7c32221 by Omar BENHAMID <omar.benhamid@smart-gts.com>: feat: google/adk-python#479 support for streamable http MCP servers for MCPToolset -- 9431bc19e6538c1b814aba0b24ff564acf046075 by Omar BENHAMID <omar.benhamid@smart-gts.com>: feat: google/adk-python#479 streamable http added to right package -- 8b4aabed45a6f0dc828beb61f12985dc7b14f3d0 by Omar BENHAMID <omar.benhamid@smart-gts.com>: feat: google/adk-python#479 streamable http : review feedbacks + sample agent COPYBARA_INTEGRATE_REVIEW=https://github.com/google/adk-python/pull/650 from omarbenhamid:feature/mcp-streamable-http 625f028784c216401d45cb1b5d4d998535ebcb00 PiperOrigin-RevId: 764419586
This commit is contained in:
committed by
Copybara-Service
parent
c7ce987676
commit
d232e6216d
8
contributing/samples/mcp_streamablehttp_agent/README.md
Normal file
8
contributing/samples/mcp_streamablehttp_agent/README.md
Normal file
@@ -0,0 +1,8 @@
|
||||
This agent connects to a local MCP server via sse.
|
||||
|
||||
To run this agent, start the local MCP server first by :
|
||||
|
||||
```bash
|
||||
uv run filesystem_server.py
|
||||
```
|
||||
|
||||
15
contributing/samples/mcp_streamablehttp_agent/__init__.py
Normal file
15
contributing/samples/mcp_streamablehttp_agent/__init__.py
Normal file
@@ -0,0 +1,15 @@
|
||||
# Copyright 2025 Google LLC
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from . import agent
|
||||
58
contributing/samples/mcp_streamablehttp_agent/agent.py
Normal file
58
contributing/samples/mcp_streamablehttp_agent/agent.py
Normal file
@@ -0,0 +1,58 @@
|
||||
# Copyright 2025 Google LLC
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
|
||||
import os
|
||||
|
||||
from google.adk.agents.llm_agent import LlmAgent
|
||||
from google.adk.tools.mcp_tool.mcp_session_manager import StreamableHTTPServerParams
|
||||
from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset
|
||||
from google.adk.tools.mcp_tool.mcp_toolset import SseServerParams
|
||||
|
||||
_allowed_path = os.path.dirname(os.path.abspath(__file__))
|
||||
|
||||
root_agent = LlmAgent(
|
||||
model='gemini-2.0-flash',
|
||||
name='enterprise_assistant',
|
||||
instruction=f"""\
|
||||
Help user accessing their file systems.
|
||||
|
||||
Allowed directory: {_allowed_path}
|
||||
""",
|
||||
tools=[
|
||||
MCPToolset(
|
||||
connection_params=StreamableHTTPServerParams(
|
||||
url='http://localhost:3000/mcp',
|
||||
),
|
||||
# don't want agent to do write operation
|
||||
# you can also do below
|
||||
# tool_filter=lambda tool, ctx=None: tool.name
|
||||
# not in [
|
||||
# 'write_file',
|
||||
# 'edit_file',
|
||||
# 'create_directory',
|
||||
# 'move_file',
|
||||
# ],
|
||||
tool_filter=[
|
||||
'read_file',
|
||||
'read_multiple_files',
|
||||
'list_directory',
|
||||
'directory_tree',
|
||||
'search_files',
|
||||
'get_file_info',
|
||||
'list_allowed_directories',
|
||||
],
|
||||
)
|
||||
],
|
||||
)
|
||||
@@ -0,0 +1,80 @@
|
||||
# Copyright 2025 Google LLC
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
import asyncio
|
||||
import os
|
||||
from pathlib import Path
|
||||
import sys
|
||||
from mcp.server.fastmcp import FastMCP
|
||||
|
||||
# Create an MCP server with a name
|
||||
mcp = FastMCP("Filesystem Server", host="localhost", port=3000)
|
||||
|
||||
|
||||
# Add a tool to read file contents
|
||||
@mcp.tool(description="Read contents of a file")
|
||||
def read_file(filepath: str) -> str:
|
||||
"""Read and return the contents of a file."""
|
||||
with open(filepath, "r") as f:
|
||||
return f.read()
|
||||
|
||||
|
||||
# Add a tool to list directory contents
|
||||
@mcp.tool(description="List contents of a directory")
|
||||
def list_directory(dirpath: str) -> list:
|
||||
"""List all files and directories in the given directory."""
|
||||
return os.listdir(dirpath)
|
||||
|
||||
|
||||
# Add a tool to get current working directory
|
||||
@mcp.tool(description="Get current working directory")
|
||||
def get_cwd() -> str:
|
||||
"""Return the current working directory."""
|
||||
return str(Path.cwd())
|
||||
|
||||
|
||||
# Graceful shutdown handler
|
||||
async def shutdown(signal, loop):
|
||||
"""Cleanup tasks tied to the service's shutdown."""
|
||||
print(f"\nReceived exit signal {signal.name}...")
|
||||
|
||||
# Get all running tasks
|
||||
tasks = [t for t in asyncio.all_tasks() if t is not asyncio.current_task()]
|
||||
|
||||
# Cancel all tasks
|
||||
for task in tasks:
|
||||
task.cancel()
|
||||
|
||||
print(f"Cancelling {len(tasks)} outstanding tasks")
|
||||
await asyncio.gather(*tasks, return_exceptions=True)
|
||||
|
||||
# Stop the loop
|
||||
loop.stop()
|
||||
print("Shutdown complete!")
|
||||
|
||||
|
||||
# Main entry point with graceful shutdown handling
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
# The MCP run function ultimately uses asyncio.run() internally
|
||||
mcp.run(transport="streamable-http")
|
||||
except KeyboardInterrupt:
|
||||
print("\nServer shutting down gracefully...")
|
||||
# The asyncio event loop has already been stopped by the KeyboardInterrupt
|
||||
print("Server has been shut down.")
|
||||
except Exception as e:
|
||||
print(f"Unexpected error: {e}")
|
||||
sys.exit(1)
|
||||
finally:
|
||||
print("Thank you for using the Filesystem MCP Server!")
|
||||
Reference in New Issue
Block a user