wehub-resource-sync
34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
import asyncio
|
|
from mcp_use import MCPAgent, MCPClient
|
|
from langchain_openai import ChatOpenAI
|
|
|
|
async def main():
|
|
client = MCPClient.from_config_file("mcp-config.json")
|
|
|
|
# Create sessions first
|
|
await client.create_all_sessions()
|
|
|
|
# Discover available tools from MCP server
|
|
tools = set()
|
|
for name in client.get_server_names():
|
|
session = client.get_session(name)
|
|
for t in await session.list_tools():
|
|
tools.add(str(getattr(t, "name", None) or getattr(t, "tool", None) or "unknown"))
|
|
|
|
print("Available MCP tools:", sorted(tools))
|
|
|
|
llm = ChatOpenAI(model="gpt-4o-mini")
|
|
agent = MCPAgent(
|
|
llm=llm,
|
|
client=client,
|
|
system_prompt="You can browse with Stagehand MCP tools and answer user queries.",
|
|
memory_enabled=True,
|
|
max_steps=20,
|
|
)
|
|
result = await agent.run("Go to example.com and extract the title")
|
|
print(result)
|
|
await client.close_all_sessions()
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|