Finish the example

Author: leawn

re_act/README.md

@@ -0,0 +1,53 @@
+# Restack AI - ReAct Example
+
+## Prerequisites
+
+- Python 3.11
+- Poetry (for dependency management)
+- Docker (for running the Restack services)
+
+## Usage
+
+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/re_act
+   ```
+
+4. Install dependencies using Poetry:
+
+   ```bash
+   poetry env use 3.11
+   ```
+
+   ```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
+   ```
+
+6. Schedule workflow
+
+   ```bash
+   poetry run schedule_workflow
+   ```
+

re_act/schedule_workflow.py

@@ -9,7 +9,11 @@ async def main():
     workflow_id = f"{int(time.time() * 1000)}-ParentWorkflow"
     runId = await client.schedule_workflow(
         workflow_name="ParentWorkflow",
-        workflow_id=workflow_id
+        workflow_id=workflow_id,
+        input={
+            "email": "[email protected]",
+            "current_accepted_applicants_count": 9
+        }
     )
 
     await client.get_workflow_result(

re_act/src/functions/generate_email_content.py

@@ -12,23 +12,30 @@ class GenerateEmailInput:
 
 @function.defn()
 async def generate_email_content(input: GenerateEmailInput):
-    
-    client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
-    response = client.chat.completions.create(
-        model="gpt-4o-mini",
-        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
+    try:
+        client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
+
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {
+                    "role": "system",
+                    "content": f"""
+                    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
+    except Exception as e:
+        log.error("Failed to generate email content", error=e)
+        raise FunctionFailure("Failed to generate email content", non_retryable=True)
 

re_act/src/functions/send_email.py

@@ -36,4 +36,4 @@ async def send_email(input: SendEmailInput):
         sg.send(message)
     except Exception as e:
         log.error("Failed to send email", error=e)
-        raise e
+        raise FunctionFailure("Failed to send email", non_retryable=True)

re_act/src/workflows/child_workflow_a.py

@@ -9,8 +9,8 @@
 @workflow.defn()
 class ChildWorkflowA:
     @workflow.run
-    async def run(self):
-        log.info(f"Sending email to [email protected]")
+    async def run(self, email: str):
+        log.info(f"Sending email to {email}")
 
         text = await workflow.step(
             generate_email_content,

re_act/src/workflows/child_workflow_b.py

@@ -9,8 +9,8 @@
 @workflow.defn()
 class ChildWorkflowB:
     @workflow.run
-    async def run(self):
-        log.info(f"Sending email to [email protected]")
+    async def run(self, email: str):
+        log.info(f"Sending email to {email}")
 
         text = await workflow.step(
             generate_email_content,

re_act/src/workflows/parent_workflow.py

@@ -1,24 +1,29 @@
 from restack_ai.workflow import workflow, log, workflow_info, import_functions
 from datetime import timedelta
 import json
-
+from dataclasses import dataclass
 from .child_workflow_a import ChildWorkflowA
 from .child_workflow_b import ChildWorkflowB
 
 with import_functions():
     from src.functions.decide import decide, DecideInput
 
+@dataclass
+class ParentWorkflowInput:
+    email: str
+    current_accepted_applicants_count: int
+
 @workflow.defn()
 class ParentWorkflow:
     @workflow.run
-    async def run(self):
+    async def run(self, input: ParentWorkflowInput):
         parent_workflow_id = workflow_info().workflow_id
 
         decide_result = await workflow.step(
             decide,
             input=DecideInput(
-                email="[email protected]",
-                current_accepted_applicants_count=9
+                email=input.email,
+                current_accepted_applicants_count=input.current_accepted_applicants_count
             ),
             start_to_close_timeout=timedelta(seconds=120)
         )
@@ -30,11 +35,13 @@ async def run(self):
             child_workflow_result = await workflow.child_execute(
                 ChildWorkflowA,
                 workflow_id=f"{parent_workflow_id}-child-a",
+                input=input.email
             )
         elif not decision:
             child_workflow_result = await workflow.child_execute(
                 ChildWorkflowB,
                 workflow_id=f"{parent_workflow_id}-child-b",
+                input=input.email
             )
 
         return child_workflow_result