Spaces:
Configuration error
Configuration error
File size: 1,096 Bytes
447ebeb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
# Create server parameters for stdio connection
import asyncio
import os
from langchain_mcp_adapters.tools import load_mcp_tools
from langchain_openai import ChatOpenAI
from langgraph.prebuilt import create_react_agent
from mcp import ClientSession
from mcp.client.sse import sse_client
async def main():
model = ChatOpenAI(model="gpt-4o", api_key="sk-12")
async with sse_client(url="http://localhost:4000/mcp/") as (read, write):
async with ClientSession(read, write) as session:
# Initialize the connection
print("Initializing session")
await session.initialize()
print("Session initialized")
# Get tools
print("Loading tools")
tools = await load_mcp_tools(session)
print("Tools loaded")
print(tools)
# # Create and run the agent
# agent = create_react_agent(model, tools)
# agent_response = await agent.ainvoke({"messages": "what's (3 + 5) x 12?"})
# Run the async function
if __name__ == "__main__":
asyncio.run(main())
|