Create tools.py

tools.py

@@ -0,0 +1,161 @@
+import random
+from datetime import datetime
+
+class BaseTool:
+    """Base class for all procurement tools"""
+    def __init__(self, name):
+        self.name = name
+    
+    def run(self, pr_data):
+        raise NotImplementedError
+
+class InventoryTool(BaseTool):
+    def __init__(self):
+        super().__init__("InventoryTool")
+    
+    def run(self, pr_data):
+        current_inventory = pr_data["Current_Inventory"]
+        required_quantity = pr_data["Quantity"]
+        
+        # Simulate checking other locations
+        other_locations_inventory = random.randint(0, required_quantity * 2)
+        total_available = current_inventory + other_locations_inventory
+        
+        if total_available >= required_quantity:
+            return {
+                "inventory_available": True,
+                "current_location": current_inventory,
+                "other_locations": other_locations_inventory,
+                "total_available": total_available,
+                "message": f"Sufficient inventory available: {total_available} units across all locations"
+            }
+        else:
+            return {
+                "inventory_available": False,
+                "current_location": current_inventory,
+                "other_locations": other_locations_inventory,
+                "total_available": total_available,
+                "shortage": required_quantity - total_available,
+                "message": f"Inventory shortage: {required_quantity - total_available} units needed"
+            }
+
+class ExternalRiskTool(BaseTool):
+    def __init__(self):
+        super().__init__("ExternalRiskTool")
+    
+    def run(self, pr_data):
+        # Simulate external risk assessment
+        category = pr_data["Category"]
+        external_disruption = pr_data["External_Disruption"]
+        
+        # Risk factors based on category and external conditions
+        risk_factors = []
+        risk_score = 0.0
+        
+        if external_disruption:
+            risk_factors.append("Supply chain disruption reported")
+            risk_score += 0.4
+        
+        # Category-specific risks
+        if category == "Raw Materials":
+            if random.random() < 0.2:  # 20% chance of commodity price volatility
+                risk_factors.append("Commodity price volatility detected")
+                risk_score += 0.3
+        
+        if category == "IT Equipment":
+            if random.random() < 0.15:  # 15% chance of semiconductor shortage
+                risk_factors.append("Semiconductor supply constraints")
+                risk_score += 0.25
+        
+        # Geographic risk simulation
+        if random.random() < 0.1:  # 10% chance of regional disruption
+            risk_factors.append("Regional weather/transport disruption")
+            risk_score += 0.2
+        
+        high_risk = risk_score > 0.3
+        
+        return {
+            "risk": high_risk,
+            "risk_score": risk_score,
+            "risk_factors": risk_factors,
+            "message": f"Risk assessment: {'HIGH RISK' if high_risk else 'LOW RISK'} - Score: {risk_score:.2f}"
+        }
+
+class ContractTool(BaseTool):
+    def __init__(self):
+        super().__init__("ContractTool")
+    
+    def run(self, pr_data):
+        contract_status = pr_data["Contract_Status"]
+        category = pr_data["Category"]
+        
+        # Simulate contract validation
+        if contract_status == "Expired":
+            days_expired = random.randint(1, 90)
+            return {
+                "contract_expired": True,
+                "days_expired": days_expired,
+                "renewal_required": True,
+                "message": f"Contract expired {days_expired} days ago - renewal required"
+            }
+        else:
+            days_remaining = random.randint(30, 365)
+            auto_renewable = category in ["Office Supplies", "Services"]
+            
+            return {
+                "contract_expired": False,
+                "days_remaining": days_remaining,
+                "auto_renewable": auto_renewable,
+                "terms_compliant": True,
+                "message": f"Contract valid for {days_remaining} days - terms compliant"
+            }
+
+class POExecutionTool(BaseTool):
+    def __init__(self):
+        super().__init__("POExecutionTool")
+    
+    def run(self, pr_data):
+        # Simulate PO creation and execution
+        po_number = f"PO-{random.randint(100000, 999999)}"
+        estimated_delivery = random.randint(3, 14)  # 3-14 days
+        
+        # Calculate costs
+        base_cost = pr_data["Budget"]
+        urgency_premium = {"Low": 0.0, "Medium": 0.05, "High": 0.15}
+        final_cost = base_cost * (1 + urgency_premium[pr_data["Urgency"]])
+        
+        return {
+            "po_executed": True,
+            "po_number": po_number,
+            "estimated_delivery_days": estimated_delivery,
+            "final_cost": final_cost,
+            "cost_variance": ((final_cost - base_cost) / base_cost) * 100,
+            "message": f"PO {po_number} created - Estimated delivery: {estimated_delivery} days, Cost: ${final_cost:,.2f}"
+        }
+
+class HumanNotificationTool(BaseTool):
+    def __init__(self):
+        super().__init__("HumanNotificationTool")
+    
+    def run(self, pr_data):
+        # Simulate human notification
+        urgency = pr_data["Urgency"]
+        notification_channels = []
+        
+        if urgency == "High":
+            notification_channels = ["Email", "SMS", "Slack"]
+            sla_hours = 2
+        elif urgency == "Medium":
+            notification_channels = ["Email", "Slack"]
+            sla_hours = 8
+        else:
+            notification_channels = ["Email"]
+            sla_hours = 24
+        
+        return {
+            "notified": True,
+            "channels": notification_channels,
+            "sla_hours": sla_hours,
+            "escalation_level": urgency,
+            "message": f"Human review requested via {', '.join(notification_channels)} - SLA: {sla_hours}h"
+        }