abinazebinoy commited on
Commit
529f177
·
1 Parent(s): 4a1062b

fix(BUG-C1): add MathInvestigator class wrapper -- all 15 investigators now importable

Browse files
ai/investigators/math_investigator.py CHANGED
@@ -109,3 +109,16 @@ def investigate(entity_id: str, entity_name: str,
109
  "evidence": evidence,
110
  "investigated_at": datetime.now().isoformat(),
111
  }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
109
  "evidence": evidence,
110
  "investigated_at": datetime.now().isoformat(),
111
  }
112
+
113
+
114
+ # BUG-C1 FIX: class wrapper -- MathematicalInvestigator is the display name
115
+ # but the import expects MathInvestigator to match the __init__.py import
116
+ class MathInvestigator:
117
+ NAME = NAME
118
+ FOCUS = FOCUS
119
+ WEIGHT = WEIGHT
120
+
121
+ def investigate(self, entity_id: str, entity_name: str,
122
+ session=None, driver=None) -> dict:
123
+ return investigate(entity_id, entity_name,
124
+ session=session, driver=driver)