Merge pull request #33 from pamelafox/hitlfix

HITL improvements

Author: pamelafox

examples/spanish/workflow_hitl_checkpoint.py

@@ -15,7 +15,6 @@
 import asyncio
 import os
 from dataclasses import dataclass
-from datetime import datetime
 from pathlib import Path
 from typing import Any
 
@@ -181,15 +180,10 @@ async def main() -> None:
     )
 
     # Verifica si hay checkpoints existentes para reanudar
-    checkpoints = await storage.list_checkpoints(workflow_name=workflow.name)
-    if checkpoints:
-        sorted_cps = sorted(checkpoints, key=lambda cp: datetime.fromisoformat(cp.timestamp))
-        latest = sorted_cps[-1]
-        print(
-            f"📂 Se encontraron {len(sorted_cps)} checkpoint(s)."
-            f" Reanudando desde el más reciente: {latest.checkpoint_id}"
-        )
-        stream = workflow.run(checkpoint_id=latest.checkpoint_id, stream=True)
+    checkpoint = await storage.get_latest(workflow_name=workflow.name)
+    if checkpoint:
+        print(f"📂 Se encontró un checkpoint. Reanudando desde el más reciente: {checkpoint.checkpoint_id}")
+        stream = workflow.run(checkpoint_id=checkpoint.checkpoint_id, stream=True)
     else:
         brief = (
             "Presenta nuestra nueva freidora de aire compacta con canasta de 5 cuartos. Menciona el precio de $89, "

examples/spanish/workflow_hitl_checkpoint_pg.py

@@ -274,11 +274,10 @@ async def main() -> None:
     )
 
     # Verifica si hay checkpoints existentes para reanudar
-    checkpoints = await storage.list_checkpoints(workflow_name=workflow.name)
-    if checkpoints:
-        latest = checkpoints[-1]  # ya ordenados por timestamp
-        print(f"📂 Se encontraron {len(checkpoints)} checkpoint(s) en PostgreSQL. Reanudando desde el más reciente.")
-        stream = workflow.run(checkpoint_id=latest.checkpoint_id, stream=True)
+    checkpoint = await storage.get_latest(workflow_name=workflow.name)
+    if checkpoint:
+        print(f"📂 Se encontró un checkpoint en PostgreSQL. Reanudando desde el más reciente.")
+        stream = workflow.run(checkpoint_id=checkpoint.checkpoint_id, stream=True)
     else:
         brief = (
             "Presenta nuestra nueva freidora de aire compacta con canasta de 5 cuartos. Menciona el precio de $89, "

examples/spanish/workflow_hitl_handoff.py

@@ -56,6 +56,7 @@ def process_return(
     return_type: Annotated[str, "Either 'refund' or 'replacement'"],
 ) -> str:
     """Procesa una devolución de producto para el pedido indicado."""
+    print(f"\n🔧 [Herramienta llamada: process_return(order_number={order_number}, return_type={return_type})]")
     return f"Devolución procesada para el pedido {order_number}: {return_type} aprobado. Correo de confirmación enviado."
 
 
@@ -70,6 +71,7 @@ def process_return(
         "return_agent para devoluciones. No puedes manejar problemas específicos tú mismo — siempre transfiere. "
         "Si un especialista acaba de terminar de ayudar, pregunta al cliente si necesita algo más "
         "— NO reenvíes al mismo especialista. "
+        "Los números de pedido son solo de 3 dígitos."
         "NO pidas información de contacto, correo electrónico ni número de teléfono. "
         "NO digas 'Adiós' hasta que el cliente confirme explícitamente que no tiene más preguntas."
     ),
@@ -80,6 +82,7 @@ def process_return(
     name="order_agent",
     instructions=(
         "Eres el especialista en seguimiento de pedidos. Ayuda al cliente a verificar el estado de su pedido. "
+        "Si el cliente necesita devolver un pedido, transfiere a return_agent. "
         "Cuando termines, transfiere de vuelta a triage_agent."
     ),
 )

examples/workflow_hitl_checkpoint.py

@@ -15,7 +15,6 @@
 import asyncio
 import os
 from dataclasses import dataclass
-from datetime import datetime
 from pathlib import Path
 from typing import Any
 
@@ -181,12 +180,10 @@ async def main() -> None:
     )
 
     # Check if there are existing checkpoints to resume from
-    checkpoints = await storage.list_checkpoints(workflow_name=workflow.name)
-    if checkpoints:
-        sorted_cps = sorted(checkpoints, key=lambda cp: datetime.fromisoformat(cp.timestamp))
-        latest = sorted_cps[-1]
-        print(f"📂 Found {len(sorted_cps)} checkpoint(s). Resuming from latest: {latest.checkpoint_id}")
-        stream = workflow.run(checkpoint_id=latest.checkpoint_id, stream=True)
+    checkpoint = await storage.get_latest(workflow_name=workflow.name)
+    if checkpoint:
+        print(f"📂 Found checkpoint. Resuming from latest: {checkpoint.checkpoint_id}")
+        stream = workflow.run(checkpoint_id=checkpoint.checkpoint_id, stream=True)
     else:
         brief = (
             "Introduce our new compact air fryer with a 5-quart basket. Mention the $89 price, "

examples/workflow_hitl_checkpoint_pg.py

@@ -274,11 +274,10 @@ async def main() -> None:
     )
 
     # Check if there are existing checkpoints to resume from
-    checkpoints = await storage.list_checkpoints(workflow_name=workflow.name)
-    if checkpoints:
-        latest = checkpoints[-1]  # already sorted by timestamp
-        print(f"📂 Found {len(checkpoints)} checkpoint(s) in PostgreSQL. Resuming from latest.")
-        stream = workflow.run(checkpoint_id=latest.checkpoint_id, stream=True)
+    checkpoint = await storage.get_latest(workflow_name=workflow.name)
+    if checkpoint:
+        print(f"📂 Found a checkpoint in PostgreSQL. Resuming from latest.")
+        stream = workflow.run(checkpoint_id=checkpoint.checkpoint_id, stream=True)
     else:
         brief = (
             "Introduce our new compact air fryer with a 5-quart basket. Mention the $89 price, "

examples/workflow_hitl_handoff.py

@@ -56,6 +56,7 @@ def process_return(
     return_type: Annotated[str, "Either 'refund' or 'replacement'"],
 ) -> str:
     """Process a product return for the given order."""
+    print(f"\n🔧 [Tool called: process_return(order_number={order_number}, return_type={return_type})]")
     return f"Return processed for order {order_number}: {return_type} approved. Confirmation email sent."
 
 
@@ -70,6 +71,7 @@ def process_return(
         "return_agent for returns. You cannot handle specific issues yourself — always hand off. "
         "If a specialist has just finished helping, ask the customer if there is anything else "
         "they need help with — do NOT re-route to the same specialist. "
+        "Order numbers are only 3 digits long."
         "Do NOT ask for contact information, email, or phone number. "
         "Do NOT say 'Goodbye' until the customer explicitly confirms they have no more questions."
     ),
@@ -79,7 +81,8 @@ def process_return(
     client=client,
     name="order_agent",
     instructions=(
-        "You are the order tracking specialist. Help the customer check order status. "
+        "You are the order tracking specialist. Help the customer check status of pending orders. "
+        "If the customer needs to return an order, hand off to return_agent. "
         "When done, hand off back to triage_agent."
     ),
 )