import asyncio
import os
from dotenv import load_dotenv
from langchain_mcp_adapters.client import MultiServerMCPClient
load_dotenv()
async def main():
# Define server configuration for MultiServerMCPClient
server_config = {
"AIServer": {
"url": "http://127.0.0.1:8000/mcp",
"transport": "streamable-http",
"headers": {
"Bearer": os.getenv("GOOGLE_API_KEY"),
},
}
}
client = MultiServerMCPClient(server_config)
async with client.session("AIServer") as session:
while True:
user_input = input("Ask AI (type 'exit' to stop): ")
if user_input.lower() == "exit":
break
response = await session.call_tool("ask_ai", {"prompt": user_input})
print(f"Response: {response.structuredContent['result']}")
if __name__ == "__main__":
asyncio.run(main())
