Add production condition retrieval smoke test

apps/ai-service/rag/instrumentation.py

@@ -162,6 +162,34 @@ class InstrumentedRetrievalService(RetrievalService):
             self._annotate_result(result)
             return result
 
+    def assess_patient_candidates(self, *args, **kwargs):
+        with stage("retrieval"):
+            self._observability_metrics.increment(
+                RETRIEVAL_ROUTE, route="patient_safety"
+            )
+            try:
+                result, assessments = super().assess_patient_candidates(
+                    *args, **kwargs
+                )
+            except Exception as exc:
+                self._record_retrieval_failure(exc)
+                raise
+            self._annotate_result(result)
+            return result, assessments
+
+    def retrieve_patient_drug_context(self, *args, **kwargs):
+        with stage("retrieval"):
+            self._observability_metrics.increment(
+                RETRIEVAL_ROUTE, route="patient_drug_safety"
+            )
+            try:
+                result = super().retrieve_patient_drug_context(*args, **kwargs)
+            except Exception as exc:
+                self._record_retrieval_failure(exc)
+                raise
+            self._annotate_result(result)
+            return result
+
     def _record_retrieval_failure(self, exc: BaseException) -> None:
         self._observability_metrics.increment(
             PROVIDER_FAILURE,