Update MeatValidatorAgent.py

MeatValidatorAgent.py

@@ -0,0 +1,66 @@
+
+class MEATValidatorAgent:
+    def __init__(self, model: str = "gpt-4o"):
+        self.llm = ChatOpenAI(model=model, temperature=0)
+
+        self.agent = Agent(
+            role="MEAT Criteria Validator",
+            goal="Determine clinical status and validate MEAT criteria for confirmed diagnoses.",
+            backstory="You are a compliance and risk adjustment specialist who classifies diagnoses "
+                      "into clinical status categories and validates MEAT criteria from chart documentation.",
+            verbose=True,
+            memory=False,
+            llm=self.llm,
+        )
+
+    def validate_one(self, diagnosis_entry: dict) -> dict:
+        """Validate MEAT criteria for a diagnosis that is confirmed (answer=yes)."""
+        diagnosis = diagnosis_entry["diagnosis"]
+        icd10 = diagnosis_entry["icd10"]
+        context = diagnosis_entry["context"]
+
+        task = Task(
+            description=(
+                f"Diagnosis: {diagnosis} (ICD-10: {icd10})\n\n"
+                f"Patient chart excerpts:\n{context}\n\n"
+                "Determine:\n"
+                "A. Clinical Status: one of {ACTIVE, MONITORING, HISTORICAL}.\n"
+                "- ACTIVE: currently under treatment, ongoing management, or acute care\n"
+                "- MONITORING: surveillance visits, follow-up, observation\n"
+                "- HISTORICAL: past condition, no current clinical impact\n\n"
+                "B. MEAT Criteria Validation (Y/N for each):\n"
+                "- MONITOR: Are signs/symptoms being tracked?\n"
+                "- EVALUATE: Are test results or treatment responses reviewed?\n"
+                "- ASSESS: Did provider order tests or make clinical assessments?\n"
+                "- TREAT: Are therapeutic interventions documented?\n\n"
+                "Output must be strict JSON:\n"
+                "{'clinical_status': 'ACTIVE/MONITORING/HISTORICAL', "
+                "'meat': {'monitor': true/false, 'evaluate': true/false, 'assess': true/false, 'treat': true/false}, "
+                "'rationale': 'short justification'}"
+            ),
+            expected_output="JSON with keys clinical_status, meat, rationale",
+            agent=self.agent,
+            json_mode=True,
+        )
+
+        crew = Crew(agents=[self.agent], tasks=[task], process=Process.sequential, verbose=True)
+        result = crew.kickoff()
+        result = json.loads(repair_json(result))
+
+        return {
+            **diagnosis_entry,
+            "clinical_status": result.get("clinical_status"),
+            "meat": result.get("meat", {}),
+            "meat_rationale": result.get("rationale", "")
+        }
+
+    def run(self, confirmed_diagnoses: list[dict]) -> list[dict]:
+        """Loop through all confirmed (yes) diagnoses and validate MEAT criteria."""
+        enriched_results = []
+        for entry in confirmed_diagnoses:
+            if entry.get("answer", "").lower() == "yes":
+                print(f"\n[INFO] Validating MEAT for: {entry['diagnosis']} ({entry['icd10']})")
+                enriched = self.validate_one(entry)
+                enriched_results.append(enriched)
+                print(f"[MEAT RESULT] {enriched}")
+        return enriched_results