Update app.py

app.py

@@ -22,12 +22,26 @@ def get_model():
 # =======================================
 @tool
 def add(a: float, b: float) -> float:
-    """Return a+b"""
+    """Add two numbers.
+    
+    Args:
+        a (float): First addend.
+        b (float): Second addend.
+    Returns:
+        float: Sum a + b.
+    """
     return a + b
 
 @tool
 def mul(a: float, b: float) -> float:
-    """Return a*b"""
+    """Multiply two numbers.
+    
+    Args:
+        a (float): First factor.
+        b (float): Second factor.
+    Returns:
+        float: Product a * b.
+    """
     return a * b
 
 def run_ex1():
@@ -75,7 +89,13 @@ def run_ex2():
 # =======================================
 @tool
 def validate_pr(pr: dict) -> dict:
-    """Return {ok: bool, errors: [...]} after checking pr_id, requester, cost_center, currency, items."""
+    """Validate basic PR fields and structure.
+    
+    Args:
+        pr (dict): Purchase Requisition object with keys: pr_id, requester, cost_center, currency, items.
+    Returns:
+        dict: {{"ok": bool, "errors": list[str]}}
+    """
     req = ["pr_id","requester","cost_center","currency","items"]
     errors = [f"Missing {k}" for k in req if k not in pr]
     if not isinstance(pr.get("items", []), list) or not pr["items"]:
@@ -84,12 +104,18 @@ def validate_pr(pr: dict) -> dict:
 
 @tool
 def create_po(pr: dict) -> dict:
-    """Create a simple PO JSON with totals and a generated po_id."""
+    """Create a simple Purchase Order (PO) from a PR.
+    
+    Args:
+        pr (dict): Validated PR dict with items[{sku, quantity, unit_price}] and currency.
+    Returns:
+        dict: PO JSON with po_id, items, subtotal, tax, total, source_pr_id.
+    """
     subtotal = 0.0
     items_out = []
     for it in pr.get("items", []):
         line = float(it["quantity"]) * float(it["unit_price"])
-        items_out.append({"sku": it["sku"], "line_total": round(line,2)})
+        items_out.append({"sku": it.get("sku","UNKNOWN"), "line_total": round(line,2)})
         subtotal += line
     tax = round(subtotal * 0.08, 2)
     total = round(subtotal + tax, 2)