send email with ai (#91)

* send email with ai

* schedule workflow

* fix openai

* workflow failure simulate

* restack ai update

Author: justrdk

email_sender/.env.example

@@ -0,0 +1,4 @@
+FROM_EMAIL=[email protected]
+SENDGRID_API_KEY=your-sendgrid-api-key
+OPENAI_API_KEY=your-openai-api-key
+TO_EMAIL=[email protected]

email_sender/README.md

@@ -0,0 +1,26 @@
+[tool.poetry]
+name = "email_sender"
+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.12.7"
+pydantic = "2.9.2"
+python-dotenv = "1.0.1"
+sendgrid = "6.11.0"
+openai = "1.56.2"
+restack-ai = "0.0.42"
+
+[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"
+schedule_failure = "schedule_workflow_failure:run_schedule_workflow_failure"

email_sender/poetry.lock

@@ -0,0 +1,45 @@
+import asyncio
+import time
+from restack_ai import Restack
+from dataclasses import dataclass
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+@dataclass
+class InputParams:
+    email_context: str
+    subject: str
+    to: str
+
+async def main():
+    client = Restack()
+
+    workflow_id = f"{int(time.time() * 1000)}-SendEmailWorkflow"
+    to_email = os.getenv("TO_EMAIL")
+    if not to_email:
+        raise Exception("TO_EMAIL environment variable is not set")
+
+    run_id = await client.schedule_workflow(
+        workflow_name="SendEmailWorkflow",
+        workflow_id=workflow_id,
+        input={
+            "email_context": "This email should contain a greeting. And telling user we have launched a new AI feature with Restack workflows. Workflows now offer logging and automatic retries when one of its steps fails. Name of user is not provided. You can set as goodbye message on the email just say 'Best regards' or something like that. No need to mention name of user or name of person sending the email.",
+            "subject": "Hello from Restack",
+            "to": to_email
+        }
+    )
+
+    await client.get_workflow_result(
+        workflow_id=workflow_id,
+        run_id=run_id
+    )
+
+    exit(0)
+
+def run_schedule_workflow():
+    asyncio.run(main())
+
+if __name__ == "__main__":
+    run_schedule_workflow()

email_sender/pyproject.toml

@@ -0,0 +1,46 @@
+import asyncio
+import time
+from restack_ai import Restack
+from dataclasses import dataclass
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+@dataclass
+class InputParams:
+    email_context: str
+    subject: str
+    to: str
+
+async def main():
+    client = Restack()
+
+    workflow_id = f"{int(time.time() * 1000)}-SendEmailWorkflow"
+    to_email = os.getenv("TO_EMAIL")
+    if not to_email:
+        raise Exception("TO_EMAIL environment variable is not set")
+
+    run_id = await client.schedule_workflow(
+        workflow_name="SendEmailWorkflow",
+        workflow_id=workflow_id,
+        input={
+            "email_context": "This email should contain a greeting. And telling user we have launched a new AI feature with Restack workflows. Workflows now offer logging and automatic retries when one of its steps fails. Name of user is not provided. You can set as goodbye message on the email just say 'Best regards' or something like that. No need to mention name of user or name of person sending the email.",
+            "subject": "Hello from Restack",
+            "to": to_email,
+            "simulate_failure": True
+        }
+    )
+
+    await client.get_workflow_result(
+        workflow_id=workflow_id,
+        run_id=run_id
+    )
+
+    exit(0)
+
+def run_schedule_workflow_failure():
+    asyncio.run(main())
+
+if __name__ == "__main__":
+    run_schedule_workflow_failure()

email_sender/schedule_workflow.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)

email_sender/schedule_workflow_failure.py

@@ -0,0 +1,45 @@
+from restack_ai.function import function, log, FunctionFailure
+from dataclasses import dataclass
+from openai import OpenAI
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+tries = 0
+
+@dataclass
+class GenerateEmailInput:
+    email_context: str
+    simulate_failure: bool = False
+
+@function.defn()
+async def generate_email_content(input: GenerateEmailInput):
+    global tries
+
+    if input.simulate_failure and tries == 0:
+        tries += 1
+        raise FunctionFailure("Simulated failure", non_retryable=False)
+    
+    if (os.environ.get("OPENAI_API_KEY") is None):
+        raise FunctionFailure("OPENAI_API_KEY is not set", non_retryable=True)
+    
+    client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
+
+    response = client.chat.completions.create(
+        model="gpt-4",
+        messages=[
+            {
+                "role": "system",
+                "content": "You are a helpful assistant that generates short emails based on the provided context."
+            },
+            {
+                "role": "user",
+                "content": f"Generate a short email based on the following context: {input.email_context}"
+            }
+        ],
+        max_tokens=150
+    )
+    
+    return response.choices[0].message.content
+