import asyncio
import os
from dotenv import load_dotenv
from crewai import Agent, Task, Crew
from stripe_agent_toolkit.crewai.toolkit import create_stripe_agent_toolkit
load_dotenv()
async def main():
# Initialize the Stripe toolkit (connects to MCP server)
stripe_agent_toolkit = await create_stripe_agent_toolkit(
secret_key=os.getenv("STRIPE_SECRET_KEY"),
)
try:
stripe_agent = Agent(
role="Stripe Agent",
goal="Integrate with Stripe effectively to support our business.",
backstory="You have been using stripe forever.",
tools=[*stripe_agent_toolkit.get_tools()],
allow_delegation=False,
verbose=True,
)
haiku_writer = Agent(
role="Haiku writer",
goal="Write a haiku",
backstory="You are really good at writing haikus.",
allow_delegation=False,
verbose=True,
)
create_payment_link = Task(
description="Create a payment link for a new product called 'test' "
"with a price of $100. The description should be a haiku",
expected_output="url",
agent=stripe_agent,
)
write_haiku = Task(
description="Write a haiku about buy bots.",
expected_output="haiku",
agent=haiku_writer,
)
crew = Crew(
agents=[stripe_agent, haiku_writer],
tasks=[create_payment_link, write_haiku],
verbose=True,
planning=True,
)
crew.kickoff()
finally:
# Clean up MCP connection
await stripe_agent_toolkit.close()
if __name__ == "__main__":
asyncio.run(main())
