Update app.py

app.py

@@ -78,7 +78,11 @@ llm_model = None
 # ==========================================
 @tool
 def parse_claim(json_data: str) -> str:
-    """Parses claim JSON data to validate structure."""
+    """Parses claim JSON data to validate structure.
+    
+    Args:
+        json_data: The raw JSON string containing the claim data.
+    """
     try:
         data = json.loads(json_data)
         claim_info = ClaimInfo.model_validate(data)
@@ -88,7 +92,11 @@ def parse_claim(json_data: str) -> str:
 
 @tool
 def is_valid_query(query: str) -> str:
-    """Validates policy standing and dates against coverage database."""
+    """Validates policy standing and dates against coverage database.
+    
+    Args:
+        query: The parsed claim JSON string.
+    """
     try:
         claim_info = ClaimInfo.model_validate_json(query)
         import csv
@@ -118,7 +126,11 @@ def is_valid_query(query: str) -> str:
 
 @tool
 def generate_policy_queries(claim_info_json: str) -> str:
-    """Generate queries to retrieve relevant policy sections based on claim info."""
+    """Generate queries to retrieve relevant policy sections based on claim info.
+    
+    Args:
+        claim_info_json: A JSON string containing the parsed claim details.
+    """
     global llm_model
     prompt = f"""
     Analyze the following auto insurance claim to identify 3-5 key policy sections to consult:
@@ -137,7 +149,11 @@ def generate_policy_queries(claim_info_json: str) -> str:
 
 @tool
 def retrieve_policy_text(queries_json: str) -> str:
-    """Retrieves policy text from ChromaDB using generated queries."""
+    """Retrieves policy text from ChromaDB using generated queries.
+    
+    Args:
+        queries_json: A JSON string containing a list of search queries.
+    """
     try:
         queries_data = json.loads(queries_json)
         query_strings = queries_data.get("queries", [])
@@ -153,7 +169,12 @@ def retrieve_policy_text(queries_json: str) -> str:
 
 @tool
 def generate_recommendation(claim_info_json: str, policy_text: str) -> str:
-    """Generate a policy recommendation based on claim info and retrieved policy text."""
+    """Generate a policy recommendation based on claim info and retrieved policy text.
+    
+    Args:
+        claim_info_json: The validated claim info in JSON format.
+        policy_text: The relevant text retrieved from the policy documents.
+    """
     global llm_model
     prompt = f"""
     Evaluate the following auto insurance claim against the policy text:
@@ -180,7 +201,12 @@ def generate_recommendation(claim_info_json: str, policy_text: str) -> str:
 
 @tool
 def finalize_decision(claim_info_json: str, recommendation_json: str) -> str:
-    """Finalize the claim decision based on the recommendation and format output."""
+    """Finalize the claim decision based on the recommendation and format output.
+    
+    Args:
+        claim_info_json: The validated claim information.
+        recommendation_json: The AI-generated recommendation output.
+    """
     try:
         claim_info = ClaimInfo.model_validate_json(claim_info_json)
         rec_data = json.loads(recommendation_json)