mirror of
https://github.com/stripe/ai.git
synced 2026-09-14 18:39:59 +08:00
sgr-stripe
48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
import asyncio
|
|
import os
|
|
from dotenv import load_dotenv
|
|
|
|
from langchain_openai import ChatOpenAI
|
|
|
|
from langgraph.prebuilt import create_react_agent
|
|
|
|
from stripe_agent_toolkit.langchain.toolkit import create_stripe_agent_toolkit
|
|
|
|
load_dotenv()
|
|
|
|
|
|
async def main():
|
|
llm = ChatOpenAI(
|
|
model="gpt-4o",
|
|
)
|
|
|
|
# Initialize the Stripe toolkit (connects to MCP server)
|
|
stripe_agent_toolkit = await create_stripe_agent_toolkit(
|
|
secret_key=os.getenv("STRIPE_SECRET_KEY"),
|
|
)
|
|
|
|
try:
|
|
tools = []
|
|
tools.extend(stripe_agent_toolkit.get_tools())
|
|
|
|
langgraph_agent_executor = create_react_agent(llm, tools)
|
|
|
|
input_state = {
|
|
"messages": """
|
|
Create a payment link for a new product called 'test' with a price
|
|
of $100. Come up with a funny description about buy bots,
|
|
maybe a haiku.
|
|
""",
|
|
}
|
|
|
|
output_state = langgraph_agent_executor.invoke(input_state)
|
|
|
|
print(output_state["messages"][-1].content)
|
|
finally:
|
|
# Clean up MCP connection
|
|
await stripe_agent_toolkit.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|