ConnectOpenAIto YouTube, OneDrive MCP Servers

import json
import os
from openai import OpenAI
from klavis import Klavis
from klavis.types import McpServerName, ConnectionType, ToolFormat

# Initialize clients
openai_client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
klavis_client = Klavis(api_key=os.getenv("KLAVIS_API_KEY"))

# Constants
OPENAI_MODEL = "gpt-4o-mini"

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

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

# Get tools from all MCP servers
youtube_tools = klavis_client.mcp_server.list_tools(
    server_url=youtube_mcp_instance.server_url,
    connection_type=ConnectionType.STREAMABLE_HTTP,
    format=ToolFormat.OPENAI,
)
onedrive_tools = klavis_client.mcp_server.list_tools(
    server_url=onedrive_mcp_instance.server_url,
    connection_type=ConnectionType.STREAMABLE_HTTP,
    format=ToolFormat.OPENAI,
)

# Combine all tools
all_tools = []
all_tools.extend(youtube_tools)
all_tools.extend(onedrive_tools)

messages = [
    {"role": "system", "content": "You are a helpful assistant."},
    {"role": "user", "content": user_message}
]
        
response = openai_client.chat.completions.create(
    model=OPENAI_MODEL,
    messages=messages,
    tools=all_tools if all_tools else None
)