mirror of
https://github.com/MervinPraison/PraisonAI.git
synced 2026-11-07 12:47:55 -10:00
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
"""
|
|
MCP WebSocket Transport Example
|
|
|
|
Demonstrates connecting to MCP servers via WebSocket (ws:// or wss://).
|
|
WebSocket transport provides bidirectional real-time communication,
|
|
ideal for cloud deployments and long-lived connections.
|
|
|
|
Features:
|
|
- Auto-detection of ws:// and wss:// URLs
|
|
- Authentication token support
|
|
- Auto-reconnect with exponential backoff
|
|
- Ping/pong keepalive
|
|
|
|
Protocol: SEP-1288 (WebSocket Transport)
|
|
"""
|
|
|
|
from praisonaiagents import Agent, MCP
|
|
|
|
# Basic WebSocket connection
|
|
agent = Agent(
|
|
name="WebSocket Assistant",
|
|
instructions="You are a helpful assistant connected via WebSocket.",
|
|
tools=MCP("ws://localhost:8080/mcp")
|
|
)
|
|
|
|
# Secure WebSocket with authentication
|
|
agent_secure = Agent(
|
|
name="Secure WebSocket Assistant",
|
|
instructions="You are a helpful assistant with secure WebSocket connection.",
|
|
tools=MCP(
|
|
"wss://api.example.com/mcp",
|
|
auth_token="Bearer your-secret-token",
|
|
timeout=60
|
|
)
|
|
)
|
|
|
|
# WebSocket with custom options
|
|
agent_custom = Agent(
|
|
name="Custom WebSocket Assistant",
|
|
tools=MCP(
|
|
"wss://api.example.com/mcp",
|
|
auth_token="your-token",
|
|
timeout=120,
|
|
debug=True
|
|
)
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
# Note: Requires a running WebSocket MCP server
|
|
# Example server: https://github.com/modelcontextprotocol/servers
|
|
|
|
print("WebSocket MCP Transport Example")
|
|
print("=" * 40)
|
|
print("This example requires a WebSocket MCP server running.")
|
|
print("URL patterns: ws://host:port/path or wss://host:port/path")
|