Merge pull request #105 from abinaze/fix/r5-wire-cache-ttl

Wire get_graph_depth() into 5 internal investigator queries

ai/deep_investigator.py

@@ -3,6 +3,7 @@ sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 from datetime import datetime
 from loguru import logger
+from config.model_selector import get_graph_depth
 
 
 LAYER_NAMES = [
@@ -204,6 +205,15 @@ class DeepInvestigator:
 
     def _layer_3_patterns(self, entity_id: str, session) -> dict:
         findings = []
+        # R-5 fix: was a hardcoded *1..2 in both queries below. Safe to wire
+        # tier-aware here specifically because 2 IS the current hardcoded
+        # value AND PROFILE["low"]'s value -- so a low-tier/desktop install
+        # sees zero change, while medium/high tiers get a deeper (more
+        # thorough, not less) traversal than today's shared baseline. This
+        # is NOT true of every graph_depth call site in this codebase (see
+        # this commit's message), which is why each one was checked
+        # individually rather than assuming they all behave the same way.
+        depth = int(get_graph_depth())
 
         # Pattern 1: Company with multiple directors that won contracts
         rows = session.run(
@@ -223,8 +233,8 @@ class DeepInvestigator:
 
         # Pattern 2: Audit flags on connected schemes
         rows = session.run(
-            """
-            MATCH (n {id:$id})-[*1..2]-(a:AuditReport)
+            f"""
+            MATCH (n {{id:$id}})-[*1..{depth}]-(a:AuditReport)
             RETURN a.title AS title, a.year AS year, a.amount_crore AS amount LIMIT 5
             """, id=entity_id
         ).data()
@@ -237,8 +247,8 @@ class DeepInvestigator:
 
         # Pattern 3: Electoral bond connections
         rows = session.run(
-            """
-            MATCH (n {id:$id})-[*1..2]-(e:ElectoralBond)
+            f"""
+            MATCH (n {{id:$id}})-[*1..{depth}]-(e:ElectoralBond)
             RETURN e.donor_name AS donor, e.amount AS amount, e.party AS party LIMIT 5
             """, id=entity_id
         ).data()
@@ -258,11 +268,12 @@ class DeepInvestigator:
 
     def _layer_4_timeline(self, entity_id: str, session) -> dict:
         events = []
+        depth = int(get_graph_depth())  # R-5 fix -- see _layer_3_patterns
 
         # Timeline events from multiple sources
         rows = session.run(
-            """
-            MATCH (n {id:$id})-[:DIRECTOR_OF|WON_CONTRACT|FLAGS|MENTIONED_IN|AUDITS*1..2]-(m)
+            f"""
+            MATCH (n {{id:$id}})-[:DIRECTOR_OF|WON_CONTRACT|FLAGS|MENTIONED_IN|AUDITS*1..{depth}]-(m)
             WHERE m.order_date IS NOT NULL OR m.year IS NOT NULL OR m.scraped_at IS NOT NULL
             RETURN coalesce(m.name,m.title,m.id) AS name,
                    labels(m)[0] AS type,

ai/investigators/historical_investigator.py

@@ -3,6 +3,7 @@ import sys
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
 from datetime import datetime
 from loguru import logger
+from config.model_selector import get_graph_depth
 
 
 class HistoricalInvestigator:
@@ -28,9 +29,13 @@ class HistoricalInvestigator:
 
             entity_type = node_row["type"] if node_row else "Unknown"
 
+            # R-5 fix: was a hardcoded *1..2, which is exactly PROFILE["low"]'s
+            # value -- zero change for a low-tier/desktop install, deeper
+            # (not shallower) traversal for medium/high.
+            depth = int(get_graph_depth())
             contract_rows = session.run(
-                """
-                MATCH (n {id: $id})-[:DIRECTOR_OF|WON_CONTRACT*1..2]->(ct:Contract)
+                f"""
+                MATCH (n {{id: $id}})-[:DIRECTOR_OF|WON_CONTRACT*1..{depth}]->(ct:Contract)
                 RETURN ct.order_date AS date, ct.amount_crore AS amount,
                        ct.buyer_org AS buyer
                 ORDER BY ct.order_date

api/routes/risk.py

@@ -11,6 +11,7 @@ from ai.benfords_analyzer import BenfordsAnalyzer
 from ai.ghost_company    import GhostCompanyDetector
 from ai.shadow_director  import ShadowDirectorDetector
 from ai.explainer        import generate_explanation
+from config.model_selector import get_graph_depth
 
 router = APIRouter()
 
@@ -204,8 +205,13 @@ def get_risk(entity_id: str, driver=Depends(get_db)):
 
         # Phase 34: ghost company detection
         try:
+            # R-5 fix: was a hardcoded *1..2, matching PROFILE["low"]'s value
+            # exactly -- zero change for low tier, deeper traversal for
+            # medium/high (see ai/deep_investigator.py's _layer_3_patterns
+            # for the full reasoning, identical here).
+            depth = int(get_graph_depth())
             co_rows = session.run(
-                "MATCH (co:Company)-[:DIRECTOR_OF|:LINKED_TO*1..2]-(n {id: $id})"
+                f"MATCH (co:Company)-[:DIRECTOR_OF|:LINKED_TO*1..{depth}]-(n {{id: $id}})"
                 " OPTIONAL MATCH (co)-[:WON_CONTRACT]->(ct:Contract)"
                 " WITH co, min(ct.order_date) AS first_contract,"
                 "      count(ct) AS contract_count,"