ConnectCrewAIto Postgres, Firecrawl Deep Research, Klavis ReportGen MCP Servers

import os
from crewai import Agent, Task, Crew, Process
from crewai_tools import MCPServerAdapter
from klavis import Klavis
from klavis.types import McpServerName, ConnectionType

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

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

firecrawl_deep_research_mcp_instance = klavis_client.mcp_server.create_server_instance(
    server_name=McpServerName.FIRECRAWL_DEEP_RESEARCH,
    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,
)

# Initialize MCP tools for each server
postgres_tools = MCPServerAdapter(postgres_mcp_instance.server_params)
firecrawl_deep_research_tools = MCPServerAdapter(firecrawl_deep_research_mcp_instance.server_params)
klavis_reportgen_tools = MCPServerAdapter(klavis_reportgen_mcp_instance.server_params)

# Create specialized agents for each service
postgres_agent = Agent(
    role="Postgres Specialist",
    goal="Handle all Postgres related tasks and data processing",
    backstory="You are an expert in Postgres operations and data analysis",
    tools=postgres_tools,
    reasoning=True,
    verbose=False
)

firecrawl_deep_research_agent = Agent(
    role="Firecrawl Deep Research Specialist",
    goal="Handle all Firecrawl Deep Research related tasks and data processing",
    backstory="You are an expert in Firecrawl Deep Research operations and data analysis",
    tools=firecrawl_deep_research_tools,
    reasoning=True,
    verbose=False
)

klavis_reportgen_agent = Agent(
    role="Klavis ReportGen Specialist",
    goal="Handle all Klavis ReportGen related tasks and data processing",
    backstory="You are an expert in Klavis ReportGen operations and data analysis",
    tools=klavis_reportgen_tools,
    reasoning=True,
    verbose=False
)

# Define collaborative tasks
research_task = Task(
    description="Gather comprehensive data from all available sources",
    expected_output="Raw data and initial findings from all services",
    agent=postgres_agent,
    markdown=True
)

analysis_task = Task(
    description="Analyze and synthesize the gathered data",
    expected_output="Comprehensive analysis with insights and recommendations",
    agent=firecrawl_deep_research_agent,
    markdown=True
)

# Create multi-agent crew
multi_agent_crew = Crew(
    agents=[postgres_agent, firecrawl_deep_research_agent, klavis_reportgen_agent],
    tasks=[research_task, analysis_task],
    verbose=False,
    process=Process.sequential
)

result = multi_agent_crew.kickoff()