ConnectFireworks AIto Stripe, Jira, Cloudflare MCP Servers

import os
import json
from fireworks.client import Fireworks
from klavis import Klavis
from klavis.types import McpServerName, ConnectionType, ToolFormat

# Initialize clients
fireworks_client = Fireworks(api_key=os.getenv("FIREWORKS_API_KEY"))
klavis_client = Klavis(api_key=os.getenv("KLAVIS_API_KEY"))

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

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

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

# Get all MCP tools
stripe_tools = klavis_client.mcp_server.list_tools(
    server_url=stripe_mcp_instance.server_url,
    connection_type=ConnectionType.STREAMABLE_HTTP,
    format=ToolFormat.OPENAI,
)
jira_tools = klavis_client.mcp_server.list_tools(
    server_url=jira_mcp_instance.server_url,
    connection_type=ConnectionType.STREAMABLE_HTTP,
    format=ToolFormat.OPENAI,
)
cloudflare_tools = klavis_client.mcp_server.list_tools(
    server_url=cloudflare_mcp_instance.server_url,
    connection_type=ConnectionType.STREAMABLE_HTTP,
    format=ToolFormat.OPENAI,
)

# Combine all tools
all_tools = []
all_tools.extend(stripe_tools.tools)
all_tools.extend(jira_tools.tools)
all_tools.extend(cloudflare_tools.tools)

messages = [
    {"role": "system", "content": "You are a helpful assistant with access to multiple data sources."},
    {"role": "user", "content": user_message}
]

response = fireworks_client.chat.completions.create(
    model="accounts/fireworks/models/llama-v3p1-70b-instruct",
    messages=messages,
    tools=all_tools
)