wehub-resource-sync
49 lines
1.0 KiB
Python
49 lines
1.0 KiB
Python
import asyncio
|
|
|
|
from langchain.agents import create_agent
|
|
from langchain_mcp_adapters.client import MultiServerMCPClient
|
|
from langchain_openai.chat_models import ChatOpenAI
|
|
|
|
from composio import Composio
|
|
|
|
composio = Composio()
|
|
session = composio.create(
|
|
user_id="user_123",
|
|
)
|
|
|
|
|
|
async def main():
|
|
try:
|
|
mcp_client = MultiServerMCPClient(
|
|
{
|
|
"composio": {
|
|
"transport": "streamable_http",
|
|
"url": session.mcp.url,
|
|
"headers": session.mcp.headers,
|
|
}
|
|
}
|
|
)
|
|
|
|
tools = await mcp_client.get_tools()
|
|
|
|
agent = create_agent(
|
|
tools=tools,
|
|
model=ChatOpenAI(model="gpt-4o"),
|
|
)
|
|
|
|
result = await agent.ainvoke(
|
|
{
|
|
"messages": [
|
|
{"role": "user", "content": "Fetch my last email and summarize?"}
|
|
]
|
|
}
|
|
)
|
|
|
|
print(result)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|