Update pipelines/requirements_pipe.py

pipelines/requirements_pipe.py

@@ -43,25 +43,22 @@ class RAGModel:
         docs = self.retriever.search(query)
         # Replace with your own synthesis and compliance assessment.
         requirements = self.llm.summarize(query=query, docs=docs)
-        compliant = self._assess_compliance(requirements, docs)
+        compliant = self._assess_compliance(requirements)
         return {
             "requirements": requirements,
             "compliant": compliant,
         }
 
-    def _assess_compliance(self, requirements: str, docs: list[str]) -> bool:
-        context = "\n\n".join(docs)
-    
-        prompt = (
-            "You are a compliance evaluator. Based on the retrieved documents and the "
-            "requirements summary, classify compliance.\n\n"
-            "Return exactly one word: COMPLIANT or NONCOMPLIANT.\n\n"
-            f"Retrieved Context:\n{context}\n\n"
-            f"Requirements Summary:\n{requirements}"
-        )
-    
-        decision = self.llm.generate(query=prompt, docs=docs).strip().upper()
-        return decision == "COMPLIANT"
+    def _assess_compliance(self, requirements: str) -> bool:
+        text = requirements.lower()
+        non_compliant_markers = [
+            "gap",
+            "missing",
+            "non-compliant",
+            "not compliant",
+            "fail",
+        ]
+        return not any(marker in text for marker in non_compliant_markers)
 
 
 class Router: