Stripe ai sdk example (#93)

* stripe ai with langchain

* indentation fix

* open ai key to model

* services and client

* update readme

* create paymentlink

* more checks

* typo

Author: justrdk

email_sender/README.md

@@ -0,0 +1,80 @@
+# Restack AI - Email Sender example
+
+This example showcases how to send emails with a Restack workflow using the sendgrid api. You can easily choose another email provider and update the code.
+You can schedule two scenarios of the workflow.
+
+1. It will be successfull and send an email.
+2. The email content generation step will fail once to showcase how Restack handles retries automatically. Once failure is caught, step will be retry automatically and rest of workflow will be executed as expected and email will be sent.
+
+
+## Prerequisites
+
+- Python 3.10 or higher
+- Poetry (for dependency management)
+- Docker (for running the Restack services)
+
+## Usage
+
+1. Run Restack local engine with Docker:
+
+   ```bash
+   docker run -d --pull always --name restack -p 5233:5233 -p 6233:6233 -p 7233:7233 ghcr.io/restackio/restack:main
+   ```
+
+2. Open the web UI to see the workflows:
+
+   ```bash
+   http://localhost:5233
+   ```
+
+3. Clone this repository:
+
+   ```bash
+   git clone https://github.com/restackio/examples-python
+   cd examples-python/examples/get-started
+   ```
+  
+4. Create .env file with: STRIPE_SECRET_KEY and OPENAI_API_KEY
+
+4. Install dependencies using Poetry:
+
+   ```bash
+   poetry env use 3.12
+   ```
+
+   ```bash
+   poetry shell
+   ```
+
+   ```bash
+   poetry install
+   ```
+
+   ```bash
+   poetry env info # Optional: copy the interpreter path to use in your IDE (e.g. Cursor, VSCode, etc.)
+   ```
+
+5. Run the services:
+
+   ```bash
+   poetry run services
+   ```
+
+   This will start the Restack service with the defined workflows and functions.
+
+6. In a new terminal, schedule the workflow:
+
+   ```bash
+   poetry shell
+   ```
+
+   ```bash
+   poetry run schedule
+   ```
+
+   This will schedule the `SendEmailWorkflow` and print the result.
+
+7. To simulate a flow where the step for sending email fails and the retry is automatically handled by Restack AI use run:
+   ```bash
+   poetry run schedule_failure
+   ```

stripe-ai/.env.example

@@ -0,0 +1,4 @@
+STRIPE_SECRET_KEY=your-stripe-secret-key
+OPENAI_API_KEY=your-openai-api-key
+LANGCHAIN_API_KEY=your-langchain-api-key
+LANGCHAIN_TRACING_V2=false

stripe-ai/README.md

@@ -0,0 +1,70 @@
+# Restack AI - Stripe Ai example
+
+This repository contains a an example on how restack can use langchain and the stripe ai sdk to create a product with a price and also create a payment link for it.
+
+## Prerequisites
+
+- Python 3.10 or higher
+- Poetry (for dependency management)
+- Docker (for running the Restack services)
+
+## Usage
+
+1. Run Restack local engine with Docker:
+
+   ```bash
+   docker run -d --pull always --name restack -p 5233:5233 -p 6233:6233 -p 7233:7233 ghcr.io/restackio/restack:main
+   ```
+
+2. Open the web UI to see the workflows:
+
+   ```bash
+   http://localhost:5233
+   ```
+
+3. Clone this repository:
+
+   ```bash
+   git clone https://github.com/restackio/examples-python
+   cd examples-python/examples/get-started
+   ```
+  
+4. Create .env file with: STRIPE_SECRET_KEY, LANGCHAIN_API_KEY and OPENAI_API_KEY
+
+4. Install dependencies using Poetry:
+
+   ```bash
+   poetry env use 3.12
+   ```
+
+   ```bash
+   poetry shell
+   ```
+
+   ```bash
+   poetry install
+   ```
+
+   ```bash
+   poetry env info # Optional: copy the interpreter path to use in your IDE (e.g. Cursor, VSCode, etc.)
+   ```
+
+5. Run the services:
+
+   ```bash
+   poetry run services
+   ```
+
+   This will start the Restack service with the defined workflows and functions.
+
+6. In a new terminal, schedule the workflow:
+
+   ```bash
+   poetry shell
+   ```
+
+   ```bash
+   poetry run schedule
+   ```
+
+   This will schedule the `CreatePaymentLinkWorkflow` and print the result.

stripe-ai/pyproject.toml

@@ -0,0 +1,28 @@
+[tool.poetry]
+name = "stripe-ai"
+version = "0.0.1"
+description = "Send emails with sendgrid"
+authors = [
+    "Restack Team <[email protected]>",
+]
+readme = "README.md"
+packages = [{include = "src"}]
+
+[tool.poetry.dependencies]
+python = ">=3.10,<4.0"
+pydantic = "2.9.2"
+python-dotenv = "1.0.1"
+openai = "1.56.2"
+restack-ai = "0.0.42"
+stripe-agent-toolkit = "^0.1.21"
+langchain = "^0.3.9"
+stripe = "^11.3.0"
+langchain-openai = "^0.2.11"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.poetry.scripts]
+services = "src.services:run_services"
+schedule = "schedule_workflow:run_schedule_workflow"

stripe-ai/schedule_workflow.py

@@ -0,0 +1,31 @@
+import asyncio
+import time
+from restack_ai import Restack
+from dotenv import load_dotenv
+
+load_dotenv()
+
+async def main():
+    client = Restack()
+
+    workflow_id = f"{int(time.time() * 1000)}-CreatePaymentLinkWorkflow"
+
+    run_id = await client.schedule_workflow(
+        workflow_name="CreatePaymentLinkWorkflow",
+        workflow_id=workflow_id,
+    )
+
+    result = await client.get_workflow_result(
+        workflow_id=workflow_id,
+        run_id=run_id
+    )
+
+    print(result)
+
+    exit(0)
+
+def run_schedule_workflow():
+    asyncio.run(main())
+
+if __name__ == "__main__":
+    run_schedule_workflow()

stripe-ai/src/__init__.py

@@ -0,0 +1,19 @@
+import os
+from restack_ai import Restack
+from restack_ai.restack import CloudConnectionOptions
+from dotenv import load_dotenv  
+
+# Load environment variables from a .env file
+load_dotenv()
+
+
+engine_id = os.getenv("RESTACK_ENGINE_ID")
+address = os.getenv("RESTACK_ENGINE_ADDRESS")
+api_key = os.getenv("RESTACK_ENGINE_API_KEY")
+
+connection_options = CloudConnectionOptions(
+    engine_id=engine_id,
+    address=address,
+    api_key=api_key
+)
+client = Restack(connection_options)

stripe-ai/src/client.py

@@ -0,0 +1,59 @@
+from langchain.agents import AgentExecutor, create_structured_chat_agent
+from langchain import hub
+from stripe_agent_toolkit.langchain.toolkit import StripeAgentToolkit
+from restack_ai.function import function, FunctionFailure
+from langchain_openai import ChatOpenAI
+
+import os
+from dotenv import load_dotenv
+from pydantic import SecretStr
+
+load_dotenv()
+
+@function.defn()
+async def create_payment_link():
+    stripe_secret_key = os.getenv("STRIPE_SECRET_KEY")
+    openai_api_key = os.getenv("OPENAI_API_KEY")
+    langchain_api_key = os.getenv("LANGCHAIN_API_KEY")
+
+    if stripe_secret_key is None:
+        raise FunctionFailure("STRIPE_SECRET_KEY is not set", non_retryable=True)
+    
+    if langchain_api_key is None:
+       raise FunctionFailure("LANGCHAIN_API_KEY is not set", non_retryable=True)
+    
+    if openai_api_key is None:
+        raise FunctionFailure("OPENAI_API_KEY is not set", non_retryable=True)
+    
+    try:
+        stripe_agent_toolkit = StripeAgentToolkit(
+            secret_key=stripe_secret_key,
+            configuration={
+            "actions": {
+                "payment_links": {
+                    "create": True,
+                },
+                "products": {
+                    "create": True,
+                },
+                "prices": {
+                    "create": True,
+                },
+            }
+            },
+        )
+
+        model = ChatOpenAI(api_key=SecretStr(openai_api_key))
+        
+        prompt = hub.pull("hwchase17/structured-chat-agent")
+
+        agent = create_structured_chat_agent(model, stripe_agent_toolkit.get_tools(), prompt)
+        agent_executor = AgentExecutor(agent=agent, tools=stripe_agent_toolkit.get_tools())
+
+        result = agent_executor.invoke({
+          "input": "Create a payment link for a new product called \"Test\" with a price of $100."
+        })
+
+        return result["output"]
+    except Exception as e:
+        raise FunctionFailure(f"Error creating payment link: {e}", non_retryable=True)

stripe-ai/src/functions/__init__.py

@@ -0,0 +1,18 @@
+import asyncio
+from src.client import client
+from src.workflows.payment_link import CreatePaymentLinkWorkflow
+from src.functions.create_payment_link import create_payment_link
+
+async def main():
+    await asyncio.gather(
+        client.start_service(
+            workflows=[CreatePaymentLinkWorkflow],
+            functions=[create_payment_link]
+        )
+    )
+
+def run_services():
+    asyncio.run(main())
+
+if __name__ == "__main__":
+    run_services()

stripe-ai/src/functions/create_payment_link.py

@@ -0,0 +1,21 @@
+from restack_ai.workflow import workflow, import_functions, log, RetryPolicy
+from datetime import timedelta
+
+with import_functions():
+    from src.functions.create_payment_link import create_payment_link
+
+@workflow.defn()
+class CreatePaymentLinkWorkflow:
+    @workflow.run
+    async def run(self):
+        log.info("CreatePaymentLinkWorkflow started", input=input)
+
+        result = await workflow.step(
+            create_payment_link,
+            retry_policy=RetryPolicy(
+                initial_interval=timedelta(seconds=10),
+                backoff_coefficient=1,
+            ),
+        )
+
+        return result