Update app/agents/tools.py

app/agents/tools.py

@@ -160,7 +160,7 @@ def run_underwriting_func(tool_input: str) -> str:
 def create_loan_application_func(tool_input: str) -> str:
     """
     Create loan application record and generate sanction letter.
-    ONLY use this after loan is APPROVED.
+    Works for APPROVED loans and ADJUST loans (after user accepts).
     
     Args:
         tool_input: JSON string with user_id, session_id
@@ -192,11 +192,12 @@ def create_loan_application_func(tool_input: str) -> str:
         
         decision = session["loan_decision"]
         
-        # Verify it's approved
-        if decision["decision"] != "APPROVED":
+        # Accept both APPROVED and ADJUST decisions
+        # ADJUST means user accepted the adjusted terms
+        if decision["decision"] not in ["APPROVED", "ADJUST"]:
             return json.dumps({
                 "success": False,
-                "error": f"Cannot create sanction letter for {decision['decision']} loan"
+                "error": f"Cannot create sanction letter for {decision['decision']} loan. Only APPROVED or ADJUST loans can generate sanction letters."
             })
         
         # Get user profile
@@ -212,7 +213,7 @@ def create_loan_application_func(tool_input: str) -> str:
             "user_id": user_id,
             "full_name": profile["full_name"],
             "email": profile["email"],
-            "requested_amount": decision["approved_amount"],  # Use approved amount
+            "requested_amount": decision["approved_amount"],  # Use approved/adjusted amount
             "requested_tenure_months": decision["tenure_months"],
             "approved_amount": decision["approved_amount"],
             "tenure_months": decision["tenure_months"],
@@ -220,7 +221,7 @@ def create_loan_application_func(tool_input: str) -> str:
             "interest_rate": decision["interest_rate"],
             "credit_score": decision["credit_score"],
             "foir": decision["foir"],
-            "decision": decision["decision"],
+            "decision": "APPROVED",  # Store as APPROVED in database (both APPROVED and ADJUST are approved)
             "risk_band": decision["risk_band"],
             "explanation": decision["explanation"],
             "created_at": datetime.utcnow(),
@@ -254,7 +255,10 @@ def create_loan_application_func(tool_input: str) -> str:
             "success": True,
             "loan_id": loan_id,
             "sanction_pdf_path": pdf_path,
-            "message": f"Sanction letter generated successfully for loan {loan_id}"
+            "approved_amount": decision["approved_amount"],
+            "emi": decision["emi"],
+            "tenure_months": decision["tenure_months"],
+            "message": f"Sanction letter generated successfully! Loan ID: {loan_id}"
         })
         
     except Exception as e: