Autonomous Multi-Agent Inventory Replenishment & Supply Network Swarm

from langgraph.graph import StateGraph, END
from typing import TypedDict, List

class SupplyState(TypedDict):
    sku_id: str
    current_inventory: int
    safety_stock: int
    supplier_lead_days: int
    risk_level: str
    rfq_approved: bool
    drafted_po: dict

def inventory_evaluator(state: SupplyState):
    deficit = state["safety_stock"] - state["current_inventory"]
    risk = "HIGH" if deficit > 0 else "NORMAL"
    return {"risk_level": risk}

def rfq_negotiation_agent(state: SupplyState):
    if state["risk_level"] == "HIGH":
        po = {
            "sku": state["sku_id"],
            "order_qty": 5000,
            "max_unit_price": 14.50,
            "target_lead_days": state["supplier_lead_days"]
        }
        return {"drafted_po": po, "rfq_approved": True}
    return {"rfq_approved": False}

graph = StateGraph(SupplyState)
graph.add_node("evaluator", inventory_evaluator)
graph.add_node("rfq_agent", rfq_negotiation_agent)
graph.set_entry_point("evaluator")
graph.add_edge("evaluator", "rfq_agent")
graph.add_edge("rfq_agent", END)
supply_engine = graph.compile()