ConnectLangChainto Perplexity, Klavis ReportGen, Affinity MCP Servers

import os
import asyncio
from klavis import Klavis
from klavis.types import McpServerName, ConnectionType
from langchain_mcp_adapters.client import MultiServerMCPClient
from langgraph.prebuilt import create_react_agent
from langchain_openai import ChatOpenAI

# Initialize clients
klavis_client = Klavis(api_key=os.getenv("KLAVIS_API_KEY"))
llm = ChatOpenAI(model="gpt-4o-mini", api_key=os.getenv("OPENAI_API_KEY"))

perplexity_mcp_instance = klavis_client.mcp_server.create_server_instance(
    server_name=McpServerName.PERPLEXITY,
    user_id="1234",
    platform_name="Klavis",
    connection_type=ConnectionType.STREAMABLE_HTTP,
)

klavis_reportgen_mcp_instance = klavis_client.mcp_server.create_server_instance(
    server_name=McpServerName.KLAVIS_REPORTGEN,
    user_id="1234",
    platform_name="Klavis",
    connection_type=ConnectionType.STREAMABLE_HTTP,
)

affinity_mcp_instance = klavis_client.mcp_server.create_server_instance(
    server_name=McpServerName.AFFINITY,
    user_id="1234",
    platform_name="Klavis",
    connection_type=ConnectionType.STREAMABLE_HTTP,
)

mcp_client = MultiServerMCPClient({
    "perplexity": {
        "transport": "streamable_http",
        "url": perplexity_mcp_instance.server_url
    },
    "klavis reportgen": {
        "transport": "streamable_http",
        "url": klavis_reportgen_mcp_instance.server_url
    },
    "affinity": {
        "transport": "streamable_http",
        "url": affinity_mcp_instance.server_url
    }
})

tools = asyncio.run(mcp_client.get_tools())

agent = create_react_agent(
    model=llm,
    tools=tools,
)

response = asyncio.run(agent.ainvoke({
    "messages": [{"role": "user", "content": "Your query here"}]
}))