Add more logic

Author: danielmillerp

examples/demos/procurement_agent/.gitignore

@@ -0,0 +1,63 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# Virtual environments
+venv/
+env/
+ENV/
+.venv
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+*~
+
+# Database files
+*.db
+*.sqlite
+*.sqlite3
+
+# Environment variables
+.env
+.env.local
+
+# Logs
+*.log
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Jupyter
+.ipynb_checkpoints/
+*.ipynb_checkpoints
+
+# Testing
+.pytest_cache/
+.coverage
+htmlcov/
+
+# UV
+.venv/
+uv.lock

examples/demos/procurement_agent/project/activities.py

@@ -0,0 +1 @@
+"""Procurement agent activities module."""

examples/demos/procurement_agent/project/activities/__init__.py

@@ -0,0 +1,174 @@
+from temporalio import activity, workflow
+import asyncio
+from datetime import timedelta, datetime
+import uuid
+import json
+
+from agentex.lib.utils.logging import make_logger
+from project.models.events import (
+    SubmitalApprovalEvent,
+    ShipmentDepartedFactoryEvent,
+    ShipmentArrivedSiteEvent,
+    InspectionFailedEvent,
+)
+from project.data.database import (
+    create_schedule_for_workflow,
+    get_schedule_for_workflow,
+    update_delivery_date_for_item_for_workflow,
+    remove_delivery_item_for_workflow,
+    update_project_end_date_for_workflow,
+)
+
+logger = make_logger(__name__)
+
+@activity.defn
+async def issue_purchase_order(event: SubmitalApprovalEvent) -> str:
+    """
+    Issues a purchase order for construction materials.
+
+    Call this when:
+    - A submittal is approved (Submittal_Approved event)
+    - Human feedback requests reissuing a purchase order
+    """
+    uuid_purchase_order = str(uuid.uuid4())
+    # wait for 5 seconds as if we were calling an API to issue a purchase order
+    await asyncio.sleep(5)
+    logger.info(f"Issuing purchase order: {event}")
+    logger.info(f"Purchase order ID: {uuid_purchase_order}")
+
+    return f"Successfully issued purchase order with ID: {uuid_purchase_order}"  
+
+@activity.defn
+async def flag_potential_issue(event: ShipmentDepartedFactoryEvent) -> str:
+    """
+    Flags a potential issue with a delivery date.
+
+    Call this when:
+    - A shipment departure creates timeline concerns (Shipment_Departed_Factory event)
+    - Human feedback identifies a potential delivery issue
+    """
+    logger.info(f"Flagging potential issue: {event}")
+    logger.info(f"Potential issue flagged with delivery date: {event.eta}")
+    # imagine this is a call to an API to flag a potential issue, perhaps a notification to a team member
+    await asyncio.sleep(1)
+    return f"Potential issue flagged with delivery date: {event.eta}"
+
+@activity.defn
+async def notify_team_shipment_arrived(event: ShipmentDepartedFactoryEvent) -> str:
+    """
+    Notifies the team that a shipment has arrived.
+
+    Call this when:
+    - A shipment arrives at the site (Shipment_Arrived_Site event)
+    - Human feedback requests team notification
+    """
+    logger.info(f"Notifying team that shipment has arrived: {event.item}")
+    logger.info(f"Team notification sent for arrival of: {event.item}")
+    # imagine this is a call to an API to notify the team that a shipment has arrived, perhaps a notification to a team member
+    await asyncio.sleep(1)
+
+    return f"Notifying team that shipment has arrived: {event.item}"
+
+@activity.defn
+async def schedule_inspection(event: ShipmentDepartedFactoryEvent) -> str:
+    """
+    Schedules an inspection for delivered materials.
+
+    Call this when:
+    - A shipment arrives at the site (Shipment_Arrived_Site event)
+    - Human feedback requests scheduling an inspection
+    """
+    inspection_date = datetime.now() + timedelta(days=1)
+    logger.info(f"Scheduling inspection for: {event.item} on {inspection_date}")
+    # imagine this is a call to an API to schedule an inspection
+    await asyncio.sleep(1)
+    return f"Scheduling inspection for {event.item} on {inspection_date}"
+
+
+
+@activity.defn
+async def create_master_construction_schedule(workflow_id: str) -> str:
+    """
+    Creates the master construction schedule for the workflow.
+
+    Call this when:
+    - The workflow is created
+
+    Args:
+        workflow_id: The Temporal workflow ID
+    """
+    logger.info(f"Creating master construction schedule for workflow: {workflow_id}")
+    await create_schedule_for_workflow(workflow_id)
+    return "Master construction schedule created for workflow"
+
+@activity.defn
+async def get_master_construction_schedule(workflow_id: str) -> str:
+    """
+    Gets the master construction schedule for the workflow.
+
+    Call this when:
+    - You want to get the master construction schedule for the workflow
+    - Human feedback requests the master construction schedule
+
+    Returns:
+        The master construction schedule for the workflow
+    """
+    schedule = await get_schedule_for_workflow(workflow_id)
+    if schedule is None:
+        return "No master construction schedule found for workflow"
+    
+    logger.info(f"Master construction schedule found for workflow: {workflow_id}")
+    logger.info(f"Master construction schedule: {schedule}")
+    return json.dumps(schedule)
+
+@activity.defn
+async def update_delivery_date_for_item(workflow_id: str, item: str, new_delivery_date: str) -> str:
+    """
+    Updates the delivery date for a specific item in the construction schedule.
+
+    Call this when:
+    - You want to update the delivery date for a specific item in the construction schedule
+    - Human feedback requests updating the delivery date for a specific item
+
+    Args:
+        workflow_id: The Temporal workflow ID
+        item: The item to update
+        new_delivery_date: The new delivery date
+    """
+    logger.info(f"Updating delivery date for item: {item} to {new_delivery_date}")
+    await update_delivery_date_for_item_for_workflow(workflow_id, item, new_delivery_date)
+    return f"Delivery date updated for item: {item} to {new_delivery_date}"
+
+@activity.defn
+async def remove_delivery_item(workflow_id: str, item: str) -> str:
+    """
+    Removes a delivery item from the construction schedule.
+
+    Call this when:
+    - You want to remove a delivery item from the construction schedule
+    - Human feedback requests removing a delivery item
+
+    Args:
+        workflow_id: The Temporal workflow ID
+        item: The item to remove
+    """
+    logger.info(f"Removing delivery item: {item}")
+    await remove_delivery_item_for_workflow(workflow_id, item)
+    return f"Delivery item removed from construction schedule: {item}"
+
+@activity.defn
+async def update_project_end_date(workflow_id: str, new_end_date: str) -> str:
+    """
+    Updates the end date for the project in the construction schedule.
+
+    Call this when:
+    - You want to update the end date for the project in the construction schedule
+    - Human feedback requests updating the end date for the project
+
+    Args:
+        workflow_id: The Temporal workflow ID
+        new_end_date: The new end date for the project
+    """
+    logger.info(f"Updating end date for project to: {new_end_date}")
+    await update_project_end_date_for_workflow(workflow_id, new_end_date)
+    return f"End date updated for project: {new_end_date}"

examples/demos/procurement_agent/project/activities/activities.py

@@ -0,0 +1 @@
+"""Procurement agent agents module."""